| 1 |
#!/usr/bin/perl |
| 2 |
|
| 3 |
package Sack::Server; |
| 4 |
|
| 5 |
use warnings; |
| 6 |
use strict; |
| 7 |
|
| 8 |
use IO::Socket::INET; |
| 9 |
use IO::Select; |
| 10 |
|
| 11 |
use Data::Dump qw(dump); |
| 12 |
use Storable qw(); |
| 13 |
use File::Slurp; |
| 14 |
use Cwd qw(abs_path); |
| 15 |
|
| 16 |
my @cloud; |
| 17 |
my $cloud_path = $ENV{CLOUD} || die "start with: CLOUD=etc/cloud perl -I/srv/Sack/lib $0\n"; |
| 18 |
@cloud = read_file $cloud_path; |
| 19 |
@cloud = map { chomp $_; $_ } @cloud; |
| 20 |
|
| 21 |
warn "# cloud ",dump( @cloud ); |
| 22 |
|
| 23 |
my $listen_port = 4444; |
| 24 |
|
| 25 |
my $node_path = abs_path $0; |
| 26 |
$node_path =~ s{Server}{Client}; |
| 27 |
|
| 28 |
my $lsn = IO::Socket::INET->new(Listen => 1, LocalPort => $listen_port, Reuse => 1) or die $!; |
| 29 |
my $sel = IO::Select->new($lsn); |
| 30 |
|
| 31 |
my $info; |
| 32 |
sub info { |
| 33 |
my $port = shift; |
| 34 |
push @{ $info->{$port} }, [ @_ ]; |
| 35 |
} |
| 36 |
|
| 37 |
sub fork_ssh { |
| 38 |
my ( $port, $host ) = @_; |
| 39 |
|
| 40 |
if ( my $pid = fork ) { |
| 41 |
# parent |
| 42 |
info $port => 'forked', $pid; |
| 43 |
return $port; |
| 44 |
|
| 45 |
} elsif ( ! defined $pid ) { |
| 46 |
warn "can't fork $host $port"; |
| 47 |
return; |
| 48 |
} else { |
| 49 |
# child |
| 50 |
my $cmd = qq|ssh -F $cloud_path.ssh -R $port:127.0.0.1:$listen_port $host $node_path $port|; |
| 51 |
warn "# exec: $cmd\n"; |
| 52 |
exec $cmd; |
| 53 |
} |
| 54 |
} |
| 55 |
|
| 56 |
my $node_port = 4000; |
| 57 |
|
| 58 |
foreach my $host ( @cloud ) { |
| 59 |
system "find /srv/Sack/ | cpio --create --dereference | ssh -T -F $cloud_path.ssh $host cpio --extract --make-directories --unconditional"; |
| 60 |
fork_ssh( $node_port++, $host ); |
| 61 |
} |
| 62 |
|
| 63 |
my $session; |
| 64 |
|
| 65 |
sub to_all { |
| 66 |
my $data = shift; |
| 67 |
foreach my $port ( keys %{ $session->{port} } ) { |
| 68 |
warn ">>>> [$port]\n"; |
| 69 |
Storable::store_fd( $data, $session->{port}->{$port} ); |
| 70 |
} |
| 71 |
} |
| 72 |
|
| 73 |
my @shard_paths; |
| 74 |
|
| 75 |
while (1) { |
| 76 |
for my $sock ($sel->can_read(1)) { |
| 77 |
if ($sock == $lsn) { |
| 78 |
my $new = $lsn->accept; |
| 79 |
$sel->add($new); |
| 80 |
$session->{peerport}->{ $new->peerport } = $new; |
| 81 |
warn "[socket] connect\n"; |
| 82 |
Storable::store_fd( { ping => 1 }, $new ); |
| 83 |
info 0 => 'ping', $new->peerport; |
| 84 |
} else { |
| 85 |
my $data = eval { Storable::fd_retrieve( $sock ) }; |
| 86 |
if ( $@ ) { |
| 87 |
delete $session->{$sock}; |
| 88 |
warn "[socket] disconnect: $@\n"; |
| 89 |
$sel->remove($sock); |
| 90 |
$sock->close; |
| 91 |
} else { |
| 92 |
warn "<<<< ", dump($data), $/; |
| 93 |
if ( my $repl = $data->{repl} ) { |
| 94 |
my $response = { repl_pid => $$ }; |
| 95 |
if ( $repl =~ m/ping/ ) { |
| 96 |
to_all { ping => 1 }; |
| 97 |
} elsif ( $repl =~ m/info/ ) { |
| 98 |
$response->{info} = $info; |
| 99 |
} elsif ( $repl =~ m{load\s*(\S+)?} ) { |
| 100 |
my $name = $1 || 'shard'; |
| 101 |
@shard_paths = glob "/tmp/sack/$name/*"; |
| 102 |
warn "loading shards ", dump( @shard_paths ); |
| 103 |
to_all { load => $name }; |
| 104 |
} else { |
| 105 |
$response->{error}->{unknown} = $data; |
| 106 |
} |
| 107 |
Storable::store_fd( $response, $sock ); |
| 108 |
} elsif ( $data->{ping} ) { |
| 109 |
my $port = $data->{port}; |
| 110 |
info $port => 'ping', $port; |
| 111 |
$session->{port}->{ $data->{port} } = $sock; |
| 112 |
} elsif ( $data->{load} eq 'shard' ) { |
| 113 |
if ( my $path = shift @shard_paths ) { |
| 114 |
warn "retrieve $path ", -s $path; |
| 115 |
my $shard = Storable::retrieve $path; |
| 116 |
warn ">>>> [", $data->{port}, "] sending shard $path\n"; |
| 117 |
Storable::store_fd( { path => $path, shard => $shard }, $sock ); |
| 118 |
} else { |
| 119 |
warn "no more shards for [", $data->{port}, "]\n"; |
| 120 |
} |
| 121 |
} else { |
| 122 |
warn "UNKNOWN ",dump($data); |
| 123 |
} |
| 124 |
} |
| 125 |
} |
| 126 |
} |
| 127 |
} |
Parent Directory
| 
Revision Log