diff -r 5b37998e522e -r 162c1de6545a web/lib/Zend/Service/Amazon/Sqs.php
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/web/lib/Zend/Service/Amazon/Sqs.php Fri Mar 11 15:05:35 2011 +0100
@@ -0,0 +1,465 @@
+_makeRequest(null, 'CreateQueue', $params);
+
+ if (!isset($result->CreateQueueResult->QueueUrl)
+ || empty($result->CreateQueueResult->QueueUrl)
+ ) {
+ if ($result->Error->Code == 'AWS.SimpleQueueService.QueueNameExists') {
+ return false;
+ } elseif ($result->Error->Code == 'AWS.SimpleQueueService.QueueDeletedRecently') {
+ // Must sleep for 60 seconds, then try re-creating the queue
+ sleep(60);
+ $retry = true;
+ $retry_count++;
+ } else {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ }
+ } else {
+ return (string) $result->CreateQueueResult->QueueUrl;
+ }
+
+ } while ($retry);
+
+ return false;
+ }
+
+ /**
+ * Delete a queue and all of it's messages
+ *
+ * Returns false if the queue is not found, true if the queue exists
+ *
+ * @param string $queue_url queue URL
+ * @return boolean
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function delete($queue_url)
+ {
+ $result = $this->_makeRequest($queue_url, 'DeleteQueue');
+
+ if ($result->Error->Code !== null) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ }
+
+ return true;
+ }
+
+ /**
+ * Get an array of all available queues
+ *
+ * @return array
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function getQueues()
+ {
+ $result = $this->_makeRequest(null, 'ListQueues');
+
+ if (isset($result->Error)) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ }
+
+ if (!isset($result->ListQueuesResult->QueueUrl)
+ || empty($result->ListQueuesResult->QueueUrl)
+ ) {
+ return array();
+ }
+
+ $queues = array();
+ foreach ($result->ListQueuesResult->QueueUrl as $queue_url) {
+ $queues[] = (string)$queue_url;
+ }
+
+ return $queues;
+ }
+
+ /**
+ * Return the approximate number of messages in the queue
+ *
+ * @param string $queue_url Queue URL
+ * @return integer
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function count($queue_url)
+ {
+ return (int)$this->getAttribute($queue_url, 'ApproximateNumberOfMessages');
+ }
+
+ /**
+ * Send a message to the queue
+ *
+ * @param string $queue_url Queue URL
+ * @param string $message Message to send to the queue
+ * @return string Message ID
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function send($queue_url, $message)
+ {
+ $params = array();
+ $params['MessageBody'] = urlencode($message);
+
+ $checksum = md5($params['MessageBody']);
+
+ $result = $this->_makeRequest($queue_url, 'SendMessage', $params);
+
+ if (!isset($result->SendMessageResult->MessageId)
+ || empty($result->SendMessageResult->MessageId)
+ ) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ } else if ((string) $result->SendMessageResult->MD5OfMessageBody != $checksum) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception('MD5 of body does not match message sent');
+ }
+
+ return (string) $result->SendMessageResult->MessageId;
+ }
+
+ /**
+ * Get messages in the queue
+ *
+ * @param string $queue_url Queue name
+ * @param integer $max_messages Maximum number of messages to return
+ * @param integer $timeout Visibility timeout for these messages
+ * @return array
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function receive($queue_url, $max_messages = null, $timeout = null)
+ {
+ $params = array();
+
+ // If not set, the visibility timeout on the queue is used
+ if ($timeout !== null) {
+ $params['VisibilityTimeout'] = (int)$timeout;
+ }
+
+ // SQS will default to only returning one message
+ if ($max_messages !== null) {
+ $params['MaxNumberOfMessages'] = (int)$max_messages;
+ }
+
+ $result = $this->_makeRequest($queue_url, 'ReceiveMessage', $params);
+
+ if (isset($result->Error)) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ }
+
+ if (!isset($result->ReceiveMessageResult->Message)
+ || empty($result->ReceiveMessageResult->Message)
+ ) {
+ // no messages found
+ return array();
+ }
+
+ $data = array();
+ foreach ($result->ReceiveMessageResult->Message as $message) {
+ $data[] = array(
+ 'message_id' => (string)$message->MessageId,
+ 'handle' => (string)$message->ReceiptHandle,
+ 'md5' => (string)$message->MD5OfBody,
+ 'body' => urldecode((string)$message->Body),
+ );
+ }
+
+ return $data;
+ }
+
+ /**
+ * Delete a message from the queue
+ *
+ * Returns true if the message is deleted, false if the deletion is
+ * unsuccessful.
+ *
+ * @param string $queue_url Queue URL
+ * @param string $handle Message handle as returned by SQS
+ * @return boolean
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function deleteMessage($queue_url, $handle)
+ {
+ $params = array();
+ $params['ReceiptHandle'] = (string)$handle;
+
+ $result = $this->_makeRequest($queue_url, 'DeleteMessage', $params);
+
+ if (isset($result->Error->Code)
+ && !empty($result->Error->Code)
+ ) {
+ return false;
+ }
+
+ // Will always return true unless ReceiptHandle is malformed
+ return true;
+ }
+
+ /**
+ * Get the attributes for the queue
+ *
+ * @param string $queue_url Queue URL
+ * @param string $attribute
+ * @return string
+ * @throws Zend_Service_Amazon_Sqs_Exception
+ */
+ public function getAttribute($queue_url, $attribute = 'All')
+ {
+ $params = array();
+ $params['AttributeName'] = $attribute;
+
+ $result = $this->_makeRequest($queue_url, 'GetQueueAttributes', $params);
+
+ if (!isset($result->GetQueueAttributesResult->Attribute)
+ || empty($result->GetQueueAttributesResult->Attribute)
+ ) {
+ require_once 'Zend/Service/Amazon/Sqs/Exception.php';
+ throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);
+ }
+
+ if(count($result->GetQueueAttributesResult->Attribute) > 1) {
+ $attr_result = array();
+ foreach($result->GetQueueAttributesResult->Attribute as $attribute) {
+ $attr_result[(string)$attribute->Name] = (string)$attribute->Value;
+ }
+ return $attr_result;
+ } else {
+ return (string) $result->GetQueueAttributesResult->Attribute->Value;
+ }
+ }
+
+ /**
+ * Make a request to Amazon SQS
+ *
+ * @param string $queue Queue Name
+ * @param string $action SQS action
+ * @param array $params
+ * @return SimpleXMLElement
+ */
+ private function _makeRequest($queue_url, $action, $params = array())
+ {
+ $params['Action'] = $action;
+ $params = $this->addRequiredParameters($queue_url, $params);
+
+ if ($queue_url === null) {
+ $queue_url = '/';
+ }
+
+ $client = self::getHttpClient();
+
+ switch ($action) {
+ case 'ListQueues':
+ case 'CreateQueue':
+ $client->setUri('http://'.$this->_sqsEndpoint);
+ break;
+ default:
+ $client->setUri($queue_url);
+ break;
+ }
+
+ $retry_count = 0;
+
+ do {
+ $retry = false;
+
+ $client->resetParameters();
+ $client->setParameterGet($params);
+
+ $response = $client->request('GET');
+
+ $response_code = $response->getStatus();
+
+ // Some 5xx errors are expected, so retry automatically
+ if ($response_code >= 500 && $response_code < 600 && $retry_count <= 5) {
+ $retry = true;
+ $retry_count++;
+ sleep($retry_count / 4 * $retry_count);
+ }
+ } while ($retry);
+
+ unset($client);
+
+ return new SimpleXMLElement($response->getBody());
+ }
+
+ /**
+ * Adds required authentication and version parameters to an array of
+ * parameters
+ *
+ * The required parameters are:
+ * - AWSAccessKey
+ * - SignatureVersion
+ * - Timestamp
+ * - Version and
+ * - Signature
+ *
+ * If a required parameter is already set in the $parameters array,
+ * it is overwritten.
+ *
+ * @param string $queue_url Queue URL
+ * @param array $parameters the array to which to add the required
+ * parameters.
+ * @return array
+ */
+ protected function addRequiredParameters($queue_url, array $parameters)
+ {
+ $parameters['AWSAccessKeyId'] = $this->_getAccessKey();
+ $parameters['SignatureVersion'] = $this->_sqsSignatureVersion;
+ $parameters['Timestamp'] = gmdate('Y-m-d\TH:i:s\Z', time()+10);
+ $parameters['Version'] = $this->_sqsApiVersion;
+ $parameters['SignatureMethod'] = $this->_sqsSignatureMethod;
+ $parameters['Signature'] = $this->_signParameters($queue_url, $parameters);
+
+ return $parameters;
+ }
+
+ /**
+ * Computes the RFC 2104-compliant HMAC signature for request parameters
+ *
+ * This implements the Amazon Web Services signature, as per the following
+ * specification:
+ *
+ * 1. Sort all request parameters (including SignatureVersion and
+ * excluding Signature, the value of which is being created),
+ * ignoring case.
+ *
+ * 2. Iterate over the sorted list and append the parameter name (in its
+ * original case) and then its value. Do not URL-encode the parameter
+ * values before constructing this string. Do not use any separator
+ * characters when appending strings.
+ *
+ * @param string $queue_url Queue URL
+ * @param array $parameters the parameters for which to get the signature.
+ *
+ * @return string the signed data.
+ */
+ protected function _signParameters($queue_url, array $paramaters)
+ {
+ $data = "GET\n";
+ $data .= $this->_sqsEndpoint . "\n";
+ if ($queue_url !== null) {
+ $data .= parse_url($queue_url, PHP_URL_PATH);
+ }
+ else {
+ $data .= '/';
+ }
+ $data .= "\n";
+
+ uksort($paramaters, 'strcmp');
+ unset($paramaters['Signature']);
+
+ $arrData = array();
+ foreach($paramaters as $key => $value) {
+ $arrData[] = $key . '=' . str_replace('%7E', '~', urlencode($value));
+ }
+
+ $data .= implode('&', $arrData);
+
+ $hmac = Zend_Crypt_Hmac::compute($this->_getSecretKey(), 'SHA256', $data, Zend_Crypt_Hmac::BINARY);
+
+ return base64_encode($hmac);
+ }
+}