Completed
Pull Request — master (#877)
by Kévin
01:55
created

DoctrineCommand::getEntityGenerator()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 12

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 12
rs 9.8666
cc 1
nc 1
nop 0
1
<?php
2
3
namespace Doctrine\Bundle\DoctrineBundle\Command;
4
5
use Doctrine\Common\Persistence\ManagerRegistry;
6
use Doctrine\DBAL\Connection;
7
use Doctrine\DBAL\Sharding\PoolingShardConnection;
8
use Doctrine\ORM\EntityManager;
9
use Doctrine\ORM\Tools\EntityGenerator;
10
use LogicException;
11
use Symfony\Component\Console\Command\Command;
12
13
/**
14
 * Base class for Doctrine console commands to extend from.
15
 *
16
 * @internal
17
 */
18
abstract class DoctrineCommand extends Command
19
{
20
    /** @var ManagerRegistry */
21
    protected $doctrine;
22
23
    public function __construct(ManagerRegistry $doctrine)
24
    {
25
        parent::__construct();
26
27
        $this->doctrine = $doctrine;
28
    }
29
30
    /**
31
     * get a doctrine entity generator
32
     *
33
     * @return EntityGenerator
34
     */
35
    protected function getEntityGenerator()
36
    {
37
        $entityGenerator = new EntityGenerator();
38
        $entityGenerator->setGenerateAnnotations(false);
39
        $entityGenerator->setGenerateStubMethods(true);
40
        $entityGenerator->setRegenerateEntityIfExists(false);
41
        $entityGenerator->setUpdateEntityIfExists(true);
42
        $entityGenerator->setNumSpaces(4);
43
        $entityGenerator->setAnnotationPrefix('ORM\\');
44
45
        return $entityGenerator;
46
    }
47
48
    /**
49
     * Get a doctrine entity manager by symfony name.
50
     *
51
     * @param string   $name
52
     * @param int|null $shardId
53
     *
54
     * @return EntityManager
55
     */
56
    protected function getEntityManager($name, $shardId = null)
57
    {
58
        $manager = $this->doctrine->getManager($name);
59
60
        if ($shardId) {
0 ignored issues
show
Bug Best Practice introduced by
The expression $shardId of type integer|null is loosely compared to true; this is ambiguous if the integer can be zero. You might want to explicitly use !== null instead.

In PHP, under loose comparison (like ==, or !=, or switch conditions), values of different types might be equal.

For integer values, zero is a special case, in particular the following results might be unexpected:

0   == false // true
0   == null  // true
123 == false // false
123 == null  // false

// It is often better to use strict comparison
0 === false // false
0 === null  // false
Loading history...
61
            if (! $manager->getConnection() instanceof PoolingShardConnection) {
62
                throw new LogicException(sprintf("Connection of EntityManager '%s' must implement shards configuration.", $name));
63
            }
64
65
            $manager->getConnection()->connect($shardId);
66
        }
67
68
        return $manager;
69
    }
70
71
    /**
72
     * Get a doctrine dbal connection by symfony name.
73
     *
74
     * @param string $name
75
     *
76
     * @return Connection
77
     */
78
    protected function getDoctrineConnection($name)
79
    {
80
        return $this->doctrine->getConnection($name);
81
    }
82
}
83