ServiceBus\Storage\Sql\AmpPosgreSQL\AmpPostgreSQLResultSet

Complexity

Total Complexity

18

Size/Duplication

Total Lines	141
Duplicated Lines	0 %

Test Coverage

Coverage

96.77%

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	affectedRows()	0	17	4
A	advance()	0	17	3
A	lastInsertId()	0	35	6
A	getCurrent()	0	20	4

<?php

/**
 * SQL databases adapters implementation.
 *
 * @author  Maksim Masiukevich <dev@async-php.com>
 * @license MIT
 * @license https://opensource.org/licenses/MIT
 */

declare(strict_types = 1);

namespace ServiceBus\Storage\Sql\AmpPosgreSQL;

use function Amp\call;
use Amp\Postgres\PgSqlCommandResult;
use Amp\Postgres\PooledResultSet;
use Amp\Postgres\PqCommandResult;
use Amp\Promise;
use Amp\Sql\ResultSet as AmpResultSet;
use Amp\Success;
use ServiceBus\Storage\Common\Exceptions\ResultSetIterationFailed;
use ServiceBus\Storage\Common\ResultSet;

/**
 *
 */
class AmpPostgreSQLResultSet implements ResultSet
{
    /**
     * @var AmpResultSet|PgSqlCommandResult|PooledResultSet|PqCommandResult
     */
    private $originalResultSet;

    /**
     * @var bool
     */
    private $advanceCalled = false;

    /**
     * @noinspection   PhpDocSignatureInspection
     *
     * @psalm-suppress TypeCoercion Assume a different data type
     *
     * @param AmpResultSet|PgSqlCommandResult|PooledResultSet|PqCommandResult $originalResultSet
     */
    public function __construct(object $originalResultSet)
    {
        $this->originalResultSet = $originalResultSet;
    }

    /**
     * {@inheritdoc}
     */
    public function advance(): Promise
    {
        $this->advanceCalled = true;

        try
        {
            if ($this->originalResultSet instanceof AmpResultSet)
            {
                return $this->originalResultSet->advance();
            }

            return new Success(false);
        }
        // @codeCoverageIgnoreStart
        catch (\Throwable $throwable)
        {
            throw new ResultSetIterationFailed($throwable->getMessage(), (int) $throwable->getCode(), $throwable);
        }
        // @codeCoverageIgnoreEnd
    }

    /**
     * {@inheritdoc}
     */
    public function getCurrent(): ?array
    {
        try
        {
            if (
                $this->originalResultSet instanceof PgSqlCommandResult ||
                $this->originalResultSet instanceof PqCommandResult
            ) {
                return null;
            }

            /** @var array<string, float|int|resource|string|null>|null $data */
            $data = $this->originalResultSet->getCurrent();

            return $data;
        }
        // @codeCoverageIgnoreStart
        catch (\Throwable $throwable)
        {
            throw new ResultSetIterationFailed($throwable->getMessage(), (int) $throwable->getCode(), $throwable);
        }
        // @codeCoverageIgnoreEnd
    }

    /**
     * {@inheritdoc}
     *
     * @psalm-suppress MixedReturnTypeCoercion
     */
    public function lastInsertId(?string $sequence = null): Promise
    {
        return call(
            function (): \Generator
            {
                try
                {
                    if ($this->originalResultSet instanceof PooledResultSet)
                    {
                        if ($this->advanceCalled === false)
                        {
                            yield $this->originalResultSet->advance();

                            $this->advanceCalled = true;
                        }

                        /** @var array<string, mixed> $result */
                        $result = $this->originalResultSet->getCurrent();

                        if (0 !== \count($result))
                        {
                            /** @var bool|int|string $value */
                            $value = \reset($result);

                            if (false !== $value)
                            {
                                return (string) $value;
                            }
                        }
                    }
                }
                // @codeCoverageIgnoreStart
                catch (\Throwable $throwable)
                {
                    throw new ResultSetIterationFailed($throwable->getMessage(), (int) $throwable->getCode(), $throwable);
                }
                // @codeCoverageIgnoreEnd
            }
        );
    }

    /**
     * {@inheritdoc}
     */
    public function affectedRows(): int
    {
        try
        {
            if (
                $this->originalResultSet instanceof PgSqlCommandResult ||
                $this->originalResultSet instanceof PqCommandResult
            ) {
                return $this->originalResultSet->getAffectedRowCount();
            }

            return 0;
        }
        // @codeCoverageIgnoreStart
        catch (\Throwable $throwable)
        {
            throw new ResultSetIterationFailed($throwable->getMessage(), (int) $throwable->getCode(), $throwable);
        }
        // @codeCoverageIgnoreEnd
    }
}