mirror of
https://github.com/nim-lang/Nim.git
synced 2025-12-29 01:14:41 +00:00
30 lines
915 B
Nim
30 lines
915 B
Nim
import asyncnet, asyncdispatch, strtabs
|
||
|
||
type
|
||
WebSocketCallback = proc (client: WebSocket, message: WebSocketMessage) {.closure, gcsafe.}
|
||
WebSocketRecvClosure = proc (ws: WebSocket): Future[string] {.gcsafe.}
|
||
|
||
WebSocketMessage = ref object
|
||
msg: string
|
||
|
||
WebSocket = ref object
|
||
socket: AsyncSocket
|
||
header: StringTableRef
|
||
onOpen: WebSocketCallback
|
||
onMessage: WebSocketCallback
|
||
onClose: WebSocketCallback
|
||
|
||
proc recv(ws: WebSocket, p: WebSocketRecvClosure): Future[string] {.async.}=
|
||
if not ws.socket.isClosed():
|
||
result = await ws.p()
|
||
if result == "":
|
||
ws.socket.close()
|
||
if ws.onClose != nil:
|
||
ws.onClose(ws, nil)
|
||
return result
|
||
|
||
proc reсvSize(ws: WebSocket, size: int): Future[string] {.async.} =
|
||
proc recvSizeClosure(ws: WebSocket): Future[string] {.async.} =
|
||
return await ws.socket.recv(size)
|
||
return await ws.recv(recvSizeClosure)
|