您的位置:首页 > 教程 > 服务端类 > Golang > Go如何使用websocket实现弹幕功能

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

2020-08-28 14:22:25 来源:易采站长站 作者:

Go如何使用websocket实现弹幕功能,用户,消息,通道,客户,客户端

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

易采站长站,站长之家为您整理了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 mainimport (    "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 utilimport (    "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实现弹幕功能的详细介绍。欢迎大家对Go如何使用websocket实现弹幕功能内容提出宝贵意见

微信扫一扫

易采站长站微信账号