11.3 gin websocket 一对一聊天

依赖包

  1. github.com/gin-gonic/gin
  2. github.com/gorilla/websocket

代码

创建ws/ws.go

  1. package ws
  3. import (
  4.     "encoding/json"
  5.     "github.com/gin-gonic/gin"
  6.     "github.com/gorilla/websocket"
  7.     "log"
  8.     "net/http"
  9. )
  11. // ClientManager is a websocket manager
  12. type ClientManager struct {
  13.     Clients    map[string]*Client
  14.     Broadcast  chan []byte
  15.     Register   chan *Client
  16.     Unregister chan *Client
  17. }
  19. // Client is a websocket client
  20. type Client struct {
  21.     ID     string
  22.     Socket *websocket.Conn
  23.     Send   chan []byte
  24. }
  26. // Message is return msg
  27. type Message struct {
  28.     Sender    string `json:"sender,omitempty"`
  29.     Recipient string `json:"recipient,omitempty"`
  30.     Content   string `json:"content,omitempty"`
  31. }
  33. // Manager define a ws server manager
  34. var Manager = ClientManager{
  35.     Broadcast:  make(chan []byte),
  36.     Register:   make(chan *Client),
  37.     Unregister: make(chan *Client),
  38.     Clients:    make(map[string]*Client),
  39. }
  41. // Start is  项目运行前, 协程开启start -> go Manager.Start()
  42. func (manager *ClientManager) Start() {
  43.     for {
  44.         log.Println("<---管道通信--->")
  45.         select {
  46.         case conn := <-Manager.Register:
  47.             log.Printf("新用户加入:%v", conn.ID)
  48.             Manager.Clients[conn.ID] = conn
  49.             jsonMessage, _ := json.Marshal(&Message{Content: "Successful connection to socket service"})
  50.             conn.Send <- jsonMessage
  51.         case conn := <-Manager.Unregister:
  52.             log.Printf("用户离开:%v", conn.ID)
  53.             if _, ok := Manager.Clients[conn.ID]; ok {
  54.                 jsonMessage, _ := json.Marshal(&Message{Content: "A socket has disconnected"})
  55.                 conn.Send <- jsonMessage
  56.                 close(conn.Send)
  57.                 delete(Manager.Clients, conn.ID)
  58.             }
  59.         case message := <-Manager.Broadcast:
  60.             MessageStruct :=Message{}
  61.             json.Unmarshal(message, &MessageStruct)
  62.             for id, conn := range Manager.Clients {
  63.                 if id!=creatId(MessageStruct.Recipient,MessageStruct.Sender){
  64.                     continue
  65.                 }
  66.                 select {
  67.                 case conn.Send <- message:
  68.                 default:
  69.                     close(conn.Send)
  70.                     delete(Manager.Clients, conn.ID)
  71.                 }
  72.             }
  73.         }
  74.     }
  75. }
  76. func creatId(uid,touid string) string {
  77.     return uid+"_"+touid
  78. }
  79. func (c *Client) Read() {
  80.     defer func() {
  81.         Manager.Unregister <- c
  82.         c.Socket.Close()
  83.     }()
  85.     for {
  86.         c.Socket.PongHandler()
  87.         _, message, err := c.Socket.ReadMessage()
  88.         if err != nil {
  89.             Manager.Unregister <- c
  90.             c.Socket.Close()
  91.             break
  92.         }
  93.         log.Printf("读取到客户端的信息:%s", string(message))
  94.         Manager.Broadcast <- message
  95.     }
  96. }
  98. func (c *Client) Write() {
  99.     defer func() {
  100.         c.Socket.Close()
  101.     }()
  103.     for {
  104.         select {
  105.         case message, ok := <-c.Send:
  106.             if !ok {
  107.                 c.Socket.WriteMessage(websocket.CloseMessage, []byte{})
  108.                 return
  109.             }
  110.             log.Printf("发送到到客户端的信息:%s", string(message))
  112.             c.Socket.WriteMessage(websocket.TextMessage, message)
  113.         }
  114.     }
  115. }
  117. //TestHandler socket 连接 中间件 作用:升级协议,用户验证,自定义信息等
  118. func WsHandler(c *gin.Context) {
  119.     uid := c.Query("uid")
  120.     touid := c.Query("to_uid")
  121.     conn, err := (&websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}).Upgrade(c.Writer, c.Request, nil)
  122.     if err != nil {
  123.         http.NotFound(c.Writer, c.Request)
  124.         return
  125.     }
  126.     //可以添加用户信息验证
  127.     client := &Client{
  128.         ID:    creatId(uid,touid),
  129.         Socket: conn,
  130.         Send:   make(chan []byte),
  131.     }
  132.     Manager.Register <- client
  133.     go client.Read()
  134.     go client.Write()
  135. }

创建main.go

  1. package main
  3. import (
  4.     "github.com/gin-gonic/gin"
  5.     "im/ws"
  6. )
  8. //server
  9. func main() {
  11.     gin.SetMode(gin.ReleaseMode) //线上环境
  13.     go ws.Manager.Start()
  14.     r := gin.Default()
  15.     r.GET("/ws",ws.WsHandler)
  16.     r.GET("/pong", func(c *gin.Context) {
  17.         c.JSON(200, gin.H{
  18.             "message": "pong",
  19.         })
  20.     })
  21.     r.Run(":8282") // listen and serve on 0.0.0.0:8080
  22. }

运行服务端

  1. go  run  mian.go

创建client.html

  1. <html>
  2. <head>
  3.     <title>Golang Chat</title>
  5. <script src="http://libs.baidu.com/jquery/1.4.2/jquery.min.js"></script>
  6. <meta charset="UTF-8" />
  7.     <script type="text/javascript">
  9.         $(function() {
  10.             function getUrlParam(name)
  11.             {
  12.                 var reg = new RegExp("(^|&)"+ name +"=([^&]*)(&|$)"); //构造一个含有目标参数的正则表达式对象
  13.                 var r = window.location.search.substr(1).match(reg);  //匹配目标参数
  14.                 if (r!=null) return unescape(r[2]); return null; //返回参数值
  15.             }
  16.             var conn;
  17.             var msg = $("#msg");
  18.             var log = $("#log");
  19.             uid=getUrlParam("uid");
  20.             to_uid=getUrlParam("to_uid");
  22.             function appendLog(msg) {
  23.                 var d = log[0]
  24.                 var doScroll = d.scrollTop == d.scrollHeight - d.clientHeight;
  25.                 msg.appendTo(log)
  26.                 if (doScroll) {
  27.                     d.scrollTop = d.scrollHeight - d.clientHeight;
  28.                 }
  29.             }
  31.             $("#form").submit(function() {
  32.                 if (!conn) {
  33.                     return false;
  34.                 }
  35.                 if (!msg.val()) {
  36.                     return false;
  37.                 }
  38.                 var json = {"sender":uid,"recipient":to_uid,"content":msg.val()}; //创建对象；
  39.                 var jsonStr = JSON.stringify(json);       //转为JSON字符串
  41.                 conn.send(jsonStr);
  42.                 msg.val("");
  43.                 return false
  44.             });
  46.             if (window["WebSocket"]) {
  47.                 conn = new WebSocket("ws://localhost:8282/ws?uid="+uid+"&to_uid="+to_uid);
  48.                 conn.onclose = function(evt) {
  49.                     appendLog($("<div><b>Connection Closed.</b></div>"))
  50.                 }
  51.                 conn.onmessage = function(evt) {
  53.                     appendLog($("<div/>").text(evt.data))
  54.                 }
  55.             } else {
  56.                 appendLog($("<div><b>WebSockets Not Support.</b></div>"))
  57.             }
  58.         });
  59.     </script>
  60.     <style type="text/css">
  61.         html {
  62.             overflow: hidden;
  63.         }
  65.         body {
  66.             overflow: hidden;
  67.             padding: 0;
  68.             margin: 0;
  69.             width: 100%;
  70.             height: 100%;
  71.             background: gray;
  72.         }
  74.         #log {
  75.             background: white;
  76.             margin: 0;
  77.             padding: 0.5em 0.5em 0.5em 0.5em;
  78.             position: absolute;
  79.             top: 0.5em;
  80.             left: 0.5em;
  81.             right: 0.5em;
  82.             bottom: 3em;
  83.             overflow: auto;
  84.         }
  86.         #form {
  87.             padding: 0 0.5em 0 0.5em;
  88.             margin: 0;
  89.             position: absolute;
  90.             bottom: 1em;
  91.             left: 0px;
  92.             width: 100%;
  93.             overflow: hidden;
  94.         }
  96.     </style>
  97. </head>
  98. <body>
  99. <div id="log"></div>
  100. <form id="form">
  101.     <input type="submit" value="发送" />
  102.     <input type="text" id="msg" size="64"/>
  103. </form>
  104. </body>
  105. </html>

自己搭建nginx或apache等web服务，分别在两个窗口运行

http://localhost/client.html?uid=1&to_uid=2

http://localhost/client.html?uid=2&to_uid=1

这样就可以聊天了

links

  • 目录
  • 上一节：
  • 下一节：