mirror of
https://github.com/Sagit-chu/flvx.git
synced 2026-10-02 08:56:38 +08:00
qaq
25 lines
510 B
Go
25 lines
510 B
Go
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
|
|
}
|