agent/MessageSystem.py at master · AssertTrue/agent · 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
import threading

class PrivateMessageSystem(threading.Thread):

    def __init__(self):

        threading.Thread.__init__(self)
        self.accessLock = threading.Lock()
        self.nextIdentifier = 1
        self.agents = dict()
        self.start()

    def send(self, aFromIdentifier, aToIdentifier, aMessage):

        agent = None

        with self.accessLock:

            if aToIdentifier in self.agents.keys():

                agent = self.agents[aToIdentifier]

        if not agent is None:

            agent._message(aFromIdentifier, aMessage)

    def register(self, aAgent):

        identifier= 0

        with self.accessLock:

            identifier = self.nextIdentifier
            self.nextIdentifier = self.nextIdentifier + 1
            self.agents[identifier] = aAgent

        return identifier

    def unregister(self, aId):

        with self.accessLock:

            if aId in self.agents.keys():

                del self.agents[aId]

messageSystem = PrivateMessageSystem()

def send(aFromIdentifier, aToIdentifier, aMessage):

    messageSystem.send(aFromIdentifier, aToIdentifier, aMessage)

def register(aAgent):

    return messageSystem.register(aAgent)

def unregister(aId):

    messageSystem.unregister(aId)