deployment.DeploymentLoader.kt Maven / Gradle / Ivy

Go to download
package com.amplitude.deployment

import com.amplitude.cohort.CohortLoader
import com.amplitude.util.getCohortIds
import com.amplitude.util.logger
import kotlinx.coroutines.Job
import kotlinx.coroutines.coroutineScope
import kotlinx.coroutines.launch
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock

class DeploymentLoader(
    private val deploymentApi: DeploymentApi,
    private val deploymentStorage: DeploymentStorage,
    private val cohortLoader: CohortLoader
) {

    companion object {
        val log by logger()
    }

    private val jobsMutex = Mutex()
    private val jobs = mutableMapOf()

    suspend fun loadDeployment(deploymentKey: String) = coroutineScope {
        log.debug("loadDeployment: - deploymentKey=$deploymentKey")
        jobsMutex.withLock {
            jobs.getOrPut(deploymentKey) {
                launch {
                    val networkFlagConfigs = deploymentApi.getFlagConfigs(deploymentKey)
                    val storageFlagConfigs = deploymentStorage.getFlagConfigs(deploymentKey) ?: listOf()
                    val networkCohortIds = networkFlagConfigs.getCohortIds()
                    val storageCohortIds = storageFlagConfigs.getCohortIds()
                    val addedCohortIds = networkCohortIds - storageCohortIds
                    if (addedCohortIds.isNotEmpty()) {
                        cohortLoader.loadCohorts(addedCohortIds, networkCohortIds)
                    }
                    deploymentStorage.putFlagConfigs(deploymentKey, networkFlagConfigs)
                    jobs.remove(deploymentKey)
                }
            }
        }.join()
        log.debug("loadDeployment: end - deploymentKey=$deploymentKey")
    }
}