org.analogweb.scala.ScalaResponseHandler.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of analogweb-scala_2.10 Show documentation

Analogweb Framework is tiny, simple, and pluggable web framework.

There is a newer version: 0.9.14

package org.analogweb.scala

import scala.util.{ Try, Success, Failure }
import scala.concurrent.Future
import org.analogweb._
import org.analogweb.core.{ DefaultResponseHandler, DefaultResponse }
import org.analogweb.scala.Execution.Implicits.defaultContext
import org.analogweb.scala.utils.Implicits._

class ScalaResponseHandler extends DefaultResponseHandler {

  override def handleResult(obj: Any, metadata: InvocationMetadata, resolver: RenderableResolver, request: RequestContext, response: ResponseContext, exh: ExceptionHandler, finder: ResponseFormatterFinder): Response = {
    val result = resolver.resolve(obj, metadata, request, response)
    def commit(a: Any) = super.handleResult(a, metadata, resolver, request, response, exh, finder).commit(request, response)
    result match {
      case r: RenderableFuture => {
        def futureResultHandler = { (request: RequestContext, response: ResponseContext) =>
          r.future.andThen { future =>
            future match {
              case Success(s) =>
                Try(commit(s)).eventually(response.ensure())
              case Failure(f) => f match {
                case e: Exception =>
                  Try(commit(exh.handleException(e))).eventually(response.ensure())
                case t => throw t
              }
            }
          }
        }
        new DefaultResponse {
          override def commit(request: RequestContext, response: ResponseContext) = {
            futureResultHandler(request, response)
          }
        }
      }
      case _ => {
        super.handleResult(result, metadata, resolver, request, response, exh, finder)
      }
    }
  }
}