73 lines
2.3 KiB
Scala
73 lines
2.3 KiB
Scala
/**
|
|
* Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
|
|
*/
|
|
|
|
package akka.http
|
|
|
|
import javax.ws.rs.core.UriBuilder
|
|
import javax.servlet.ServletConfig
|
|
import java.io.File
|
|
|
|
import akka.actor.BootableActorLoaderService
|
|
import akka.util.Bootable
|
|
|
|
import org.eclipse.jetty.xml.XmlConfiguration
|
|
import org.eclipse.jetty.server.{ Handler, Server }
|
|
import org.eclipse.jetty.server.handler.{ HandlerList, HandlerCollection, ContextHandler }
|
|
import java.net.URL
|
|
import akka.AkkaException
|
|
|
|
/**
|
|
* Handles the Akka Comet Support (load/unload)
|
|
*/
|
|
trait EmbeddedAppServer extends Bootable {
|
|
self: BootableActorLoaderService ⇒
|
|
|
|
import akka.config.Config._
|
|
|
|
val REST_HOSTNAME = config.getString("akka.http.hostname", "localhost")
|
|
val REST_PORT = config.getInt("akka.http.port", 9998)
|
|
|
|
val isRestEnabled = config.getList("akka.enabled-modules").exists(_ == "http")
|
|
|
|
protected var server: Option[Server] = None
|
|
|
|
protected def findJettyConfigXML: Option[URL] =
|
|
Option(applicationLoader.getOrElse(this.getClass.getClassLoader).getResource("microkernel-server.xml")) orElse
|
|
HOME.map(home ⇒ new File(home + "/config/microkernel-server.xml").toURI.toURL)
|
|
|
|
abstract override def onLoad = {
|
|
super.onLoad
|
|
if (isRestEnabled) {
|
|
|
|
val configuration = new XmlConfiguration(findJettyConfigXML.getOrElse(sys.error("microkernel-server.xml not found!")))
|
|
|
|
System.setProperty("jetty.port", REST_PORT.toString)
|
|
System.setProperty("jetty.host", REST_HOSTNAME)
|
|
|
|
HOME.foreach(home ⇒ System.setProperty("jetty.home", home + "/deploy/root"))
|
|
|
|
server = Option(configuration.configure.asInstanceOf[Server]) map { s ⇒ //Set the correct classloader to our contexts
|
|
applicationLoader foreach { loader ⇒
|
|
//We need to provide the correct classloader to the servlets
|
|
def setClassLoader(handlers: Seq[Handler]) {
|
|
handlers foreach {
|
|
case c: ContextHandler ⇒ c.setClassLoader(loader)
|
|
case c: HandlerCollection ⇒ setClassLoader(c.getHandlers)
|
|
case _ ⇒
|
|
}
|
|
}
|
|
setClassLoader(s.getHandlers)
|
|
}
|
|
//Start the server
|
|
s.start()
|
|
s
|
|
}
|
|
}
|
|
}
|
|
|
|
abstract override def onUnload = {
|
|
super.onUnload
|
|
server foreach { _.stop() }
|
|
}
|
|
}
|