yanistyle/sing-box-extended
1
0
Fork 0
mirror of https://github.com/shtorm-7/sing-box-extended.git synced 2026-06-19 01:14:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

clash-api: Fix websocket connection not closed after config reload via SIGHUP

Browse Source 
Co-authored-by: TraitMan <traitman@maildog.top>
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
traitman
2026-02-28 18:13:53 +08:00
committed by 世界
parent 28c1f2c51e
commit 46453be6dd
2 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
experimental/clashapi/connections.go
View File

@@ -2,6 +2,7 @@ package clashapi
import ( import (
"bytes" "bytes"
"context"
"net/http" "net/http"
"strconv" "strconv"
"time" "time"
@@ -17,15 +18,15 @@ import (
"github.com/gofrs/uuid/v5" "github.com/gofrs/uuid/v5"
) )
func connectionRouter(router adapter.Router, trafficManager *trafficontrol.Manager) http.Handler { func connectionRouter(ctx context.Context, router adapter.Router, trafficManager *trafficontrol.Manager) http.Handler {
r := chi.NewRouter() r := chi.NewRouter()
r.Get("/", getConnections(trafficManager)) r.Get("/", getConnections(ctx, trafficManager))
r.Delete("/", closeAllConnections(router, trafficManager)) r.Delete("/", closeAllConnections(router, trafficManager))
r.Delete("/{id}", closeConnection(trafficManager)) r.Delete("/{id}", closeConnection(trafficManager))
return r return r
} }
func getConnections(trafficManager *trafficontrol.Manager) func(w http.ResponseWriter, r *http.Request) { func getConnections(ctx context.Context, trafficManager *trafficontrol.Manager) func(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Upgrade") != "websocket" { if r.Header.Get("Upgrade") != "websocket" {
snapshot := trafficManager.Snapshot() snapshot := trafficManager.Snapshot()
@@ -67,7 +68,12 @@ func getConnections(trafficManager *trafficontrol.Manager) func(w http.ResponseW
tick := time.NewTicker(time.Millisecond * time.Duration(interval)) tick := time.NewTicker(time.Millisecond * time.Duration(interval))
defer tick.Stop() defer tick.Stop()
for range tick.C { for {
select {
case <-ctx.Done():
return
case <-tick.C:
}
if err = sendSnapshot(); err != nil { if err = sendSnapshot(); err != nil {
break break
} }

13
experimental/clashapi/server.go
View File

@@ -116,12 +116,12 @@ func NewServer(ctx context.Context, logFactory log.ObservableFactory, options op
r.Use(authentication(options.Secret)) r.Use(authentication(options.Secret))
r.Get("/", hello(options.ExternalUI != "")) r.Get("/", hello(options.ExternalUI != ""))
r.Get("/logs", getLogs(logFactory)) r.Get("/logs", getLogs(logFactory))
r.Get("/traffic", traffic(trafficManager)) r.Get("/traffic", traffic(s.ctx, trafficManager))
r.Get("/version", version) r.Get("/version", version)
r.Mount("/configs", configRouter(s, logFactory)) r.Mount("/configs", configRouter(s, logFactory))
r.Mount("/proxies", proxyRouter(s, s.router)) r.Mount("/proxies", proxyRouter(s, s.router))
r.Mount("/rules", ruleRouter(s.router)) r.Mount("/rules", ruleRouter(s.router))
r.Mount("/connections", connectionRouter(s.router, trafficManager)) r.Mount("/connections", connectionRouter(s.ctx, s.router, trafficManager))
r.Mount("/providers/proxies", proxyProviderRouter()) r.Mount("/providers/proxies", proxyProviderRouter())
r.Mount("/providers/rules", ruleProviderRouter()) r.Mount("/providers/rules", ruleProviderRouter())
r.Mount("/script", scriptRouter()) r.Mount("/script", scriptRouter())
@@ -303,7 +303,7 @@ type Traffic struct {
Down int64 `json:"down"` Down int64 `json:"down"`
} }
func traffic(trafficManager *trafficontrol.Manager) func(w http.ResponseWriter, r *http.Request) { func traffic(ctx context.Context, trafficManager *trafficontrol.Manager) func(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) { return func(w http.ResponseWriter, r *http.Request) {
var conn net.Conn var conn net.Conn
if r.Header.Get("Upgrade") == "websocket" { if r.Header.Get("Upgrade") == "websocket" {
@@ -324,7 +324,12 @@ func traffic(trafficManager *trafficontrol.Manager) func(w http.ResponseWriter,
defer tick.Stop() defer tick.Stop()
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
uploadTotal, downloadTotal := trafficManager.Total() uploadTotal, downloadTotal := trafficManager.Total()
for range tick.C { for {
select {
case <-ctx.Done():
return
case <-tick.C:
}
buf.Reset() buf.Reset()
uploadTotalNew, downloadTotalNew := trafficManager.Total() uploadTotalNew, downloadTotalNew := trafficManager.Total()
err := json.NewEncoder(buf).Encode(Traffic{ err := json.NewEncoder(buf).Encode(Traffic{