use request context for controller connections

canonical · Jul 16, 2024 · fca9ff3 · fca9ff3
1 parent 4bfd2a9
commit fca9ff3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/jujuapi/websocket.go b/internal/jujuapi/websocket.go
@@ -55,7 +55,7 @@ func (s *apiServer) ServeWS(ctx context.Context, conn *websocket.Conn) {
 	controllerRoot := newControllerRoot(s.jimm, s.params, identityId)
 	s.cleanup = controllerRoot.cleanup
 	Dblogger := controllerRoot.newAuditLogger()
-	serveRoot(context.Background(), controllerRoot, Dblogger, conn)
+	serveRoot(ctx, controllerRoot, Dblogger, conn)
 }
 
 // Kill implements the rpc.Killer interface.