71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import socket
|
|
import json
|
|
import threading
|
|
import time
|
|
|
|
|
|
class TCPClient:
|
|
def __init__(self, host='127.0.0.1', port=8889):
|
|
self.host = host
|
|
self.port = port
|
|
self.client_socket = None
|
|
self.is_connected = False
|
|
self.reconnect_interval = 5 # 重连间隔(秒)
|
|
self.is_running = False
|
|
|
|
def connect(self):
|
|
"""连接到TCP服务端"""
|
|
try:
|
|
self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.client_socket.connect((self.host, self.port))
|
|
self.is_connected = True
|
|
print(f"已连接到 {self.host}:{self.port}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"连接到服务端失败: {e}")
|
|
self.is_connected = False
|
|
return False
|
|
|
|
def start(self):
|
|
"""启动客户端并维持连接"""
|
|
self.is_running = True
|
|
reconnect_thread = threading.Thread(target=self._reconnect_worker, daemon=True)
|
|
reconnect_thread.start()
|
|
|
|
def _reconnect_worker(self):
|
|
"""重连工作线程"""
|
|
while self.is_running:
|
|
if not self.is_connected:
|
|
print(f"尝试重新连接到 {self.host}:{self.port}...")
|
|
self.connect()
|
|
time.sleep(self.reconnect_interval)
|
|
|
|
def send_data(self, data):
|
|
"""发送数据到服务端"""
|
|
if not self.is_connected or not self.client_socket:
|
|
print("客户端未连接到服务端")
|
|
return False
|
|
|
|
try:
|
|
json_data = json.dumps(data, ensure_ascii=False)
|
|
# 添加换行符作为结束标记
|
|
self.client_socket.sendall((json_data + "\n").encode('utf-8'))
|
|
print(f"已发送数据: {json_data}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"发送数据时出错: {e}")
|
|
self.is_connected = False
|
|
return False
|
|
|
|
def stop(self):
|
|
"""停止客户端"""
|
|
self.is_running = False
|
|
self.is_connected = False
|
|
if self.client_socket:
|
|
try:
|
|
self.client_socket.close()
|
|
except Exception as e:
|
|
print(f"关闭客户端套接字时出错: {e}")
|
|
print("TCP客户端已停止")
|
|
|