support BatchGetItem and BatchWriteItem

Author: kitar

src/Kitar/Dynamodb/Model/Model.php

@@ -354,6 +354,10 @@ public function __call($method, $parameters)
             "putItem",
             "deleteItem",
             "updateItem",
+            "batchGetItem",
+            "batchPutItem",
+            "batchDeleteItem",
+            "batchWriteItem",
             "scan",
             "filter",
             "filterIn",

src/Kitar/Dynamodb/Query/Builder.php

@@ -41,6 +41,20 @@ class Builder extends BaseBuilder
         'remove' => []
     ];
 
+    /**
+     * Keys array for BatchGetItem
+     * https://docs.aws.amazon.com/amazondynamodb/latest/APIReference/API_BatchGetItem.html
+     * @var array
+     */
+    public $batch_get_keys = [];
+
+    /**
+     * RequestItems array for BatchWriteItem
+     * https://docs.aws.amazon.com/amazondynamodb/latest/APIReference/API_BatchWriteItem.html
+     * @var array
+     */
+    public $batch_write_request_items = [];
+
     /**
      * ScanIndexForward option.
      */
@@ -312,6 +326,48 @@ public function updateItem($item)
         return $this->process('updateItem', 'processSingleItem');
     }
 
+    public function batchGetItem($keys)
+    {
+        $this->batch_get_keys = $keys;
+
+        return $this->process('batchGetItem', 'processBatchGetItems');
+    }
+
+    public function batchPutItem($items)
+    {
+        $this->batch_write_request_items = array_map(function ($item) {
+            return [
+                'PutRequest' => [
+                    'Item' => $item,
+                ],
+            ];
+        }, $items);
+
+        return $this->batchWriteItem();
+    }
+
+    public function batchDeleteItem($keys)
+    {
+        $this->batch_write_request_items = array_map(function ($key) {
+            return [
+                'DeleteRequest' => [
+                    'Key' => $key,
+                ],
+            ];
+        }, $keys);
+
+        return $this->batchWriteItem();
+    }
+
+    public function batchWriteItem($request_items = [])
+    {
+        if (! empty($request_items)) {
+            $this->batch_write_request_items = $request_items;
+        }
+
+        return $this->process('batchWriteItem', null);
+    }
+
     /**
      * @inheritdoc
      */
@@ -539,6 +595,8 @@ protected function process($query_method, $processor_method)
             $this->grammar->compileKey($this->key),
             $this->grammar->compileItem($this->item),
             $this->grammar->compileUpdates($this->updates),
+            $this->grammar->compileBatchGetRequestItems($this->from, $this->batch_get_keys),
+            $this->grammar->compileBatchWriteRequestItems($this->from, $this->batch_write_request_items),
             $this->grammar->compileDynamodbLimit($this->limit),
             $this->grammar->compileScanIndexForward($this->scan_index_forward),
             $this->grammar->compileExclusiveStartKey($this->exclusive_start_key),

src/Kitar/Dynamodb/Query/Grammar.php

@@ -135,6 +135,54 @@ public function compileUpdates($updates)
         ];
     }
 
+    public function compileBatchGetRequestItems($table_name, $keys)
+    {
+        if (empty($keys)) {
+            return [];
+        }
+
+        $marshaler = $this->marshaler;
+        $marshaled_items = array_map(function ($key) use ($marshaler) {
+            return $marshaler->marshalItem($key);
+        }, $keys);
+
+        $table_name = $this->tablePrefix . $table_name;
+
+        return [
+            'RequestItems' => [
+                $table_name => [
+                    'Keys' => $marshaled_items,
+                ],
+            ]
+        ];
+    }
+
+    public function compileBatchWriteRequestItems($table_name, $request_items)
+    {
+        if (empty($request_items)) {
+            return [];
+        }
+
+        $marshaler = $this->marshaler;
+        $marshaled_items = array_map(function ($request_item) use ($marshaler) {
+            return array_map(function ($request_body) use ($marshaler) {
+                $marshaled = [];
+                foreach ($request_body as $key => $body) {
+                    $marshaled[$key] = $marshaler->marshalItem($body);
+                }
+                return $marshaled;
+            }, $request_item);
+        }, $request_items);
+
+        $table_name = $this->tablePrefix . $table_name;
+
+        return [
+            'RequestItems' => [
+                $table_name => $marshaled_items,
+            ],
+        ];
+    }
+
     /**
      * Compile the Limit attribute.
      *

src/Kitar/Dynamodb/Query/Processor.php

@@ -78,4 +78,29 @@ public function processMultipleItems(Result $awsResponse, $modelClass = null)
             return $item;
         });
     }
+
+    public function processBatchGetItems(Result $awsResponse, $modelClass = null)
+    {
+        $response = $this->unmarshal($awsResponse);
+
+        if (empty($modelClass)) {
+            return $response;
+        }
+
+        $items = collect();
+
+        foreach ($response['Responses'] as $_ => $table_items) {
+            foreach ($table_items as $item) {
+                $item = (new $modelClass)->newFromBuilder($item);
+                $items->push($item);
+            }
+        }
+
+        unset($response['Responses']);
+
+        return $items->map(function ($item) use ($response) {
+            $item->setMeta($response);
+            return $item;
+        });
+    }
 }