91 lines
3.1 KiB
Scala
91 lines
3.1 KiB
Scala
/**
|
|
* Copyright (C) 2009-2010 Scalable Solutions AB <http://scalablesolutions.se>
|
|
*/
|
|
|
|
package se.scalablesolutions.akka.kernel
|
|
|
|
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 javax.servlet.ServletConfig
|
|
|
|
import se.scalablesolutions.akka.actor.BootableActorLoaderService
|
|
import se.scalablesolutions.akka.util.{Bootable, Logging}
|
|
import se.scalablesolutions.akka.comet.{ AkkaServlet }
|
|
|
|
/**
|
|
* Handles the Akka Comet Support (load/unload)
|
|
*/
|
|
trait EmbeddedAppServer 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 {
|
|
override def init(sc : ServletConfig) : Unit = {
|
|
val cl = Thread.currentThread.getContextClassLoader
|
|
try {
|
|
Thread.currentThread.setContextClassLoader(applicationLoader.get)
|
|
super.init(sc)
|
|
}
|
|
finally {
|
|
Thread.currentThread.setContextClassLoader(cl)
|
|
}
|
|
}
|
|
})
|
|
|
|
adapter.setContextPath(uri.getPath)
|
|
adapter.addInitParameter("cometSupport",
|
|
"org.atmosphere.container.GrizzlyCometSupport")
|
|
adapter.addInitParameter("com.sun.jersey.config.property.resourceConfigClass",
|
|
"com.sun.jersey.api.core.PackagesResourceConfig")
|
|
|
|
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
|
|
jerseySelectorThread foreach { (t) => {
|
|
log.info("Shutting down REST service (Jersey)")
|
|
t.stopEndpoint
|
|
}
|
|
}
|
|
}
|
|
}
|