基于go websocket写一个聊天室

上一篇介绍了一下go websocket，这篇blog主要是通过websocket写一个聊天室。
先写前端

</style>
</head>
<body>
<div id="log"></div>
<form id="form"><input type="submit" value="Send" /><input type="text" id="msg" size="64"/>
</form>
</body>
</html>

这里面有三个组件，一个log记录，一个是msg消息发送，一个是发送按钮。下面给这个两个组件添加事件，

<script type="text/javascript">
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://" + document.location.host + "/ws");conn.onclose = function (evt) {var item = document.createElement("div");item.innerHTML = "<b>Connection closed.</b>";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 = "<b>Your browser does not support WebSockets.</b>";appendLog(item);}
};
</script>

这个里面先对发送按钮添加事件，每当按下时候执行send，将数据发送到后端；当后端发送数据到前端，就执行appendLog追加内容。好了，介绍完这个就介绍后端。后端里面先介绍主程序

func main() {flag.Parse()hub := newHub()go hub.run()http.HandleFunc("/", serveHome)http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {serveWs(hub, w, r)})err := http.ListenAndServe(*addr, nil)if err != nil {log.Fatal("ListenAndServe: ", err)}
}

这个里面hub是一个消息的分发器，就是把消息发送到每个客户端。这个里面把消息广播到每个聊天的client。

func (h *Hub) run() {for {select {case client := <-h.register:h.clients[client] = truecase client := <-h.unregister:if _, ok := h.clients[client]; ok {delete(h.clients, client)close(client.send)}case message := <-h.broadcast:for client := range h.clients {select {case client.send <- message:default:close(client.send)delete(h.clients, client)}}}}
}

还有一个是serverWs的实现

    go client.writePump()go client.readPump()

就是负责client的读和写。

func (c *Client) readPump() {defer func() {c.hub.unregister <- cc.conn.Close()}()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 {_, message, err := c.conn.ReadMessage()if err != nil {if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {log.Printf("error: %v", err)}break}message = bytes.TrimSpace(bytes.Replace(message, newline, space, -1))c.hub.broadcast <- message}
}

读很简单，通过ReadMessage读取
写也是类似，

func (c *Client) writePump() {ticker := time.NewTicker(pingPeriod)defer func() {ticker.Stop()c.conn.Close()}()for {select {case message, ok := <-c.send:c.conn.SetWriteDeadline(time.Now().Add(writeWait))if !ok {// The hub closed the channel.c.conn.WriteMessage(websocket.CloseMessage, []byte{})return}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))if err := c.conn.WriteMessage(websocket.PingMessage, []byte{}); err != nil {return}}}
}

那么这个聊天室就写好了。

基于go websocket写一个聊天室相关推荐

java 聊天室开源_用java WebSocket做一个聊天室
最近一个项目中,需要用到Java的websocket新特性,于是就学了一下,感觉这技术还挺好玩的,瞬间知道网页上面的那些在线客服是怎么做的了. 先看图: 实现了多客户机进行实时通讯. 下面看代码项目结 ...
php 利用redis写一个聊天室,使用Redis完成聊天室功能
Redis提供了Pub/Sub(发布/订阅)模式的消息机制.发布者向指定频道发布消息,订阅了该频道的订阅者就可以获取消息.通过该机制,我们可以完成聊天室.公告牌等功能. 首先,来介绍下关于pub/su ...
WebSocket实现一个聊天室
相关参考: 阮一峰老师:WebSocket 教程 MDN:WebSocket 简书:nodejs-websocket介绍文章目录一.WebSocket 介绍二.WebSocket 基本使用 2. ...
Springboot+WebSocket实现一个聊天室
具体步骤为什么需要 WebSocket? 区别引入依赖配置config 客户端客户端为什么需要 WebSocket? 因为 HTTP 协议有一个缺陷:通信只能由客户端发起举例来说,我们想了 ...
php 利用redis写一个聊天室,Redis实现多人多聊天室功能
本文为大家分享了Redis支持多人多聊天室功能的设计代码,供大家参考,具体内容如下设计原理左边的一个数据域,代表两个聊天室,聊天室id分别是827,729 在聊天室827里,有2个人,分别是jas ...
Linux下基于socket和多线程的聊天室小程序
转载:http://blog.csdn.net/robot__man/article/details/52460733 要求:基于TCP编写,一个聊天室最多100人. 客户端: 1.用户需要登录, ...
基于Node.js + WebSocket 的简易聊天室
代码地址如下: http://www.demodashi.com/demo/13282.html Node.js聊天室运行说明 Node.js的本质就是运行在服务端的JavaScript.Node.j ...
javaweb课程设计:基于websocket的网络聊天室（所有的资源和代码还有详细步骤我都会提供）
1 课程设计目的和任务本项目的是实现在web应用上进行多人聊天,为以后在大型项目中实现客服在线服务做一个测试,提前了解HTML5新特性,熟练掌握websocket技术. 2 课程设计的主要内容实现 ...
[源码和文档分享]基于Netty和WebSocket的Web聊天室
一.背景伴随着Internet的发展与宽带技术的普及,人们可以通过Internet交换动态数据,展示新产品,与人进行沟通并进行电子商务贸易.作为构成网站的重要组成部分,留言管理系统为人们的交流提供了 ...
workerman-chat（PHP开发的基于Websocket协议的聊天室框架）（thinkphp也是支持socket聊天的）...
workerman-chat(PHP开发的基于Websocket协议的聊天室框架)(thinkphp也是支持socket聊天的) 一.总结 1.下面链接里面还有一个来聊的php聊天室源码可以学习 2. ...

基于go websocket写一个聊天室

基于go websocket写一个聊天室相关推荐

最新文章

热门文章