140 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/*
 | 
						|
 * Copyright 2012 Google Inc.
 | 
						|
 *
 | 
						|
 * Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
 * you may not use this file except in compliance with the License.
 | 
						|
 * You may obtain a copy of the License at
 | 
						|
 *
 | 
						|
 *     http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 *
 | 
						|
 * Unless required by applicable law or agreed to in writing, software
 | 
						|
 * distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
 * See the License for the specific language governing permissions and
 | 
						|
 * limitations under the License.
 | 
						|
 */
 | 
						|
 | 
						|
/**
 | 
						|
 * @author Chirag Shah <chirags@google.com>
 | 
						|
 */
 | 
						|
class W3TCG_Google_Http_Batch
 | 
						|
{
 | 
						|
  /** @var string Multipart Boundary. */
 | 
						|
  private $boundary;
 | 
						|
 | 
						|
  /** @var array service requests to be executed. */
 | 
						|
  private $requests = array();
 | 
						|
 | 
						|
  /** @var W3TCG_Google_Client */
 | 
						|
  private $client;
 | 
						|
 | 
						|
  private $expected_classes = array();
 | 
						|
 | 
						|
  private $base_path;
 | 
						|
 | 
						|
  public function __construct(W3TCG_Google_Client $client, $boundary = false)
 | 
						|
  {
 | 
						|
    $this->client = $client;
 | 
						|
    $this->base_path = $this->client->getBasePath();
 | 
						|
    $this->expected_classes = array();
 | 
						|
    $boundary = (false == $boundary) ? mt_rand() : $boundary;
 | 
						|
    $this->boundary = str_replace('"', '', $boundary);
 | 
						|
  }
 | 
						|
 | 
						|
  public function add(W3TCG_Google_Http_Request $request, $key = false)
 | 
						|
  {
 | 
						|
    if (false == $key) {
 | 
						|
      $key = mt_rand();
 | 
						|
    }
 | 
						|
 | 
						|
    $this->requests[$key] = $request;
 | 
						|
  }
 | 
						|
 | 
						|
  public function execute()
 | 
						|
  {
 | 
						|
    $body = '';
 | 
						|
 | 
						|
    /** @var W3TCG_Google_Http_Request $req */
 | 
						|
    foreach ($this->requests as $key => $req) {
 | 
						|
      $body .= "--{$this->boundary}\n";
 | 
						|
      $body .= $req->toBatchString($key) . "\n";
 | 
						|
      $this->expected_classes["response-" . $key] = $req->getExpectedClass();
 | 
						|
    }
 | 
						|
 | 
						|
    $body = rtrim($body);
 | 
						|
    $body .= "\n--{$this->boundary}--";
 | 
						|
 | 
						|
    $url = $this->base_path . '/batch';
 | 
						|
    $httpRequest = new W3TCG_Google_Http_Request($url, 'POST');
 | 
						|
    $httpRequest->setRequestHeaders(
 | 
						|
        array('Content-Type' => 'multipart/mixed; boundary=' . $this->boundary)
 | 
						|
    );
 | 
						|
 | 
						|
    $httpRequest->setPostBody($body);
 | 
						|
    $response = $this->client->getIo()->makeRequest($httpRequest);
 | 
						|
 | 
						|
    return $this->parseResponse($response);
 | 
						|
  }
 | 
						|
 | 
						|
  public function parseResponse(W3TCG_Google_Http_Request $response)
 | 
						|
  {
 | 
						|
    $contentType = $response->getResponseHeader('content-type');
 | 
						|
    $contentType = explode(';', $contentType);
 | 
						|
    $boundary = false;
 | 
						|
    foreach ($contentType as $part) {
 | 
						|
      $part = (explode('=', $part, 2));
 | 
						|
      if (isset($part[0]) && 'boundary' == trim($part[0])) {
 | 
						|
        $boundary = $part[1];
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    $body = $response->getResponseBody();
 | 
						|
    if ($body) {
 | 
						|
      $body = str_replace("--$boundary--", "--$boundary", $body);
 | 
						|
      $parts = explode("--$boundary", $body);
 | 
						|
      $responses = array();
 | 
						|
 | 
						|
      foreach ($parts as $part) {
 | 
						|
        $part = trim($part);
 | 
						|
        if (!empty($part)) {
 | 
						|
          list($metaHeaders, $part) = explode("\r\n\r\n", $part, 2);
 | 
						|
          $metaHeaders = $this->client->getIo()->getHttpResponseHeaders($metaHeaders);
 | 
						|
 | 
						|
          $status = substr($part, 0, strpos($part, "\n"));
 | 
						|
          $status = explode(" ", $status);
 | 
						|
          $status = $status[1];
 | 
						|
 | 
						|
          list($partHeaders, $partBody) = $this->client->getIo()->ParseHttpResponse($part, false);
 | 
						|
          $response = new W3TCG_Google_Http_Request("");
 | 
						|
          $response->setResponseHttpCode($status);
 | 
						|
          $response->setResponseHeaders($partHeaders);
 | 
						|
          $response->setResponseBody($partBody);
 | 
						|
 | 
						|
          // Need content id.
 | 
						|
          $key = $metaHeaders['content-id'];
 | 
						|
 | 
						|
          if (isset($this->expected_classes[$key]) &&
 | 
						|
              strlen($this->expected_classes[$key]) > 0) {
 | 
						|
            $class = $this->expected_classes[$key];
 | 
						|
            $response->setExpectedClass($class);
 | 
						|
          }
 | 
						|
 | 
						|
          try {
 | 
						|
            $response = W3TCG_Google_Http_REST::decodeHttpResponse($response);
 | 
						|
            $responses[$key] = $response;
 | 
						|
          } catch (W3TCG_Google_Service_Exception $e) {
 | 
						|
            // Store the exception as the response, so succesful responses
 | 
						|
            // can be processed.
 | 
						|
            $responses[$key] = $e;
 | 
						|
          }
 | 
						|
        }
 | 
						|
      }
 | 
						|
 | 
						|
      return $responses;
 | 
						|
    }
 | 
						|
 | 
						|
    return null;
 | 
						|
  }
 | 
						|
}
 |