gin：结合gorilla实现webSocket

Love The Way You Lie 2022-04-24 11:40 204阅读 0赞

gin框架写rest接口特别6，性能也很好。  
本文介绍一下gin和gorilla结合创建websocket

### 文章目录 ###

*  1. gin
 *  2. gorilla
 *  3. js代码
 *  5. 测试

# 1. gin #

github地址：https://github.com/gin-gonic/gin

安装gin的代码包

> go get github.com/gin-gonic/gin

package main  
    
    import (  
       "github.com/gin-gonic/gin"  
    )  
    
    func main() {
          
       bindAddress := "localhost:2303"  
       r := gin.Default()  
       //监听 get请求  /test路径
       r.GET("/test", func(c *gin.Context) {
          
          c.JSON(200,[]string{
        "123", "321"})  
       })  
       r.Run(bindAddress)  
    }

上面是个简单的gin请求例子，更多使用方式请查看官方介绍

# 2. gorilla #

gorilla拥有很多类库，这次我们使用的是它的webSocket

go get github.com/gorilla/websocket
    1
    package main  
    
    import (  
       "github.com/gin-gonic/gin"  
       "github.com/gorilla/websocket" 
       "net/http"
    )  
    
    
    var upGrader = websocket.Upgrader{
          
       CheckOrigin: func (r *http.Request) bool {
          
          return true  
       },  
    }  
    
    //webSocket请求ping 返回pong  
    func ping(c *gin.Context) {
          
       //升级get请求为webSocket协议
       ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)  
       if err != nil {
          
          return  
       }  
       defer ws.Close()  
       for {
        
          //读取ws中的数据  
          mt, message, err := ws.ReadMessage()  
          if err != nil {
          
             break  
          }  
          if string(message) == "ping" {
          
             message = []byte("pong")  
          }  
          //写入ws数据
          err = ws.WriteMessage(mt, message)  
          if err != nil {
          
             break  
          }  
       }  
    }  
    
    func main() {
          
       bindAddress := "localhost:2303"  
       r := gin.Default()  
       r.GET("/ping", ping)  
       r.Run(bindAddress)  
    }

# 3. js代码 #

# 5. 测试 #

浏览器访问html代码，查看浏览器调试模式中的Console输出

Connection open …  
Received Message: Hello WebSockets!