mirror of
https://github.com/logos-storage/nim-websock.git
synced 2026-01-03 06:03:07 +00:00
a1ae7d2c70
* Implement websocket server. * Implement websocket client. * Run nimpretty. * Remove commented code. * Address comments. * Address comments on websocket server. * Use seq[byte] to store data. * Working bytes conversion. * Remove result from return * Refactor the code. * Minor change. * Add test. * Add websocket test and fix closing handshake. * Add MsgReader to read data in external buffer.
42 lines
1.3 KiB
Nim
42 lines
1.3 KiB
Nim
import ../src/ws, ../src/http, chronos, chronicles, httputils, stew/byteutils
|
|
|
|
proc cb(transp: StreamTransport, header: HttpRequestHeader) {.async.} =
|
|
info "Handling request:", uri = header.uri()
|
|
if header.uri() == "/ws":
|
|
info "Initiating web socket connection."
|
|
try:
|
|
var ws = await newWebSocket(header, transp, "myfancyprotocol")
|
|
if ws.readyState == Open:
|
|
info "Websocket handshake completed."
|
|
else:
|
|
error "Failed to open websocket connection."
|
|
return
|
|
|
|
while true:
|
|
# Only reads header for data frame.
|
|
let msgReader = await ws.nextMessageReader()
|
|
|
|
# Read the frame payload in buffer.
|
|
let buffer = newSeq[byte](100)
|
|
var recvData :seq[byte]
|
|
while msgReader.error != EOFError:
|
|
msgReader.readMessage(buffer)
|
|
recvData.add buffer
|
|
if ws.readyState == ReadyState.Closed:
|
|
return
|
|
info "Response: ", data = recvData
|
|
await ws.send(recvData)
|
|
|
|
except WebSocketError:
|
|
error "WebSocket error:", exception = getCurrentExceptionMsg()
|
|
|
|
discard await transp.sendHTTPResponse(HttpVersion11, Http200, "Hello World")
|
|
await transp.closeWait()
|
|
|
|
when isMainModule:
|
|
let address = "127.0.0.1:8888"
|
|
var httpServer = newHttpServer(address, cb)
|
|
httpServer.start()
|
|
echo "Server started..."
|
|
waitFor httpServer.join()
|