init

go-gost/x/handler/forward/local/traffic_bridge.go

@@ -0,0 +1,24 @@
+package local
+
+import (
+	"sync"
+)
+
+var (
+	globalTrafficRecorder TrafficRecorder
+	trafficMutex          sync.RWMutex
+)
+
+// SetGlobalTrafficRecorder 设置全局流量记录器
+func SetGlobalTrafficRecorder(recorder TrafficRecorder) {
+	trafficMutex.Lock()
+	defer trafficMutex.Unlock()
+	globalTrafficRecorder = recorder
+}
+
+// GetGlobalTrafficRecorder 获取全局流量记录器
+func GetGlobalTrafficRecorder() TrafficRecorder {
+	trafficMutex.RLock()
+	defer trafficMutex.RUnlock()
+	return globalTrafficRecorder
+}