Merge pull request faif#34 from HanWenfang/master

Sakis Kasampalis · Sakis Kasampalis · commit ee37556a3c2e · 2013-12-25T16:37:44.000-08:00
publish_subscribe pattern
diff --git a/README.md b/README.md
@@ -31,4 +31,5 @@ Current Patterns:
 * template
 * command		
 * memento		
-* visitor
+* visitor
+* publish_subscribe
diff --git a/publish_subscribe.py b/publish_subscribe.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+
+'''
+Reference: http://www.slideshare.net/ishraqabd/publish-subscribe-model-overview-13368808
+Author: https://github.com/HanWenfang
+'''
+
+class Provider:
+	def __init__(self):
+		self.msgQueue = []
+		self.subscribers = {}
+
+	def notify(self, msg):
+		self.msgQueue.append(msg)
+
+	def subscribe(self,msg, subscriber):
+		if not msg in self.subscribers:
+			self.subscribers[msg] = []
+			self.subscribers[msg].append(subscriber) #unfair
+		else:
+			self.subscribers[msg].append(subscriber)
+	
+	def unSubscribe(self,msg, subscriber):
+		self.subscribers[msg].remove(subscriber)
+
+	def update(self):
+		for msg in self.msgQueue:
+			if msg in self.subscribers:
+				for sub in self.subscribers[msg]:
+					sub.run(msg)
+		self.msgQueue = []
+
+class Publisher:
+	def __init__(self, msgCenter):
+		self.provider = msgCenter
+	
+	def publish(self, msg):
+		self.provider.notify(msg)
+
+
+class Subscriber:
+	def __init__(self,name,msgCenter):
+		self.name = name
+		self.provider = msgCenter
+
+	def subscribe(self, msg):
+		self.provider.subscribe(msg, self)
+
+	def run(self, msg):
+		print "%s got %s"%(self.name, msg)
+
+
+def main():
+	messageCenter = Provider()
+
+	fftv = Publisher(messageCenter)
+
+	jim = Subscriber("jim", messageCenter)
+	jim.subscribe("cartoon")
+	jack = Subscriber("jack", messageCenter)
+	jack.subscribe("music")
+	gee = Subscriber("gee", messageCenter)
+	gee.subscribe("movie")
+
+	fftv.publish("cartoon")
+	fftv.publish("music")
+	fftv.publish("ads")
+	fftv.publish("movie")
+	fftv.publish("cartoon")
+	fftv.publish("cartoon")
+	fftv.publish("movie")
+	fftv.publish("blank")
+
+	messageCenter.update()
+
+if __name__ == "__main__":
+	main()
