diff --git a/MP2/peer.py b/MP2/peer.py index f26beee..8c102d2 100644 --- a/MP2/peer.py +++ b/MP2/peer.py @@ -1,26 +1,32 @@ from datetime import datetime, timezone import socket -def main(): - INTRODUCER_IP = "172.22.156.3" - IP = socket.gethostbyname(socket.gethostname()) - SERVER_PORT = 6767 - INTRODUCER_FLAG = (INTRODUCER_IP == IP) - - members_list = { - (IP, SERVER_PORT): {"heart_beat": 1, "last_seen": utc_time()} - } - - server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) - server.bind((IP, SERVER_PORT)) - - if not INTRODUCER_FLAG: - server.sendto(b"JOIN", (INTRODUCER_IP, SERVER_PORT)) - data, addr = server.recvfrom(4096) - print(addr, data) - def utc_time(): return datetime.now(timezone.utc) -if __name__ == "__main__": - main() + +INTRODUCER_IP = "172.22.156.3" +IP = socket.gethostbyname(socket.gethostname()) +SERVER_PORT = 6767 +INTRODUCER_FLAG = (INTRODUCER_IP == IP) + +#check if this VM is the introducer and add it self +members_list = {} +if INTRODUCER_FLAG: + members_list[(IP, PORT)] = { + "heart_beat": 1, + "last_seen": utc_time() + } + + +#start UDP socket +server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) +server.bind((IP, SERVER_PORT)) + +if not INTRODUCER_FLAG: + server.sendto(b"JOIN", (INTRODUCER_IP, SERVER_PORT)) + +while True: + data, addr = server.recvfrom(4096) + +#increment heart beat \ No newline at end of file