Aggregator-based-RPL/client.py at main · raziqueA/Aggregator-based-RPL · 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
54
55
56
57
58
59
60
from message import Message
class Client:
    def __init__(self, id):
        self.id = id
        self.server_id = 1
        self.parent = -1
        self.msg_count = 0
        self.children = []
        self.child_count = 0
        self.message_storage = [None]*10

    def send_msg(self, node_objects, init_time):
        msg = Message(f"hello_{self.msg_count}_({self.id})")
        msg.sender_ip = self.id
        msg.sender_mac = self.id
        msg.dest_ip = self.server_id
        msg.dest_mac = self.server_id
        self.msg_count = self.msg_count+1
        msg.init_time = init_time
        msg.nums_send = msg.nums_send + 1
        if self.parent != -1:
            print(f"{self.id} : Sending {msg.message} to {self.server_id} via {self.parent}.")
            print(f"{self.parent} is parent of {self.id}")
            node_objects[self.parent].recive(node_objects, msg)


    def recive(self, node_objects, msg):
        # self.msg_count = self.msg_count+1
        # self.msg = Message(f"Hello_{self.msg_count}")
        msg.sender_mac = self.id
        msg.dest_mac = self.parent
        print(f"{self.id} : recived msg from {msg.sender_ip} to send to {msg.dest_ip}, rightnow i have it.")

        # print(f"{self.id} : recived msg from {msg.sender_ip} to send to {msg.dest_ip} via {self.parent}.")
        self.message_storage[msg.sender_ip] = msg
        # if self.parent != -1:
        #     node_objects[self.parent].forward(node_objects, msg)


    def forward(self, node_objects, send_to):
        # self.msg_count = self.msg_count+1
        # self.msg = Message(f"Hello_{self.msg_count}")
        # msg.sender_mac = self.id
        # msg.dest_mac = self.parent
        if(self.message_storage[send_to] == None):
            print(f"Unofficial {self.id}: i do not have file from {send_to} - so doing nothing")
            return
        # print(f"{self.id} : forwarding msg from {msg.sender_ip} to {msg.dest_ip} via {self.parent}.")
        else:
            # msg retrive kiya
            msg = self.message_storage[send_to]
            msg.nums_rec = msg.nums_rec + 1
            msg.nums_send = msg.nums_send + 1
            self.message_storage[send_to] = None
            if self.parent != -1:
                node_objects[self.parent].recive(node_objects, msg)

    def getTrxCount(self):
        return {self.msg_count}