ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent-MP/MP.pm
(Generate patch)

Comparing AnyEvent-MP/MP.pm (file contents):
Revision 1.2 by root, Fri Jul 31 20:55:46 2009 UTC vs.
Revision 1.22 by root, Tue Aug 4 18:33:30 2009 UTC

4 4
5=head1 SYNOPSIS 5=head1 SYNOPSIS
6 6
7 use AnyEvent::MP; 7 use AnyEvent::MP;
8 8
9 NODE # returns this node identifier
10 $NODE # contains this node identifier 9 $NODE # contains this node's noderef
10 NODE # returns this node's noderef
11 NODE $port # returns the noderef of the port
11 12
12 snd $port, type => data...; 13 snd $port, type => data...;
14
15 $SELF # receiving/own port id in rcv callbacks
13 16
14 rcv $port, smartmatch => $cb->($port, @msg); 17 rcv $port, smartmatch => $cb->($port, @msg);
15 18
16 # examples: 19 # examples:
17 rcv $port2, ping => sub { snd $_[0], "pong"; 0 }; 20 rcv $port2, ping => sub { snd $_[0], "pong"; 0 };
27This module (-family) implements a simple message passing framework. 30This module (-family) implements a simple message passing framework.
28 31
29Despite its simplicity, you can securely message other processes running 32Despite its simplicity, you can securely message other processes running
30on the same or other hosts. 33on the same or other hosts.
31 34
35At the moment, this module family is severly brokena nd underdocumented,
36so do not use. This was uploaded mainly to reserve the CPAN namespace -
37stay tuned!
38
32=head1 CONCEPTS 39=head1 CONCEPTS
33 40
34=over 4 41=over 4
35 42
36=item port 43=item port
37 44
38A port is something you can send messages to with the C<snd> function, and 45A port is something you can send messages to with the C<snd> function, and
39you can register C<rcv> handlers with. All C<rcv> handlers will receive 46you can register C<rcv> handlers with. All C<rcv> handlers will receive
40messages they match, messages will not be queued. 47messages they match, messages will not be queued.
41 48
42=item port id - C<pid@host#portname> 49=item port id - C<noderef#portname>
43 50
44A port id is always the node id, a hash-mark (C<#>) as separator, followed 51A port id is always the noderef, a hash-mark (C<#>) as separator, followed
45by a port name. 52by a port name (a printable string of unspecified format).
46
47A port name can be a well known port (basically an identifier/bareword),
48or a generated name, consisting of node id, a dot (C<.>), and an
49identifier.
50 53
51=item node 54=item node
52 55
53A node is a single process containing at least one port - the node 56A node is a single process containing at least one port - the node
54port. You can send messages to node ports to let them create new ports, 57port. You can send messages to node ports to let them create new ports,
55among other things. 58among other things.
56 59
57Initially, nodes are either private (single-process only) or hidden 60Initially, nodes are either private (single-process only) or hidden
58(connected to a father node only). Only when they epxlicitly "go public" 61(connected to a master node only). Only when they epxlicitly "become
59can you send them messages form unrelated other nodes. 62public" can you send them messages from unrelated other nodes.
60 63
61Public nodes automatically connect to all other public nodes in a network 64=item noderef - C<host:port,host:port...>, C<id@noderef>, C<id>
62when they connect, creating a full mesh.
63 65
64=item node id - C<host:port>, C<id@host>, C<id>
65
66A node ID is a string that either uniquely identifies a given node (For 66A noderef is a string that either uniquely identifies a given node (for
67private and hidden nodes), or contains a recipe on how to reach a given 67private and hidden nodes), or contains a recipe on how to reach a given
68node (for public nodes). 68node (for public nodes).
69 69
70=back 70=back
71 71
72=head1 FUNCTIONS 72=head1 VARIABLES/FUNCTIONS
73 73
74=over 4 74=over 4
75 75
76=cut 76=cut
77 77
78package AnyEvent::MP; 78package AnyEvent::MP;
79 79
80use AnyEvent::MP::Util ();
81use AnyEvent::MP::Node; 80use AnyEvent::MP::Base;
82use AnyEvent::MP::Transport;
83 81
84use utf8;
85use common::sense; 82use common::sense;
86 83
87use Carp (); 84use Carp ();
88 85
89use AE (); 86use AE ();
90 87
91use base "Exporter"; 88use base "Exporter";
92 89
93our $VERSION = '0.0'; 90our $VERSION = '0.02';
94our @EXPORT = qw(NODE $NODE $PORT snd rcv _any_); 91our @EXPORT = qw(
92 NODE $NODE *SELF node_of _any_
93 become_slave become_public
94 snd rcv mon kil reg psub
95 port
96);
95 97
96our $DEFAULT_SECRET; 98our $SELF;
97our $DEFAULT_PORT = "4040";
98 99
99our $CONNECT_INTERVAL = 5; # new connect every 5s, at least 100sub _self_die() {
100our $CONNECT_TIMEOUT = 30; # includes handshake 101 my $msg = $@;
102 $msg =~ s/\n+$// unless ref $msg;
103 kil $SELF, die => $msg;
104}
101 105
102sub default_secret { 106=item $thisnode = NODE / $NODE
103 unless (defined $DEFAULT_SECRET) { 107
104 if (open my $fh, "<$ENV{HOME}/.aemp-secret") { 108The C<NODE> function returns, and the C<$NODE> variable contains
105 sysread $fh, $DEFAULT_SECRET, -s $fh; 109the noderef of the local node. The value is initialised by a call
110to C<become_public> or C<become_slave>, after which all local port
111identifiers become invalid.
112
113=item $noderef = node_of $portid
114
115Extracts and returns the noderef from a portid or a noderef.
116
117=item $SELF
118
119Contains the current port id while executing C<rcv> callbacks or C<psub>
120blocks.
121
122=item SELF, %SELF, @SELF...
123
124Due to some quirks in how perl exports variables, it is impossible to
125just export C<$SELF>, all the symbols called C<SELF> are exported by this
126module, but only C<$SELF> is currently used.
127
128=item snd $portid, type => @data
129
130=item snd $portid, @msg
131
132Send the given message to the given port ID, which can identify either
133a local or a remote port, and can be either a string or soemthignt hat
134stringifies a sa port ID (such as a port object :).
135
136While the message can be about anything, it is highly recommended to use a
137string as first element (a portid, or some word that indicates a request
138type etc.).
139
140The message data effectively becomes read-only after a call to this
141function: modifying any argument is not allowed and can cause many
142problems.
143
144The type of data you can transfer depends on the transport protocol: when
145JSON is used, then only strings, numbers and arrays and hashes consisting
146of those are allowed (no objects). When Storable is used, then anything
147that Storable can serialise and deserialise is allowed, and for the local
148node, anything can be passed.
149
150=item kil $portid[, @reason]
151
152Kill the specified port with the given C<@reason>.
153
154If no C<@reason> is specified, then the port is killed "normally" (linked
155ports will not be kileld, or even notified).
156
157Otherwise, linked ports get killed with the same reason (second form of
158C<mon>, see below).
159
160Runtime errors while evaluating C<rcv> callbacks or inside C<psub> blocks
161will be reported as reason C<< die => $@ >>.
162
163Transport/communication errors are reported as C<< transport_error =>
164$message >>.
165
166=item $guard = mon $portid, $cb->(@reason)
167
168=item $guard = mon $portid, $otherport
169
170=item $guard = mon $portid, $otherport, @msg
171
172Monitor the given port and do something when the port is killed.
173
174In the first form, the callback is simply called with any number
175of C<@reason> elements (no @reason means that the port was deleted
176"normally"). Note also that I<< the callback B<must> never die >>, so use
177C<eval> if unsure.
178
179In the second form, the other port will be C<kil>'ed with C<@reason>, iff
180a @reason was specified, i.e. on "normal" kils nothing happens, while
181under all other conditions, the other port is killed with the same reason.
182
183In the last form, a message of the form C<@msg, @reason> will be C<snd>.
184
185Example: call a given callback when C<$port> is killed.
186
187 mon $port, sub { warn "port died because of <@_>\n" };
188
189Example: kill ourselves when C<$port> is killed abnormally.
190
191 mon $port, $self;
192
193Example: send us a restart message another C<$port> is killed.
194
195 mon $port, $self => "restart";
196
197=cut
198
199sub mon {
200 my ($noderef, $port, $cb) = ((split /#/, shift, 2), shift);
201
202 my $node = $NODE{$noderef} || add_node $noderef;
203
204 #TODO: ports must not be references
205 if (!ref $cb or "AnyEvent::MP::Port" eq ref $cb) {
206 if (@_) {
207 # send a kill info message
208 my (@msg) = ($cb, @_);
209 $cb = sub { snd @msg, @_ };
106 } else { 210 } else {
107 $DEFAULT_SECRET = AnyEvent::MP::Util::nonce 32; 211 # simply kill other port
212 my $port = $cb;
213 $cb = sub { kil $port, @_ if @_ };
108 } 214 }
109 } 215 }
110 216
111 $DEFAULT_SECRET 217 $node->monitor ($port, $cb);
112}
113 218
114our $UNIQ = sprintf "%x.%x", $$, time; # per-process/node unique cookie 219 defined wantarray
115our $PUBLIC = 0; 220 and AnyEvent::Util::guard { $node->unmonitor ($port, $cb) }
116our $NODE;
117our $PORT;
118
119our %NODE; # node id to transport mapping, or "undef", for local node
120our %PORT; # local ports
121our %LISTENER; # local transports
122
123sub NODE() { $NODE }
124
125{
126 use POSIX ();
127 my $nodename = (POSIX::uname)[1];
128 $NODE = "$$\@$nodename";
129} 221}
130 222
131sub _ANY_() { 1 } 223=item $guard = mon_guard $port, $ref, $ref...
132sub _any_() { \&_ANY_ }
133 224
134sub add_node { 225Monitors the given C<$port> and keeps the passed references. When the port
226is killed, the references will be freed.
227
228Optionally returns a guard that will stop the monitoring.
229
230This function is useful when you create e.g. timers or other watchers and
231want to free them when the port gets killed:
232
233 $port->rcv (start => sub {
234 my $timer; $timer = mon_guard $port, AE::timer 1, 1, sub {
235 undef $timer if 0.9 < rand;
236 });
237 });
238
239=cut
240
241sub mon_guard {
135 my ($noderef) = @_; 242 my ($port, @refs) = @_;
136 243
137 return $NODE{$noderef} 244 mon $port, sub { 0 && @refs }
138 if exists $NODE{$noderef}; 245}
139 246
140 for (split /,/, $noderef) { 247=item $local_port = port
141 return $NODE{$noderef} = $NODE{$_} 248
142 if exists $NODE{$_}; 249Create a new local port object that supports message matching.
250
251=item $portid = port { my @msg = @_; $finished }
252
253Creates a "mini port", that is, a very lightweight port without any
254pattern matching behind it, and returns its ID.
255
256The block will be called for every message received on the port. When the
257callback returns a true value its job is considered "done" and the port
258will be destroyed. Otherwise it will stay alive.
259
260The message will be passed as-is, no extra argument (i.e. no port id) will
261be passed to the callback.
262
263If you need the local port id in the callback, this works nicely:
264
265 my $port; $port = miniport {
266 snd $otherport, reply => $port;
267 };
268
269=cut
270
271sub port(;&) {
272 my $id = "$UNIQ." . $ID++;
273 my $port = "$NODE#$id";
274
275 if (@_) {
276 my $cb = shift;
277 $PORT{$id} = sub {
278 local $SELF = $port;
279 eval {
280 &$cb
281 and kil $id;
282 };
283 _self_die if $@;
284 };
285 } else {
286 my $self = bless {
287 id => "$NODE#$id",
288 }, "AnyEvent::MP::Port";
289
290 $PORT_DATA{$id} = $self;
291 $PORT{$id} = sub {
292 local $SELF = $port;
293
294 eval {
295 for (@{ $self->{rc0}{$_[0]} }) {
296 $_ && &{$_->[0]}
297 && undef $_;
298 }
299
300 for (@{ $self->{rcv}{$_[0]} }) {
301 $_ && [@_[1 .. @{$_->[1]}]] ~~ $_->[1]
302 && &{$_->[0]}
303 && undef $_;
304 }
305
306 for (@{ $self->{any} }) {
307 $_ && [@_[0 .. $#{$_->[1]}]] ~~ $_->[1]
308 && &{$_->[0]}
309 && undef $_;
310 }
311 };
312 _self_die if $@;
313 };
143 } 314 }
144 315
145 # for indirect sends, use a different class 316 $port
146 my $node = new AnyEvent::MP::Node::Direct $noderef;
147
148 $NODE{$_} = $node
149 for $noderef, split /,/, $noderef;
150
151 $node
152} 317}
153 318
319=item reg $portid, $name
320
321Registers the given port under the name C<$name>. If the name already
322exists it is replaced.
323
324A port can only be registered under one well known name.
325
326A port automatically becomes unregistered when it is killed.
327
328=cut
329
330sub reg(@) {
331 my ($portid, $name) = @_;
332
333 $REG{$name} = $portid;
334}
335
336=item rcv $portid, tagstring => $callback->(@msg), ...
337
338=item rcv $portid, $smartmatch => $callback->(@msg), ...
339
340=item rcv $portid, [$smartmatch...] => $callback->(@msg), ...
341
342Register callbacks to be called on matching messages on the given port.
343
344The callback has to return a true value when its work is done, after
345which is will be removed, or a false value in which case it will stay
346registered.
347
348The global C<$SELF> (exported by this module) contains C<$portid> while
349executing the callback.
350
351Runtime errors wdurign callback execution will result in the port being
352C<kil>ed.
353
354If the match is an array reference, then it will be matched against the
355first elements of the message, otherwise only the first element is being
356matched.
357
358Any element in the match that is specified as C<_any_> (a function
359exported by this module) matches any single element of the message.
360
361While not required, it is highly recommended that the first matching
362element is a string identifying the message. The one-string-only match is
363also the most efficient match (by far).
364
365=cut
366
154sub snd($@) { 367sub rcv($@) {
155 my ($noderef, $port) = split /#/, shift, 2; 368 my ($noderef, $port) = split /#/, shift, 2;
156 369
157 add_node $noderef 370 ($NODE{$noderef} || add_node $noderef) == $NODE{""}
158 unless exists $NODE{$noderef}; 371 or Carp::croak "$noderef#$port: rcv can only be called on local ports, caught";
159 372
160 $NODE{$noderef}->send ([$port, [@_]]); 373 my $self = $PORT_DATA{$port}
161} 374 or Carp::croak "$noderef#$port: rcv can only be called on message matching ports, caught";
162 375
163sub _inject { 376 "AnyEvent::MP::Port" eq ref $self
164 my ($port, $msg) = @{+shift}; 377 or Carp::croak "$noderef#$port: rcv can only be called on message matching ports, caught";
165 378
166 $port = $PORT{$port} 379 while (@_) {
167 or return; 380 my ($match, $cb) = splice @_, 0, 2;
168 381
169 use Data::Dumper; 382 if (!ref $match) {
170 warn Dumper $msg; 383 push @{ $self->{rc0}{$match} }, [$cb];
171} 384 } elsif (("ARRAY" eq ref $match && !ref $match->[0])) {
172 385 my ($type, @match) = @$match;
173sub normalise_noderef($) { 386 @match
174 my ($noderef) = @_; 387 ? push @{ $self->{rcv}{$match->[0]} }, [$cb, \@match]
175 388 : push @{ $self->{rc0}{$match->[0]} }, [$cb];
176 my $cv = AE::cv;
177 my @res;
178
179 $cv->begin (sub {
180 my %seen;
181 my @refs;
182 for (sort { $a->[0] <=> $b->[0] } @res) {
183 push @refs, $_->[1] unless $seen{$_->[1]}++
184 }
185 shift->send (join ",", @refs);
186 });
187
188 $noderef = $DEFAULT_PORT unless length $noderef;
189
190 my $idx;
191 for my $t (split /,/, $noderef) {
192 my $pri = ++$idx;
193
194 #TODO: this should be outside normalise_noderef and in become_public
195 if ($t =~ /^\d*$/) {
196 my $nodename = (POSIX::uname)[1];
197
198 $cv->begin;
199 AnyEvent::Socket::resolve_sockaddr $nodename, $t || "aemp=$DEFAULT_PORT", "tcp", 0, undef, sub {
200 for (@_) {
201 my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
202 push @res, [
203 $pri += 1e-5,
204 AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
205 ];
206 }
207 $cv->end;
208 };
209
210# my (undef, undef, undef, undef, @ipv4) = gethostbyname $nodename;
211#
212# for (@ipv4) {
213# push @res, [
214# $pri,
215# AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $_, $t || $DEFAULT_PORT,
216# ];
217# }
218 } else { 389 } else {
219 my ($host, $port) = AnyEvent::Socket::parse_hostport $t, "aemp=$DEFAULT_PORT" 390 push @{ $self->{any} }, [$cb, $match];
220 or Carp::croak "$t: unparsable transport descriptor";
221
222 $cv->begin;
223 AnyEvent::Socket::resolve_sockaddr $host, $port, "tcp", 0, undef, sub {
224 for (@_) {
225 my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
226 push @res, [
227 $pri += 1e-5,
228 AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
229 ];
230 }
231 $cv->end;
232 }
233 } 391 }
234 } 392 }
235
236 $cv->end;
237
238 $cv
239} 393}
240 394
241sub become_public { 395=item $closure = psub { BLOCK }
242 return if $PUBLIC;
243 396
244 my $noderef = join ",", ref $_[0] ? @{+shift} : shift; 397Remembers C<$SELF> and creates a closure out of the BLOCK. When the
245 my @args = @_; 398closure is executed, sets up the environment in the same way as in C<rcv>
399callbacks, i.e. runtime errors will cause the port to get C<kil>ed.
246 400
247 $NODE = (normalise_noderef $noderef)->recv; 401This is useful when you register callbacks from C<rcv> callbacks:
248 402
249 my $self = new AnyEvent::MP::Node::Self noderef => $NODE; 403 rcv delayed_reply => sub {
250 404 my ($delay, @reply) = @_;
251 $NODE{""} = $self; # empty string == local node 405 my $timer = AE::timer $delay, 0, psub {
252 406 snd @reply, $SELF;
253 for my $t (split /,/, $NODE) {
254 $NODE{$t} = $self;
255
256 my ($host, $port) = AnyEvent::Socket::parse_hostport $t;
257
258 $LISTENER{$t} = AnyEvent::MP::Transport::mp_server $host, $port,
259 @args,
260 on_error => sub {
261 die "on_error<@_>\n";#d#
262 },
263 on_connect => sub {
264 my ($tp) = @_;
265
266 $NODE{$tp->{remote_id}} = $_[0];
267 },
268 sub {
269 my ($tp) = @_;
270
271 $NODE{"$tp->{peerhost}:$tp->{peerport}"} = $tp;
272 },
273 ; 407 };
408 };
409
410=cut
411
412sub psub(&) {
413 my $cb = shift;
414
415 my $port = $SELF
416 or Carp::croak "psub can only be called from within rcv or psub callbacks, not";
417
418 sub {
419 local $SELF = $port;
420
421 if (wantarray) {
422 my @res = eval { &$cb };
423 _self_die if $@;
424 @res
425 } else {
426 my $res = eval { &$cb };
427 _self_die if $@;
428 $res
429 }
274 } 430 }
275
276 $PUBLIC = 1;
277} 431}
432
433=back
434
435=head1 FUNCTIONS FOR NODES
436
437=over 4
438
439=item become_public endpoint...
440
441Tells the node to become a public node, i.e. reachable from other nodes.
442
443If no arguments are given, or the first argument is C<undef>, then
444AnyEvent::MP tries to bind on port C<4040> on all IP addresses that the
445local nodename resolves to.
446
447Otherwise the first argument must be an array-reference with transport
448endpoints ("ip:port", "hostname:port") or port numbers (in which case the
449local nodename is used as hostname). The endpoints are all resolved and
450will become the node reference.
451
452=cut
453
454=back
455
456=head1 NODE MESSAGES
457
458Nodes understand the following messages sent to them. Many of them take
459arguments called C<@reply>, which will simply be used to compose a reply
460message - C<$reply[0]> is the port to reply to, C<$reply[1]> the type and
461the remaining arguments are simply the message data.
462
463=over 4
464
465=cut
466
467=item lookup => $name, @reply
468
469Replies with the port ID of the specified well-known port, or C<undef>.
470
471=item devnull => ...
472
473Generic data sink/CPU heat conversion.
474
475=item relay => $port, @msg
476
477Simply forwards the message to the given port.
478
479=item eval => $string[ @reply]
480
481Evaluates the given string. If C<@reply> is given, then a message of the
482form C<@reply, $@, @evalres> is sent.
483
484Example: crash another node.
485
486 snd $othernode, eval => "exit";
487
488=item time => @reply
489
490Replies the the current node time to C<@reply>.
491
492Example: tell the current node to send the current time to C<$myport> in a
493C<timereply> message.
494
495 snd $NODE, time => $myport, timereply => 1, 2;
496 # => snd $myport, timereply => 1, 2, <time>
278 497
279=back 498=back
280 499
281=head1 SEE ALSO 500=head1 SEE ALSO
282 501

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines