1 |
dpavlin |
163 |
#!/usr/bin/perl |
2 |
|
|
|
3 |
dpavlin |
179 |
package Sack::Server; |
4 |
|
|
|
5 |
dpavlin |
163 |
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 |
dpavlin |
169 |
use File::Slurp; |
14 |
dpavlin |
176 |
use Cwd qw(abs_path); |
15 |
dpavlin |
163 |
|
16 |
dpavlin |
200 |
use lib '/srv/Sack/lib'; |
17 |
|
|
use Sack::Merge; |
18 |
dpavlin |
210 |
use Sack::Server::HTTP; |
19 |
dpavlin |
200 |
|
20 |
dpavlin |
179 |
my @cloud; |
21 |
dpavlin |
209 |
my $cloud_path = $ENV{CLOUD} || "etc/cloud"; |
22 |
|
|
die "start with: CLOUD=etc/cloud perl -I/srv/Sack/lib $0\n" unless -e $cloud_path; |
23 |
dpavlin |
169 |
@cloud = read_file $cloud_path; |
24 |
|
|
@cloud = map { chomp $_; $_ } @cloud; |
25 |
|
|
|
26 |
|
|
warn "# cloud ",dump( @cloud ); |
27 |
|
|
|
28 |
dpavlin |
163 |
my $listen_port = 4444; |
29 |
dpavlin |
210 |
my $http_port = 4480; |
30 |
dpavlin |
163 |
|
31 |
dpavlin |
176 |
my $node_path = abs_path $0; |
32 |
dpavlin |
181 |
$node_path =~ s{Server}{Client}; |
33 |
dpavlin |
163 |
|
34 |
dpavlin |
210 |
my $lsn = IO::Socket::INET->new(Listen => 1, LocalPort => $listen_port, Reuse => 1) or die "$listen_port $!"; |
35 |
|
|
my $www = IO::Socket::INET->new(Listen => 1, LocalPort => $http_port, Reuse => 1) or die "$http_port $!"; |
36 |
dpavlin |
163 |
my $sel = IO::Select->new($lsn); |
37 |
dpavlin |
210 |
$sel->add( $www ); |
38 |
dpavlin |
163 |
|
39 |
dpavlin |
166 |
my $info; |
40 |
|
|
sub info { |
41 |
|
|
my $port = shift; |
42 |
|
|
push @{ $info->{$port} }, [ @_ ]; |
43 |
|
|
} |
44 |
|
|
|
45 |
dpavlin |
179 |
sub fork_ssh { |
46 |
dpavlin |
163 |
my ( $port, $host ) = @_; |
47 |
|
|
|
48 |
|
|
if ( my $pid = fork ) { |
49 |
|
|
# parent |
50 |
dpavlin |
166 |
info $port => 'forked', $pid; |
51 |
dpavlin |
163 |
return $port; |
52 |
|
|
|
53 |
|
|
} elsif ( ! defined $pid ) { |
54 |
|
|
warn "can't fork $host $port"; |
55 |
|
|
return; |
56 |
|
|
} else { |
57 |
|
|
# child |
58 |
dpavlin |
169 |
my $cmd = qq|ssh -F $cloud_path.ssh -R $port:127.0.0.1:$listen_port $host $node_path $port|; |
59 |
dpavlin |
163 |
warn "# exec: $cmd\n"; |
60 |
|
|
exec $cmd; |
61 |
|
|
} |
62 |
|
|
} |
63 |
|
|
|
64 |
|
|
my $node_port = 4000; |
65 |
|
|
|
66 |
dpavlin |
169 |
foreach my $host ( @cloud ) { |
67 |
dpavlin |
176 |
system "find /srv/Sack/ | cpio --create --dereference | ssh -T -F $cloud_path.ssh $host cpio --extract --make-directories --unconditional"; |
68 |
dpavlin |
179 |
fork_ssh( $node_port++, $host ); |
69 |
dpavlin |
163 |
} |
70 |
|
|
|
71 |
|
|
my $session; |
72 |
|
|
|
73 |
dpavlin |
188 |
sub to_all { |
74 |
|
|
my $data = shift; |
75 |
|
|
foreach my $port ( keys %{ $session->{port} } ) { |
76 |
|
|
warn ">>>> [$port]\n"; |
77 |
|
|
Storable::store_fd( $data, $session->{port}->{$port} ); |
78 |
|
|
} |
79 |
|
|
} |
80 |
|
|
|
81 |
dpavlin |
192 |
my @shard_paths; |
82 |
|
|
|
83 |
dpavlin |
163 |
while (1) { |
84 |
|
|
for my $sock ($sel->can_read(1)) { |
85 |
|
|
if ($sock == $lsn) { |
86 |
|
|
my $new = $lsn->accept; |
87 |
|
|
$sel->add($new); |
88 |
dpavlin |
174 |
$session->{peerport}->{ $new->peerport } = $new; |
89 |
dpavlin |
164 |
warn "[socket] connect\n"; |
90 |
dpavlin |
163 |
Storable::store_fd( { ping => 1 }, $new ); |
91 |
dpavlin |
166 |
info 0 => 'ping', $new->peerport; |
92 |
dpavlin |
210 |
} elsif ( $sock == $www ) { |
93 |
|
|
my $client = $www->accept; |
94 |
|
|
Sack::Server::HTTP::request( $client, sub { |
95 |
|
|
warn "XXX callback ",dump( @_ ); |
96 |
|
|
my ( $send, $method, $param ) = @_; |
97 |
|
|
print $send "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n" |
98 |
|
|
, dump( $info ) |
99 |
|
|
; |
100 |
|
|
} ); |
101 |
dpavlin |
163 |
} else { |
102 |
dpavlin |
165 |
my $data = eval { Storable::fd_retrieve( $sock ) }; |
103 |
|
|
if ( $@ ) { |
104 |
|
|
delete $session->{$sock}; |
105 |
|
|
warn "[socket] disconnect: $@\n"; |
106 |
|
|
$sel->remove($sock); |
107 |
|
|
$sock->close; |
108 |
|
|
} else { |
109 |
dpavlin |
163 |
warn "<<<< ", dump($data), $/; |
110 |
dpavlin |
196 |
|
111 |
|
|
if ( my $path = $data->{shard} ) { |
112 |
|
|
push @{ $info->{shard}->{ $data->{port} } }, $path; |
113 |
|
|
} |
114 |
|
|
|
115 |
dpavlin |
192 |
if ( my $repl = $data->{repl} ) { |
116 |
|
|
my $response = { repl_pid => $$ }; |
117 |
|
|
if ( $repl =~ m/ping/ ) { |
118 |
dpavlin |
188 |
to_all { ping => 1 }; |
119 |
dpavlin |
192 |
} elsif ( $repl =~ m/info/ ) { |
120 |
dpavlin |
175 |
$response->{info} = $info; |
121 |
dpavlin |
192 |
} elsif ( $repl =~ m{load\s*(\S+)?} ) { |
122 |
|
|
my $name = $1 || 'shard'; |
123 |
|
|
@shard_paths = glob "/tmp/sack/$name/*"; |
124 |
dpavlin |
203 |
warn "loading shard $name from ", dump( @shard_paths ); |
125 |
dpavlin |
192 |
to_all { load => $name }; |
126 |
dpavlin |
196 |
} elsif ( $repl =~ m{view\s*(\S+)?} ) { |
127 |
|
|
my $path = $1 || '/srv/Sack/views/00.demo.pl'; |
128 |
|
|
my $code = read_file $path; |
129 |
dpavlin |
200 |
Sack::Merge->clean; |
130 |
dpavlin |
196 |
to_all { view => $code, path => $path }; |
131 |
|
|
$response->{view}->{$path}->{running}; |
132 |
|
|
} elsif ( $repl =~ m{debug\s*(.+)?} ) { |
133 |
|
|
to_all { debug => $1 }; |
134 |
dpavlin |
200 |
} elsif ( $repl =~ m{out} ) { |
135 |
|
|
my $out = Sack::Merge->out; |
136 |
|
|
warn "out ",dump( $out ); |
137 |
|
|
$response->{out} = $out; |
138 |
dpavlin |
203 |
} elsif ( $repl =~ m{clean} ) { |
139 |
|
|
delete $info->{shard}; |
140 |
|
|
to_all { clean => 1 }; |
141 |
dpavlin |
189 |
} else { |
142 |
|
|
$response->{error}->{unknown} = $data; |
143 |
dpavlin |
174 |
} |
144 |
dpavlin |
166 |
Storable::store_fd( $response, $sock ); |
145 |
|
|
} elsif ( $data->{ping} ) { |
146 |
dpavlin |
174 |
my $port = $data->{port}; |
147 |
|
|
info $port => 'ping', $port; |
148 |
|
|
$session->{port}->{ $data->{port} } = $sock; |
149 |
dpavlin |
192 |
} elsif ( $data->{load} eq 'shard' ) { |
150 |
|
|
if ( my $path = shift @shard_paths ) { |
151 |
|
|
warn "retrieve $path ", -s $path; |
152 |
|
|
my $shard = Storable::retrieve $path; |
153 |
|
|
warn ">>>> [", $data->{port}, "] sending shard $path\n"; |
154 |
|
|
Storable::store_fd( { path => $path, shard => $shard }, $sock ); |
155 |
|
|
} else { |
156 |
|
|
warn "no more shards for [", $data->{port}, "]\n"; |
157 |
|
|
} |
158 |
dpavlin |
200 |
} elsif ( defined $data->{out} ) { |
159 |
|
|
Sack::Merge->add( $data->{out} ); |
160 |
dpavlin |
192 |
} else { |
161 |
|
|
warn "UNKNOWN ",dump($data); |
162 |
dpavlin |
164 |
} |
163 |
dpavlin |
163 |
} |
164 |
|
|
} |
165 |
|
|
} |
166 |
|
|
} |