golang websocket 服务端的实现

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《golang websocket 服务端的实现》，介绍一下服务端、websocket，希望对大家的知识积累有所帮助，助力实战开发！

创建一个websocket的服务端

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 

<p>理论要掌握，实操不能落！以上关于《golang websocket 服务端的实现》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！</p>