git.gnu.io has moved to IP address 209.51.188.249 -- please double check where you are logging in.

Queue_item.php 3.65 KB
Newer Older
1 2 3 4
<?php
/**
 * Table Definition for queue_item
 */
5
require_once INSTALLDIR.'/classes/Memcached_DataObject.php';
6

7
class Queue_item extends Managed_DataObject
8 9 10 11 12
{
    ###START_AUTOCODE
    /* the code below is auto generated do not remove the above tag */

    public $__table = 'queue_item';                      // table name
13 14
    public $id;                              // int(4)  primary_key not_null
    public $frame;                           // blob not_null
15
    public $transport;                       // varchar(32)
16
    public $created;                         // datetime()   not_null
17
    public $claimed;                         // datetime()
18 19 20

    /* the code above is auto generated do not remove the tag below */
    ###END_AUTOCODE
21

22 23 24 25 26 27
    public static function schemaDef()
    {
        return array(
            'fields' => array(
                'id' => array('type' => 'serial', 'not null' => true, 'description' => 'unique identifier'),
                'frame' => array('type' => 'blob', 'not null' => true, 'description' => 'data: object reference or opaque string'),
28
                'transport' => array('type' => 'varchar', 'length' => 32, 'not null' => true, 'description' => 'queue for what? "email", "xmpp", "sms", "irc", ...'),
29 30 31 32 33 34 35 36 37 38
                'created' => array('type' => 'datetime', 'not null' => true, 'description' => 'date this record was created'),
                'claimed' => array('type' => 'datetime', 'description' => 'date this item was claimed'),
            ),
            'primary key' => array('id'),
            'indexes' => array(
                'queue_item_created_idx' => array('created'),
            ),
        );
    }

39 40 41 42
    /**
     * @param mixed $transports name of a single queue or array of queues to pull from
     *                          If not specified, checks all queues in the system.
     */
43
    static function top($transports=null, array $ignored_transports=array()) {
44 45

        $qi = new Queue_item();
46 47 48
        if ($transports) {
            if (is_array($transports)) {
                // @fixme use safer escaping
49
                $list = implode("','", array_map(array($qi, 'escape'), $transports));
50 51 52 53
                $qi->whereAdd("transport in ('$list')");
            } else {
                $qi->transport = $transports;
            }
54
        }
55 56 57 58 59
        if (!empty($ignored_transports)) {
            // @fixme use safer escaping
            $list = implode("','", array_map(array($qi, 'escape'), $ignored_transports));
            $qi->whereAdd("transport NOT IN ('$list')");
        }
60
        $qi->orderBy('created');
Evan Prodromou's avatar
Evan Prodromou committed
61
        $qi->whereAdd('claimed is null');
62 63 64

        $qi->limit(1);

Evan Prodromou's avatar
Evan Prodromou committed
65
        $cnt = $qi->find(true);
66 67

        if ($cnt) {
68 69 70
            // XXX: potential race condition
            // can we force it to only update if claimed is still null
            // (or old)?
71
            common_log(LOG_INFO, 'claiming queue item id = ' . $qi->getID() . ' for transport ' . $qi->transport);
72 73 74 75
            $orig = clone($qi);
            $qi->claimed = common_sql_now();
            $result = $qi->update($orig);
            if ($result) {
76
                common_log(LOG_DEBUG, 'claim succeeded.');
77 78
                return $qi;
            } else {
mmn's avatar
mmn committed
79
                common_log(LOG_ERR, 'claim of queue item id= ' . $qi->getID() . ' for transport ' . $qi->transport . ' failed.');
80 81
            }
        }
Evan Prodromou's avatar
Evan Prodromou committed
82 83
        $qi = null;
        return null;
84
    }
85 86 87 88

    /**
     * Release a claimed item.
     */
89
    function releaseClaim()
90 91
    {
        // DB_DataObject doesn't let us save nulls right now
92
        $sql = sprintf("UPDATE queue_item SET claimed=NULL WHERE id=%d", $this->getID());
93 94 95 96 97
        $this->query($sql);

        $this->claimed = null;
        $this->encache();
    }
98
}