ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent-MP/MP/Node.pm
Revision: 1.18
Committed: Wed Aug 12 21:39:59 2009 UTC (14 years, 9 months ago) by root
Branch: MAIN
Changes since 1.17: +105 -96 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, $noderef) = @_;
25
26 $self = bless { noderef => $noderef }, $self;
27
28 $self->transport_reset;
29
30 $self
31 }
32
33 sub send {
34 &{$_[0]->{send}}
35 }
36
37 package AnyEvent::MP::Node::External;
38
39 use base "AnyEvent::MP::Node";
40
41 # called at init time, mostly sets {send}
42 sub transport_reset {
43 my ($self) = @_;
44
45 delete $self->{transport};
46
47 $self->{send} = sub {
48 push @{$_[0]{queue}}, $_[1];
49 $_[0]->connect;
50 };
51 }
52
53 # called only after successful handshake
54 sub transport_error {
55 my ($self, @reason) = @_;
56
57 delete $self->{queue};
58 $self->transport_reset;
59
60 if (my $mon = delete $self->{lmon}) {
61 $_->(@reason) for map @$_, values %$mon;
62 }
63 }
64
65 # called after handshake was successful
66 sub transport_connect {
67 my ($self, $transport) = @_;
68
69 $self->transport_reset
70 if $self->{transport};
71
72 delete $self->{connect_w};
73 delete $self->{connect_to};
74
75 $self->{transport} = $transport;
76
77 $self->{send} = sub {
78 $transport->send ($_[1]);
79 };
80
81 $transport->send ($_)
82 for @{ delete $self->{queue} || [] };
83 }
84
85 sub connect {
86 my ($self) = @_;
87
88 Scalar::Util::weaken $self;
89
90 $self->{connect_to} ||= AE::timer
91 $AnyEvent::MP::Config::CFG{monitor_timeout} || $AnyEvent::MP::Kernel::MONITOR_TIMEOUT,
92 0,
93 sub {
94 $self->transport_error (transport_error => $self->{noderef}, "unable to connect");
95 };
96
97 unless ($self->{connect_w}) {
98 my @endpoints;
99 my %trial;
100
101 $self->{connect_w} = AE::timer
102 0,
103 $AnyEvent::MP::Config::CFG{connect_interval} || $AnyEvent::MP::Kernel::CONNECT_INTERVAL,
104 sub {
105 @endpoints = split /,/, $self->{noderef}
106 unless @endpoints;
107
108 my $endpoint = shift @endpoints;
109
110 $trial{$endpoint} ||= do {
111 my ($host, $port) = AnyEvent::Socket::parse_hostport $endpoint
112 or return $AnyEvent::MP::Kernel::WARN->("$self->{noderef}: not a resolved node reference.");
113
114 AnyEvent::MP::Transport::mp_connect
115 $host, $port,
116 sub { delete $trial{$endpoint} }
117 ;
118 };
119 }
120 ;
121 }
122 }
123
124 sub kill {
125 my ($self, $port, @reason) = @_;
126
127 $self->send (["", kil => $port, @reason]);
128 }
129
130 sub monitor {
131 my ($self, $portid, $cb) = @_;
132
133 my $list = $self->{lmon}{$portid} ||= [];
134
135 $self->send (["", mon1 => $portid])
136 unless @$list;
137
138 push @$list, $cb;
139 }
140
141 sub unmonitor {
142 my ($self, $portid, $cb) = @_;
143
144 my $list = $self->{lmon}{$portid}
145 or return;
146
147 @$list = grep $_ != $cb, @$list;
148
149 unless (@$list) {
150 $self->send (["", mon0 => $portid]);
151 delete $self->{monitor}{$portid};
152 }
153 }
154
155 package AnyEvent::MP::Node::Direct;
156
157 use base "AnyEvent::MP::Node::External";
158
159 package AnyEvent::MP::Node::Indirect;
160
161 use base "AnyEvent::MP::Node::Direct";
162
163 sub connect {
164 my ($self) = @_;
165
166 $self->transport_error (transport_error => $self->{noderef}, "unable to connect to indirect node");
167 }
168
169 package AnyEvent::MP::Node::Self;
170
171 use base "AnyEvent::MP::Node";
172
173 sub transport_reset {
174 my ($self) = @_;
175
176 $self->{send} = sub {
177 local $AnyEvent::MP::Kernel::SRCNODE = $_[0];
178 AnyEvent::MP::Kernel::_inject (@{ $_[1] });
179 };
180 }
181
182 sub kill {
183 my ($self, $port, @reason) = @_;
184
185 delete $AnyEvent::MP::Kernel::PORT{$port};
186 delete $AnyEvent::MP::Kernel::PORT_DATA{$port};
187
188 my $mon = delete $AnyEvent::MP::Kernel::LMON{$port}
189 or !@reason
190 or $AnyEvent::MP::Kernel::WARN->("unmonitored local port $port died with reason: @reason");
191
192 $_->(@reason) for values %$mon;
193 }
194
195 sub monitor {
196 my ($self, $portid, $cb) = @_;
197
198 return $cb->(no_such_port => "cannot monitor nonexistent port")
199 unless exists $AnyEvent::MP::Kernel::PORT{$portid};
200
201 $AnyEvent::MP::Kernel::LMON{$portid}{$cb+0} = $cb;
202 }
203
204 sub unmonitor {
205 my ($self, $portid, $cb) = @_;
206
207 delete $AnyEvent::MP::Kernel::LMON{$portid}{$cb+0};
208 }
209
210 =head1 SEE ALSO
211
212 L<AnyEvent::MP>.
213
214 =head1 AUTHOR
215
216 Marc Lehmann <schmorp@schmorp.de>
217 http://home.schmorp.de/
218
219 =cut
220
221 1
222