{"id":208003,"date":"2025-07-08T09:58:57","date_gmt":"2025-07-08T01:58:57","guid":{"rendered":"https:\/\/server.hk\/cnblog\/208003\/"},"modified":"2025-07-08T09:58:57","modified_gmt":"2025-07-08T01:58:57","slug":"%e4%bb%a5%e5%be%aa%e7%8e%af%e6%96%b9%e5%bc%8f%e5%90%91%e5%ae%a2%e6%88%b7%e7%ab%af%e6%8f%90%e4%be%9b%e6%b6%88%e6%81%af%e7%9a%84-websocket-%e6%9c%8d%e5%8a%a1%e5%99%a8","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/208003\/","title":{"rendered":"\u4ee5\u5faa\u73af\u65b9\u5f0f\u5411\u5ba2\u6237\u7aef\u63d0\u4f9b\u6d88\u606f\u7684 WebSocket \u670d\u52a1\u5668"},"content":{"rendered":"

<\/b> <\/p>\n

\u5f53\u524d\u4f4d\u7f6e\uff1a ><\/span> ><\/span> ><\/span> ><\/span> \u4ee5\u5faa\u73af\u65b9\u5f0f\u5411\u5ba2\u6237\u7aef\u63d0\u4f9b\u6d88\u606f\u7684 WebSocket \u670d\u52a1\u5668<\/span><\/p>\n

\u6765\u6e90\uff1astackoverflow<\/span>
\n2024-04-27 20:00:53<\/span>
\n<\/i>0\u6d4f\u89c8<\/span>
\n<\/i>\u6536\u85cf<\/span> <\/p>\n

\u76ee\u524d\u4e0a\u5df2\u7ecf\u6709\u5f88\u591a\u5173\u4e8eGolang<\/span>\u7684\u6587\u7ae0\u4e86\uff0c\u81ea\u5df1\u5728\u521d\u6b21\u9605\u8bfb\u8fd9\u4e9b\u6587\u7ae0\u4e2d\uff0c\u4e5f\u89c1\u8bc6\u5230\u4e86\u5f88\u591a\u5b66\u4e60\u601d\u8def\uff1b\u90a3\u4e48\u672c\u6587\u300a\u4ee5\u5faa\u73af\u65b9\u5f0f\u5411\u5ba2\u6237\u7aef\u63d0\u4f9b\u6d88\u606f\u7684 WebSocket \u670d\u52a1\u5668\u300b<\/span>\uff0c\u4e5f\u5e0c\u671b\u80fd\u5e2e\u52a9\u5230\u5927\u5bb6\uff0c\u5982\u679c\u9605\u8bfb\u5b8c\u540e\u771f\u7684\u5bf9\u4f60\u5b66\u4e60Golang<\/span>\u6709\u5e2e\u52a9\uff0c\u6b22\u8fce\u52a8\u52a8\u624b\u6307\uff0c\u8bc4\u8bba\u7559\u8a00\u5e76\u5206\u4eab~<\/p>\n

\u95ee\u9898\u5185\u5bb9
\n <\/p>\n

\u6211\u5728 go \u4e2d\u6709\u4e00\u4e2a\u4f7f\u7528 gorilla websocket \u5305\u7684 websocket \u670d\u52a1\u5668\u3002\u5728\u6b64\u9636\u6bb5\uff0c\u6211\u5c06\u53ea\u6709\u4e00\u53f0\u670d\u52a1\u5668\u4e3a 5 \u4e2a\u5ba2\u6237\u7aef\u63d0\u4f9b\u670d\u52a1\u3002\u6211\u4ece\u4e0a\u6e38\u6536\u5230\u4e00\u4e9b\u6d88\u606f\u5230 websocket \u670d\u52a1\u5668\u3002\u6211\u7684\u76ee\u7684\u662f\u4e0d\u5c06\u6240\u6709\u6d88\u606f\u5e7f\u64ad\u5230\u8fde\u63a5\u7684\u5ba2\u6237\u7aef\u3002\u6211\u53ea\u60f3\u4ee5\u5faa\u73af\u65b9\u5f0f\u5c06\u6d88\u606f\u7684\u4e00\u4efd\u526f\u672c\u53d1\u9001\u7ed9\u8fde\u63a5\u7684\u5ba2\u6237\u7aef\u3002\u54ea\u4e2a\u5ba2\u6237\u7aef\u83b7\u53d6\u5b83\u5e76\u4e0d\u91cd\u8981\uff0c\u53ea\u8981\u53ea\u6709\u4e00\u4e2a\u5ba2\u6237\u7aef\u83b7\u53d6\u5b83\u5373\u53ef\u3002<\/p>\n

\u6211\u5c1d\u8bd5\u7684\u89e3\u51b3\u65b9\u6848<\/strong> \u6211\u6709\u4e00\u4e2a\u7b80\u5355\u7684 go \u670d\u52a1\u5668\uff0c\u521b\u5efa\u4e86\u4e00\u4e2a\u6211\u6b63\u5728\u63a5\u6536\u7684\u5ba2\u6237\u7aef\u6c60\uff08websocket \u8fde\u63a5\uff09\u3002\u4f46\u662f\uff0c\u6211\u6ca1\u6709\u770b\u5230\u4efb\u4f55\u5982\u4e0a\u6240\u8ff0\u7684\u5faa\u73af\u6d88\u606f\u7684\u9009\u9879\u3002\u6211\u6240\u6709\u7684\u5ba2\u6237\u90fd\u6536\u5230\u4e86\u8fd9\u4e2a\u6d88\u606f\u3002\u5982\u4f55\u53ea\u5411\u8fde\u63a5\u7684\u5ba2\u6237\u7aef\u53d1\u9001\u4e00\u4efd\u6d88\u606f\u526f\u672c\uff0c\u800c\u4e0d\u662f\u5411\u6240\u6709\u5ba2\u6237\u7aef\u5e7f\u64ad\u3002<\/p>\n

\u6291\u5236\u5668<\/strong> \u6211\u7684\u4ee3\u7801\u53d6\u81ea\u5728\u7ebf\u8d44\u6e90\u5e76\u6839\u636e\u6211\u7684\u8981\u6c42\u8fdb\u884c\u4e86\u4fee\u6539\u3002\u6211\u5bf9 go \u548c websockets \u8fd8\u6bd4\u8f83\u964c\u751f\u3002 \u4f7f\u7528 websockets \u662f\u5426\u53ef\u4ee5\u505a\u5230\u8fd9\u4e00\u70b9\uff1f<\/strong><\/p>\n

main.go<\/h2>\n
package main\n\nimport (\n    \"fmt\"\n    \"net\/http\"\n\n    \"github.com\/realtime-chat-go-react\/backend\/pkg\/websocket\"\n)\n\nfunc servews(pool *websocket.pool, w http.responsewriter, r *http.request) {\n    fmt.println(\"websocket endpoint hit\")\n    conn, err := websocket.upgrade(w, r)\n    if err != nil {\n        fmt.fprintf(w, \"%+v\\n\", err)\n    }\n\n    client := &websocket.client{\n        conn: conn,\n        pool: pool,\n    }\n\n    pool.register <- client\n    client.read()\n}\n\nfunc setuproutes() {\n    pool := websocket.newpool()\n    go pool.start()\n\n    http.handlefunc(\"\/ws\", func(w http.responsewriter, r *http.request) {\n        servews(pool, w, r)\n    })\n}\n\nfunc main() {\n    setuproutes()\n    err := http.listenandserve(\":8080\",nil)\n\n    if err != nil {\n        fmt.println(err)\n    }\n}<\/pre>\n
websocket.go<\/h2>\n
package websocket\n\nimport (\n    \"log\"\n    \"net\/http\"\n\n    \"github.com\/gorilla\/websocket\"\n)\n\nvar upgrader = websocket.upgrader{\n    readbuffersize:  1024,\n    writebuffersize: 1024,\n}\n\nvar wslist []*websocket.conn\n\nfunc upgrade(w http.responsewriter, r *http.request) (*websocket.conn, error) {\n    upgrader.checkorigin = func(r *http.request) bool { return true }\n    conn, err := upgrader.upgrade(w, r, nil)\n    wslist = append(wslist, conn) \/\/creating a list here to store all websocket clients.\n\n    if err != nil {\n        log.println(err)\n        return nil, err\n    }\n\n    return conn, nil\n}<\/pre>\n
pool.go<\/h2>\n
package websocket\n\nimport \"fmt\"\n\ntype pool struct {\n    register   chan *client\n    unregister chan *client\n    clients    map[*client]bool\n    broadcast  chan message\n}\n\nfunc newpool() *pool {\n    return &pool{\n        register:   make(chan *client),\n        unregister: make(chan *client),\n        clients:    make(map[*client]bool),\n        broadcast:  make(chan message),\n    }\n}\n\nfunc (pool *pool) start() {\n    for {\n        select {\n        case client := <-pool.register:\n            pool.clients[client] = true\n            fmt.println(\"size of connection pool: \", len(pool.clients))\n            for client, _ := range pool.clients {\n                fmt.println(client)\n                client.conn.writejson(message{type: 1, body: \"new user joined...\"})\n            }\n            break\n        case client := <-pool.unregister:\n            delete(pool.clients, client)\n            fmt.println(\"size of connection pool: \", len(pool.clients))\n            for client, _ := range pool.clients {\n                client.conn.writejson(message{type: 1, body: \"user disconnected...\"})\n            }\n            break\n        case message := <-pool.broadcast:     \/\/this is where i need to modify the code but not sure how\n            fmt.println(\"sending message to all clients in pool\")\n            for client, _ := range pool.clients {\n                if err := client.conn.writejson(message); err != nil {\n                    fmt.println(err)\n                    return\n                }\n            }\n        }\n    }\n}<\/pre>\n
client.go<\/h2>\n
package websocket\n\nimport (\n    \"fmt\"\n    \"log\"\n    \"sync\"\n\n    \"github.com\/gorilla\/websocket\"\n)\n\ntype Client struct {\n    ID   string\n    Conn *websocket.Conn\n    Pool *Pool\n    mu   sync.Mutex\n}\n\ntype Message struct {\n    Type int    `json:\"type\"`\n    Body string `json:\"body\"`\n}\n\nfunc (c *Client) Read() {\n    defer func() {\n        c.Pool.Unregister <- c\n        c.Conn.Close()\n    }()\n\n    for {\n        messageType, p, err := c.Conn.ReadMessage()\n        if err != nil {\n            log.Println(err)\n            return\n        }\n\n        message := Message{Type: messageType, Body: string(p)}\n        c.Pool.Broadcast <- message\n        fmt.Printf(\"Message Received: %+v\\n\", message)\n\n    }\n}<\/pre>\n
 <\/p>\n
\u89e3\u51b3\u65b9\u6848<\/h2>\n
 <\/p>\n
\u4fee\u6539\u6c60\u4ee5\u5c06\u5ba2\u6237\u7aef\u5b58\u50a8\u5728\u5207\u7247\u800c\u4e0d\u662f\u6620\u5c04\u4e2d\u3002\u6dfb\u52a0\u5b57\u6bb5\u6765\u8bb0\u5f55\u4ee5\u524d\u4f7f\u7528\u8fc7\u7684\u5ba2\u6237\u7aef\u7684\u7d22\u5f15\u3002<\/p>\n
type pool struct {\n    register   chan *client\n    unregister chan *client\n    clients    []*client\n    broadcast  chan message\n    prevclientindex int\n}<\/pre>\n
\u5faa\u73af\u800c\u4e0d\u662f\u5e7f\u64ad\uff1a<\/p>\n
case message := <-pool.broadcast: \n    if len(pool.clients) == 0 {\n        continue\n    }\n    pool.prevclientindex++\n    if pool.prevclientindex >= len(pool.clients) {\n       pool.prevclientindex = 0\n    }\n    client := pool.clients[pool.prevclientindex]\n    if err := client.conn.writejson(message); err != nil {\n        \/\/ handle error\n        ...<\/pre>\n
\u6ce8\u518c\u9644\u52a0\u5230\u5207\u7247\uff1a<\/p>\n
case client := <-pool.register:\n    pool.clients = append(pool.clients, client)\n    ...<\/pre>\n
\u53d6\u6d88\u6ce8\u518c\u5c06\u4ece\u5207\u7247\u4e2d\u5220\u9664\u5ba2\u6237\u7aef\uff1a<\/p>\n
case client := <-pool.Unregister:\n    j := 0\n    for _, c := range pool.Clients {\n       if c != client {\n          c.Clients[j] = c\n          j++\n       }\n    }\n    pool.Clients = pool.Clients[:j]\n    ...<\/pre>\n
\u7406\u8bba\u8981\u638c\u63e1\uff0c\u5b9e\u64cd\u4e0d\u80fd\u843d\uff01\u4ee5\u4e0a\u5173\u4e8e\u300a\u4ee5\u5faa\u73af\u65b9\u5f0f\u5411\u5ba2\u6237\u7aef\u63d0\u4f9b\u6d88\u606f\u7684 WebSocket \u670d\u52a1\u5668\u300b\u7684\u8be6\u7ec6\u4ecb\u7ecd\uff0c\u5927\u5bb6\u90fd\u638c\u63e1\u4e86\u5427\uff01\u5982\u679c\u60f3\u8981\u7ee7\u7eed\u63d0\u5347\u81ea\u5df1\u7684\u80fd\u529b\uff0c\u90a3\u4e48\u5c31\u6765\u5173\u6ce8\u516c\u4f17\u53f7\u5427\uff01<\/p>\n","protected":false},"excerpt":{"rendered":"
\u5f53\u524d\u4f4d\u7f6e\uff1a > > ...<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4925],"tags":[],"class_list":["post-208003","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/208003","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=208003"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/208003\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=208003"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=208003"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=208003"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}