web/Zend/Service/WindowsAzure/Storage/Batch.php
changeset 0 4eba9c11703f
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/web/Zend/Service/WindowsAzure/Storage/Batch.php	Mon Dec 13 18:29:26 2010 +0100
@@ -0,0 +1,248 @@
+<?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_Service_WindowsAzure
+ * @subpackage Storage
+ * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license    http://framework.zend.com/license/new-bsd     New BSD License
+ * @version    $Id: Batch.php 23167 2010-10-19 17:53:31Z mabe $
+ */
+
+/**
+ * @see Zend_Service_WindowsAzure_Exception
+ */
+require_once 'Zend/Service/WindowsAzure/Exception.php';
+
+/**
+ * @see Zend_Service_WindowsAzure_Storage_BatchStorageAbstract
+ */
+require_once 'Zend/Service/WindowsAzure/Storage/BatchStorageAbstract.php';
+
+/**
+ * @category   Zend
+ * @package    Zend_Service_WindowsAzure
+ * @subpackage Storage
+ * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)
+ * @license    http://framework.zend.com/license/new-bsd     New BSD License
+ */
+class Zend_Service_WindowsAzure_Storage_Batch
+{	
+    /**
+     * Storage client the batch is defined on
+     * 
+     * @var Zend_Service_WindowsAzure_Storage_BatchStorageAbstract
+     */
+    protected $_storageClient = null;
+    
+    /**
+     * For table storage?
+     * 
+     * @var boolean
+     */
+    protected $_forTableStorage = false;
+    
+    /**
+     * Base URL
+     * 
+     * @var string
+     */
+    protected $_baseUrl;
+    
+    /**
+     * Pending operations
+     * 
+     * @var unknown_type
+     */
+    protected $_operations = array();
+    
+    /**
+     * Does the batch contain a single select?
+     * 
+     * @var boolean
+     */
+    protected $_isSingleSelect = false;
+    
+    /**
+     * Creates a new Zend_Service_WindowsAzure_Storage_Batch
+     * 
+     * @param Zend_Service_WindowsAzure_Storage_BatchStorageAbstract $storageClient Storage client the batch is defined on
+     */
+    public function __construct(Zend_Service_WindowsAzure_Storage_BatchStorageAbstract $storageClient = null, $baseUrl = '')
+    {
+        $this->_storageClient = $storageClient;
+        $this->_baseUrl       = $baseUrl;
+        $this->_beginBatch();
+    }
+    
+	/**
+	 * Get base URL for creating requests
+	 *
+	 * @return string
+	 */
+	public function getBaseUrl()
+	{
+		return $this->_baseUrl;
+	}
+    
+    /**
+     * Starts a new batch operation set
+     * 
+     * @throws Zend_Service_WindowsAzure_Exception
+     */
+    protected function _beginBatch()
+    {
+        $this->_storageClient->setCurrentBatch($this);
+    }
+    
+    /**
+     * Cleanup current batch
+     */
+    protected function _clean()
+    {
+        unset($this->_operations);
+        $this->_storageClient->setCurrentBatch(null);
+        $this->_storageClient = null;
+        unset($this);
+    }
+
+	/**
+	 * Enlist operation in current batch
+	 *
+	 * @param string $path Path
+	 * @param string $queryString Query string
+	 * @param string $httpVerb HTTP verb the request will use
+	 * @param array $headers x-ms headers to add
+	 * @param boolean $forTableStorage Is the request for table storage?
+	 * @param mixed $rawData Optional RAW HTTP data to be sent over the wire
+	 * @throws Zend_Service_WindowsAzure_Exception
+	 */
+	public function enlistOperation($path = '/', $queryString = '', $httpVerb = Zend_Http_Client::GET, $headers = array(), $forTableStorage = false, $rawData = null)
+	{
+	    // Set _forTableStorage
+	    if ($forTableStorage) {
+	        $this->_forTableStorage = true;
+	    }
+	    
+	    // Set _isSingleSelect
+	    if ($httpVerb == Zend_Http_Client::GET) {
+	        if (count($this->_operations) > 0) {
+	            throw new Zend_Service_WindowsAzure_Exception("Select operations can only be performed in an empty batch transaction.");
+	        }
+	        $this->_isSingleSelect = true;
+	    }
+	    
+	    // Clean path
+		if (strpos($path, '/') !== 0) {
+			$path = '/' . $path;
+		}
+			
+		// Clean headers
+		if ($headers === null) {
+		    $headers = array();
+		}
+		    
+		// URL encoding
+		$path           = Zend_Service_WindowsAzure_Storage::urlencode($path);
+		$queryString    = Zend_Service_WindowsAzure_Storage::urlencode($queryString);
+
+		// Generate URL
+		$requestUrl     = $this->getBaseUrl() . $path . $queryString;
+		
+		// Generate $rawData
+		if ($rawData === null) {
+		    $rawData = '';
+		}
+		    
+		// Add headers
+		if ($httpVerb != Zend_Http_Client::GET) {
+    		$headers['Content-ID'] = count($this->_operations) + 1;
+    		if ($httpVerb != Zend_Http_Client::DELETE) {
+    		    $headers['Content-Type'] = 'application/atom+xml;type=entry';
+    		}
+    		$headers['Content-Length'] = strlen($rawData);
+		}
+		    
+		// Generate $operation
+		$operation = '';
+		$operation .= $httpVerb . ' ' . $requestUrl . ' HTTP/1.1' . "\n";
+		foreach ($headers as $key => $value)
+		{
+		    $operation .= $key . ': ' . $value . "\n";
+		}
+		$operation .= "\n";
+		
+		// Add data
+		$operation .= $rawData;
+
+		// Store operation
+		$this->_operations[] = $operation;	        
+	}
+    
+    /**
+     * Commit current batch
+     * 
+     * @return Zend_Http_Response
+     * @throws Zend_Service_WindowsAzure_Exception
+     */
+    public function commit()
+    {
+        // Perform batch
+        $response = $this->_storageClient->performBatch($this->_operations, $this->_forTableStorage, $this->_isSingleSelect);
+        
+        // Dispose
+        $this->_clean();
+        
+        // Parse response
+        $errors = null;
+        preg_match_all('/<message (.*)>(.*)<\/message>/', $response->getBody(), $errors);
+        
+        // Error?
+        if (count($errors[2]) > 0) {
+            throw new Zend_Service_WindowsAzure_Exception('An error has occured while committing a batch: ' . $errors[2][0]);
+        }
+        
+        // Return
+        return $response;
+    }
+    
+    /**
+     * Rollback current batch
+     */
+    public function rollback()
+    {
+        // Dispose
+        $this->_clean();
+    }
+    
+    /**
+     * Get operation count
+     * 
+     * @return integer
+     */
+    public function getOperationCount()
+    {
+        return count($this->_operations);
+    }
+    
+    /**
+     * Is single select?
+     * 
+     * @return boolean
+     */
+    public function isSingleSelect()
+    {
+        return $this->_isSingleSelect;
+    }
+}