… | |
… | |
72 | |
72 | |
73 | =cut |
73 | =cut |
74 | |
74 | |
75 | package AnyEvent::MP; |
75 | package AnyEvent::MP; |
76 | |
76 | |
77 | use AnyEvent::MP::Util (); |
|
|
78 | use AnyEvent::MP::Node; |
77 | use AnyEvent::MP::Base; |
79 | use AnyEvent::MP::Transport; |
|
|
80 | |
78 | |
81 | use utf8; |
|
|
82 | use common::sense; |
79 | use common::sense; |
83 | |
80 | |
84 | use Carp (); |
81 | use Carp (); |
85 | |
82 | |
86 | use AE (); |
83 | use AE (); |
87 | |
84 | |
88 | use base "Exporter"; |
85 | use base "Exporter"; |
89 | |
86 | |
90 | our $VERSION = '0.01'; |
87 | our $VERSION = '0.02'; |
91 | our @EXPORT = qw(NODE $NODE $PORT snd rcv _any_); |
88 | our @EXPORT = qw( |
92 | |
89 | NODE $NODE $PORT snd rcv _any_ |
93 | our $DEFAULT_SECRET; |
90 | create_port create_port_on |
94 | our $DEFAULT_PORT = "4040"; |
91 | create_miniport |
95 | |
92 | become_slave become_public |
96 | our $CONNECT_INTERVAL = 5; # new connect every 5s, at least |
93 | ); |
97 | our $CONNECT_TIMEOUT = 30; # includes handshake |
|
|
98 | |
|
|
99 | sub default_secret { |
|
|
100 | unless (defined $DEFAULT_SECRET) { |
|
|
101 | if (open my $fh, "<$ENV{HOME}/.aemp-secret") { |
|
|
102 | sysread $fh, $DEFAULT_SECRET, -s $fh; |
|
|
103 | } else { |
|
|
104 | $DEFAULT_SECRET = AnyEvent::MP::Util::nonce 32; |
|
|
105 | } |
|
|
106 | } |
|
|
107 | |
|
|
108 | $DEFAULT_SECRET |
|
|
109 | } |
|
|
110 | |
94 | |
111 | =item NODE / $NODE |
95 | =item NODE / $NODE |
112 | |
96 | |
113 | The C<NODE ()> function and the C<$NODE> variable contain the noderef of |
97 | The C<NODE ()> function and the C<$NODE> variable contain the noderef of |
114 | the local node. The value is initialised by a call to C<become_public> or |
98 | the local node. The value is initialised by a call to C<become_public> or |
115 | C<become_slave>, after which all local port identifiers become invalid. |
99 | C<become_slave>, after which all local port identifiers become invalid. |
116 | |
100 | |
117 | =cut |
|
|
118 | |
|
|
119 | our $UNIQ = sprintf "%x.%x", $$, time; # per-process/node unique cookie |
|
|
120 | our $ID = "a0"; |
|
|
121 | our $PUBLIC = 0; |
|
|
122 | our $NODE; |
|
|
123 | our $PORT; |
|
|
124 | |
|
|
125 | our %NODE; # node id to transport mapping, or "undef", for local node |
|
|
126 | our %PORT; # local ports |
|
|
127 | our %LISTENER; # local transports |
|
|
128 | |
|
|
129 | sub NODE() { $NODE } |
|
|
130 | |
|
|
131 | { |
|
|
132 | use POSIX (); |
|
|
133 | my $nodename = (POSIX::uname)[1]; |
|
|
134 | $NODE = "$$\@$nodename"; |
|
|
135 | } |
|
|
136 | |
|
|
137 | sub _ANY_() { 1 } |
|
|
138 | sub _any_() { \&_ANY_ } |
|
|
139 | |
|
|
140 | sub add_node { |
|
|
141 | my ($noderef) = @_; |
|
|
142 | |
|
|
143 | return $NODE{$noderef} |
|
|
144 | if exists $NODE{$noderef}; |
|
|
145 | |
|
|
146 | for (split /,/, $noderef) { |
|
|
147 | return $NODE{$noderef} = $NODE{$_} |
|
|
148 | if exists $NODE{$_}; |
|
|
149 | } |
|
|
150 | |
|
|
151 | # for indirect sends, use a different class |
|
|
152 | my $node = new AnyEvent::MP::Node::Direct $noderef; |
|
|
153 | |
|
|
154 | $NODE{$_} = $node |
|
|
155 | for $noderef, split /,/, $noderef; |
|
|
156 | |
|
|
157 | $node |
|
|
158 | } |
|
|
159 | |
|
|
160 | =item snd $portid, type => @data |
101 | =item snd $portid, type => @data |
161 | |
102 | |
162 | =item snd $portid, @msg |
103 | =item snd $portid, @msg |
163 | |
104 | |
164 | Send the given message to the given port ID, which can identify either a |
105 | Send the given message to the given port ID, which can identify either |
165 | local or a remote port. |
106 | a local or a remote port, and can be either a string or soemthignt hat |
|
|
107 | stringifies a sa port ID (such as a port object :). |
166 | |
108 | |
167 | While the message can be about anything, it is highly recommended to use |
109 | While the message can be about anything, it is highly recommended to use a |
168 | a constant string as first element. |
110 | string as first element (a portid, or some word that indicates a request |
|
|
111 | type etc.). |
169 | |
112 | |
170 | The message data effectively becomes read-only after a call to this |
113 | The message data effectively becomes read-only after a call to this |
171 | function: modifying any argument is not allowed and can cause many |
114 | function: modifying any argument is not allowed and can cause many |
172 | problems. |
115 | problems. |
173 | |
116 | |
… | |
… | |
175 | JSON is used, then only strings, numbers and arrays and hashes consisting |
118 | JSON is used, then only strings, numbers and arrays and hashes consisting |
176 | of those are allowed (no objects). When Storable is used, then anything |
119 | of those are allowed (no objects). When Storable is used, then anything |
177 | that Storable can serialise and deserialise is allowed, and for the local |
120 | that Storable can serialise and deserialise is allowed, and for the local |
178 | node, anything can be passed. |
121 | node, anything can be passed. |
179 | |
122 | |
180 | =cut |
123 | =item $local_port = create_port |
181 | |
124 | |
182 | sub snd(@) { |
125 | Create a new local port object. See the next section for allowed methods. |
183 | my ($noderef, $port) = split /#/, shift, 2; |
|
|
184 | |
126 | |
185 | add_node $noderef |
127 | =cut |
186 | unless exists $NODE{$noderef}; |
|
|
187 | |
128 | |
188 | $NODE{$noderef}->send (["$port", [@_]]); |
129 | sub create_port { |
189 | } |
130 | my $id = "$AnyEvent::MP::Base::UNIQ." . ++$AnyEvent::MP::Base::ID; |
190 | |
131 | |
|
|
132 | my $self = bless { |
|
|
133 | id => "$NODE#$id", |
|
|
134 | names => [$id], |
|
|
135 | }, "AnyEvent::MP::Port"; |
|
|
136 | |
|
|
137 | $AnyEvent::MP::Base::PORT{$id} = sub { |
|
|
138 | unshift @_, $self; |
|
|
139 | |
|
|
140 | for (@{ $self->{rc0}{$_[1]} }) { |
|
|
141 | $_ && &{$_->[0]} |
|
|
142 | && undef $_; |
|
|
143 | } |
|
|
144 | |
|
|
145 | for (@{ $self->{rcv}{$_[1]} }) { |
|
|
146 | $_ && [@_[1 .. @{$_->[1]}]] ~~ $_->[1] |
|
|
147 | && &{$_->[0]} |
|
|
148 | && undef $_; |
|
|
149 | } |
|
|
150 | |
|
|
151 | for (@{ $self->{any} }) { |
|
|
152 | $_ && [@_[0 .. $#{$_->[1]}]] ~~ $_->[1] |
|
|
153 | && &{$_->[0]} |
|
|
154 | && undef $_; |
|
|
155 | } |
|
|
156 | }; |
|
|
157 | |
|
|
158 | $self |
|
|
159 | } |
|
|
160 | |
|
|
161 | =item $portid = create_miniport { } |
|
|
162 | |
|
|
163 | Creates a "mini port", that is, a port without much #TODO |
|
|
164 | |
|
|
165 | =cut |
|
|
166 | |
|
|
167 | sub create_miniport(&) { |
|
|
168 | my $cb = shift; |
|
|
169 | my $id = "$AnyEvent::MP::Base::UNIQ." . ++$AnyEvent::MP::Base::ID; |
|
|
170 | |
|
|
171 | $AnyEvent::MP::Base::PORT{$id} = sub { |
|
|
172 | # unshift @_, "$NODE#$id"; |
|
|
173 | &$cb |
|
|
174 | and delete $AnyEvent::MP::Base::PORT{$id}; |
|
|
175 | }; |
|
|
176 | |
|
|
177 | "$NODE#$id" |
|
|
178 | } |
|
|
179 | |
|
|
180 | package AnyEvent::MP::Port; |
|
|
181 | |
|
|
182 | =back |
|
|
183 | |
|
|
184 | =head1 METHODS FOR PORT OBJECTS |
|
|
185 | |
|
|
186 | =over 4 |
|
|
187 | |
|
|
188 | =item "$port" |
|
|
189 | |
|
|
190 | A port object stringifies to its port ID, so can be used directly for |
|
|
191 | C<snd> operations. |
|
|
192 | |
|
|
193 | =cut |
|
|
194 | |
|
|
195 | use overload |
|
|
196 | '""' => sub { $_[0]{id} }, |
|
|
197 | fallback => 1; |
|
|
198 | |
191 | =item rcv $portid, type => $callback->(@msg) |
199 | =item $port->rcv (type => $callback->($port, @msg)) |
192 | |
200 | |
193 | =item rcv $portid, $smartmatch => $callback->(@msg) |
201 | =item $port->rcv ($smartmatch => $callback->($port, @msg)) |
194 | |
202 | |
195 | =item rcv $portid, [$smartmatch...] => $callback->(@msg) |
203 | =item $port->rcv ([$smartmatch...] => $callback->($port, @msg)) |
196 | |
204 | |
197 | Register a callback on the port identified by C<$portid>, which I<must> be |
205 | Register a callback on the given port. |
198 | a local port. |
|
|
199 | |
206 | |
200 | The callback has to return a true value when its work is done, after |
207 | The callback has to return a true value when its work is done, after |
201 | which is will be removed, or a false value in which case it will stay |
208 | which is will be removed, or a false value in which case it will stay |
202 | registered. |
209 | registered. |
203 | |
210 | |
… | |
… | |
213 | also the most efficient match (by far). |
220 | also the most efficient match (by far). |
214 | |
221 | |
215 | =cut |
222 | =cut |
216 | |
223 | |
217 | sub rcv($@) { |
224 | sub rcv($@) { |
218 | my ($port, $match, $cb) = @_; |
225 | my ($self, $match, $cb) = @_; |
219 | |
|
|
220 | my $port = $PORT{$port} |
|
|
221 | or do { |
|
|
222 | my ($noderef, $lport) = split /#/, $port; |
|
|
223 | "AnyEvent::MP::Node::Self" eq ref $NODE{$noderef} |
|
|
224 | or Carp::croak "$port: can only rcv on local ports"; |
|
|
225 | |
|
|
226 | $PORT{$lport} |
|
|
227 | or Carp::croak "$port: port does not exist"; |
|
|
228 | |
|
|
229 | $PORT{$port} = $PORT{$lport} # also return |
|
|
230 | }; |
|
|
231 | |
226 | |
232 | if (!ref $match) { |
227 | if (!ref $match) { |
233 | push @{ $port->{rc0}{$match} }, [$cb]; |
228 | push @{ $self->{rc0}{$match} }, [$cb]; |
234 | } elsif (("ARRAY" eq ref $match && !ref $match->[0])) { |
229 | } elsif (("ARRAY" eq ref $match && !ref $match->[0])) { |
235 | my ($type, @match) = @$match; |
230 | my ($type, @match) = @$match; |
236 | @match |
231 | @match |
237 | ? push @{ $port->{rcv}{$match->[0]} }, [$cb, \@match] |
232 | ? push @{ $self->{rcv}{$match->[0]} }, [$cb, \@match] |
238 | : push @{ $port->{rc0}{$match->[0]} }, [$cb]; |
233 | : push @{ $self->{rc0}{$match->[0]} }, [$cb]; |
239 | } else { |
234 | } else { |
240 | push @{ $port->{any} }, [$cb, $match]; |
235 | push @{ $self->{any} }, [$cb, $match]; |
241 | } |
236 | } |
242 | } |
237 | } |
243 | |
238 | |
244 | sub _inject { |
239 | =item $port->register ($name) |
245 | my ($port, $msg) = @{+shift}; |
|
|
246 | |
240 | |
247 | $port = $PORT{$port} |
241 | Registers the given port under the well known name C<$name>. If the name |
248 | or return; |
242 | already exists it is replaced. |
249 | |
243 | |
250 | @_ = @$msg; |
244 | A port can only be registered under one well known name. |
251 | |
245 | |
252 | for (@{ $port->{rc0}{$msg->[0]} }) { |
246 | =cut |
253 | $_ && &{$_->[0]} |
|
|
254 | && undef $_; |
|
|
255 | } |
|
|
256 | |
247 | |
257 | for (@{ $port->{rcv}{$msg->[0]} }) { |
248 | sub register { |
258 | $_ && [@_[1..$#{$_->[1]}]] ~~ $_->[1] |
249 | my ($self, $name) = @_; |
259 | && &{$_->[0]} |
|
|
260 | && undef $_; |
|
|
261 | } |
|
|
262 | |
250 | |
263 | for (@{ $port->{any} }) { |
251 | $self->{wkname} = $name; |
264 | $_ && [@_[0..$#{$_->[1]}]] ~~ $_->[1] |
252 | $AnyEvent::MP::Base::WKP{$name} = "$self"; |
265 | && &{$_->[0]} |
|
|
266 | && undef $_; |
|
|
267 | } |
|
|
268 | } |
253 | } |
269 | |
254 | |
270 | sub normalise_noderef($) { |
255 | =item $port->destroy |
|
|
256 | |
|
|
257 | Explicitly destroy/remove/nuke/vaporise the port. |
|
|
258 | |
|
|
259 | Ports are normally kept alive by there mere existance alone, and need to |
|
|
260 | be destroyed explicitly. |
|
|
261 | |
|
|
262 | =cut |
|
|
263 | |
|
|
264 | sub destroy { |
271 | my ($noderef) = @_; |
265 | my ($self) = @_; |
272 | |
266 | |
273 | my $cv = AE::cv; |
267 | delete $AnyEvent::MP::Base::WKP{ $self->{wkname} }; |
274 | my @res; |
|
|
275 | |
268 | |
276 | $cv->begin (sub { |
269 | delete $AnyEvent::MP::Base::PORT{$_} |
277 | my %seen; |
270 | for @{ $self->{names} }; |
278 | my @refs; |
|
|
279 | for (sort { $a->[0] <=> $b->[0] } @res) { |
|
|
280 | push @refs, $_->[1] unless $seen{$_->[1]}++ |
|
|
281 | } |
|
|
282 | shift->send (join ",", @refs); |
|
|
283 | }); |
|
|
284 | |
|
|
285 | $noderef = $DEFAULT_PORT unless length $noderef; |
|
|
286 | |
|
|
287 | my $idx; |
|
|
288 | for my $t (split /,/, $noderef) { |
|
|
289 | my $pri = ++$idx; |
|
|
290 | |
|
|
291 | #TODO: this should be outside normalise_noderef and in become_public |
|
|
292 | if ($t =~ /^\d*$/) { |
|
|
293 | my $nodename = (POSIX::uname)[1]; |
|
|
294 | |
|
|
295 | $cv->begin; |
|
|
296 | AnyEvent::Socket::resolve_sockaddr $nodename, $t || "aemp=$DEFAULT_PORT", "tcp", 0, undef, sub { |
|
|
297 | for (@_) { |
|
|
298 | my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3]; |
|
|
299 | push @res, [ |
|
|
300 | $pri += 1e-5, |
|
|
301 | AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service |
|
|
302 | ]; |
|
|
303 | } |
|
|
304 | $cv->end; |
|
|
305 | }; |
|
|
306 | |
|
|
307 | # my (undef, undef, undef, undef, @ipv4) = gethostbyname $nodename; |
|
|
308 | # |
|
|
309 | # for (@ipv4) { |
|
|
310 | # push @res, [ |
|
|
311 | # $pri, |
|
|
312 | # AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $_, $t || $DEFAULT_PORT, |
|
|
313 | # ]; |
|
|
314 | # } |
|
|
315 | } else { |
|
|
316 | my ($host, $port) = AnyEvent::Socket::parse_hostport $t, "aemp=$DEFAULT_PORT" |
|
|
317 | or Carp::croak "$t: unparsable transport descriptor"; |
|
|
318 | |
|
|
319 | $cv->begin; |
|
|
320 | AnyEvent::Socket::resolve_sockaddr $host, $port, "tcp", 0, undef, sub { |
|
|
321 | for (@_) { |
|
|
322 | my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3]; |
|
|
323 | push @res, [ |
|
|
324 | $pri += 1e-5, |
|
|
325 | AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service |
|
|
326 | ]; |
|
|
327 | } |
|
|
328 | $cv->end; |
|
|
329 | } |
|
|
330 | } |
|
|
331 | } |
|
|
332 | |
|
|
333 | $cv->end; |
|
|
334 | |
|
|
335 | $cv |
|
|
336 | } |
271 | } |
337 | |
272 | |
338 | sub become_public { |
273 | =back |
339 | return if $PUBLIC; |
|
|
340 | |
274 | |
341 | my $noderef = join ",", ref $_[0] ? @{+shift} : shift; |
275 | =head1 FUNCTIONS FOR NODES |
342 | my @args = @_; |
|
|
343 | |
276 | |
344 | $NODE = (normalise_noderef $noderef)->recv; |
277 | =over 4 |
345 | |
278 | |
346 | for my $t (split /,/, $NODE) { |
279 | =item mon $noderef, $callback->($noderef, $status, $) |
347 | $NODE{$t} = $NODE{""}; |
|
|
348 | |
280 | |
349 | my ($host, $port) = AnyEvent::Socket::parse_hostport $t; |
281 | Monitors the given noderef. |
350 | |
282 | |
351 | $LISTENER{$t} = AnyEvent::MP::Transport::mp_server $host, $port, |
283 | =item become_public endpoint... |
352 | @args, |
|
|
353 | on_error => sub { |
|
|
354 | die "on_error<@_>\n";#d# |
|
|
355 | }, |
|
|
356 | on_connect => sub { |
|
|
357 | my ($tp) = @_; |
|
|
358 | |
284 | |
359 | $NODE{$tp->{remote_id}} = $_[0]; |
285 | Tells the node to become a public node, i.e. reachable from other nodes. |
360 | }, |
|
|
361 | sub { |
|
|
362 | my ($tp) = @_; |
|
|
363 | |
286 | |
364 | $NODE{"$tp->{peerhost}:$tp->{peerport}"} = $tp; |
287 | If no arguments are given, or the first argument is C<undef>, then |
365 | }, |
288 | AnyEvent::MP tries to bind on port C<4040> on all IP addresses that the |
366 | ; |
289 | local nodename resolves to. |
367 | } |
|
|
368 | |
290 | |
369 | $PUBLIC = 1; |
291 | Otherwise the first argument must be an array-reference with transport |
370 | } |
292 | endpoints ("ip:port", "hostname:port") or port numbers (in which case the |
|
|
293 | local nodename is used as hostname). The endpoints are all resolved and |
|
|
294 | will become the node reference. |
|
|
295 | |
|
|
296 | =cut |
371 | |
297 | |
372 | =back |
298 | =back |
373 | |
299 | |
374 | =head1 NODE MESSAGES |
300 | =head1 NODE MESSAGES |
375 | |
301 | |
… | |
… | |
380 | |
306 | |
381 | =over 4 |
307 | =over 4 |
382 | |
308 | |
383 | =cut |
309 | =cut |
384 | |
310 | |
385 | ############################################################################# |
311 | =item wkp => $name, @reply |
386 | # self node code |
|
|
387 | |
312 | |
388 | sub _new_port($) { |
313 | Replies with the port ID of the specified well-known port, or C<undef>. |
389 | my ($name) = @_; |
|
|
390 | |
314 | |
391 | my ($noderef, $portname) = split /#/, $name; |
315 | =item devnull => ... |
392 | |
316 | |
393 | $PORT{$name} = |
317 | Generic data sink/CPU heat conversion. |
394 | $PORT{$portname} = { |
|
|
395 | names => [$name, $portname], |
|
|
396 | }; |
|
|
397 | } |
|
|
398 | |
|
|
399 | $NODE{""} = new AnyEvent::MP::Node::Self noderef => $NODE; |
|
|
400 | _new_port ""; |
|
|
401 | |
318 | |
402 | =item relay => $port, @msg |
319 | =item relay => $port, @msg |
403 | |
320 | |
404 | Simply forwards the message to the given port. |
321 | Simply forwards the message to the given port. |
405 | |
|
|
406 | =cut |
|
|
407 | |
|
|
408 | rcv "", relay => \&snd; |
|
|
409 | |
322 | |
410 | =item eval => $string[ @reply] |
323 | =item eval => $string[ @reply] |
411 | |
324 | |
412 | Evaluates the given string. If C<@reply> is given, then a message of the |
325 | Evaluates the given string. If C<@reply> is given, then a message of the |
413 | form C<@reply, $@, @evalres> is sent. |
326 | form C<@reply, $@, @evalres> is sent. |
414 | |
327 | |
415 | Example: crash another node. |
328 | Example: crash another node. |
416 | |
329 | |
417 | snd $othernode, eval => "exit"; |
330 | snd $othernode, eval => "exit"; |
418 | |
331 | |
419 | =cut |
|
|
420 | |
|
|
421 | rcv "", eval => sub { |
|
|
422 | my (undef, $string, @reply) = @_; |
|
|
423 | my @res = eval $string; |
|
|
424 | snd @reply, "$@", @res if @reply; |
|
|
425 | }; |
|
|
426 | |
|
|
427 | =item time => @reply |
332 | =item time => @reply |
428 | |
333 | |
429 | Replies the the current node time to C<@reply>. |
334 | Replies the the current node time to C<@reply>. |
430 | |
335 | |
431 | Example: tell the current node to send the current time to C<$myport> in a |
336 | Example: tell the current node to send the current time to C<$myport> in a |
432 | C<timereply> message. |
337 | C<timereply> message. |
433 | |
338 | |
434 | snd $NODE, time => $myport, timereply => 1, 2; |
339 | snd $NODE, time => $myport, timereply => 1, 2; |
435 | # => snd $myport, timereply => 1, 2, <time> |
340 | # => snd $myport, timereply => 1, 2, <time> |
436 | |
341 | |
437 | =cut |
|
|
438 | |
|
|
439 | rcv "", time => sub { shift; snd @_, AE::time }; |
|
|
440 | |
|
|
441 | =back |
342 | =back |
442 | |
343 | |
443 | =head1 SEE ALSO |
344 | =head1 SEE ALSO |
444 | |
345 | |
445 | L<AnyEvent>. |
346 | L<AnyEvent>. |