gorilla/websocket 사용하기

gorilla/websocket 라이브러리

import "github.com/gorilla/websocket"

업그레이더

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

CheckOrigin 및 보안

upgrader.CheckOrigin = func(r *http.Request) bool {
    return r.Header.Get("Origin") == "https://myapp.com"
}

처리기에서 업그레이드하기

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        return
    }
    defer conn.Close()
    // read/write loop here
}

메시지 읽기

mt, data, err := conn.ReadMessage()
if err != nil {
    return // client closed or error
}
fmt.Printf("got %d bytes\n", len(data))
_ = mt

메시지 쓰기

err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))
if err != nil {
    return
}

에코 반복

for {
    mt, msg, err := conn.ReadMessage()
    if err != nil {
        break
    }
    conn.WriteMessage(mt, msg)
}

JSON 도우미

type Msg struct {
    Text string // tag: json text
}
var m Msg
conn.ReadJSON(&m)
conn.WriteJSON(Msg{Text: "ack"})

읽기 하나, 쓰기 하나 규칙

경로 등록

http.HandleFunc("/ws", wsHandler)
http.ListenAndServe(":8080", nil)

실행 가능: JSON 왕복 비유

package main

import (
    "encoding/json"
    "fmt"
)

func main() {
    in := []byte("{\"text\":\"ping\"}")
    var m map[string]string
    json.Unmarshal(in, &m)
    reply, _ := json.Marshal(map[string]string{"text": m["text"] + "-ack"})
    fmt.Println(string(reply))
}

빠른 확인

복습