X-Git-Url: http://git.rot13.org/?p=angular-mojolicious.git;a=blobdiff_plain;f=couchdb-trigger.pl;h=f23e505e453c4c2879bbbeb37dbf912c9b245a3a;hp=d39f0079e6e26b4aa2736a69788220b7116feedd;hb=a9035c2e30dec2e21ec6558b7ff5fe24efec5aeb;hpb=d44d1dd771e6c733ee519b6ee7fae239e5b17566 diff --git a/couchdb-trigger.pl b/couchdb-trigger.pl index d39f007..f23e505 100755 --- a/couchdb-trigger.pl +++ b/couchdb-trigger.pl @@ -15,28 +15,27 @@ use strict; use lib 'common/mojo/lib'; -use Mojo::Client; +use Mojo::UserAgent; use Mojo::JSON; use Time::HiRes qw(time); +use Data::Dump qw(dump); -my $url = 'http://localhost:5984/monitor'; +my ( $url, $trigger_path ) = @ARGV; -sub _trigger { - my $trigger = $_[0]->{trigger}; - if ( my $command = $trigger->{command} ) { - # FIXME SECURITY HOLE - my $output = $trigger->{output} = `$command`; +$url ||= 'http://localhost:5984/monitor'; +$trigger_path ||= 'trigger/shell.pm' ; - $trigger->{output} = - [ map { [ split (/\s+/,$_) ] } split(/\n/,$output) ] - if $trigger->{format} =~ m/table/i; - } - return $trigger; -} +our $database = $1 if $url =~ m{/(\w+)/?$}; + +require $trigger_path if -e $trigger_path; + +my $uid = `hostname -s`; +chomp $uid; +$uid .= ':' . $$; my $seq = 0; -my $client = Mojo::Client->new; +my $client = Mojo::UserAgent->new; our $json = Mojo::JSON->new; sub info { warn $_[0], " ",$json->encode($_[1]),$/ } sub debug { info "# $_[0]", $_[1] } @@ -52,12 +51,9 @@ while( ! $error ) { $tx->res->body(sub{ my ( $content, $body ) = @_; - debug 'BODY' => $body; + return if length($body) == 0; # empty chunk, heartbeat? - if ( length($body) == 0 ) { - warn "# empty chunk, heartbeat?\n"; - return; - } + debug 'BODY' => $body; foreach ( split(/\r?\n/, $body) ) { # we can get multiple documents in one chunk @@ -77,42 +73,46 @@ while( ! $error ) { debug 'change' => $change; - if ( my $trigger = $change->{doc}->{trigger} ) { - if ( exists $trigger->{active} ) { - debug 'trigger.active', $change->{doc}->{trigger}->{active}; - } else { - $trigger->{active} = [ time() ]; + if ( filter($change) ) { + + if ( ! exists $change->{doc}->{trigger}->{active} ) { + $change->{doc}->{trigger}->{active} = [ $uid, time() ]; debug 'TRIGGER start PUT ', $change->{doc}; - $client->put( "$url/$id" => $json->encode( $change->{doc} ) => sub { - my ($client,$tx) = @_; - if ($tx->error) { - if ( $tx->res->code == 409 ) { - info "TRIGGER ABORTED started on another worker? ", $tx->error; + my $client = Mojo::UserAgent->new; + my $res = $client->put( "$url/$id" => $json->encode( $change->{doc} ) )->res; +warn "code ", $res->code, dump( $res->json ); + if ( $res->code == 409 ) { + info "TRIGGER ABORTED started on another worker? ", $res->error; + next; + } elsif ( $res->code != 201 ) { + info "ERROR $url/$id ", $res->code; + } + + + } elsif ( $change->{doc}->{trigger}->{active}->[0] eq $uid ) { + if ( exists $change->{doc}->{trigger}->{active}->[2] ) { + warn "allready executed"; + next; + } else { + + debug "TRIGGER execute ", $change->{doc}; + trigger( $change ); + + push @{ $change->{doc}->{trigger}->{active} }, time(), 0; # last timestamp +warn "change ",dump $change; + + my $client = Mojo::UserAgent->new; + my $res = $client->put( "$url/$id" => $json->encode( $change->{doc} ) )->res; +warn "code ", $res->code; + if ( my $json = $res->json ) { +warn dump($json); + $change->{doc}->{_rev} = $json->{rev}; + info "TRIGGER finish ", $change->{doc}; } else { - info "ERROR ", $tx->error; - } - } else { - my $res = $tx->res->json; - $change->{doc}->{_rev} = $res->{rev}; - - debug "TRIGGER execute ", $change->{doc}; - _trigger( $change->{doc} ); - - push @{ $trigger->{active} }, time(), 0; # last timestamp - - $client->put( "$url/$id" => $json->encode( $change->{doc} ) => sub { - my ($client,$tx) = @_; - if ($tx->error) { - info "ERROR", $tx->error; - } else { - my $res = $tx->res->json; - $change->{doc}->{_rev} = $res->{rev}; - info "TRIGGER finish ", $change->{doc}; - } - })->process; + info "ERROR $url/$id", $tx->error; } - })->process; + } } } } else {