Merge pull request #2 from mailgun/maxim/develop

Cancel prev state health check on config update (merge from v3)
mailgun · Oct 11, 2023 · ac2cd7a · ac2cd7a
2 parents a48500e + 580b1cf
commit ac2cd7a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/server/routerfactory.go b/pkg/server/routerfactory.go
@@ -31,6 +31,8 @@ type RouterFactory struct {
 
 	chainBuilder *middleware.ChainBuilder
 	tlsManager   *tls.Manager
+
+	cancelPrevState func()
 }
 
 // NewRouterFactory creates a new RouterFactory.
@@ -65,7 +67,12 @@ func NewRouterFactory(staticConfiguration static.Configuration, managerFactory *
 
 // CreateRouters creates new TCPRouters and UDPRouters.
 func (f *RouterFactory) CreateRouters(rtConf *runtime.Configuration) (map[string]*tcprouter.Router, map[string]udptypes.Handler) {
-	ctx := context.Background()
+	if f.cancelPrevState != nil {
+		f.cancelPrevState()
+	}
+
+	var ctx context.Context
+	ctx, f.cancelPrevState = context.WithCancel(context.Background())
 
 	// HTTP
 	serviceManager := f.managerFactory.Build(rtConf)