AbstractPdoCommand::queryInternal() - Code Metrics - Inspection of "Logger context" - yiisoft/db - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — master (#794)

by Def

created 2024-01-04 09:59 UTC

AbstractPdoCommand::queryInternal() A

↳ Parent: AbstractPdoCommand

Complexity

Conditions	3
Paths	4

Size

Total Lines	27
Code Lines	13

Duplication

Lines	0
Ratio	0 %

Importance

Changes	1
Bugs	0	Features	0

Metric	Value
cc	3
eloc	13
c	1
b	0
f	0
nc	4
nop	1
dl	0
loc	27
rs	9.8333

<?php

declare(strict_types=1);

namespace Yiisoft\Db\Driver\Pdo;

use PDO;
use PDOException;
use PDOStatement;
use Psr\Log\LoggerAwareInterface;
use Psr\Log\LoggerAwareTrait;
use Psr\Log\LogLevel;
use Throwable;
use Yiisoft\Db\Command\AbstractCommand;
use Yiisoft\Db\Command\Param;
use Yiisoft\Db\Command\ParamInterface;
use Yiisoft\Db\Exception\ConvertException;
use Yiisoft\Db\Exception\Exception;
use Yiisoft\Db\Exception\InvalidParamException;
use Yiisoft\Db\Profiler\Context\CommandContext;
use Yiisoft\Db\Profiler\ProfilerAwareInterface;
use Yiisoft\Db\Profiler\ProfilerAwareTrait;
use Yiisoft\Db\Query\Data\DataReader;
use Yiisoft\Db\QueryBuilder\QueryBuilderInterface;

/**
 * Represents a database command that can be executed using a PDO (PHP Data Object) database connection.
 *
 * It's an abstract class that provides a common interface for building and executing various types of statements
 * such as {@see cancel()}, {@see execute()}, {@see insert()}, {@see update()}, {@see delete()}, etc., using a PDO
 * connection.
 *
 * It also provides methods for binding parameter values and retrieving query results.
 */
abstract class AbstractPdoCommand extends AbstractCommand implements PdoCommandInterface, LoggerAwareInterface, ProfilerAwareInterface
{
    use LoggerAwareTrait;
    use ProfilerAwareTrait;

    /**
     * @var PDOStatement|null Represents a prepared statement and, after the statement is executed, an associated
     * result set.
     *
     * @link https://www.php.net/manual/en/class.pdostatement.php
     */
    protected PDOStatement|null $pdoStatement = null;

    public function __construct(protected PdoConnectionInterface $db)
    {
    }

    /**
     * This method mainly sets {@see pdoStatement} to be `null`.
     */
    public function cancel(): void
    {
        $this->pdoStatement = null;
    }

    public function getPdoStatement(): PDOStatement|null
    {
        return $this->pdoStatement;
    }

    public function bindParam(
        int|string $name,
        mixed &$value,
        int|null $dataType = null,
        int|null $length = null,
        mixed $driverOptions = null
    ): static {
        $this->prepare();

        if ($dataType === null) {
            $dataType = $this->db->getSchema()->getDataType($value);
        }

        if ($length === null) {
            $this->pdoStatement?->bindParam($name, $value, $dataType);
        } elseif ($driverOptions === null) {
            $this->pdoStatement?->bindParam($name, $value, $dataType, $length);
        } else {
            $this->pdoStatement?->bindParam($name, $value, $dataType, $length, $driverOptions);
        }

        return $this;
    }

    public function bindValue(int|string $name, mixed $value, int|null $dataType = null): static
    {
        if ($dataType === null) {
            $dataType = $this->db->getSchema()->getDataType($value);
        }

        $this->params[$name] = new Param($value, $dataType);

        return $this;
    }

    public function bindValues(array $values): static
    {
        if (empty($values)) {
            return $this;
        }

        /**
         * @psalm-var array<string, int>|ParamInterface|int $value
         */
        foreach ($values as $name => $value) {
            if ($value instanceof ParamInterface) {
                $this->params[$name] = $value;
            } else {
                $type = $this->db->getSchema()->getDataType($value);
                $this->params[$name] = new Param($value, $type);
            }
        }

        return $this;
    }

    public function prepare(bool|null $forRead = null): void
    {
        if (isset($this->pdoStatement)) {
            $this->bindPendingParams();

            return;
        }

        $sql = $this->getSql();

        /**
         * If SQL is empty, there will be {@see \ValueError} on prepare pdoStatement.
         *
         * @link https://php.watch/versions/8.0/ValueError
         */
        if ($sql === '') {
            return;
        }

        $pdo = $this->db->getActivePDO($sql, $forRead);

        try {
            $this->pdoStatement = $pdo?->prepare($sql);
            $this->bindPendingParams();
        } catch (PDOException $e) {
            $message = $e->getMessage() . "\nFailed to prepare SQL: $sql";
            /** @psalm-var array|null $errorInfo */
            $errorInfo = $e->errorInfo ?? null;

            throw new Exception($message, $errorInfo, $e);
        }
    }

    /**
     * Binds pending parameters registered via {@see bindValue()} and {@see bindValues()}.
     *
     * Note that this method requires an active {@see pdoStatement}.
     */
    protected function bindPendingParams(): void
    {
        foreach ($this->params as $name => $value) {
            $this->pdoStatement?->bindValue($name, $value->getValue(), $value->getType());
        }
    }

    protected function getQueryBuilder(): QueryBuilderInterface
    {
        return $this->db->getQueryBuilder();
    }

    protected function getQueryMode(int $queryMode): string
    {
        return match ($queryMode) {
            self::QUERY_MODE_EXECUTE => 'execute',
            self::QUERY_MODE_ROW => 'queryOne',
            self::QUERY_MODE_ALL => 'queryAll',
            self::QUERY_MODE_COLUMN => 'queryColumn',
            self::QUERY_MODE_CURSOR => 'query',
            self::QUERY_MODE_SCALAR => 'queryScalar',
            self::QUERY_MODE_ROW | self::QUERY_MODE_EXECUTE => 'insertWithReturningPks'
        };
    }

    /**
     * Executes a prepared statement.
     *
     * It's a wrapper around {@see PDOStatement::execute()} to support transactions and retry handlers.
     *
     * @param string|null $rawSql Deprecated. Use `null` value. Will be removed in version 2.0.0.
     *
     * @throws Exception
     * @throws Throwable
     */
    protected function internalExecute(string|null $rawSql): void
    {
        $attempt = 0;

        while (true) {
            try {
                if (
                    ++$attempt === 1
                    && $this->isolationLevel !== null
                    && $this->db->getTransaction() === null
                ) {
                    $this->db->transaction(
                        fn () => $this->internalExecute($rawSql),
                        $this->isolationLevel
                    );
                } else {
                    $this->pdoStatement?->execute();
                }
                break;
            } catch (PDOException $e) {
                $rawSql = $rawSql ?: $this->getRawSql();
                $e = (new ConvertException($e, $rawSql))->run();

                if ($this->retryHandler === null || !($this->retryHandler)($e, $attempt)) {
                    throw $e;
                }
            }
        }
    }

    /**
     * @throws InvalidParamException
     */
    protected function internalGetQueryResult(int $queryMode): mixed
    {
        if ($queryMode === self::QUERY_MODE_CURSOR) {
            return new DataReader($this);
        }

        if ($queryMode === self::QUERY_MODE_EXECUTE) {
            return $this->pdoStatement?->rowCount() ?? 0;
        }

        if ($this->is($queryMode, self::QUERY_MODE_ROW)) {
            /** @psalm-var array|false $result */
            $result = $this->pdoStatement?->fetch(PDO::FETCH_ASSOC);
        } elseif ($this->is($queryMode, self::QUERY_MODE_SCALAR)) {
            /** @psalm-var mixed $result */
            $result = $this->pdoStatement?->fetchColumn();
        } elseif ($this->is($queryMode, self::QUERY_MODE_COLUMN)) {
            /** @psalm-var mixed $result */
            $result = $this->pdoStatement?->fetchAll(PDO::FETCH_COLUMN);
        } elseif ($this->is($queryMode, self::QUERY_MODE_ALL)) {
            /** @psalm-var mixed $result */
            $result = $this->pdoStatement?->fetchAll(PDO::FETCH_ASSOC);
        } else {
            throw new InvalidParamException("Unknown query mode '$queryMode'");
        }

        $this->pdoStatement?->closeCursor();

        return $result;
    }

    protected function queryInternal(int $queryMode): mixed
    {
        $logCategory = self::class . '::' . $this->getQueryMode($queryMode);

        if ($this->logger !== null) {
            $rawSql = $this->getRawSql();
            $this->logger->log(LogLevel::INFO, $rawSql, [$logCategory, LogTypes::KEY => LogTypes::TYPE_QUERY]);
        }

        $queryContext = new CommandContext(__METHOD__, $logCategory, $this->getSql(), $this->getParams());

        /**
         * @psalm-var string $rawSql
         * @psalm-suppress RedundantConditionGivenDocblockType
         * @psalm-suppress DocblockTypeContradiction
         */
        $this->profiler?->begin($rawSql ??= $this->getRawSql(), $queryContext);

        try {
            /** @psalm-var mixed $result */
            $result = parent::queryInternal($queryMode);
        } catch (Throwable $e) {
            $this->profiler?->end($rawSql, $queryContext->setException($e));
            throw $e;
        }
        $this->profiler?->end($rawSql, $queryContext);

        return $result;
    }

    /**
     * Refreshes table schema, which was marked by {@see requireTableSchemaRefresh()}.
     */
    protected function refreshTableSchema(): void
    {
        if ($this->refreshTableName !== null) {
            $this->db->getSchema()->refreshTableSchema($this->refreshTableName);
        }
    }
}


1			<?php
2
3			declare(strict_types=1);
4
5			namespace Yiisoft\Db\Driver\Pdo;
6
7			use PDO;
8			use PDOException;
9			use PDOStatement;
10			use Psr\Log\LoggerAwareInterface;
11			use Psr\Log\LoggerAwareTrait;
12			use Psr\Log\LogLevel;
13			use Throwable;
14			use Yiisoft\Db\Command\AbstractCommand;
15			use Yiisoft\Db\Command\Param;
16			use Yiisoft\Db\Command\ParamInterface;
17			use Yiisoft\Db\Exception\ConvertException;
18			use Yiisoft\Db\Exception\Exception;
19			use Yiisoft\Db\Exception\InvalidParamException;
20			use Yiisoft\Db\Profiler\Context\CommandContext;
21			use Yiisoft\Db\Profiler\ProfilerAwareInterface;
22			use Yiisoft\Db\Profiler\ProfilerAwareTrait;
23			use Yiisoft\Db\Query\Data\DataReader;
24			use Yiisoft\Db\QueryBuilder\QueryBuilderInterface;
25
26			/**
27			* Represents a database command that can be executed using a PDO (PHP Data Object) database connection.
28			*
29			* It's an abstract class that provides a common interface for building and executing various types of statements
30			* such as {@see cancel()}, {@see execute()}, {@see insert()}, {@see update()}, {@see delete()}, etc., using a PDO
31			* connection.
32			*
33			* It also provides methods for binding parameter values and retrieving query results.
34			*/
35			abstract class AbstractPdoCommand extends AbstractCommand implements PdoCommandInterface, LoggerAwareInterface, ProfilerAwareInterface
36			{
37			use LoggerAwareTrait;
38			use ProfilerAwareTrait;
39
40			/**
41			* @var PDOStatement\|null Represents a prepared statement and, after the statement is executed, an associated
42			* result set.
43			*
44			* @link https://www.php.net/manual/en/class.pdostatement.php
45			*/
46			protected PDOStatement\|null $pdoStatement = null;
47
48			public function __construct(protected PdoConnectionInterface $db)
49			{
50			}
51
52			/**
53			* This method mainly sets {@see pdoStatement} to be `null`.
54			*/
55			public function cancel(): void
56			{
57			$this->pdoStatement = null;
58			}
59
60			public function getPdoStatement(): PDOStatement\|null
61			{
62			return $this->pdoStatement;
63			}
64
65			public function bindParam(
66			int\|string $name,
67			mixed &$value,
68			int\|null $dataType = null,
69			int\|null $length = null,
70			mixed $driverOptions = null
71			): static {
72			$this->prepare();
73
74			if ($dataType === null) {
75			$dataType = $this->db->getSchema()->getDataType($value);
76			}
77
78			if ($length === null) {
79			$this->pdoStatement?->bindParam($name, $value, $dataType);
80			} elseif ($driverOptions === null) {
81			$this->pdoStatement?->bindParam($name, $value, $dataType, $length);
82			} else {
83			$this->pdoStatement?->bindParam($name, $value, $dataType, $length, $driverOptions);
84			}
85
86			return $this;
87			}
88
89			public function bindValue(int\|string $name, mixed $value, int\|null $dataType = null): static
90			{
91			if ($dataType === null) {
92			$dataType = $this->db->getSchema()->getDataType($value);
93			}
94
95			$this->params[$name] = new Param($value, $dataType);
96
97			return $this;
98			}
99
100			public function bindValues(array $values): static
101			{
102			if (empty($values)) {
103			return $this;
104			}
105
106			/**
107			* @psalm-var array<string, int>\|ParamInterface\|int $value
108			*/
109			foreach ($values as $name => $value) {
110			if ($value instanceof ParamInterface) {
111			$this->params[$name] = $value;
112			} else {
113			$type = $this->db->getSchema()->getDataType($value);
114			$this->params[$name] = new Param($value, $type);
115			}
116			}
117
118			return $this;
119			}
120
121			public function prepare(bool\|null $forRead = null): void
122			{
123			if (isset($this->pdoStatement)) {
124			$this->bindPendingParams();
125
126			return;
127			}
128
129			$sql = $this->getSql();
130
131			/**
132			* If SQL is empty, there will be {@see \ValueError} on prepare pdoStatement.
133			*
134			* @link https://php.watch/versions/8.0/ValueError
135			*/
136			if ($sql === '') {
137			return;
138			}
139
140			$pdo = $this->db->getActivePDO($sql, $forRead);
141
142			try {
143			$this->pdoStatement = $pdo?->prepare($sql);
144			$this->bindPendingParams();
145			} catch (PDOException $e) {
146			$message = $e->getMessage() . "\nFailed to prepare SQL: $sql";
147			/** @psalm-var array\|null $errorInfo */
148			$errorInfo = $e->errorInfo ?? null;
149
150			throw new Exception($message, $errorInfo, $e);
151			}
152			}
153
154			/**
155			* Binds pending parameters registered via {@see bindValue()} and {@see bindValues()}.
156			*
157			* Note that this method requires an active {@see pdoStatement}.
158			*/
159			protected function bindPendingParams(): void
160			{
161			foreach ($this->params as $name => $value) {
162			$this->pdoStatement?->bindValue($name, $value->getValue(), $value->getType());
163			}
164			}
165
166			protected function getQueryBuilder(): QueryBuilderInterface
167			{
168			return $this->db->getQueryBuilder();
169			}
170
171			protected function getQueryMode(int $queryMode): string
172			{
173			return match ($queryMode) {
174			self::QUERY_MODE_EXECUTE => 'execute',
175			self::QUERY_MODE_ROW => 'queryOne',
176			self::QUERY_MODE_ALL => 'queryAll',
177			self::QUERY_MODE_COLUMN => 'queryColumn',
178			self::QUERY_MODE_CURSOR => 'query',
179			self::QUERY_MODE_SCALAR => 'queryScalar',
180			self::QUERY_MODE_ROW \| self::QUERY_MODE_EXECUTE => 'insertWithReturningPks'
181			};
182			}
183
184			/**
185			* Executes a prepared statement.
186			*
187			* It's a wrapper around {@see PDOStatement::execute()} to support transactions and retry handlers.
188			*
189			* @param string\|null $rawSql Deprecated. Use `null` value. Will be removed in version 2.0.0.
190			*
191			* @throws Exception
192			* @throws Throwable
193			*/
194			protected function internalExecute(string\|null $rawSql): void
195			{
196			$attempt = 0;
197
198			while (true) {
199			try {
200			if (
201			++$attempt === 1
202			&& $this->isolationLevel !== null
203			&& $this->db->getTransaction() === null
204			) {
205			$this->db->transaction(
206			fn () => $this->internalExecute($rawSql),
207			$this->isolationLevel
208			);
209			} else {
210			$this->pdoStatement?->execute();
211			}
212			break;
213			} catch (PDOException $e) {
214			$rawSql = $rawSql ?: $this->getRawSql();
215			$e = (new ConvertException($e, $rawSql))->run();
216
217			if ($this->retryHandler === null \|\| !($this->retryHandler)($e, $attempt)) {
218			throw $e;
219			}
220			}
221			}
222			}
223
224			/**
225			* @throws InvalidParamException
226			*/
227			protected function internalGetQueryResult(int $queryMode): mixed
228			{
229			if ($queryMode === self::QUERY_MODE_CURSOR) {
230			return new DataReader($this);
231			}
232
233			if ($queryMode === self::QUERY_MODE_EXECUTE) {
234			return $this->pdoStatement?->rowCount() ?? 0;
235			}
236
237			if ($this->is($queryMode, self::QUERY_MODE_ROW)) {
238			/** @psalm-var array\|false $result */
239			$result = $this->pdoStatement?->fetch(PDO::FETCH_ASSOC);
240			} elseif ($this->is($queryMode, self::QUERY_MODE_SCALAR)) {
241			/** @psalm-var mixed $result */
242			$result = $this->pdoStatement?->fetchColumn();
243			} elseif ($this->is($queryMode, self::QUERY_MODE_COLUMN)) {
244			/** @psalm-var mixed $result */
245			$result = $this->pdoStatement?->fetchAll(PDO::FETCH_COLUMN);
246			} elseif ($this->is($queryMode, self::QUERY_MODE_ALL)) {
247			/** @psalm-var mixed $result */
248			$result = $this->pdoStatement?->fetchAll(PDO::FETCH_ASSOC);
249			} else {
250			throw new InvalidParamException("Unknown query mode '$queryMode'");
251			}
252
253			$this->pdoStatement?->closeCursor();
254
255			return $result;
256			}
257
258			protected function queryInternal(int $queryMode): mixed
259			{
260			$logCategory = self::class . '::' . $this->getQueryMode($queryMode);
261
262			if ($this->logger !== null) {
263			$rawSql = $this->getRawSql();
264			$this->logger->log(LogLevel::INFO, $rawSql, [$logCategory, LogTypes::KEY => LogTypes::TYPE_QUERY]);
265			}
266
267			$queryContext = new CommandContext(__METHOD__, $logCategory, $this->getSql(), $this->getParams());
268
269			/**
270			* @psalm-var string $rawSql
271			* @psalm-suppress RedundantConditionGivenDocblockType
272			* @psalm-suppress DocblockTypeContradiction
273			*/
274			$this->profiler?->begin($rawSql ??= $this->getRawSql(), $queryContext);
			0 ignored issues – show Comprehensibility Best Practice introduced 2023-11-27 08:17 UTC by Report Bug Copy Issue Report The variable `$rawSql` does not seem to be defined for all execution paths leading up to this point. Loading history...
275			try {
276			/** @psalm-var mixed $result */
277			$result = parent::queryInternal($queryMode);
278			} catch (Throwable $e) {
279			$this->profiler?->end($rawSql, $queryContext->setException($e));
280			throw $e;
281			}
282			$this->profiler?->end($rawSql, $queryContext);
283
284			return $result;
285			}
286
287			/**
288			* Refreshes table schema, which was marked by {@see requireTableSchemaRefresh()}.
289			*/
290			protected function refreshTableSchema(): void
291			{
292			if ($this->refreshTableName !== null) {
293			$this->db->getSchema()->refreshTableSchema($this->refreshTableName);
294			}
295			}
296			}
297

yiisoft / db

Pull Request — master (#794)

AbstractPdoCommand::queryInternal() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like