基于 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 回帖 • 408 关注

相关帖子

欢迎来到这里!

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

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

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

    1 回复
  • xhaoxiong

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

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

推荐标签 标签

  • IPFS

    IPFS(InterPlanetary File System,星际文件系统)是永久的、去中心化保存和共享文件的方法,这是一种内容可寻址、版本化、点对点超媒体的分布式协议。请浏览 IPFS 入门笔记了解更多细节。

    20 引用 • 245 回帖 • 232 关注
  • FFmpeg

    FFmpeg 是一套可以用来记录、转换数字音频、视频,并能将其转化为流的开源计算机程序。

    22 引用 • 31 回帖 • 12 关注
  • WordPress

    WordPress 是一个使用 PHP 语言开发的博客平台,用户可以在支持 PHP 和 MySQL 数据库的服务器上架设自己的博客。也可以把 WordPress 当作一个内容管理系统(CMS)来使用。WordPress 是一个免费的开源项目,在 GNU 通用公共许可证(GPLv2)下授权发布。

    45 引用 • 113 回帖 • 321 关注
  • Love2D

    Love2D 是一个开源的, 跨平台的 2D 游戏引擎。使用纯 Lua 脚本来进行游戏开发。目前支持的平台有 Windows, Mac OS X, Linux, Android 和 iOS。

    14 引用 • 53 回帖 • 506 关注
  • 小薇

    小薇是一个用 Java 写的 QQ 聊天机器人 Web 服务,可以用于社群互动。

    由于 Smart QQ 从 2019 年 1 月 1 日起停止服务,所以该项目也已经停止维护了!

    34 引用 • 467 回帖 • 689 关注
  • PHP

    PHP(Hypertext Preprocessor)是一种开源脚本语言。语法吸收了 C 语言、 Java 和 Perl 的特点,主要适用于 Web 开发领域,据说是世界上最好的编程语言。

    164 引用 • 406 回帖 • 524 关注
  • 锤子科技

    锤子科技(Smartisan)成立于 2012 年 5 月,是一家制造移动互联网终端设备的公司,公司的使命是用完美主义的工匠精神,打造用户体验一流的数码消费类产品(智能手机为主),改善人们的生活质量。

    4 引用 • 31 回帖 • 6 关注
  • FlowUs

    FlowUs.息流 个人及团队的新一代生产力工具。

    让复杂的信息管理更轻松、自由、充满创意。

    1 引用 • 2 关注
  • CAP

    CAP 指的是在一个分布式系统中, Consistency(一致性)、 Availability(可用性)、Partition tolerance(分区容错性),三者不可兼得。

    11 引用 • 5 回帖 • 552 关注
  • WebClipper

    Web Clipper 是一款浏览器剪藏扩展,它可以帮助你把网页内容剪藏到本地。

    3 引用 • 9 回帖 • 2 关注
  • IDEA

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

    180 引用 • 400 回帖
  • 互联网

    互联网(Internet),又称网际网络,或音译因特网、英特网。互联网始于 1969 年美国的阿帕网,是网络与网络之间所串连成的庞大网络,这些网络以一组通用的协议相连,形成逻辑上的单一巨大国际网络。

    96 引用 • 330 回帖
  • 大疆创新

    深圳市大疆创新科技有限公司(DJI-Innovations,简称 DJI),成立于 2006 年,是全球领先的无人飞行器控制系统及无人机解决方案的研发和生产商,客户遍布全球 100 多个国家。通过持续的创新,大疆致力于为无人机工业、行业用户以及专业航拍应用提供性能最强、体验最佳的革命性智能飞控产品和解决方案。

    2 引用 • 14 回帖 • 2 关注
  • Facebook

    Facebook 是一个联系朋友的社交工具。大家可以通过它和朋友、同事、同学以及周围的人保持互动交流,分享无限上传的图片,发布链接和视频,更可以增进对朋友的了解。

    4 引用 • 15 回帖 • 448 关注
  • Kubernetes

    Kubernetes 是 Google 开源的一个容器编排引擎,它支持自动化部署、大规模可伸缩、应用容器化管理。

    108 引用 • 54 回帖
  • API

    应用程序编程接口(Application Programming Interface)是一些预先定义的函数,目的是提供应用程序与开发人员基于某软件或硬件得以访问一组例程的能力,而又无需访问源码,或理解内部工作机制的细节。

    76 引用 • 421 回帖
  • Gitea

    Gitea 是一个开源社区驱动的轻量级代码托管解决方案,后端采用 Go 编写,采用 MIT 许可证。

    4 引用 • 16 回帖 • 7 关注
  • CloudFoundry

    Cloud Foundry 是 VMware 推出的业界第一个开源 PaaS 云平台,它支持多种框架、语言、运行时环境、云平台及应用服务,使开发人员能够在几秒钟内进行应用程序的部署和扩展,无需担心任何基础架构的问题。

    5 引用 • 18 回帖 • 151 关注
  • 微服务

    微服务架构是一种架构模式,它提倡将单一应用划分成一组小的服务。服务之间互相协调,互相配合,为用户提供最终价值。每个服务运行在独立的进程中。服务于服务之间才用轻量级的通信机制互相沟通。每个服务都围绕着具体业务构建,能够被独立的部署。

    96 引用 • 155 回帖
  • 笔记

    好记性不如烂笔头。

    303 引用 • 777 回帖
  • 数据库

    据说 99% 的性能瓶颈都在数据库。

    330 引用 • 612 回帖
  • RESTful

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

    30 引用 • 114 回帖 • 8 关注
  • Scala

    Scala 是一门多范式的编程语言,集成面向对象编程和函数式编程的各种特性。

    13 引用 • 11 回帖 • 101 关注
  • 单点登录

    单点登录(Single Sign On)是目前比较流行的企业业务整合的解决方案之一。SSO 的定义是在多个应用系统中,用户只需要登录一次就可以访问所有相互信任的应用系统。

    9 引用 • 25 回帖 • 9 关注
  • OpenStack

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

    10 引用 • 9 关注
  • Bug

    Bug 本意是指臭虫、缺陷、损坏、犯贫、窃听器、小虫等。现在人们把在程序中一些缺陷或问题统称为 bug(漏洞)。

    76 引用 • 1738 回帖 • 3 关注
  • 电影

    这是一个不能说的秘密。

    120 引用 • 597 回帖 • 1 关注