CheeseAPI-Websocket 1.0.2

一款基于CheeseAPI的升级款Websocket插件。

CheeseAPI_Websocket

介绍

一款基于CheeseAPI的升级款Websocket插件，它能够解决在多worker下websocket的通讯问题，前提是需要引入redis。

安装

目前仅保证支持3.11及以上的python。

pip install CheeseAPI_Websocket

使用

CheeseAPI_Websocket是CheeseAPI的一款插件，它需要依赖于CheeseAPI才能运行。

import threading, time

from CheeseAPI import app, WebsocketClient, Response
from CheeseAPI_Websocket import websocket

app.modules.append('CheeseAPI_Websocket') # 加入模块

websocket.init() # 初始化redis连接

@app.route.websocket('/')
class Test(WebsocketClient):
    ...

# 创建一个线程，在非协程环境下发送Websocket
@app.handle.server_afterStartingHandle
def test():
    def test0():
        while True:
            websocket.send('/', '你好')
            time.sleep(1)
threading.Thread(target = test, daemon = True).start()

# 在协程环境下发送Websocket
@app.route.post('/websocket')
async def test1():
    await websocket.async_send('/', '世界')
    return Response()

if __name__ == '__main__':
    app.run()

若传输的数据量过大，请使用原生的send方法，以避免为redis带来过大的负担。

Websocket

from CheeseAPI_Websocket import websocket

websocket.init(host: IPv4 = app.server.host, port: Port = 6379, db: NonNegativeInt = 0)

初始化redis连接。

websocket.send(path: str, message: str | bytes | dict | list, sid: str | List[str] | Literal['*'] = '*')

发送消息，支持广播。

await websocket.async_send(path: str, message: str | bytes | dict | list, sid: str | List[str] | Literal['*'] = '*')

发送消息，支持广播。

websocket.close(path: str, sid: str | List[str] | Literal['*'] = '*')

关闭连接，支持广播。

await websocket.async_close(path: str, sid: str | List[str] | Literal['*'] = '*')

关闭连接，支持广播。