OMS/app/taskmgr/lib/connecter/rabbitmq/stocket.php

<?php

/**
 * rabbitmq 访问对像
 */
use PhpAmqpLib\Connection\AMQPConnection;
use PhpAmqpLib\Message\AMQPMessage;

class taskmgr_connecter_rabbitmq_stocket extends taskmgr_connecter_abstract implements taskmgr_connecter_interface{

    //连接配置信息
    protected $_amqp_config = null;
    //交换机名
    protected $_amqp_exchange_name = null;
    //对列名
    protected $_amqp_queue_name = null; 
    //rabbitmq 连接
    protected $_amqp_connect = null;
    //rabbitmq 通道
    protected $_amqp_channel = null;
    //已定义的发布对列
    protected $_publishQueue = array();

    /**
     * __destruct
     * @return mixed 返回值
     */

    public function __destruct(){
        //销毁类对象的时候断开mq连接
        $this->disconnect();
    }

    /**
     * 初始化数据访问对像
     * 
     * @param string $task 任务标识
     * @return void
     */
    public function load($task, $config) {

        $queue_prefix = $config['QUEUE_PREFIX'] ? $config['QUEUE_PREFIX'] : 'ERP';
        $config['ROUTER'] = sprintf($config['ROUTER'], strtolower($task));

        $exchangeName = sprintf('%s_TASK_%s_EXCHANGE', $queue_prefix, strtoupper($task));
        $queueName =sprintf('%s_TASK_%s_QUEUE', $queue_prefix, strtoupper($task));

        $isConnect = $this->connect(array('config' => $config, 'exchangeName' => $exchangeName, 'queueName' => $queueName));
        $this->length();

        return $isConnect;
    }
    
    /**
     * 连接 rabbitmq 服务器 
     *  
     * @param $config Array 连接参数 
     * @param $exchangeName String 交换机名称 
     * @param $queueName String 对列名 
     */
    public function connect($cfg) {

        //分解参数
        $config         = $cfg['config'];
        $exchangeName   = $cfg['exchangeName'];
        $queueName      = $cfg['queueName'];

        if (!$this->_validCfg($config)) {

            return false;
        }

        $exchangeName = trim($exchangeName);
        if (empty($exchangeName)) {

            return false;
        }

        $queueName = trim($queueName);
        if (empty($queueName)) {

            return false;
        }
       
        //断开原有链接
        $this->disconnect();

        //保存参数
        $this->_amqp_config = $config;
        $this->_amqp_exchange_name = $exchangeName;
        $this->_amqp_queue_name = $queueName;

        return true;
    }

    /**
     * 断开amqp连接 
     *  
     * @param void 
     * @retrun void 
     */
    public function disconnect() {

        if (is_object($this->_amqp_connect)) {

            unset($this->_amqp_channel);
            $this->_amqp_channel = null;
            $this->_amqp_connect->close();
            unset($this->_amqp_connect);
            $this->_amqp_connect = null;
        }
    }

    /**
     * 向对列提交信息 
     *  
     * @param $message String 信息内容体 
     * @param $router String  
     * @return boolean 
     */
    public function publish($message, $router) {

        if ($this->_amqpChannel()) {

            $msg = new AMQPMessage($message, array('content_type' => 'text/plain', 'delivery_mode' => 2));
            return $this->_amqp_channel->basic_publish($msg, $this->_amqp_exchange_name);
        } else {

            return false;
        }
    }

    /**
     * 获取队列信息 
     *  
     * @param void 
     * @return mixed 
     */
    public function get() {

        if ($this->_amqpChannel()) {

            return $this->_amqp_channel->basic_get($this->_amqp_queue_name);
        } else {

            return null;
        }
    }

    /**
     * 使用 block 模式 
     *  
     * @param void 
     * @return void 
     */
    public function consume($function) {

        if ($this->_amqpChannel()) {
            
            $this->_amqp_channel->basic_consume($this->_amqp_queue_name, $this->_consumer_tag, false, false, false, false, $function);

            while (count($this->_amqp_channel->callbacks)) {
                $this->_amqp_channel->wait();
            }
        } else {
            
            return null;
        }
    }

    /**
     * 获取队列信息条数 
     *  
     * @param void 
     * @return integer 
     */
    public function length() {

        if ($this->_amqpChannel()) {
            
            //return $this->_amqp_channel->queue_declare($this->_amqp_queue_name);
            return 1;
        } else {

            return 0;
        }
    }

    /**
     * 
     */
    public function ack($tagId) {

        if ($this->_amqpChannel()) {
            
            $this->_amqp_channel->ack($tagId);
        } else {

            return null;
        }
    }

    /**
     * 
     */
    public function nack($tagId) {

        if ($this->_amqpChannel()) {

            $this->_amqp_channel->nack($tagId);
        } else {

            return null;
        }
    }

    /**
     * 创建AMQP连接 
     *  
     * @param void 
     * @return boolean 
     */
    protected function _amqpConnect() {

        if (is_object($this->_amqp_connect)) {

            if (!$this->_amqp_connect->isConnected()) {

                //重新连接
                return $this->_amqp_connect->reconnect();
            } else {

                return true;
            }
        } else {

            $this->_amqp_connect = new AMQPConnection(
                $this->_amqp_config['HOST'],
                $this->_amqp_config['PORT'],
                $this->_amqp_config['USER'],
                $this->_amqp_config['PASSWD'],
                $this->_amqp_config['VHOST']);
            return true; //$this->_amqp_connect->connect();
        }
    }

    /**
     * 创建 amqp channel 
     * @param void 
     * @return Boolean 
     */
    protected function _amqpChannel() {

        if ($this->_amqpConnect()) {

            if (is_object($this->_amqp_channel)) {

                return true;
            } else {

                unset($this->_amqp_channel);
                $this->_amqp_channel = $this->_amqp_connect->channel();
                $this->_amqp_channel->exchange_declare($this->_amqp_exchange_name, 'topic', false, true, false);
                $this->_amqp_channel->queue_declare($this->_amqp_queue_name, false, true, false, false);
                $this->_amqp_channel->queue_bind($this->_amqp_queue_name, $this->_amqp_exchange_name);
                return true;

                //return $this->_amqp_channel->isConnected(); 
            }
        } else {

            return false;
        }
    }

    /**
     * rabbitmq 连接参数检查 
     *  
     * @param $config Array 要检查的参数数组 
     * @return Boolean 
     */
    protected function _validCfg($config) {

        //先只做简单检查 ，后期可能需对参数做完整检查
        if (!is_array($config) || empty($config)) {

            return false;
        } else {

            return true;
        }
    }

}