mirror of
https://github.com/cloudwu/skynet.git
synced 2026-07-23 19:43:09 +00:00
add ws/wss server and client support
This commit is contained in:
90
examples/simplewebsocket.lua
Executable file
90
examples/simplewebsocket.lua
Executable file
@@ -0,0 +1,90 @@
|
||||
local skynet = require "skynet"
|
||||
local socket = require "skynet.socket"
|
||||
local service = require "skynet.service"
|
||||
local websocket = require "http.websocket"
|
||||
|
||||
local handle = {}
|
||||
local MODE = ...
|
||||
|
||||
if MODE == "agent" then
|
||||
function handle.connect(id)
|
||||
print("ws connect from: " .. tostring(id))
|
||||
end
|
||||
|
||||
function handle.handshake(id, header)
|
||||
print("ws handshake from: " .. tostring(id))
|
||||
print("----header-----")
|
||||
for k,v in pairs(header) do
|
||||
print(k,v)
|
||||
end
|
||||
print("--------------")
|
||||
end
|
||||
|
||||
function handle.message(id, msg)
|
||||
websocket.write(id, msg)
|
||||
end
|
||||
|
||||
function handle.ping(id)
|
||||
print("ws ping from: " .. tostring(id) .. "\n")
|
||||
end
|
||||
|
||||
function handle.pong(id)
|
||||
print("ws pong from: " .. tostring(id))
|
||||
end
|
||||
|
||||
function handle.close(id, code, reason)
|
||||
print("ws close from: " .. tostring(id), code, reason)
|
||||
end
|
||||
|
||||
function handle.error(id)
|
||||
print("ws error from: " .. tostring(id))
|
||||
end
|
||||
|
||||
skynet.start(function ()
|
||||
skynet.dispatch("lua", function (_,_, id, protocol)
|
||||
websocket.accept(id, handle, protocol)
|
||||
end)
|
||||
end)
|
||||
|
||||
else
|
||||
local function simple_echo_client_service(protocol)
|
||||
local skynet = require "skynet"
|
||||
local websocket = require "http.websocket"
|
||||
local url = string.format("%s://127.0.0.1:9948/", protocol)
|
||||
local ws_id = websocket.connect(url)
|
||||
while true do
|
||||
local msg = "hello world!"
|
||||
websocket.write(ws_id, msg)
|
||||
print(">: " .. msg)
|
||||
local resp, close_reason = websocket.read(ws_id)
|
||||
print("<: " .. (resp and resp or "[Close] " .. close_reason))
|
||||
if not resp then
|
||||
print("echo server close.")
|
||||
break
|
||||
end
|
||||
websocket.ping(ws_id)
|
||||
skynet.sleep(100)
|
||||
end
|
||||
end
|
||||
|
||||
skynet.start(function ()
|
||||
local agent = {}
|
||||
for i= 1, 20 do
|
||||
agent[i] = skynet.newservice(SERVICE_NAME, "agent")
|
||||
end
|
||||
local balance = 1
|
||||
local protocol = "ws"
|
||||
local id = socket.listen("0.0.0.0", 9948)
|
||||
skynet.error(string.format("Listen websocket port 9948 protocol:%s", protocol))
|
||||
socket.start(id, function(id, addr)
|
||||
print(string.format("accept client socket_id: %s addr:%s", id, addr))
|
||||
skynet.send(agent[balance], "lua", id, protocol)
|
||||
balance = balance + 1
|
||||
if balance > #agent then
|
||||
balance = 1
|
||||
end
|
||||
end)
|
||||
-- test echo client
|
||||
service.new("websocket_echo_client", simple_echo_client_service, protocol)
|
||||
end)
|
||||
end
|
||||
Reference in New Issue
Block a user