Class SingleSessionWSHandler

pywebio/platform/tornado.py:315–358

    thread = threading.current_thread()

    class SingleSessionWSHandler(_webio_handler(cdn=False)):
        session: ScriptModeSession = None
        instance: typing.ClassVar = None
        closed = False

        def send_msg_to_client(self, session):
            for msg in session.get_task_commands():
                try:
                    self.write_message(json.dumps(msg))
                except TypeError as e:
                    logger.exception('Data serialization error: %s\n'
                                     'This may be because you pass the wrong type of parameter to the function'
                                     ' of PyWebIO.\nData content: %s', e, msg)

        def open(self):
            if SingleSessionWSHandler.session is None:
                SingleSessionWSHandler.instance = self
                session_info = get_session_info_from_headers(self.request.headers)
                session_info['user_ip'] = self.request.remote_ip
                session_info['request'] = self.request
                session_info['backend'] = 'tornado'
                session_info['protocol'] = 'websocket'
                self.session = SingleSessionWSHandler.session = ScriptModeSession(
                    thread, session_info=session_info,
                    on_task_command=self.send_msg_to_client,
                    loop=asyncio.get_event_loop())
                websocket_conn_opened.set()
            else:
                self.close()

        def on_message(self, data):
            if isinstance(data, bytes):
                event = deserialize_binary_event(data)
            else:
                event = json.loads(data)
            if event is None:
                return
            self.session.send_client_event(event)

        def on_close(self):
            if self.session is not None:
                self.session.close()
                self.closed = True
                logger.debug('ScriptModeSession closed')

    async def wait_to_stop_loop(server):
        """当只剩当前线程和Daemon线程运行时，关闭Server