Golang如何实现简单聊天室_Golang WebSocket实时通信实践

使用gorilla/websocket库搭建WebSocket服务器,通过Upgrade将HTTP连接升级为WebSocket连接,实现客户端与服务端的双向通信。2. 定义全局clients map和broadcast通道,利用协程监听广播消息并推送给所有活跃连接,实现消息的实时群发。

golang如何实现简单聊天室_golang websocket实时通信实践

用Golang实现一个简单聊天室,核心是利用WebSocket建立客户端与服务端的长连接,实现消息的实时双向通信。下面一步步带你完成一个基础但完整的聊天室demo。

1. 搭建WebSocket服务器

Go标准库没有内置WebSocket支持,需借助第三方库,推荐使用 gorilla/websocket,安装方式:

go get github.com/gorilla/websocket

创建主服务文件 main.go,初始化路由和WebSocket处理函数:

package main

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

var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool {return true // 允许跨域},}

func handleConnections(w http.ResponseWriter, r *http.Request) {// 升级HTTP连接为WebSocketws, err := upgrader.Upgrade(w, r, nil)if err != nil {log.Fatal(err)return}defer ws.Close()

// 新用户加入后,持续监听消息for {    var msg string    err := ws.ReadJSON(&msg)    if err != nil {        log.Printf("读取消息失败: %v", err)        break    }    // 广播消息给所有客户端(后续实现)    broadcastMessage(msg, ws)}

}

立即学习“go语言免费学习笔记(深入)”;

func main() {http.HandleFunc("/ws", handleConnections)http.Handle("/", http.FileServer(http.Dir("./public")))log.Println("服务器运行在 :8080")log.Fatal(http.ListenAndServe(":8080", nil))}

2. 管理客户端连接

需要一个全局结构来保存所有活跃的WebSocket连接,以便广播消息。

定义 clients map 和广播通道:

var (    clients   = make(map[*websocket.Conn]bool) // 所有连接    broadcast = make(chan string)              // 广播消息通道)

启动一个后台协程,监听广播通道并推送消息给每个客户端:

func handleMessages() {    for {        msg := <-broadcast        for client := range clients {            err := client.WriteJSON(msg)            if err != nil {                log.Printf("发送消息失败: %v", err)                client.Close()                delete(clients, client)            }        }    }}

在 main 函数中启动该协程:

go handleMessages()

修改 handleConnections,在连接建立时注册客户端:

clients[ws] = true

3. 实现消息广播

当某个客户端发来消息,将其推入 broadcast 通道:

func broadcastMessage(msg string, sender *websocket.Conn) {    broadcast <- msg}

这样 handleMessages 协程会自动将消息发送给所有在线用户。

4. 编写前端页面

在项目目录下创建 public/index.html:

  Go 聊天室  

聊天室

const ws = new WebSocket("ws://localhost:8080/ws");const chatBox = document.getElementById("chat");const input = document.getElementById("msg");

ws.onmessage = function(event) {  const div = document.createElement("div");  div.textContent = event.data;  chatBox.appendChild(div);};function send() {  if (input.value) {    ws.send(input.value);    input.value = "";  }}

以上就是Golang如何实现简单聊天室_Golang WebSocket实时通信实践的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1428238.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月16日 22:17:44
下一篇 2025年12月16日 22:17:55

相关推荐

发表回复

登录后才能评论
关注微信