python 心跳 socket

原创

mob649e81593bda 2023-07-15 12:47:20 ©著作权

文章标签 客户端服务器 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e81593bda的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 心跳 Socket

在网络通信中，心跳是一种用于维持连接的机制。它通过定期发送或接收消息来检测连接是否正常，如果连接断开，则可以快速做出相应的处理。在本文中，我们将使用 Python 的 socket 模块来实现心跳机制。

什么是 Socket

Socket 是一种在网络上进行通信的一种机制，它提供了一种进行网络编程的方法。通过 Socket，我们可以建立客户端和服务器之间的连接，并在连接上发送和接收数据。

Python 提供了 socket 模块，它包含了各种函数和类，可以方便地创建和使用 Socket。使用 Socket，我们可以实现各种网络通信功能，包括心跳。

实现心跳机制

首先，我们需要创建一个服务器和一个客户端。服务器将定期向客户端发送心跳消息，而客户端接收并回复心跳消息以表示连接正常。

服务器端

import socket
import time

def send_heartbeat(conn):
    while True:
        conn.send(b"Heartbeat")
        time.sleep(5)

def main():
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind(("localhost", 8888))
    server_socket.listen(1)
    
    print("Server started, waiting for connections...")
    
    conn, addr = server_socket.accept()
    print("Connection from:", addr)
    
    send_heartbeat(conn)

if __name__ == "__main__":
    main()

在上面的代码中，我们首先创建了一个服务器套接字，并绑定到本地主机的8888端口。然后，我们使用 listen 函数开始监听来自客户端的连接。

一旦有客户端连接成功，我们就会得到一个连接对象 conn 和客户端的地址 addr。然后，我们进入了一个循环，不断向客户端发送心跳消息。

客户端

import socket

def receive_heartbeat(conn):
    while True:
        data = conn.recv(1024)
        if data:
            print("Received heartbeat:", data.decode())
            conn.send(b"Heartbeat reply")

def main():
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.connect(("localhost", 8888))
    
    receive_heartbeat(client_socket)

if __name__ == "__main__":
    main()

在上面的代码中，我们首先创建了一个客户端套接字，并使用 connect 函数连接到服务器的地址和端口。

然后，我们进入一个循环，不断接收来自服务器的心跳消息。一旦接收到消息，我们打印出来，并回复一个心跳回复消息。