82 lines
3 KiB
Scala
Executable file
82 lines
3 KiB
Scala
Executable file
/**
|
|
* Copyright (C) 2009 Scalable Solutions.
|
|
*/
|
|
|
|
package se.scalablesolutions.akka.rest
|
|
|
|
import se.scalablesolutions.akka.config.ConfiguratorRepository
|
|
import se.scalablesolutions.akka.util.Logging
|
|
|
|
import com.sun.jersey.api.core.ResourceConfig
|
|
import com.sun.jersey.spi.container.servlet.ServletContainer
|
|
import com.sun.jersey.spi.container.WebApplication
|
|
|
|
import javax.servlet.{ServletConfig}
|
|
import javax.servlet.http.{HttpServletRequest, HttpServletResponse}
|
|
|
|
import org.atmosphere.cpr.{AtmosphereServletProcessor, AtmosphereEvent}
|
|
import org.atmosphere.cpr.AtmosphereServlet.AtmosphereHandlerWrapper
|
|
import org.atmosphere.container.GrizzlyCometSupport
|
|
import org.atmosphere.handler.ReflectorServletProcessor
|
|
import org.atmosphere.core.{JerseyBroadcaster}
|
|
|
|
import java.net.URLClassLoader
|
|
import java.io.InputStream
|
|
|
|
import scala.collection.jcl.Conversions._
|
|
|
|
/**
|
|
* @author <a href="http://jonasboner.com">Jonas Bonér</a>
|
|
*/
|
|
class AkkaServlet extends ServletContainer with AtmosphereServletProcessor with Logging {
|
|
|
|
override def initiate(rc: ResourceConfig, wa: WebApplication) = {
|
|
akka.Kernel.boot // will boot if not already booted by 'main'
|
|
val configurators = ConfiguratorRepository.getConfigurators
|
|
|
|
rc.getClasses.addAll(configurators.flatMap(_.getComponentInterfaces))
|
|
rc.getProperties.put("com.sun.jersey.spi.container.ResourceFilters", akka.Config.config.getString("akka.rest.filters").getOrElse(""))
|
|
|
|
wa.initiate(rc, new ActorComponentProviderFactory(configurators))
|
|
}
|
|
|
|
// Borrowed from AbstractReflectorAtmosphereHandler
|
|
override def onMessage(event: AtmosphereEvent[HttpServletRequest, HttpServletResponse]): AtmosphereEvent[_, _] = {
|
|
|
|
val response = event.getResponse
|
|
val data = if(event.getMessage ne null) event.getMessage.toString else null
|
|
val isUsingStream = try {
|
|
response.getWriter
|
|
false
|
|
} catch { case e: IllegalStateException => true }
|
|
|
|
if (isUsingStream) {
|
|
if (data != null) response.getOutputStream.write(data.getBytes)
|
|
response.getOutputStream.flush
|
|
} else {
|
|
if (data != null) response.getWriter.write(data)
|
|
response.getWriter.flush
|
|
}
|
|
event
|
|
}
|
|
|
|
override def onEvent(event: AtmosphereEvent[HttpServletRequest, HttpServletResponse]): AtmosphereEvent[_, _] = {
|
|
event.getRequest.setAttribute(ReflectorServletProcessor.ATMOSPHERE_EVENT, event)
|
|
event.getRequest.setAttribute(ReflectorServletProcessor.ATMOSPHERE_HANDLER, this)
|
|
service(event.getRequest, event.getResponse)
|
|
event
|
|
}
|
|
}
|
|
|
|
class AkkaCometServlet extends org.atmosphere.cpr.AtmosphereServlet {
|
|
override def init(sconf: ServletConfig) = {
|
|
val servlet = new AkkaServlet
|
|
config = new AtmosphereConfig { ah = servlet }
|
|
atmosphereHandlers.put("/*", new AtmosphereHandlerWrapper(servlet, new JerseyBroadcaster))
|
|
setCometSupport(new GrizzlyCometSupport(config))
|
|
getCometSupport.init(sconf)
|
|
servlet.init(sconf)
|
|
}
|
|
|
|
override def loadAtmosphereDotXml(is: InputStream, urlc: URLClassLoader) = () //Hide it
|
|
}
|