hooks/zmq.py
changeset 8211 543e1579ba0d
child 8350 e1c05bf6fdeb
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hooks/zmq.py	Fri Feb 10 16:20:35 2012 +0100
@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# copyright 2012 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
+#
+# This file is part of CubicWeb.
+#
+# CubicWeb is free software: you can redistribute it and/or modify it under the
+# terms of the GNU Lesser General Public License as published by the Free
+# Software Foundation, either version 2.1 of the License, or (at your option)
+# any later version.
+#
+# CubicWeb is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Lesser General Public License along
+# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
+
+from cubicweb.server import hook
+
+class ZMQStopHook(hook.Hook):
+    __regid__ = 'zmqstop'
+    events = ('server_shutdown',)
+
+    def __call__(self):
+        self.repo.app_instances_bus.stop()
+
+class ZMQStartHook(hook.Hook):
+    __regid__ = 'zmqstart'
+    events = ('server_startup',)
+
+    def __call__(self):
+        config = self.repo.config
+        address_pub = config.get('zmq-address-pub')
+        if not address_pub:
+            return
+        from cubicweb.server import cwzmq
+        self.repo.app_instances_bus = cwzmq.ZMQComm()
+        self.repo.app_instances_bus.add_publisher(address_pub)
+        def clear_cache_callback(msg):
+            self.debug('clear_caches: %s', ' '.join(msg))
+            self.repo.clear_caches(msg[1:])
+        self.repo.app_instances_bus.add_subscription('delete', clear_cache_callback)
+        for address in config.get('zmq-address-sub'):
+            self.repo.app_instances_bus.add_subscriber(address)
+        self.repo.app_instances_bus.start()
+