基于 golang websocket 的聊天室 demo

本贴最后更新于 1903 天前,其中的信息可能已经时异事殊

测试聊天室 demo

具体代码来自github.com

main.go(入口函数)

package main  
  
import (  
  "log"  
 "net/http")  
  
func serveHome(w http.ResponseWriter, r *http.Request) {  
  if r.Method != "GET" {  
  http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)  
  return  
  }  
  http.ServeFile(w, r, "home.html")  
}  
  
func main() {  
 hub := newHub()  
  //开启一个调度器  
  go hub.run()  
  //demo页面  
  http.HandleFunc("/", serveHome)  
  //websocket 握手  
  http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {  
  serveWs(hub, w, r)  
 }) err := http.ListenAndServe(":9017", nil)  
  if err != nil {  
  log.Fatal("ListenAndServe: ", err)  
 }}

hub.go(调度代码,通过 map 对 client 的维护,以及消息的广播 )

package main  
  
import "log"  
  
type Hub struct {  
  //用于维护用户的map  
  clients map[*Client]bool  
  
  //用于广播给用户的chan  
  broadcast chan []byte  
  
  //用于用户订阅的chan  
  register chan *Client  
  
  //用于用户取消订阅的chan  
  unregister chan *Client  
}  
  
//实例化一个调度器  
func newHub() *Hub {  
  return &Hub{  
 broadcast:  make(chan []byte),  
  register:   make(chan *Client),  
  unregister: make(chan *Client),  
  clients:    make(map[*Client]bool),  
  }  
}  
  
//开始运行调度器  
func (h *Hub) run() {  
  
  for {  
  select {  
  case client := <-h.register:  
         log.Printf("客户端 %d: 订阅\n", client.id)  
 h.clients[client] = true  
  case client := <-h.unregister:  
         log.Printf("客户端 %d: 取消订阅\n", client.id)  
  if _, ok := h.clients[client]; ok {  
  delete(h.clients, client)  
  close(client.send)  
 }  case message := <-h.broadcast:  
         log.Println("通过调度器把数据塞给client的chan,然后client.writePump广播消息")  
  for client := range h.clients {  
  select {  
  case client.send <- message:  
               log.Printf("将消息发送给客户端%d\n", client.id)  
  default:  
               close(client.send)  
  delete(h.clients, client)  
 } } } }}

client.go(client 信息的发送与接收)

package main  
  
import (  
  "bytes"  
 "log" "net/http" "time"  
 "github.com/gorilla/websocket")  
  
var iii int  
  
const (  
  //写入数据允许的等待时间  
  writeWait = 10 * time.Second  
  
  //pong的周期  
  pongWait = 60 * time.Second  
  
  //ping的周期 (需要小于pong的周期)  
  pingPeriod = (pongWait * 9) / 10  
  
  //消息最大的字节数  
  maxMessageSize = 512  
)  
  
var (  
 newline = []byte{'\n'}  
 space = []byte{' '}  
)  
  
var upgrader = websocket.Upgrader{  
 ReadBufferSize:  1024,  
  WriteBufferSize: 1024,  
  //允许跨域  
  CheckOrigin: func(r *http.Request) bool {  
  return true  
  },  
}  
  
// Client is a middleman between the websocket connection and the hub.  
type Client struct {  
  //客户端的调度器  
  hub *Hub  
  
  //每个client实例化一个链接  
  conn *websocket.Conn  
  
  //从调度器中的向客户端发送数据  
  send chan []byte  
  
  //每个客户端的唯一标识  
  id int  
}  
  
//从链接中进行读取  
func (c *Client) readPump() {  
  defer func() {  
 c.hub.unregister <- c  
 c.conn.Close()  
 }()  log.Println("设置读取信息的最大值")  
 c.conn.SetReadLimit(maxMessageSize)  
 c.conn.SetReadDeadline(time.Now().Add(pongWait))  
 c.conn.SetPongHandler(func(string) error { c.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil })  
  
  for {  
  log.Println("读取连接中的信息")  
 _, message, err := c.conn.ReadMessage()  
  if err != nil {  
  if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {  
  log.Printf("error: %v", err)  
 }  return  
  }  
  log.Println("去除消息中的空格和换行")  
 message = bytes.TrimSpace(bytes.Replace(message, newline, space, -1))  
 c.hub.broadcast <- message  
 }}  
  
//从chan中获取信息进行写入  
func (c *Client) writePump() {  
 ticker := time.NewTicker(pingPeriod)  
  defer func() {  
 ticker.Stop()  
 c.conn.Close()  
 }()  for {  
  select {  
  
  case message, ok := <-c.send:  
         log.Println("writePump 接收到调度器send message")  
 c.conn.SetWriteDeadline(time.Now().Add(writeWait))  
  if !ok {  
 c.conn.WriteMessage(websocket.CloseMessage, []byte{})  
  return  
  }  
  log.Println("指定下一条发送消息的类型")  
 w, err := c.conn.NextWriter(websocket.TextMessage)  
  if err != nil {  
  return  
  }  
 w.Write(message)  
  
  // Add queued chat messages to the current websocket message.  
  n := len(c.send)  
  for i := 0; i < n; i++ {  
 w.Write(newline)  
 w.Write(<-c.send)  
 }  
  if err := w.Close(); err != nil {  
  return  
  }  
  case <-ticker.C:  
         c.conn.SetWriteDeadline(time.Now().Add(writeWait))  
  log.Println("接收到定时器的心跳ping-pong")  
  if err := c.conn.WriteMessage(websocket.PingMessage, nil); err != nil {  
  return  
  }  
 } }}  
  
//每新增一个连接初始化一个客户端进行维护  
func serveWs(hub *Hub, w http.ResponseWriter, r *http.Request) {  
  
 conn, err := upgrader.Upgrade(w, r, nil)  
  if err != nil {  
  log.Println(err)  
  return  
  }  
  //用于查看用户(测试的假数据)  
  iii++  
  log.Println("初始化一个新的客户端")  
 client := &Client{hub: hub, conn: conn, send: make(chan []byte, 256), id: iii}  
 client.hub.register <- client  
  
  //客户端写入  
  log.Println("开启client.writePump")  
  go client.writePump()  
  
  log.Println("开启client.readPump")  
  //客户端读取  
  go client.readPump()  
}
<html>
<head>
    <title>Chat Example<title>
    <script>
   window.onload = function () {  
  var conn;  
 var msg = document.getElementById("msg");  
 var log = document.getElementById("log");  
  
 function appendLog(item) {  
  var doScroll = log.scrollTop \> log.scrollHeight \- log.clientHeight \- 1;  
  log.appendChild(item);  
 if (doScroll) {  
  log.scrollTop = log.scrollHeight \- log.clientHeight;  
  }  
 }  
  document.getElementById("form").onsubmit = function () {  
  if (!conn) {  
  return false;  
  }  
  if (!msg.value) {  
  return false;  
  }  
  conn.send(msg.value);  
  msg.value = "";  
 return false;  };  
 if (window\["WebSocket"\]) {  
  conn = new WebSocket("ws://123.207.1.120:9017/ws");  
  conn.onclose = function (evt) {  
  var item = document.createElement("div");  
  item.innerHTML = "Connection closed.";  
  appendLog(item);  
  };  
  conn.onmessage = function (evt) {  
  var messages = evt.data.split('\\n');  
 for (var i = 0; i < messages.length; i++) {  
  var item = document.createElement("div");  
  item.innerText = messages\[i\];  
  appendLog(item);  
  }  
 };  
  } else {  
  var item = document.createElement("div");  
  item.innerHTML = "Your browser does not support WebSockets.";  
  appendLog(item);  
  }  
};
    </script>
<style type="text/css">
html {  
  overflow: hidden;  
}  
  
body {  
  overflow: hidden;  
  padding: 0;  
  margin: 0;  
  width: 100%;  
  height: 100%;  
  background: gray;  
}  
  
#log {  
  background: white;  
  margin: 0;  
  padding: 0.5em 0.5em 0.5em 0.5em;  
  position: absolute;  
  top: 0.5em;  
  left: 0.5em;  
  right: 0.5em;  
  bottom: 3em;  
  overflow: auto;  
}  
  
#form {  
  padding: 0 0.5em 0 0.5em;  
  margin: 0;  
  position: absolute;  
  bottom: 1em;  
  left: 0px;  
  width: 100%;  
  overflow: hidden;  
}
</style>
<body>
  <div id="log"></div>
  <form id="form">
  <input type="submit" value="Send"/>
  <input type="text" id="msg" size="64">
  </form>
</body>

</html>



  • golang

    Go 语言是 Google 推出的一种全新的编程语言,可以在不损失应用程序性能的情况下降低代码的复杂性。谷歌首席软件工程师罗布派克(Rob Pike)说:我们之所以开发 Go,是因为过去 10 多年间软件开发的难度令人沮丧。Go 是谷歌 2009 发布的第二款编程语言。

    491 引用 • 1383 回帖 • 370 关注
  • WebSocket

    WebSocket 是 HTML5 中定义的一种新协议,它实现了浏览器与服务器之间的全双工通信(full-duplex)。

    48 引用 • 206 回帖 • 407 关注

相关帖子

欢迎来到这里!

我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。

注册 关于
请输入回帖内容 ...
  • broadcast 和 client.send 这两个 chan 建议加上缓冲,不然会造成大量阻塞甚至死锁

    这个 websocket 包虽然有自动 pingpong 功能,但是建议加上客户端 ping 服务端然后重连的功能

    1 回复
  • xhaoxiong

    刚初步入手关于 golang 网络编程方面的一些知识,感谢指点

xhaoxiong
站在巨人的肩膀上学习与创新

推荐标签 标签

  • 大数据

    大数据(big data)是指无法在一定时间范围内用常规软件工具进行捕捉、管理和处理的数据集合,是需要新处理模式才能具有更强的决策力、洞察发现力和流程优化能力的海量、高增长率和多样化的信息资产。

    89 引用 • 113 回帖
  • OAuth

    OAuth 协议为用户资源的授权提供了一个安全的、开放而又简易的标准。与以往的授权方式不同之处是 oAuth 的授权不会使第三方触及到用户的帐号信息(如用户名与密码),即第三方无需使用用户的用户名与密码就可以申请获得该用户资源的授权,因此 oAuth 是安全的。oAuth 是 Open Authorization 的简写。

    36 引用 • 103 回帖 • 3 关注
  • Flume

    Flume 是一套分布式的、可靠的,可用于有效地收集、聚合和搬运大量日志数据的服务架构。

    9 引用 • 6 回帖 • 594 关注
  • NGINX

    NGINX 是一个高性能的 HTTP 和反向代理服务器,也是一个 IMAP/POP3/SMTP 代理服务器。 NGINX 是由 Igor Sysoev 为俄罗斯访问量第二的 Rambler.ru 站点开发的,第一个公开版本 0.1.0 发布于 2004 年 10 月 4 日。

    311 引用 • 546 回帖 • 58 关注
  • RESTful

    一种软件架构设计风格而不是标准,提供了一组设计原则和约束条件,主要用于客户端和服务器交互类的软件。基于这个风格设计的软件可以更简洁,更有层次,更易于实现缓存等机制。

    30 引用 • 114 回帖 • 8 关注
  • 生活

    生活是指人类生存过程中的各项活动的总和,范畴较广,一般指为幸福的意义而存在。生活实际上是对人生的一种诠释。生活包括人类在社会中与自己息息相关的日常活动和心理影射。

    228 引用 • 1450 回帖
  • TGIF

    Thank God It's Friday! 感谢老天,总算到星期五啦!

    284 引用 • 4481 回帖 • 651 关注
  • 域名

    域名(Domain Name),简称域名、网域,是由一串用点分隔的名字组成的 Internet 上某一台计算机或计算机组的名称,用于在数据传输时标识计算机的电子方位(有时也指地理位置)。

    43 引用 • 208 回帖
  • Kotlin

    Kotlin 是一种在 Java 虚拟机上运行的静态类型编程语言,由 JetBrains 设计开发并开源。Kotlin 可以编译成 Java 字节码,也可以编译成 JavaScript,方便在没有 JVM 的设备上运行。在 Google I/O 2017 中,Google 宣布 Kotlin 成为 Android 官方开发语言。

    19 引用 • 33 回帖 • 22 关注
  • BND

    BND(Baidu Netdisk Downloader)是一款图形界面的百度网盘不限速下载器,支持 Windows、Linux 和 Mac,详细介绍请看这里

    107 引用 • 1281 回帖 • 21 关注
  • 又拍云

    又拍云是国内领先的 CDN 服务提供商,国家工信部认证通过的“可信云”,乌云众测平台认证的“安全云”,为移动时代的创业者提供新一代的 CDN 加速服务。

    21 引用 • 37 回帖 • 505 关注
  • JetBrains

    JetBrains 是一家捷克的软件开发公司,该公司位于捷克的布拉格,并在俄国的圣彼得堡及美国麻州波士顿都设有办公室,该公司最为人所熟知的产品是 Java 编程语言开发撰写时所用的集成开发环境:IntelliJ IDEA

    18 引用 • 54 回帖 • 1 关注
  • GitLab

    GitLab 是利用 Ruby 一个开源的版本管理系统,实现一个自托管的 Git 项目仓库,可通过 Web 界面操作公开或私有项目。

    46 引用 • 72 回帖 • 1 关注
  • Solidity

    Solidity 是一种智能合约高级语言,运行在 [以太坊] 虚拟机(EVM)之上。它的语法接近于 JavaScript,是一种面向对象的语言。

    3 引用 • 18 回帖 • 347 关注
  • 职场

    找到自己的位置,萌新烦恼少。

    126 引用 • 1699 回帖
  • IBM

    IBM(国际商业机器公司)或万国商业机器公司,简称 IBM(International Business Machines Corporation),总公司在纽约州阿蒙克市。1911 年托马斯·沃森创立于美国,是全球最大的信息技术和业务解决方案公司,拥有全球雇员 30 多万人,业务遍及 160 多个国家和地区。

    16 引用 • 53 回帖 • 104 关注
  • 运维

    互联网运维工作,以服务为中心,以稳定、安全、高效为三个基本点,确保公司的互联网业务能够 7×24 小时为用户提供高质量的服务。

    148 引用 • 257 回帖 • 1 关注
  • GAE

    Google App Engine(GAE)是 Google 管理的数据中心中用于 WEB 应用程序的开发和托管的平台。2008 年 4 月 发布第一个测试版本。目前支持 Python、Java 和 Go 开发部署。全球已有数十万的开发者在其上开发了众多的应用。

    14 引用 • 42 回帖 • 677 关注
  • IDEA

    IDEA 全称 IntelliJ IDEA,是一款 Java 语言开发的集成环境,在业界被公认为最好的 Java 开发工具之一。IDEA 是 JetBrains 公司的产品,这家公司总部位于捷克共和国的首都布拉格,开发人员以严谨著称的东欧程序员为主。

    180 引用 • 400 回帖
  • 钉钉

    钉钉,专为中国企业打造的免费沟通协同多端平台, 阿里巴巴出品。

    15 引用 • 67 回帖 • 381 关注
  • Openfire

    Openfire 是开源的、基于可拓展通讯和表示协议 (XMPP)、采用 Java 编程语言开发的实时协作服务器。Openfire 的效率很高,单台服务器可支持上万并发用户。

    6 引用 • 7 回帖 • 87 关注
  • 深度学习

    深度学习(Deep Learning)是机器学习的分支,是一种试图使用包含复杂结构或由多重非线性变换构成的多个处理层对数据进行高层抽象的算法。

    40 引用 • 40 回帖
  • 爬虫

    网络爬虫(Spider、Crawler),是一种按照一定的规则,自动地抓取万维网信息的程序。

    106 引用 • 275 回帖
  • OpenStack

    OpenStack 是一个云操作系统,通过数据中心可控制大型的计算、存储、网络等资源池。所有的管理通过前端界面管理员就可以完成,同样也可以通过 Web 接口让最终用户部署资源。

    10 引用 • 9 关注
  • Vditor

    Vditor 是一款浏览器端的 Markdown 编辑器,支持所见即所得、即时渲染(类似 Typora)和分屏预览模式。它使用 TypeScript 实现,支持原生 JavaScript、Vue、React 和 Angular。

    308 引用 • 1658 回帖 • 1 关注
  • Redis

    Redis 是一个开源的使用 ANSI C 语言编写、支持网络、可基于内存亦可持久化的日志型、Key-Value 数据库,并提供多种语言的 API。从 2010 年 3 月 15 日起,Redis 的开发工作由 VMware 主持。从 2013 年 5 月开始,Redis 的开发由 Pivotal 赞助。

    284 引用 • 247 回帖 • 211 关注
  • 电影

    这是一个不能说的秘密。

    120 引用 • 597 回帖