pl.allegro.tech.servicemesh.envoycontrol.config.containers.ToxiproxyContainer.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.

There is a newer version: 0.22.4

package pl.allegro.tech.servicemesh.envoycontrol.config.containers

import eu.rekawek.toxiproxy.ToxiproxyClient
import org.testcontainers.containers.wait.strategy.Wait
import pl.allegro.tech.servicemesh.envoycontrol.config.testcontainers.GenericContainer
import java.util.LinkedList

class ToxiproxyContainer(exposedPortsCount: Int = 0) :
    GenericContainer("shopify/toxiproxy:2.1.4") {

    companion object {
        const val internalToxiproxyPort = 8474
    }

    val client by lazy { ToxiproxyClient("localhost", getMappedPort(internalToxiproxyPort)) }

    private val freeExposedPorts = (1..exposedPortsCount).map { portOffset ->
        val port = internalToxiproxyPort + portOffset
        this.addExposedPort(port)
        port
    }.let { LinkedList(it) }

    override fun configure() {
        super.configure()
        this.addExposedPort(internalToxiproxyPort)
        waitingFor(Wait.forHttp("/version").forPort(internalToxiproxyPort))
    }

    fun createProxy(targetIp: String, targetPort: Int): String {
        val listenPort = freeExposedPorts.pop()
        client.createProxy(
            "$targetIp:$targetPort",
            "$allInterfaces:$listenPort",
            "$targetIp:$targetPort"
        ).enable()
        return "http://$containerIpAddress:${getMappedPort(listenPort)}"
    }
}