ResultSet::TaskRetries: add get_seconds_to_next_retry

Get the number of seconds before the next retriable task is ready.

This number is specifically intended to be used as a timeout, where
`undef` means never time out.
This commit is contained in:
Graham Christensen 2021-08-26 12:38:33 -04:00
parent c4134c8e84
commit 29738364fb
2 changed files with 86 additions and 0 deletions

View file

@ -0,0 +1,41 @@
package Hydra::Schema::ResultSet::TaskRetries;
use strict;
use warnings;
use utf8;
use base 'DBIx::Class::ResultSet';
use List::Util qw(max);
=head2 get_seconds_to_next_retry
Query the database to identify how soon the next retryable task is due
for being attempted again.
If there are no tasks to be reattempted it returns undef.
If a task's scheduled retry has passed, it returns 0.
Otherwise, returns the number of seconds from now to look for work.
=cut
sub get_seconds_to_next_retry {
my ($self) = @_;
my $next_retry = $self->search(
{}, # any task
{
order_by => {
-asc => 'retry_at'
},
rows => 1,
}
)->get_column('retry_at')->first;
if (defined($next_retry)) {
return max(0, $next_retry - time());
} else {
return undef;
}
}
1;

View file

@ -0,0 +1,45 @@
use strict;
use warnings;
use Setup;
my %ctx = test_init();
require Hydra::Schema;
require Hydra::Model::DB;
use Test2::V0;
my $db = Hydra::Model::DB->new;
hydra_setup($db);
my $taskretries = $db->resultset('TaskRetries');
subtest "get_seconds_to_next_retry" => sub {
subtest "Without any records in the database" => sub {
is($taskretries->get_seconds_to_next_retry(), undef, "Without any records our next retry moment is forever away.");
};
subtest "With only tasks whose retry timestamps are in the future" => sub {
$taskretries->create({
channel => "bogus",
pluginname => "bogus",
payload => "bogus",
attempts => 1,
retry_at => time() + 100,
});
is($taskretries->get_seconds_to_next_retry(), within(100, 2), "We should retry in roughly 100 seconds");
};
subtest "With tasks whose retry timestamp are in the past" => sub {
$taskretries->create({
channel => "bogus",
pluginname => "bogus",
payload => "bogus",
attempts => 1,
retry_at => time() - 100,
});
is($taskretries->get_seconds_to_next_retry(), 0, "We should retry immediately");
}
};
done_testing;