add newly created agent endpoints to the map

Signed-off-by: Alice-Lilith <[email protected]>
ngrok · Jan 16, 2025 · e226689 · e226689
1 parent 3ef67b6
commit e226689
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/tunneldriver/driver.go b/pkg/tunneldriver/driver.go
@@ -480,6 +480,8 @@ func (td *TunnelDriver) CreateAgentEndpoint(ctx context.Context, name string, sp
 	if err != nil {
 		return err
 	}
+	td.agentEndpoints.Add(name, tun)
+	td.tunnels[name] = tun
 
 	upstreamPort, err := strconv.Atoi(upstreamURL.Port())
 	if err != nil {