mirror of https://github.com/usbharu/Hideout.git
chore: KJOB依存を完全になくした
This commit is contained in:
parent
bb5cef6c22
commit
7324e0b0e1
|
|
@ -188,10 +188,8 @@ dependencies {
|
||||||
implementation(libs.bundles.exposed)
|
implementation(libs.bundles.exposed)
|
||||||
implementation(libs.bundles.coroutines)
|
implementation(libs.bundles.coroutines)
|
||||||
implementation(libs.bundles.ktor.client)
|
implementation(libs.bundles.ktor.client)
|
||||||
implementation(libs.bundles.serialization)
|
|
||||||
implementation(libs.bundles.apache.tika)
|
implementation(libs.bundles.apache.tika)
|
||||||
implementation(libs.bundles.openapi)
|
implementation(libs.bundles.openapi)
|
||||||
implementation(libs.bundles.kjob)
|
|
||||||
implementation(libs.bundles.owl.producer)
|
implementation(libs.bundles.owl.producer)
|
||||||
implementation(libs.bundles.spring.boot.oauth2)
|
implementation(libs.bundles.spring.boot.oauth2)
|
||||||
implementation(libs.bundles.spring.boot.data.mongodb)
|
implementation(libs.bundles.spring.boot.data.mongodb)
|
||||||
|
|
|
||||||
|
|
@ -16,7 +16,6 @@
|
||||||
|
|
||||||
package dev.usbharu.hideout.activitypub.service.activity.follow
|
package dev.usbharu.hideout.activitypub.service.activity.follow
|
||||||
|
|
||||||
import com.fasterxml.jackson.databind.ObjectMapper
|
|
||||||
import dev.usbharu.hideout.activitypub.domain.model.Follow
|
import dev.usbharu.hideout.activitypub.domain.model.Follow
|
||||||
import dev.usbharu.hideout.activitypub.service.common.AbstractActivityPubProcessor
|
import dev.usbharu.hideout.activitypub.service.common.AbstractActivityPubProcessor
|
||||||
import dev.usbharu.hideout.activitypub.service.common.ActivityPubProcessContext
|
import dev.usbharu.hideout.activitypub.service.common.ActivityPubProcessContext
|
||||||
|
|
@ -29,7 +28,6 @@ import org.springframework.stereotype.Service
|
||||||
@Service
|
@Service
|
||||||
class APFollowProcessor(
|
class APFollowProcessor(
|
||||||
transaction: Transaction,
|
transaction: Transaction,
|
||||||
private val objectMapper: ObjectMapper,
|
|
||||||
private val owlProducer: OwlProducer,
|
private val owlProducer: OwlProducer,
|
||||||
) :
|
) :
|
||||||
AbstractActivityPubProcessor<Follow>(transaction) {
|
AbstractActivityPubProcessor<Follow>(transaction) {
|
||||||
|
|
|
||||||
|
|
@ -1,331 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (C) 2024 usbharu
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package dev.usbharu.hideout.core.infrastructure.kjobexposed
|
|
||||||
|
|
||||||
import kjob.core.job.JobProgress
|
|
||||||
import kjob.core.job.JobSettings
|
|
||||||
import kjob.core.job.JobStatus
|
|
||||||
import kjob.core.job.ScheduledJob
|
|
||||||
import kjob.core.repository.JobRepository
|
|
||||||
import kotlinx.coroutines.Dispatchers
|
|
||||||
import kotlinx.coroutines.flow.Flow
|
|
||||||
import kotlinx.coroutines.flow.asFlow
|
|
||||||
import kotlinx.serialization.encodeToString
|
|
||||||
import kotlinx.serialization.json.*
|
|
||||||
import org.jetbrains.exposed.dao.id.LongIdTable
|
|
||||||
import org.jetbrains.exposed.sql.*
|
|
||||||
import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
|
|
||||||
import org.jetbrains.exposed.sql.SqlExpressionBuilder.inList
|
|
||||||
import org.jetbrains.exposed.sql.SqlExpressionBuilder.plus
|
|
||||||
import org.jetbrains.exposed.sql.transactions.experimental.newSuspendedTransaction
|
|
||||||
import org.jetbrains.exposed.sql.transactions.transaction
|
|
||||||
import java.time.Clock
|
|
||||||
import java.time.Instant
|
|
||||||
import java.util.*
|
|
||||||
|
|
||||||
class ExposedJobRepository(
|
|
||||||
private val database: Database,
|
|
||||||
private val tableName: String,
|
|
||||||
private val clock: Clock,
|
|
||||||
private val json: Json
|
|
||||||
) :
|
|
||||||
JobRepository {
|
|
||||||
|
|
||||||
class Jobs(tableName: String) : LongIdTable(tableName) {
|
|
||||||
val status = text("status")
|
|
||||||
val runAt = long("runAt").nullable()
|
|
||||||
val statusMessage = text("statusMessage").nullable()
|
|
||||||
val retries = integer("retries")
|
|
||||||
val kjobId = char("kjobId", 36).nullable()
|
|
||||||
val createdAt = long("createdAt")
|
|
||||||
val updatedAt = long("updatedAt")
|
|
||||||
val jobId = text("jobId")
|
|
||||||
val name = text("name")
|
|
||||||
val properties = text("properties").nullable()
|
|
||||||
val step = integer("step")
|
|
||||||
val max = integer("max").nullable()
|
|
||||||
val startedAt = long("startedAt").nullable()
|
|
||||||
val completedAt = long("completedAt").nullable()
|
|
||||||
}
|
|
||||||
|
|
||||||
val jobs: Jobs = Jobs(tableName)
|
|
||||||
|
|
||||||
fun createTable() {
|
|
||||||
transaction(database) {
|
|
||||||
SchemaUtils.create(jobs)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("InjectDispatcher")
|
|
||||||
suspend fun <T> query(block: suspend () -> T): T = newSuspendedTransaction(Dispatchers.IO) { block() }
|
|
||||||
|
|
||||||
override suspend fun completeProgress(id: String): Boolean {
|
|
||||||
val now = Instant.now(clock).toEpochMilli()
|
|
||||||
return query {
|
|
||||||
jobs.update({ jobs.id eq id.toLong() }) {
|
|
||||||
it[jobs.completedAt] = now
|
|
||||||
it[jobs.updatedAt] = now
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun exist(jobId: String): Boolean {
|
|
||||||
return query {
|
|
||||||
jobs.selectAll().where(jobs.jobId eq jobId).empty().not()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("SuspendFunWithFlowReturnType")
|
|
||||||
override suspend fun findNext(names: Set<String>, status: Set<JobStatus>, limit: Int): Flow<ScheduledJob> {
|
|
||||||
return query {
|
|
||||||
jobs.selectAll().where(
|
|
||||||
jobs.status.inList(list = status.map { it.name })
|
|
||||||
.and(if (names.isEmpty()) Op.TRUE else jobs.name.inList(names))
|
|
||||||
).limit(limit)
|
|
||||||
.map { it.toScheduledJob() }.asFlow()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun get(id: String): ScheduledJob? {
|
|
||||||
val single = query { jobs.selectAll().where(jobs.id eq id.toLong()).singleOrNull() } ?: return null
|
|
||||||
return single.toScheduledJob()
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun reset(id: String, oldKjobId: UUID?): Boolean {
|
|
||||||
return query {
|
|
||||||
jobs.update({
|
|
||||||
jobs.id eq id.toLong() and if (oldKjobId == null) {
|
|
||||||
jobs.kjobId.isNull()
|
|
||||||
} else {
|
|
||||||
jobs.kjobId eq oldKjobId.toString()
|
|
||||||
}
|
|
||||||
}) {
|
|
||||||
it[jobs.status] = JobStatus.CREATED.name
|
|
||||||
it[jobs.statusMessage] = null
|
|
||||||
it[jobs.kjobId] = null
|
|
||||||
it[jobs.step] = 0
|
|
||||||
it[jobs.max] = null
|
|
||||||
it[jobs.startedAt] = null
|
|
||||||
it[jobs.completedAt] = null
|
|
||||||
it[jobs.updatedAt] = Instant.now(clock).toEpochMilli()
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun save(jobSettings: JobSettings, runAt: Instant?): ScheduledJob {
|
|
||||||
val now = Instant.now(clock)
|
|
||||||
val scheduledJob =
|
|
||||||
ScheduledJob(
|
|
||||||
id = "",
|
|
||||||
status = JobStatus.CREATED,
|
|
||||||
runAt = runAt,
|
|
||||||
statusMessage = null,
|
|
||||||
retries = 0,
|
|
||||||
kjobId = null,
|
|
||||||
createdAt = now,
|
|
||||||
updatedAt = now,
|
|
||||||
settings = jobSettings,
|
|
||||||
progress = JobProgress(0)
|
|
||||||
)
|
|
||||||
val id = query {
|
|
||||||
jobs.insert {
|
|
||||||
it[jobs.status] = scheduledJob.status.name
|
|
||||||
it[jobs.createdAt] = scheduledJob.createdAt.toEpochMilli()
|
|
||||||
it[jobs.updatedAt] = scheduledJob.updatedAt.toEpochMilli()
|
|
||||||
it[jobs.jobId] = scheduledJob.settings.id
|
|
||||||
it[jobs.name] = scheduledJob.settings.name
|
|
||||||
it[jobs.properties] = scheduledJob.settings.properties.stringify()
|
|
||||||
it[jobs.runAt] = scheduledJob.runAt?.toEpochMilli()
|
|
||||||
it[jobs.statusMessage] = null
|
|
||||||
it[jobs.retries] = 0
|
|
||||||
it[jobs.kjobId] = null
|
|
||||||
it[jobs.step] = 0
|
|
||||||
it[jobs.max] = null
|
|
||||||
it[jobs.startedAt] = null
|
|
||||||
it[jobs.completedAt] = null
|
|
||||||
}[jobs.id].value
|
|
||||||
}
|
|
||||||
return scheduledJob.copy(id = id.toString())
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun setProgressMax(id: String, max: Long): Boolean {
|
|
||||||
val now = Instant.now(clock).toEpochMilli()
|
|
||||||
return query {
|
|
||||||
jobs.update({ jobs.id eq id.toLong() }) {
|
|
||||||
it[jobs.max] = max.toInt()
|
|
||||||
it[jobs.updatedAt] = now
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun startProgress(id: String): Boolean {
|
|
||||||
val now = Instant.now(clock).toEpochMilli()
|
|
||||||
return query {
|
|
||||||
jobs.update({ jobs.id eq id.toLong() }) {
|
|
||||||
it[jobs.startedAt] = now
|
|
||||||
it[jobs.updatedAt] = now
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun stepProgress(id: String, step: Long): Boolean {
|
|
||||||
val now = Instant.now(clock).toEpochMilli()
|
|
||||||
return query {
|
|
||||||
jobs.update({ jobs.id eq id.toLong() }) {
|
|
||||||
it[jobs.step] = jobs.step + step.toInt()
|
|
||||||
it[jobs.updatedAt] = now
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun update(
|
|
||||||
id: String,
|
|
||||||
oldKjobId: UUID?,
|
|
||||||
kjobId: UUID?,
|
|
||||||
status: JobStatus,
|
|
||||||
statusMessage: String?,
|
|
||||||
retries: Int
|
|
||||||
): Boolean {
|
|
||||||
return query {
|
|
||||||
jobs.update({
|
|
||||||
(jobs.id eq id.toLong()) and if (oldKjobId == null) {
|
|
||||||
jobs.kjobId.isNull()
|
|
||||||
} else {
|
|
||||||
jobs.kjobId eq oldKjobId.toString()
|
|
||||||
}
|
|
||||||
}) {
|
|
||||||
it[jobs.status] = status.name
|
|
||||||
it[jobs.retries] = retries
|
|
||||||
it[jobs.updatedAt] = Instant.now(clock).toEpochMilli()
|
|
||||||
it[jobs.id] = id.toLong()
|
|
||||||
it[jobs.statusMessage] = statusMessage
|
|
||||||
it[jobs.kjobId] = kjobId.toString()
|
|
||||||
} == 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("CyclomaticComplexMethod")
|
|
||||||
private fun String?.parseJsonMap(): Map<String, Any> {
|
|
||||||
this ?: return emptyMap()
|
|
||||||
return json.parseToJsonElement(this).jsonObject.mapValues { (_, el) ->
|
|
||||||
if (el is JsonObject) {
|
|
||||||
val t = el["t"]?.run { jsonPrimitive.content } ?: error("Cannot get jsonPrimitive")
|
|
||||||
val value = el["v"]?.jsonArray ?: error("Cannot get jsonArray")
|
|
||||||
when (t) {
|
|
||||||
"s" -> value.map { it.jsonPrimitive.content }
|
|
||||||
"d" -> value.map { it.jsonPrimitive.double }
|
|
||||||
"l" -> value.map { it.jsonPrimitive.long }
|
|
||||||
"i" -> value.map { it.jsonPrimitive.int }
|
|
||||||
"b" -> value.map { it.jsonPrimitive.boolean }
|
|
||||||
else -> error("Unknown type prefix '$t'")
|
|
||||||
}.toList()
|
|
||||||
} else {
|
|
||||||
val content = el.jsonPrimitive.content
|
|
||||||
val t = content.substringBefore(':')
|
|
||||||
val value = content.substringAfter(':')
|
|
||||||
when (t) {
|
|
||||||
"s" -> value
|
|
||||||
"d" -> value.toDouble()
|
|
||||||
"l" -> value.toLong()
|
|
||||||
"i" -> value.toInt()
|
|
||||||
"b" -> value.toBoolean()
|
|
||||||
else -> error("Unknown type prefix '$t'")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("CyclomaticComplexMethod")
|
|
||||||
private fun Map<String, Any>.stringify(): String? {
|
|
||||||
if (isEmpty()) {
|
|
||||||
return null
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("UNCHECKED_CAST")
|
|
||||||
fun listSerialize(value: List<*>): JsonElement {
|
|
||||||
return if (value.isEmpty()) {
|
|
||||||
buildJsonObject {
|
|
||||||
put("t", "s")
|
|
||||||
putJsonArray("v") {}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
val (t, values) = when (val item = value.first()) {
|
|
||||||
is Double -> "d" to (value as List<Double>).map(::JsonPrimitive)
|
|
||||||
is Long -> "l" to (value as List<Long>).map(::JsonPrimitive)
|
|
||||||
is Int -> "i" to (value as List<Int>).map(::JsonPrimitive)
|
|
||||||
is String -> "s" to (value as List<String>).map(::JsonPrimitive)
|
|
||||||
is Boolean -> "b" to (value as List<Boolean>).map(::JsonPrimitive)
|
|
||||||
else -> error("Cannot serialize unsupported list property value: $item")
|
|
||||||
}
|
|
||||||
buildJsonObject {
|
|
||||||
put("t", t)
|
|
||||||
put("v", JsonArray(values))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fun createJsonPrimitive(string: String, value: Any) = JsonPrimitive("$string:$value")
|
|
||||||
|
|
||||||
val jsonObject = JsonObject(
|
|
||||||
mapValues { (_, value) ->
|
|
||||||
when (value) {
|
|
||||||
is List<*> -> listSerialize(value)
|
|
||||||
is Double -> createJsonPrimitive("d", value)
|
|
||||||
is Long -> createJsonPrimitive("l", value)
|
|
||||||
is Int -> createJsonPrimitive("i", value)
|
|
||||||
is String -> createJsonPrimitive("s", value)
|
|
||||||
is Boolean -> createJsonPrimitive("b", value)
|
|
||||||
else -> error("Cannot serialize unsupported property value: $value")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
return json.encodeToString(jsonObject)
|
|
||||||
}
|
|
||||||
|
|
||||||
private fun ResultRow.toScheduledJob(): ScheduledJob {
|
|
||||||
val single = this
|
|
||||||
|
|
||||||
return ScheduledJob(
|
|
||||||
id = single[jobs.id].value.toString(),
|
|
||||||
status = JobStatus.valueOf(single[jobs.status]),
|
|
||||||
runAt = single[jobs.runAt]?.let { Instant.ofEpochMilli(it) },
|
|
||||||
statusMessage = single[jobs.statusMessage],
|
|
||||||
retries = single[jobs.retries],
|
|
||||||
kjobId = single[jobs.kjobId]?.let {
|
|
||||||
try {
|
|
||||||
@Suppress("SwallowedException")
|
|
||||||
UUID.fromString(it)
|
|
||||||
} catch (ignored: IllegalArgumentException) {
|
|
||||||
null
|
|
||||||
}
|
|
||||||
},
|
|
||||||
createdAt = Instant.ofEpochMilli(single[jobs.createdAt]),
|
|
||||||
updatedAt = Instant.ofEpochMilli(single[jobs.updatedAt]),
|
|
||||||
settings = JobSettings(
|
|
||||||
id = single[jobs.jobId],
|
|
||||||
name = single[jobs.name],
|
|
||||||
properties = single[jobs.properties].parseJsonMap()
|
|
||||||
),
|
|
||||||
progress = JobProgress(
|
|
||||||
step = single[jobs.step].toLong(),
|
|
||||||
max = single[jobs.max]?.toLong(),
|
|
||||||
startedAt = single[jobs.startedAt]?.let { Instant.ofEpochMilli(it) },
|
|
||||||
completedAt = single[jobs.completedAt]?.let { Instant.ofEpochMilli(it) }
|
|
||||||
)
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,65 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (C) 2024 usbharu
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package dev.usbharu.hideout.core.infrastructure.kjobexposed
|
|
||||||
|
|
||||||
import kjob.core.BaseKJob
|
|
||||||
import kjob.core.KJob
|
|
||||||
import kjob.core.KJobFactory
|
|
||||||
import kotlinx.coroutines.runBlocking
|
|
||||||
import org.jetbrains.exposed.sql.Database
|
|
||||||
import java.time.Clock
|
|
||||||
|
|
||||||
class ExposedKJob(config: Configuration) : BaseKJob<ExposedKJob.Configuration>(config) {
|
|
||||||
|
|
||||||
private val database: Database = config.connectionDatabase ?: Database.connect(
|
|
||||||
requireNotNull(config.connectionString),
|
|
||||||
requireNotNull(config.driverClassName)
|
|
||||||
)
|
|
||||||
|
|
||||||
override val jobRepository: ExposedJobRepository
|
|
||||||
get() = ExposedJobRepository(database, config.jobTableName, Clock.systemUTC(), config.json)
|
|
||||||
|
|
||||||
override val lockRepository: ExposedLockRepository
|
|
||||||
get() = ExposedLockRepository(database, config, clock)
|
|
||||||
|
|
||||||
override fun start(): KJob {
|
|
||||||
jobRepository.createTable()
|
|
||||||
lockRepository.createTable()
|
|
||||||
return super.start()
|
|
||||||
}
|
|
||||||
|
|
||||||
override fun shutdown(): Unit = runBlocking {
|
|
||||||
super.shutdown()
|
|
||||||
lockRepository.clearExpired()
|
|
||||||
}
|
|
||||||
|
|
||||||
companion object : KJobFactory<ExposedKJob, Configuration> {
|
|
||||||
override fun create(configure: Configuration.() -> Unit): KJob = ExposedKJob(Configuration().apply(configure))
|
|
||||||
}
|
|
||||||
|
|
||||||
class Configuration : BaseKJob.Configuration() {
|
|
||||||
var connectionString: String? = null
|
|
||||||
var driverClassName: String? = null
|
|
||||||
var connectionDatabase: Database? = null
|
|
||||||
|
|
||||||
var jobTableName: String = "kjobJobs"
|
|
||||||
|
|
||||||
var lockTableName: String = "kjobLocks"
|
|
||||||
|
|
||||||
var expireLockInMinutes: Long = 5L
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,91 +0,0 @@
|
||||||
/*
|
|
||||||
* Copyright (C) 2024 usbharu
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package dev.usbharu.hideout.core.infrastructure.kjobexposed
|
|
||||||
|
|
||||||
import kjob.core.job.Lock
|
|
||||||
import kjob.core.repository.LockRepository
|
|
||||||
import kotlinx.coroutines.Dispatchers
|
|
||||||
import org.jetbrains.exposed.dao.id.UUIDTable
|
|
||||||
import org.jetbrains.exposed.sql.*
|
|
||||||
import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
|
|
||||||
import org.jetbrains.exposed.sql.SqlExpressionBuilder.greater
|
|
||||||
import org.jetbrains.exposed.sql.transactions.experimental.newSuspendedTransaction
|
|
||||||
import org.jetbrains.exposed.sql.transactions.transaction
|
|
||||||
import java.time.Clock
|
|
||||||
import java.time.Instant
|
|
||||||
import java.util.*
|
|
||||||
import kotlin.time.Duration.Companion.minutes
|
|
||||||
|
|
||||||
class ExposedLockRepository(
|
|
||||||
private val database: Database,
|
|
||||||
private val config: ExposedKJob.Configuration,
|
|
||||||
private val clock: Clock
|
|
||||||
) : LockRepository {
|
|
||||||
|
|
||||||
class Locks(tableName: String) : UUIDTable(tableName) {
|
|
||||||
val updatedAt = long("updatedAt")
|
|
||||||
val expiresAt = long("expiresAt")
|
|
||||||
}
|
|
||||||
|
|
||||||
val locks: Locks = Locks(config.lockTableName)
|
|
||||||
|
|
||||||
fun createTable() {
|
|
||||||
transaction(database) {
|
|
||||||
SchemaUtils.create(locks)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Suppress("InjectDispatcher")
|
|
||||||
suspend fun <T> query(block: suspend () -> T): T = newSuspendedTransaction(Dispatchers.IO) { block() }
|
|
||||||
|
|
||||||
override suspend fun exists(id: UUID): Boolean {
|
|
||||||
val now = Instant.now(clock)
|
|
||||||
return query {
|
|
||||||
locks.selectAll().where(locks.id eq id and locks.expiresAt.greater(now.toEpochMilli())).empty().not()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun ping(id: UUID): Lock {
|
|
||||||
val now = Instant.now(clock)
|
|
||||||
val expiresAt = now.plusSeconds(config.expireLockInMinutes.minutes.inWholeSeconds)
|
|
||||||
val lock = Lock(id, now)
|
|
||||||
query {
|
|
||||||
if (locks.selectAll().where(locks.id eq id).limit(1)
|
|
||||||
.map { Lock(it[locks.id].value, Instant.ofEpochMilli(it[locks.expiresAt])) }.isEmpty()
|
|
||||||
) {
|
|
||||||
locks.insert {
|
|
||||||
it[locks.id] = id
|
|
||||||
it[locks.updatedAt] = now.toEpochMilli()
|
|
||||||
it[locks.expiresAt] = expiresAt.toEpochMilli()
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
locks.update({ locks.id eq id }) {
|
|
||||||
it[locks.updatedAt] = now.toEpochMilli()
|
|
||||||
it[locks.expiresAt] = expiresAt.toEpochMilli()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return lock
|
|
||||||
}
|
|
||||||
|
|
||||||
suspend fun clearExpired() {
|
|
||||||
val now = Instant.now(clock).toEpochMilli()
|
|
||||||
query {
|
|
||||||
locks.deleteWhere { locks.expiresAt greater now }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue