correct SQL reformating
[APKPM.git] / 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         return $self->render('gnuplot', img => '', gnuplot => '') unless $sql;
83
84         # re-format SQL
85         $sql =~ s/\s+(from|where|order|limit|join)/\n$1/gs;
86         $self->param( sql => $sql );
87
88         my $ret = $gearman->req( 'SUBMIT_JOB', 'Store_sql', '', $sql );
89         die "no result for $sql" unless $ret;
90
91         $ret = Mojo::JSON->new->decode( $ret );
92
93         return $self->render('gnuplot', sql => $sql, img => '', gnuplot => $ret->{error} )
94                 if exists $ret->{error};
95
96         my $dir = $self->app->home->rel_dir('public');
97
98         my $name = $sql;
99         $name =~ s/\W+//gs;
100         $name .= '.png';
101
102 warn "# $sql -> $name";
103
104         mkdir "$dir/gnuplot" unless -e "$dir/gnuplot";
105         my $png = "$dir/gnuplot/$name"; # FIXME
106
107         my @c = @{ $ret->{columns} };
108         warn "first column not timestamp" unless $c[0] eq 'timestamp';
109
110         my $gnuplot = qq|
111 set terminal png
112 set output '$png'
113
114 set xdata time
115 set timefmt "%Y-%m-%dT%H:%M:%S"
116 set format x "%H:%M\\n%d"
117
118 |;
119
120         my $plot = 'plot ';
121         my $with = $self->param('with') || 'dots';
122         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
123                 $gnuplot .= $plot; $plot = ',';
124                 $gnuplot .= qq| '-' using 1:2 with $with title "$c[$i]" |;
125         }
126         $gnuplot .= "\n";
127
128         foreach my $i ( 1 .. $#{ $ret->{rows}->[0] } ) {
129
130                 foreach my $row ( @{ $ret->{rows} } ) {
131                         my $date = $row->[0];
132                         $date =~ s/ /T/g;
133                         $date =~ s/\.\d+//;
134                         $gnuplot .= join(" ", $date, $row->[$i])."\n";
135                 }
136                 $gnuplot .= "e\n";
137         }
138
139         open(my $fd, '|-', 'gnuplot') || die "gnuplot: $!";
140         print $fd $gnuplot;
141         close($fd);
142
143         $gnuplot = '' unless $self->param('include_gnuplot');
144
145         $self->render('gnuplot', sql => $sql, img => "/gnuplot/$name", gnuplot => $gnuplot);
146 };
147
148 get '/_redis' => sub {
149         my $self = shift;
150
151         my $redis = Redis->new;
152
153         my $status;
154         foreach my $k ( $redis->keys('poll.*') ) {
155                 $status->{$k} = eval { $redis->scard($k) } || $redis->get($k);
156         }
157
158         warn "## ", dump $status;
159         _render_jsonp( $self, Mojo::JSON->new->encode({ status => $status }) );
160 };
161
162 app->start;