add UnregisterCounter method to stats manager

2026-10-31 08:14:37 +00:00 · 2020-02-03 13:53:14 +08:00
parent 5dd85717fc
commit 86d5bd866b
2 changed files with 18 additions and 0 deletions
@@ -63,6 +63,18 @@ func (m *Manager) RegisterCounter(name string) (stats.Counter, error) {
 	return c, nil
 }

+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
+}
+
 func (m *Manager) GetCounter(name string) stats.Counter {
 	m.access.RLock()
 	defer m.access.RUnlock()