From 854e065c6d468953a56a6d6fe862ad922b39f581 Mon Sep 17 00:00:00 2001
From: MuJ <muji3203380067@outlook.com>
Date: Sun, 4 Feb 2024 21:36:00 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E6=AD=A5=E5=AE=8C=E6=88=90=E4=BA=86?=
 =?UTF-8?q?=E6=8E=A5=E6=94=B6=E5=92=8C=E5=8F=91=E9=80=81=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E7=9A=84=E4=BB=A3=E7=A0=81=E7=BC=96=E5=86=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 listen.py | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 post.py   | 18 ++++++++++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 listen.py
 create mode 100644 post.py

diff --git a/listen.py b/listen.py
new file mode 100644
index 0000000..7f91efd
--- /dev/null
+++ b/listen.py
@@ -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()  # 停止监听
diff --git a/post.py b/post.py
new file mode 100644
index 0000000..e952314
--- /dev/null
+++ b/post.py
@@ -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()
\ No newline at end of file