Allow overriding of response type in call data

packaged · Sep 12, 2018 · ccd0bf2 · ccd0bf2
1 parent 28282a1
commit ccd0bf2
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/src/Response/ApiCallData.php b/src/Response/ApiCallData.php
@@ -10,18 +10,15 @@ class ApiCallData
   protected $_type;
   protected $_result;
 
-  public static function create(
-    $type, $result, $statusCode, $statusMessage, $totalTime, $executionTime,
-    $callTime
-  )
+  public static function create($type, $result, $statusCode, $statusMessage, $totalTime, $executionTime, $callTime)
   {
-    $callData = new static;
+    $callData = new static();
     $callData->_setStatus($statusCode, $statusMessage);
     $callData->_setRawResult($result);
     $callData->_setResponseType($type);
     $callData->_totalTime = $totalTime;
-    $callData->_execTime  = $executionTime;
-    $callData->_callTime  = $callTime;
+    $callData->_execTime = $executionTime;
+    $callData->_callTime = $callTime;
     return $callData;
   }
 
@@ -38,8 +35,8 @@ public function getRawResult()
 
   protected function _setStatus($code, $message = '')
   {
-    $this->_status          = new \stdClass();
-    $this->_status->code    = $code;
+    $this->_status = new \stdClass();
+    $this->_status->code = $code;
     $this->_status->message = $message;
     return $this;
   }
@@ -55,6 +52,19 @@ public function getResponseType()
     return $this->_type;
   }
 
+  /**
+   * Override response type
+   *
+   * @param $type
+   *
+   * @return $this
+   */
+  public function setResponseType($type)
+  {
+    $this->_type = $type;
+    return $this;
+  }
+
   /**
    * Get the time taken to process the call internally on the server
    *