用PHP 写的基于 Memcache 的 Queue

PHP代码

 <?php  
    class MQ{  
        public static $client;  
        private static $m_real;  
        private static $m_front;  
        private static $m_data = array();  
        const QUEUE_MAX_NUM = 100000000;  
        const QUEUE_FRONT_KEY = '_queue_item_front';  
        const QUEUE_REAL_KEY = '_queue_item_real';  
        public static function setupMq($conf) {  
            self::$client = memcache_pconnect($conf);  
            self::$m_real = memcache_get(self::$client, self::QUEUE_REAL_KEY);  
            self::$m_front = memcache_get(self::$client, self::QUEUE_FRONT_KEY);  
            if (!isset(self::$m_real) || emptyempty(self::$m_real)) {  
                self::$real= 0;  
            }  
            if (!isset(self::$m_front) || emptyempty(self::$m_front)) {  
                self::$m_front = 0;  
            }  
            return self::$client;  
        }  
        public static function add($queue, $data) {  
            $result = false;  
            if (self::$m_real < self::QUEUE_MAX_NUM) {  
                if (memcache_add(self::$client, $queue.self::$m_real, $data)) {  
                    self::mqRealChange();  
                    $result = true;  
                }  
            }  

            return $result;  
        }  
        public static function get($key, $count) {  
            $num = 0;  
            for ($i=self::$m_front;$i<self::$m_front + $count;$i++) {  
                if ($dataTmp = memcache_get(self::$client, $key.$i)) {  
                    self::$m_data[] = $dataTmp;  
                    memcache_delete(self::$client, $key.$i);  
                    $num++;  
                }  
            }  
            if ($num>0) {  
                self::mqFrontChange($num);  
            }  
            return self::$m_data;  
        }  
        private static function mqRealChange() {  
            memcache_add(self::$client, self::QUEUE_REAL_KEY, 0);  
            self::$m_real = memcache_increment(self::$client, self::QUEUE_REAL_KEY, 1);  
        }  

        private static function mqFrontChange($num) {  
            memcache_add(self::$client, self::QUEUE_FRONT_KEY, 0);  
            self::$m_front = memcache_increment(self::$client, self::QUEUE_FRONT_KEY, $num);  
        }  
        public static function mflush($memcache_obj) {  
            memcache_flush($memcache_obj);  
        }  
        public static function Debug() {  
            echo 'real:'.self::$m_real."<br>/r/n";  
            echo 'front:'.self::$m_front."<br>/r/n";  
            echo 'wait for process data:'.intval(self::$m_real - self::$m_front);  
            echo "<br>/r/n";  
            echo '<pre>';  
            print_r(self::$m_data);  
            echo '<pre>';  
        }  
    }  
    define('FLUSH_MQ',0);//CLEAN ALL DATA  
    define('IS_ADD',0);//SET DATA  
    $mobj = MQ::setupMq('127.0.0.1','11211');  
    if (FLUSH_MQ) {  
        MQ::mflush($mobj);  
    } else {  
        if (IS_ADD) {  
            MQ::add('user_sync', '1test');  
            MQ::add('user_sync', '2test');  
            MQ::add('user_sync', '3test');  
            MQ::add('user_sync', '4test');  
            MQ::add('user_sync', '5test');  
            MQ::add('user_sync', '6test');  
        } else {  
            MQ::get('user_sync', 10);  
        }  

    }  
    MQ::Debug();  ?>

使用方法

MQ::setupMq('127.0.0.1','11211');//连接

MQ::add($key, $value);//添加数据到队列

MQ::add($key, $value);//添加数据到队列

MQ::add($key, $value);//添加数据到队列

MQ::add($key, $value);//添加数据到队列

MQ::add($key, $value);//添加数据到队列

MQ::add($key, $value);//添加数据到队列

MQ:get($key, 10);//取出一定数量的数据