> ### 第四例 聊天室
* 客戶端
~~~
<!DOCTYPE HTML>
<html >
<head >
<meta charset="utf-8" >
<script src="http://libs.baidu.com/jquery/2.1.4/jquery.min.js" ></script >
</head >
<body >
<form >
<div id="div1" ></div >
<hr >
<textarea style="width: 300px;height: 100px;" id="content" ></textarea >
<input type="button" value="發送" onclick="Send()" >
</form >
</body >
<script >
function GetQueryString(name) {
var reg = new RegExp("(^|&)" + name + "=([^&]*)(&|$)");
var r = window.location.search.substr(1).match(reg);
if (r != null) return unescape(r[2]);
return null;
}
//?roomId=1&userId=1000&userName=winnie&connType=1
var roomId = GetQueryString("roomId");
var userId = GetQueryString("userId");
var userName = GetQueryString("userName");
var connType = GetQueryString("connType");
var wsCopy;
//服務端地址
//var wsUrl = "ws://127.0.0.1:8095/message?roomId=22&userId=222&userName=王五&connType=2";
var wsUrl = "ws://127.0.0.1:8095/message?roomId=" + roomId + "&userId=" + userId + "&userName=" + userName + "&connType=" + connType;
//心跳檢測
var heartTimeOut = 10 * 1000; //10s (1000 毫秒= 1 秒) - 設置時間比nginx超時時間短)
//斷線重連
var reConnTimeOut = 2 * 1000;
//心跳檢測
var heartCheck = {
timer: null,
timeout: heartTimeOut,
start: function (ws) {
this.timer = setInterval(function () {
console.log("heartCheck");
ws.send("ping www.51websocket.cn");
}, this.timeout);
}
};
//斷線重連
var reConnection = {
timer: null,
timeout: reConnTimeOut,
url: wsUrl,
start: function (ws) {
if (!ws.token) {
this.timer = setTimeout(function () {
console.log("ReConnect");
createWebSocket(this.url);
}, this.timeout);
}
}
};
//創建連接
createWebSocket(wsUrl);
function createWebSocket() {
try {
var ws = new WebSocket(wsUrl);
init(ws);
} catch (e) {
console.log('catch');
reConnection.start();
}
}
function init(ws) {
//TODO : 連接關閉時觸發
ws.onclose = function () {
clearInterval(heartCheck.timer);
if (this.token == undefined) { //多種錯誤事件, 只會觸發一個斷線重連
this.token = false;
} else if (this.token == false) {
this.token = true;
}
console.log('Close');
reConnection.start(this);
};
//通信發生錯誤時觸發
ws.onerror = function () {
clearInterval(heartCheck.timer);
if (this.token == undefined) {
this.token = false;
} else if (this.token == false) {
this.token = true;
}
console.log('Error');
reConnection.start(this);
};
//連接建立時觸發
ws.onopen = function () {
clearTimeout(reConnection.timer);
console.log('Connect');
//心跳檢測重置
heartCheck.start(this);
wsCopy = ws;
};
//客戶端接收服務端數據時觸發
ws.onmessage = function (event) {
$("#div1").append("<h3>" + event.data + "</h3>");
}
}
//向服務端發送消息
function Send() {
wsCopy.send($("#content").val());
$("#content").val("");
}
</script >
</html >
~~~
* 服務端
~~~
package main
//如果是服務掛了, 重啟的話, 重數據庫里讀取聊天信息
import (
"chatbox/zdbmodel"
"chatbox/zlogs"
"golang.org/x/net/websocket"
"net/http"
"strconv"
"fmt"
"time"
"strings"
)
//直播間信息
type LiveRoom struct {
user map[int]*UserInfo //直播間用戶
ch chan []byte //發送彈幕
msg [][]byte //歷史彈幕
}
//用戶信息
type UserInfo struct {
id int //用戶Id
name string //用戶名稱
conn *websocket.Conn //當前用戶的連接
connType int //用戶類型 1-游客 2-會員
}
//公共函數
func commAtoi(s string) (int) {
if n, err := strconv.Atoi(s); err == nil {
return n
}
return 0
}
var list map[int]*LiveRoom
var roomUser map[int]*UserInfo
var roomCh chan []byte
var roomMsg [][]byte
func svrConnHandler(conn *websocket.Conn) {
//TODO : 表單數據處理
r := conn.Request()
r.ParseForm()
roomId := r.Form["roomId"][0]
userId := r.Form["userId"][0]
userName := r.Form["userName"][0]
connType := r.Form["connType"][0]
rId := commAtoi(roomId)
uId := commAtoi(userId)
cType := commAtoi(connType)
//TODO : 構建結構體
uInfo := &UserInfo{
id: uId,
name: userName,
conn: conn,
connType: cType,
}
_, ok := list[rId]
if !ok {
roomUser = make(map[int]*UserInfo, 100)
roomCh = make(chan []byte, 100)
roomMsg = make([][]byte, 100)
}
roomUser[uId] = uInfo
list[rId] = &LiveRoom{
user: roomUser,
ch: roomCh,
msg: roomMsg,
}
//TODO : 接收客戶端消息
go func() {
var content [1024]byte
for {
connUserInfo := list[rId].user[uId]
n, err := connUserInfo.conn.Read(content[:])
if err != nil {
list[rId].user[uId].conn.Close()
delete(list[rId].user, uId)
break
}
if n > 0 {
if string(content[:n]) == string("ping www.51websocket.cn") {
content = [1024]byte{}
continue
}
content2 := append([]byte(fmt.Sprint(userId, ":::", userName, ":::")), content[:n]...)
list[rId].ch <- content2
content = [1024]byte{}
}
}
}()
//TODO : 向客戶端發送消息
if !ok {
for {
select {
case content := <-list[rId].ch:
message := strings.Split(string(content), ":::")
list[rId].msg = append(list[rId].msg, []byte("<span class='name'>"+message[1]+"："+"</span>"+message[2]))
for _, v := range list[rId].user {
v.conn.Write([]byte("<span class='name'>" + message[1] + "：" + "</span>" + message[2]))
}
}
} //<--向客戶端發送消息-->
} else {
for {
time.Sleep(time.Second * 60 * 60)
}
}
}
func main() {
list = make(map[int]*LiveRoom, 10)
zlogs.InitLog()
zdbmodel.InitDb()
http.Handle("/message", websocket.Handler(svrConnHandler))
err := http.ListenAndServe(":8095", nil)
zlogs.ErrorLog(err)
}
~~~
- 第一例 留言板
- 第二例 gRPC使用例子
- 第三例 基于go-micro做服務注冊和服務發現
- 第四例 聊天室
- 第五例 工具庫 第五例 并發安全字典
- dao
- common
- common.go
- config
- config.go
- gorm
- grom.go
- sqlx
- sqlx.go
- kafka
- kafka.go
- log
- log.go
- log2.go
- redis
- redis.go
- zookeeper
- zookeeper.go
- init
- main.go
- 第六例 原生sql操作
- 第七例 sqlx操作
- 第八例 Redis數據庫(gomodule/redigo)
- 第九例 Redis消息隊列
- 第十例 Redis集群連接
- 第十一例 Zookeeper操作
- 第十二例 Kafka操作
- 第十三例 NSQ操作
- 第十四例 二分查找
- 第十五例 交換排序 - 冒泡排序
- 第十六例 插入排序 - 直接插入排序
- 第十七例 插入排序 - 希爾排序
- 第十八例 交換排序 - 快速排序
- 第十九例 算法求解應用
- 第二十例 pprof性能分析
- 第二一例 CPU信息采集
- 第二二例 Heap信息采集
- 第二三例 Http信息采集
- 第二四例 單元測試(功能測試)
- 第二五例 基準測試(壓力測試/性能測試)
- 第二六例 gdb調試
- 第二七例 json序列化和反序列化
- 第二八例 protobuf序列化和反序列化
- 第二九例 包管理工具 go vendor
- 第三十例 包管理工具 go mod
- 第三一例 zip壓縮
- 第三二例 交叉編譯
- 第三三例 線上環境部署
- 第三四例 業務:實現固定周期維護
- 第三五例 聊天室(精簡版)
- 第三六例 并發安全字典
- 第三七例 導出Excel表格
- 第三八例 導出CSV表格
- 第三九例 聊天室(高并發)
- 第四十例 JWT (Json Web Token)
- 第四一例 雪花算法生成 Id
- 第四二例 對稱加密 AES
- 第四三例 非對稱加密 RSA
- 第四四例 簽名算法 SHA1
- 第四五例 數據庫操作 gorm
- 第四六例 數據庫操作 gorm 集合
- 數據庫連接和創建表
- 查詢 - 分頁
- 查詢所有數據
- 查詢單條數據
- 插入一條或多條數據
- 更新一條或多條數據
- 更新一條或多條數據（有零值）
- 第四七例 RSA(MD5WithRSA 算法)簽名和驗簽方式
- 第四八例 線上部署腳本
- 第四九例 Elasticsearch
- 第五十例 對象池
- 第五十一例 相關閱讀