mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
53557e38b6
- Incorrect name being shown on dashboard "Proxies page" - Apps being shown when homepage.show is false - Load balanced routes are shown on homepage instead of the load balancer - Route with idlewatcher will now be removed on container destroy - Idlewatcher panic - Performance improvement - Idlewatcher infinitely loading - Reload stucked / not working properly - Streams stuck on shutdown / reload - etc... Added: - support idlewatcher for loadbalanced routes - partial implementation for stream type idlewatcher Issues: - graceful shutdown
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package v1
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/coder/websocket"
|
|
"github.com/coder/websocket/wsjson"
|
|
U "github.com/yusing/go-proxy/internal/api/v1/utils"
|
|
"github.com/yusing/go-proxy/internal/common"
|
|
"github.com/yusing/go-proxy/internal/config"
|
|
"github.com/yusing/go-proxy/internal/server"
|
|
"github.com/yusing/go-proxy/internal/utils"
|
|
)
|
|
|
|
func Stats(w http.ResponseWriter, r *http.Request) {
|
|
U.RespondJSON(w, r, getStats())
|
|
}
|
|
|
|
func StatsWS(w http.ResponseWriter, r *http.Request) {
|
|
localAddresses := []string{"127.0.0.1", "10.0.*.*", "172.16.*.*", "192.168.*.*"}
|
|
originPats := make([]string, len(config.Value().MatchDomains)+len(localAddresses))
|
|
|
|
if len(originPats) == 0 {
|
|
U.Logger.Warnf("no match domains configured, accepting websocket request from all origins")
|
|
originPats = []string{"*"}
|
|
} else {
|
|
for i, domain := range config.Value().MatchDomains {
|
|
originPats[i] = "*." + domain
|
|
}
|
|
originPats = append(originPats, localAddresses...)
|
|
}
|
|
if common.IsDebug {
|
|
originPats = []string{"*"}
|
|
}
|
|
conn, err := websocket.Accept(w, r, &websocket.AcceptOptions{
|
|
OriginPatterns: originPats,
|
|
})
|
|
if err != nil {
|
|
U.Logger.Errorf("/stats/ws failed to upgrade websocket: %s", err)
|
|
return
|
|
}
|
|
/* trunk-ignore(golangci-lint/errcheck) */
|
|
defer conn.CloseNow()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for range ticker.C {
|
|
stats := getStats()
|
|
if err := wsjson.Write(ctx, conn, stats); err != nil {
|
|
U.Logger.Errorf("/stats/ws failed to write JSON: %s", err)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func getStats() map[string]any {
|
|
return map[string]any{
|
|
"proxies": config.Statistics(),
|
|
"uptime": utils.FormatDuration(server.GetProxyServer().Uptime()),
|
|
}
|
|
}
|