|
5 | 5 | Author: https://github.com/HanWenfang |
6 | 6 | ''' |
7 | 7 |
|
| 8 | + |
8 | 9 | class Provider: |
9 | | - def __init__(self): |
10 | | - self.msgQueue = [] |
11 | | - self.subscribers = {} |
12 | | - |
13 | | - def notify(self, msg): |
14 | | - self.msgQueue.append(msg) |
15 | | - |
16 | | - def subscribe(self,msg, subscriber): |
17 | | - if not msg in self.subscribers: |
18 | | - self.subscribers[msg] = [] |
19 | | - self.subscribers[msg].append(subscriber) #unfair |
20 | | - else: |
21 | | - self.subscribers[msg].append(subscriber) |
22 | | - |
23 | | - def unSubscribe(self,msg, subscriber): |
24 | | - self.subscribers[msg].remove(subscriber) |
25 | | - |
26 | | - def update(self): |
27 | | - for msg in self.msgQueue: |
28 | | - if msg in self.subscribers: |
29 | | - for sub in self.subscribers[msg]: |
30 | | - sub.run(msg) |
31 | | - self.msgQueue = [] |
| 10 | + def __init__(self): |
| 11 | + self.msgQueue = [] |
| 12 | + self.subscribers = {} |
| 13 | + |
| 14 | + def notify(self, msg): |
| 15 | + self.msgQueue.append(msg) |
| 16 | + |
| 17 | + def subscribe(self, msg, subscriber): |
| 18 | + if not msg in self.subscribers: |
| 19 | + self.subscribers[msg] = [] |
| 20 | + self.subscribers[msg].append(subscriber) #unfair |
| 21 | + else: |
| 22 | + self.subscribers[msg].append(subscriber) |
| 23 | + |
| 24 | + def unSubscribe(self, msg, subscriber): |
| 25 | + self.subscribers[msg].remove(subscriber) |
| 26 | + |
| 27 | + def update(self): |
| 28 | + for msg in self.msgQueue: |
| 29 | + if msg in self.subscribers: |
| 30 | + for sub in self.subscribers[msg]: |
| 31 | + sub.run(msg) |
| 32 | + self.msgQueue = [] |
| 33 | + |
32 | 34 |
|
33 | 35 | class Publisher: |
34 | | - def __init__(self, msgCenter): |
35 | | - self.provider = msgCenter |
36 | | - |
37 | | - def publish(self, msg): |
38 | | - self.provider.notify(msg) |
| 36 | + def __init__(self, msgCenter): |
| 37 | + self.provider = msgCenter |
| 38 | + |
| 39 | + def publish(self, msg): |
| 40 | + self.provider.notify(msg) |
39 | 41 |
|
40 | 42 |
|
41 | 43 | class Subscriber: |
42 | | - def __init__(self,name,msgCenter): |
43 | | - self.name = name |
44 | | - self.provider = msgCenter |
| 44 | + def __init__(self, name, msgCenter): |
| 45 | + self.name = name |
| 46 | + self.provider = msgCenter |
45 | 47 |
|
46 | | - def subscribe(self, msg): |
47 | | - self.provider.subscribe(msg, self) |
| 48 | + def subscribe(self, msg): |
| 49 | + self.provider.subscribe(msg, self) |
48 | 50 |
|
49 | | - def run(self, msg): |
50 | | - print "%s got %s"%(self.name, msg) |
| 51 | + def run(self, msg): |
| 52 | + print "%s got %s" % (self.name, msg) |
51 | 53 |
|
52 | 54 |
|
53 | 55 | def main(): |
54 | | - messageCenter = Provider() |
| 56 | + messageCenter = Provider() |
| 57 | + |
| 58 | + fftv = Publisher(messageCenter) |
55 | 59 |
|
56 | | - fftv = Publisher(messageCenter) |
| 60 | + jim = Subscriber("jim", messageCenter) |
| 61 | + jim.subscribe("cartoon") |
| 62 | + jack = Subscriber("jack", messageCenter) |
| 63 | + jack.subscribe("music") |
| 64 | + gee = Subscriber("gee", messageCenter) |
| 65 | + gee.subscribe("movie") |
57 | 66 |
|
58 | | - jim = Subscriber("jim", messageCenter) |
59 | | - jim.subscribe("cartoon") |
60 | | - jack = Subscriber("jack", messageCenter) |
61 | | - jack.subscribe("music") |
62 | | - gee = Subscriber("gee", messageCenter) |
63 | | - gee.subscribe("movie") |
| 67 | + fftv.publish("cartoon") |
| 68 | + fftv.publish("music") |
| 69 | + fftv.publish("ads") |
| 70 | + fftv.publish("movie") |
| 71 | + fftv.publish("cartoon") |
| 72 | + fftv.publish("cartoon") |
| 73 | + fftv.publish("movie") |
| 74 | + fftv.publish("blank") |
64 | 75 |
|
65 | | - fftv.publish("cartoon") |
66 | | - fftv.publish("music") |
67 | | - fftv.publish("ads") |
68 | | - fftv.publish("movie") |
69 | | - fftv.publish("cartoon") |
70 | | - fftv.publish("cartoon") |
71 | | - fftv.publish("movie") |
72 | | - fftv.publish("blank") |
| 76 | + messageCenter.update() |
73 | 77 |
|
74 | | - messageCenter.update() |
75 | 78 |
|
76 | 79 | if __name__ == "__main__": |
77 | | - main() |
| 80 | + main() |
0 commit comments