Skip to content

Commit

Permalink
Replace restart_tasks and delete_tasks methods with the relative Kart…
Browse files Browse the repository at this point in the history
…onBackend methods
  • Loading branch information
Antelox authored Jun 8, 2023
1 parent cc91adb commit 2f6ef45
Showing 1 changed file with 6 additions and 21 deletions.
27 changes: 6 additions & 21 deletions karton/dashboard/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@
request,
send_from_directory,
)
from karton.core import Producer
from karton.core.backend import KartonMetrics
from karton.core.base import KartonBase
from karton.core.inspect import KartonAnalysis, KartonQueue, KartonState
Expand All @@ -43,21 +42,6 @@
)


def restart_tasks(tasks: List[Task]) -> None:
identity = "karton.dashboard-retry"
producer = Producer(identity=identity)

for task in tasks:
# spawn a new task and mark the original one as finished
producer.send_task(task.fork_task())
karton.backend.set_task_status(task=task, status=TaskState.FINISHED)


def cancel_tasks(tasks: List[Task]) -> None:
for task in tasks:
karton.backend.set_task_status(task=task, status=TaskState.FINISHED)


class TaskView:
"""
All problems in computer science can be solved by another
Expand Down Expand Up @@ -263,7 +247,8 @@ def restart_crashed_queue_tasks(queue_name):
if not queue:
return jsonify({"error": "Queue doesn't exist"}), 404

restart_tasks(queue.crashed_tasks)
for task in queue.crashed_tasks:
karton.backend.restart_task(task)
return redirect(request.referrer)


Expand All @@ -274,7 +259,7 @@ def cancel_crashed_queue_tasks(queue_name):
if not queue:
return jsonify({"error": "Queue doesn't exist"}), 404

cancel_tasks(queue.crashed_tasks)
karton.backend.delete_tasks(queue.crashed_tasks)
return redirect(request.referrer)


Expand All @@ -285,7 +270,7 @@ def cancel_pending_queue_tasks(queue_name):
if not queue:
return jsonify({"error": "Queue doesn't exist"}), 404

cancel_tasks(queue.pending_tasks)
karton.backend.delete_tasks(queue.pending_tasks)
return redirect(request.referrer)


Expand All @@ -295,7 +280,7 @@ def restart_task(task_id):
if not task:
return jsonify({"error": "Task doesn't exist"}), 404

restart_tasks([task])
karton.backend.restart_task(task)
return redirect(request.referrer)


Expand All @@ -305,7 +290,7 @@ def cancel_task(task_id):
if not task:
return jsonify({"error": "Task doesn't exist"}), 404

cancel_tasks([task])
karton.backend.delete_tasks([task])
return redirect(request.referrer)


Expand Down

0 comments on commit 2f6ef45

Please sign in to comment.