kopia lustrzana https://github.com/cirospaciari/socketify.py
31 wiersze
1.0 KiB
Python
31 wiersze
1.0 KiB
Python
|
from socketify import App, AppOptions, OpCode, CompressOptions
|
||
|
import asyncio
|
||
|
|
||
|
def ws_open(ws):
|
||
|
print('A WebSocket got connected!')
|
||
|
ws.send("Hello World!", OpCode.TEXT)
|
||
|
|
||
|
def ws_message(ws, message, opcode):
|
||
|
print(message, opcode)
|
||
|
#Ok is false if backpressure was built up, wait for drain
|
||
|
ok = ws.send(message, opcode)
|
||
|
|
||
|
async def ws_upgrade(res, req, socket_context):
|
||
|
key = req.get_header("sec-websocket-key")
|
||
|
protocol = req.get_header("sec-websocket-protocol")
|
||
|
extensions = req.get_header("sec-websocket-extensions")
|
||
|
await asyncio.sleep(2)
|
||
|
res.upgrade(key, protocol, extensions, socket_context)
|
||
|
|
||
|
app = App()
|
||
|
app.ws("/*", {
|
||
|
'compression': CompressOptions.SHARED_COMPRESSOR,
|
||
|
'max_payload_length': 16 * 1024 * 1024,
|
||
|
'idle_timeout': 12,
|
||
|
'open': ws_open,
|
||
|
'message': ws_message,
|
||
|
'upgrade': ws_upgrade
|
||
|
})
|
||
|
app.any("/", lambda res,req: res.end("Nothing to see here!'"))
|
||
|
app.listen(3000, lambda config: print("Listening on port http://localhost:%d now\n" % (config.port)))
|
||
|
app.run()
|