net.spals.appbuilder.app.examples.finatra.sample.SampleFinatraWebApp.scala Maven / Gradle / Ivy

Go to download
package net.spals.appbuilder.app.examples.finatra.sample

import com.twitter.finatra.http.routing.HttpRouter
import net.spals.appbuilder.app.finatra.FinatraWebApp
import net.spals.appbuilder.config.service.ServiceScan
import net.spals.appbuilder.filestore.core.FileStore
import net.spals.appbuilder.graph.model.ServiceGraphFormat
import net.spals.appbuilder.keystore.core.KeyStore
import net.spals.appbuilder.mapstore.core.MapStore
import net.spals.appbuilder.message.core.{MessageConsumer, MessageProducer}
import net.spals.appbuilder.model.core.ModelSerializer
import org.mockito.Mockito.mock

/**
  * A full sample [[FinatraWebApp]] which uses all default services
  * and bindings.
  *
  * @author tkral
  */
object SampleFinatraWebAppMain extends SampleFinatraWebApp

class SampleFinatraWebApp extends FinatraWebApp {

  // Use mock HttpRouter for testing purposes. Normally,
  // this would use the production HttpRouter shipped with
  // Finatra
  val mockRouter = mock(classOf[HttpRouter])

  enableServiceGraph(ServiceGraphFormat.ASCII)
  setServiceConfigFromClasspath("config/sample-finatra-service.conf")
  setServiceScan(new ServiceScan.Builder()
    .addServicePackages("net.spals.appbuilder.app.examples.finatra.sample")
    .addDefaultServices(classOf[FileStore])
    .addDefaultServices(classOf[KeyStore])
    .addDefaultServices(classOf[MapStore])
    .addDefaultServices(classOf[MessageConsumer], classOf[MessageProducer])
    .addDefaultServices(classOf[ModelSerializer])
    .build())
  addModule(new SampleFinatraGuiceModule)
  addModule(new SampleFinatraTwitterModule)
  build()

  // Example of override for standard Finstra hooks and callbacks
  override protected def configureHttp(router: HttpRouter): Unit = {
    // Swap in the mock router for testing purposes
    super.configureHttp(mockRouter)
  }
}