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 |
263 |
// text-decoration: line-through; |
195 |
|
|
} |
196 |
|
|
|
197 |
|
|
.ping { |
198 |
|
|
color: #ff0; |
199 |
|
|
} |
200 |
|
|
|
201 |
|
|
.load { |
202 |
dpavlin |
219 |
color: #f00; |
203 |
|
|
} |
204 |
dpavlin |
212 |
|
205 |
dpavlin |
263 |
.view { |
206 |
|
|
color: #00f; |
207 |
|
|
} |
208 |
|
|
|
209 |
|
|
.ready { |
210 |
dpavlin |
219 |
color: #0f0; |
211 |
|
|
} |
212 |
dpavlin |
222 |
|
213 |
|
|
.shards { |
214 |
|
|
font-family: monospace; |
215 |
dpavlin |
250 |
// line-height: 0.8em; |
216 |
dpavlin |
222 |
color: #0f0; |
217 |
|
|
} |
218 |
|
|
|
219 |
|
|
.send { |
220 |
|
|
color: #ff0; |
221 |
|
|
} |
222 |
|
|
|
223 |
|
|
.wait { |
224 |
|
|
color: #f00; |
225 |
|
|
} |
226 |
|
|
|
227 |
|
|
|
228 |
dpavlin |
219 |
</style> |
229 |
dpavlin |
232 |
|
230 |
|
|
|; |
231 |
|
|
|
232 |
|
|
print $send qq|<h1>Views</h1><ul>| |
233 |
|
|
, join("\n", map { |
234 |
|
|
my $view = $_; |
235 |
|
|
my $html = ''; |
236 |
|
|
if ( my $s = $info->{view}->{$view} ) { |
237 |
|
|
my @nodes = sort keys %$s; |
238 |
|
|
my $total = 0; |
239 |
dpavlin |
237 |
$html .= qq|<table><tr><th>node</th><th>rows</th><th>shards</th><th>∑</th><th>merge</th></tr>|; |
240 |
dpavlin |
232 |
foreach my $node ( @nodes ) { |
241 |
|
|
my $h = ''; |
242 |
|
|
my $shard = 0; |
243 |
|
|
foreach my $path ( sort keys %{ $s->{$node} } ) { |
244 |
|
|
my $affected = $s->{$node}->{$path}; |
245 |
dpavlin |
233 |
my $n = '?'; |
246 |
|
|
$n = $1 if $path =~ m{/(\w)\w+/\d+}; |
247 |
dpavlin |
235 |
my $class = $affected == 0 ? 'style="color: #888"' : ''; |
248 |
|
|
$h .= qq|<tt title="$affected $path"$class>$n</tt>|; |
249 |
dpavlin |
232 |
$shard += $affected; |
250 |
|
|
$total += $affected; |
251 |
|
|
} |
252 |
dpavlin |
247 |
my $merge = $info->{merge}->{$view}->{$node} || '?'; |
253 |
dpavlin |
237 |
$html .= qq|<tr><td><tt>$node</tt></td><td>$shard</td><td>$h</td><td>$total</td><td>$merge</td></tr>\n|; |
254 |
dpavlin |
232 |
} |
255 |
|
|
|
256 |
|
|
$html .= qq|</table>|; |
257 |
|
|
}; |
258 |
|
|
qq|<li><a href="$view">$view</a>$html| |
259 |
|
|
} glob '/srv/Sack/views/*/*.pl' ) |
260 |
|
|
, qq|</ul>| |
261 |
|
|
; |
262 |
|
|
|
263 |
|
|
my $out = Sack::Merge->out; |
264 |
dpavlin |
253 |
if ( my $li = join("\n", map { qq|<li><input type=checkbox name=filter value="$_"><a href="/out/$_" target="$_">$_</a>| } keys %$out ) ) { |
265 |
|
|
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|; |
266 |
dpavlin |
241 |
} |
267 |
dpavlin |
232 |
|
268 |
dpavlin |
263 |
print $send qq|<h1>Nodes</h1>|; |
269 |
|
|
foreach my $node ( keys %{ $info->{node} } ) { |
270 |
|
|
my $class = join(' ' |
271 |
|
|
, map { $_->[0] } @{ $info->{node}->{$node} } |
272 |
|
|
); |
273 |
|
|
$class .= ' ' . ( $info->{pending}->{$node} || 'ready' ); |
274 |
|
|
print $send qq|<tt class="$class">$node</tt>\n|; |
275 |
|
|
} |
276 |
dpavlin |
219 |
|
277 |
dpavlin |
249 |
print $send qq|<h1>Shards</h1><ul>|; |
278 |
dpavlin |
221 |
foreach my $path ( glob '/tmp/sack/*' ) { |
279 |
dpavlin |
222 |
print $send qq|<li><a href="$path">$path</a><div class="shards">|; |
280 |
dpavlin |
221 |
foreach my $s ( sort grep { m/$path/ } keys %{ $info->{shard} } ) { |
281 |
dpavlin |
222 |
my $i = $info->{shard}->{$s}; |
282 |
dpavlin |
249 |
print $send qq|<span class="$i" title="$s">$i</span> |; |
283 |
dpavlin |
221 |
} |
284 |
dpavlin |
222 |
print $send qq|</div>|; |
285 |
dpavlin |
221 |
} |
286 |
|
|
print $send qq|</ul>|; |
287 |
dpavlin |
219 |
|
288 |
dpavlin |
233 |
if ( $param->{info} ) { |
289 |
|
|
print $send qq|<a href="?info=0">hide info</a>|; |
290 |
|
|
print $send '<pre>', dump($info), '</pre>' |
291 |
|
|
} else { |
292 |
|
|
print $send qq|<a href="?info=1">show info</a>|; |
293 |
|
|
} |
294 |
dpavlin |
232 |
return 1; |
295 |
dpavlin |
210 |
} ); |
296 |
dpavlin |
163 |
} else { |
297 |
dpavlin |
165 |
my $data = eval { Storable::fd_retrieve( $sock ) }; |
298 |
|
|
if ( $@ ) { |
299 |
|
|
delete $session->{$sock}; |
300 |
|
|
warn "[socket] disconnect: $@\n"; |
301 |
|
|
$sel->remove($sock); |
302 |
|
|
$sock->close; |
303 |
|
|
} else { |
304 |
dpavlin |
232 |
warn "<<<< ", dump($data), $/ if $debug; |
305 |
dpavlin |
196 |
|
306 |
|
|
if ( my $path = $data->{shard} ) { |
307 |
dpavlin |
211 |
$info->{shard}->{ $path } = $data->{port}; |
308 |
|
|
# FIXME will need push for multiple copies of shards |
309 |
dpavlin |
196 |
} |
310 |
|
|
|
311 |
dpavlin |
192 |
if ( my $repl = $data->{repl} ) { |
312 |
|
|
my $response = { repl_pid => $$ }; |
313 |
|
|
if ( $repl =~ m/ping/ ) { |
314 |
dpavlin |
188 |
to_all { ping => 1 }; |
315 |
dpavlin |
192 |
} elsif ( $repl =~ m/info/ ) { |
316 |
dpavlin |
175 |
$response->{info} = $info; |
317 |
dpavlin |
192 |
} elsif ( $repl =~ m{load\s*(\S+)?} ) { |
318 |
dpavlin |
221 |
load_shard $1; |
319 |
dpavlin |
196 |
} elsif ( $repl =~ m{view\s*(\S+)?} ) { |
320 |
dpavlin |
232 |
run_view $1; |
321 |
dpavlin |
200 |
} elsif ( $repl =~ m{out} ) { |
322 |
|
|
my $out = Sack::Merge->out; |
323 |
|
|
warn "out ",dump( $out ); |
324 |
|
|
$response->{out} = $out; |
325 |
dpavlin |
203 |
} elsif ( $repl =~ m{clean} ) { |
326 |
|
|
delete $info->{shard}; |
327 |
|
|
to_all { clean => 1 }; |
328 |
dpavlin |
232 |
} elsif ( $repl eq 'exit' ) { |
329 |
|
|
to_all { exit => 1 }; |
330 |
|
|
sleep 1; |
331 |
|
|
exit; |
332 |
dpavlin |
247 |
} elsif ( $repl eq '.' ) { |
333 |
|
|
$response->{'.'} = [ @responses ]; |
334 |
|
|
@responses = (); |
335 |
|
|
} elsif ( $repl =~ m{(\w+)\s*(.+)?} ) { |
336 |
|
|
to_all { $1 => $2 }; |
337 |
dpavlin |
189 |
} else { |
338 |
dpavlin |
247 |
$response->{error}->{repl} = $repl; |
339 |
dpavlin |
174 |
} |
340 |
dpavlin |
166 |
Storable::store_fd( $response, $sock ); |
341 |
|
|
} elsif ( $data->{ping} ) { |
342 |
dpavlin |
174 |
my $port = $data->{port}; |
343 |
|
|
info $port => 'ping', $port; |
344 |
|
|
$session->{port}->{ $data->{port} } = $sock; |
345 |
dpavlin |
232 |
} elsif ( defined $data->{load} && $data->{load} eq 'shard' ) { |
346 |
dpavlin |
221 |
if ( my $path = shift @shard_load_queue ) { |
347 |
|
|
$info->{shard}->{$path} = 'read'; |
348 |
dpavlin |
192 |
my $shard = Storable::retrieve $path; |
349 |
dpavlin |
221 |
$info->{shard}->{$path} = 'send'; |
350 |
dpavlin |
192 |
warn ">>>> [", $data->{port}, "] sending shard $path\n"; |
351 |
|
|
Storable::store_fd( { path => $path, shard => $shard }, $sock ); |
352 |
|
|
} else { |
353 |
|
|
warn "no more shards for [", $data->{port}, "]\n"; |
354 |
dpavlin |
263 |
delete $info->{pending}->{ $data->{port} }; |
355 |
dpavlin |
192 |
} |
356 |
dpavlin |
235 |
} elsif ( exists $data->{out} ) { |
357 |
|
|
my $added = Sack::Merge->add( $data->{out} ) if defined $data->{out}; |
358 |
|
|
$info->{merge}->{ $data->{view} }->{ $data->{port} } = $added; |
359 |
|
|
$info->{view }->{ $data->{view} }->{ $data->{port} } = $data->{on_shard}; |
360 |
dpavlin |
248 |
# refresh shard allocation |
361 |
|
|
$info->{shard}->{ $_ } = $data->{port} foreach keys %{ $data->{on_shard} }; |
362 |
dpavlin |
263 |
delete $info->{pending}->{ $data->{port} }; |
363 |
dpavlin |
247 |
} elsif ( exists $data->{port} ) { |
364 |
|
|
push @responses, $data; |
365 |
|
|
warn "# ",dump($data),$/; |
366 |
dpavlin |
192 |
} else { |
367 |
|
|
warn "UNKNOWN ",dump($data); |
368 |
dpavlin |
164 |
} |
369 |
dpavlin |
163 |
} |
370 |
|
|
} |
371 |
|
|
} |
372 |
|
|
} |