CommunityID/libs/Zend/Queue/Adapter/Array.php

351 lines
9.5 KiB
PHP

<?php
/**
* Zend Framework
*
* LICENSE
*
* This source file is subject to the new BSD license that is bundled
* with this package in the file LICENSE.txt.
* It is also available through the world-wide-web at this URL:
* http://framework.zend.com/license/new-bsd
* If you did not receive a copy of the license and are unable to
* obtain it through the world-wide-web, please send an email
* to license@zend.com so we can send you a copy immediately.
*
* @category Zend
* @package Zend_Queue
* @subpackage Adapter
* @copyright Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
* @license http://framework.zend.com/license/new-bsd New BSD License
* @version $Id: Array.php 16881 2009-07-20 13:24:19Z mikaelkael $
*/
/**
* @see Zend_Queue_Adapter_AdapterAbstract
*/
require_once 'Zend/Queue/Adapter/AdapterAbstract.php';
/**
* Class for using a standard PHP array as a queue
*
* @category Zend
* @package Zend_Queue
* @subpackage Adapter
* @copyright Copyright (c) 2005-2009 Zend Technologies USA Inc. (http://www.zend.com)
* @license http://framework.zend.com/license/new-bsd New BSD License
*/
class Zend_Queue_Adapter_Array extends Zend_Queue_Adapter_AdapterAbstract
{
/**
* @var array
*/
protected $_data = array();
/**
* Constructor
*
* @param array|Zend_Config $options
* @param Zend_Queue|null $queue
* @return void
*/
public function __construct($options, Zend_Queue $queue = null)
{
parent::__construct($options, $queue);
}
/********************************************************************
* Queue management functions
*********************************************************************/
/**
* Does a queue already exist?
*
* Throws an exception if the adapter cannot determine if a queue exists.
* use isSupported('isExists') to determine if an adapter can test for
* queue existance.
*
* @param string $name
* @return boolean
*/
public function isExists($name)
{
return array_key_exists($name, $this->_data);
}
/**
* Create a new queue
*
* Visibility timeout is how long a message is left in the queue "invisible"
* to other readers. If the message is acknowleged (deleted) before the
* timeout, then the message is deleted. However, if the timeout expires
* then the message will be made available to other queue readers.
*
* @param string $name queue name
* @param integer $timeout default visibility timeout
* @return boolean
*/
public function create($name, $timeout=null)
{
if ($this->isExists($name)) {
return false;
}
if ($timeout === null) {
$timeout = self::CREATE_TIMEOUT_DEFAULT;
}
$this->_data[$name] = array();
return true;
}
/**
* Delete a queue and all of it's messages
*
* Returns false if the queue is not found, true if the queue exists
*
* @param string $name queue name
* @return boolean
*/
public function delete($name)
{
$found = isset($this->_data[$name]);
if ($found) {
unset($this->_data[$name]);
}
return $found;
}
/**
* Get an array of all available queues
*
* Not all adapters support getQueues(), use isSupported('getQueues')
* to determine if the adapter supports this feature.
*
* @return array
*/
public function getQueues()
{
return array_keys($this->_data);
}
/**
* Return the approximate number of messages in the queue
*
* @param Zend_Queue $queue
* @return integer
* @throws Zend_Queue_Exception
*/
public function count(Zend_Queue $queue=null)
{
if ($queue === null) {
$queue = $this->_queue;
}
if (!isset($this->_data[$queue->getName()])) {
/**
* @see Zend_Queue_Exception
*/
require_once 'Zend/Queue/Exception.php';
throw new Zend_Queue_Exception('Queue does not exist');
}
return count($this->_data[$queue->getName()]);
}
/********************************************************************
* Messsage management functions
*********************************************************************/
/**
* Send a message to the queue
*
* @param string $message Message to send to the active queue
* @param Zend_Queue $queue
* @return Zend_Queue_Message
* @throws Zend_Queue_Exception
*/
public function send($message, Zend_Queue $queue=null)
{
if ($queue === null) {
$queue = $this->_queue;
}
if (!$this->isExists($queue->getName())) {
require_once 'Zend/Queue/Exception.php';
throw new Zend_Queue_Exception('Queue does not exist:' . $queue->getName());
}
// create the message
$data = array(
'message_id' => md5(uniqid(rand(), true)),
'body' => $message,
'md5' => md5($message),
'handle' => null,
'created' => time(),
'queue_name' => $queue->getName(),
);
// add $data to the "queue"
$this->_data[$queue->getName()][] = $data;
$options = array(
'queue' => $queue,
'data' => $data,
);
$classname = $queue->getMessageClass();
if (!class_exists($classname)) {
require_once 'Zend/Loader.php';
Zend_Loader::loadClass($classname);
}
return new $classname($options);
}
/**
* Get messages in the queue
*
* @param integer $maxMessages Maximum number of messages to return
* @param integer $timeout Visibility timeout for these messages
* @param Zend_Queue $queue
* @return Zend_Queue_Message_Iterator
*/
public function receive($maxMessages = null, $timeout = null, Zend_Queue $queue = null)
{
if ($maxMessages === null) {
$maxMessages = 1;
}
if ($timeout === null) {
$timeout = self::RECEIVE_TIMEOUT_DEFAULT;
}
if ($queue === null) {
$queue = $this->_queue;
}
$data = array();
$start_time = microtime(true);
$count = 0;
$temp = &$this->_data[$queue->getName()];
foreach ($temp as $key=>&$msg) {
if (($msg['handle'] === null)
|| ($msg['timeout'] + $timeout < $start_time)
) {
$msg['handle'] = md5(uniqid(rand(), true));
$msg['timeout'] = microtime(true);
$data[] = $msg;
$count++;
}
if ($count >= $maxMessages) {
break;
}
}
$options = array(
'queue' => $queue,
'data' => $data,
'messageClass' => $queue->getMessageClass(),
);
$classname = $queue->getMessageSetClass();
if (!class_exists($classname)) {
require_once 'Zend/Loader.php';
Zend_Loader::loadClass($classname);
}
return new $classname($options);
}
/**
* Delete a message from the queue
*
* Returns true if the message is deleted, false if the deletion is
* unsuccessful.
*
* @param Zend_Queue_Message $message
* @return boolean
* @throws Zend_Queue_Exception
*/
public function deleteMessage(Zend_Queue_Message $message)
{
// load the queue
$queue = &$this->_data[$message->queue_name];
foreach ($queue as $key => &$msg) {
if ($msg['handle'] == $message->handle) {
unset($queue[$key]);
return true;
}
}
return false;
}
/********************************************************************
* Supporting functions
*********************************************************************/
/**
* Return a list of queue capabilities functions
*
* $array['function name'] = true or false
* true is supported, false is not supported.
*
* @param string $name
* @return array
*/
public function getCapabilities()
{
return array(
'create' => true,
'delete' => true,
'send' => true,
'receive' => true,
'deleteMessage' => true,
'getQueues' => true,
'count' => true,
'isExists' => true,
);
}
/********************************************************************
* Functions that are not part of the Zend_Queue_Adapter_Abstract
*********************************************************************/
/**
* serialize
*/
public function __sleep()
{
return array('_data');
}
/*
* These functions are debug helpers.
*/
/**
* returns underlying _data array
* $queue->getAdapter()->getData();
*
* @return $this;
*/
public function getData()
{
return $this->_data;
}
/**
* sets the underlying _data array
* $queue->getAdapter()->setData($data);
*
* @param $data array
* @return $this;
*/
public function setData($data)
{
$this->_data = $data;
return $this;
}
}