mirror of
https://github.com/wyx2685/V2bX.git
synced 2025-01-23 10:28:13 -05:00
27 lines
481 B
Go
27 lines
481 B
Go
|
package hy2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/InazumaV/V2bX/common/counter"
|
||
|
|
)
|
||
|
|
|
||
|
|
type HookServer struct {
|
||
|
|
Tag string
|
||
|
|
Counter sync.Map
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *HookServer) Log(id string, tx, rx uint64) (ok bool) {
|
||
|
|
if c, ok := h.Counter.Load(h.Tag); ok {
|
||
|
|
c.(*counter.TrafficCounter).Rx(id, int(rx))
|
||
|
|
c.(*counter.TrafficCounter).Tx(id, int(rx))
|
||
|
|
return true
|
||
|
|
} else {
|
||
|
|
c := counter.NewTrafficCounter()
|
||
|
|
h.Counter.Store(h.Tag, c)
|
||
|
|
c.Rx(id, int(rx))
|
||
|
|
c.Tx(id, int(rx))
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|