Improve events and web sockets

[lgpl/argeo-commons.git] / org.argeo.cms.ee / src / org / argeo / cms / websocket / server / EventEndpoint.java

diff --git a/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/EventEndpoint.java b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/EventEndpoint.java
new file mode 100644 (file)
index 0000000..a6b2a4d
--- /dev/null
+++ b/org.argeo.cms.ee/src/org/argeo/cms/websocket/server/EventEndpoint.java
@@ -0,0 +1,50 @@
+package org.argeo.cms.websocket.server;
+
+import java.io.IOException;
+import java.util.Map;
+
+import javax.websocket.OnClose;
+import javax.websocket.OnOpen;
+import javax.websocket.RemoteEndpoint;
+import javax.websocket.Session;
+import javax.websocket.server.PathParam;
+import javax.websocket.server.ServerEndpoint;
+
+import org.argeo.api.cms.CmsContext;
+import org.argeo.api.cms.CmsEventSubscriber;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.FrameworkUtil;
+
+@ServerEndpoint(value = "/event/{topic}", configurator = CmsWebSocketConfigurator.class)
+public class EventEndpoint implements CmsEventSubscriber {
+       private BundleContext bc = FrameworkUtil.getBundle(TestEndpoint.class).getBundleContext();
+
+       private RemoteEndpoint.Basic remote;
+       private CmsContext cmsContext;
+
+//     private String topic = "cms";
+
+       @OnOpen
+       public void onOpen(Session session, @PathParam("topic") String topic) {
+               if (bc != null) {
+                       cmsContext = bc.getService(bc.getServiceReference(CmsContext.class));
+                       cmsContext.addEventSubscriber(topic, this);
+               }
+               remote = session.getBasicRemote();
+
+       }
+
+       @OnClose
+       public void onClose(@PathParam("topic") String topic) {
+               cmsContext.removeEventSubscriber(topic, this);
+       }
+
+       @Override
+       public void onEvent(String topic, Map<String, Object> properties) {
+               try {
+                       remote.sendText(topic + ": " + properties);
+               } catch (IOException e) {
+                       throw new IllegalStateException(e);
+               }
+       }
+}