From c0e86faa78c9038cbed69604ded96f012fd0d6e7 Mon Sep 17 00:00:00 2001
From: Graham Christensen <graham@grahamc.com>
Date: Thu, 26 Aug 2021 17:08:22 -0400
Subject: [PATCH] hydra-notify: call TaskRetries->save_task if a task fails

---
 src/script/hydra-notify | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/script/hydra-notify b/src/script/hydra-notify
index 42b1978e..de8f64f2 100755
--- a/src/script/hydra-notify
+++ b/src/script/hydra-notify
@@ -73,7 +73,15 @@ GetOptions(
 my $db = Hydra::Model::DB->new();
 
 my @plugins = Hydra::Plugin->instantiate(db => $db, config => $config);
-my $task_dispatcher = Hydra::TaskDispatcher->new($db, $prom, [@plugins]);
+my $task_dispatcher = Hydra::TaskDispatcher->new(
+    $db,
+    $prom,
+    [@plugins],
+    sub {
+        my ($task) = @_;
+        $db->resultset("TaskRetries")->save_task($task);
+    }
+);
 
 my $dbh = $db->storage->dbh;