return simple table dump sa json

[APKPM.git] / web_ui.pl

diff --git a/web_ui.pl b/web_ui.pl
index 7eb6563..7fef2fc 100755 (executable)
--- a/web_ui.pl
+++ b/web_ui.pl
@@ -60,6 +60,13 @@ get '/g/:call/:args' => [ args => qr/.*/ ] => sub {
        _render_jsonp( $self, $ret );
 };
 
+get '/table/:table' => sub {
+       my $self = shift;
+       my $sql = "select * from " . $self->param('table') . " limit " . ( $self->param('limit') || 1 );
+       my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql );
+       _render_jsonp( $self, $ret );
+};
+
 get '/_g/status' => sub {
        my $self = shift;
 
@@ -79,6 +86,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;
 
@@ -97,14 +137,7 @@ get '/gnuplot' => sub {
        $sql =~ s/\s+(from|where|order|limit|join)/\n$1/gs;
        $self->param( sql => $sql );
 
-       $gearman->timeout(15);
-       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};
@@ -193,5 +226,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( "\$ <b><tt>$command</tt></b>\n\n<pre>$output</pre>" );
+};
+
+# create pid file
+open(my $pid, '>', '/tmp/apkpm.web_ui.pid');
+print $pid "$$\n";
+close $pid;
 
 app->start;