add registry info delete

7e18deea · luxq · 8851b53e · 7e18deea
Commit 7e18deea authored Mar 12, 2024 by luxq
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

registry.go registry/registry.go +9 -0

No files found.
--- a/registry/registry.go
+++ b/registry/registry.go
@@ -94,6 +94,15 @@ func (s *Registry) Start() {
 	}
 }

+func (s *Registry) Clear() {
+	instance := os.Getenv(REGISTRY_SERVICE_INSTANCE)
+	for _, opt := range s.opts {
+		instance = opt()
+	}
+	k := fmt.Sprintf("%s%s", common.GetServiceKeyPrefix(s.register.ServiceType()), instance)
+	s.rdb.Del(context.Background(), k)
+}
+
 func (s *Registry) Stop() {
 	close(s.quit)
 }