Fixes the internal state update (#17)

Author: theely

src/internal/protocol/node_table.go

@@ -106,6 +106,7 @@ func MarkSelfAsBootstrap() {
 			Connected:     true,
 		}
 		value, err := json.Marshal(peer)
+		UpdateNodeTableHook(key, value)
 		common.ReportError(err, "Error while marshalling peer")
 		if err := store.Put(ctx, key, value); err != nil {
 			common.Logger.Error("Error while registering bootstrap: ", err)

src/internal/protocol/registrar.go

@@ -133,6 +133,7 @@ func provideService(service Service) {
 	}
 	common.Logger.Info("Registering LLM service: ", myself)
 	value, err := json.Marshal(myself)
+	UpdateNodeTableHook(key, value)
 	common.ReportError(err, "Error while marshalling peer")
 	err = store.Put(ctx, key, value)
 	if err != nil {
@@ -157,6 +158,7 @@ func ReannounceLocalServices() {
 		common.Logger.Error("Error marshalling self during reannounce: ", err)
 		return
 	}
+	UpdateNodeTableHook(key, value)
 	if err := store.Put(ctx, key, value); err != nil {
 		common.Logger.Warn("Failed to reannounce local services: ", err)
 	} else {