skynet/examples/simplewebsocket.lua

local skynet = require "skynet"
local socket = require "skynet.socket"
local service = require "skynet.service"
local websocket = require "http.websocket"
local cjson = require "cjson"
local cjson_encode = cjson.encode

local handle = {}
local MODE = ...

-- if MODE == "agent" then
--     function handle.connect(id)
--         print("ws connect from: " .. tostring(id))
--     end

--     function handle.handshake(id, header, url)
--         local addr = websocket.addrinfo(id)
--         print("ws handshake from: " .. tostring(id), "url", url, "addr:", addr)
--         print("----header-----")
--         for k,v in pairs(header) do
--             print(k,v)
--         end
--         print("--------------")
--     end

--     function handle.message(id, msg, msg_type)
--         assert(msg_type == "binary" or msg_type == "text")
--         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, addr)
--             local ok, err = websocket.accept(id, handle, protocol, addr)
--             if not ok then
--                 print(err)
--             end
--         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/test_websocket", 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, addr)
--             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

local function simple_echo_client_service(protocol)
    local skynet = require "skynet"
    local websocket = require "http.websocket"
    local url = string.format("%s://192.168.31.100:8001", protocol)
    print("echo server 3333333333333")
    skynet.sleep(100)
    local ws_id = websocket.connect(url)

    local tab = {
        type = "login",
        session = 1,
        content = {
            sid = 1,
            account = 4090,
            channel = "xiaohei",
        }
    }
    local msg = cjson_encode(tab)
    print(">: " .. msg)
    websocket.write(ws_id, msg)
    skynet.sleep(600)

    -- skynet.sleep(10000)
    -- websocket.ping(ws_id)
    -- websocket.write(ws_id, "hello")
    -- websocket.ping(ws_id)

        -- 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

skynet.start(function()
    simple_echo_client_service("ws")
end)