SweetSea/Techmino
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

再调整websocket库

Browse Source
This commit is contained in:
MrZ626
2021-02-17 00:16:11 +08:00
parent 063de09d3f
commit 8fdef2b7f5
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Zframework/websocket.lua
View File

@@ -27,8 +27,7 @@ function WS.new()
return m return m
end end
local seckey="osT3F7mvlojIvf3/8uIsJQ==" function WS:connect(server,path,body)
function WS:connect(server,path,head,body)
local host,port=unpack(splitStr(server,":")) local host,port=unpack(splitStr(server,":"))
local SOCK=self.socket local SOCK=self.socket
local res,err=SOCK:connect(host,port or 80) local res,err=SOCK:connect(host,port or 80)
@@ -44,8 +43,7 @@ function WS:connect(server,path,head,body)
"Content-Type: application/json\r\n".. "Content-Type: application/json\r\n"..
"Content-Length: "..#body.."\r\n".. "Content-Length: "..#body.."\r\n"..
"Sec-WebSocket-Version: 13\r\n".. "Sec-WebSocket-Version: 13\r\n"..
"Sec-WebSocket-Key: "..seckey.."\r\n".. "Sec-WebSocket-Key: osT3F7mvlojIvf3/8uIsJQ==\r\n\r\n"..--secKey
(head or"").."\r\n"..
body body
) )
repeat res=SOCK:receive("*l")until res=="" repeat res=SOCK:receive("*l")until res==""
@@ -86,8 +84,8 @@ local OPcode={
ping=9, ping=9,
pong=10 pong=10
} }
function WS:send(message,type) function WS:send(message,op)
_send(self.socket,OPcode[type] or 2--[[binary]],message) _send(self.socket,OPcode[op] or 2--[[binary]],message)
end end
function WS:read() function WS:read()
@@ -115,7 +113,7 @@ function WS:read()
res=SOCK:receive(length) res=SOCK:receive(length)
local closeCode local closeCode
if OPCODE==9 then--ping if OPCODE==9 then--ping
self:pong(res) self:send(res,10--[[pong]])
elseif OPCODE==8 then--close elseif OPCODE==8 then--close
closeCode=shl(res:byte(1),8)+res:byte(2) closeCode=shl(res:byte(1),8)+res:byte(2)
res=sub(res,3,-3) res=sub(res,3,-3)