X-Git-Url: http://git.rot13.org/?a=blobdiff_plain;f=web_ui.pl;h=4dbef87911743ef3dfaecfb942810342e0ca0459;hb=58b4d2a9e1f08c7bd3f0b1f7d27e427c74153818;hp=22d445581e5f74c195d33f9fea163aba417374b4;hpb=0b8729d4799a725da5fa4e07b6aa08ec39c34312;p=APKPM.git diff --git a/web_ui.pl b/web_ui.pl index 22d4455..4dbef87 100755 --- a/web_ui.pl +++ b/web_ui.pl @@ -6,6 +6,11 @@ use Redis; use lib '/srv/MojoX-Gearman/lib'; use MojoX::Gearman; +use lib 'lib'; +use APKPM::Model; +use Redis; +use Encode; + use Data::Dump qw(dump); plugin 'tag_helpers'; @@ -29,6 +34,7 @@ get '/ping_http' => sub { my $gearman = MojoX::Gearman->new; #( ioloop => Mojo::IOLoop->singleton ); $gearman->server( $ENV{GEARMAN} || 'localhost:4730' ); +$gearman->timeout( 15 ); get '/ping_g' => sub { my $self = shift; @@ -50,9 +56,34 @@ sub _render_jsonp { get '/g/:call/:args' => [ args => qr/.*/ ] => sub { my $self = shift; - my $ret = $gearman->req( 'SUBMIT_JOB', $self->param('call'), '', $self->param('args') ); - warn $self->param('call'), " = ", dump($ret), "\n"; - die "no result for ", $self->param('call'), ' args: ', $self->param('args') unless defined $ret; + + my $call = $self->param('call'); + my $args = $self->param('args'); + + if ( $call =~ m/^(LDAP|CRM)_search/ ) { + my $k = $1 . '.*' . $args . '*'; + my $redis = Redis->new; + if ( my @k = $redis->keys( $k ) ) { + my $ret = join(',', map { $redis->get($_) } @k ); + return _render_jsonp( $self, "[$ret]" ); + } + } + + my $ret = $gearman->req( 'SUBMIT_JOB', $call, '', $args ); + warn "$call = ", dump($ret), "\n"; + die "no result for $call args: $args" unless defined $ret; + _render_jsonp( $self, $ret ); +}; + +get '/table/:table' => sub { + my $self = shift; + my $sql = "select * from " . $self->param('table'); + if ( my $username = $self->param('username') ) { + $sql .= " where username = '$username' "; + } + $sql .= " limit " . ( $self->param('limit') || 1 ); + + my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql ); _render_jsonp( $self, $ret ); }; @@ -75,6 +106,39 @@ get '/_g/status' => sub { _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) ); }; +sub _gearman_redis { + my ( $job, $params, $timeout ) = @_; + + my $key = $params; + $key =~ s/\W+/_/gs; + $key =~ s/\s+/_/gs; + $key = "sql.$key"; + warn "# _gearman_redis $key [$timeout s]"; + + my $redis = Redis->new; + my $ret; + + if ( $ret = $redis->get($key) ) { + warn "redis hit $key\n"; + } else { + $gearman->timeout($timeout); + $ret = $gearman->req( 'SUBMIT_JOB', $job, '', $params ); + die "no result for $params" unless $ret; + + $redis->set($key => $ret); + } + + $redis->expire($key => 60); # refresh redis key timeout + + $ret = Mojo::JSON->new->decode( $ret ); + + if ( ! exists $ret->{error} && ref $ret->{columns} ne 'ARRAY' ) { + $ret->{error} = 'query run longer than ' . $gearman->timeout . ' s timeout'; + } + + return $ret; +} + get '/gnuplot' => sub { my $self = shift; @@ -93,13 +157,7 @@ get '/gnuplot' => sub { $sql =~ s/\s+(from|where|order|limit|join)/\n$1/gs; $self->param( sql => $sql ); - my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql ); - die "no result for $sql" unless $ret; - - $ret = Mojo::JSON->new->decode( $ret ); - - $ret->{error} = 'query run longer than ' . $gearman->timeout . ' s timeout' - if ! exists $ret->{error} && ref $ret->{columns} ne 'ARRAY'; + my $ret = _gearman_redis( 'Store_sql' => $sql, $self->param('timeout') ); return $self->render('gnuplot', sql => $sql, img => '', gnuplot => $ret->{error} ) if exists $ret->{error}; @@ -179,15 +237,7 @@ set xtics nomirror rotate by -90 get '/_redis' => sub { my $self = shift; - my $redis = Redis->new; - - my $status; - foreach my $k ( $redis->keys('poll.*') ) { - $status->{$k} = eval { $redis->scard($k) } || $redis->get($k); - } - - warn "## ", dump $status; - _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) ); + _render_jsonp( $self, Mojo::JSON->new->encode({ status => APKPM::Model->redis_status }) ); }; get '/user' => sub { @@ -196,5 +246,18 @@ get '/user' => sub { $self->render('user'); }; +get '/psql/:command' => sub { + my $self = shift; + my $command = $self->param('command'); + $command =~ s/^_/\\/ && warn "internal psql command $command"; + $command = sprintf "psql -c '%s' apkpm", $command; + my $output = `$command`; # FIXME unsecure + $self->render_text( "\$ $command\n\n
$output
" ); +}; + +# create pid file +open(my $pid, '>', '/tmp/apkpm.web_ui.pid'); +print $pid "$$\n"; +close $pid; app->start;