AnyEvent-MP/MP/Node.pm

=head1 NAME

AnyEvent::MP::Node - a single processing node (CPU/process...)

=head1 SYNOPSIS

   use AnyEvent::MP::Node;

=head1 DESCRIPTION

=cut

package AnyEvent::MP::Node;

use common::sense;

use AE ();
use AnyEvent::Util ();
use AnyEvent::Socket ();

use AnyEvent::MP::Transport ();

use base Exporter::;

our $VERSION = '0.0';

our $DEFAULT_PORT = "4040";

sub normalise_noderef($) {
   my ($noderef) = @_;

   my $cv = AE::cv;
   my @res;

   $cv->begin (sub {
      my %seen;
      my @refs;
      for (sort { $a->[0] <=> $b->[0] } @res) {
         push @refs, $_->[1] unless $seen{$_->[1]}++
      }
      shift->send (join ",", @refs);
   });

   $noderef = $DEFAULT_PORT unless length $noderef;

   my $idx;
   for my $t (split /,/, $noderef) {
      my $pri = ++$idx;
      
      #TODO: this should be outside normalise_noderef and in become_public
      if ($t =~ /^\d*$/) {
         require POSIX;
         my $nodename = (POSIX::uname ())[1];

         $cv->begin;
         AnyEvent::Socket::resolve_sockaddr $nodename, $t || "aemp=$DEFAULT_PORT", "tcp", 0, undef, sub {
            for (@_) {
               my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
               push @res, [
                  $pri += 1e-5,
                  AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
               ];
            }
            $cv->end;
         };

#         my (undef, undef, undef, undef, @ipv4) = gethostbyname $nodename;
#
#         for (@ipv4) {
#            push @res, [
#               $pri,
#               AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $_, $t || $DEFAULT_PORT,
#            ];
#         }
      } else {
         my ($host, $port) = AnyEvent::Socket::parse_hostport $t, "aemp=$DEFAULT_PORT"
            or Carp::croak "$t: unparsable transport descriptor";

         $cv->begin;
         AnyEvent::Socket::resolve_sockaddr $host, $port, "tcp", 0, undef, sub {
            for (@_) {
               my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
               push @res, [
                  $pri += 1e-5,
                  AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
               ];
            }
            $cv->end;
         }
      }
   }

   $cv->end;

   $cv
}

sub new {
   my ($class, $noderef) = @_;

   bless { noderef => $noderef }, $class
}

package AnyEvent::MP::Node::Direct;

use base "AnyEvent::MP::Node";

sub send {
   my ($self, $msg) = @_;

   if ($self->{transport}) {
      $self->{transport}->send ($msg);
   } elsif ($self->{queue}) {
      push @{ $self->{queue} }, $msg;
   } else {
      $self->{queue} = [$msg];
      $self->connect;
   }
}

sub kill {
   my ($self, $port, @reason) = @_;

   $self->send (["", kil => $port, @reason]);

#   delete $AnyEvent::MP::Base::PORT{$port};

#   my $mon = delete $AnyEvent::MP::Base::LMON{$port}
#      or return;

#   $_->(@reason) for values %$mon;
}

sub monitor {
   my ($self, $portid, $cb) = @_;

   return $cb->()
      if $self->{failed};

   my $list = $self->{lmon}{$portid} ||= [];

   $self->send (["", mon1 => $portid])
      unless @$list;

   push @$list, $cb;
}

sub unmonitor {
   my ($self, $portid, $cb) = @_;

   my $list = $self->{lmon}{$portid}
      or return;

   @$list = grep $_ != $cb, @$list;

   unless (@$list) {
      $self->send (["", mon0 => $portid]);
      delete $self->{monitor}{$portid};
   }
}

sub set_transport {
   my ($self, $transport) = @_;

   $self->clr_transport
      if $self->{transport};

   delete $self->{trial};
   delete $self->{next_connect};
   delete $self->{failed};

   if (
      exists $self->{remote_uniq}
      && $self->{remote_uniq} ne $transport->{remote_uniq}
   ) {
      # uniq changed, drop queue
      delete $self->{queue};
      #TODO: "DOWN"
   }

   $self->{remote_uniq} = $transport->{remote_uniq};
   $self->{transport}   = $transport;

   $transport->send ($_)
      for @{ delete $self->{queue} || [] };
}

sub clr_transport {
   my ($self) = @_;

   delete $self->{transport};
   $self->{failed} = 1;

   if (my $mon = delete $self->{monitor}) {
      $_->() for map @$_, values %$mon;
   }

   $self->connect;
}

sub connect {
   my ($self) = @_;

   Scalar::Util::weaken $self;

   unless (exists $self->{n_noderef}) {
      return if $self->{n_noderef_}++;
      (AnyEvent::MP::Node::normalise_noderef ($self->{noderef}))->cb (sub {
         $self or return;
         delete $self->{n_noderef_};
         my $noderef = shift->recv;

         $self->{n_noderef} = $noderef;

         $AnyEvent::MP::Base::NODE{$_} = $self
            for split /,/, $noderef;

         $self->connect;
      });
      return;
   }

   $self->{retry} ||= [split /,/, $self->{n_noderef}];

   my $endpoint = shift @{ $self->{retry} };

   if (defined $endpoint) {
      $self->{trial}{$endpoint} ||= do {
         my ($host, $port) = AnyEvent::Socket::parse_hostport $endpoint
            or return;

         my ($w, $g);

         $w = AE::timer $AnyEvent::MP::Base::CONNECT_TIMEOUT, 0, sub {
            delete $self->{trial}{$endpoint};
         };
         $g = AnyEvent::MP::Transport::mp_connect
            $host, $port,
            sub {
               delete $self->{trial}{$endpoint}
                  unless @_;
               $g = shift;
            };
         ;

         [$w, \$g]
      };
   } else {
      delete $self->{retry};
   }

   $self->{next_connect} = AE::timer $AnyEvent::MP::Base::CONNECT_INTERVAL, 0, sub {
      $self->connect;
   };
}

package AnyEvent::MP::Node::Self;

use base "AnyEvent::MP::Node";

sub set_transport {
   die "FATAL error, set_transport was called";
}

sub send {
   local $AnyEvent::MP::Base::SRCNODE = $_[0];
   AnyEvent::MP::Base::_inject (@{ $_[1] });
}

sub kill {
   my ($self, $port, @reason) = @_;

   delete $AnyEvent::MP::Base::PORT{$port};

   my $mon = delete $AnyEvent::MP::Base::LMON{$port}
      or return;

   $_->(@reason) for values %$mon;
}

sub monitor {
   my ($self, $portid, $cb) = @_;

   return $cb->()
      unless exists $AnyEvent::MP::Base::PORT{$portid};

   $AnyEvent::MP::Base::LMON{$portid}{$cb+0} = $cb;
}

sub unmonitor {
   my ($self, $portid, $cb) = @_;

   delete $AnyEvent::MP::Base::LMON{$portid}{$cb+0};
}

=head1 SEE ALSO

L<AnyEvent>.

=head1 AUTHOR

 Marc Lehmann <schmorp@schmorp.de>
 http://home.schmorp.de/

=cut

1

Revision:	1.7
Committed:	Tue Aug 4 14:10:51 2009 UTC (14 years, 9 months ago) by root
Branch:	MAIN
Changes since 1.6:	+24 -0 lines
Log Message:	* empty log message *
#	Content
1	=head1 NAME
2
3	AnyEvent::MP::Node - a single processing node (CPU/process...)
4
5	=head1 SYNOPSIS
6
7	use AnyEvent::MP::Node;
8
9	=head1 DESCRIPTION
10
11	=cut
12
13	package AnyEvent::MP::Node;
14
15	use common::sense;
16
17	use AE ();
18	use AnyEvent::Util ();
19	use AnyEvent::Socket ();
20
21	use AnyEvent::MP::Transport ();
22
23	use base Exporter::;
24
25	our $VERSION = '0.0';
26
27	our $DEFAULT_PORT = "4040";
28
29	sub normalise_noderef($) {
30	my ($noderef) = @_;
31
32	my $cv = AE::cv;
33	my @res;
34
35	$cv->begin (sub {
36	my %seen;
37	my @refs;
38	for (sort { $a->[0] <=> $b->[0] } @res) {
39	push @refs, $_->[1] unless $seen{$_->[1]}++
40	}
41	shift->send (join ",", @refs);
42	});
43
44	$noderef = $DEFAULT_PORT unless length $noderef;
45
46	my $idx;
47	for my $t (split /,/, $noderef) {
48	my $pri = ++$idx;
49
50	#TODO: this should be outside normalise_noderef and in become_public
51	if ($t =~ /^\d*$/) {
52	require POSIX;
53	my $nodename = (POSIX::uname ())[1];
54
55	$cv->begin;
56	AnyEvent::Socket::resolve_sockaddr $nodename, $t \|\| "aemp=$DEFAULT_PORT", "tcp", 0, undef, sub {
57	for (@_) {
58	my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
59	push @res, [
60	$pri += 1e-5,
61	AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
62	];
63	}
64	$cv->end;
65	};
66
67	# my (undef, undef, undef, undef, @ipv4) = gethostbyname $nodename;
68	#
69	# for (@ipv4) {
70	# push @res, [
71	# $pri,
72	# AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $_, $t \|\| $DEFAULT_PORT,
73	# ];
74	# }
75	} else {
76	my ($host, $port) = AnyEvent::Socket::parse_hostport $t, "aemp=$DEFAULT_PORT"
77	or Carp::croak "$t: unparsable transport descriptor";
78
79	$cv->begin;
80	AnyEvent::Socket::resolve_sockaddr $host, $port, "tcp", 0, undef, sub {
81	for (@_) {
82	my ($service, $host) = AnyEvent::Socket::unpack_sockaddr $_->[3];
83	push @res, [
84	$pri += 1e-5,
85	AnyEvent::Socket::format_hostport AnyEvent::Socket::format_address $host, $service
86	];
87	}
88	$cv->end;
89	}
90	}
91	}
92
93	$cv->end;
94
95	$cv
96	}
97
98	sub new {
99	my ($class, $noderef) = @_;
100
101	bless { noderef => $noderef }, $class
102	}
103
104	package AnyEvent::MP::Node::Direct;
105
106	use base "AnyEvent::MP::Node";
107
108	sub send {
109	my ($self, $msg) = @_;
110
111	if ($self->{transport}) {
112	$self->{transport}->send ($msg);
113	} elsif ($self->{queue}) {
114	push @{ $self->{queue} }, $msg;
115	} else {
116	$self->{queue} = [$msg];
117	$self->connect;
118	}
119	}
120
121	sub kill {
122	my ($self, $port, @reason) = @_;
123
124	$self->send (["", kil => $port, @reason]);
125
126	# delete $AnyEvent::MP::Base::PORT{$port};
127
128	# my $mon = delete $AnyEvent::MP::Base::LMON{$port}
129	# or return;
130
131	# $_->(@reason) for values %$mon;
132	}
133
134	sub monitor {
135	my ($self, $portid, $cb) = @_;
136
137	return $cb->()
138	if $self->{failed};
139
140	my $list = $self->{lmon}{$portid} \|\|= [];
141
142	$self->send (["", mon1 => $portid])
143	unless @$list;
144
145	push @$list, $cb;
146	}
147
148	sub unmonitor {
149	my ($self, $portid, $cb) = @_;
150
151	my $list = $self->{lmon}{$portid}
152	or return;
153
154	@$list = grep $_ != $cb, @$list;
155
156	unless (@$list) {
157	$self->send (["", mon0 => $portid]);
158	delete $self->{monitor}{$portid};
159	}
160	}
161
162	sub set_transport {
163	my ($self, $transport) = @_;
164
165	$self->clr_transport
166	if $self->{transport};
167
168	delete $self->{trial};
169	delete $self->{next_connect};
170	delete $self->{failed};
171
172	if (
173	exists $self->{remote_uniq}
174	&& $self->{remote_uniq} ne $transport->{remote_uniq}
175	) {
176	# uniq changed, drop queue
177	delete $self->{queue};
178	#TODO: "DOWN"
179	}
180
181	$self->{remote_uniq} = $transport->{remote_uniq};
182	$self->{transport} = $transport;
183
184	$transport->send ($_)
185	for @{ delete $self->{queue} \|\| [] };
186	}
187
188	sub clr_transport {
189	my ($self) = @_;
190
191	delete $self->{transport};
192	$self->{failed} = 1;
193
194	if (my $mon = delete $self->{monitor}) {
195	$_->() for map @$_, values %$mon;
196	}
197
198	$self->connect;
199	}
200
201	sub connect {
202	my ($self) = @_;
203
204	Scalar::Util::weaken $self;
205
206	unless (exists $self->{n_noderef}) {
207	return if $self->{n_noderef_}++;
208	(AnyEvent::MP::Node::normalise_noderef ($self->{noderef}))->cb (sub {
209	$self or return;
210	delete $self->{n_noderef_};
211	my $noderef = shift->recv;
212
213	$self->{n_noderef} = $noderef;
214
215	$AnyEvent::MP::Base::NODE{$_} = $self
216	for split /,/, $noderef;
217
218	$self->connect;
219	});
220	return;
221	}
222
223	$self->{retry} \|\|= [split /,/, $self->{n_noderef}];
224
225	my $endpoint = shift @{ $self->{retry} };
226
227	if (defined $endpoint) {
228	$self->{trial}{$endpoint} \|\|= do {
229	my ($host, $port) = AnyEvent::Socket::parse_hostport $endpoint
230	or return;
231
232	my ($w, $g);
233
234	$w = AE::timer $AnyEvent::MP::Base::CONNECT_TIMEOUT, 0, sub {
235	delete $self->{trial}{$endpoint};
236	};
237	$g = AnyEvent::MP::Transport::mp_connect
238	$host, $port,
239	sub {
240	delete $self->{trial}{$endpoint}
241	unless @_;
242	$g = shift;
243	};
244	;
245
246	[$w, \$g]
247	};
248	} else {
249	delete $self->{retry};
250	}
251
252	$self->{next_connect} = AE::timer $AnyEvent::MP::Base::CONNECT_INTERVAL, 0, sub {
253	$self->connect;
254	};
255	}
256
257	package AnyEvent::MP::Node::Self;
258
259	use base "AnyEvent::MP::Node";
260
261	sub set_transport {
262	die "FATAL error, set_transport was called";
263	}
264
265	sub send {
266	local $AnyEvent::MP::Base::SRCNODE = $_[0];
267	AnyEvent::MP::Base::_inject (@{ $_[1] });
268	}
269
270	sub kill {
271	my ($self, $port, @reason) = @_;
272
273	delete $AnyEvent::MP::Base::PORT{$port};
274
275	my $mon = delete $AnyEvent::MP::Base::LMON{$port}
276	or return;
277
278	$_->(@reason) for values %$mon;
279	}
280
281	sub monitor {
282	my ($self, $portid, $cb) = @_;
283
284	return $cb->()
285	unless exists $AnyEvent::MP::Base::PORT{$portid};
286
287	$AnyEvent::MP::Base::LMON{$portid}{$cb+0} = $cb;
288	}
289
290	sub unmonitor {
291	my ($self, $portid, $cb) = @_;
292
293	delete $AnyEvent::MP::Base::LMON{$portid}{$cb+0};
294	}
295
296	=head1 SEE ALSO
297
298	L<AnyEvent>.
299
300	=head1 AUTHOR
301
302	Marc Lehmann <schmorp@schmorp.de>
303	http://home.schmorp.de/
304
305	=cut
306
307	1
308