GoLang中socket心跳检测的实现

尉迟珍瑞

心跳检测是确保服务端与客户端之间稳定性的一种机制，具体实现方法为每间隔固定时间相互发送消息并响应。
服务端代码：

1. package main
   
2. 
   
3. import (
   
4.     "fmt"
   
5.     "net"
   
6.     "time"
   
7. )
   
8. 
   
9. func main() {
   
10.     // 启动 TCP 服务器
    
11.     listener, err := net.Listen("tcp", ":8080")
    
12.     if err != nil {
    
13.         fmt.Println("Error starting server:", err)
    
14.         return
    
15.     }
    
16.     defer listener.Close()
    
17. 
    
18.     fmt.Println("Server is listening on port 8080")
    
19. 
    
20.     for {
    
21.         conn, err := listener.Accept()
    
22.         if err != nil {
    
23.             fmt.Println("Error accepting connection:", err)
    
24.             continue
    
25.         }
    
26.         go handleConnection(conn)
    
27.     }
    
28. }
    
29. 
    
30. func handleConnection(conn net.Conn) {
    
31.     defer conn.Close()
    
32.     buffer := make([]byte, 1024)
    
33. 
    
34.     for {
    
35.         // 读取数据
    
36.         n, err := conn.Read(buffer)
    
37.         if err != nil {
    
38.             fmt.Println("Connection closed:", err)
    
39.             return
    
40.         }
    
41.         fmt.Println("Received:", string(buffer[:n]))
    
42. 
    
43.         // 响应心跳消息
    
44.         if string(buffer[:n]) == "pong" {
    
45.             conn.Write([]byte("pong"))
    
46.         }
    
47.     }
    
48. }

复制代码

客户端代码：

1. package main
   
2. 
   
3. import (
   
4.     "fmt"
   
5.     "net"
   
6.     "time"
   
7. )
   
8. 
   
9. func main() {
   
10.     // 连接到服务器
    
11.     conn, err := net.Dial("tcp", "localhost:8080")
    
12.     if err != nil {
    
13.         fmt.Println("Error connecting to server:", err)
    
14.         return
    
15.     }
    
16.     defer conn.Close()
    
17. 
    
18.     // 定义心跳间隔
    
19.     ticker := time.NewTicker(2 * time.Second)
    
20.     defer ticker.Stop()
    
21. 
    
22.     go func() {
    
23.         for {
    
24.             <-ticker.C
    
25.             _, err := conn.Write([]byte("ping"))
    
26.             if err != nil {
    
27.                 fmt.Println("Error sending heartbeat:", err)
    
28.                 return
    
29.             }
    
30.             fmt.Println("ping")
    
31.         }
    
32.     }()
    
33. 
    
34.     // 读取服务器响应
    
35.     buffer := make([]byte, 1024)
    
36.     for {
    
37.         n, err := conn.Read(buffer)
    
38.         if err != nil {
    
39.             fmt.Println("Error reading from server:", err)
    
40.             return
    
41.         }
    
42.         fmt.Println("Received:", string(buffer[:n]))
    
43.     }
    
44. }

复制代码

到此这篇关于GoLang中socket心跳检测的实现的文章就介绍到这了,更多相关GoLang socket心跳检测内容请搜索晓枫资讯以前的文章或继续浏览下面的相关文章希望大家以后多多支持晓枫资讯！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

小璇子1028

感谢楼主，顶。