106 lines
3.7 KiB
Scala
106 lines
3.7 KiB
Scala
/**
|
|
* Copyright (C) 2009 Scalable Solutions.
|
|
*/
|
|
|
|
package se.scalablesolutions.akka.kernel.jersey
|
|
|
|
import kernel.Kernel
|
|
import config.ConfiguratorRepository
|
|
import util.Logging
|
|
|
|
import com.sun.jersey.api.core.{DefaultResourceConfig, ResourceConfig,ClasspathResourceConfig}
|
|
import com.sun.jersey.spi.container.servlet.ServletContainer
|
|
import com.sun.jersey.spi.container.WebApplication
|
|
|
|
import com.sun.jersey.server.impl.component.{IoCResourceFactory}
|
|
|
|
import javax.servlet.{ServletConfig}
|
|
import javax.servlet.http.{HttpServletRequest,HttpServletResponse}
|
|
|
|
import org.atmosphere.cpr.{AtmosphereServlet,AtmosphereServletProcessor,AtmosphereEvent,DefaultBroadcaster}
|
|
import org.atmosphere.cpr.AtmosphereServlet.{AtmosphereHandlerWrapper}
|
|
import org.atmosphere.container.GrizzlyCometSupport
|
|
import org.atmosphere.handler.ReflectorServletProcessor
|
|
import org.atmosphere.core.{JerseyBroadcaster}
|
|
|
|
import java.util.{HashSet, HashMap}
|
|
import java.net.{URL,URLClassLoader}
|
|
import java.io.{InputStream}
|
|
|
|
import scala.collection.jcl.Conversions._
|
|
|
|
|
|
class AkkaServlet extends ServletContainer with AtmosphereServletProcessor with Logging {
|
|
|
|
override def initiate(rc: ResourceConfig, wa: WebApplication) = {
|
|
|
|
Kernel.boot // will boot if not already booted by 'main'
|
|
val configurators = ConfiguratorRepository.getConfiguratorsFor(getServletContext)
|
|
|
|
rc.getClasses.addAll(configurators.flatMap(_.getComponentInterfaces))
|
|
rc.getProperties.put("com.sun.jersey.spi.container.ResourceFilters","org.atmosphere.core.AtmosphereFilter")
|
|
//rc.getFeatures.put("com.sun.jersey.config.feature.Redirect", true)
|
|
//rc.getFeatures.put("com.sun.jersey.config.feature.ImplicitViewables",true)
|
|
|
|
wa.initiate(rc,new ActorComponentProviderFactory(configurators))
|
|
}
|
|
|
|
//Borrowed from AbstractReflectorAtmosphereHandler
|
|
override def onMessage(event : AtmosphereEvent[HttpServletRequest,HttpServletResponse]) : AtmosphereEvent[_,_] =
|
|
{
|
|
//log.info("onMessage: " + event.getMessage.toString)
|
|
|
|
if(event.getMessage ne null)
|
|
{
|
|
var isUsingStream = false
|
|
try {
|
|
event.getResponse.getWriter
|
|
} catch {
|
|
case e: IllegalStateException => isUsingStream = true
|
|
}
|
|
|
|
val data = event.getMessage.toString
|
|
|
|
if (isUsingStream){
|
|
if(data != null)
|
|
event.getResponse.getOutputStream.write(data.getBytes)
|
|
event.getResponse.getOutputStream.flush
|
|
} else {
|
|
event.getResponse.getWriter.write(data)
|
|
event.getResponse.getWriter.flush
|
|
}
|
|
}
|
|
else
|
|
log.info("Null event message :/ req[ " + event.getRequest + "] res[" +event.getResponse + "]")
|
|
|
|
event
|
|
}
|
|
|
|
override def onEvent(event : AtmosphereEvent[HttpServletRequest,HttpServletResponse]) : AtmosphereEvent[_,_] =
|
|
{
|
|
//log.info("onEvent: " + event.getMessage)
|
|
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
|
|
this.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
|
|
}
|