0

0

Golang实现简单聊天室WebSocket服务

P粉602998670

P粉602998670

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

|

507人浏览过

|

来源于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存储连接,并用互斥锁保护并发安全。

有一导航
有一导航

有一导航延续了美国Groupon网站一贯的简约风格和购物流程,致力于打造中国本土化的精品消费限时团购网站,您会发现网站的页面非常简单,简单到每天只有一款产品。 产品通常不是实物,而是生活消费领域的各类服务型产品,比如服装、饰品、数码、化妆品、培训、健身等各类商品,用户只需在线购买,三分钟就可轻松买到超低折扣的团购产品!

下载
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页面测试聊天功能:

a style="color:#f60; text-decoration:underline;" title= "html" href="https://www.php.cn/zt/15763.html" target="_blank">html>







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

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

    热门AI工具

    更多
    DeepSeek
    DeepSeek

    幻方量化公司旗下的开源大模型平台

    豆包大模型
    豆包大模型

    字节跳动自主研发的一系列大型语言模型

    通义千问
    通义千问

    阿里巴巴推出的全能AI助手

    腾讯元宝
    腾讯元宝

    腾讯混元平台推出的AI助手

    文心一言
    文心一言

    文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

    讯飞写作
    讯飞写作

    基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

    即梦AI
    即梦AI

    一站式AI创作平台,免费AI图片和视频生成。

    ChatGPT
    ChatGPT

    最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

    相关专题

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

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

    182

    2024.02.23

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

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

    229

    2024.02.23

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

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

    343

    2024.02.23

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

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

    209

    2024.03.05

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

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

    394

    2024.05.21

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

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

    220

    2025.06.09

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

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

    193

    2025.06.10

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

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

    398

    2025.06.17

    俄罗斯Yandex引擎入口
    俄罗斯Yandex引擎入口

    2026年俄罗斯Yandex搜索引擎最新入口汇总,涵盖免登录、多语言支持、无广告视频播放及本地化服务等核心功能。阅读专题下面的文章了解更多详细内容。

    158

    2026.01.28

    热门下载

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

    相关下载

    更多

    精品课程

    更多
    相关推荐
    /
    热门推荐
    /
    最新课程
    Bootstrap 5教程
    Bootstrap 5教程

    共46课时 | 3.1万人学习

    AngularJS教程
    AngularJS教程

    共24课时 | 3.1万人学习

    CSS教程
    CSS教程

    共754课时 | 24.7万人学习

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

    Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号