Skip to content
Snippets Groups Projects
Select Git revision
  • main default protected
  • v0.0.3
  • IotAmak-0.0.1
  • 0.0.2
4 results
Blame Permalink
scheduler.py 2.99 KiB
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109 






from time import sleep

from schedulable import Schedulable


class Scheduler(Schedulable):

    def __init__(self):

        Schedulable.__init__(self, client_id="Scheduler")
        self.sleep_between_cycle = 5

        self.nbr_agent = 0
        self.subscribe("amas/agent/new", self.update_nbr_agent)
        self.subscribe("amas/action_done", self.update_schedulable)
        self.subscribe("env/action_done", self.update_schedulable)

        self.agent_waiting = 0
        self.schedulable_waiting = 0

        print("Init done")

    def update_schedulable(self, client, userdata, message):
        self.schedulable_waiting += 1
        print("__Schedulable is waiting")

    def update_nbr_agent(self, client, userdata, message):
        self.nbr_agent += 1
        self.subscribe("agent/" + str(message.payload.decode("utf-8")) + "/cycle_done", self.agent_done)
        print("__Update agent : ",self.nbr_agent, str(message.payload.decode("utf-8")))

    def agent_done(self, client, userdata, message):
        self.agent_waiting += 1
        print("__Agent done")

    def wait_agent(self):
        while self.agent_waiting < self.nbr_agent:
            sleep(self.wait_delay)
        self.agent_waiting = 0

    def wait_schedulable(self):
        while self.schedulable_waiting < 2:
            sleep(self.wait_delay)
        self.schedulable_waiting = 0

    def first_part(self) -> None:
        """
        first part of a cycle
        """
        self.client.publish("scheduler/schedulable/wakeup", "")
        # Amas on cycle begin
        # Environment on cycle begin
        self.wait_schedulable()

    def main_part(self) -> None:
        """
        main part of a cycle
        """
        self.client.publish("scheduler/agent/wakeup", "")
        # Agent doing phase 1
        self.wait_agent()
        self.client.publish("scheduler/agent/wakeup", "")
        # agent doing phase 2
        self.wait_agent()

    def last_part(self) -> None:
        """
        last part of a cycle
        """
        self.client.publish("scheduler/schedulable/wakeup", "")
        # Amas on cycle end
        # Environment on cycle end
        self.wait_schedulable()

    def run(self) -> None:
        """
        main part of amak core
        """

        # wait that all schedulable have init
        print("Waiting schedulable")
        self.wait_schedulable()
        # Wait that all agent have init
        print("Waiting agents :", self.nbr_agent)
        self.wait_agent()

        while not self.exit_bool:
            print("Cycle : ", self.nbr_cycle)

            print("-First part")
            self.first_part()
            print("-Main part")
            self.main_part()
            print("-Last part")
            self.last_part()

            # sleep(self.sleep_between_cycle)
            input()
            self.nbr_cycle += 1

        self.client.publish("scheduler/schedulable/wakeup", "")
        self.client.publish("scheduler/agent/wakeup", "")
        sleep(2)


if __name__ == '__main__':
    s = Scheduler()
    s.run()