socket tcp服务端断开连接后python重连接

原创

mob64ca12d70c79 2025-01-08 03:33:49 ©著作权

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

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

Python Socket TCP 服务端断开连接后的重连接

在网络编程中，TCP（传输控制协议）是一种面向连接的通信协议。当TCP服务端与客户端之间的连接断开时，客户端需要能够重新连接到服务端。这篇文章将带你深入了解如何在Python中实现TCP客户端的重连接逻辑，并通过代码示例演示这种实现方式。

什么是TCP连接？

TCP是一个面向连接的协议，它确保数据在两个节点之间顺序传输和完整性。TCP连接的过程包括连接的建立、数据传输和连接的断开。

当客户端与服务端之间的连接异常中断时，客户端应该能够检测到这种断开并尝试重新连接。实现这一功能的挑战在于，客户端需要处理不同的异常情况，以便稳定地恢复连接。

TCP连接的状态

为了帮助理解TCP的连接重建过程，下面是一个简单的状态图：

stateDiagram
    [*] --> Unconnected
    Unconnected --> Connecting : try to connect
    Connecting --> Connected : connection established
    Connecting --> Unconnected : connection failed
    Connected --> Unconnected : lost connection
    Unconnected --> Connecting : retry connecting

Python Socket 模块

Python提供了socket模块来进行网络通信。使用这个模块，我们可以轻松创建TCP客户端与服务端。

TCP客户端示例

下面是一个简单的TCP客户端示例，它能够在服务端断开连接时自动重连。

import socket
import time

# 定义服务端信息
SERVER_HOST = '127.0.0.1'  # 服务端主机
SERVER_PORT = 65432          # 服务端端口

def create_socket():
    """创建TCP Socket"""
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    return s

def connect_to_server(s):
    """连接到服务端"""
    while True:
        try:
            s.connect((SERVER_HOST, SERVER_PORT))
            print("连接到服务端成功。")
            return
        except (socket.error, ConnectionRefusedError):
            print("连接失败，尝试重新连接...")
            time.sleep(5)  # 等待5秒钟后重试连接

def main():
    sock = create_socket()
    connect_to_server(sock)

    while True:
        try:
            message = input("请输入要发送给服务端的信息（输入 'exit' 关闭）:")
            if message.lower() == 'exit':
                print("关闭连接...")
                sock.close()
                break
            
            sock.sendall(message.encode('utf-8'))
            data = sock.recv(1024)
            print(f"从服务端接收到的数据: {data.decode('utf-8')}")

        except (socket.error, ConnectionResetError):
            print("连接丢失，正在尝试重连...")
            sock.close()
            sock = create_socket()
            connect_to_server(sock)

if __name__ == "__main__":
    main()