add message type

cloudwu · Mar 30, 2020 · fd27676 · fd27676
1 parent 7efa799
commit fd27676
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/examples/simplewebsocket.lua b/examples/simplewebsocket.lua
@@ -21,7 +21,8 @@ if MODE == "agent" then
         print("--------------")
     end
 
-    function handle.message(id, msg)
+    function handle.message(id, msg, msg_type)
+        assert(msg_type == "binary" or msg_type == "text")
         websocket.write(id, msg)
     end
 

diff --git a/lualib/http/websocket.lua b/lualib/http/websocket.lua
@@ -273,7 +273,7 @@ local function resolve_accept(self)
             try_handle(self, "pong")
         else
             if fin and #recv_buf == 0 then
-                try_handle(self, "message", payload_data)
+                try_handle(self, "message", payload_data, op)
             else
                 recv_buf[#recv_buf+1] = payload_data
                 recv_count = recv_count + #payload_data
@@ -282,7 +282,7 @@ local function resolve_accept(self)
                 end
                 if fin then
                     local s = table.concat(recv_buf)
-                    try_handle(self, "message", s)
+                    try_handle(self, "message", s, op)
                     recv_buf = {}  -- clear recv_buf
                     recv_count = 0
                 end