|
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