mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-08-10 05:10:58 +00:00
fix(xray): synchronize lifecycle state (#6138)
* fix(xray): synchronize lifecycle snapshots Protect process replacement and result caching with a lifecycle state object, so read paths keep one process snapshot while restarts swap state safely. Bound version probing to prevent a stalled binary from holding the restart lock. * test(xray): cover concurrent lifecycle reads Exercise status, result, and traffic reads while the managed process is replaced, so the race detector guards the lifecycle snapshot boundary. * fix(xray): guard process config snapshots Synchronize hot-applied config snapshots, keep Telegram reads on one lifecycle snapshot, and strengthen lifecycle timeout and concurrency regression coverage. --------- Co-authored-by: PathGao <gaoyanbo@gaoyanbodeMacBook-Air.local>
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
package xray
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestRefreshVersionTimesOut(t *testing.T) {
|
||||
if runtime.GOOS == "windows" {
|
||||
t.Skip("shell fixture is Unix-only")
|
||||
}
|
||||
|
||||
dir := t.TempDir()
|
||||
t.Setenv("XUI_BIN_FOLDER", dir)
|
||||
binaryPath := filepath.Join(dir, GetBinaryName())
|
||||
if err := os.WriteFile(binaryPath, []byte("#!/bin/sh\nexec sleep 1\n"), 0o700); err != nil {
|
||||
t.Fatalf("write xray fixture: %v", err)
|
||||
}
|
||||
|
||||
previousTimeout := xrayVersionTimeout
|
||||
xrayVersionTimeout = 20 * time.Millisecond
|
||||
t.Cleanup(func() { xrayVersionTimeout = previousTimeout })
|
||||
|
||||
p := newProcess(&Config{})
|
||||
started := time.Now()
|
||||
p.refreshVersion()
|
||||
elapsed := time.Since(started)
|
||||
if elapsed < xrayVersionTimeout {
|
||||
t.Fatalf("refreshVersion duration = %s, want at least %s", elapsed, xrayVersionTimeout)
|
||||
}
|
||||
if elapsed > 500*time.Millisecond {
|
||||
t.Fatalf("refreshVersion duration = %s, want under 500ms", elapsed)
|
||||
}
|
||||
if got := p.GetXrayVersion(); got != "Unknown" {
|
||||
t.Fatalf("version = %q, want Unknown", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestProcessConfigSnapshotsAreRaceSafe(t *testing.T) {
|
||||
p := NewProcess(&Config{})
|
||||
first := &Config{}
|
||||
second := &Config{}
|
||||
var wg sync.WaitGroup
|
||||
|
||||
wg.Go(func() {
|
||||
for range 1000 {
|
||||
p.SetConfig(first)
|
||||
p.SetConfig(second)
|
||||
}
|
||||
})
|
||||
for range 4 {
|
||||
wg.Go(func() {
|
||||
for range 1000 {
|
||||
if p.GetConfig() == nil {
|
||||
t.Error("config = nil")
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
Reference in New Issue
Block a user