在Play Framework

时间:2015-12-22 04:18:40

标签: scala playframework proxy

如何使用Play Framework执行完整代理?

我想保持请求和响应的标题和正文完好无损。基本上,客户端和服务器都是透明的代理层。

注意:我有一些工作。当SO允许我发布时会发布。

1 个答案:

答案 0 :(得分:2)

这就是我最终的结果。

通过我的(不全面)测试,这适用于所有具有不同体型的方法。

请注意我对_.head的使用。我没有挖掘为什么标头具有类型Map[String, Seq[String]]。我可能会删除重复的标题内容(例如,标题中包含多个Content-Type)。也许加入Seq[String];是更好的方式。

import play.api.libs.ws._
import play.api.libs.iteratee.Enumerator
import play.api.mvc._

def proxy(proxyUrl: String) = Action.async(BodyParsers.parse.raw) { request =>
  // filter out the Host and potentially any other header we don't want
  val headers: Seq[(String, String)] = request.headers.toSimpleMap.toSeq.filter {
    case (headerStr, _) if headerStr != "Host" => true
    case _ => false
  }
  val wsRequestBase: WSRequestHolder = WS.url(s"http://localhost:9000/$proxyUrl") // set the proxy path
    .withMethod(request.method) // set our HTTP method
    .withHeaders(headers : _*) // Set our headers, function takes var args so we need to "explode" the Seq to var args
    .withQueryString(request.queryString.mapValues(_.head).toSeq: _*) // similarly for query strings
  // depending on whether we have a body, append it in our request
  val wsRequest: WSRequestHolder = request.body.asBytes() match {
    case Some(bytes) => wsRequestBase.withBody(bytes)
    case None => wsRequestBase
  }
  wsRequest
    .stream() // send the request. We want to process the response body as a stream
    .map { case (responseHeader: WSResponseHeaders, bodyStream: Enumerator[Array[Byte]]) => // we want to read the raw bytes for the body
      // Content stream is an enumerator. It's a 'stream' that generates Array[Byte]
      new Result(
        new ResponseHeader(responseHeader.status),
        bodyStream
      )
      .withHeaders(responseHeader.headers.mapValues(_.head).toSeq: _*)
    }
}

routes文件条目看起来像这样:

GET     /proxy/*proxyUrl                   @controllers.Application.proxy(proxyUrl: String)

您需要其他行来支持其他方法(例如POST)

随意建议编辑。