TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad\OneToManyDataLoader

Complexity

Total Complexity

6

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	load()	0	10	2
A	__construct()	0	5	1
A	get()	0	10	3

<?php


namespace TheCodingMachine\TDBM\QueryFactory\SmartEagerLoad;

use Doctrine\DBAL\Connection;
use TheCodingMachine\TDBM\TDBMException;

class OneToManyDataLoader
{
    /**
     * @var Connection
     */
    private $connection;
    /**
     * @var string
     */
    private $sql;
    /**
     * @var string
     */
    private $fkColumn;
    /**
     * @var array<string, array<int, array<string, mixed>>> Array of rows, indexed by foreign key.
     */
    private $data;

    public function __construct(Connection $connection, string $sql, string $fkColumn)
    {
        $this->connection = $connection;
        $this->sql = $sql;
        $this->fkColumn = $fkColumn;
    }

    /**
     * @return array<string, array<string, mixed>> Rows, indexed by ID.
     */
    private function load(): array
    {
        $results = $this->connection->fetchAll($this->sql);

        $data = [];
        foreach ($results as $row) {
            $data[$row[$this->fkColumn]][] = $row;
        }

        return $data;
    }

    /**
     * Returns the DB row with the given ID.
     * Loads all rows if necessary.
     * Throws an exception if nothing found.
     *
     * @param string $id
     * @return array<string, mixed>
     */
    public function get(string $id): array
    {
        if ($this->data === null) {
            $this->data = $this->load();
        }

        if (!isset($this->data[$id])) {
            return [];
        }
        return $this->data[$id];
    }
}