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 Data::Dump qw(dump);
  10
  11 plugin 'tag_helpers';
  12 # Documentation browser under "/perldoc" (this plugin requires Perl 5.10)
  13 plugin 'pod_renderer';
  14
  15 app->secret('apkpm');
  16
  17 get '/' => sub {
  18         my $self = shift;
  19         $self->render('index');
  20 };
  21
  22 get '/ping_http' => sub {
  23         my $self = shift;
  24         my $gearman = $self->client;
  25         my $pong = $gearman->post( "http://localhost:4780/ping" => { Connection => 'close' } => "some data" )->res->body;
  26         warn "ping = $pong";
  27         $self->render( 'ping', pong => $pong );
  28 };
  29
  30 my $gearman = MojoX::Gearman->new; #( ioloop => Mojo::IOLoop->singleton );
  31 $gearman->server( $ENV{GEARMAN} || 'localhost:4730' );
  32
  33 get '/ping_g' => sub {
  34         my $self = shift;
  35         my $pong = $gearman->req( 'SUBMIT_JOB', 'ping', '', "some data 2" );
  36         warn "ping = $pong";
  37         $self->render( 'ping', pong => $pong );
  38 };
  39
  40 sub _render_jsonp {
  41         my ( $self, $json ) = @_;
  42         #my $data = $self->render( json => $json, partial => 1 );
  43         my $data = $json;
  44         if ( my $callback = $self->param('callback') ) {
  45                 $data = "$callback($data)";
  46         }
  47         $self->render( data => $data, format => 'js' );
  48 }
  49
  50
  51 get '/g/:call/:args' => [ args => qr/.*/ ] => sub {
  52         my $self = shift;
  53         my $ret = $gearman->req( 'SUBMIT_JOB', $self->param('call'), '', $self->param('args') );
  54         warn $self->param('call'), " = ", dump($ret), "\n";
  55         die "no result for ", $self->param('call'), ' args: ', $self->param('args') unless defined $ret;
  56         _render_jsonp( $self, $ret );
  57 };
  58
  59 get '/_g/status' => sub {
  60         my $self = shift;
  61
  62         my $ret = $gearman->req( 'TEXT' => 'status' );
  63         warn "# status:\n$ret";
  64
  65         my @c = qw(function total running available);
  66
  67         my $status;
  68         foreach my $l ( split(/\n/,$ret) ) {
  69                 my @v = split(/\t/, $l);
  70                 my $h;
  71                 $h->{$c[$_]} = $v[$_] foreach 0 .. $#v;
  72                 push @$status, $h;
  73         }
  74         warn "## ", dump $status;
  75         _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) );
  76 };
  77
  78 get '/gnuplot' => sub {
  79         my $self = shift;
  80
  81         my $sql = $self->param('sql');
  82
  83         if ( my $timeout = $self->param('timeout') ) {
  84                 warn "set timout to $timeout";
  85                 $gearman->timeout( $timeout );
  86         } else {
  87                 $self->param( timeout => $gearman->timeout );
  88         }
  89
  90         return $self->render('gnuplot', img => '', gnuplot => '') unless $sql;
  91
  92         # re-format SQL
  93         $sql =~ s/\s+(from|where|order|limit|join)/\n$1/gs;
  94         $self->param( sql => $sql );
  95
  96         my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql );
  97         die "no result for $sql" unless $ret;
  98
  99         $ret = Mojo::JSON->new->decode( $ret );
 100
 101         $ret->{error} = 'query run longer than ' . $gearman->timeout . ' s timeout'
 102                 if ! exists $ret->{error} && ref $ret->{columns} ne 'ARRAY';
 103
 104         return $self->render('gnuplot', sql => $sql, img => '', gnuplot => $ret->{error} )
 105                 if exists $ret->{error};
 106
 107         my $dir = $self->app->home->rel_dir('public');
 108
 109         my $name = $sql;
 110         $name =~ s/\W+//gs;
 111         $name .= '.png';
 112
 113 warn "# $sql -> $name";
 114
 115         mkdir "$dir/gnuplot" unless -e "$dir/gnuplot";
 116         my $png = "$dir/gnuplot/$name"; # FIXME
 117
 118         my @c = @{ $ret->{columns} };
 119         warn "first column not timestamp" unless $c[0] eq 'timestamp';
 120
 121         my $gnuplot = qq|
 122 set terminal png
 123 set output '$png'
 124
 125 set xdata time
 126 set timefmt "%Y-%m-%dT%H:%M:%S"
 127 set format x "%H:%M\\n%d"
 128
 129 |;
 130
 131         my $plot = 'plot ';
 132         my $with = $self->param('with') || 'dots';
 133         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
 134                 $gnuplot .= $plot; $plot = ',';
 135                 $gnuplot .= qq| '-' using 1:2 with $with title "$c[$i]" |;
 136         }
 137         $gnuplot .= "\n";
 138
 139         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
 140
 141                 foreach my $row ( @{ $ret->{rows} } ) {
 142                         my $date = $row->[0];
 143                         $date =~ s/ /T/g;
 144                         $date =~ s/\.\d+//;
 145                         $gnuplot .= join(" ", $date, $row->[$i])."\n";
 146                 }
 147                 $gnuplot .= "e\n";
 148         }
 149
 150         open(my $fd, '|-', 'gnuplot') || die "gnuplot: $!";
 151         print $fd $gnuplot;
 152         close($fd);
 153
 154         $gnuplot = '' unless $self->param('include_gnuplot');
 155
 156         $self->render('gnuplot', sql => $sql, img => "/gnuplot/$name", gnuplot => $gnuplot);
 157 };
 158
 159 get '/_redis' => sub {
 160         my $self = shift;
 161
 162         my $redis = Redis->new;
 163
 164         my $status;
 165         foreach my $k ( $redis->keys('poll.*') ) {
 166                 $status->{$k} = eval { $redis->scard($k) } || $redis->get($k);
 167         }
 168
 169         warn "## ", dump $status;
 170         _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) );
 171 };
 172
 173 app->start;