修改tcp
This commit is contained in:
parent
14066cb7ac
commit
06b8edf26f
@ -8,7 +8,8 @@ from scipy.fft import fft, fftfreq
|
||||
import matplotlib.pyplot as plt
|
||||
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas, NavigationToolbar2QT
|
||||
from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QPushButton, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QFrame, QGridLayout, QStatusBar, QSizePolicy, QFileDialog
|
||||
from PyQt5.QtCore import Qt
|
||||
from PyQt5.QtCore import Qt, QTimer, pyqtSlot, QByteArray
|
||||
from PyQt5.QtNetwork import QTcpSocket, QAbstractSocket
|
||||
import matplotlib
|
||||
import struct
|
||||
import time
|
||||
@ -80,6 +81,18 @@ class SocketClientApp(QMainWindow):
|
||||
self.setWindowTitle("Socket Client & Data Plotter")
|
||||
self.setGeometry(100, 100, 700, 600) # 设置初始尺寸
|
||||
|
||||
self.buffer = QByteArray()
|
||||
self.reconnect_timer = QTimer(self)
|
||||
self.reconnect_timer.timeout.connect(self.attempt_reconnect)
|
||||
self.reconnect_interval = 5000
|
||||
self.max_reconnect_attempts = 5
|
||||
self.current_reconnect_attempts = 0
|
||||
|
||||
self.socket = QTcpSocket(self)
|
||||
self.socket.readyRead.connect(self.on_ready_read)
|
||||
self.socket.disconnected.connect(self.on_socket_disconnected)
|
||||
self.socket.errorOccurred.connect(self.on_socket_error)
|
||||
|
||||
# 组件初始化
|
||||
self.ip_label = QLabel("IP 地址:")
|
||||
self.port_label = QLabel("端口号:")
|
||||
@ -209,30 +222,37 @@ class SocketClientApp(QMainWindow):
|
||||
crc += byte
|
||||
return crc & 0xFF # 只保留最低字节
|
||||
def connect_to_server(self):
|
||||
""" 连接到服务器 """
|
||||
self.ip = self.ip_input.text()
|
||||
self.port = self.port_input.text()
|
||||
ip = self.ip_input.text()
|
||||
port = int(self.port_input.text())
|
||||
self.socket.abort()
|
||||
self.socket.connectToHost(ip, port)
|
||||
|
||||
if not self.ip or not self.port:
|
||||
self.status_bar.showMessage("状态: 请输入有效的 IP 和端口号")
|
||||
def on_socket_disconnected(self):
|
||||
self.status_bar.showMessage("状态: 连接断开，正在重连...")
|
||||
self.get_data_button.setEnabled(False)
|
||||
self.reconnect_timer.start(self.reconnect_interval)
|
||||
|
||||
def on_socket_error(self, error):
|
||||
self.status_bar.showMessage(f"状态: 错误 - {self.socket.errorString()}")
|
||||
|
||||
def attempt_reconnect(self):
|
||||
if self.current_reconnect_attempts >= self.max_reconnect_attempts:
|
||||
self.reconnect_timer.stop()
|
||||
self.status_bar.showMessage("状态: 重连失败，请检查网络")
|
||||
return
|
||||
self.current_reconnect_attempts += 1
|
||||
self.connect_to_server()
|
||||
|
||||
try:
|
||||
self.port = int(self.port)
|
||||
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.socket.settimeout(2)
|
||||
# 设置 keep-alive 选项
|
||||
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
|
||||
self.socket.connect((self.ip, self.port))
|
||||
def on_ready_read(self):
|
||||
self.buffer += self.socket.readAll()
|
||||
|
||||
self.get_data_button.setEnabled(True) # 连接成功，启用获取数据按钮
|
||||
self.upgrade_button.setEnabled(True)
|
||||
self.upgrade_button_test.setEnabled(True)
|
||||
self.status_bar.showMessage(f"状态: 连接成功! 服务器 {self.ip}:{self.port}")
|
||||
|
||||
except Exception as e:
|
||||
self.status_bar.showMessage(f"状态: 连接失败 - {str(e)}")
|
||||
def receive_data(self, length: int):
|
||||
while len(self.buffer) < length:
|
||||
if not self.socket.waitForReadyRead(1000):
|
||||
raise ConnectionError("等待数据超时")
|
||||
data = self.buffer[:length]
|
||||
self.buffer = self.buffer[length:]
|
||||
return bytes(data)
|
||||
|
||||
def parse_package_head(self,data: bytes) -> Tuple[Optional[dict], Optional[bytes]]:
|
||||
"""
|
||||
@ -255,19 +275,6 @@ class SocketClientApp(QMainWindow):
|
||||
'result_code': result_code
|
||||
}, data[HEADER_SIZE:]
|
||||
|
||||
def receive_data(self,length: int):
|
||||
"""
|
||||
从socket接收指定长度的数据
|
||||
"""
|
||||
data = bytearray()
|
||||
while len(data) < length:
|
||||
chunk = self.socket.recv(length - len(data))
|
||||
if not chunk:
|
||||
raise ConnectionError("Connection closed")
|
||||
data.extend(chunk)
|
||||
print(f"data size {len(data)}")
|
||||
return bytes(data)
|
||||
|
||||
def process_packet(self):
|
||||
""" 循环接收数据，直到接收完整 """
|
||||
try:
|
||||
@ -312,7 +319,7 @@ class SocketClientApp(QMainWindow):
|
||||
""" 获取数据并绘制 """
|
||||
try:
|
||||
self.status_bar.showMessage("状态: 正在获取数据...")
|
||||
self.socket.sendall(bytes([0xAA,0x55,0xAA,0x01,0x00,0x00,0x01,0x00,0x77,0x01,0x00,0x01,0x00,0x00,0x00])) # 发送数据
|
||||
self.socket.write(bytes([0xAA,0x55,0xAA,0x01,0x00,0x00,0x01,0x00,0x77,0x01,0x00,0x01,0x00,0x00,0x00])) # 发送数据
|
||||
data = self.process_packet() # 接收所有数据
|
||||
data = np.frombuffer(data, dtype=np.int32) # 根据实际数据格式转换
|
||||
for i in range(min(100, len(data))): # 确保不超过数据长度
|
||||
@ -379,7 +386,7 @@ class SocketClientApp(QMainWindow):
|
||||
print(f"Upgrade length: {upgrade_len}, CRC: {crc}")
|
||||
print(f"Total packet size: {len(full_packet)} bytes")
|
||||
|
||||
self.socket.sendall(full_packet)
|
||||
self.socket.write(full_packet)
|
||||
print("Upgrade packet ready to send (commented out actual send code)")
|
||||
except Exception as e:
|
||||
print(f"Upgrade failed: {str(e)}")
|
||||
@ -425,11 +432,11 @@ class SocketClientApp(QMainWindow):
|
||||
print(f"Upgrade length: {upgrade_len}, CRC: {crc}")
|
||||
print(f"Total packet size: {len(full_packet)} bytes")
|
||||
|
||||
self.socket.sendall(full_packet)
|
||||
self.socket.write(full_packet)
|
||||
sleep(15)
|
||||
self.socket.close()
|
||||
self.connect_to_server()
|
||||
self.socket.sendall(full_packet)
|
||||
self.socket.write(full_packet)
|
||||
print("Upgrade packet ready to send (commented out actual send code)")
|
||||
except Exception as e:
|
||||
print(f"Upgrade failed: {str(e)}")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user