feat: Server could disabled

innoai-tech · Feb 29, 2024 · 9f36a4c · 9f36a4c
1 parent 1c81c65
commit 9f36a4c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/configuration/lifecycle.go b/pkg/configuration/lifecycle.go
@@ -83,6 +83,12 @@ func serve(ctx context.Context, stopCh chan os.Signal, configuratorServers ...Se
 	for i := range configuratorServers {
 		server := configuratorServers[i]
 
+		if d, ok := server.(CanDisabled); ok {
+			if d.Disabled(ctx) {
+				continue
+			}
+		}
+
 		g.Go(func() error {
 			err := server.Serve(c)
 			go func() {
@@ -158,3 +164,7 @@ type Server interface {
 type CanShutdown interface {
 	Shutdown(ctx context.Context) error
 }
+
+type CanDisabled interface {
+	Disabled(ctx context.Context) bool
+}