-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Environment variable HTTP4S_BACKEND to help development
- Loading branch information
Showing
3 changed files
with
58 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
53 changes: 46 additions & 7 deletions
53
...s/src/main/scala/com.snowplowanalytics.snowplow.collectors.scalastream/CollectorApp.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,19 +1,58 @@ | ||
package com.snowplowanalytics.snowplow.collectors.scalastream | ||
|
||
import cats.implicits._ | ||
import cats.effect.{ExitCode, IO} | ||
import cats.effect.kernel.Resource | ||
import com.comcast.ip4s.IpLiteralSyntax | ||
import org.http4s.server.Server | ||
import org.http4s.ember.server.EmberServerBuilder | ||
import org.http4s.blaze.server.BlazeServerBuilder | ||
import org.http4s.netty.server.NettyServerBuilder | ||
import org.typelevel.log4cats.Logger | ||
import org.typelevel.log4cats.slf4j.Slf4jLogger | ||
|
||
import java.net.InetSocketAddress | ||
import scala.concurrent.duration.DurationLong | ||
|
||
object CollectorApp { | ||
|
||
implicit private def unsafeLogger: Logger[IO] = | ||
Slf4jLogger.getLogger[IO] | ||
|
||
def run(): IO[ExitCode] = | ||
buildHttpServer().use(_ => IO.never).as(ExitCode.Success) | ||
|
||
private def buildHttpServer() = | ||
EmberServerBuilder | ||
.default[IO] | ||
.withHost(ipv4"0.0.0.0") | ||
.withPort(port"8080") | ||
.withHttpApp(new CollectorRoutes[IO].value) | ||
.build | ||
private def buildHttpServer(): Resource[IO, Server] = | ||
sys.env.get("HTTP4S_BACKEND").map(_.toUpperCase()) match { | ||
case Some("EMBER") | None => buildEmberServer | ||
case Some("BLAZE") => buildBlazeServer | ||
case Some("NETTY") => buildNettyServer | ||
case Some(other) => throw new IllegalArgumentException(s"Unrecognized http4s backend $other") | ||
} | ||
|
||
private def buildEmberServer = | ||
Resource.eval(Logger[IO].info("Building ember server")) >> | ||
EmberServerBuilder | ||
.default[IO] | ||
.withHost(ipv4"0.0.0.0") | ||
.withPort(port"8080") | ||
.withHttpApp(new CollectorRoutes[IO].value) | ||
.withIdleTimeout(610.seconds) | ||
.build | ||
|
||
private def buildBlazeServer: Resource[IO, Server] = | ||
Resource.eval(Logger[IO].info("Building blaze server")) >> | ||
BlazeServerBuilder[IO] | ||
.bindSocketAddress(new InetSocketAddress(8080)) | ||
.withHttpApp(new CollectorRoutes[IO].value) | ||
.withIdleTimeout(610.seconds) | ||
.resource | ||
|
||
private def buildNettyServer: Resource[IO, Server] = | ||
Resource.eval(Logger[IO].info("Building netty server")) >> | ||
NettyServerBuilder[IO] | ||
.bindLocal(8080) | ||
.withHttpApp(new CollectorRoutes[IO].value) | ||
.withIdleTimeout(610.seconds) | ||
.resource | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters