go get github.com/gorilla/websocket

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

// 創(chuàng)建一個(gè) upgrader 對(duì)象，用于升級(jí) HTTP 連接為 WebSocket 連接
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    // 將 HTTP 連接升級(jí)為 WebSocket 連接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade failed:", err)
        return
    }

    defer conn.Close()

    // 處理 WebSocket 會(huì)話
    for {
        // 讀取客戶端發(fā)送的消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("Read message failed:", err)
            break
        }

        // 處理客戶端發(fā)送的消息
        log.Printf("Received message: %s", message)

        // 回復(fù)客戶端消息
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println("Write message failed:", err)
            break
        }
    }
}

package main

import (
    "log"
    "os"
    "os/signal"
    "time"

    "github.com/gorilla/websocket"
)

func main() {
    // 創(chuàng)建一個(gè) dialer 對(duì)象，用于建立 WebSocket 連接
    dialer := websocket.Dialer{
        HandshakeTimeout: 10 * time.Second,
    }

    // 建立 WebSocket 連接
    conn, _, err := dialer.Dial("ws://localhost:8080/ws", nil)
    if err != nil {
        log.Fatal("Dial failed:", err)
    }
    defer conn.Close()

    // 啟動(dòng)一個(gè) goroutine 來(lái)接收服務(wù)器推送的消息
    go func() {
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("Read message failed:", err)
                break
            }

            log.Printf("Received message: %s", message)
        }
    }()

    // 向服務(wù)器發(fā)送消息
    message := []byte("Hello, WebSocket!")
    err = conn.WriteMessage(websocket.TextMessage, message)
    if err != nil {
        log.Println("Write message failed:", err)
        return
    }

    // 等待用戶按下 Ctrl+C 終止程序
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, os.Interrupt)
    <-interrupt
}

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

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

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade failed:", err)
        return
    }

    defer conn.Close()

    // 創(chuàng)建一個(gè)獨(dú)立的會(huì)話
    session := NewSession()

    // 處理會(huì)話
    session.Handle(conn)
}

type Session struct {
    conn *websocket.Conn
}

func NewSession() *Session {
    return &Session{}
}

func (s *Session) Handle(conn *websocket.Conn) {
    s.conn = conn

    go s.readLoop()
    go s.writeLoop()
}

func (s *Session) readLoop() {
    for {
        messageType, message, err := s.conn.ReadMessage()
        if err != nil {
            log.Println("Read message failed:", err)
            break
        }

        log.Printf("Received message: %s", message)
    }
}

func (s *Session) writeLoop() {
    for {
        select {
        // 從消息隊(duì)列中獲取消息并發(fā)送給客戶端
        case message := <-s.messageQueue:
            err := s.conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                log.Println("Write message failed:", err)
            }
        }
    }
}