当前位置:   article > 正文

go 短连接和长连接 超时处理_go实现tcp长链接,心跳、以及超时

go实现tcp长链接,心跳、以及超时

go 短连接和长连接 超时处理


作为一个可能会和很多Client进行通讯交互的Server,首先要保证的就是整个Server运行状态的稳定性,因此在和Client建立连接通讯的时候,确保连接的及时断开非常重要,否则一旦和多个客户端建立不关闭的长连接,对于服务器资源的占用是很可怕的。因此,我们需要针对可能出现的短连接和长连接,设定不同的限制策略。

    针对短连接,我们可以使用golang中的net包自带的timeout函数,一共有三个,分别是:


[plain]  view plain  copy
  1. func (*IPConn) SetDeadline  
  2. func (c *IPConn) SetDeadline(t time.Time) error  
  3.   
  4. func (*IPConn) SetReadDeadline  
  5. func (c *IPConn) SetReadDeadline(t time.Time) error  
  6.   
  7. func (*IPConn) SetWriteDeadline  
  8. func (c *IPConn) SetWriteDeadline(t time.Time) error</span>  


    如果想要给服务器设置短连接的timeout,我们就可以这么写:

[plain]  view plain  copy
  1. netListen, err := net.Listen("tcp", Port)  
  2.     Log("Waiting for clients")  
  3.     for {  
  4.         conn, err := netListen.Accept()  
  5.         if err != nil {  
  6.             continue  
  7.         }  
  8.   
  9.         conn.SetReadDeadline(time.Now().Add(time.Duration(10) * time.Second))</span>  
    这里的三个函数都是用于设置每次socket连接能够维持的最长时间,一旦超过设置的timeout后,便会在Server端自动断开连接。其中SetReadline, SetWriteline设置的是读取和写入的最长持续时间,而SetDeadline则同时包含了 SetReadline, SetWriteline两个函数。

    通过这样设定,每个和Server通讯的Client连接时长最长也不会超过10s了~~


    搞定短连接后,接下来就是针对长连接的处理策略了~~

    作为长连接,由于我们往往很难确定什么时候会中断连接,因此并不能像处理短连接那样简单粗暴的设定一个timeout就可以搞定,而在Golang的net包中,并没有针对长连接的函数,因此需要我们自己设计并实现针对长连接的处理策略啦~

    针对socke长连接,常见的做法是在Server和Socket之间设计通讯机制,当两者之间没有信息交互时,双方便会定时发送数据包(心跳),以维持连接状态。


    这种方法是目前使用相对比较多的做法,但是开销相对也较大,特别是当Server和多个client保持长连接的时候,并发会比较高,考虑到公司的业务需求,我最后选择了逻辑相对简单,开销相对较小的策略:

    当Server每次收到Client发到的信息之后,便会开始心跳计时,如果在心跳计时结束之前没有再次收到Client发来的信息,那么便会断开跟Client的连接。而一旦在设定时间内再次收到Client发来的信息,那么Server便会重置计时器,再次重新进行心跳计时,直到超时断开连接为止。

下面就是实现该计时的代码:

[plain]  view plain  copy
  1. //长连接入口  
  2. func handleConnection(conn net.Conn,timeout int) {  
  3.   
  4.     buffer := make([]byte, 2048)  
  5.     for {  
  6.         n, err := conn.Read(buffer)  
  7.   
  8.         if err != nil {  
  9.             LogErr(conn.RemoteAddr().String(), " connection error: ", err)  
  10.             return  
  11.         }  
  12.         Data :=(buffer[:n])  
  13.         messnager := make(chan byte)  
  14.         postda :=make(chan byte)  
  15.         //心跳计时  
  16.         go HeartBeating(conn,messnager,timeout)  
  17.         //检测每次Client是否有数据传来  
  18.         go GravelChannel(Data,messnager)  
  19.         Log( "receive data length:",n)  
  20.         Log(conn.RemoteAddr().String(), "receive data string:", string(Data  
  21.   
  22.     }  
  23. }  
  24.   
  25. //心跳计时,根据GravelChannel判断Client是否在设定时间内发来信息  
  26. func HeartBeating(conn net.Conn, readerChannel chan byte,timeout int) {  
  27.         select {  
  28.         case fk := <-readerChannel:  
  29.             Log(conn.RemoteAddr().String(), "receive data string:", string(fk))  
  30.             conn.SetDeadline(time.Now().Add(time.Duration(timeout) * time.Second))  
  31.             //conn.SetReadDeadline(time.Now().Add(time.Duration(5) * time.Second))  
  32.             break  
  33.         case <-time.After(time.Second*5):  
  34.             Log("It's really weird to get Nothing!!!")  
  35.             conn.Close()  
  36.         }  
  37.   
  38. }  
  39.   
  40. func GravelChannel(n []byte,mess chan byte){  
  41.     for _ , v := range n{  
  42.         mess <- v  
  43.     }  
  44.     close(mess)  
  45. }  
  46.   
  47.   
  48. func Log(v ...interface{}) {  
  49.     log.Println(v...)  
  50. }  


    这样,就可以成功实现对于长连接的处理了~~,我们可以这么进行测试:

[plain]  view plain  copy
  1. func sender(conn net.Conn) {  
  2.     for i := 0; i <5; i++ {  
  3.         words:= strconv.Itoa(i)+"This is a test for long conn"   
  4.         conn.Write([]byte(words))  
  5.         time.Sleep(2*time.Second)  
  6.   
  7.     }  
  8.     fmt.Println("send over")  
  9.   
  10. }  

    可以发现,Sender函数中time.Sleep阻塞的时间设定的比Server中的timeout短的时候,Client端的信息可以自由的发送到循环结束,而当我们设定Sender函数的阻塞时间较长时,就只能发出第一次循环的信息。


声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/230047
推荐阅读
相关标签
  

闽ICP备14008679号