Merge pull request #12243 from lukidzi/cleanup-context

fix(kds): remove context from map on stream close
kumahq · Dec 11, 2024 · f1fa968 · f1fa968
2 parents 2f495a5 + aeaf45f
commit f1fa968
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/kds/v2/server/tenant_callback.go b/pkg/kds/v2/server/tenant_callback.go
@@ -5,6 +5,7 @@ import (
 	"errors"
 	"sync"
 
+	envoy_core "github.com/envoyproxy/go-control-plane/envoy/config/core/v3"
 	envoy_sd "github.com/envoyproxy/go-control-plane/envoy/service/discovery/v3"
 	envoy_xds "github.com/envoyproxy/go-control-plane/pkg/server/v3"
 
@@ -50,3 +51,9 @@ func (c *tenancyCallbacks) OnStreamDeltaRequest(streamID int64, request *envoy_s
 	util.FillTenantMetadata(tenantID, request.Node)
 	return nil
 }
+
+func (c *tenancyCallbacks) OnDeltaStreamClosed(streamID int64, _ *envoy_core.Node) {
+	c.Lock()
+	delete(c.streamToCtx, streamID)
+	c.Unlock()
+}