/[meteor]/googlecode.com/svn/trunk/Meteor/Subscriber.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Annotation of /googlecode.com/svn/trunk/Meteor/Subscriber.pm

Parent Directory Parent Directory | Revision Log Revision Log


Revision 45 - (hide annotations)
Mon Feb 4 19:02:49 2008 UTC (16 years, 1 month ago) by knops.gerd
File size: 9264 byte(s)
• Mode-specific configurations
• New ~channelinfo~ header keyword
• New ChannelInfoTemplate parameter

1 knops.gerd 11 #!/usr/bin/perl -w
2     ###############################################################################
3     # Meteor
4     # An HTTP server for the 2.0 web
5     # Copyright (c) 2006 contributing authors
6     #
7     # Subscriber.pm
8     #
9     # Description:
10     # A Meteor Subscriber
11     #
12     ###############################################################################
13     #
14     # This program is free software; you can redistribute it and/or modify it
15     # under the terms of the GNU General Public License as published by the Free
16     # Software Foundation; either version 2 of the License, or (at your option)
17     # any later version.
18     #
19     # This program is distributed in the hope that it will be useful, but WITHOUT
20     # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
21     # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
22     # more details.
23     #
24     # You should have received a copy of the GNU General Public License along
25     # with this program; if not, write to the Free Software Foundation, Inc.,
26     # 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
27     #
28     # For more information visit www.meteorserver.org
29     #
30     ###############################################################################
31    
32     package Meteor::Subscriber;
33     ###############################################################################
34     # Configuration
35     ###############################################################################
36    
37     use strict;
38    
39     use Meteor::Connection;
40     use Meteor::Channel;
41     use Meteor::Document;
42    
43     @Meteor::Subscriber::ISA=qw(Meteor::Connection);
44    
45     our %PersistentConnections=();
46 knops.gerd 25 our $NumAcceptedConnections=0;
47 knops.gerd 37
48 knops.gerd 11
49     ###############################################################################
50     # Factory methods
51     ###############################################################################
52     sub newFromServer {
53     my $class=shift;
54    
55     my $self=$class->SUPER::newFromServer(shift);
56    
57     $self->{'headerBuffer'}='';
58     $self->{'MessageCount'}=0;
59     $self->{'MaxMessageCount'}=0;
60    
61     $self->{'ConnectionStart'}=time;
62     my $maxTime=$::CONF{'MaxTime'};
63     if($maxTime>0)
64     {
65     $self->{'ConnectionTimeLimit'}=$self->{'ConnectionStart'}+$maxTime;
66     }
67    
68 knops.gerd 25 $::Statistics->{'current_subscribers'}++;
69     $::Statistics->{'subscriber_connections_accepted'}++;
70    
71 knops.gerd 11 $self;
72     }
73    
74     ###############################################################################
75     # Class methods
76     ###############################################################################
77     sub deleteSubscriberWithID {
78     my $class=shift;
79     my $id=shift;
80    
81     if(exists($PersistentConnections{$id}))
82     {
83 knops.gerd 44 $PersistentConnections{$id}->close();
84 knops.gerd 11 }
85     }
86    
87     sub pingPersistentConnections {
88     my $class=shift;
89    
90     my @cons=values %PersistentConnections;
91    
92 knops.gerd 45 map { $_->ping() } @cons;
93 knops.gerd 11 }
94    
95     sub checkPersistentConnectionsForMaxTime {
96     my $class=shift;
97    
98     my $time=time;
99     my @cons=values %PersistentConnections;
100    
101     map { $_->checkForMaxTime($time) } @cons;
102     }
103    
104 knops.gerd 25 sub numSubscribers {
105    
106     return scalar(keys %PersistentConnections);
107     }
108    
109 knops.gerd 11 ###############################################################################
110     # Instance methods
111     ###############################################################################
112     sub processLine {
113     my $self=shift;
114     my $line=shift;
115    
116     # Once the header was processed we ignore any input
117     return unless(exists($self->{'headerBuffer'}));
118    
119     if($line ne '')
120     {
121     #
122     # Accumulate header
123     #
124     $self->{'headerBuffer'}.="$line\n";
125     }
126     else
127     {
128     #
129     # Empty line signals end of header.
130     # Analyze header, register with appropiate channel
131     # and send pending messages.
132     #
133 andrew.betts 32 # GET $::CONF{'SubscriberDynamicPageAddress'}/hostid/streamtype/channeldefs HTTP/1.1
134 knops.gerd 11 #
135     # Find the 'GET' line
136     #
137 andrew.betts 32 if($self->{'headerBuffer'}=~/GET\s+$::CONF{'SubscriberDynamicPageAddress'}\/([0-9a-z]+)\/([0-9a-z]+)\/(\S+)/i)
138 knops.gerd 11 {
139 andrew.betts 32 my $subscriberID=$1;
140     $self->{'mode'}=$2;
141 knops.gerd 45 my $persist=$self->getConf('Persist');
142    
143 andrew.betts 32 if ($self->{'mode'} eq "xhrinteractive" || $self->{'mode'} eq "iframe" || $self->{'mode'} eq "serversent" || $self->{'mode'} eq "longpoll") {
144     $persist=1;
145     $self->{'MaxMessageCount'}=1 unless(!($self->{'mode'} eq "longpoll"));
146     }
147     if ($self->{'mode'} eq "iframe") {
148     $self->{'HeaderTemplateNumber'}=1;
149     } else {
150     $self->{'HeaderTemplateNumber'}=2;
151     }
152 knops.gerd 45
153     my $maxTime=$self->getConf('MaxTime');
154     if($maxTime>0)
155     {
156     $self->{'ConnectionTimeLimit'}=$self->{'ConnectionStart'}+$maxTime;
157     }
158    
159 andrew.betts 32 my @channelData=split('/',$3);
160 knops.gerd 13 my $channels={};
161 andrew.betts 32 my $channelName;
162     my $offset;
163 andrew.betts 35 foreach my $chandef (@channelData) {
164     if($chandef=~/^([a-z0-9]+)(.(r|b|h)([0-9]*))?$/i) {
165 andrew.betts 32 $channelName = $1;
166     $channels->{$channelName}->{'startIndex'} = undef;
167 andrew.betts 35 if ($3) {
168     $offset = $4;
169     if ($3 eq 'r') { $channels->{$channelName}->{'startIndex'} = $offset; }
170     if ($3 eq 'b') { $channels->{$channelName}->{'startIndex'} = -$offset; }
171     if ($3 eq 'h') { $channels->{$channelName}->{'startIndex'} = 0; }
172 knops.gerd 13 }
173 knops.gerd 11 }
174     }
175    
176 knops.gerd 13 delete($self->{'headerBuffer'});
177    
178 andrew.betts 32 if($persist)
179 knops.gerd 11 {
180     $self->{'subscriberID'}=$subscriberID;
181     $self->deleteSubscriberWithID($subscriberID);
182     $PersistentConnections{$subscriberID}=$self;
183     }
184    
185 knops.gerd 13 if(scalar(keys %{$channels}))
186 knops.gerd 11 {
187     $self->emitOKHeader();
188 knops.gerd 17 $self->setChannels($channels,$persist);
189     $self->close(1) unless($persist);
190 knops.gerd 11 return;
191     }
192     }
193 andrew.betts 32 elsif($self->{'headerBuffer'}=~/GET\s+\/disconnect\/(\S+)/)
194     {
195     $self->deleteSubscriberWithID($1);
196     $self->emitOKHeader();
197     $self->close(1);
198     return;
199     }
200 knops.gerd 11 elsif($self->{'headerBuffer'}=~/GET\s+([^\s\?]+)/)
201     {
202     Meteor::Document->serveFileToClient($1,$self);
203     $self->close(1);
204     return;
205     }
206    
207     #
208     # If we fall through we did not understand the request
209     #
210     $self->emitErrorHeader();
211     }
212     }
213    
214 knops.gerd 13 sub setChannels {
215 knops.gerd 11 my $self=shift;
216 knops.gerd 13 my $channels=shift;
217 knops.gerd 17 my $persist=shift;
218 knops.gerd 11
219 knops.gerd 13 foreach my $channelName (keys %{$channels})
220     {
221     my $startIndex=$channels->{$channelName}->{'startIndex'};
222    
223     my $channel=Meteor::Channel->channelWithName($channelName);
224    
225     $self->{'channels'}->{$channelName}=$channel if($persist);
226    
227     $channel->addSubscriber($self,$startIndex,$persist);
228     }
229 knops.gerd 11 }
230    
231     sub emitOKHeader {
232     my $self=shift;
233    
234     $self->emitHeader('200 OK');
235     }
236    
237     sub emitErrorHeader {
238     my $self=shift;
239    
240     $self->emitHeader('404 Not Found');
241 knops.gerd 25 $::Statistics->{'errors_served'}++;
242 knops.gerd 11
243     # close up shop here!
244     $self->close();
245     }
246    
247     sub emitHeader {
248     my $self=shift;
249     my $status=shift;
250    
251     my $header=undef;
252     if(exists($self->{'HeaderTemplateNumber'}))
253     {
254     my $hn='HeaderTemplate'.$self->{'HeaderTemplateNumber'};
255    
256 knops.gerd 45 $header=$self->getConf($hn);
257 knops.gerd 11 }
258 knops.gerd 45 $header=$self->getConf('HeaderTemplate') unless(defined($header));
259 knops.gerd 11
260 knops.gerd 45 $header=~s/~([^~]*)~/
261 knops.gerd 11 if(!defined($1) || $1 eq '')
262     {
263     '~';
264     }
265     elsif($1 eq 'server')
266     {
267     $::PGM;
268     }
269     elsif($1 eq 'status')
270     {
271     $status;
272     }
273     elsif($1 eq 'servertime')
274     {
275     time;
276     }
277 knops.gerd 45 elsif($1 eq 'channelinfo')
278     {
279     Meteor::Channel->listChannelsUsingTemplate($self->getConf('ChannelInfoTemplate'));
280     }
281 knops.gerd 11 else
282     {
283     '';
284     }
285     /gex;
286    
287 andrew.betts 39 $self->write($header);
288 knops.gerd 11 }
289    
290 knops.gerd 45 sub sendMessages {
291 knops.gerd 11 my $self=shift;
292    
293 knops.gerd 45 my $numMessages=0;
294     my $msgTemplate=$self->getConf('Messagetemplate');
295     my $msgData='';
296 knops.gerd 25
297 knops.gerd 45 foreach my $message (@_)
298     {
299     $msgData.=$message->messageWithTemplate($msgTemplate);
300     $numMessages++;
301     }
302 knops.gerd 11
303 knops.gerd 45 return if($numMessages<1);
304 knops.gerd 25
305 knops.gerd 45 $self->write($msgData);
306 knops.gerd 11
307 knops.gerd 45 $::Statistics->{'messages_served'}+=$numMessages;
308    
309     my $msgCount=$self->{'MessageCount'};
310     $msgCount+=$numMessages;
311     $self->{'MessageCount'}=$msgCount;
312    
313     my $maxMsg=$self->getConf('MaxMessages');
314 knops.gerd 11 if(defined($maxMsg) && $maxMsg>0 && $msgCount>=$maxMsg)
315     {
316     $self->close(1);
317     }
318    
319     if($self->{'MaxMessageCount'}>0 && $msgCount>=$self->{'MaxMessageCount'})
320     {
321     $self->close(1);
322     }
323 knops.gerd 45
324 knops.gerd 11 }
325    
326 knops.gerd 45 sub ping {
327     my $self=shift;
328     my $msg=$self->getConf('PingMessage');
329    
330     $self->write($msg);
331     }
332    
333 knops.gerd 13 sub closeChannel {
334     my $self=shift;
335     my $channelName=shift;
336    
337     return unless(exists($self->{'channels'}->{$channelName}));
338    
339     my $channel=$self->{'channels'}->{$channelName};
340     $channel->removeSubscriber($self);
341    
342     delete($self->{'channels'}->{$channelName});
343    
344     $self->close() if(scalar(keys %{$self->{'channels'}})==0);
345     }
346    
347 knops.gerd 11 sub close {
348     my $self=shift;
349     my $noShutdownMsg=shift;
350    
351 knops.gerd 13 foreach my $channelName (keys %{$self->{'channels'}})
352     {
353     my $channel=$self->{'channels'}->{$channelName};
354     $channel->removeSubscriber($self);
355     }
356     delete($self->{'channels'});
357 knops.gerd 11
358     if(exists($self->{'subscriberID'}))
359     {
360     delete($PersistentConnections{$self->{'subscriberID'}});
361     }
362    
363     #
364     # Send shutdown message unless remote closed or
365     # connection not yet established
366     #
367     unless($noShutdownMsg || $self->{'remoteClosed'} || exists($self->{'headerBuffer'}))
368     {
369 knops.gerd 45 my $msg=$self->getConf('SubscriberShutdownMsg');
370 knops.gerd 11 if(defined($msg) && $msg ne '')
371     {
372     $self->write($msg);
373     }
374     }
375    
376     $self->SUPER::close();
377     }
378    
379 knops.gerd 37 sub didClose {
380    
381     $::Statistics->{'current_subscribers'}--;
382     }
383    
384 knops.gerd 11 sub checkForMaxTime {
385     my $self=shift;
386     my $time=shift;
387    
388     $self->close(1) if(exists($self->{'ConnectionTimeLimit'}) && $self->{'ConnectionTimeLimit'}<$time);
389     }
390    
391 knops.gerd 45 sub getConf {
392     my $self=shift;
393     my $key=shift;
394    
395     if(exists($self->{'mode'}) && $self->{'mode'} ne '')
396     {
397     my $k=$key.'.'.$self->{'mode'};
398    
399     return $::CONF{$k} if(exists($::CONF{$k}));
400     }
401    
402     $::CONF{$key};
403     }
404    
405 knops.gerd 11 1;
406 andrew.betts 3 ############################################################################EOF

  ViewVC Help
Powered by ViewVC 1.1.26