Fix memory leak in websocket handler

harsh97 · harsh97 · commit fd0a474cd1f6 · 2024-06-04T08:20:12.000+05:30
diff --git a/ads/aqua/extension/ui_websocket_handler.py b/ads/aqua/extension/ui_websocket_handler.py
@@ -112,6 +112,7 @@ def on_message_processed(self, future: concurrent.futures.Future):
             )
             raise e
         finally:
+            self.future_message_map.pop(future)
             # Send the response back to the client on the event thread
             IOLoop.current().run_sync(lambda: self.write_message(response.to_json()))
 

Original file line number	Diff line number	Diff line change
`@@ -112,6 +112,7 @@ def on_message_processed(self, future: concurrent.futures.Future):`
`112`	`112`	`)`
`113`	`113`	`raise e`
`114`	`114`	`finally:`
	`115`	`+ self.future_message_map.pop(future)`
`115`	`116`	`# Send the response back to the client on the event thread`
`116`	`117`	`IOLoop.current().run_sync(lambda: self.write_message(response.to_json()))`
`117`	`118`