golang websocket 服务端

在go中创建websocket服务

基础组件 虽然golang官网提供的功能包中有websocket服务相关内容但部分功能不全所以引用第三方包
包地址 github.com/gorilla/websocket

创建一个websocket的服务端

  • websocket服务其实就是在http上升级而来许多地方与http相同
package smile

import (
    "errors"
    "log"
    "net/http"
    "sync"
    "time"

    "github.com/gorilla/websocket"
)

const (
    // 允许等待的写入时间
    writeWait = 10 * time.Second

    // Time allowed to read the next pong message from the peer.
    pongWait = 60 * time.Second

    // Send pings to peer with this period. Must be less than pongWait.
    pingPeriod = (pongWait * 9) / 10

    // Maximum message size allowed from peer.
    maxMessageSize = 512
)

// 最大的连接ID,每次连接都加1 处理
var maxConnId int64

// 客户端读写消息
type wsMessage struct {
    // websocket.TextMessage 消息类型
    messageType int
    data        []byte
}

// ws 的所有连接
// 用于广播
var wsConnAll map[int64]*wsConnection

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    // 允许所有的CORS 跨域请求,正式环境可以关闭
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

// 客户端连接
type wsConnection struct {
    wsSocket *websocket.Conn // 底层websocket
    inChan   chan *wsMessage // 读队列
    outChan  chan *wsMessage // 写队列

    mutex     sync.Mutex // 避免重复关闭管道,加锁处理
    isClosed  bool
    closeChan chan byte // 关闭通知
    id        int64
}

func wsHandler(resp http.ResponseWriter, req *http.Request) {
    // 应答客户端告知升级连接为websocket
    wsSocket, err := upgrader.Upgrade(resp, req, nil)
    if err != nil {
        log.Println("升级为websocket失败", err.Error())
        return
    }
    maxConnId++
    // TODO 如果要控制连接数可以计算,wsConnAll长度
    // 连接数保持一定数量,超过的部分不提供服务
    wsConn := &wsConnection{
        wsSocket:  wsSocket,
        inChan:    make(chan *wsMessage, 1000),
        outChan:   make(chan *wsMessage, 1000),
        closeChan: make(chan byte),
        isClosed:  false,
        id:        maxConnId,
    }
    wsConnAll[maxConnId] = wsConn
    log.Println("当前在线人数", len(wsConnAll))

    // 处理器,发送定时信息,避免意外关闭
    go wsConn.processLoop()
    // 读协程
    go wsConn.wsReadLoop()
    // 写协程
    go wsConn.wsWriteLoop()
}

// 处理队列中的消息
func (wsConn *wsConnection) processLoop() {
    // 处理消息队列中的消息
    // 获取到消息队列中的消息,处理完成后,发送消息给客户端
    for {
        msg, err := wsConn.wsRead()
        if err != nil {
            log.Println("获取消息出现错误", err.Error())
            break
        }
        log.Println("接收到消息", string(msg.data))
        // 修改以下内容把客户端传递的消息传递给处理程序
        err = wsConn.wsWrite(msg.messageType, msg.data)
        if err != nil {
            log.Println("发送消息给客户端出现错误", err.Error())
            break
        }
    }
}

// 处理消息队列中的消息
func (wsConn *wsConnection) wsReadLoop() {
    // 设置消息的最大长度
    wsConn.wsSocket.SetReadLimit(maxMessageSize)
    wsConn.wsSocket.SetReadDeadline(time.Now().Add(pongWait))
    for {
        // 读一个message
        msgType, data, err := wsConn.wsSocket.ReadMessage()
        if err != nil {
            websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure)
            log.Println("消息读取出现错误", err.Error())
            wsConn.close()
            return
        }
        req := &wsMessage{
            msgType,
            data,
        }
        // 放入请求队列,消息入栈
        select {
        case wsConn.inChan <- req:
        case <-wsConn.closeChan:
            return
        }
    }
}

// 发送消息给客户端
func (wsConn *wsConnection) wsWriteLoop() {
    ticker := time.NewTicker(pingPeriod)
    defer func() {
        ticker.Stop()
    }()
    for {
        select {
        // 取一个应答
        case msg := <-wsConn.outChan:
            // 写给websocket
            if err := wsConn.wsSocket.WriteMessage(msg.messageType, msg.data); err != nil {
                log.Println("发送消息给客户端发生错误", err.Error())
                // 切断服务
                wsConn.close()
                return
            }
        case <-wsConn.closeChan:
            // 获取到关闭通知
            return
        case <-ticker.C:
            // 出现超时情况
            wsConn.wsSocket.SetWriteDeadline(time.Now().Add(writeWait))
            if err := wsConn.wsSocket.WriteMessage(websocket.PingMessage, nil); err != nil {
                return
            }
        }
    }
}

// 写入消息到队列中
func (wsConn *wsConnection) wsWrite(messageType int, data []byte) error {
    select {
    case wsConn.outChan <- &wsMessage{messageType, data}:
    case <-wsConn.closeChan:
        return errors.New("连接已经关闭")
    }
    return nil
}

// 读取消息队列中的消息
func (wsConn *wsConnection) wsRead() (*wsMessage, error) {
    select {
    case msg := <-wsConn.inChan:
        // 获取到消息队列中的消息
        return msg, nil
    case <-wsConn.closeChan:

    }
    return nil, errors.New("连接已经关闭")
}

// 关闭连接
func (wsConn *wsConnection) close() {
    log.Println("关闭连接被调用了")
    wsConn.wsSocket.Close()
    wsConn.mutex.Lock()
    defer wsConn.mutex.Unlock()
    if wsConn.isClosed == false {
        wsConn.isClosed = true
        // 删除这个连接的变量
        delete(wsConnAll, wsConn.id)
        close(wsConn.closeChan)
    }
}

// 对所有用户进行广播
func broadcastUsers(messageType int, data string) {
    for _, ws := range wsConnAll {
        ws.wsWrite(messageType, []byte(data))
    }
}

// 启动程序
func StartWebsocket(addrPort string) {
    wsConnAll = make(map[int64]*wsConnection)
    http.HandleFunc("/ws", wsHandler)
    http.ListenAndServe(addrPort, nil)
}

启动调用

// broadcastUsers 这个方法是对所有连接进行广播
smile.StartWebsocket("0.0.0.0:8080")
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 195,898评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,401评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,058评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,539评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,382评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,319评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,706评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,370评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,664评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,715评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,476评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,326评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,730评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,003评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,275评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,683评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,877评论 2 335

推荐阅读更多精彩内容