diff --git a/app/stats/stats.go b/app/stats/stats.go index 7bb56b287..ce9e36920 100644 --- a/app/stats/stats.go +++ b/app/stats/stats.go @@ -126,6 +126,18 @@ func (m *Manager) RegisterCounter(name string) (stats.Counter, error) { return c, nil } +// UnregisterCounter implements stats.Manager. +func (m *Manager) UnregisterCounter(name string) error { + m.access.Lock() + defer m.access.Unlock() + if _, found := m.counters[name]; !found { + return newError("Counter ", name, " was not found.") + } + newError("remove counter ", name).AtDebug().WriteToLog() + delete(m.counters, name) + return nil +} + // GetCounter implements stats.Manager. func (m *Manager) GetCounter(name string) stats.Counter { m.access.RLock() diff --git a/features/stats/stats.go b/features/stats/stats.go index 2a0acec2b..6ac0da26a 100644 --- a/features/stats/stats.go +++ b/features/stats/stats.go @@ -38,6 +38,7 @@ type Manager interface { // RegisterCounter registers a new counter to the manager. The identifier string must not be empty, and unique among other counters. RegisterCounter(string) (Counter, error) + UnregisterCounter(string) error // GetCounter returns a counter by its identifier. GetCounter(string) Counter @@ -87,6 +88,11 @@ func (NoopManager) RegisterCounter(string) (Counter, error) { return nil, newError("not implemented") } +// UnregisterCounter implements Manager. +func (NoopManager) UnregisterCounter(string) error { + return newError("not implemented") +} + // GetCounter implements Manager. func (NoopManager) GetCounter(string) Counter { return nil