lib/dbqueuemanager.php

   1 <?php
   2 /**
   3  * StatusNet, the distributed open-source microblogging tool
   4  *
   5  * Simple-minded queue manager for storing items in the database
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  QueueManager
  23  * @package   StatusNet
  24  * @author    Evan Prodromou <evan@status.net>
  25  * @author    Brion Vibber <brion@status.net>
  26  * @copyright 2009-2010 StatusNet, Inc.
  27  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  28  * @link      http://status.net/
  29  */
  30
  31 class DBQueueManager extends QueueManager
  32 {
  33     /**
  34      * Saves an object reference into the queue item table.
  35      * @return boolean true on success
  36      * @throws ServerException on failure
  37      */
  38     public function enqueue($object, $queue)
  39     {
  40         $qi = new Queue_item();
  41
  42         $qi->frame     = $this->encode($object);
  43         $qi->transport = $queue;
  44         $qi->created   = common_sql_now();
  45         $result        = $qi->insert();
  46
  47         if ($result === false) {
  48             common_log_db_error($qi, 'INSERT', __FILE__);
  49             throw new ServerException('DB error inserting queue item');
  50         }
  51
  52         $this->stats('enqueued', $queue);
  53
  54         return true;
  55     }
  56
  57     /**
  58      * Poll every 10 seconds for new events during idle periods.
  59      * We'll look in more often when there's data available.
  60      *
  61      * @return int seconds
  62      */
  63     public function pollInterval()
  64     {
  65         return 10;
  66     }
  67
  68     /**
  69      * Run a polling cycle during idle processing in the input loop.
  70      * @return boolean true if we should poll again for more data immediately
  71      */
  72     public function poll()
  73     {
  74         //$this->_log(LOG_DEBUG, 'Checking for notices...');
  75         $qi = Queue_item::top($this->activeQueues(), $this->getIgnoredTransports());
  76         if (!$qi instanceof Queue_item) {
  77             //$this->_log(LOG_DEBUG, 'No notices waiting; idling.');
  78             return false;
  79         }
  80
  81         try {
  82             $item = $this->decode($qi->frame);
  83         } catch (Exception $e) {
  84             $this->_log(LOG_INFO, "[{$qi->transport}] Discarding: "._ve($e->getMessage()));
  85             $this->_done($qi);
  86             return true;
  87         }
  88
  89         $rep = $this->logrep($item);
  90         $this->_log(LOG_DEBUG, 'Got '._ve($rep).' for transport '._ve($qi->transport));
  91
  92         try {
  93             $handler = $this->getHandler($qi->transport);
  94             $result = $handler->handle($item);
  95         } catch (NoQueueHandlerException $e) {
  96             $this->noHandlerFound($qi, $rep);
  97             return true;
  98         } catch (NoResultException $e) {
  99             $this->_log(LOG_ERR, "[{$qi->transport}:$rep] ".get_class($e).' thrown ('._ve($e->getMessage()).'), ignoring queue_item '._ve($qi->getID()));
 100             $result = true;
 101         } catch (AlreadyFulfilledException $e) {
 102             $this->_log(LOG_ERR, "[{$qi->transport}:$rep] ".get_class($e).' thrown ('._ve($e->getMessage()).'), ignoring queue_item '._ve($qi->getID()));
 103             $result = true;
 104         } catch (Exception $e) {
 105             $this->_log(LOG_ERR, "[{$qi->transport}:$rep] Exception (".get_class($e).') thrown: '._ve($e->getMessage()));
 106             $result = false;
 107         }
 108
 109         if ($result) {
 110             $this->_log(LOG_INFO, "[{$qi->transport}:$rep] Successfully handled item");
 111             $this->_done($qi);
 112         } else {
 113             $this->_log(LOG_INFO, "[{$qi->transport}:$rep] Failed to handle item");
 114             $this->_fail($qi);
 115         }
 116         return true;
 117     }
 118
 119     // What to do if no handler was found. For example, the OpportunisticQM
 120     // should avoid deleting items just because it can't reach XMPP queues etc.
 121     protected function noHandlerFound(Queue_item $qi, $rep=null) {
 122         $this->_log(LOG_INFO, "[{$qi->transport}:{$rep}] No handler for queue {$qi->transport}; discarding.");
 123         $this->_done($qi);
 124     }
 125
 126     /**
 127      * Delete our claimed item from the queue after successful processing.
 128      *
 129      * @param QueueItem $qi
 130      */
 131     protected function _done(Queue_item $qi)
 132     {
 133         if (empty($qi->claimed)) {
 134             $this->_log(LOG_WARNING, "Reluctantly releasing unclaimed queue item {$qi->id} from {$qi->transport}");
 135         }
 136         $qi->delete();
 137
 138         $this->stats('handled', $qi->transport);
 139     }
 140
 141     /**
 142      * Free our claimed queue item for later reprocessing in case of
 143      * temporary failure.
 144      *
 145      * @param QueueItem $qi
 146      */
 147     protected function _fail(Queue_item $qi, $releaseOnly=false)
 148     {
 149         if (empty($qi->claimed)) {
 150             $this->_log(LOG_WARNING, "[{$qi->transport}:item {$qi->id}] Ignoring failure for unclaimed queue item");
 151         } else {
 152             $qi->releaseClaim();
 153         }
 154
 155         if (!$releaseOnly) {
 156             $this->stats('error', $qi->transport);
 157         }
 158     }
 159 }