在Tornado中保留websocket连接列表[副本]

import tornado.ioloop
import tornado.web
import tornado.websocket

# 全局连接列表
connected_clients = set()

class WebSocketHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print("WebSocket opened")
        connected_clients.add(self)

    def on_message(self, message):
        print(f"Received message: {message}")
        # 可以在这里处理消息并推送给其他客户端

    def on_close(self):
        print("WebSocket closed")
        connected_clients.remove(self)

class MainHandler(tornado.web.RequestHandler):
    def get(self):
        self.write("Hello, world")

def make_app():
    return tornado.web.Application([
        (r"/", MainHandler),
        (r"/websocket", WebSocketHandler),
    ])

if __name__ == "__main__":
    app = make_app()
    app.listen(8888)
    tornado.ioloop.IOLoop.current().start()

import threading

lock = threading.Lock()

class WebSocketHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        with lock:
            connected_clients.add(self)

    def on_close(self):
        with lock:
            connected_clients.remove(self)