Shippinno\Job\Infrastructure\Persistence\Doctrine\ManagerRegistryAwareTrait

Complexity

Total Complexity

7

Size/Duplication

Total Lines	50
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

16.66%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	setManagerRegistry()	0	4	1
A	flush()	0	13	3
A	clear()	0	6	2
A	hasEntityManager()	0	4	1

<?php

namespace Shippinno\Job\Infrastructure\Persistence\Doctrine;

use Doctrine\Common\Persistence\ManagerRegistry;
use Doctrine\ORM\EntityManager;

trait ManagerRegistryAwareTrait
{
    /**
     * @var ManagerRegistry|null $managerRegistry
     */
    protected $managerRegistry;

    /**
     * @param ManagerRegistry|null $managerRegistry
     */
    public function setManagerRegistry(?ManagerRegistry $managerRegistry)
    {
        $this->managerRegistry = $managerRegistry;
    }

    /**
     * @return void
     */
    public function flush(): void
    {
        if ($this->hasEntityManager()) {
            /** @var EntityManager $entityManager */
            $entityManager =  $this->managerRegistry->getManager();
            $connection = $entityManager->getConnection();
            if (!$connection->ping()) {
                $connection->close();
                $connection->connect();
            }
            $entityManager->flush();
        }
    }

    /**
     * @return void
     */
    public function clear(): void
    {
        if ($this->hasEntityManager()) {
            $this->managerRegistry->getManager()->clear();
        }
    }

    /**
     * @return bool
     */
    protected function hasEntityManager(): bool
    {
        return null !== $this->managerRegistry;
    }
}