com.pubnub.internal.eventengine.EffectDispatcher.kt Maven / Gradle / Ivy

Go to download
package com.pubnub.internal.eventengine

import org.slf4j.LoggerFactory
import java.util.concurrent.ConcurrentHashMap
import java.util.concurrent.ExecutorService
import java.util.concurrent.Executors

internal class EffectDispatcher(
    private val effectFactory: EffectFactory,
    private val effectSource: Source,
    private val managedEffects: ConcurrentHashMap = ConcurrentHashMap(),
    private val executorService: ExecutorService = Executors.newSingleThreadExecutor(),
) {
    private val log = LoggerFactory.getLogger(EffectDispatcher::class.java)

    fun start() {
        executorService.submit {
            try {
                while (true) {
                    val invocation = effectSource.take()
                    dispatch(invocation)
                }
            } catch (e: InterruptedException) {
                Thread.currentThread().interrupt()
            }
        }
    }

    fun stop() {
        executorService.shutdownNow()
    }

    internal fun dispatch(effectInvocation: T) {
        log.trace("Dispatching effect: $effectInvocation")
        when (val type = effectInvocation.type) {
            is Cancel -> {
                managedEffects.remove(type.idToCancel)?.cancel()
            }

            is Managed -> {
                managedEffects.remove(effectInvocation.id)?.cancel()
                val managedEffect = effectFactory.create(effectInvocation) as? ManagedEffect ?: return
                managedEffects[effectInvocation.id] = managedEffect
                managedEffect.runEffect()
            }

            is NonManaged -> {
                effectFactory.create(effectInvocation)?.runEffect()
            }
        }
    }
}