TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-15 00:30:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

进程退出时停止上传带宽数据

Browse Source
This commit is contained in:
GoEdgeLab
2023-04-05 11:34:15 +08:00
parent d13018c51f
commit 31d1fc229c
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/stats/bandwidth_stat_manager.go
View File

@@ -36,6 +36,8 @@ func init() {
}) })
events.On(events.EventQuit, func() { events.On(events.EventQuit, func() {
SharedBandwidthStatManager.Cancel()
err := SharedBandwidthStatManager.Save() err := SharedBandwidthStatManager.Save()
if err != nil { if err != nil {
remotelogs.Error("STAT", "save bandwidth stats failed: "+err.Error()) remotelogs.Error("STAT", "save bandwidth stats failed: "+err.Error())
@@ -277,6 +279,11 @@ func (this *BandwidthStatManager) Save() error {
return os.WriteFile(this.cacheFile, data, 0666) return os.WriteFile(this.cacheFile, data, 0666)
} }
// Cancel 取消上传
func (this *BandwidthStatManager) Cancel() {
this.ticker.Stop()
}
// 从本地缓存文件中恢复数据 // 从本地缓存文件中恢复数据
func (this *BandwidthStatManager) recover() { func (this *BandwidthStatManager) recover() {
cacheData, err := os.ReadFile(this.cacheFile) cacheData, err := os.ReadFile(this.cacheFile)