mirror of
https://github.com/SagerNet/sing-box.git
synced 2026-04-13 20:28:32 +10:00
76 lines
2.6 KiB
Go
76 lines
2.6 KiB
Go
package libbox
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/sagernet/sing-box/common/networkquality"
|
|
)
|
|
|
|
type NetworkQualityTest struct {
|
|
ctx context.Context
|
|
cancel context.CancelFunc
|
|
}
|
|
|
|
func NewNetworkQualityTest() *NetworkQualityTest {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
return &NetworkQualityTest{ctx: ctx, cancel: cancel}
|
|
}
|
|
|
|
func (t *NetworkQualityTest) Start(configURL string, handler NetworkQualityTestHandler) {
|
|
t.StartWithSerialAndRuntime(configURL, false, NetworkQualityDefaultMaxRuntimeSeconds, handler)
|
|
}
|
|
|
|
func (t *NetworkQualityTest) StartWithSerial(configURL string, serial bool, handler NetworkQualityTestHandler) {
|
|
t.StartWithSerialAndRuntime(configURL, serial, NetworkQualityDefaultMaxRuntimeSeconds, handler)
|
|
}
|
|
|
|
func (t *NetworkQualityTest) StartWithSerialAndRuntime(configURL string, serial bool, maxRuntimeSeconds int32, handler NetworkQualityTestHandler) {
|
|
go func() {
|
|
httpClient := networkquality.NewHTTPClient(nil)
|
|
defer httpClient.CloseIdleConnections()
|
|
|
|
result, err := networkquality.Run(networkquality.Options{
|
|
ConfigURL: configURL,
|
|
HTTPClient: httpClient,
|
|
Serial: serial,
|
|
MaxRuntime: time.Duration(maxRuntimeSeconds) * time.Second,
|
|
Context: t.ctx,
|
|
OnProgress: func(p networkquality.Progress) {
|
|
handler.OnProgress(&NetworkQualityProgress{
|
|
Phase: int32(p.Phase),
|
|
DownloadCapacity: p.DownloadCapacity,
|
|
UploadCapacity: p.UploadCapacity,
|
|
DownloadRPM: p.DownloadRPM,
|
|
UploadRPM: p.UploadRPM,
|
|
IdleLatencyMs: p.IdleLatencyMs,
|
|
ElapsedMs: p.ElapsedMs,
|
|
DownloadCapacityAccuracy: int32(p.DownloadCapacityAccuracy),
|
|
UploadCapacityAccuracy: int32(p.UploadCapacityAccuracy),
|
|
DownloadRPMAccuracy: int32(p.DownloadRPMAccuracy),
|
|
UploadRPMAccuracy: int32(p.UploadRPMAccuracy),
|
|
})
|
|
},
|
|
})
|
|
if err != nil {
|
|
handler.OnError(err.Error())
|
|
return
|
|
}
|
|
handler.OnResult(&NetworkQualityResult{
|
|
DownloadCapacity: result.DownloadCapacity,
|
|
UploadCapacity: result.UploadCapacity,
|
|
DownloadRPM: result.DownloadRPM,
|
|
UploadRPM: result.UploadRPM,
|
|
IdleLatencyMs: result.IdleLatencyMs,
|
|
DownloadCapacityAccuracy: int32(result.DownloadCapacityAccuracy),
|
|
UploadCapacityAccuracy: int32(result.UploadCapacityAccuracy),
|
|
DownloadRPMAccuracy: int32(result.DownloadRPMAccuracy),
|
|
UploadRPMAccuracy: int32(result.UploadRPMAccuracy),
|
|
})
|
|
}()
|
|
}
|
|
|
|
func (t *NetworkQualityTest) Cancel() {
|
|
t.cancel()
|
|
}
|