/[Sack]/trunk/lib/Sack/Server.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Annotation of /trunk/lib/Sack/Server.pm

Parent Directory Parent Directory | Revision Log Revision Log


Revision 265 - (hide annotations)
Thu Feb 4 18:15:59 2010 UTC (14 years, 3 months ago) by dpavlin
File size: 9725 byte(s)
add pending class to nodes only if there is anything pending

1 dpavlin 163 #!/usr/bin/perl
2    
3 dpavlin 179 package Sack::Server;
4    
5 dpavlin 163 use warnings;
6     use strict;
7    
8 dpavlin 232 my $debug = 0;
9    
10 dpavlin 163 use IO::Socket::INET;
11     use IO::Select;
12    
13     use Data::Dump qw(dump);
14     use Storable qw();
15 dpavlin 169 use File::Slurp;
16 dpavlin 176 use Cwd qw(abs_path);
17 dpavlin 163
18 dpavlin 200 use lib '/srv/Sack/lib';
19     use Sack::Merge;
20 dpavlin 210 use Sack::Server::HTTP;
21 dpavlin 213 use Sack::Server::HTML;
22 dpavlin 241 use Sack::Server::Gnuplot;
23 dpavlin 200
24 dpavlin 179 my @cloud;
25 dpavlin 209 my $cloud_path = $ENV{CLOUD} || "etc/cloud";
26     die "start with: CLOUD=etc/cloud perl -I/srv/Sack/lib $0\n" unless -e $cloud_path;
27 dpavlin 169 @cloud = read_file $cloud_path;
28     @cloud = map { chomp $_; $_ } @cloud;
29    
30     warn "# cloud ",dump( @cloud );
31    
32 dpavlin 163 my $listen_port = 4444;
33 dpavlin 210 my $http_port = 4480;
34 dpavlin 163
35 dpavlin 176 my $node_path = abs_path $0;
36 dpavlin 181 $node_path =~ s{Server}{Client};
37 dpavlin 163
38 dpavlin 210 my $lsn = IO::Socket::INET->new(Listen => 1, LocalPort => $listen_port, Reuse => 1) or die "$listen_port $!";
39     my $www = IO::Socket::INET->new(Listen => 1, LocalPort => $http_port, Reuse => 1) or die "$http_port $!";
40 dpavlin 163 my $sel = IO::Select->new($lsn);
41 dpavlin 210 $sel->add( $www );
42 dpavlin 163
43 dpavlin 166 my $info;
44     sub info {
45     my $port = shift;
46 dpavlin 211 push @{ $info->{node}->{$port} }, [ @_ ];
47 dpavlin 166 }
48    
49 dpavlin 179 sub fork_ssh {
50 dpavlin 163 my ( $port, $host ) = @_;
51    
52     if ( my $pid = fork ) {
53     # parent
54 dpavlin 166 info $port => 'forked', $pid;
55 dpavlin 163 return $port;
56    
57     } elsif ( ! defined $pid ) {
58     warn "can't fork $host $port";
59     return;
60     } else {
61     # child
62 dpavlin 169 my $cmd = qq|ssh -F $cloud_path.ssh -R $port:127.0.0.1:$listen_port $host $node_path $port|;
63 dpavlin 163 warn "# exec: $cmd\n";
64     exec $cmd;
65     }
66     }
67    
68     my $node_port = 4000;
69    
70 dpavlin 169 foreach my $host ( @cloud ) {
71 dpavlin 176 system "find /srv/Sack/ | cpio --create --dereference | ssh -T -F $cloud_path.ssh $host cpio --extract --make-directories --unconditional";
72 dpavlin 179 fork_ssh( $node_port++, $host );
73 dpavlin 163 }
74    
75     my $session;
76    
77 dpavlin 263 sub all_ports { keys %{ $session->{port} } }
78    
79 dpavlin 188 sub to_all {
80     my $data = shift;
81 dpavlin 263 foreach my $port ( all_ports ) {
82 dpavlin 232 warn ">>>> [$port]\n" if $debug;
83 dpavlin 188 Storable::store_fd( $data, $session->{port}->{$port} );
84     }
85     }
86    
87 dpavlin 221 our @shard_load_queue;
88     sub load_shard {
89     my $shard = shift @_ || return;
90 dpavlin 192
91 dpavlin 221 warn "# load_shard $shard\n";
92    
93 dpavlin 260 $shard .= '/*' if -d $shard;
94 dpavlin 221
95 dpavlin 260 my @shards = glob $shard;
96    
97 dpavlin 221 if ( ! @shards ) {
98     warn "no shards for $shard\n";
99     return;
100     }
101    
102 dpavlin 247 foreach my $s ( @shards ) {
103 dpavlin 254 if ( my $node = $info->{shard}->{$s} ) {
104     next if $node =~ m{^\d+$}; # not node number
105     }
106 dpavlin 247 $info->{shard}->{$s} = 'wait';
107     push @shard_load_queue, $s;
108     warn "queued $s for loading\n";
109     }
110 dpavlin 221 to_all { load => $shard };
111 dpavlin 263 $info->{pending}->{$_} = 'load' foreach all_ports;
112 dpavlin 221 }
113    
114 dpavlin 232 sub run_view {
115     my ( $path ) = @_;
116     if ( ! -r $path ) {
117     warn "ERROR view $path: $!";
118     return;
119     }
120     my $code = read_file $path;
121     Sack::Merge->clean;
122     delete( $info->{view} );
123 dpavlin 235 delete( $info->{merge} );
124 dpavlin 248 delete( $info->{shard} );
125 dpavlin 263 $info->{pending}->{$_} = 'view' foreach all_ports;
126 dpavlin 232 to_all { code => $code, view => $path };
127     };
128    
129 dpavlin 247 our @responses;
130    
131 dpavlin 163 while (1) {
132     for my $sock ($sel->can_read(1)) {
133     if ($sock == $lsn) {
134     my $new = $lsn->accept;
135     $sel->add($new);
136 dpavlin 174 $session->{peerport}->{ $new->peerport } = $new;
137 dpavlin 164 warn "[socket] connect\n";
138 dpavlin 163 Storable::store_fd( { ping => 1 }, $new );
139 dpavlin 210 } elsif ( $sock == $www ) {
140     my $client = $www->accept;
141     Sack::Server::HTTP::request( $client, sub {
142     my ( $send, $method, $param ) = @_;
143 dpavlin 212
144     if ( $method =~ m{views} ) {
145 dpavlin 232 run_view $method;
146 dpavlin 212 print $send "HTTP/1.0 302 $method\r\nLocation: /\r\n\r\n";
147 dpavlin 213 return 1;
148 dpavlin 212 } elsif ( $method =~ m{^/tmp/sack} ) {
149 dpavlin 221 load_shard $method;
150 dpavlin 212 print $send "HTTP/1.0 302 $method\r\nLocation: /\r\n\r\n";
151 dpavlin 213 return 1;
152     } elsif ( $method =~ m{^/out/(.+)} ) {
153     print $send "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\n\r\n";
154     Sack::Server::HTML::send_out( $send, Sack::Merge->out, $1, $param );
155     return 1;
156 dpavlin 241 } elsif ( $method =~ m{^/gnuplot} ) {
157     eval {
158 dpavlin 245 my $path = Sack::Server::Gnuplot::date( Sack::Merge->out, $param );
159 dpavlin 241 if ( -e $path ) {
160     print $send "HTTP/1.0 200 OK\r\nContent-Type: image/png\r\n\r\n";
161     open(my $fh, '<', $path) || die $path;
162     my $b;
163     while ( read($fh, $b, 4096) ) {
164     print $send $b;
165     }
166     return 1;
167 dpavlin 245 } else {
168     print $send "HTTP/1.0 404 no graph\r\n\r\n";
169     return 1;
170 dpavlin 241 }
171     };
172     warn "ERROR: $@" if $@;
173 dpavlin 212 }
174    
175 dpavlin 263 my $refresh = $param->{refresh};
176     $refresh = 1 if keys %{ $info->{pending} };
177    
178 dpavlin 262 print $send "HTTP/1.0 200 OK\r\nContent-Type: text/html"
179 dpavlin 263 , ( $refresh ? "\r\nRefresh: $refresh" : '' )
180 dpavlin 262 , "\r\n\r\n"
181 dpavlin 263 , ( $refresh ? qq|<span style="color: #888; float:right">$refresh</span>| : '' )
182 dpavlin 262 ;
183 dpavlin 212
184 dpavlin 232 print $send qq|
185 dpavlin 212
186 dpavlin 232 <style type="text/css">
187 dpavlin 213
188 dpavlin 232 .running {
189     color: #f00;
190     }
191    
192    
193 dpavlin 219 .forked {
194 dpavlin 265 color: #888;
195 dpavlin 263 }
196    
197     .ping {
198 dpavlin 265 color: #0f0;
199 dpavlin 263 }
200    
201     .load {
202 dpavlin 219 color: #f00;
203     }
204 dpavlin 212
205 dpavlin 263 .view {
206     color: #00f;
207     }
208    
209 dpavlin 222 .shards {
210     font-family: monospace;
211 dpavlin 250 // line-height: 0.8em;
212 dpavlin 222 color: #0f0;
213     }
214    
215     .send {
216     color: #ff0;
217     }
218    
219     .wait {
220     color: #f00;
221     }
222    
223    
224 dpavlin 219 </style>
225 dpavlin 232
226     |;
227    
228 dpavlin 264 my $out = Sack::Merge->out;
229     if ( my $li = join("\n", map { qq|<li><input type=checkbox name=filter value="$_"><a href="/out/$_" target="$_">$_</a>| } keys %$out ) ) {
230     print $send qq|<h1>Results</h1><form method="get" target="gnuplot" action="/gnuplot/"><ul>$li</ul><input type=submit value="Show checked"></form><img src="/gnuplot/">\n|;
231     }
232    
233 dpavlin 232 print $send qq|<h1>Views</h1><ul>|
234     , join("\n", map {
235     my $view = $_;
236     my $html = '';
237     if ( my $s = $info->{view}->{$view} ) {
238     my @nodes = sort keys %$s;
239     my $total = 0;
240 dpavlin 237 $html .= qq|<table><tr><th>node</th><th>rows</th><th>shards</th><th>&sum;</th><th>merge</th></tr>|;
241 dpavlin 232 foreach my $node ( @nodes ) {
242     my $h = '';
243     my $shard = 0;
244     foreach my $path ( sort keys %{ $s->{$node} } ) {
245     my $affected = $s->{$node}->{$path};
246 dpavlin 233 my $n = '?';
247     $n = $1 if $path =~ m{/(\w)\w+/\d+};
248 dpavlin 235 my $class = $affected == 0 ? 'style="color: #888"' : '';
249     $h .= qq|<tt title="$affected $path"$class>$n</tt>|;
250 dpavlin 232 $shard += $affected;
251     $total += $affected;
252     }
253 dpavlin 247 my $merge = $info->{merge}->{$view}->{$node} || '?';
254 dpavlin 237 $html .= qq|<tr><td><tt>$node</tt></td><td>$shard</td><td>$h</td><td>$total</td><td>$merge</td></tr>\n|;
255 dpavlin 232 }
256    
257     $html .= qq|</table>|;
258     };
259     qq|<li><a href="$view">$view</a>$html|
260     } glob '/srv/Sack/views/*/*.pl' )
261     , qq|</ul>|
262     ;
263    
264 dpavlin 263 print $send qq|<h1>Nodes</h1>|;
265     foreach my $node ( keys %{ $info->{node} } ) {
266     my $class = join(' '
267     , map { $_->[0] } @{ $info->{node}->{$node} }
268     );
269 dpavlin 265 if ( my $pending = $info->{pending}->{$node} ) {
270     $class .= " $pending";
271     }
272 dpavlin 263 print $send qq|<tt class="$class">$node</tt>\n|;
273     }
274 dpavlin 219
275 dpavlin 249 print $send qq|<h1>Shards</h1><ul>|;
276 dpavlin 221 foreach my $path ( glob '/tmp/sack/*' ) {
277 dpavlin 222 print $send qq|<li><a href="$path">$path</a><div class="shards">|;
278 dpavlin 221 foreach my $s ( sort grep { m/$path/ } keys %{ $info->{shard} } ) {
279 dpavlin 222 my $i = $info->{shard}->{$s};
280 dpavlin 249 print $send qq|<span class="$i" title="$s">$i</span> |;
281 dpavlin 221 }
282 dpavlin 222 print $send qq|</div>|;
283 dpavlin 221 }
284     print $send qq|</ul>|;
285 dpavlin 219
286 dpavlin 233 if ( $param->{info} ) {
287     print $send qq|<a href="?info=0">hide info</a>|;
288     print $send '<pre>', dump($info), '</pre>'
289     } else {
290     print $send qq|<a href="?info=1">show info</a>|;
291     }
292 dpavlin 232 return 1;
293 dpavlin 210 } );
294 dpavlin 163 } else {
295 dpavlin 165 my $data = eval { Storable::fd_retrieve( $sock ) };
296     if ( $@ ) {
297     delete $session->{$sock};
298     warn "[socket] disconnect: $@\n";
299     $sel->remove($sock);
300     $sock->close;
301     } else {
302 dpavlin 232 warn "<<<< ", dump($data), $/ if $debug;
303 dpavlin 196
304     if ( my $path = $data->{shard} ) {
305 dpavlin 211 $info->{shard}->{ $path } = $data->{port};
306     # FIXME will need push for multiple copies of shards
307 dpavlin 196 }
308    
309 dpavlin 192 if ( my $repl = $data->{repl} ) {
310     my $response = { repl_pid => $$ };
311     if ( $repl =~ m/ping/ ) {
312 dpavlin 188 to_all { ping => 1 };
313 dpavlin 192 } elsif ( $repl =~ m/info/ ) {
314 dpavlin 175 $response->{info} = $info;
315 dpavlin 192 } elsif ( $repl =~ m{load\s*(\S+)?} ) {
316 dpavlin 221 load_shard $1;
317 dpavlin 196 } elsif ( $repl =~ m{view\s*(\S+)?} ) {
318 dpavlin 232 run_view $1;
319 dpavlin 200 } elsif ( $repl =~ m{out} ) {
320     my $out = Sack::Merge->out;
321     warn "out ",dump( $out );
322     $response->{out} = $out;
323 dpavlin 203 } elsif ( $repl =~ m{clean} ) {
324     delete $info->{shard};
325     to_all { clean => 1 };
326 dpavlin 232 } elsif ( $repl eq 'exit' ) {
327     to_all { exit => 1 };
328     sleep 1;
329     exit;
330 dpavlin 247 } elsif ( $repl eq '.' ) {
331     $response->{'.'} = [ @responses ];
332     @responses = ();
333     } elsif ( $repl =~ m{(\w+)\s*(.+)?} ) {
334     to_all { $1 => $2 };
335 dpavlin 189 } else {
336 dpavlin 247 $response->{error}->{repl} = $repl;
337 dpavlin 174 }
338 dpavlin 166 Storable::store_fd( $response, $sock );
339     } elsif ( $data->{ping} ) {
340 dpavlin 174 my $port = $data->{port};
341     info $port => 'ping', $port;
342     $session->{port}->{ $data->{port} } = $sock;
343 dpavlin 232 } elsif ( defined $data->{load} && $data->{load} eq 'shard' ) {
344 dpavlin 221 if ( my $path = shift @shard_load_queue ) {
345     $info->{shard}->{$path} = 'read';
346 dpavlin 192 my $shard = Storable::retrieve $path;
347 dpavlin 221 $info->{shard}->{$path} = 'send';
348 dpavlin 192 warn ">>>> [", $data->{port}, "] sending shard $path\n";
349     Storable::store_fd( { path => $path, shard => $shard }, $sock );
350     } else {
351     warn "no more shards for [", $data->{port}, "]\n";
352 dpavlin 263 delete $info->{pending}->{ $data->{port} };
353 dpavlin 192 }
354 dpavlin 235 } elsif ( exists $data->{out} ) {
355     my $added = Sack::Merge->add( $data->{out} ) if defined $data->{out};
356     $info->{merge}->{ $data->{view} }->{ $data->{port} } = $added;
357     $info->{view }->{ $data->{view} }->{ $data->{port} } = $data->{on_shard};
358 dpavlin 248 # refresh shard allocation
359     $info->{shard}->{ $_ } = $data->{port} foreach keys %{ $data->{on_shard} };
360 dpavlin 263 delete $info->{pending}->{ $data->{port} };
361 dpavlin 247 } elsif ( exists $data->{port} ) {
362     push @responses, $data;
363     warn "# ",dump($data),$/;
364 dpavlin 192 } else {
365     warn "UNKNOWN ",dump($data);
366 dpavlin 164 }
367 dpavlin 163 }
368     }
369     }
370     }

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26