bunny-lab/Borealis-Github-Replica
1
0
Fork 0
mirror of https://github.com/bunny-lab-io/Borealis.git synced 2025-12-16 12:05:48 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fixed Connect Button in Remote Shell

Browse Source
This commit is contained in:
Nicole Rappe
2025-12-07 17:36:48 -07:00
parent 8bbd6b86ed
commit d6b701efca
2 changed files with 31 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
Data/Engine/services/WebSocket/Agent/reverse_tunnel_orchestrator.py
View File

@@ -668,7 +668,23 @@ class ReverseTunnelService:
server.close(code=code, reason=reason)
except Exception:
self.logger.debug("protocol server close failed tunnel_id=%s", tunnel_id, exc_info=True)
if tunnel_id in self._protocol_servers:
try:
self._protocol_servers.pop(tunnel_id, None)
except Exception:
pass
self._push_stop_to_agent(lease, reason=reason)
websocket = self._agent_sockets.pop(tunnel_id, None)
if websocket is not None:
try:
self.lease_manager.mark_agent_disconnected(tunnel_id)
except Exception:
pass
try:
if self._loop:
self._loop.call_soon_threadsafe(asyncio.create_task, websocket.close())
except Exception:
self.logger.debug("agent websocket close failed tunnel_id=%s", tunnel_id, exc_info=True)
self.release_bridge(tunnel_id, reason=reason)
return True