这次的就直接发放代码截图吧，应该是用go语言做后台一个简易的聊天，这里没用到什么特别的知识，最朴实的来实现效果，主要目的是分享下h5怎么用websocket，go搭建websocket服务的主要部分。
go代码部分：
1 // WebChat project main.go
2 package main
3
4 import (
5 "fmt"
6 "net/http"
7 "time"
8
9 "encoding/json"
10
11 "strings"
12
13 "golang.org/x/net/websocket"
14 )
15
16 //全局信息
17 var datas Datas
18 var users map[*websocket.Conn]string
19
20 func main() {
21 fmt.Println("启动时间")
22 fmt.Println(time.Now())
23
24 //初始化
25 datas = Datas{}
26 users = make(map[*websocket.Conn]string)
27
28 //绑定效果页面
29 http.HandleFunc("/", h_index)
30 //绑定socket方法
31 http.Handle("/webSocket", websocket.Handler(h_webSocket))
32 //开始监听
33 http.ListenAndServe(":8", nil)
34 }
35
36 func h_index(w http.ResponseWriter, r *http.Request) {
37
38 http.ServeFile(w, r, "index.html")
39 }
40
41 func h_webSocket(ws *websocket.Conn) {
42
43 var userMsg UserMsg
44 var data string
45 for {
46
47 //判断是否重复连接
48 if _, ok := users[ws]; !ok {
49 users[ws] = "匿名"
50 }
51 userMsgsLen := len(datas.UserMsgs)
52 fmt.Println("UserMsgs", userMsgsLen, "users长度：", len(users))
53
54 //有消息时，全部分发送数据
55 if userMsgsLen > 0 {
56 b, errMarshl := json.Marshal(datas)
57 if errMarshl != nil {
58 fmt.Println("全局消息内容异常...")
59 break
60 }
61 for key, _ := range users {
62 errMarshl = websocket.Message.Send(key, string(b))
63 if errMarshl != nil {
64 //移除出错的链接
65 delete(users, key)
66 fmt.Println("发送出错...")
67 break
68 }
69 }
70 datas.UserMsgs = make([]UserMsg, 0)
71 }
72
73 fmt.Println("开始解析数据...")
74 err := websocket.Message.Receive(ws, &data)
75 fmt.Println("data：", data)
76 if err != nil {
77 //移除出错的链接
78 delete(users, ws)
79 fmt.Println("接收出错...")
80 break
81 }
82
83 data = strings.Replace(data, "\n", "", 0)
84 err = json.Unmarshal([]byte(data), &userMsg)
85 if err != nil {
86 fmt.Println("解析数据异常...")
87 break
88 }
89 fmt.Println("请求数据类型：", userMsg.DataType)
90
91 switch userMsg.DataType {
92 case "send":
93 //赋值对应的昵称到ws
94 if _, ok := users[ws]; ok {
95 users[ws] = userMsg.UserName
96
97 //清除连接人昵称信息
98 datas.UserDatas = make([]UserData, 0)
99 //重新加载当前在线连接人
100 for _, item := range users {
101
102 userData := UserData{UserName: item}
103 datas.UserDatas = append(datas.UserDatas, userData)
104 }
105 }
106 datas.UserMsgs = append(datas.UserMsgs, userMsg)
107 }
108 }
109
110 }
111
112 type UserMsg struct {
113 UserName string
114 Msg string
115 DataType string
116 }
117
118 type UserData struct {
119 UserName string
120 }
121
122 type Datas struct {
123 UserMsgs []UserMsg
124 UserDatas []UserData
125 }
html代码部分：
1 <!DOCTYPE html>
2 <html lang="zh-CN">
3 <head>
4 <title></title>
5 <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1, user-scalable=no">
6 <!-- 新 Bootstrap 核心 CSS 文件 -->
7 <link rel="stylesheet" href="//cdn.bootcss.com/bootstrap/3.3.5/css/bootstrap.min.css">
8 <script src="//cdn.bootcss.com/jquery/1.11.3/jquery.min.js"></script>
9 <!-- <script src="//cdn.bootcss.com/bootstrap/3.3.5/js/bootstrap.min.js"></script>-->
10 </head>
11 <body>
12 <div class="container">
13 <div>内容：</div>
14 <div class="list-group" id="divShow">
15 <!--<div class="list-group-item list-group-item-success">1111</div>
16 <div class="list-group-item list-group-item-info">1111</div>
17 <div class="list-group-item list-group-item-warning">1111</div>
18 <div class="list-group-item list-group-item-danger">1111</div>-->
19 </div>
20 <div class="list-group" id="divUsers">
21 在线：<br />
22 <!--<div class="btn btn-default">111</div>-->
23
24 </div>
25 <div>
26 昵称：<input class="form-control" id="txtUserName" value="红领巾" type="text" maxlength="20" style="width: 30%; margin-bottom: 15px" />
27 聊聊：<textarea class="form-control" id="txtContent" autofocus rows="6" placeholder="想聊的内容" maxlength="200" required style="width: 60%; "></textarea>
28 <button class="btn btn-default" id="btnSend" style="margin-top:15px">发 送</button>
29 </div>
30 </div>
31 </body>
32 </html>
33
34 <script>
35
36 var tool = function () {
37
38 var paperLoopNum = 0;
39 var paperTempleArr = [
40 ‘<div class="list-group-item list-group-item-success">{0}</div>‘,
41 ‘<div class="list-group-item list-group-item-info">{0}</div>‘,
42 ‘<div class="list-group-item list-group-item-warning">{0}</div>‘,
43 ‘<div class="list-group-item list-group-item-danger">{0}</div>‘
44 ];
45
46 return {
47
48 paperDiv: function (val) {
49
50 var hl = paperTempleArr[paperLoopNum];
51 paperLoopNum++;
52 if (paperLoopNum >= paperTempleArr.length) { paperLoopNum = 0; }
53
54 return this.formart(hl, [val])
55 },
56 formart: function (str, arrVal) {
57
58 for (var i = 0; i < arrVal.length; i++) {
59 str = str.replace("{" + i + "}", arrVal[i]);
60 }
61 return str;
62 }
63 }
64 }
65
66 function showMsg(id, hl, isAppend) {
67
68 if (!isAppend) { $("#" + id).html(hl); } else {
69 $("#" + id).append(hl);
70 }
71 }
72
73 $(function () {
74
75 //初始化工具方法
76 var tl = new tool();
77
78 var wsUrl = "ws://172.16.9.6:8/webSocket";
79 ws = new WebSocket(wsUrl);
80
81 try {
82
83 ws.onopen = function () {
84
85 //showMsg("divShow", tl.paperDiv("连接服务器-成功"));
86 }
87
88 ws.onclose = function () {
89 if (ws) {
90 ws.close();
91 ws = null;
92 }
93 showMsg("divShow", tl.paperDiv("连接服务器-关闭"), true);
94 }
95
96 ws.onmessage = function (result) {
97
98 //console.log(result.data);
99 var data = JSON.parse(result.data);
100 $(data.UserMsgs).each(function (i, item) {
101 showMsg("divShow", tl.paperDiv("【" + item.UserName + "】：" + item.Msg), true);
102 });
103
104 var userDataShow = [];
105 $(data.UserDatas).each(function (i, item) {
106
107 userDataShow.push(‘<div class="btn btn-default">‘ + item.UserName + ‘</div>‘);
108
109 });
110 showMsg("divUsers", userDataShow.join(‘‘), false);
111 }
112
113 ws.onerror = function () {
114 if (ws) {
115 ws.close();
116 ws = null;
117 }
118 showMsg("divShow", tl.paperDiv("连接服务器-关闭"), true);
119 }
120
121 } catch (e) {
122
123 alert(e.message);
124 }
125 $("#btnSend").on("click", function () {
126
127 var tContentObj = $("#txtContent");
128 var tContent = $.trim( tContentObj.val()).replace("/[\n]/g", "");
129 var tUserName = $.trim( $("#txtUserName").val()); tUserName = tUserName.length <= 0 ? "匿名" : tUserName;
130 if (tContent.length <= 0 || $.trim(tContent).length <= 0) { alert("请输入发送内容!"); return; }
131 if (ws == null) { alert("连接失败，请F5刷新页面!"); return; }
132
133 var request = tl.formart(‘{"UserName": "{0}", "DataType": "{1}", "Msg": "{2}" }‘,
134 [tUserName, "send", tContent]);
135 ws.send(request);
136 tContentObj.val("");
137 tContentObj.val($.trim(tContentObj.val()).replace("/[\n]/g", ""));
138 });
139 $("#txtContent").on("keydown", function (event) {
140
141 if (event.keyCode == 13) {
142
143 $("#btnSend").trigger("click");
144 }
145 });
146 })
147
148 </script>
效果图：
主要的备注都写在代码里面了，希望更多的朋友相互分享交流。
时间： 2024-09-29 19:38:52