linux tcp 断线重连

import socket
import time

def connect_to_server(host, port):
    while True:
        try:
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sock.connect((host, port))
            print("Connected to server successfully!")
            return sock
        except ConnectionRefusedError:
            print("Connection refused, retrying in 5 seconds...")
            time.sleep(5)
        except Exception as e:
            print(f"An error occurred: {e}, retrying in 5 seconds...")
            time.sleep(5)

def main():
    host = '127.0.0.1'
    port = 12345
    sock = connect_to_server(host, port)
    
    # 进行数据传输的逻辑
    try:
        while True:
            data = sock.recv(1024)
            if not data:
                break
            print(f"Received: {data.decode()}")
    except ConnectionResetError:
        print("Connection lost, attempting to reconnect...")
        sock = connect_to_server(host, port)
    finally:
        sock.close()

if __name__ == "__main__":
    main()