]> git.mxchange.org Git - quix0rs-gnu-social.git/commitdiff
split public stream to its own queue handler
authorEvan Prodromou <evan@prodromou.name>
Fri, 29 Aug 2008 18:17:02 +0000 (14:17 -0400)
committerEvan Prodromou <evan@prodromou.name>
Fri, 29 Aug 2008 18:17:02 +0000 (14:17 -0400)
Add another queue handler for the public stream. Should further
parallelize the work of sending out messages.

darcs-hash:20080829181702-84dde-594505aa73d2380b13bd98917b70b02bac597d12.gz

lib/jabber.php
lib/util.php
scripts/publicqueuehandler.php [new file with mode: 0755]
scripts/startdaemons.sh
scripts/stopdaemons.sh
scripts/xmppdaemon.php
scripts/xmppqueuehandler.php

index 215cd55374bc487eaabe1fd47b3cd6cc48d6818b..171dff4df47cd2c0b4719f1a44c72bb4c3c8a498 100644 (file)
@@ -24,32 +24,32 @@ require_once('XMPPHP/XMPP.php');
 # XXX: something of a hack to work around problems with the XMPPHP lib
 
 class Laconica_XMPP extends XMPPHP_XMPP {
-    
+
     function messageplus($to, $body, $type = 'chat', $subject = null, $payload = null) {
                $to       = htmlspecialchars($to);
                $body   = htmlspecialchars($body);
                $subject = htmlspecialchars($subject);
-               
+
                $jid = jabber_daemon_address();
-               
+
                $out = "<message from='$jid' to='$to' type='$type'>";
                if($subject) $out .= "<subject>$subject</subject>";
                $out .= "<body>$body</body>";
                if($payload) $out .= $payload;
                $out .= "</message>";
-               
+
                $cnt = strlen($out);
                common_log(LOG_DEBUG, "Sending $cnt chars to $to");
                $this->send($out);
                common_log(LOG_DEBUG, 'Done.');
     }
-       
+
        public function presence($status = null, $show = 'available', $to = null, $type='available', $priority=NULL) {
                if($type == 'available') $type = '';
                $to      = htmlspecialchars($to);
                $status = htmlspecialchars($status);
                if($show == 'unavailable') $type = 'unavailable';
-               
+
                $out = "<presence";
                if($to) $out .= " to='$to'";
                if($type) $out .= " type='$type'";
@@ -62,7 +62,7 @@ class Laconica_XMPP extends XMPPHP_XMPP {
                        if(!is_null($priority)) $out .= "<priority>$priority</priority>";
                        $out .= "</presence>";
                }
-               
+
                $this->send($out);
        }
 }
@@ -105,7 +105,7 @@ function jabber_connect($resource=NULL, $status=NULL, $priority=NULL) {
                                                                );
                $conn->autoSubscribe();
                $conn->useEncryption(common_config('xmpp', 'encryption'));
-               
+
                if (!$conn) {
                        return false;
                }
@@ -141,7 +141,7 @@ function jabber_send_notice($to, $notice) {
 # Extra stuff defined by Twitter, needed by twitter clients
 
 function jabber_format_entry($profile, $notice) {
-       
+
        $noticeurl = common_local_url('shownotice',
                                                                  array('notice' => $notice->id));
        $msg = jabber_format_notice($profile, $notice);
@@ -167,7 +167,7 @@ function jabber_format_entry($profile, $notice) {
        $html .= ($notice->rendered) ? $notice->rendered : common_render_content($notice->content, $notice);
        $html .= "\n</body>\n";
        $html .= "\n</html>\n";
-       
+
        $event = "<event xmlns='http://jabber.org/protocol/pubsub#event'>\n";
     $event .= "<items xmlns='http://jabber.org/protocol/pubsub' ";
        $event .= "node='" . common_local_url('public') . "'>\n";
@@ -229,6 +229,7 @@ function jabber_special_presence($type, $to=NULL, $show=NULL, $status=NULL) {
 }
 
 function jabber_broadcast_notice($notice) {
+
        if (!common_config('xmpp', 'enabled')) {
                return true;
        }
@@ -268,7 +269,7 @@ function jabber_broadcast_notice($notice) {
        # XXX: use a join here rather than looping through results
        $sub = new Subscription();
        $sub->subscribed = $notice->profile_id;
-        
+
        if ($sub->find()) {
                while ($sub->fetch()) {
                        $user = User::staticGet($sub->subscriber);
@@ -289,14 +290,20 @@ function jabber_broadcast_notice($notice) {
                        }
                }
        }
+       
+       return true;
+}
+
+function jabber_public_notice($notice) {
 
        # Now, users who want everything
-       
+
        $public = common_config('xmpp', 'public');
-       
+
        # FIXME PRIV don't send out private messages here
-       # XXX: should we send out non-local messages if public,localonly = false? I think not
-       
+       # XXX: should we send out non-local messages if public,localonly
+       # = false? I think not
+
        if ($public && $notice->is_local) {
                foreach ($public as $address) {
                                common_log(LOG_INFO,
@@ -305,7 +312,7 @@ function jabber_broadcast_notice($notice) {
                                jabber_send_notice($address, $notice);
                }
        }
-       
+
        return true;
 }
 
index 496c6f3d2ce6f36c50b1b4bc76c45d742e773500..c6cdfbcb949ab0d447802ce55461f9b53cbd5fb3 100644 (file)
@@ -1070,7 +1070,7 @@ function common_broadcast_notice($notice, $remote=false) {
 # Stick the notice on the queue
 
 function common_enqueue_notice($notice) {
-       foreach (array('jabber', 'omb', 'sms') as $transport) {
+       foreach (array('jabber', 'omb', 'sms', 'public') as $transport) {
                $qi = new Queue_item();
                $qi->notice_id = $notice->id;
                $qi->transport = $transport;
@@ -1126,6 +1126,12 @@ function common_real_broadcast($notice, $remote=false) {
                        common_log(LOG_ERR, 'Error in sms broadcast for notice ' . $notice->id);
                }
        }
+       if ($success) {
+               $success = jabber_public_notice($notice);
+               if (!$success) {
+                       common_log(LOG_ERR, 'Error in public broadcast for notice ' . $notice->id);
+               }
+       }
        // XXX: broadcast notices to other IM
        return $success;
 }
diff --git a/scripts/publicqueuehandler.php b/scripts/publicqueuehandler.php
new file mode 100755 (executable)
index 0000000..0d95a48
--- /dev/null
@@ -0,0 +1,66 @@
+#!/usr/bin/env php
+<?php
+/*
+ * Laconica - a distributed open-source microblogging tool
+ * Copyright (C) 2008, Controlez-Vous, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+# Abort if called from a web server
+if (isset($_SERVER) && array_key_exists('REQUEST_METHOD', $_SERVER)) {
+       print "This script must be run from the command line\n";
+       exit();
+}
+
+define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
+define('LACONICA', true);
+
+require_once(INSTALLDIR . '/lib/common.php');
+require_once(INSTALLDIR . '/lib/jabber.php');
+require_once(INSTALLDIR . '/lib/queuehandler.php');
+
+set_error_handler('common_error_handler');
+
+class PublicQueueHandler extends QueueHandler {
+       
+       function transport() {
+               return 'public';
+       }
+       
+       function start() {
+               # Low priority; we don't want to receive messages
+               $this->conn = jabber_connect($this->_id, NULL, -1);
+               return !is_null($this->conn);
+       }
+
+       function handle_notice($notice) {
+               return jabber_public_notice($notice);
+       }
+       
+       function finish() {
+       }
+}
+
+mb_internal_encoding('UTF-8');
+
+$resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-public');
+
+$handler = new XmppQueueHandler($resource);
+
+if ($handler->start()) {
+       $handler->handle_queue();
+}
+
+$handler->finish();
index 37d30cdb4094301e17d726131b5352b73b8e4f44..8c054417de965b1077a6ba6c939699e1a11ef7a1 100755 (executable)
@@ -22,6 +22,7 @@ export INSTALLDIR=$1
 
 /sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/xmppdaemon.php -b -m --pidfile=/var/run/xmppdaemon.pid
 /sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/xmppqueuehandler.php -b -m --pidfile=/var/run/xmppqueuehandler.pid
+/sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/publicqueuehandler.php -b -m --pidfile=/var/run/publicqueuehandler.pid
 /sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/xmppconfirmhandler.php -b -m --pidfile=/var/run/xmppconfirmhandler.pid
 /sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/smsqueuehandler.php -b -m --pidfile=/var/run/smsqueuehandler.pid
 /sbin/start-stop-daemon -S --exec $INSTALLDIR/scripts/ombqueuehandler.php -b -m --pidfile=/var/run/ombqueuehandler.pid
index d620472f05d2b4cde3dcbc83747c26e43ad11838..6814101cb2790463ac8b5e6291ebdad25fbc0611 100755 (executable)
@@ -22,6 +22,7 @@ export INSTALLDIR=$1
 
 /sbin/start-stop-daemon -K -m --pidfile=/var/run/xmppdaemon.pid
 /sbin/start-stop-daemon -K -m --pidfile=/var/run/xmppqueuehandler.pid
+/sbin/start-stop-daemon -K -m --pidfile=/var/run/publicqueuehandler.pid
 /sbin/start-stop-daemon -K -m --pidfile=/var/run/xmppconfirmhandler.pid
 /sbin/start-stop-daemon -K -m --pidfile=/var/run/smsqueuehandler.pid
 /sbin/start-stop-daemon -K -m --pidfile=/var/run/ombqueuehandler.pid
index b013a34c9da6d9465664b6e056f8f111c03b076f..153ab5149e174718720558afa824b337514576da 100755 (executable)
@@ -259,7 +259,7 @@ class XMPPDaemon {
 
 mb_internal_encoding('UTF-8');
 
-$resource = ($argc > 1) ? $argv[1] : NULL;
+$resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-listen');
 
 $daemon = new XMPPDaemon($resource);
 
index 8fe8b1360a1fe6a6375cb7c3b6ddb0430b8c18e0..3f632d5a8748f65cafc0b018632aaf667f7695cd 100755 (executable)
@@ -41,7 +41,7 @@ class XmppQueueHandler extends QueueHandler {
        
        function start() {
                # Low priority; we don't want to receive messages
-               $this->conn = jabber_connect($this->resource, NULL, -100);
+               $this->conn = jabber_connect($this->_id, NULL, -1);
                return !is_null($this->conn);
        }
 
@@ -55,7 +55,7 @@ class XmppQueueHandler extends QueueHandler {
 
 mb_internal_encoding('UTF-8');
 
-$resource = ($argc > 1) ? $argv[1] : NULL;
+$resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-queuehandler');
 
 $handler = new XmppQueueHandler($resource);