First commit

client/UDPClient.py

@@ -0,0 +1,53 @@
+from SocketClient import SocketClient
+import time
+import socket
+import select
+
+
+class UDPClient(SocketClient):
+    _timestamps: set[float] = set()
+    _last_check = time.time()
+
+    def Connect(self) -> bool:
+        self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self.socket.setblocking(False)
+        return True
+
+    def Send(self, data: str):
+        now = time.time()
+        data = "%s/%f" % (data, now)
+        self.socket.sendto(data.encode(), (self.ip, self.port))
+        self.logger.info("UDP_SENT %s" % data)
+        self._timestamps.add(now)
+        print(self._timestamps)
+        self.Receive()
+        self.staleCheck()
+
+    def Reconnect(self) -> bool:
+        return True
+
+    def Receive(self):
+        (incoming, _, _) = select.select([self.socket], [], [], 0)
+        if incoming:
+            data, addr = self.socket.recvfrom(1024)
+            timestamp = data.split(b"/")[1]
+            print(timestamp.decode())
+            self.logger.info("UDP_SUCCESS %s (%s)" % (data.decode(), addr))
+            self._timestamps.discard(float(timestamp.decode()))
+            print(self._timestamps)
+
+    def staleCheck(self):
+        if (time.time() - self._last_check) < 1:
+            return
+
+        timedout_timestamps = set()
+        for timestamp in self._timestamps:
+            if (self._last_check - timestamp) < 10:
+                continue
+            self.logger.error("UDP_FAIL %f" % timestamp)
+            print(self._timestamps)
+            timedout_timestamps.add(timestamp)
+
+        self._last_check = time.time()
+        for timestamp in timedout_timestamps:
+            self._timestamps.discard(timestamp)