web_ui.pl

   1 #!/usr/bin/env perl
   2
   3 use Mojolicious::Lite;
   4 use Redis;
   5
   6 use lib '/srv/MojoX-Gearman/lib';
   7 use MojoX::Gearman;
   8
   9 use lib 'lib';
  10 use APKPM::Model;
  11
  12 use Data::Dump qw(dump);
  13
  14 plugin 'tag_helpers';
  15 # Documentation browser under "/perldoc" (this plugin requires Perl 5.10)
  16 plugin 'pod_renderer';
  17
  18 app->secret('apkpm');
  19
  20 get '/' => sub {
  21         my $self = shift;
  22         $self->render('index');
  23 };
  24
  25 get '/ping_http' => sub {
  26         my $self = shift;
  27         my $gearman = $self->client;
  28         my $pong = $gearman->post( "http://localhost:4780/ping" => { Connection => 'close' } => "some data" )->res->body;
  29         warn "ping = $pong";
  30         $self->render( 'ping', pong => $pong );
  31 };
  32
  33 my $gearman = MojoX::Gearman->new; #( ioloop => Mojo::IOLoop->singleton );
  34 $gearman->server( $ENV{GEARMAN} || 'localhost:4730' );
  35 $gearman->timeout( 15 );
  36
  37 get '/ping_g' => sub {
  38         my $self = shift;
  39         my $pong = $gearman->req( 'SUBMIT_JOB', 'ping', '', "some data 2" );
  40         warn "ping = $pong";
  41         $self->render( 'ping', pong => $pong );
  42 };
  43
  44 sub _render_jsonp {
  45         my ( $self, $json ) = @_;
  46         #my $data = $self->render( json => $json, partial => 1 );
  47         my $data = $json;
  48         if ( my $callback = $self->param('callback') ) {
  49                 $data = "$callback($data)";
  50         }
  51         $self->render( data => $data, format => 'js' );
  52 }
  53
  54
  55 get '/g/:call/:args' => [ args => qr/.*/ ] => sub {
  56         my $self = shift;
  57         my $ret = $gearman->req( 'SUBMIT_JOB', $self->param('call'), '', $self->param('args') );
  58         warn $self->param('call'), " = ", dump($ret), "\n";
  59         die "no result for ", $self->param('call'), ' args: ', $self->param('args') unless defined $ret;
  60         _render_jsonp( $self, $ret );
  61 };
  62
  63 get '/table/:table' => sub {
  64         my $self = shift;
  65         my $sql = "select * from " . $self->param('table') . " limit " . ( $self->param('limit') || 1 );
  66         my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql );
  67         _render_jsonp( $self, $ret );
  68 };
  69
  70 get '/_g/status' => sub {
  71         my $self = shift;
  72
  73         my $ret = $gearman->req( 'TEXT' => 'status' );
  74         warn "# status:\n$ret";
  75
  76         my @c = qw(function total running available);
  77
  78         my $status;
  79         foreach my $l ( split(/\n/,$ret) ) {
  80                 my @v = split(/\t/, $l);
  81                 my $h;
  82                 $h->{$c[$_]} = $v[$_] foreach 0 .. $#v;
  83                 push @$status, $h;
  84         }
  85         warn "## ", dump $status;
  86         _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) );
  87 };
  88
  89 sub _gearman_redis {
  90         my ( $job, $params, $timeout ) = @_;
  91
  92         my $key = $params;
  93         $key =~ s/\W+/_/gs;
  94         $key =~ s/\s+/_/gs;
  95         $key = "sql.$key";
  96         warn "# _gearman_redis $key [$timeout s]";
  97
  98         my $redis = Redis->new;
  99         my $ret;
 100
 101         if ( $ret = $redis->get($key) ) {
 102                 warn "redis hit $key\n";
 103         } else {
 104                 $gearman->timeout($timeout);
 105                 $ret = $gearman->req( 'SUBMIT_JOB', $job, '', $params );
 106                 die "no result for $params" unless $ret;
 107
 108                 $redis->set($key => $ret);
 109         }
 110
 111         $redis->expire($key => 60); # refresh redis key timeout
 112
 113         $ret = Mojo::JSON->new->decode( $ret );
 114
 115         if ( ! exists $ret->{error} && ref $ret->{columns} ne 'ARRAY' ) {
 116                 $ret->{error} = 'query run longer than ' . $gearman->timeout . ' s timeout';
 117         }
 118
 119         return $ret;
 120 }
 121
 122 get '/gnuplot' => sub {
 123         my $self = shift;
 124
 125         my $sql = $self->param('sql');
 126
 127         if ( my $timeout = $self->param('timeout') ) {
 128                 warn "set timout to $timeout";
 129                 $gearman->timeout( $timeout );
 130         } else {
 131                 $self->param( timeout => $gearman->timeout );
 132         }
 133
 134         return $self->render('gnuplot', img => '', gnuplot => '') unless $sql;
 135
 136         # re-format SQL
 137         $sql =~ s/\s+(from|where|order|limit|join)/\n$1/gs;
 138         $self->param( sql => $sql );
 139
 140         my $ret = _gearman_redis( 'Store_sql' => $sql, $self->param('timeout') );
 141
 142         return $self->render('gnuplot', sql => $sql, img => '', gnuplot => $ret->{error} )
 143                 if exists $ret->{error};
 144
 145         my $dir = $self->app->home->rel_dir('public');
 146
 147         my $name = $sql;
 148         $name =~ s/\W+//gs;
 149         $name .= '.png';
 150
 151 warn "# $sql -> $name";
 152
 153         mkdir "$dir/gnuplot" unless -e "$dir/gnuplot";
 154         my $png = "$dir/gnuplot/$name"; # FIXME
 155
 156         my @c = @{ $ret->{columns} };
 157
 158         my $gnuplot = qq|
 159 set terminal png
 160 set output '$png'
 161 |;
 162
 163         my $del; # delimiter between date and time
 164
 165         if ( $ret->{rows}->[0]->[0] =~ m/(\d{4}-\d{2}-\d{2})(.)?(\d{2}:\d{2}:\d{2})?/ ) {
 166
 167                 my ( $date,$time );
 168                 ( $date, $del, $time ) = ( $1,$2,$3 );
 169
 170                 my $fmt = '%Y-%m-%d';
 171                 $fmt   .= 'T' if $del;
 172                 $fmt   .= '%H:%M:%S' if $time;
 173
 174                 my $format_x = $time ? '%d %H:%M' : '%Y-%m-%d';
 175                 $gnuplot .= qq|
 176
 177 set xdata time
 178 set timefmt "$fmt"
 179
 180 set format x "$format_x"
 181 set xtics nomirror rotate by -90
 182
 183 |;
 184
 185         } else {
 186                 warn "first column not timestamp";
 187         }
 188
 189         my $plot = 'plot ';
 190         my $with = $self->param('with') || 'dots';
 191         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
 192                 $gnuplot .= $plot; $plot = ',';
 193                 $gnuplot .= qq| '-' using 1:2 with $with title "$c[$i]" |;
 194         }
 195         $gnuplot .= "\n";
 196
 197         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
 198
 199                 foreach my $row ( @{ $ret->{rows} } ) {
 200                         my $date = $row->[0];
 201                         $date =~ s/\Q$del\E/T/g if $del;
 202                         $date =~ s/\.\d+//;
 203                         $gnuplot .= join(" ", $date, $row->[$i])."\n";
 204                 }
 205                 $gnuplot .= "e\n";
 206         }
 207
 208         open(my $fd, '|-', 'gnuplot') || die "gnuplot: $!";
 209         print $fd $gnuplot;
 210         close($fd);
 211
 212         $gnuplot = '' unless $self->param('include_gnuplot');
 213
 214         $self->render('gnuplot', sql => $sql, img => "/gnuplot/$name", gnuplot => $gnuplot);
 215 };
 216
 217 get '/_redis' => sub {
 218         my $self = shift;
 219
 220         _render_jsonp( $self, Mojo::JSON->new->encode({ status => APKPM::Model->redis_status }) );
 221 };
 222
 223 get '/user' => sub {
 224         my $self = shift;
 225
 226         $self->render('user');
 227 };
 228
 229 get '/psql/:command' => sub {
 230         my $self = shift;
 231         my $command = $self->param('command');
 232         $command =~ s/^_/\\/ && warn "internal psql command $command";
 233         $command = sprintf "psql -c '%s' apkpm", $command;
 234         my $output = `$command`; # FIXME unsecure
 235         $self->render_text( "\$ <b><tt>$command</tt></b>\n\n<pre>$output</pre>" );
 236 };
 237
 238 # create pid file
 239 open(my $pid, '>', '/tmp/apkpm.web_ui.pid');
 240 print $pid "$$\n";
 241 close $pid;
 242
 243 app->start;