134 lines
5.0 KiB
Python
134 lines
5.0 KiB
Python
import socket
|
||
import threading
|
||
import json
|
||
import time
|
||
from datetime import datetime
|
||
|
||
class TCPServer:
|
||
def __init__(self, host='localhost', port=8080):
|
||
self.host = host
|
||
self.port = port
|
||
self.server_socket = None
|
||
self.client_socket = None
|
||
self.client_address = None
|
||
self.running = False
|
||
|
||
def start_server(self):
|
||
"""启动TCP服务端"""
|
||
try:
|
||
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
self.server_socket.bind((self.host, self.port))
|
||
self.server_socket.listen(1)
|
||
self.running = True
|
||
|
||
print(f"TCP服务端已启动，监听 {self.host}:{self.port}")
|
||
|
||
while self.running:
|
||
try:
|
||
self.client_socket, self.client_address = self.server_socket.accept()
|
||
print(f"客户端已连接: {self.client_address}")
|
||
|
||
# 启动客户端处理线程
|
||
client_thread = threading.Thread(target=self.handle_client)
|
||
client_thread.daemon = True
|
||
client_thread.start()
|
||
|
||
except Exception as e:
|
||
if self.running:
|
||
print(f"接受客户端连接时出错: {e}")
|
||
|
||
except Exception as e:
|
||
print(f"启动服务端时出错: {e}")
|
||
finally:
|
||
self.stop_server()
|
||
|
||
def handle_client(self):
|
||
"""处理客户端通信"""
|
||
try:
|
||
while self.running and self.client_socket:
|
||
try:
|
||
# 设置接收超时
|
||
self.client_socket.settimeout(1.0)
|
||
|
||
# 接收客户端数据
|
||
data = self.client_socket.recv(1024)
|
||
if not data:
|
||
break
|
||
|
||
# 解析JSON数据
|
||
message = data.decode('utf-8')
|
||
json_data = json.loads(message)
|
||
|
||
# 处理客户端返回的异常数据
|
||
self.process_client_data(json_data)
|
||
|
||
except socket.timeout:
|
||
# 超时继续循环
|
||
continue
|
||
except json.JSONDecodeError as e:
|
||
print(f"解析JSON数据时出错: {e}")
|
||
except Exception as e:
|
||
print(f"处理客户端数据时出错: {e}")
|
||
|
||
except Exception as e:
|
||
print(f"客户端处理线程出错: {e}")
|
||
finally:
|
||
self.disconnect_client()
|
||
|
||
def process_client_data(self, json_data):
|
||
"""处理客户端返回的数据"""
|
||
try:
|
||
cmd = json_data.get("cmd")
|
||
timestamp = json_data.get("timestamp")
|
||
|
||
if cmd == "production_error":
|
||
print(f"收到异常生产通知 - 时间: {timestamp}")
|
||
# 调用update_custom_table_status更新数据库状态
|
||
try:
|
||
update_custom_table_status("异常生产")
|
||
print("数据库状态已更新为: 异常生产")
|
||
except Exception as e:
|
||
print(f"更新数据库状态时出错: {e}")
|
||
elif cmd == "interrupt_error":
|
||
print(f"收到中断生产通知 - 时间: {timestamp}")
|
||
# 调用update_custom_table_status更新数据库状态
|
||
try:
|
||
update_custom_table_status("中断生产")
|
||
print("数据库状态已更新为: 中断生产")
|
||
except Exception as e:
|
||
print(f"更新数据库状态时出错: {e}")
|
||
else:
|
||
print(f"收到未知命令: {cmd}")
|
||
|
||
except Exception as e:
|
||
print(f"处理客户端数据时出错: {e}")
|
||
|
||
def send_data(self, data):
|
||
"""向客户端发送数据"""
|
||
try:
|
||
if self.client_socket and self.running:
|
||
json_data = json.dumps(data, ensure_ascii=False)
|
||
self.client_socket.send((json_data + "\n").encode('utf-8'))
|
||
print(f"已发送数据: {json_data}")
|
||
return True
|
||
except Exception as e:
|
||
print(f"发送数据时出错: {e}")
|
||
return False
|
||
|
||
def disconnect_client(self):
|
||
"""断开客户端连接"""
|
||
if self.client_socket:
|
||
self.client_socket.close()
|
||
self.client_socket = None
|
||
print("客户端已断开连接")
|
||
|
||
def stop_server(self):
|
||
"""停止服务端"""
|
||
self.running = False
|
||
if self.client_socket:
|
||
self.client_socket.close()
|
||
if self.server_socket:
|
||
self.server_socket.close()
|
||
print("TCP服务端已停止")
|