deployment.DeploymentLoader.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evaluation-proxy-core Show documentation
Core package for Amplitude's evaluation proxy.
The newest version!
package com.amplitude.deployment

import com.amplitude.cohort.CohortLoader
import com.amplitude.util.Loader
import com.amplitude.util.getAllCohortIds
import com.amplitude.util.getFlagVersion
import com.amplitude.util.isNewerThan
import com.amplitude.util.logger
import kotlinx.coroutines.launch

internal class DeploymentLoader(
    private val deploymentApi: DeploymentApi,
    private val deploymentStorage: DeploymentStorage,
    private val cohortLoader: CohortLoader,
) {
    companion object {
        val log by logger()
    }

    private val loader = Loader()

    suspend fun loadDeployment(deploymentKey: String) {
        log.trace("loadDeployment: - deploymentKey=$deploymentKey")
        loader.load(deploymentKey) {
            val networkFlags = deploymentApi.getFlagConfigs(deploymentKey)
            // Remove flags that are no longer deployed.
            val networkFlagKeys = networkFlags.map { it.key }.toSet()
            val storageFlags = deploymentStorage.getAllFlags(deploymentKey)
            val storageFlagKeys = storageFlags.map { it.key }.toSet()
            for (flagToRemove in storageFlagKeys - networkFlagKeys) {
                log.debug("Removing flag: $flagToRemove")
                deploymentStorage.removeFlag(deploymentKey, flagToRemove)
            }
            // Load cohorts for each flag independently then put the
            // flag into storage.
            for (flag in networkFlags) {
                val cohortIds = flag.getAllCohortIds()
                val storageFlag = storageFlags[flag.key]
                // Download cohorts for flags with cohorts and put the flag into storage
                if (!flag.isNewerThan(storageFlag)) continue
                log.debug("Found newer flag: ${flag.key} v${flag.getFlagVersion()} newer than v${storageFlag?.getFlagVersion()}")
                if (cohortIds.isNotEmpty()) {
                    launch {
                        cohortLoader.loadCohorts(cohortIds)
                        deploymentStorage.putFlag(deploymentKey, flag)
                    }
                } else {
                    deploymentStorage.putFlag(deploymentKey, flag)
                }
            }
        }
        log.trace("loadDeployment: end - deploymentKey=$deploymentKey")
    }
}