ahuston-0/hydra
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Abstract over postgres' LISTEN/NOTIFY

Browse Source 
This lets us test the event loop if we wanted, and lets us
test the listening behavior in isolation.
This commit is contained in:
Graham Christensen
2021-06-17 16:27:27 -04:00
parent 9bce425c33
commit 5027003285
3 changed files with 194 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/script/hydra-notify
View File

@ -3,9 +3,9 @@
use strict;
use utf8;
use Hydra::Plugin;
use Hydra::PostgresListener;
use Hydra::Helper::Nix;
use Hydra::Helper::AddBuilds;
use IO::Select;
use Getopt::Long;
STDERR->autoflush(1);
@ -26,9 +26,10 @@ my @plugins = Hydra::Plugin->instantiate(db => $db, config => $config);
my $dbh = $db->storage->dbh;
$dbh->do("listen build_started");
$dbh->do("listen build_finished");
$dbh->do("listen step_finished");
my $listener = Hydra::PostgresListener->new($dbh);
$listener->subscribe("build_started");
$listener->subscribe("build_finished");
$listener->subscribe("step_finished");
sub buildStarted {
my ($buildId) = @_;
@ -112,15 +113,13 @@ for my $build ($db->resultset('Builds')->search(
# Process incoming notifications.
my $fd = $dbh->func("getfd");
my $sel = IO::Select->new($fd);
while (!$queued_only) {
$sel->can_read;
my $messages = $listener->block_for_messages();
while (my $message = $messages->()) {
while (my $notify = $dbh->func("pg_notifies")) {
my ($channelName, $pid, $payload) = @$notify;
my $channelName = $message->{"channel"};
my $pid = $message->{"pid"};
my $payload = $message->{"payload"};
#print STDERR "got '$channelName' from $pid: $payload\n";
my @payload = split /\t/, $payload;