135/netdisco
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

hack to make functional and OO interface

Browse Source
This commit is contained in:
Oliver Gorwits
2014-05-08 22:05:10 +01:00
parent b8c706a2e7
commit 54510a1560
2 changed files with 156 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

148
Netdisco/lib/App/Netdisco/Daemon/JobQueue/PostgreSQL.pm
View File

@@ -1,146 +1,16 @@
package App::Netdisco::Daemon::JobQueue::PostgreSQL;
use Dancer qw/:moose :syntax :script/;
use Dancer::Plugin::DBIC 'schema';
use Net::Domain 'hostfqdn';
use Try::Tiny;
use App::Netdisco::JobQueue::PostgreSQL ();
use Role::Tiny;
use namespace::clean;
#jq_get
#jq_getlocal
#jq_queued
#jq_lock
#jq_defer
#jq_complete
#jq_insert
sub jq_get { shift and App::Netdisco::JobQueue::PostgreSQL::jq_get(@_) }
sub jq_getlocal { shift and App::Netdisco::JobQueue::PostgreSQL::jq_getlocal(@_) }
sub jq_queued { shift and App::Netdisco::JobQueue::PostgreSQL::jq_queued(@_) }
sub jq_lock { shift and App::Netdisco::JobQueue::PostgreSQL::jq_lock(@_) }
sub jq_defer { shift and App::Netdisco::JobQueue::PostgreSQL::jq_defer(@_) }
sub jq_complete { shift and App::Netdisco::JobQueue::PostgreSQL::jq_complete(@_) }
sub jq_insert { shift and App::Netdisco::JobQueue::PostgreSQL::jq_insert(@_) }
sub jq_get {
my ($self, $num_slots) = @_;
my @returned = ();
my $rs = schema('netdisco')->resultset('Admin')
->search(
{status => 'queued'},
{order_by => 'random()', rows => ($num_slots || 1)},
);
while (my $job = $rs->next) {
my $job_type = setting('job_types')->{$job->action} or next;
push @returned, schema('daemon')->resultset('Admin')
->new_result({ $job->get_columns, type => $job_type });
}
return @returned;
}
sub jq_getlocal {
my $self = shift;
my $fqdn = hostfqdn || 'localhost';
my @returned = ();
my $rs = schema('netdisco')->resultset('Admin')
->search({status => "queued-$fqdn"});
while (my $job = $rs->next) {
my $job_type = setting('job_types')->{$job->action} or next;
push @returned, schema('daemon')->resultset('Admin')
->new_result({ $job->get_columns, type => $job_type });
}
return @returned;
}
sub jq_queued {
my ($self, $job_type) = @_;
return schema('netdisco')->resultset('Admin')->search({
device => { '!=' => undef},
action => $job_type,
status => { -like => 'queued%' },
})->get_column('device')->all;
}
sub jq_lock {
my ($self, $job) = @_;
my $fqdn = hostfqdn || 'localhost';
my $happy = 0;
# lock db row and update to show job has been picked
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})
->update({ status => "queued-$fqdn" });
});
$happy = 1;
};
return $happy;
}
sub jq_defer {
my ($self, $job) = @_;
my $happy = 0;
# lock db row and update to show job is available
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})
->update({ status => 'queued' });
});
$happy = 1;
};
return $happy;
}
sub jq_complete {
my ($self, $job) = @_;
my $happy = 0;
# lock db row and update to show job is done/error
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})->update({
status => $job->status,
log => $job->log,
finished => $job->finished,
});
});
$happy = 1;
};
return $happy;
}
sub jq_insert {
my ($self, $jobs) = @_;
$jobs = [$jobs] if ref [] ne ref $jobs;
my $happy = 0;
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')->populate([
map {{
device => $_->{device},
port => $_->{port},
action => $_->{action},
subaction => ($_->{extra} || $_->{subaction}),
username => $_->{username},
userip => $_->{userip},
status => 'queued',
}} @$jobs
]);
});
$happy = 1;
};
return $happy;
}
true;
1;

147
Netdisco/lib/App/Netdisco/JobQueue/PostgreSQL.pm Normal file
View File

@@ -0,0 +1,147 @@
package App::Netdisco::JobQueue::PostgreSQL;
use Dancer qw/:moose :syntax :script/;
use Dancer::Plugin::DBIC 'schema';
use Net::Domain 'hostfqdn';
use Try::Tiny;
use base 'Exporter';
our @EXPORT = ();
our @EXPORT_OK = qw/
jq_get
jq_getlocal
jq_queued
jq_lock
jq_defer
jq_complete
jq_insert
/;
our %EXPORT_TAGS = ( all => \@EXPORT_OK );
sub jq_get {
my $num_slots = shift;
my @returned = ();
my $rs = schema('netdisco')->resultset('Admin')
->search(
{status => 'queued'},
{order_by => 'random()', rows => ($num_slots || 1)},
);
while (my $job = $rs->next) {
my $job_type = setting('job_types')->{$job->action} or next;
push @returned, schema('daemon')->resultset('Admin')
->new_result({ $job->get_columns, type => $job_type });
}
return @returned;
}
sub jq_getlocal {
my $fqdn = hostfqdn || 'localhost';
my @returned = ();
my $rs = schema('netdisco')->resultset('Admin')
->search({status => "queued-$fqdn"});
while (my $job = $rs->next) {
my $job_type = setting('job_types')->{$job->action} or next;
push @returned, schema('daemon')->resultset('Admin')
->new_result({ $job->get_columns, type => $job_type });
}
return @returned;
}
sub jq_queued {
my $job_type = shift;
return schema('netdisco')->resultset('Admin')->search({
device => { '!=' => undef},
action => $job_type,
status => { -like => 'queued%' },
})->get_column('device')->all;
}
sub jq_lock {
my $job = shift;
my $fqdn = hostfqdn || 'localhost';
my $happy = 0;
# lock db row and update to show job has been picked
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})
->update({ status => "queued-$fqdn" });
});
$happy = 1;
};
return $happy;
}
sub jq_defer {
my $job = shift;
my $happy = 0;
# lock db row and update to show job is available
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})
->update({ status => 'queued' });
});
$happy = 1;
};
return $happy;
}
sub jq_complete {
my $job = shift;
my $happy = 0;
# lock db row and update to show job is done/error
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')
->find($job->id, {for => 'update'})->update({
status => $job->status,
log => $job->log,
finished => $job->finished,
});
});
$happy = 1;
};
return $happy;
}
sub jq_insert {
my $jobs = shift;
$jobs = [$jobs] if ref [] ne ref $jobs;
my $happy = 0;
try {
schema('netdisco')->txn_do(sub {
schema('netdisco')->resultset('Admin')->populate([
map {{
device => $_->{device},
port => $_->{port},
action => $_->{action},
subaction => ($_->{extra} || $_->{subaction}),
username => $_->{username},
userip => $_->{userip},
status => 'queued',
}} @$jobs
]);
});
$happy = 1;
};
return $happy;
}
true;