Small changes
This commit is contained in:
parent
aab76f32b3
commit
6eb72f8486
@ -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)
|
||||
|
||||
@ -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) {}
|
||||
|
||||
@ -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) {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user