gin框架中使用websocket发送消息及群聊

一、依赖包的安装

• 1、在go语言中常见的websocket包有以下两个

  • github地址,优先选择别人封装好的
  • 官方包

• 2、选择set集合包，链接地址

• 3、在gin框架中使用，链接地址

二、在gin中使用websocket

• 1、接入鉴权

  websocket也可以和普通api接口一样的做一个接口鉴权(token机制),如果验证通过可以继续往下走,没有验证不能往下走

  func Chat(ctx *gin.Context) {
       
       
  	var upGrader = websocket.Upgrader{
       
       
  		CheckOrigin: func(r *http.Request) bool {
       
       
  			// 根据鉴权的方式来处理,如果不想鉴权的就直接返回true,如果需要鉴权就要根据判断来返回true，或者false
  			return true
  		},
  	}
  }
  

• 2、实现鉴权处理

  func Chat(ctx *gin.Context) {
       
       
    // 根据url地址上获取当前用户id和token
  	userId := ctx.DefaultQuery("userId", "")
  	token := ctx.DefaultQuery("token", "")
  	userIdInt, _ := strconv.ParseInt(userId, 10, 64)
  	isValied := checkToken(userIdInt, token)
  	var upGrader = websocket.Upgrader{
       
       
  		CheckOrigin: func(r *http.Request) bool {
       
       
  			// 根据判断token的方法来鉴权,如果没token就返回false
  			return isValied
  		},
  	}
  }
  
  // 内部使用判断token合法
  func checkToken(userId int64, token string) bool {
       
       
  	...
  	return true
  }
  

• 3、将普通的get请求升级为websocket请求

  ...
  //升级get请求为webSocket协议
  conn, err := upGrader.Upgrade(ctx.Writer, ctx.Request, nil)
  if