ru.fix.armeria.dynamic.request.endpoint.DynamicAddressEndpoints.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jfix-armeria-dynamic-request Show documentation

https://github.com/ru-fix/

There is a newer version: 1.2.0

Show newest version

package ru.fix.armeria.dynamic.request.endpoint

import com.linecorp.armeria.client.Endpoint
import com.linecorp.armeria.client.endpoint.DynamicEndpointGroup
import com.linecorp.armeria.client.endpoint.EndpointGroup
import com.linecorp.armeria.client.endpoint.EndpointSelectionStrategy
import ru.fix.dynamic.property.api.DynamicProperty
import java.util.concurrent.CompletableFuture

class SocketAddress(
    val host: String,
    val port: Int
) {
    fun asEndpoint(): Endpoint = Endpoint.of(host, port)
}

object DynamicAddressEndpoints {

    @JvmStatic
    fun dynamicAddressEndpoint(addressProperty: DynamicProperty): EndpointGroup =
        DynamicAddressEndpoint(addressProperty)

    @JvmStatic
    fun dynamicAddressListEndpointGroup(
        addressListProperty: DynamicProperty>,
        endpointSelectionStrategy: EndpointSelectionStrategy = defaultEndpointSelectionStrategy
    ): EndpointGroup =
        DynamicAddressListEndpointGroup(addressListProperty, endpointSelectionStrategy)

}

private val defaultEndpointSelectionStrategy = EndpointSelectionStrategy.roundRobin()

private abstract class BaseDynamicPropertyEndpointGroup(
    property: DynamicProperty,
    asEndpoints: T.() -> List,
    endpointSelectionStrategy: EndpointSelectionStrategy = defaultEndpointSelectionStrategy
) : DynamicEndpointGroup(endpointSelectionStrategy) {
    private val propertySubscription = property.createSubscription()

    init {
        propertySubscription.setAndCallListener { _, newValue ->
            val newEndpoints = newValue.asEndpoints()
            setEndpoints(newEndpoints)
        }
    }

    override fun doCloseAsync(future: CompletableFuture<*>) {
        propertySubscription.close()
        super.doCloseAsync(future)
    }
}

private class DynamicAddressEndpoint(addressProperty: DynamicProperty) :
    BaseDynamicPropertyEndpointGroup(addressProperty, asEndpoints = { listOf(asEndpoint()) })

private class DynamicAddressListEndpointGroup(
    addressListProperty: DynamicProperty>,
    endpointSelectionStrategy: EndpointSelectionStrategy = defaultEndpointSelectionStrategy
) :
    BaseDynamicPropertyEndpointGroup>(
        addressListProperty,
        asEndpoints = { this.map { it.asEndpoint() } },
        endpointSelectionStrategy = endpointSelectionStrategy
    )