raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 11

tweaks...

Browse source
This commit is contained in:
Viktor Klang 2009-07-29 00:10:53 +02:00
parent aad4a54b8b
commit 3adb704c40
5 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

21
kernel/src/main/scala/jersey/AkkaServlet.scala
View file

@ -45,11 +45,12 @@ class AkkaServlet extends ServletContainer with AtmosphereServletProcessor with
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)
if(event.getMessage != null)
{
var isUsingStream = false
try {
@ -58,16 +59,19 @@ class AkkaServlet extends ServletContainer with AtmosphereServletProcessor with
case e: IllegalStateException => isUsingStream = true
}
val data = event.getMessage.toString
if (isUsingStream){
event.getResponse.getOutputStream.write(event.getMessage.toString.getBytes);
if(data != null)
event.getResponse.getOutputStream.write(data.getBytes)
event.getResponse.getOutputStream.flush
} else {
event.getResponse.getWriter.write(event.getMessage.toString)
event.getResponse.getWriter.write(data)
event.getResponse.getWriter.flush
}
}
else
log.info(new NullPointerException, "Null event message :/")
log.info("Null event message :/ req[ " + event.getRequest + "] res[" +event.getResponse + "]")
event
}
@ -84,14 +88,15 @@ class AkkaServlet extends ServletContainer with AtmosphereServletProcessor with
}
}
class AkkaCometServlet extends org.atmosphere.cpr.AtmosphereServlet with Logging
class AkkaCometServlet extends org.atmosphere.cpr.AtmosphereServlet
{
override def init(sconf : ServletConfig) = {
val servlet = new AkkaServlet
this.config = new AtmosphereConfig
atmosphereHandlers.put("", new AtmosphereHandlerWrapper(servlet,new JerseyBroadcaster()))
setCometSupport(new GrizzlyCometSupport(new AtmosphereConfig{ ah = servlet }))
atmosphereHandlers.put("", new AtmosphereHandlerWrapper(servlet,new JerseyBroadcaster))
setCometSupport(new GrizzlyCometSupport(config))
getCometSupport.init(sconf)
servlet.init(sconf)