TechEmpower.FrameworkBenchmarks/frameworks/Scala/fintrospect/src/main/scala/FintrospectBenchmarkServer.scala

import java.util.TimeZone.getTimeZone

import com.twitter.finagle.http.path.Root
import com.twitter.finagle.http.{Request, Response}
import com.twitter.finagle.stats.NullStatsReceiver
import com.twitter.finagle.tracing.NullTracer
import com.twitter.finagle.{Filter, Http}
import com.twitter.util.{Await, NullMonitor}
import io.fintrospect.ModuleSpec
import io.fintrospect.configuration.Host
import io.fintrospect.renderers.simplejson.SimpleJson
import org.apache.commons.lang.time.FastDateFormat.getInstance

import scala.util.Properties

object FintrospectBenchmarkServer extends App {

  val dateFormat = getInstance("EEE, d MMM yyyy HH:mm:ss 'GMT'", getTimeZone("GMT"))

  val addServerAndDate = Filter.mk[Request, Response, Request, Response] { (req, svc) =>
    svc(req).map(resp => {
      resp.headerMap("Server") = "Example"
      resp.headerMap("Date") = dateFormat.format(System.currentTimeMillis())
      resp
    })
  }

  val dbHost = Properties.envOrNone("DBHOST").map(Host(_)).getOrElse(Host.localhost)
  val database = Database(dbHost)

  val module = ModuleSpec(Root, SimpleJson())
    .withRoute(JsonRoute())
    .withRoute(PlainTextRoute())
    .withRoute(FortunesRoute(database))
    .withRoutes(DatabaseRoutes(database))

  Await.ready(
    Http.server
      .withCompressionLevel(0)
      .withStatsReceiver(NullStatsReceiver)
      .withTracer(NullTracer)
      .withMonitor(NullMonitor)
      .serve(":9000", addServerAndDate.andThen(module.toService))
  )
}