pl.allegro.tech.servicemesh.envoycontrol.config.envoy.CallStats.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.envoy

import pl.allegro.tech.servicemesh.envoycontrol.config.service.EchoServiceExtension

class CallStats(private val serviceExtensions: List) {
    var failedHits: Int = 0
    var totalHits: Int = 0

    private var containerHits: MutableMap =
        serviceExtensions.associate { it.container().containerId to 0 }.toMutableMap()

    fun hits(extension: EchoServiceExtension) = containerHits[extension.container().containerId] ?: 0

    fun addResponse(response: ResponseWithBody) {
        serviceExtensions
            .map { it.container() }
            .firstOrNull { response.isFrom(it) }
            ?.let { containerHits.compute(it.containerId) { _, i -> i?.inc() } }
        if (!response.isOk()) failedHits++
        totalHits++
    }
}