【笔记】Go通过Websocket协议建立服务端和客户端

发表于 2023-07-12 更新于 2025-12-15 阅读次数：

前言

Go通过Websocket协议（ws://）建立服务端和客户端

下载依赖

1	go get github.com/gorilla/websocket

服务端

<port>：监听端口

package main

import (
    "fmt"
    "github.com/gorilla/websocket"
    "net/http"
)

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

func handler(w http.ResponseWriter, r *http.Request) {
    // HTTP协议升级为Websocket协议
    conn, _ := UP.Upgrade(w, r, nil)
    // 持续接收请求
    for {
        m, p, e := conn.ReadMessage()
        if e != nil {
            fmt.Println("连接已断开")
            break
        }
        fmt.Println(m, string(p))
        if string(p) == "ping" {
            // 发送一个消息
            conn.WriteMessage(websocket.TextMessage, []byte("pong"))
        }
    }
}

func main() {
    http.HandleFunc("/", handler)
    fmt.Println("服务端已启动")
    http.ListenAndServe("0.0.0.0:<port>", nil)
}

客户端

<ip>：服务端IP地址
<port>：服务端端口

package main

import (
    "fmt"
    "github.com/gorilla/websocket"
    "os"
    "runtime"
    "time"
)

func main() {
start:
    // 获取连接
    dl := websocket.Dialer{}
    conn, _, e := dl.Dial("ws://<ip>:<port>", nil)
    if e != nil {
        // 连接失败，延时2秒重新连接
        time.Sleep(2 * time.Second)
        goto start
    }
    defer conn.Close()
    
    // 持续发送消息
    for {
        // 发送一个消息
        conn.WriteMessage(websocket.TextMessage, []byte("ping"))
        // 接收一个消息
        m, p, e := conn.ReadMessage()
        if e != nil {
            // 连接失败，延时2秒重新连接
            time.Sleep(2 * time.Second)
            goto start
        }
        fmt.Println(m, string(p))
    }
}