Go如何使用websocket实现弹幕功能

下面由Golang教程栏目给大家Go使用websocket实现弹幕功能的方法，希望对需要的朋友有所帮助！

使用websocket协议，客户端发送一个消息，服务端广播到所有有效连接中。主要思路：1.封装*websocket.conn，用client结构表示一个客户端。2.维持一个map[client]bool，表示有效的客户端映射，用于广播消息3.除了处理websocket连接外，还要开启一个广播协程，监听客户端连接，断开，发弹幕事件。

推荐：《go语言教程》

主要的结构：

type Client struct{     wsConnect *websocket.Conn     inChan chan []byte     outChan chan []byte     closeChan chan byte     Name string //客户的名称     Id string //客户id，唯一     mutex sync.Mutex  // 对closeChan关闭上锁     IsClosed bool  // 防止closeChan被关闭多次 } type Message struct {     EventType byte  `json:"type"`       // 0表示用户发布消息；1表示用户进入；2表示用户退出     Name string     `json:"name"`       // 用户名称     Message string  `json:"message"`    // 消息内容 }     clients = make(map [*util.Client] bool)      // 用户组映射     join = make(chan *util.Client, 10)        // 用户加入通道     leave = make(chan *util.Client, 10)       // 用户退出通道     message = make(chan Message, 10)    // 消息通道

server端代码

package main  import (     "encoding/json"     "fmt"     "github.com/gorilla/websocket"     "goGin/server/util"     "net/http" )  var(     upgrader = websocket.Upgrader{         // 允许跨域         CheckOrigin:func(r *http.Request) bool{             return true         },     }     clients = make(map [*util.Client] bool)      // 用户组映射     join = make(chan *util.Client, 10)        // 用户加入通道     leave = make(chan *util.Client, 10)       // 用户退出通道     message = make(chan Message, 10)    // 消息通道 ) type Message struct {     EventType byte  `json:"type"`       // 0表示用户发布消息；1表示用户进入；2表示用户退出     Name string     `json:"name"`       // 用户名称     Message string  `json:"message"`    // 消息内容 }  func wsHandler(w http.ResponseWriter , r *http.Request){     var(         wsConn *websocket.Conn         err error         client *util.Client         data []byte     )     r.ParseForm() //返回一个map,并且赋值给r.Form     name := r.Form["name"][0]     id := r.Form["id"][0]      if wsConn , err = upgrader.Upgrade(w,r,nil); err != nil{         return     }      if client , err = util.InitConnection(wsConn); err != nil{         goto ERR     }     client.Id = id     client.Name = name      // 如果用户列表中没有该用户     if !clients[client] {         join <- client     }      for {         if data , err = client.ReadMessage();err != nil{ //一直读消息，没有消息就阻塞             goto ERR         }         var msg Message         msg.EventType = 0         msg.Name = client.Name         msg.Message = string(data)         message <- msg     }  ERR:     leave<-client//这个客户断开     client.Close()  }  func broadcaster() {     for {         select {         // 消息通道中有消息则执行，否则堵塞         case msg := <-message:             // 将数据编码成json形式，data是[]byte类型             // json.Marshal()只会编码结构体中公开的属性(即大写字母开头的属性)             data, err := json.Marshal(msg)             if err != nil {                 return             }             for client := range clients {                 if client.IsClosed == true {                     leave<-client//这个客户断开                     continue                 }                 // fmt.Println("=======the json message is", string(data))  // 转换成字符串类型便于查看                 if client.WriteMessage(data) != nil {                     continue //发送失败就跳过                 }             }          // 有用户加入         case client := <-join:             clients[client] = true  // 将用户加入映射             // 将用户加入消息放入消息通道             var msg Message             msg.Name = client.Name             msg.EventType = 1             msg.Message = fmt.Sprintf("%s join in, there are %d preson in room", client.Name, len(clients))             message <- msg          // 有用户退出         case client := <-leave:             // 如果该用户已经被删除             if !clients[client] {                 break             }             delete(clients, client) // 将用户从映射中删除             // 将用户退出消息放入消息通道             var msg Message             msg.Name = client.Name             msg.EventType = 2             msg.Message = fmt.Sprintf("%s leave, there are %d preson in room", client.Name, len(clients))             message <- msg         }     } }  func main(){     go broadcaster()     http.HandleFunc("/ws",wsHandler)     http.ListenAndServe("0.0.0.0:7777",nil) }

封装client

package util import (     "github.com/gorilla/websocket"     "sync"     "errors" ) type Client struct{     wsConnect *websocket.Conn     inChan chan []byte     outChan chan []byte     closeChan chan byte     Name string //客户的名称     Id string //客户id，唯一      mutex sync.Mutex  // 对closeChan关闭上锁     IsClosed bool  // 防止closeChan被关闭多次 } func InitConnection(wsConn *websocket.Conn)(conn *Client ,err error){     conn = &Client{         wsConnect:wsConn,         inChan: make(chan []byte,1000),         outChan: make(chan []byte,1000),         closeChan: make(chan byte,1),         IsClosed:false,     }     // 启动读协程     go conn.readLoop();     // 启动写协程     go conn.writeLoop();     return } func (conn *Client)ReadMessage()(data []byte , err error){     select{     case data = <- conn.inChan:     case <- conn.closeChan:         err = errors.New("connection is closeed")     }     return } func (conn *Client)WriteMessage(data []byte)(err error){     select{     case conn.outChan <- data:     case <- conn.closeChan:         err = errors.New("connection is closeed")     }     return } func (conn *Client)Close(){     // 线程安全，可多次调用     conn.wsConnect.Close()     // 利用标记，让closeChan只关闭一次     conn.mutex.Lock()     if !conn.IsClosed {         close(conn.closeChan)         conn.IsClosed = true     }     conn.mutex.Unlock() }  func (conn *Client)readLoop(){     var(         data []byte         err error     )     for{         if _, data , err = conn.wsConnect.ReadMessage(); err != nil{             goto ERR         }         //阻塞在这里，等待inChan有空闲位置         select{         case conn.inChan <- data:         case <- conn.closeChan:        // closeChan 感知 conn断开             goto ERR         }      }  ERR:     conn.Close() }  func (conn *Client)writeLoop(){     var(         data []byte         err error     )      for{         select{         case data= <- conn.outChan:         case <- conn.closeChan:             goto ERR         }         if err = conn.wsConnect.WriteMessage(websocket.TextMessage , data); err != nil{             goto ERR         }     }  ERR:     conn.Close()  }

客户端代码

<!DOCTYPE html> <html> <head>     <title>go websocket</title>     <meta charset="utf-8" /> </head> <body> <script type="text/javascript">     var wsUri ="ws://127.0.0.1:7777/ws?name=aaa&id=112";     var output;      function init() {         output = document.getElementById("output");         testWebSocket();     }      function testWebSocket() {         websocket = new WebSocket(wsUri);         websocket.onopen = function(evt) {             onOpen(evt)         };         websocket.onclose = function(evt) {             onClose(evt)         };         websocket.onmessage = function(evt) {             onMessage(evt)         };         websocket.onerror = function(evt) {             onError(evt)         };     }      function onOpen(evt) {         writeToScreen("CONNECTED");         // doSend("WebSocket rocks");     }      function onClose(evt) {         writeToScreen("DISCONNECTED");     }      function onMessage(evt) {         writeToScreen('<span style="color: blue;">RESPONSE: '+ evt.data+'</span>');         // websocket.close();     }      function onError(evt) {         writeToScreen('<span style="color: red;">ERROR:</span> '+ evt.data);     }      function doSend(message) {         // writeToScreen("SENT: " + message);         websocket.send(message);     }      function writeToScreen(message) {         var pre = document.createElement("p");         pre.style.wordWrap = "break-word";         pre.innerHTML = message;         output.appendChild(pre);     }      window.addEventListener("load", init, false);     function sendBtnClick(){         var msg = document.getElementById("input").value;         doSend(msg);         document.getElementById("input").value = '';     }     function closeBtnClick(){         websocket.close();     } </script> <h2>WebSocket Test</h2> <input type="text" id="input"></input> <button onclick="sendBtnClick()" >send</button> <button onclick="closeBtnClick()" >close</button> <div id="output"></div>  </body> </html>

以上就是Go如何使用websocket实现弹幕功能的详细内容，更多内容请关注技术你好其它相关文章！