golang websocket 服务端的实现

 更新时间:2019年09月04日 08:14:15   作者:小柒的另一个世界  
这篇文章主要介绍了golang 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 <- 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 StartWebsocket(addrPort string) {
  wsConnAll = make(map[int64]*wsConnection)
  http.HandleFunc("/ws", wsHandler)
  http.ListenAndServe(addrPort, nil)
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

相关文章

  • 在Go中使用JSON(附demo)

    在Go中使用JSON(附demo)

    Go开发人员经常需要处理JSON内容,本文主要介绍了在Go中使用JSON,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-02-02
  • Go编译32位GNU静态链接库的方法

    Go编译32位GNU静态链接库的方法

    Go链接库系统的难用可谓是人尽皆知,不同Go版本编译出来的不兼容,而且只支持GNU的,不能编译出Windows上的dll和lib。这篇文章给大家介绍Go编译32位GNU静态链接库的方法,感兴趣的朋友一起看看吧
    2020-05-05
  • GoLang jwt无感刷新与SSO单点登录限制解除方法详解

    GoLang jwt无感刷新与SSO单点登录限制解除方法详解

    这篇文章主要介绍了GoLang jwt无感刷新与SSO单点登录限制解除方法,JWT是一个签名的JSON对象,通常用作Oauth2的Bearer token,JWT包括三个用.分割的部分。本文将利用JWT进行认证和加密,感兴趣的可以了解一下
    2023-03-03
  • 详解Go语言中用 os/exec 执行命令的五种方法

    详解Go语言中用 os/exec 执行命令的五种方法

    这篇文章主要介绍了Go语言中用 os/exec 执行命令的五种方法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-11-11
  • go语言计算两个时间的时间差方法

    go语言计算两个时间的时间差方法

    这篇文章主要介绍了go语言计算两个时间的时间差方法,涉及Python操作时间的技巧,需要的朋友可以参考下
    2015-03-03
  • Go语言中init函数与匿名函数使用浅析

    Go语言中init函数与匿名函数使用浅析

    这篇文章主要介绍了Go语言中init函数与匿名函数使用浅析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习吧
    2023-01-01
  • Golang中omitempty关键字的具体实现

    Golang中omitempty关键字的具体实现

    本文主要介绍了Golang中omitempty关键字的具体实现,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-01-01
  • 使用Go语言简单模拟Python的生成器

    使用Go语言简单模拟Python的生成器

    这篇文章主要介绍了使用Go语言简单模拟Python的生成器,Python的generator是非常酷的功能,用Go实现的代码也较为简洁,需要的朋友可以参考下
    2015-08-08
  • go语言中匿名函数的作用域陷阱详解

    go语言中匿名函数的作用域陷阱详解

    GO语言的匿名函数(anonymous function),其实就是闭包.是指不需要定义函数名的一种函数实现方式,下面这篇文章主要给大家介绍了关于go语言中匿名函数作用域陷阱的相关资料,需要的朋友可以参考下
    2022-05-05
  • golang coroutine 的等待与死锁用法

    golang coroutine 的等待与死锁用法

    这篇文章主要介绍了golang coroutine 的等待与死锁用法详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-05-05

最新评论