main.com.wisetrack.sdk.scheduler.SingleThreadFutureScheduler.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdk-core
WiseTrack Android Library
The newest version!
package com.wisetrack.sdk.scheduler

import com.wisetrack.sdk.WiseTrackFactory
import java.util.concurrent.Callable
import java.util.concurrent.ScheduledFuture
import java.util.concurrent.ScheduledThreadPoolExecutor
import java.util.concurrent.TimeUnit

/**
@author hamed (@hamed-hsb)
 * @since 09th October 2021
 */

class SingleThreadFutureScheduler(source: String, doKeepAlive: Boolean) : FutureScheduler {
    private var scheduledThreadPoolExecutor: ScheduledThreadPoolExecutor? = null

    init {
        scheduledThreadPoolExecutor = ScheduledThreadPoolExecutor(
            1,
            ThreadFactoryWrapper(source),
            { runnable, _ ->
                // Logs rejected runnable rejected from the entering the pool
                WiseTrackFactory.getLogger().warn(
                    "Runnable [${runnable}] rejected from [$source] "
                )
            }
        )
        if (!doKeepAlive) {
            scheduledThreadPoolExecutor!!.setKeepAliveTime(10L, TimeUnit.MILLISECONDS)
            scheduledThreadPoolExecutor!!.allowCoreThreadTimeOut(true)
        }
    }

    override fun scheduleFuture(command: Runnable?, millisecondDelay: Long): ScheduledFuture<*>? {
        return scheduledThreadPoolExecutor!!.schedule(
            RunnableWrapper(command),
            millisecondDelay,
            TimeUnit.MILLISECONDS
        )
    }

    override fun scheduleFutureWithFixedDelay(
        command: Runnable?,
        initialMillisecondDelay: Long,
        millisecondDelay: Long
    ): ScheduledFuture<*>? {
        return scheduledThreadPoolExecutor!!.scheduleWithFixedDelay(
            RunnableWrapper(command),
            initialMillisecondDelay,
            millisecondDelay,
            TimeUnit.MILLISECONDS
        )

    }

    override fun  scheduleFutureWithReturn(
        callable: Callable?,
        millisecondDelay: Long
    ): ScheduledFuture? {
        return scheduledThreadPoolExecutor!!.schedule(Callable {
            try {
                return@Callable callable!!.call()
            } catch (t: Throwable) {
                WiseTrackFactory.getLogger().error(
                    "Callable error [${t.message}] of type [${t.javaClass.canonicalName}]"
                )
                return@Callable null
            }
        }, millisecondDelay, TimeUnit.MILLISECONDS)
    }

    override fun teardown() {
        scheduledThreadPoolExecutor!!.shutdownNow()
    }
}