LibraryCatalog\Service\Repository\PdoTrait

Complexity

Total Complexity

18

Size/Duplication

Total Lines	131
Duplicated Lines	0 %

Test Coverage

Coverage

74.51%

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	fetchOne()	0	13	3
A	fetchList()	0	30	6
A	prepareConnection()	0	13	2
B	insert()	0	30	7

<?php

declare(strict_types=1);

namespace LibraryCatalog\Service\Repository;

trait PdoTrait
{
    /** @var \PDO */
    protected \PDO $pdo;
    /** @var bool */
    protected bool $pdoReady = false;
    /** @var \Closure */
    protected \Closure $connection;

    /**
     * @param string $host
     * @param string $user
     * @param string $password
     * @param string $dbname
     */
    protected function prepareConnection(string $host, string $user, string $password, string $dbname): void
    {
        $this->connection = function () use ($host, $user, $password, $dbname): \PDO {
            if (!$this->pdoReady) {
                $this->pdo = new \PDO(
                    sprintf('mysql:dbname=%s;host=%s', $dbname, $host),
                    $user,
                    $password,
                    array(\PDO::ATTR_PERSISTENT => true)
                );
                $this->pdoReady = true;
            }
            return $this->pdo;
        };
    }

    /**
     * @param string $table
     * @param mixed $id
     * @return array|null
     * @throws Exception
     */
    protected function fetchOne(string $table, $id): ?array
    {
        /** @var \PDO $pdo */
        $pdo = ($this->connection)();

        $stmt = $pdo->prepare("SELECT * FROM $table WHERE id = ? LIMIT 1");
        if ($stmt === false) {
            throw new Exception("Can not init PDO: " . json_encode($pdo->errorInfo()), (int)$pdo->errorCode());
        }
        $stmt->execute([$id]);
        $res = $stmt->fetch(\PDO::FETCH_ASSOC);
        // Convert false to null.
        return $res ?: null;
    }

    /**
     * @param string $table
     * @param string $condition
     * @param array $conditionData
     * @param int $limit
     * @param string $order
     * @return array
     * @throws Exception
     */
    protected function fetchList(
        string $table,
        string $condition = '',
        array $conditionData = [],
        int $limit = 1000,
        string $order = ''
    ): array {
        /** @var \PDO $pdo */
        $pdo = ($this->connection)();

        // Prepare SQL with params.
        $sql = "SELECT * FROM $table";
        if ($condition != '') {
            $sql .= ' WHERE ' . $condition;
        }
        if ($order != '') {
            $sql .= ' ORDER BY ' . $order;
        }
        if ($limit > 0) {
            $sql .= ' LIMIT ' . $limit;
        }

        $stmt = $pdo->prepare($sql);
        if ($stmt === false) {
            throw new Exception("Can not init PDO: " . json_encode($pdo->errorInfo()), (int)$pdo->errorCode());
        }
        $stmt->execute($conditionData);
        $res = $stmt->fetchAll(\PDO::FETCH_ASSOC);
        // Convert false to empty result.
        return $res ?: [];
    }

    /**
     * Inserts data and returns its ID.
     *
     * @param string $table
     * @param array $data
     * @return mixed
     * @throws Exception
     */
    protected function insert(string $table, array $data)
    {
        $rowNames = '';
        $rowValues = '';
        foreach ($data as $name => $value) {
            if ($value !== null) {
                if ($rowNames !== '') {
                    $rowNames .= ',';
                }
                if ($rowValues !== '') {
                    $rowValues .= ',';
                }
                $rowNames .= $name;
                $rowValues .= ':' . $name;
            } else {
                unset($data[$name]);
            }
        }

        /** @var \PDO $pdo */
        $pdo = ($this->connection)();
        $stmt = $pdo->prepare("INSERT INTO $table ($rowNames) VALUES ($rowValues)");
        if ($stmt === false) {
            throw new Exception("Can not init PDO: " . json_encode($pdo->errorInfo()), (int)$pdo->errorCode());
        }
        if (!$stmt->execute($data)) {
            throw new Exception("Can not insert data in PDO: " . json_encode($pdo->errorInfo()), (int)$pdo->errorCode());
        }

        return $pdo->lastInsertId();
    }
}