Skip to content

Commit 2588656

Browse files
committed
FIX remove repeated ws from stickys
1 parent 8509138 commit 2588656

File tree

1 file changed

+15
-15
lines changed
  • src/gh/components/DF_websocket_listener

1 file changed

+15
-15
lines changed

src/gh/components/DF_websocket_listener/code.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -35,36 +35,36 @@ def RunScript(self,
3535
prefix = 'ws'
3636

3737
# Persistent state across runs
38-
sc.sticky.setdefault(f'{prefix}_ws_server', None)
39-
sc.sticky.setdefault(f'{prefix}_ws_loop', None)
40-
sc.sticky.setdefault(f'{prefix}_ws_thread', None)
38+
sc.sticky.setdefault(f'{prefix}_server', None)
39+
sc.sticky.setdefault(f'{prefix}_loop', None)
40+
sc.sticky.setdefault(f'{prefix}_thread', None)
4141
sc.sticky.setdefault(f'{prefix}_last_pcd', None)
4242
sc.sticky.setdefault(f'{prefix}_loaded_pcd', None)
43-
sc.sticky.setdefault(f'{prefix}_ws_logs', [])
44-
sc.sticky.setdefault(f'{prefix}_ws_thread_started', False)
43+
sc.sticky.setdefault(f'{prefix}_logs', [])
44+
sc.sticky.setdefault(f'{prefix}_thread_started', False)
4545
sc.sticky.setdefault(f'{prefix}_prev_start', False)
4646
sc.sticky.setdefault(f'{prefix}_prev_stop', False)
4747
sc.sticky.setdefault(f'{prefix}_prev_load', False)
4848

49-
logs = sc.sticky[f'{prefix}_ws_logs']
49+
logs = sc.sticky[f'{prefix}_logs']
5050

5151
# STOP server
52-
if i_stop and sc.sticky.pop(f'{prefix}_ws_thread_started', False):
53-
server = sc.sticky.pop(f'{prefix}_ws_server', None)
54-
loop = sc.sticky.pop(f'{prefix}_ws_loop', None)
52+
if i_stop and sc.sticky.pop(f'{prefix}_thread_started', False):
53+
server = sc.sticky.pop(f'{prefix}_server', None)
54+
loop = sc.sticky.pop(f'{prefix}_loop', None)
5555
if server and loop:
5656
try:
5757
server.close()
5858
asyncio.run_coroutine_threadsafe(server.wait_closed(), loop)
5959
logs.append("WebSocket server close initiated")
6060
except Exception as e:
6161
logs.append(f"Error closing server: {e}")
62-
sc.sticky[f'{prefix}_ws_thread'] = None
62+
sc.sticky[f'{prefix}_thread'] = None
6363
logs.append("Cleared previous WebSocket server flag")
6464
ghenv.Component.ExpireSolution(True) # noqa: F821
6565

6666
# START server
67-
if i_start and not sc.sticky[f'{prefix}_ws_thread_started']:
67+
if i_start and not sc.sticky[f'{prefix}_thread_started']:
6868

6969
async def echo(ws, path):
7070
logs.append("[GH] Client connected")
@@ -84,11 +84,11 @@ async def echo(ws, path):
8484

8585
async def server_coro():
8686
loop = asyncio.get_running_loop()
87-
sc.sticky[f'{prefix}_ws_loop'] = loop
87+
sc.sticky[f'{prefix}_loop'] = loop
8888

8989
logs.append(f"server_coro starting on {i_host}:{i_port}")
9090
server = await serve(echo, i_host, i_port)
91-
sc.sticky[f'{prefix}_ws_server'] = server
91+
sc.sticky[f'{prefix}_server'] = server
9292
logs.append(f"Listening on ws://{i_host}:{i_port}")
9393
await server.wait_closed()
9494
logs.append("Server coroutine exited")
@@ -101,8 +101,8 @@ def run_server():
101101

102102
t = threading.Thread(target=run_server, daemon=True)
103103
t.start()
104-
sc.sticky[f'{prefix}_ws_thread'] = t
105-
sc.sticky[f'{prefix}_ws_thread_started'] = True
104+
sc.sticky[f'{prefix}_thread'] = t
105+
sc.sticky[f'{prefix}_thread_started'] = True
106106
ghenv.Component.ExpireSolution(True) # noqa: F821
107107

108108
# LOAD buffered PCD on i_load rising edge

0 commit comments

Comments
 (0)