讲师中心 微信公众号
AI工具推荐 视频效率加速

Golang实现简单聊天室WebSocket服务

酷磊大大_2469

酷磊大大_2469

发布时间:2025-09-29 12:12:02

|

527人浏览过

|

来源于php中文网

原创

答案:使用Golang和gorilla/websocket库可实现一个简单的WebSocket聊天室,通过HTTP服务升级连接、管理客户端并广播消息。首先安装gorilla/websocket库,创建/ws路由并用Upgrader升级为WebSocket连接,允许跨域;接着定义clients映射和broadcast通道,用RWMutex保证并发安全,在wsHandler中注册连接并读取消息发送至广播通道;然后在main函数中启动handleBroadcast协程,循环监听broadcast并将消息写入所有活跃连接,出错时关闭连接并从clients中删除;最后编写HTML页面测试,多个浏览器窗口可实时收发消息,完成基础聊天功能。

golang实现简单聊天室websocket服务

用Golang实现一个简单的WebSocket聊天室服务,核心在于使用gorilla/websocket库建立长连接,并通过广播机制让所有客户端实时收发消息。整个服务不需要前端框架,后端即可完成连接管理与消息分发。

1. 安装依赖并初始化WebSocket服务

Go语言中操作WebSocket最常用的库是gorilla/websocket。先通过以下命令安装:

go get github.com/gorilla/websocket

然后创建一个HTTP服务,将特定路径(如/ws)升级为WebSocket连接:

代码示例:
创建main.go文件:

package main

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

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

func wsHandler(w http.ResponseWriter, r *http.Request) {
  conn, err := upgrader.Upgrade(w, r, nil)
  if err != nil {
    log.Printf("升级失败: %v", err)
    return
  }
  defer conn.Close()
  log.Println("新客户端已连接")

  // 简单回显测试
  for {
    _, msg, err := conn.ReadMessage()
    if err != nil {
      log.Printf("读取消息错误: %v", err)
      break
    }
    log.Printf("收到消息: %s", msg)
    conn.WriteMessage(websocket.TextMessage, []byte("echo: "+string(msg)))
  }
}

func main() {
  http.HandleFunc("/ws", wsHandler)
  log.Println("服务启动在 :8080")
  log.Fatal(http.ListenAndServe(":8080", nil))
}

2. 实现客户端连接管理

聊天室需要维护多个客户端连接。可以定义一个Client结构体和一个全局的clients集合来管理活跃连接。

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

使用map[*websocket.Conn]bool存储连接,并用互斥锁保护并发安全。

github-repo-analyzer
github-repo-analyzer

当用户提及 GitHub URL,或要求分析、审查、探索或理解任意 GitHub 项目、仓库或代码库时,请使用此技能。触发条件包括以下短语:“分析这个项目”、“看看这个仓库”、“analyze this repo”、“what does this project do”,或消息中包含任何 github.com URL。

下载
var (
  clients = make(map[*websocket.Conn]bool)
  broadcast = make(chan Message)
  mu sync.RWMutex
)

type Message struct {
  Data string json:"data"
}

在wsHandler中注册新连接,并将其加入clients:

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

  mu.Lock()
  clients[conn] = true
  mu.Unlock()

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

  mu.Lock()
  delete(clients, conn)
  mu.Unlock()
}

3. 广播消息给所有在线用户

启动一个独立的广播协程,在main函数中监听broadcast通道,把接收到的消息发送给每一个当前连接的客户端。

func main() {
  http.HandleFunc("/ws", wsHandler)
  go handleBroadcast()
  log.Println("聊天室服务运行在 :8080")
  log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleBroadcast() {
  for {
    msg :=     data := []byte(msg.Data)
    mu.RLock()
    for conn := range clients {
      err := conn.WriteMessage(websocket.TextMessage, data)
      if err != nil {
        log.Printf("发送失败: %v", err)
        conn.Close()
        mu.Lock()
        delete(clients, conn)
        mu.Unlock()
      }
    }
    mu.RUnlock()
  }
}

4. 测试聊天室功能

写一个简单的HTML页面测试聊天功能:







    <script><br> const ws = new WebSocket("ws://localhost:8080/ws");<br> ws.onmessage = function(event) {<br> const li = document.createElement("li");<br> li.textContent = event.data;<br> document.getElementById("logs").appendChild(li);<br> };<br> function send() {<br> const input = document.getElementById("msg");<br> ws.send(input.value);<br> input.value = "";<br> }<br> </script>

    打开多个浏览器窗口,输入消息点击发送,所有客户端都能实时看到内容。

    基本上就这些。这个简单聊天室实现了连接管理、消息广播和基础异常处理,适合学习WebSocket通信模型。后续可扩展用户名、房间、私聊等功能。

    热门AI工具

    更多
    VibeKnow
    VibeKnow Hot

    一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

    超级麦吉

    一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

    Lovart
    Lovart Hot

    一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

    WorkBuddy

    一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

    蛙蛙写作

    一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

    Laper
    Laper Hot

    Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

    咔片AIPPT

    一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

    DeepSeek

    DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

    豆包大模型

    豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

    相关专题

    更多
    golang如何定义变量
    golang如何定义变量

    golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

    519

    2024.02.23

    golang有哪些数据转换方法
    golang有哪些数据转换方法

    golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

    616

    2024.02.23

    golang常用库有哪些
    golang常用库有哪些

    golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

    1216

    2024.02.23

    golang和python的区别是什么
    golang和python的区别是什么

    golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

    791

    2024.03.05

    golang是免费的吗
    golang是免费的吗

    golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

    1766

    2024.05.21

    golang结构体相关大全
    golang结构体相关大全

    本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

    5054

    2025.06.09

    golang相关判断方法
    golang相关判断方法

    本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

    2334

    2025.06.10

    golang数组使用方法
    golang数组使用方法

    本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

    4206

    2025.06.17

    Vite 项目如何打包和部署
    Vite 项目如何打包和部署

    Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

    60

    2026.10.28

    热门下载

    更多
    网站特效
    /
    网站源码
    /
    网站素材
    /
    前端模板

    精品课程

    更多
    相关推荐
    /
    热门推荐
    /
    最新课程
    GDB 17.2 官方文档集合
    GDB 17.2 官方文档集合

    共0课时 | 0人学习

    Bootstrap 入门安装配置
    Bootstrap 入门安装配置

    共0课时 | 0人学习

    38+ PhpStorm 提示和技巧
    38+ PhpStorm 提示和技巧

    共1课时 | 228人学习

    关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
    php中文网:公益在线php培训,帮助PHP学习者快速成长!
    关注服务号
    PHP中文网订阅号
    每天精选资源文章推送

    Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn