Guillermoandrae\DynamoDb\Operation\BatchWriteItemOperation

Complexity

Total Complexity

12

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	2
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	setPutRequest()	0	3	1
A	execute()	0	7	2
A	setDeleteRequest()	0	3	1
A	setWriteRequest()	0	17	5
A	__construct()	0	12	3

<?php

namespace Guillermoandrae\DynamoDb\Operation;

use Aws\DynamoDb\DynamoDbClient;
use Aws\DynamoDb\Exception\DynamoDbException;
use Aws\DynamoDb\Marshaler;
use Guillermoandrae\DynamoDb\Contract\AbstractBatchItemOperation;
use Guillermoandrae\DynamoDb\Factory\ExceptionFactory;
use Guillermoandrae\DynamoDb\Helper\Inflector;

/**
 * BatchWriteItem operation.
 *
 * @author Guillermo A. Fisher <me@guillermoandraefisher.com>
 * @link https://docs.aws.amazon.com/aws-sdk-php/v3/api/api-dynamodb-2012-08-10.html#batchwriteitem
 */
final class BatchWriteItemOperation extends AbstractBatchItemOperation
{
    /**
     * Registers the DynamoDb client, Marshaler, and the mapping of tables and primary keys with this object.
     *
     * @param DynamoDbClient $client The DynamoDB client.
     * @param Marshaler $marshaler The Marshaler.
     * @param array $deleteItems OPTIONAL The table(s) and associated primary key attributes to use when deleting.
     * @param array $putItems OPTIONAL The table(s) and associated items to add.
     */
    public function __construct(
        DynamoDbClient $client,
        Marshaler      $marshaler,
        array          $deleteItems = [],
        array          $putItems = []
    ) {
        parent::__construct($client, $marshaler);
        if (!empty($deleteItems)) {
            $this->setDeleteRequest($deleteItems);
        }
        if (!empty($putItems)) {
            $this->setPutRequest($putItems);
        }
    }

    /**
     * Registers the DeleteRequest options.
     *
     * @param array $deleteItems The table(s) and associated primary key attributes to use when deleting.
     * @return BatchWriteItemOperation
     * @see BatchWriteItemOperation::setWriteRequest()
     */
    public function setDeleteRequest(array $deleteItems): BatchWriteItemOperation
    {
        return $this->setWriteRequest('Delete', $deleteItems);
    }

    /**
     * Registers the PutRequest options.
     *
     * @param array $putItems The table(s) and associated items to add.
     * @return BatchWriteItemOperation
     * @see BatchWriteItemOperation::setWriteRequest()
     */
    public function setPutRequest(array $putItems): BatchWriteItemOperation
    {
        return $this->setWriteRequest('Put', $putItems);
    }

    /**
     * Registers request options.
     *
     * @param string $type The request type (Put or Delete).
     * @param array $requestItems The table(s) and associated options.
     * @return BatchWriteItemOperation
     */
    private function setWriteRequest(string $type, array $requestItems): BatchWriteItemOperation
    {
        $type = Inflector::camelize($type);
        $keyName = ($type == 'Put') ? 'Item' : 'Key';
        foreach ($requestItems as $tableName => $items) {
            if (!isset($this->requestItems[$tableName])) {
                $this->requestItems[$tableName] = [];
            }
            foreach ($items as $item) {
                $this->requestItems[$tableName][] = [
                    sprintf('%sRequest', $type) => [
                        $keyName => $this->getMarshaler()->marshalItem($item)
                    ]
                ];
            }
        }
        return $this;
    }

    public function execute(): bool
    {
        try {
            $this->getClient()->batchWriteItem($this->toArray());
            return true;
        } catch (DynamoDbException $ex) {
            throw ExceptionFactory::factory($ex);
        }
    }
}