org.http4k.k8s.Http4kK8sServer.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http4k-platform-k8s Show documentation

http4k K8S integration tooling

The newest version!

package org.http4k.k8s

import org.http4k.config.Environment
import org.http4k.config.Environment.Companion.ENV
import org.http4k.config.EnvironmentKey.k8s.HEALTH_PORT
import org.http4k.config.EnvironmentKey.k8s.SERVICE_PORT
import org.http4k.core.HttpHandler
import org.http4k.k8s.health.Health
import org.http4k.server.Http4kServer
import org.http4k.server.ServerConfig
import org.http4k.server.asServer

/**
 * A K8S server consists of a main application and a health application, running on 2 different ports.
 * This class provides unified start/stop control.
 */
class Http4kK8sServer(private val main: Http4kServer, private val health: Http4kServer) : Http4kServer {
    override fun port() = main.port()
    fun healthPort() = health.port()

    override fun start(): Http4kServer = apply {
        main.start()
        health.start()
    }

    override fun stop(): Http4kServer = apply {
        health.stop()
        main.stop()
    }
}

fun HttpHandler.asK8sServer(
    serverConfig: (port: Int) -> ServerConfig,
    port: Int = 8000,
    healthApp: HttpHandler = Health(),
    healthPort: Int = 8001
) = Http4kK8sServer(asServer(serverConfig(port)), healthApp.asServer(serverConfig(healthPort)))

fun HttpHandler.asK8sServer(
    serverConfig: (port: Int) -> ServerConfig,
    env: Environment = ENV,
    healthApp: HttpHandler = Health()
) = asK8sServer(serverConfig, SERVICE_PORT(env), healthApp, HEALTH_PORT(env))