Golang长连接-基于WebSocket

Golang长连接-基于WebSocket Websocket简介 WebSocket可以实现客户端与服务器间双向、基于消息的文本或二进制数据传输。它是浏览器中最靠近套接字的API。但WebSocket连接远远不是一个网络套...

Golang长连接-基于WebSocket

Websocket简介

WebSocket可以实现客户端与服务器间双向、基于消息的文本或二进制数据传输。它是浏览器中最靠近套接字的API。但WebSocket连接远远不是一个网络套接字，因为浏览器在这个简单的API之后隐藏了所有的复杂性，而且还提供了更多服务：

连接协商和同源策略；
与既有HTTP基础设施的互操作；
基于消息的通信和高效消息分帧；
子协议协商及可扩展能力。

WebSocket资源URL采用了自定义模式：ws表示纯文本通信（如ws://example.com/socket），wss表示使用加密信道通信（TCP+TLS）。

长连接实现

本例用gin框架，引入github.com/gorilla/websocket包

服务端核心代码:

ping函数，将请求升级为websocket。
只有Get请求才能进行升级，具体原因可以查看websocket源码
当客户端请求ping时，便建立了长连接，服务端读取客户端数据，如果数据为ping，则返回pong，如果不为ping，则把输入的内容返回。

package main
import (
   "net/http"
   "github.com/gin-gonic/gin"
   "github.com/gorilla/websocket"
)
var upGrader = websocket.Upgrader{
   CheckOrigin: func (r *http.Request) bool {
      return true
   },
}
func setupRouter() *gin.Engine {
   r := gin.Default()
   r.LoadHTMLGlob("templates/*")
   // Ping test
   r.GET("/ping", ping)
   r.GET("/longconnecthtml",longconnecthtml)
   return r
}
func longconnecthtml(c *gin.Context)  {
   c.HTML(http.StatusOK, "longconnect.tmpl",gin.H{})
}
func ping(c *gin.Context) {
   //c.String(http.StatusOK, "ok")
   //升级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() {
   r := setupRouter()
   // Listen and Server in 0.0.0.0:8080
   r.Run(":9090")
}

客户端核心代码：

要使用websocket，客户端必须发起websocket请求
websocket的建立和使用也很方便，主要涉及

*new WebSocket：创建websocket对象

*onopen：连接建立时触发

*onmessage：客户端接收服务端数据时触发

*onerror：通信发生错误时触发

*onclose：连接关闭时触发

该代码块的主要功能是和服务端建立连接，在文本框输入信息，将信息发送给服务端，并将服务端返回内容显示出来

<!DOCTYPE HTML>
<html>
   <head>
   <meta charset="utf-8">
   <title>长连接测试</title>
   <textarea id="inp_send" class="form-control" style="height:100px;" placeholder="发送的内容"></textarea>
   <button type="button" id="btn_send" class="btn btn-info" onclick="fun_sendto();">发送（ctrl+回车）</button>
<script type="text/javascript" src="https://dss0.bdstatic.com/5aV1bjqh_Q23odCf/static/superman/js/lib/jquery-1-edb203c114.10.2.js"></script>
      <script type="text/javascript">
        var ws = new WebSocket("ws://localhost:9090/ping");
        //连接打开时触发
        ws.onopen = function(evt) {
            console.log("Connection open ...");
            ws.send("Hello WebSockets!");
        };
        //接收到消息时触发
        ws.onmessage = function(evt) {
            console.log("Received Message: " + evt.data);
        };
        //连接关闭时触发
        ws.onclose = function(evt) {
            console.log("Connection closed.");
        };
        function fun_sendto(){
            var content = $("#inp_send").val();
            ws.send(content);
        }
      </script>
   </head>
   <body>
   </body>
</html>

服务端运行起来之后，请求http://localhost:9090/longconnecthtml即可查看效果

长连接展示

浏览器，请求ping接口，便可看到该请求为websocket请求，而且为pending状态
也可以查看到在发起请求时，request header中有很多新的参数，具体含义大家可以看我提供的参考资料
console中显示的是服务端推送到客户端的数据

注意事项

长连接可以实现双向通信，但是以占用连接为前提的，如果请求量较大，需要考虑资源问题
服务部署情况对效果影响很大。例如，如果只部署北京机房供全国使用，当服务端向客户端推送数据时，不同地区可能会有很大延迟。

更多相关技术内容咨询欢迎前往并持续关注六星社区了解详情。

程序员编程交流QQ群：805358732

如果你想用Python开辟副业赚钱，但不熟悉爬虫与反爬虫技术，没有接单途径，也缺乏兼职经验
关注下方微信公众号：Python编程学习圈，获取价值999元全套Python入门到进阶的学习资料以及教程，还有Python技术交流群一起交流学习哦。

发表于 2021-07-10 10:41
阅读 ( 570 )
分类：Golang

Golang长连接-基于WebSocket

你可能感兴趣的文章

相关问题

0 条评论

作家榜 »