Ray\AuraSqlModule\AuraSqlConnectionInterceptor

Complexity

Total Complexity

4

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	3
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getConnection()	0	8	2
A	invoke()	0	9	1
A	__construct()	0	6	1

<?php

declare(strict_types=1);

namespace Ray\AuraSqlModule;

use Aura\Sql\ConnectionLocatorInterface;
use Aura\Sql\ExtendedPdoInterface;
use Override;
use Ray\Aop\MethodInterceptor;
use Ray\Aop\MethodInvocation;
use Ray\AuraSqlModule\Annotation\Read;
use ReflectionProperty;

use function in_array;

final class AuraSqlConnectionInterceptor implements MethodInterceptor
{
    public const PROP = 'pdo';

    /** @phpstan-param array<string> $readsMethods */
    public function __construct(
        private ConnectionLocatorInterface $connectionLocator,
        /** @var string[] */
        #[Read]
        private array $readsMethods
    ) {
    }

    /**
     * {@inheritDoc}
     */
    #[Override]
    public function invoke(MethodInvocation $invocation)
    {
        $connection = $this->getConnection($invocation);
        $object = $invocation->getThis();
        $ref = new ReflectionProperty($object, self::PROP);
        $ref->setValue($object, $connection);

        return $invocation->proceed();
    }

    /** @param MethodInvocation<object> $invocation */
    private function getConnection(MethodInvocation $invocation): ExtendedPdoInterface
    {
        $methodName = $invocation->getMethod()->name;
        if (in_array($methodName, $this->readsMethods, true)) {
            return $this->connectionLocator->getRead();
        }

        return $this->connectionLocator->getWrite();
    }
}