初步完成了接收和发送数据的代码编写

This commit is contained in:
MuJ 2024-02-04 21:36:00 +08:00
parent 0f281d487d
commit 854e065c6d
2 changed files with 82 additions and 0 deletions

64
listen.py Normal file
View File

@ -0,0 +1,64 @@
import socket
import threading
class VoltageReceiver():
def __init__(self):
self.data = "" # 数据存储
self.active = False # 控制线程的开关
self.dataready = 0 # 数据是否准备好的标志位
self.thread = None # 线程存储
self.sock = None # 套接字存储
def receive_data(self):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.bind(('localhost', 9996))
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.sock.listen(1)
while self.active: # 只有当 self.active 为 True 时,才开始监听
conn, addr = self.sock.accept()
while conn:
try:
data = conn.recv(1024)
except socket.error as e:
print("Remote host has disconnected.") # 打印提醒
break
if not data or not self.active: # 如果没有数据或 self.active 为 False 则跳出循环
break
data = data.decode()
if self.dataready:
# 新的数据已经准备好,而旧的数据还没被取走,发出警告
self.alert()
with threading.Lock(): # 使用锁确保线程安全
self.data = data
self.dataready = 1
conn.close()
def start(self): # 开始监听
if not self.thread or not self.thread.is_alive(): # 如果线程未创建或已结束,那么创建一个新线程
self.active = True
self.thread = threading.Thread(target=self.receive_data)
self.thread.start()
def stop(self): # 停止监听
self.active = False
if self.sock: # 关闭套接字,使得 sock.accept() 退出阻塞
self.sock.close()
def get_data(self): # 获取数据
with threading.Lock(): # 使用锁确保线程安全
self.dataready = 0
return self.data
def alert(self): # 报警函数
print("Warning: New data has arrived but the old data has not been taken away.")
if __name__ == "__main__": # 测试代码
receiver = VoltageReceiver() # 创建一个 VoltageReceiver 的实例
receiver.start() # 开始监听
while receiver.active:
if receiver.dataready: # 当数据准备好时,打印数据
print(receiver.get_data())
receiver.stop() # 停止监听

18
post.py Normal file
View File

@ -0,0 +1,18 @@
import socket
# 创建一个socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 建立连接:
s.connect(('localhost', 9996))
while True:
try:
# 输入需要发送的数据
data = input("Enter the voltage data you want to send (or 'exit' to quit): ")
if data.lower() == 'exit':
break
# 发送数据:
s.send(data.encode())
except KeyboardInterrupt:
break
# 关闭连接
s.close()