WebSocket 클라이언트

서버와의 실시간 양방향 통신을 위한 WebSocket 클라이언트.

로딩

local websocket = require("websocket")

연결

기본 연결

local client, err = websocket.connect("wss://api.example.com/ws")
if err then
    return nil, err
end

옵션과 함께

local client, err = websocket.connect("wss://api.example.com/ws", {
    headers = {
        ["Authorization"] = "Bearer " .. token
    },
    protocols = {"graphql-ws"},
    dial_timeout = "10s",
    read_timeout = "30s",
    compression = websocket.COMPRESSION.CONTEXT_TAKEOVER
})

반환: Client, error

연결 옵션

타임아웃 형식: 숫자는 밀리초, 문자열은 Go duration 형식 ("5s", "1m")을 사용합니다.

메시지 보내기

텍스트 메시지

local ok, err = client:send("Hello, Server!")
if err then
    return nil, err
end

-- JSON 전송
client:send(json.encode({
    type = "subscribe",
    channel = "orders"
}))

바이너리 메시지

client:send(binary_data, websocket.BINARY)

반환: boolean, error

Ping

client:ping()

반환: boolean, error

메시지 받기

channel() 메서드는 메시지 수신을 위한 채널을 반환합니다. 멀티플렉싱을 위해 channel.select와 함께 작동합니다.

기본 수신

local ch = client:channel()

local msg, ok = ch:receive()
if ok then
    print("Type:", msg.type)  -- "text" 또는 "binary"
    print("Data:", msg.data)
end

메시지 루프

local ch = client:channel()

while true do
    local msg, ok = ch:receive()
    if not ok then
        break  -- 연결 닫힘
    end

    if msg.type == "text" then
        local data = json.decode(msg.data)
        handle_message(data)
    end
end

Select와 함께

local ch = client:channel()
local timeout = time.after("30s")

while true do
    local r = channel.select {
        ch:case_receive(),
        timeout:case_receive()
    }

    if r.channel == timeout then
        client:ping()  -- Keep-alive
        timeout = time.after("30s")
    else
        local data = json.decode(r.value.data)
        process(data)
    end
end

메시지 객체

연결 닫기

-- 정상 닫기 (코드 1000)
client:close()

-- 코드와 이유와 함께
client:close(websocket.CLOSE_CODES.NORMAL, "Session ended")

-- 에러 닫기
client:close(websocket.CLOSE_CODES.INTERNAL_ERROR, "Processing failed")

반환: boolean, error

상수

메시지 타입

-- 숫자 (send용)
websocket.TEXT    -- 1
websocket.BINARY  -- 2

-- 문자열 (수신된 메시지 type 필드)
websocket.TYPE_TEXT    -- "text"
websocket.TYPE_BINARY  -- "binary"
websocket.TYPE_PING    -- "ping"
websocket.TYPE_PONG    -- "pong"
websocket.TYPE_CLOSE   -- "close"

압축 모드

websocket.COMPRESSION.DISABLED         -- 0 (압축 없음)
websocket.COMPRESSION.CONTEXT_TAKEOVER -- 1 (슬라이딩 윈도우)
websocket.COMPRESSION.NO_CONTEXT       -- 2 (메시지별)

닫기 코드

client:close(websocket.CLOSE_CODES.NORMAL, "Done")

예제

실시간 채팅

local function connect_chat(room_id, on_message)
    local client, err = websocket.connect("wss://chat.example.com/ws", {
        headers = {["Authorization"] = "Bearer " .. token}
    })
    if err then
        return nil, err
    end

    -- 방 참여
    client:send(json.encode({
        type = "join",
        room = room_id
    }))

    -- 메시지 루프
    local ch = client:channel()
    while true do
        local msg, ok = ch:receive()
        if not ok then break end

        local data = json.decode(msg.data)
        on_message(data)
    end

    client:close()
end

Keep-Alive와 함께 가격 스트림

local client = websocket.connect("wss://stream.example.com/prices")

client:send(json.encode({
    action = "subscribe",
    symbols = {"BTC-USD", "ETH-USD"}
}))

local ch = client:channel()
local heartbeat = time.after("30s")

while true do
    local r = channel.select {
        ch:case_receive(),
        heartbeat:case_receive()
    }

    if r.channel == heartbeat then
        client:ping()
        heartbeat = time.after("30s")
    elseif not r.ok then
        break  -- 연결 닫힘
    else
        local price = json.decode(r.value.data)
        update_price(price.symbol, price.value)
    end
end

client:close()

권한

WebSocket 연결은 보안 정책 평가 대상입니다.

보안 액션

정책 설정은 보안 모델을 참조하세요.

에러

local client, err = websocket.connect(url)
if err then
    if errors.is(err, errors.PERMISSION_DENIED) then
        print("Access denied:", err:message())
    elseif err:retryable() then
        print("Temporary error:", err:message())
    end
    return nil, err
end

에러 처리는 에러 처리를 참조하세요.