Fijxu/http3-ytproxy
1
0
Fork 0
Code Issues 4 Pull requests Projects Releases Packages Wiki Activity Actions

feat: WIP stat to calculate the rate of 403 videoplayback requests
All checks were successful
CI / build (push) Successful in 51s
Details

Browse source
This commit is contained in:
Fijxu 2025-02-11 23:45:21 -03:00
parent f66ae3187f
commit 53d04ac79b
Signed by: Fijxu
GPG key ID: 32C1DDF333EDA6A4
1 changed files with 40 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
main.go
View file

@ -156,16 +156,19 @@ func (cw *ConnectionWatcher) OnStateChange(conn net.Conn, state http.ConnState)
var cw ConnectionWatcher var cw ConnectionWatcher
type statusJson struct { type statusJson struct {
Version string `json:"version"` Version string `json:"version"`
Uptime time.Duration `json:"uptime"` Uptime time.Duration `json:"uptime"`
RequestCount int64 `json:"requestCount"` RequestCount int64 `json:"requestCount"`
RequestPerSecond int64 `json:"requestPerSecond"` RequestPerSecond int64 `json:"requestPerSecond"`
RequestPerMinute int64 `json:"requestPerMinute"` RequestPerMinute int64 `json:"requestPerMinute"`
TotalConnEstablished int64 `json:"totalEstablished"` TotalConnEstablished int64 `json:"totalEstablished"`
EstablishedConnections int64 `json:"establishedConnections"` EstablishedConnections int64 `json:"establishedConnections"`
ActiveConnections int64 `json:"activeConnections"` ActiveConnections int64 `json:"activeConnections"`
IdleConnections int64 `json:"idleConnections"` IdleConnections int64 `json:"idleConnections"`
RequestsForbidden struct { RequestsForbiddenPerSec struct {
Videoplayback int64 `json:"videoplayback"`
}
RequestsForbidden struct {
Videoplayback int64 `json:"videoplayback"` Videoplayback int64 `json:"videoplayback"`
Vi int64 `json:"vi"` Vi int64 `json:"vi"`
Ggpht int64 `json:"ggpht"` Ggpht int64 `json:"ggpht"`
@ -182,6 +185,11 @@ var stats_ = statusJson{
EstablishedConnections: 0, EstablishedConnections: 0,
ActiveConnections: 0, ActiveConnections: 0,
IdleConnections: 0, IdleConnections: 0,
RequestsForbiddenPerSec: struct {
Videoplayback int64 `json:"videoplayback"`
}{
Videoplayback: 0,
},
RequestsForbidden: struct { RequestsForbidden: struct {
Videoplayback int64 `json:"videoplayback"` Videoplayback int64 `json:"videoplayback"`
Vi int64 `json:"vi"` Vi int64 `json:"vi"`
@ -317,6 +325,27 @@ func requestPerMinute() {
} }
} }
func forbiddenRequestsPerSec() {
var last int64
for {
time.Sleep(1 * time.Second)
current := stats_.RequestsForbidden.Videoplayback
stats_.RequestsForbiddenPerSec.Videoplayback = current - last
// fmt.Println(aux)
// if aux > 15 {
// body := "{\"status\":\"stopped\"}\""
// request, err := http.NewRequest("PUT", "http://gluetun:8000", strings.NewReader(body))
// if err != nil {
// log.Panic(err)
// }
// client.Do(request)
// }
// metrics.RequestPerSecond.Set(float64(stats_.RequestPerSecond))
// time.Sleep(60 * time.Second)
last = current
}
}
func beforeMisc(next http.HandlerFunc) http.HandlerFunc { func beforeMisc(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, req *http.Request) { return func(w http.ResponseWriter, req *http.Request) {
defer panicHandler(w) defer panicHandler(w)
@ -479,6 +508,7 @@ func main() {
go requestPerSecond() go requestPerSecond()
go requestPerMinute() go requestPerMinute()
go forbiddenRequestsPerSec()
ln, err := net.Listen("tcp", host+":"+port) ln, err := net.Listen("tcp", host+":"+port)
if err != nil { if err != nil {