codex/ping_server.py at main · ubcaerodesign/codex · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import time
import zmq
import threading
from ubcaerodesign.networking import make_zmq_socket

server_ip = "144.126.213.3"
server_port_send = 5555
server_port_receive = 5557


def send_ping(server_ip, server_port, server_status):
    ping_pub = make_zmq_socket(server_ip, server_port, "ping_sender", "connect", "pub")
    while not server_status.is_set():
        try:
            ping_pub.send_string("PING")
        except Exception as e:
            print(f"Failed to send PING: {e}")
        time.sleep(0.5)


def receive_ping(server_ip, server_port, server_status):
    ping_sub = make_zmq_socket(
        server_ip, server_port, "ping_receiver", "connect", "sub"
    )
    ping_sub.setsockopt(zmq.RCVTIMEO, 1000)  # wait 1s to see if server is running
    while not server_status.is_set():
        try:
            message = ping_sub.recv_string()
            if message:
                print("Server is running")
                server_status.set()
        except zmq.Again:
            print("Server not running")
            server_status.set()


if __name__ == "__main__":
    server_status = threading.Event()

    send_thread = threading.Thread(
        target=send_ping, args=(server_ip, server_port_send, server_status), daemon=True
    )
    receive_thread = threading.Thread(
        target=receive_ping,
        args=(server_ip, server_port_receive, server_status),
        daemon=True,
    )

    send_thread.start()
    receive_thread.start()

    send_thread.join()
    receive_thread.join()