com.pubnub.api.presence.eventengine.effect.DelayedHeartbeatEffect.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pubnub-kotlin Show documentation

PubNub is a cross-platform client-to-client (1:1 and 1:many) push service in the cloud, capable of broadcasting real-time messages to millions of web and mobile clients simultaneously, in less than a quarter second!

There is a newer version: 10.2.0

Show newest version

package com.pubnub.api.presence.eventengine.effect

import com.pubnub.api.PubNubException
import com.pubnub.api.endpoints.remoteaction.RemoteAction
import com.pubnub.api.eventengine.ManagedEffect
import com.pubnub.api.eventengine.Sink
import com.pubnub.api.presence.eventengine.event.PresenceEvent
import com.pubnub.api.subscribe.eventengine.effect.RetryPolicy
import com.pubnub.extension.scheduleWithDelay
import org.slf4j.LoggerFactory
import java.util.concurrent.ScheduledExecutorService
import java.util.concurrent.ScheduledFuture

internal class DelayedHeartbeatEffect(
    private val heartbeatRemoteAction: RemoteAction,
    private val presenceEventSink: Sink,
    private val policy: RetryPolicy,
    private val executorService: ScheduledExecutorService,
    private val delayedHeartbeatInvocation: PresenceEffectInvocation.DelayedHeartbeat
) : ManagedEffect {
    private val log = LoggerFactory.getLogger(DelayedHeartbeatEffect::class.java)

    @Transient
    private var scheduled: ScheduledFuture<*>? = null

    @Transient
    private var cancelled = false

    @Synchronized
    override fun runEffect() {
        log.trace("Running DelayedHeartbeatEffect")
        if (cancelled) {
            return
        }

        val delay = policy.nextDelay(delayedHeartbeatInvocation.attempts)
        if (delay == null) {
            presenceEventSink.add(PresenceEvent.HeartbeatGiveup(delayedHeartbeatInvocation.reason ?: PubNubException("Unknown error")))
            return
        }

        scheduled = executorService.scheduleWithDelay(delay) {
            heartbeatRemoteAction.async { _, status ->
                if (status.error) {
                    presenceEventSink.add(PresenceEvent.HeartbeatFailure(status.exception ?: PubNubException("Unknown error")))
                } else {
                    presenceEventSink.add(PresenceEvent.HeartbeatSuccess)
                }
            }
        }
    }

    @Synchronized
    override fun cancel() {
        if (cancelled) {
            return
        }
        cancelled = true
        heartbeatRemoteAction.silentCancel()
        scheduled?.cancel(true)
    }
}