golang tcp 连接空闲监控 golang 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>

[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两个函数。

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

[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. }