69 lines
2.5 KiB
Kotlin
69 lines
2.5 KiB
Kotlin
package dev.inmo.krontab
|
|
|
|
import android.content.Context
|
|
import androidx.work.*
|
|
|
|
const val krontabTemplateWorkField = "krontabTemplate"
|
|
|
|
/**
|
|
* Will [enqueueKronSchedulerWork] with [KronScheduler] from [krontabTemplate] and call [setUpRequest] on setting up
|
|
* [OneTimeWorkRequest.Builder] with [Data] which will be used to [OneTimeWorkRequest.Builder.setInputData] after
|
|
* [setUpRequest] completed
|
|
*/
|
|
suspend inline fun <reified T : KrontabTemplateSchedulerWork> Context.enqueueKrontabTemplateSchedulerWork(
|
|
workName: String,
|
|
krontabTemplate: KrontabTemplate,
|
|
existingWorkPolicy: ExistingWorkPolicy = ExistingWorkPolicy.REPLACE,
|
|
noinline setUpRequest: suspend OneTimeWorkRequest.Builder.(
|
|
data: Data
|
|
) -> Unit = {}
|
|
) = enqueueKronSchedulerWork(workName, krontabTemplate.toKronScheduler(), T::class.java, existingWorkPolicy) {
|
|
val data = workDataOf(
|
|
krontabTemplateWorkField to krontabTemplate
|
|
)
|
|
setUpRequest(data)
|
|
setInputData(data)
|
|
}
|
|
|
|
/**
|
|
* Extend this class in case you wish to base on [KrontabTemplate]. It will automatically handle request of
|
|
* [kronScheduler] and put it in [setUpRequest]
|
|
*/
|
|
abstract class KrontabTemplateSchedulerWork(
|
|
context: Context,
|
|
workerParams: WorkerParameters
|
|
) : KronSchedulerWork(context, workerParams) {
|
|
/**
|
|
* Will try to get [KrontabTemplate] from [getInputData] by key [krontabTemplateWorkField]
|
|
*
|
|
* @see setUpRequest
|
|
*/
|
|
protected val krontabTemplate: KrontabTemplate?
|
|
get() = inputData.getString(krontabTemplateWorkField)
|
|
|
|
/**
|
|
* Override this methods instead of old [setUpRequest] in case you wish to set up some work request parameters
|
|
*
|
|
* @param data This parameter will be used to put data inside of [OneTimeWorkRequest.Builder] after this method
|
|
* will be completed
|
|
*/
|
|
protected open suspend fun OneTimeWorkRequest.Builder.setUpRequest(data: Data) {}
|
|
|
|
/**
|
|
* Will automatically put [krontabTemplate] into work data, call [setUpRequest] with future [Data] object and then
|
|
* call [OneTimeWorkRequest.Builder.setInputData] with that [Data] object
|
|
*/
|
|
override suspend fun OneTimeWorkRequest.Builder.setUpRequest() {
|
|
val data = workDataOf(
|
|
krontabTemplateWorkField to krontabTemplate,
|
|
)
|
|
setUpRequest(data)
|
|
setInputData(data)
|
|
}
|
|
|
|
/**
|
|
* Will return [KronScheduler] in case if [krontabTemplate] was not null
|
|
*/
|
|
override suspend fun kronScheduler(): KronScheduler? = krontabTemplate ?.toKronScheduler()
|
|
}
|