manager: wire up tunnels changed notifier

1 files changed, 10 insertions, 24 deletions

diff --git a/conf/storewatcher.go b/conf/storewatcher.go
index 4f9c2ef7..7a5ab387 100644
--- a/conf/storewatcher.go
+++ b/conf/storewatcher.go
@@ -5,34 +5,20 @@
 
 package conf
 
-import "reflect"
-
-type StoreCallback func()
+type storeCallback struct {
+	cb func()
+}
 
-var storeCallbacks []StoreCallback
+var storeCallbacks = make(map[*storeCallback]bool)
 
-func RegisterStoreChangeCallback(cb StoreCallback) {
+func RegisterStoreChangeCallback(cb func()) *storeCallback {
 	startWatchingConfigDir()
 	cb()
-	storeCallbacks = append(storeCallbacks, cb)
+	s := &storeCallback{cb}
+	storeCallbacks[s] = true
+	return s
 }
 
-func UnregisterStoreChangeCallback(cb StoreCallback) {
-	//TODO: this function is ridiculous, doing slow iteration like this and reflection too.
-
-	index := -1
-	for i, e := range storeCallbacks {
-		if reflect.ValueOf(e).Pointer() == reflect.ValueOf(cb).Pointer() {
-			index = i
-			break
-		}
-	}
-	if index == -1 {
-		return
-	}
-	newList := storeCallbacks[0:index]
-	if index < len(storeCallbacks)-1 {
-		newList = append(newList, storeCallbacks[index+1:]...)
-	}
-	storeCallbacks = newList
+func UnregisterStoreChangeCallback(cb *storeCallback) {
+	delete(storeCallbacks, cb)
 }