Homec4science
Diffusion SHRINE MedCo Fork (branches/1.22.5) Browse

shrine-medco/apps/dashboard-app/src/main/scala/net/shrine/dashboard/httpclient/HttpClientDirectives.scala
30e27302f5bbbranches/1.22.5

Last Change
Raw File

HttpClientDirectives.scala
View Options

package net.shrine.dashboard.httpclient
import java.io.InputStream
import net.shrine.log.Loggable
import spray.can.Http
import akka.io.IO
import akka.actor.{ActorRef, ActorSystem}
import spray.http.{HttpEntity, StatusCodes, HttpRequest, HttpResponse, Uri}
import spray.routing.{RequestContext, Route}
import akka.pattern.ask
import scala.concurrent.{Await, Future, blocking}
import scala.concurrent.duration.DurationInt
import scala.language.postfixOps
import scala.concurrent.ExecutionContext.Implicits.global
/**
* From https://github.com/bthuillier/spray/commit/d31fc1b5e1415e1b908fe7d1f01f364a727e2593 with extra bits from http://www.cakesolutions.net/teamblogs/http-proxy-with-spray .
* Replace when Spray has its own version.
*
* @author david
* @since 9/14/15
*/
trait HttpClientDirectives extends Loggable {
/**
* Proxy the request to the specified base uri appended with the unmatched path.
*
*/
def forwardUnmatchedPath(baseUri: Uri)(implicit system: ActorSystem): Route = {
def completeWithEntityAsString(httpResponse:HttpResponse,uri:Uri):Route = {
ctx => {
ctx.complete(httpResponse.entity.asString)
}
}
requestWithUnmatchedPath(baseUri,handleCommonErrorsOrRoute(completeWithEntityAsString))
}
/**
* Make the request to the specified base uri appended with the unmatched path, then use the returned entity (as a string) to complete the route.
*
*/
def requestWithUnmatchedPath(baseUri:Uri, route:(HttpResponse,Uri) => Route)(implicit system: ActorSystem): Route = {
ctx => {
val resourceUri = baseUri.withPath(baseUri.path.++(ctx.unmatchedPath))
val httpResponse = if(resourceUri.scheme == "classpath") ClasspathResourceHttpClient.loadFromResource(resourceUri.path.toString())
else HttpClient.webApiCall(HttpRequest(ctx.request.method,resourceUri))
route(httpResponse,resourceUri)(ctx)
}
}
/**
* proxy the request to the specified uri with the unmatched path, then use the returned entity (as a string) to complete the route.
*
*/
def requestUriThenRoute(resourceUri:Uri, route:(HttpResponse,Uri) => Route)(implicit system: ActorSystem): Route = {
ctx => {
val httpResponse = if(resourceUri.scheme == "classpath") ClasspathResourceHttpClient.loadFromResource(resourceUri.path.toString())
else HttpClient.webApiCall(HttpRequest(ctx.request.method,resourceUri))
route(httpResponse,resourceUri)(ctx)
}
}
def handleCommonErrorsOrRoute(route:(HttpResponse,Uri) => Route)(httpResponse: HttpResponse,uri:Uri): Route = {
ctx => {
if(httpResponse.status != StatusCodes.OK) {
//todo create and report a problem
val ctxCopy: RequestContext = ctx.withHttpResponseMapped(_.copy(status = httpResponse.status))
ctxCopy.complete(s"$uri replied with $httpResponse")
}
else route(httpResponse,uri)(ctx)
}
}
}
object HttpClientDirectives extends HttpClientDirectives
/**
* A simple HttpClient to use inside the HttpDirectives
*/
object HttpClient extends Loggable {
def webApiCall(request:HttpRequest)(implicit system: ActorSystem): HttpResponse = {
val transport: ActorRef = IO(Http)(system)
debug(s"Requesting $request")
blocking {
val future:Future[HttpResponse] = for {
response <- transport.ask(request)(10 seconds).mapTo[HttpResponse]
} yield response
Await.result(future,10 seconds)
}
}
}
/**
* For testing, get an HttpResponse for a classpath resource
*/
object ClasspathResourceHttpClient {
def loadFromResource(resourceName:String):HttpResponse = {
blocking {
val cleanResourceName = if (resourceName.startsWith ("/") ) resourceName.drop(1)
else resourceName
val classLoader = getClass.getClassLoader
val is: InputStream = classLoader.getResourceAsStream (cleanResourceName)
val string:String = scala.io.Source.fromInputStream (is).mkString
HttpResponse(entity = HttpEntity(string))
}
}
}
c4science · Help