pl.allegro.tech.servicemesh.envoycontrol.config.consul.ConsulOperations.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of envoy-control-tests Show documentation
Production-ready Control Plane for Service Mesh based on Envoy Proxy.
The newest version!
package pl.allegro.tech.servicemesh.envoycontrol.config.consul

import com.ecwid.consul.v1.ConsulClient
import com.ecwid.consul.v1.QueryParams
import com.ecwid.consul.v1.agent.model.NewService
import com.ecwid.consul.v1.catalog.model.CatalogService
import pl.allegro.tech.servicemesh.envoycontrol.config.envoy.EnvoyContainer
import pl.allegro.tech.servicemesh.envoycontrol.config.envoy.EnvoyExtension
import pl.allegro.tech.servicemesh.envoycontrol.config.service.ServiceExtension
import java.util.UUID

class ConsulOperations(port: Int) {

    private val client = ConsulClient("localhost", port)

    fun registerService(
        id: String = UUID.randomUUID().toString(),
        name: String,
        address: String,
        port: Int,
        registerDefaultCheck: Boolean = false,
        tags: List = listOf("a"),
        beforeRegistration: (NewService) -> Unit = {}
    ): String {
        val service = NewService().also {
            it.id = id
            it.name = name
            it.address = address
            it.port = port
            it.tags = tags
            it.check = if (registerDefaultCheck) NewService.Check().also { check ->
                check.http = "http://$address:$port"
                check.interval = "3s"
            } else NewService.Check()
        }
        beforeRegistration.invoke(service)
        client.agentServiceRegister(service)
        return service.id
    }

    fun registerService(
        extension: ServiceExtension<*>,
        id: String = UUID.randomUUID().toString(),
        name: String,
        registerDefaultCheck: Boolean = false,
        tags: List = listOf("a"),
        beforeRegistration: (NewService) -> Unit = {}
    ) = registerService(
        id,
        name,
        extension.container().ipAddress(),
        extension.container().port(),
        registerDefaultCheck,
        tags,
        beforeRegistration
    )

    fun registerServiceWithEnvoyOnIngress(
        extension: EnvoyExtension,
        id: String = UUID.randomUUID().toString(),
        name: String,
        registerDefaultCheck: Boolean = false,
        tags: List = listOf("a")
    ) = registerService(
        id,
        name,
        extension.container.ipAddress(),
        EnvoyContainer.INGRESS_LISTENER_CONTAINER_PORT,
        registerDefaultCheck,
        tags
    )

    fun registerServiceWithEnvoyOnEgress(
        extension: EnvoyExtension,
        id: String = UUID.randomUUID().toString(),
        name: String,
        registerDefaultCheck: Boolean = false,
        tags: List = listOf("a")
    ) = registerService(
        id,
        name,
        extension.container.ipAddress(),
        EnvoyContainer.EGRESS_LISTENER_CONTAINER_PORT,
        registerDefaultCheck,
        tags
    )

    fun deregisterService(id: String) {
        client.agentServiceDeregister(id)
    }

    fun getService(serviceName: String, params: QueryParams = QueryParams.DEFAULT): List =
        client.getCatalogService(serviceName, params).value ?: emptyList()

    fun deregisterAll() {
        registeredServices().forEach { deregisterService(it) }
    }

    fun anyRpcOperation(): String {
        return leader()
    }

    fun leader(): String {
        return client.statusLeader.value
    }

    fun peers(): List {
        return client.statusPeers.value
    }

    private fun registeredServices() =
        client.agentServices
            .value
            .values
            .filter { !it.service.contains("envoy-control") }
            .map { it.id }
}