optimize websocket api

sniper00 · Jul 18, 2024 · 549faf9 · 549faf9
1 parent dfc5d7f
commit 549faf9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/example/network_websocket.lua b/example/network_websocket.lua
@@ -17,10 +17,12 @@ websocket.on_accept(function(fd, request)
 end)
 
 websocket.wson("message",function(fd, msg)
+    print(moon.decode(msg, "Z"))
+
     -- binary frame
-    -- socket.write(fd, msg)
+    -- websocket.write(fd, msg)
+
     -- text frame
-    print(moon.decode(msg, "Z"))
     websocket.write_text(fd, moon.decode(msg, "Z"))
 end)
 

diff --git a/lualib/moon/http/websocket.lua b/lualib/moon/http/websocket.lua
@@ -30,7 +30,11 @@ local socket_data_type = {
 ---| 'ping'
 ---| 'pong'
 
-local websocket = {}
+---@class websocket
+---@field write fun(fd:integer, data:string|buffer_ptr|buffer_shr_ptr):boolean send binary data
+local websocket = {
+    write = socket.write,
+}
 
 function websocket.close(fd)
     socket.close(fd)