mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
1.9 KiB
102 lines
1.9 KiB
package stats
|
|
|
|
//go:generate errorgen
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"v2ray.com/core"
|
|
"v2ray.com/core/features/stats"
|
|
)
|
|
|
|
// Counter is an implementation of stats.Counter.
|
|
type Counter struct {
|
|
value int64
|
|
}
|
|
|
|
// Value implements stats.Counter.
|
|
func (c *Counter) Value() int64 {
|
|
return atomic.LoadInt64(&c.value)
|
|
}
|
|
|
|
// Set implements stats.Counter.
|
|
func (c *Counter) Set(newValue int64) int64 {
|
|
return atomic.SwapInt64(&c.value, newValue)
|
|
}
|
|
|
|
// Add implements stats.Counter.
|
|
func (c *Counter) Add(delta int64) int64 {
|
|
return atomic.AddInt64(&c.value, delta)
|
|
}
|
|
|
|
// Manager is an implementation of stats.Manager.
|
|
type Manager struct {
|
|
access sync.RWMutex
|
|
counters map[string]*Counter
|
|
}
|
|
|
|
func NewManager(ctx context.Context, config *Config) (*Manager, error) {
|
|
m := &Manager{
|
|
counters: make(map[string]*Counter),
|
|
}
|
|
|
|
v := core.FromContext(ctx)
|
|
if v != nil {
|
|
if err := v.RegisterFeature(m); err != nil {
|
|
return nil, newError("failed to register StatManager").Base(err)
|
|
}
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
func (*Manager) Type() interface{} {
|
|
return stats.ManagerType()
|
|
}
|
|
|
|
func (m *Manager) RegisterCounter(name string) (stats.Counter, error) {
|
|
m.access.Lock()
|
|
defer m.access.Unlock()
|
|
|
|
if _, found := m.counters[name]; found {
|
|
return nil, newError("Counter ", name, " already registered.")
|
|
}
|
|
newError("create new counter ", name).AtDebug().WriteToLog()
|
|
c := new(Counter)
|
|
m.counters[name] = c
|
|
return c, nil
|
|
}
|
|
|
|
func (m *Manager) GetCounter(name string) stats.Counter {
|
|
m.access.RLock()
|
|
defer m.access.RUnlock()
|
|
|
|
if c, found := m.counters[name]; found {
|
|
return c
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *Manager) Visit(visitor func(string, stats.Counter) bool) {
|
|
m.access.RLock()
|
|
defer m.access.RUnlock()
|
|
|
|
for name, c := range m.counters {
|
|
if !visitor(name, c) {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// Start implements common.Runnable.
|
|
func (m *Manager) Start() error {
|
|
return nil
|
|
}
|
|
|
|
// Close implement common.Closable.
|
|
func (m *Manager) Close() error {
|
|
return nil
|
|
}
|