1 files changed, 30 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/agent/functions/ext.go b/src/go/collectors/go.d.plugin/agent/functions/ext.go
new file mode 100644
index 000000000..28c717d88
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/agent/functions/ext.go
@@ -0,0 +1,30 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package functions
+
+func (m *Manager) Register(name string, fn func(Function)) {
+	if fn == nil {
+		m.Warningf("not registering '%s': nil function", name)
+		return
+	}
+
+	m.mux.Lock()
+	defer m.mux.Unlock()
+
+	if _, ok := m.FunctionRegistry[name]; !ok {
+		m.Debugf("registering function '%s'", name)
+	} else {
+		m.Warningf("re-registering function '%s'", name)
+	}
+	m.FunctionRegistry[name] = fn
+}
+
+func (m *Manager) Unregister(name string) {
+	m.mux.Lock()
+	defer m.mux.Unlock()
+
+	if _, ok := m.FunctionRegistry[name]; !ok {
+		delete(m.FunctionRegistry, name)
+		m.Debugf("unregistering function '%s'", name)
+	}
+}