phpMemcache消息队列类

<?php

/**
 * Memcache 消息队列类
 */
class QMC {

    const PREFIX = 'ASDFASDFFWQKE';

    /**
     * 初始化 mc
     * @staticvar string $mc
     * @return Memcache
     */
    static private function mc_init() {
        static $mc = null;
        if (is_null($mc)) {
            $mc = new Memcache;
            $mc->connect('127.0.0.1', 11211);
        }
        return $mc;
    }

    /**
     * mc 计数器 , 增加计数并返回新的计数
     * @param string $key 计数器
     * @param int $offset 计数增量 , 可为负数 .0 为不改变计数
     * @param int $time 时间
     * @return int/false 失败是返回 false, 成功时返回更新计数器后的计数
     */
    static public function set_counter($key, $offset, $time = 0) {
        $mc = self::mc_init();
        $val = $mc->get($key);
        if (!is_numeric($val) || $val < 0) {
            $ret = $mc->set($key, 0, $time);
            if (!$ret)
                return false;
            $val = 0;
        }
        $offset = intval($offset);
        if ($offset > 0) {
            return $mc->increment($key, $offset);
        } elseif ($offset < 0) {
            return $mc->decrement($key, -$offset);
        }
        return $val;
    }

    /**
     * 写入队列
     * @param string $key
     * @param mixed $value
     * @return bool
     */
    static public function input($key, $value) {
        $mc = self::mc_init();
        $w_key = self::PREFIX . $key . 'W';
        $v_key = self::PREFIX . $key . self::set_counter($w_key, 1);
        return $mc->set($v_key, $value);
    }

    /**
     * 读取队列里的数据
     * @param string $key
     * @param int $max 最多读取条数
     * @return array
     */
    static public function output($key, $max = 100) {
        $out = array();
        $mc = self::mc_init();
        $r_key = self::PREFIX . $key . 'R';
        $w_key = self::PREFIX . $key . 'W';
        $r_p = self::set_counter($r_key, 0); // 读指针
        $w_p = self::set_counter($w_key, 0); // 写指针
        if ($r_p == 0)
            $r_p = 1;
        while ($w_p >= $r_p) {
            if (--$max < 0)
                break;
            $v_key = self::PREFIX . $key . $r_p;
            $r_p = self::set_counter($r_key, 1);
            $out[] = $mc->get($v_key);
            $mc->delete($v_key);
        }
        return $out;
    }
}

/**
 * 使用方法 :
 */
QMC::input($key, $value); // 写入队列
$list = QMC::output($key); // 读取队列
?>

 

posted @ 2015-01-28 21:57  timelesszhuang  阅读(262)  评论(0编辑  收藏  举报