/[meteor]/googlecode.com/svn/trunk/Meteor/Channel.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

Diff of /googlecode.com/svn/trunk/Meteor/Channel.pm

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 9 by andrew.betts, Fri Dec 8 16:52:58 2006 UTC revision 58 by andrew.betts, Thu Mar 6 08:29:39 2008 UTC
# Line 1  Line 1 
1  #!/usr/bin/perl -w  #!/usr/bin/perl -w
2  ###############################################################################  ###############################################################################
3  #   Meteor  #   Meteor
4  #   An HTTP server for the 2.0 web  #   An HTTP server for the 2.0 web
5  #   Copyright (c) 2006 contributing authors  #   Copyright (c) 2006 contributing authors
6  #  #
7  #   Subscriber.pm  #   Subscriber.pm
8  #  #
9  #       Description:  #       Description:
10  #       A Meteor Channel  #       A Meteor Channel
11  #  #
12  ###############################################################################  ###############################################################################
13  #  #
14  #   This program is free software; you can redistribute it and/or modify it  #   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  #   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)  #   Software Foundation; either version 2 of the License, or (at your option)
17  #   any later version.  #   any later version.
18  #  #
19  #   This program is distributed in the hope that it will be useful, but WITHOUT  #   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  #   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
21  #   FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for  #   FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
22  #   more details.  #   more details.
23  #  #
24  #   You should have received a copy of the GNU General Public License along  #   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.,  #   with this program; if not, write to the Free Software Foundation, Inc.,
26  #   59 Temple Place, Suite 330, Boston, MA 02111-1307 USA  #   59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
27  #  #
28  #   For more information visit www.meteorserver.org  #   For more information visit www.meteorserver.org
29  #  #
30  ###############################################################################  ###############################################################################
31    
32  package Meteor::Channel;  package Meteor::Channel;
33  ###############################################################################  ###############################################################################
34  # Configuration  # Configuration
35  ###############################################################################  ###############################################################################
36                    
37          use strict;          use strict;
38                    
39          use Meteor::Message;          use Meteor::Message;
40                    
41          our %Channels=();          our %Channels=();
42          our $MessageID=0;          our $MessageID=0;
43    
44  ###############################################################################  ###############################################################################
45  # Class methods  # Class methods
46  ###############################################################################  ###############################################################################
47  sub channelWithName {  sub channelWithName {
48          my $class=shift;          my $class=shift;
49          my $channelName=shift;          my $channelName=shift;
50          my $avoidCreation=shift;          my $avoidCreation=shift;
51                    
52          unless(exists($Channels{$channelName}))          unless(exists($Channels{$channelName}))
53          {          {
54                  return undef if($avoidCreation);                  return undef if($avoidCreation);
55                  #                  #
56                  # Create new channel                  # Create new channel
57                  #                  #
58                  $Channels{$channelName}=$class->newChannel($channelName);                  $Channels{$channelName}=$class->newChannel($channelName);
59                                    
60                  &::syslog('debug',"New channel $channelName");                  &::syslog('debug',"New channel $channelName");
61          }          }
62                    
63          return $Channels{$channelName};          return $Channels{$channelName};
64  }  }
65    
66  sub listChannels {  sub listChannels {
67          my $class=shift;          my $class=shift;
68                    
69          my $list='';          my $list='';
70          foreach my $channelName (sort keys %Channels)          foreach my $channelName (sort keys %Channels)
71          {          {
72                  my $channel=$Channels{$channelName};                  my $channel=$Channels{$channelName};
73                                    
74                  $list.=$channelName.'('.$channel->messageCount().'/'.$channel->subscriberCount().")$::CRLF";                  $list.=$channelName.'('.$channel->messageCount().'/'.$channel->subscriberCount().")$::CRLF";
75          }          }
76                    
77          $list;          $list;
78  }  }
79    
80  sub deleteChannel {  sub deleteChannel {
81          my $class=shift;          my $class=shift;
82          my $channelName=shift;          my $channelName=shift;
83                    
84          delete($Channels{$channelName});          delete($Channels{$channelName});
85  }  }
86    
87  sub trimMessageStoresByTimestamp {  sub trimMessageStoresByTimestamp {
88          my $class=shift;          my $class=shift;
89          my $minTimeStamp=shift;          my $minTimeStamp=shift;
90                    
91          return unless($minTimeStamp);          return unless($minTimeStamp);
92                    
93          map { $_->trimMessageStoreByTimestamp($minTimeStamp) } (values %Channels);          map { $_->trimMessageStoreByTimestamp($minTimeStamp) } (values %Channels);
94  }  }
95    
96  sub clearAllBuffers {  sub clearAllBuffers {
97          my $class=shift;          my $class=shift;
98                    
99          map { $_->clearBuffer() } (values %Channels);          map { $_->clearBuffer() } (values %Channels);
100  }  }
101    
102  ###############################################################################  sub numChannels {
103  # Factory methods          
104  ###############################################################################          return scalar(keys %Channels);
105  sub new {  }
106          #  
107          # Create a new empty instance  ###############################################################################
108          #  # Factory methods
109          my $class=shift;  ###############################################################################
110            sub new {
111          my $obj={};          #
112                    # Create a new empty instance
113          bless($obj,$class);          #
114  }          my $class=shift;
115                    
116  sub newChannel {          my $obj={};
117          #          
118          # new instance from new server connection          bless($obj,$class);
119          #  }
120          my $self=shift->new();          
121            sub newChannel {
122          my $name=shift;          #
123          $self->{'name'}=$name;          # new instance from new server connection
124                    #
125          $self->{'subscribers'}=[];          my $self=shift->new();
126          $self->{'messages'}=[];          
127                    my $name=shift;
128          $self;          $self->{'name'}=$name;
129  }          
130            $self->{'subscribers'}=[];
131  sub DESTROY {          $self->{'messages'}=[];
132          my $self=shift;          
133                    $self;
134          my @subscribers=@{$self->{'subscribers'}};  }
135          map { $_->close() } @subscribers;  
136  }  sub DESTROY {
137            my $self=shift;
138  ###############################################################################          
139  # Instance methods          my @subscribers=@{$self->{'subscribers'}};
140  ###############################################################################          map { $_->closeChannel($self->{'name'}) } @subscribers;
141  sub name {  }
142          shift->{'name'};  
143  }  ###############################################################################
144    # Instance methods
145  sub addSubscriber {  ###############################################################################
146          my $self=shift;  sub name {
147          my $subscriber=shift;          shift->{'name'};
148          my $startId=shift;  }
149          my $persist=shift;  
150            sub addSubscriber {
151          # Note: negative $startId means go back that many messages          my $self=shift;
152                    my $subscriber=shift;
153          push(@{$self->{'subscribers'}},$subscriber) if($persist);          my $startId=shift;
154                    my $persist=shift;
155          my $startIndex=$self->indexForMessageID($startId);          my $mode=shift || '';
156          return unless(defined($startIndex));          my $userAgent=shift || '';
157                    
158          my $msgCount=scalar(@{$self->{'messages'}});          # Note: negative $startId means go back that many messages
159          my $txt='';          my $startIndex=$self->indexForMessageID($startId);
160                    my $logStartIndex = $startIndex || $self->lastMsgID() || 0;
161          $startIndex=0 if($startIndex<0);          
162                    push(@{$self->{'subscribers'}},$subscriber) if($persist);
163          while($startIndex<$msgCount)          
164          {          &::syslog('info','',
165                  my $message=$self->{'messages'}->[$startIndex++];                  'joinchannel',
166                                    $subscriber->{'ip'},
167                  $txt.=$message->message();                  $subscriber->{'subscriberID'},
168          }                  $self->{'name'},
169                            $mode,
170          $subscriber->sendMessage($txt);                  $logStartIndex,
171  }                  $userAgent
172            );
173  sub removeSubscriber {          
174          my $self=shift;          return unless(defined($startIndex));
175          my $subscriber=shift;          
176                    my $msgCount=scalar(@{$self->{'messages'}});
177          my $idx=undef;          my $txt='';
178          for(my $i=0;$i<scalar(@{$self->{'subscribers'}});$i++)          
179          {          $startIndex=0 if($startIndex<0);
180                  if($self->{'subscribers'}->[$i]==$subscriber)          
181                  {          if($startIndex<$msgCount) {
182                          $idx=$i;                  $subscriber->sendMessages(@{$self->{'messages'}}[$startIndex..$msgCount-1]);
183                          last;          }
184                  }  }
185          }  
186            sub removeSubscriber {
187          if(defined($idx))          my $self=shift;
188          {          my $subscriber=shift;
189                  splice(@{$self->{'subscribers'}},$idx,1);          my $reason=shift ||'unknown';
190          }          
191                    my $idx=undef;
192          $self->checkExpiration();          my $numsubs = scalar(@{$self->{'subscribers'}});
193  }  
194            for (my $i=0; $i<$numsubs; $i++) {
195  sub subscriberCount {                  if($self->{'subscribers'}->[$i]==$subscriber) {
196          my $self=shift;                          $idx=$i;
197                                    last;
198          scalar(@{$self->{'subscribers'}});                  }
199  }          }
200            
201  sub addMessage {          if(defined($idx))
202          my $self=shift;          {
203          my $messageText=shift;                  splice(@{$self->{'subscribers'}},$idx,1);
204                            
205          my $message=Meteor::Message->newWithID($MessageID++);                  my $timeConnected = time - $subscriber->{'ConnectionStart'};
206          $message->setText($messageText);                          &::syslog('info','',
207          push(@{$self->{'messages'}},$message);                          'leavechannel',
208                                    $subscriber->{'ip'},
209          $self->trimMessageStoreBySize();                          $subscriber->{'subscriberID'},
210                                    $self->{'name'},
211          my $text=$message->message();                          $timeConnected,
212          map { $_->sendMessage($text) } @{$self->{'subscribers'}};                          $subscriber->{'MessageCount'},
213  }                          $subscriber->{'bytesWritten'},
214                            $reason
215  sub messageCount {                  );
216          my $self=shift;          }
217                    
218          scalar(@{$self->{'messages'}});          $self->checkExpiration();
219  }  }
220    
221  sub trimMessageStoreBySize {  sub subscriberCount {
222          my $self=shift;          my $self=shift;
223                    
224          my $numMessages=scalar(@{$self->{'messages'}});          scalar(@{$self->{'subscribers'}});
225            }
226          if($numMessages>$::CONF{'MaxMessagesPerChannel'})  
227          {  sub addMessage {
228                  splice(@{$self->{'messages'}},0,-$::CONF{'MaxMessagesPerChannel'});          my $self=shift;
229          }          my $messageText=shift;
230  }          
231            my $message=Meteor::Message->newWithID($MessageID++);
232  sub trimMessageStoreByTimestamp {          $message->setText($messageText);
233          my $self=shift;          $message->setChannelName($self->{'name'});
234          my $ts=shift;          push(@{$self->{'messages'}},$message);
235                    &::syslog('debug',"New message ".$message->{"id"}." on channel ".$self->{'name'});
236          while(scalar(@{$self->{'messages'}})>0 && $self->{'messages'}->[0]->timestamp()<$ts)          
237          {          $self->trimMessageStoreBySize();
238                  my $msg=shift(@{$self->{'messages'}});          
239          }          map { $_->sendMessages($message) } @{$self->{'subscribers'}};
240                    
241          $self->checkExpiration();          $message;
242  }  }
243    
244  sub clearBuffer {  sub messageCount {
245          my $self=shift;          my $self=shift;
246                    
247          $self->{'messages'}=[];          scalar(@{$self->{'messages'}});
248            }
249          $self->checkExpiration();  
250  }  sub trimMessageStoreBySize {
251            my $self=shift;
252  sub checkExpiration {          
253          my $self=shift;          my $numMessages=scalar(@{$self->{'messages'}});
254                    
255          if($self->messageCount()==0 && $self->subscriberCount()==0)          if($numMessages>$::CONF{'MaxMessagesPerChannel'})
256          {          {
257                  my $name=$self->name();                  splice(@{$self->{'messages'}},0,-$::CONF{'MaxMessagesPerChannel'});
258                  &::syslog('debug',"Channel expired: $name");          }
259                  $self->deleteChannel($name);  }
260          }  
261  }  sub trimMessageStoreByTimestamp {
262            my $self=shift;
263  sub indexForMessageID {          my $ts=shift;
264          my $self=shift;          
265          my $id=shift;          while(scalar(@{$self->{'messages'}})>0 && $self->{'messages'}->[0]->timestamp()<$ts)
266                    {
267          # the messages is always sorted by ID, so we can                  my $msg=shift(@{$self->{'messages'}});
268          # use a binary search to find the message.          }
269          # return undef if there are no messages or the          
270          # ID is that of the last message.          $self->checkExpiration();
271          # Otherwise return the ID of the found message  }
272          # of if no message with that ID exists the one  
273          # with the next higher ID  sub clearBuffer {
274          #          my $self=shift;
275          return undef unless(defined($id));          
276                    $self->{'messages'}=[];
277          my $numMessages=scalar(scalar(@{$self->{'messages'}}));          
278                    $self->checkExpiration();
279          return undef unless($numMessages);  }
280          return -1 unless($id ne '');  
281            sub checkExpiration {
282          # Note: negative $id means go back that many messages          my $self=shift;
283          return $numMessages+$id if($id<0);          
284                    if($self->messageCount()==0 && $self->subscriberCount()==0)
285          my $low=0;          {
286          my $high=$numMessages-1;                  my $name=$self->name();
287          my $mid;                  &::syslog('debug',"Channel expired: $name");
288          my $cond;                  $self->deleteChannel($name);
289          while($low<=$high)          }
290          {  }
291                  $mid=($low+$high)>>1;  
292                  $cond=$id <=> $self->{'messages'}->[$mid]->id();  sub indexForMessageID {
293                  if($cond<0)          my $self=shift;
294                  {          my $id=shift;
295                          $high=$mid-1;          
296                  }          # the messages is always sorted by ID, so we can
297                  elsif($cond>0)          # use a binary search to find the message.
298                  {          # return undef if there are no messages or the
299                          $low=$mid+1;          # ID is that of the last message.
300                  }          # Otherwise return the ID of the found message
301                  else          # of if no message with that ID exists the one
302                  {          # with the next higher ID
303                          return $mid;          #
304                  }          return undef unless(defined($id));
305          }          
306                    my $numMessages=scalar(@{$self->{'messages'}});
307          return undef if($low>=$numMessages);          
308                    return undef unless($numMessages);
309          return $low;          return -1 unless($id ne '');
310  }          
311            # Note: negative $id means go back that many messages
312  1;          return $numMessages+$id if($id<0);
313            
314            my $low=0;
315            my $high=$numMessages-1;
316            my $mid;
317            my $cond;
318            while($low<=$high)
319            {
320                    $mid=($low+$high)>>1;
321                    $cond=$id <=> $self->{'messages'}->[$mid]->id();
322                    if($cond<0)
323                    {
324                            $high=$mid-1;
325                    }
326                    elsif($cond>0)
327                    {
328                            $low=$mid+1;
329                    }
330                    else
331                    {
332                            return $mid;
333                    }
334            }
335            
336            return undef if($low>=$numMessages);
337            
338            return $low;
339    }
340    
341    sub lastMsgID {
342            my $self=shift;
343            my $numMessages=scalar(@{$self->{'messages'}});
344            return undef unless($numMessages>0);
345            @{$self->{'messages'}}[-1]->id();
346    }
347    
348    sub descriptionWithTemplate {
349            my $self=shift;
350            my $template=shift;
351            
352            return '' unless(defined($template) && $template ne '');
353            
354            $template=~s/~([a-zA-Z0-9_]*)~/
355                    if(!defined($1) || $1 eq '') {
356                            '~';
357                    } elsif($1 eq 'messageCount') {
358                            $self->messageCount();
359                    } elsif($1 eq 'subscriberCount') {
360                            $self->subscriberCount();
361                    } elsif($1 eq 'lastMsgID') {
362                            $self->lastMsgID() || 1;
363                    } elsif($1 eq 'name') {
364                            $self->{'name'};
365                    } else {
366                            '';
367                    }
368            /gex;
369            
370            $template;
371    }
372    
373    1;
374  ############################################################################EOF  ############################################################################EOF
Legend:
Removed from v.9  
changed lines
  Added in v.58
  ViewVC Help
Powered by ViewVC 1.1.26