DBFaker\Generators\ForeignKeyColumnGenerator

Complexity

Total Complexity

3

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__invoke()	0	8	2
A	__construct()	0	5	1

<?php

namespace DBFaker\Generators;

use DBFaker\Helpers\DBFakerSchemaManager;
use DBFaker\Helpers\PrimaryKeyRegistry;
use DBFaker\Helpers\SchemaHelper;
use Doctrine\DBAL\Schema\Column;
use Doctrine\DBAL\Schema\ForeignKeyConstraint;
use Doctrine\DBAL\Schema\Table;

class ForeignKeyColumnGenerator implements FakeDataGeneratorInterface
{
    /**
     * @var Column
     */
    private $foreignColumn;

    /**
     * @var PrimaryKeyRegistry
     */
    private $foreignPkRegistry;

    /**
     * @var bool
     */
    private $generateUniqueValues;

    /**
     * @var mixed[]
     */
    private $alreadyGeneratedValues = [];


    /**
     * ForeignKeyColumnGenerator constructor.
     * @param Table $table
     * @param Column $column
     * @param PrimaryKeyRegistry $foreignPkRegistry
     * @param ForeignKeyConstraint $fk
     * @param DBFakerSchemaManager $schemaManager
     * @param SchemaHelper $schemaHelper
     * @throws \DBFaker\Exceptions\SchemaLogicException
     * @throws \Doctrine\DBAL\DBALException
     * @throws \Doctrine\DBAL\Schema\SchemaException
     */
    public function __construct(Table $table, Column $column, PrimaryKeyRegistry $foreignPkRegistry, ForeignKeyConstraint $fk, DBFakerSchemaManager $schemaManager, SchemaHelper $schemaHelper)
    {
        $this->foreignColumn = $schemaManager->getForeignColumn($table, $column);
        $this->foreignPkRegistry = $foreignPkRegistry;
        $this->generateUniqueValues = $schemaHelper->isForeignKeyAlsoUniqueIndex($fk);
    }

    /**
     * @return mixed
     * @throws \Doctrine\DBAL\DBALException
     * @throws \Exception
     */
    public function __invoke()
    {
        $randomPk = $this->foreignPkRegistry->loadValuesFromTable()->getRandomValue($this->alreadyGeneratedValues);
        $value = $randomPk[$this->foreignColumn->getName()];
        if ($this->generateUniqueValues) {
            $this->alreadyGeneratedValues[] = $randomPk;
        }
        return $value;
    }
}