DBFaker\Generators\UniqueAbleGenerator

Complexity

Total Complexity

8

Size/Duplication

Total Lines	78
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	storeObjectInGeneratedValues()	0	4	2
A	isUnique()	0	10	2
A	__invoke()	0	14	3
A	__construct()	0	4	1

<?php
namespace DBFaker\Generators;

use DBFaker\DBFaker;
use DBFaker\Exceptions\MaxNbOfIterationsForUniqueValueException;
use Doctrine\DBAL\Schema\Column;

abstract class UniqueAbleGenerator implements FakeDataGeneratorInterface
{

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

    /**
     * @var Column
     */
    private $column;

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

    /**
     * ComplexObjectGenerator constructor.
     * @param Column $column
     * @param bool $generateUniqueValues
     */
    public function __construct(Column $column, bool $generateUniqueValues = false)
    {
        $this->generateUniqueValues = $generateUniqueValues;
        $this->column = $column;
    }

    /**
     * @return mixed
     */
    public function __invoke()
    {
        $object = $this->generateRandomValue($this->column);
        $iterations = 1;
        while (!$this->isUnique($object)) {
            $object = $this->generateRandomValue($this->column);
            $iterations++;
            if ($iterations > DBFaker::MAX_ITERATIONS_FOR_UNIQUE_VALUE) {
                throw new MaxNbOfIterationsForUniqueValueException('Unable to generate a unique value in less then maximumn allowed iterations.');
            }
        }
        $this->storeObjectInGeneratedValues($object);

        return $object;
    }

    /**
     * @param Column $column
     * @return mixed
     */
    abstract protected function generateRandomValue(Column $column);

    /**
     * @param mixed $object
     */
    private function storeObjectInGeneratedValues($object) : void
    {
        if ($this->generateUniqueValues) {
            $this->generatedValues[] = $object;
        }
    }

    /**
     * @param mixed $object
     * @return bool
     */
    private function isUnique($object) : bool
    {
        if (!$this->generateUniqueValues) {
            return true;
        }

        $filtered = array_filter($this->generatedValues, function ($value) use ($object) {
            return $object === $value;
        });
        return count($filtered) === 0;
    }
}