ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent-MP/MP/Node.pm
Revision: 1.30
Committed: Fri Aug 28 20:57:42 2009 UTC (14 years, 9 months ago) by root
Branch: MAIN
Changes since 1.29: +3 -6 lines
Log Message:
*** empty log message ***

File Contents

# 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 sub new {
24 my ($self, $id) = @_;
25
26 $self = bless { id => $id }, $self;
27
28 $self->init;
29 $self->transport_reset;
30
31 $self
32 }
33
34 sub init {
35 #
36 }
37
38 sub send {
39 &{ shift->{send} }
40 }
41
42 # nodes reachable via the network
43 package AnyEvent::MP::Node::External;
44
45 use base "AnyEvent::MP::Node";
46
47 # called at init time, mostly sets {send}
48 sub transport_reset {
49 my ($self) = @_;
50
51 delete $self->{transport};
52
53 Scalar::Util::weaken $self;
54
55 $self->{send} = sub {
56 push @{$self->{queue}}, shift;
57 $self->connect;
58 };
59 }
60
61 # called only after successful handshake
62 sub transport_error {
63 my ($self, @reason) = @_;
64
65 my $no_transport = !$self->{transport};
66
67 delete $self->{connect_w};
68 delete $self->{connect_to};
69
70 delete $self->{queue};
71 $self->transport_reset;
72
73 if (my $mon = delete $self->{lmon}) {
74 $_->(@reason) for map @$_, values %$mon;
75 }
76
77 AnyEvent::MP::Kernel::_inject_nodeevent ($self, 0, @reason)
78 unless $no_transport;
79 }
80
81 # called after handshake was successful
82 sub transport_connect {
83 my ($self, $transport) = @_;
84
85 delete $self->{trial};
86
87 $self->transport_error (transport_error => "switched connections")
88 if $self->{transport};
89
90 delete $self->{connect_w};
91 delete $self->{connect_to};
92
93 $self->{transport} = $transport;
94
95 my $transport_send = $transport->can ("send");
96
97 $self->{send} = sub {
98 $transport_send->($transport, $_[0]);
99 };
100
101 AnyEvent::MP::Kernel::_inject_nodeevent ($self, 1);
102
103 $transport->send ($_)
104 for @{ delete $self->{queue} || [] };
105 }
106
107 sub connect {
108 my ($self, @addresses) = @_;
109
110 return if $self->{transport};
111
112 Scalar::Util::weaken $self;
113
114 $self->{connect_to} ||= AE::timer
115 $AnyEvent::MP::Config::CFG{monitor_timeout} || $AnyEvent::MP::Kernel::MONITOR_TIMEOUT,
116 0,
117 sub {
118 $self->transport_error (transport_error => $self->{id}, "unable to connect");
119 };
120
121 return unless @addresses;
122
123 $AnyEvent::MP::Kernel::WARN->(9, "connecting to $self->{id} with [@addresses]");
124
125 unless ($self->{connect_w}) {
126 my @endpoints;
127
128 $self->{connect_w} = AE::timer
129 rand,
130 $AnyEvent::MP::Config::CFG{connect_interval} || $AnyEvent::MP::Kernel::CONNECT_INTERVAL,
131 sub {
132 @endpoints = @addresses
133 unless @endpoints;
134
135 my $endpoint = shift @endpoints;
136
137 $AnyEvent::MP::Kernel::WARN->(9, "connecting to $self->{id} at $endpoint");
138
139 $self->{trial}{$endpoint} ||= do {
140 my ($host, $port) = AnyEvent::Socket::parse_hostport $endpoint
141 or return $AnyEvent::MP::Kernel::WARN->(1, "$self->{id}: not a resolved node reference.");
142
143 AnyEvent::MP::Transport::mp_connect
144 $host, $port,
145 sub { delete $self->{trial}{$endpoint} },
146 };
147 }
148 ;
149 }
150 }
151
152 sub kill {
153 my ($self, $port, @reason) = @_;
154
155 $self->send (["", kil => $port, @reason]);
156 }
157
158 sub monitor {
159 my ($self, $portid, $cb) = @_;
160
161 my $list = $self->{lmon}{$portid} ||= [];
162
163 $self->send (["", mon1 => $portid])
164 unless @$list || !length $portid;
165
166 push @$list, $cb;
167 }
168
169 sub unmonitor {
170 my ($self, $portid, $cb) = @_;
171
172 my $list = $self->{lmon}{$portid}
173 or return;
174
175 @$list = grep $_ != $cb, @$list;
176
177 unless (@$list) {
178 $self->send (["", mon0 => $portid]);
179 delete $self->{monitor}{$portid};
180 }
181 }
182
183 # used for direct slave connections as well
184 package AnyEvent::MP::Node::Direct;
185
186 use base "AnyEvent::MP::Node::External";
187
188 package AnyEvent::MP::Node::Self;
189
190 use base "AnyEvent::MP::Node";
191
192 sub connect {
193 # we are trivially connected
194 }
195
196 sub transport_reset {
197 my ($self) = @_;
198
199 Scalar::Util::weaken $self;
200
201 $self->{send} = sub {
202 local $AnyEvent::MP::Kernel::SRCNODE = $self;
203 AnyEvent::MP::Kernel::_inject (@{ $_[0] });
204 };
205 }
206
207 sub transport_connect {
208 my ($self, $tp) = @_;
209
210 $AnyEvent::MP::Kernel::WARN->(9, "I refuse to talk to myself ($tp->{peerhost}:$tp->{peerport})");
211 }
212
213 sub kill {
214 my ($self, $port, @reason) = @_;
215
216 delete $AnyEvent::MP::Kernel::PORT{$port};
217 delete $AnyEvent::MP::Kernel::PORT_DATA{$port};
218
219 my $mon = delete $AnyEvent::MP::Kernel::LMON{$port}
220 or !@reason
221 or $AnyEvent::MP::Kernel::WARN->(2, "unmonitored local port $port died with reason: @reason");
222
223 $_->(@reason) for values %$mon;
224 }
225
226 sub monitor {
227 my ($self, $portid, $cb) = @_;
228
229 return $cb->(no_such_port => "cannot monitor nonexistent port")
230 unless exists $AnyEvent::MP::Kernel::PORT{$portid};
231
232 $AnyEvent::MP::Kernel::LMON{$portid}{$cb+0} = $cb;
233 }
234
235 sub unmonitor {
236 my ($self, $portid, $cb) = @_;
237
238 delete $AnyEvent::MP::Kernel::LMON{$portid}{$cb+0};
239 }
240
241 =head1 SEE ALSO
242
243 L<AnyEvent::MP>.
244
245 =head1 AUTHOR
246
247 Marc Lehmann <schmorp@schmorp.de>
248 http://home.schmorp.de/
249
250 =cut
251
252 1
253