3 use base qw(Gearman::Driver::Worker);
5 use Time::HiRes qw(time);
6 use Data::Dump qw(dump);
14 with 'APKPM::Gearman::Client';
15 with 'APKPM::Gearman';
19 sub poll : Job : Encode(e_json) {
20 my ( $self, $job, $workload ) = @_;
22 my $redis = Redis->new;
23 $redis->del( $_ ) foreach $redis->keys('CPE.*');
25 my $start = $self->datetime_now;
27 $redis->set( 'CPE.start' => $start );
29 my $entries = $self->do( 'LDAP_search' => "(&(cn=$workload*)(dhcpStatements=*))" );
31 my $taskset = $self->gc->new_task_set;
34 foreach my $entry ( @$entries ) {
36 my $username = $entry->{cn} || die "no cn in ", dump($entry);
38 if ( $entry->{dhcpStatements} !~ m/fixed-address\s+(\S+)/ ) {
39 $redis->sadd('CPE.error.no-ip' => $username);
44 if ( $entry->{dhcpOption} !~ m/vendor-class-identifier\s\"([^"]+)\"/ ) {
45 $redis->sadd('CPE.error.vendor' => $username);
51 $taskset->add_task('CPE_info', "$ip $username $vendor", {
52 on_complete => sub { push @$results, ${$_[0]} }
54 $redis->sadd('CPE.queued' => $ip);
56 $vendor =~ s/\s+/_/g; # sadd dies on spaces in keys
57 $redis->sadd("CPE.vendor.$vendor" => $ip); # FIXME
63 my $finish = $self->datetime_now;
64 $redis->set( 'CPE.finish' => $finish );
67 foreach my $k ( $redis->keys('CPE.*') ) {
71 $poll->{$n} = eval { $redis->scard($k) } || $redis->get($k) };
74 warn "# poll = ",dump $poll;
75 $self->do_background_json('Store_insert', { _table => 'CPE_poll', %$poll });
77 warn "# results = ", dump $results;
81 sub info : Job : Decode(d_array) : Encode(e_json) {
82 my ( $self, $job, $workload ) = @_;
84 my ( $ip, $username, $vendor ) = @$workload;
86 my $redis = Redis->new;
88 return { error => "invalid workload", expected => "ip username" } unless $ip && $username;
90 my $p = Net::Ping->new;
93 my ( $ok, $rtt, $ping_ip ) = $p->ping( $ip );
96 $redis->sadd( 'CPE.ping.ok' => $ip );
98 $redis->sadd( 'CPE.ping.error' => $ip );
99 return { error => "ping $ip" };
102 $self->do_background_json( 'Store_insert', {
104 username => $username,
105 timestamp => $self->datetime_now,
110 if ( $vendor =~ m/SAMSUNG/ ) {
111 $vendor = 'Davolink';
112 } elsif ( $vendor =~ m/zte/ ) {
113 $vendor = 'EasyGateway';
115 $redis->sadd( 'CPE.skipped' => $username );
120 $redis->sadd( "CPE.$vendor.queued" => $ip );
122 my $module = 'H1::' . $vendor;
123 my $ret = eval { $module->info( $ip ) };
125 warn "# $module ",dump($ret);
128 $self->do_background_json( 'Store_insert', {
131 username => $username,
132 timestamp => $self->datetime_now,
133 h => $self->to_hstore( $ret ),
135 $redis->sadd( "CPE.$vendor.ok" => $ip );
137 $redis->sadd( "CPE.$vendor.error" => $ip );
141 return { ip => $ip, rtt => $rtt };