golang语言websocket编程

2018-07-18 08:47:59   catherine

先安装依赖包：

  1. go get github.com/gorilla/websocket

server.go

  1. package main
  3. import (
  4.     "encoding/json"
  5.     "fmt"
  6.     "net/http"
  8.     "github.com/gorilla/websocket"
  9.     "github.com/satori/go.uuid"
  10. )
  12. type ClientManager struct {
  13.     clients    map[*Client]bool
  14.     broadcast  chan []byte
  15.     register   chan *Client
  16.     unRegister chan *Client
  17. }
  19. type Client struct {
  20.     id     string
  21.     socket *websocket.Conn
  22.     send   chan []byte
  23. }
  25. type Message struct {
  26.     Sender    string `json:"sender,omitempty"`
  27.     Recipient string `json:"recipient,omitempty"`
  28.     Content   string `json:"content,omitempty"`
  29. }
  31. var manager = ClientManager{
  32.     broadcast:  make(chan []byte),
  33.     register:   make(chan *Client),
  34.     unRegister: make(chan *Client),
  35.     clients:    make(map[*Client]bool),
  36. }
  38. func (manager *ClientManager) start() {
  39.     for {
  40.         select {
  41.         case conn := <-manager.register:
  42.             manager.clients[conn] = true
  43.             jsonMessage, _ := json.Marshal(&Message{Content: conn.id + "/A new socket has connected."})
  44.             manager.send(jsonMessage, conn)
  45.         case conn := <-manager.unRegister:
  46.             if _, ok := manager.clients[conn]; ok {
  47.                 close(conn.send)
  48.                 delete(manager.clients, conn)
  49.                 jsonMessage, _ := json.Marshal(&Message{Content: conn.id + "/A socket has disconnected."})
  50.                 manager.send(jsonMessage, conn)
  51.             }
  52.         case message := <-manager.broadcast:
  53.             for conn := range manager.clients {
  54.                 select {
  55.                 case conn.send <- message:
  56.                 default:
  57.                     close(conn.send)
  58.                     delete(manager.clients, conn)
  59.                 }
  60.             }
  61.         }
  62.     }
  63. }
  65. func (manager *ClientManager) send(message []byte, ignore *Client) {
  66.     for conn := range manager.clients {
  67.         if conn != ignore {
  68.             conn.send <- message
  69.         }
  70.     }
  71. }
  73. func (c *Client) read() {
  74.     defer func() {
  75.         manager.unRegister <- c
  76.         c.socket.Close()
  77.     }()
  79.     for {
  80.         _, message, err := c.socket.ReadMessage()
  81.         if err != nil {
  82.             manager.unRegister <- c
  83.             c.socket.Close()
  84.             break
  85.         }
  86.         var content map[string]string
  87.         json.Unmarshal(message, &content)
  89.         if user, ok := content["user"]; ok {
  90.             fmt.Print(user)
  91.         }
  93.         jsonMessage, _ := json.Marshal(&Message{Sender: c.id, Content: string(message)})
  94.         manager.broadcast <- jsonMessage
  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.             }
  111.             c.socket.WriteMessage(websocket.TextMessage, message)
  112.         }
  113.     }
  114. }
  116. func main() {
  117.     fmt.Println("Starting application...")
  118.     go manager.start()
  119.     http.HandleFunc("/ws", wsPage)
  120.     http.ListenAndServe(":8081", nil)
  121. }
  123. func wsPage(res http.ResponseWriter, req *http.Request) {
  124.     u := &websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}
  125.     conn, err := u.Upgrade(res, req, nil)
  126.     if err != nil {
  127.         http.NotFound(res, req)
  128.         return
  129.     }
  130.     newUUID, err := uuid.NewV4()
  131.     if err != nil {
  132.         http.Error(res, "generate  uuid error, please try again!", 900)
  133.         return
  134.     }
  135.     client := &Client{id: newUUID.String(), socket: conn, send: make(chan []byte)}
  137.     manager.register <- client
  139.     go client.read()
  140.     go client.write()
  141. }

client.go

  1. package main
  3. import (
  4.     "encoding/json"
  5.     "fmt"
  6.     "net/http"
  8.     "github.com/gorilla/websocket"
  9.     "github.com/satori/go.uuid"
  10. )
  12. type ClientManager struct {
  13.     clients    map[*Client]bool
  14.     broadcast  chan []byte
  15.     register   chan *Client
  16.     unRegister chan *Client
  17. }
  19. type Client struct {
  20.     id     string
  21.     socket *websocket.Conn
  22.     send   chan []byte
  23. }
  25. type Message struct {
  26.     Sender    string `json:"sender,omitempty"`
  27.     Recipient string `json:"recipient,omitempty"`
  28.     Content   string `json:"content,omitempty"`
  29. }
  31. var manager = ClientManager{
  32.     broadcast:  make(chan []byte),
  33.     register:   make(chan *Client),
  34.     unRegister: make(chan *Client),
  35.     clients:    make(map[*Client]bool),
  36. }
  38. func (manager *ClientManager) start() {
  39.     for {
  40.         select {
  41.         case conn := <-manager.register:
  42.             manager.clients[conn] = true
  43.             jsonMessage, _ := json.Marshal(&Message{Content: conn.id + "/A new socket has connected."})
  44.             manager.send(jsonMessage, conn)
  45.         case conn := <-manager.unRegister:
  46.             if _, ok := manager.clients[conn]; ok {
  47.                 close(conn.send)
  48.                 delete(manager.clients, conn)
  49.                 jsonMessage, _ := json.Marshal(&Message{Content: conn.id + "/A socket has disconnected."})
  50.                 manager.send(jsonMessage, conn)
  51.             }
  52.         case message := <-manager.broadcast:
  53.             for conn := range manager.clients {
  54.                 select {
  55.                 case conn.send <- message:
  56.                 default:
  57.                     close(conn.send)
  58.                     delete(manager.clients, conn)
  59.                 }
  60.             }
  61.         }
  62.     }
  63. }
  65. func (manager *ClientManager) send(message []byte, ignore *Client) {
  66.     for conn := range manager.clients {
  67.         if conn != ignore {
  68.             conn.send <- message
  69.         }
  70.     }
  71. }
  73. func (c *Client) read() {
  74.     defer func() {
  75.         manager.unRegister <- c
  76.         c.socket.Close()
  77.     }()
  79.     for {
  80.         _, message, err := c.socket.ReadMessage()
  81.         if err != nil {
  82.             manager.unRegister <- c
  83.             c.socket.Close()
  84.             break
  85.         }
  86.         var content map[string]string
  87.         json.Unmarshal(message, &content)
  89.         if user, ok := content["user"]; ok {
  90.             fmt.Print(user)
  91.         }
  93.         jsonMessage, _ := json.Marshal(&Message{Sender: c.id, Content: string(message)})
  94.         manager.broadcast <- jsonMessage
  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.             }
  111.             c.socket.WriteMessage(websocket.TextMessage, message)
  112.         }
  113.     }
  114. }
  116. func main() {
  117.     fmt.Println("Starting application...")
  118.     go manager.start()
  119.     http.HandleFunc("/ws", wsPage)
  120.     http.ListenAndServe(":8081", nil)
  121. }
  123. func wsPage(res http.ResponseWriter, req *http.Request) {
  124.     u := &websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}
  125.     conn, err := u.Upgrade(res, req, nil)
  126.     if err != nil {
  127.         http.NotFound(res, req)
  128.         return
  129.     }
  130.     newUUID, err := uuid.NewV4()
  131.     if err != nil {
  132.         http.Error(res, "generate  uuid error, please try again!", 900)
  133.         return
  134.     }
  135.     client := &Client{id: newUUID.String(), socket: conn, send: make(chan []byte)}
  137.     manager.register <- client
  139.     go client.read()
  140.     go client.write()
  141. }

index.html

  1. <html>
  2. <head>
  3. <title>Golang Chat</title>
  4. <script type="text/javascript" src="http://ajax.googleapis.com/ajax/libs/jquery/1.9.0/jquery.min.js"></script>
  5. <script type="text/javascript">
  6.     $(function() {
  7.     var conn;
  8.     var msg = $("#msg");
  9.     var log = $("#log");
  11.     function appendLog(msg) {
  12.         var d = log[0]
  13.         var doScroll = d.scrollTop == d.scrollHeight - d.clientHeight;
  14.         msg.appendTo(log)
  15.         if (doScroll) {
  16.             d.scrollTop = d.scrollHeight - d.clientHeight;
  17.         }
  18.     }
  20.     function strMapToObj(strMap) {
  21.         let obj= Object.create(null);
  22.         for (let[k,v] of strMap) {
  23.             obj[k] = v;
  24.         }
  25.         return obj;
  26.     }
  28.     $("#form").submit(function() {
  29.         if (!conn) {
  30.             return false;
  31.         }
  32.         if (!msg.val()) {
  33.             return false;
  34.         }
  35.         var Student = function () {
  36.             this.msg = msg.val();
  37.             this.user = '121212-454564';
  38.         };
  39.         conn.send(JSON.stringify(new Student));
  40.         msg.val("");
  41.         return false
  42.     });
  44.     if (window["WebSocket"]) {
  45.         conn = new WebSocket("ws://localhost:8081/ws");
  46.         conn.onerror = function (ev) {
  47.             console.log(ev)
  48.         }
  49.         conn.onclose = function(evt) {
  50.             console.log(evt)
  51.             appendLog($("<div><b>Connection Closed.</b></div>"))
  52.         }
  53.         conn.onmessage = function(evt) {
  54.             console.log(evt)
  55.             appendLog($("<div/>").text(evt.data))
  56.         }
  57.     } else {
  58.         appendLog($("<div><b>WebSockets Not Support.</b></div>"))
  59.     }
  60.     });
  61. </script>
  62. <style type="text/css">
  63. html {
  64.     overflow: hidden;
  65. }
  67. body {
  68.     overflow: hidden;
  69.     padding: 0;
  70.     margin: 0;
  71.     width: 100%;
  72.     height: 100%;
  73.     background: gray;
  74. }
  76. #log {
  77.     background: white;
  78.     margin: 0;
  79.     padding: 0.5em 0.5em 0.5em 0.5em;
  80.     position: absolute;
  81.     top: 0.5em;
  82.     left: 0.5em;
  83.     right: 0.5em;
  84.     bottom: 3em;
  85.     overflow: auto;
  86. }
  88. #form {
  89.     padding: 0 0.5em 0 0.5em;
  90.     margin: 0;
  91.     position: absolute;
  92.     bottom: 1em;
  93.     left: 0px;
  94.     width: 100%;
  95.     overflow: hidden;
  96. }
  98. </style>
  99. </head>
  100. <body>
  101. <div id="log"></div>
  102. <form id="form">
  103.     <input type="submit" value="发送" />
  104.     <input type="text" id="msg" size="64"/>
  105. </form>
  106. </body>
  107. </html>

来源：
https://blog.csdn.net/wangshubo1989/article/details/78250790

分类: web

标签:   golang   websocket  

微信号：腾讯云折扣返现

技术讨论QQ群： 863102326

作者: catherine

User Image

搜索

标签

study ab amap apache apahe awk aws bat centos CFS chrome cmd cnpm composer consul crontab css curl cygwin devops di docker docker,docker-compose ethereum excel fiddler fluentd framework front-end git gitgui github glide go golang gorm grafana gzip ioc item2 iterm2 javascript jenkins jsonp kafka laradock laravel larval linux liunux log mac mac, wi-fi macos magento mariaDB minikube mongoDB msp mysql netbeans nginx nodejs nohup npm nsq oracle php php-fpm php7 phpstorm php扩展 Protobuf python redis scp server shell soap socket socket5 sql sre ssdb ssh ssl study sublime swift system td-agent uml v2ray vagrant vagrnat vim vpn vue vue.js webpack webrtc websocket webtatic windows windows7 word wps xdebug yarn yii2 yum zookeeper 世界国家 互联网 以太坊 分类 前端 小程序 打印机 排序算法 搞笑 权限 粤语 缓存 网络 虚拟机 视频 设计模式 项目管理
热门文章
友情链接