FinagleProxySample

FinagleProxySample.scala

package com.ponkotuy.proxy

import com.twitter.finagle.builder.ClientBuilder
import com.twitter.finagle.{Http, Service, http}
import org.jboss.netty.handler.codec.http.{HttpMethod, HttpResponse, HttpRequest}
import com.twitter.util.{Await, Future}
import com.github.theon.uri.Uri

class FinagleProxySample(hosts: String, port: Int, inter: Intercepter) {
  val client = ClientBuilder().codec(http.Http()).hosts(hosts).build()

  val service = new Service[HttpRequest, HttpResponse] {
    def apply(req: HttpRequest): Future[HttpResponse] = {
      val res = if (req.getMethod == HttpMethod.POST) {
        val uri = Uri.parseUri(req.getUri)
        req.setUri(uri.pathRaw)
        client.apply(req)
      } else {
        client.apply(req)
      }
      res.foreach(rs => inter.input(req, rs))
      res
    }
  }
  val server = Http.serve(s":$port", service)

  def start(): Unit = {
    Await.ready(server)
  }
}

trait Intercepter {
  def input(req: HttpRequest, res: HttpResponse): Unit
}