Small changes

This commit is contained in:
Brage Skjønborg 2026-01-31 14:10:29 +01:00
parent aab76f32b3
commit 6eb72f8486
3 changed files with 8 additions and 8 deletions

View File

@ -95,7 +95,7 @@ abstract class TaskListener(val taskType: TaskType = TaskType.CPU_INTENSIVE): Ta
override fun onError(task: Task, exception: Exception) {
reporter?.log(task.taskId, "Error processing task: ${exception.message}")
exception.printStackTrace()
reporter?.markFailed(task.taskId)
reporter?.markFailed(task.referenceId, task.taskId)
}
override fun onComplete(task: Task, result: Event?) {
@ -107,7 +107,7 @@ abstract class TaskListener(val taskType: TaskType = TaskType.CPU_INTENSIVE): Ta
}
override fun onCancelled(task: Task) {
reporter!!.markCancelled(task.taskId)
reporter!!.markCancelled(task.referenceId, task.taskId)
currentJob?.cancel()
heartbeatRunner?.cancel()
currentTask = null
@ -134,8 +134,8 @@ interface TaskReporter {
fun markClaimed(taskId: UUID, workerId: String)
fun updateLastSeen(taskId: UUID)
fun markCompleted(taskId: UUID)
fun markFailed(taskId: UUID)
fun markCancelled(taskId: UUID)
fun markFailed(referenceId: UUID, taskId: UUID)
fun markCancelled(referenceId: UUID, taskId: UUID)
fun updateProgress(taskId: UUID, progress: Int)
fun log(taskId: UUID, message: String)
fun publishEvent(event: Event)

View File

@ -37,8 +37,8 @@ class TaskListenerTest {
override fun markClaimed(taskId: UUID, workerId: String) { claimed = true }
override fun markCompleted(taskId: UUID) { consumed = true }
override fun markFailed(taskId: UUID) { consumed = true }
override fun markCancelled(taskId: UUID) {}
override fun markFailed(referenceId: UUID, taskId: UUID) { consumed = true }
override fun markCancelled(referenceId: UUID, taskId: UUID) {}
override fun updateProgress(taskId: UUID, progress: Int) {}
override fun publishEvent(event: Event) { events.add(event) }
override fun updateLastSeen(taskId: UUID) {}

View File

@ -45,8 +45,8 @@ class TaskPollerImplementationTest : TestBase() {
override fun markClaimed(taskId: UUID, workerId: String) {}
override fun updateLastSeen(taskId: UUID) {}
override fun markCompleted(taskId: UUID) {}
override fun markFailed(taskId: UUID) {}
override fun markCancelled(taskId: UUID) {}
override fun markFailed(referenceId: UUID,taskId: UUID) {}
override fun markCancelled(referenceId: UUID,taskId: UUID) {}
override fun updateProgress(taskId: UUID, progress: Int) {}
override fun log(taskId: UUID, message: String) {}
override fun publishEvent(event: Event) {