Update smalltalk.java

Author: m0ver

src/tinystruct/examples/smalltalk.java

@@ -22,6 +22,8 @@
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import javax.servlet.http.HttpSessionEvent;
+import javax.servlet.http.HttpSessionListener;
 
 import org.tinystruct.AbstractApplication;
 import org.tinystruct.ApplicationException;
@@ -34,7 +36,7 @@
 import org.tinystruct.transfer.http.upload.ContentDisposition;
 import org.tinystruct.transfer.http.upload.MultipartFormData;
 
-public class smalltalk extends AbstractApplication {
+public class smalltalk extends AbstractApplication implements HttpSessionListener {
 
   private static final long TIMEOUT = 1000;
   private final Map<String, Map<String, Queue<Builder>>> groups = Collections.synchronizedMap(new HashMap<String, Map<String, Queue<Builder>>>());
@@ -72,7 +74,7 @@ public smalltalk index() {
       if ((sessions = this.groups.get(meeting_code)) == null) {
         this.groups.put(meeting_code.toString(), sessions = new HashMap<String, Queue<Builder>>());
       }
-      
+
       final String sessionId = request.getSession().getId();
       if (sessions.get(sessionId) == null) {
         sessions.put(sessionId, new ArrayDeque<Builder>());
@@ -86,7 +88,7 @@ public smalltalk index() {
     if ((topic = this.getVariable(meeting_code.toString())) != null) {
       this.setVariable("topic", topic.getValue().toString().replaceAll("[\r\n]", "<br />"), true);
     }
-    
+
     return this;
   }
 
@@ -311,5 +313,44 @@ protected String filter(String text) {
   public String version() {
     return "Welcome to use tinystruct 2.0";
   }
-  
+
+  @Override
+  public void sessionCreated(HttpSessionEvent arg0) {
+    Object meeting_code = arg0.getSession().getAttribute("meeting_code");
+    if ( meeting_code == null ) {
+      meeting_code = java.util.UUID.randomUUID().toString();
+      arg0.getSession().setAttribute("meeting_code", meeting_code);
+
+      System.out.println("New meeting generated by HttpSessionListener:" + meeting_code);
+    }
+
+    Map<String, Queue<Builder>> sessions;
+    synchronized (groups) {
+      if ((sessions = groups.get(meeting_code)) == null) {
+        groups.put(meeting_code.toString(), sessions = new HashMap<String, Queue<Builder>>());
+      }
+
+      final String sessionId = arg0.getSession().getId();
+      if (sessions.get(sessionId) == null) {
+        sessions.put(sessionId, new ArrayDeque<Builder>());
+      }
+
+      groups.notifyAll();
+    }
+  }
+
+  @Override
+  public void sessionDestroyed(HttpSessionEvent arg0) {
+    Object meeting_code = arg0.getSession().getAttribute("meeting_code");
+    String sessionId = arg0.getSession().getId();
+    if ( meeting_code != null ) {
+      Map<String, Queue<Builder>> sessions;
+      synchronized (groups) {
+        if ((sessions = groups.get(meeting_code)) != null) {
+          sessions.remove(sessionId);
+          groups.notifyAll();
+        }
+      }
+    }
+  }
 }