Consolidated akka-security, akka-rest, akka-comet and akka-servlet into akka-http

akka-http/src/main/scala/BootableCometActorService.scala

@@ -0,0 +1,72 @@
+/**
+ * Copyright (C) 2009-2010 Scalable Solutions AB <http://scalablesolutions.se>
+ */
+ 
+package se.scalablesolutions.akka.comet
+
+import com.sun.grizzly.http.SelectorThread
+import com.sun.grizzly.http.servlet.ServletAdapter
+import com.sun.grizzly.standalone.StaticStreamAlgorithm
+
+import javax.ws.rs.core.UriBuilder
+
+import se.scalablesolutions.akka.actor.BootableActorLoaderService
+import se.scalablesolutions.akka.util.{Bootable, Logging}
+
+/**
+ * Handles the Akka Comet Support (load/unload)
+ */
+trait BootableCometActorService extends Bootable with Logging {
+  self : BootableActorLoaderService =>
+  
+  import se.scalablesolutions.akka.config.Config._
+  
+  val REST_HOSTNAME = config.getString("akka.rest.hostname", "localhost")
+  val REST_URL = "http://" + REST_HOSTNAME
+  val REST_PORT = config.getInt("akka.rest.port", 9998)
+  
+  protected var jerseySelectorThread: Option[SelectorThread] = None
+  
+  abstract override def onLoad   = {
+    super.onLoad
+    if (config.getBool("akka.rest.service", true)) {
+    
+      val uri = UriBuilder.fromUri(REST_URL).port(REST_PORT).build()
+
+      val scheme = uri.getScheme
+      if (!scheme.equalsIgnoreCase("http")) throw new IllegalArgumentException(
+        "The URI scheme, of the URI " + REST_URL + ", must be equal (ignoring case) to 'http'")
+        
+      log.info("Attempting to start REST service on uri [%s]",uri)
+
+      val adapter = new ServletAdapter
+      adapter.setHandleStaticResources(true)
+      adapter.setServletInstance(new AkkaServlet)
+      adapter.setContextPath(uri.getPath)
+      adapter.addInitParameter("cometSupport", "org.atmosphere.container.GrizzlyCometSupport")
+      if (HOME.isDefined) adapter.addRootFolder(HOME.get + "/deploy/root")
+      log.info("REST service root path [%s] and context path [%s]", adapter.getRootFolders, adapter.getContextPath)
+
+      val ah = new com.sun.grizzly.arp.DefaultAsyncHandler
+      ah.addAsyncFilter(new com.sun.grizzly.comet.CometAsyncFilter)
+      jerseySelectorThread = Some(new SelectorThread).map { t =>
+          t.setAlgorithmClassName(classOf[StaticStreamAlgorithm].getName)
+          t.setPort(REST_PORT)
+          t.setAdapter(adapter)
+          t.setEnableAsyncExecution(true)
+          t.setAsyncHandler(ah)
+          t.listen
+          t 
+      }
+      log.info("REST service started successfully. Listening to port [%s]", REST_PORT)
+    }
+  }
+  
+  abstract override def onUnload = {
+    super.onUnload    
+    if (jerseySelectorThread.isDefined) {
+      log.info("Shutting down REST service (Jersey)")
+      jerseySelectorThread.get.stopEndpoint
+    }
+  }
+}