|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace Doctrine\ODM\MongoDB\Tools\Console\Command\Schema; |
|
4
|
|
|
|
|
5
|
|
|
use Doctrine\ODM\MongoDB\Mapping\ClassMetadata; |
|
6
|
|
|
use Doctrine\ODM\MongoDB\SchemaManager; |
|
7
|
|
|
use Symfony\Component\Console\Input\InputInterface; |
|
8
|
|
|
use Symfony\Component\Console\Input\InputOption; |
|
9
|
|
|
use Symfony\Component\Console\Output\OutputInterface; |
|
10
|
|
|
|
|
11
|
|
|
class CreateCommand extends AbstractCommand |
|
12
|
|
|
{ |
|
13
|
|
|
private $createOrder = array(self::COLLECTION, self::INDEX); |
|
14
|
|
|
|
|
15
|
|
|
private $timeout; |
|
16
|
|
|
|
|
17
|
|
|
protected function configure() |
|
18
|
|
|
{ |
|
19
|
|
|
$this |
|
20
|
|
|
->setName('odm:schema:create') |
|
21
|
|
|
->addOption('class', 'c', InputOption::VALUE_REQUIRED, 'Document class to process (default: all classes)') |
|
22
|
|
|
->addOption('timeout', 't', InputOption::VALUE_OPTIONAL, 'Timeout (ms) for acknowledged index creation') |
|
23
|
|
|
->addOption(self::COLLECTION, null, InputOption::VALUE_NONE, 'Create collections') |
|
24
|
|
|
->addOption(self::INDEX, null, InputOption::VALUE_NONE, 'Create indexes') |
|
25
|
|
|
->setDescription('Create databases, collections and indexes for your documents') |
|
26
|
|
|
; |
|
27
|
|
|
} |
|
28
|
|
|
|
|
29
|
|
|
protected function execute(InputInterface $input, OutputInterface $output) |
|
30
|
|
|
{ |
|
31
|
|
|
$create = array_filter($this->createOrder, function ($option) use ($input) { |
|
32
|
|
|
return $input->getOption($option); |
|
33
|
|
|
}); |
|
34
|
|
|
|
|
35
|
|
|
// Default to the full creation order if no options were specified |
|
36
|
|
|
$create = empty($create) ? $this->createOrder : $create; |
|
37
|
|
|
|
|
38
|
|
|
$class = $input->getOption('class'); |
|
39
|
|
|
|
|
40
|
|
|
$timeout = $input->getOption('timeout'); |
|
41
|
|
|
$this->timeout = isset($timeout) ? (int) $timeout : null; |
|
42
|
|
|
|
|
43
|
|
|
$sm = $this->getSchemaManager(); |
|
44
|
|
|
$isErrored = false; |
|
45
|
|
|
|
|
46
|
|
View Code Duplication |
foreach ($create as $option) { |
|
|
|
|
|
|
47
|
|
|
try { |
|
48
|
|
|
if (isset($class)) { |
|
49
|
|
|
$this->{'processDocument' . ucfirst($option)}($sm, $class); |
|
50
|
|
|
} else { |
|
51
|
|
|
$this->{'process' . ucfirst($option)}($sm); |
|
52
|
|
|
} |
|
53
|
|
|
$output->writeln(sprintf( |
|
54
|
|
|
'Created <comment>%s%s</comment> for <info>%s</info>', |
|
55
|
|
|
$option, |
|
56
|
|
|
(isset($class) ? (self::INDEX === $option ? '(es)' : '') : (self::INDEX === $option ? 'es' : 's')), |
|
57
|
|
|
($class ?? 'all classes') |
|
58
|
|
|
)); |
|
59
|
|
|
} catch (\Exception $e) { |
|
60
|
|
|
$output->writeln('<error>' . $e->getMessage() . '</error>'); |
|
61
|
|
|
$isErrored = true; |
|
62
|
|
|
} |
|
63
|
|
|
} |
|
64
|
|
|
|
|
65
|
|
|
return $isErrored ? 255 : 0; |
|
66
|
|
|
} |
|
67
|
|
|
|
|
68
|
|
|
protected function processDocumentCollection(SchemaManager $sm, $document) |
|
69
|
|
|
{ |
|
70
|
|
|
$sm->createDocumentCollection($document); |
|
71
|
|
|
} |
|
72
|
|
|
|
|
73
|
|
|
protected function processCollection(SchemaManager $sm) |
|
74
|
|
|
{ |
|
75
|
|
|
$sm->createCollections(); |
|
76
|
|
|
} |
|
77
|
|
|
|
|
78
|
|
|
protected function processDocumentDb(SchemaManager $sm, $document) |
|
79
|
|
|
{ |
|
80
|
|
|
throw new \BadMethodCallException('A database is created automatically by MongoDB (>= 3.0).'); |
|
81
|
|
|
} |
|
82
|
|
|
|
|
83
|
|
|
protected function processDb(SchemaManager $sm) |
|
84
|
|
|
{ |
|
85
|
|
|
throw new \BadMethodCallException('A database is created automatically by MongoDB (>= 3.0).'); |
|
86
|
|
|
} |
|
87
|
|
|
|
|
88
|
|
|
protected function processDocumentIndex(SchemaManager $sm, $document) |
|
89
|
|
|
{ |
|
90
|
|
|
$sm->ensureDocumentIndexes($document, $this->timeout); |
|
91
|
|
|
} |
|
92
|
|
|
|
|
93
|
|
|
protected function processIndex(SchemaManager $sm) |
|
94
|
|
|
{ |
|
95
|
|
|
$sm->ensureIndexes($this->timeout); |
|
96
|
|
|
} |
|
97
|
|
|
|
|
98
|
|
|
protected function processDocumentProxy(SchemaManager $sm, $document) |
|
|
|
|
|
|
99
|
|
|
{ |
|
100
|
|
|
$classMetadata = $this->getMetadataFactory()->getMetadataFor($document); |
|
101
|
|
|
|
|
102
|
|
|
if (!$classMetadata->isEmbeddedDocument && !$classMetadata->isMappedSuperclass && !$classMetadata->isQueryResultDocument) { |
|
|
|
|
|
|
103
|
|
|
$this->getDocumentManager()->getProxyFactory()->generateProxyClasses(array($classMetadata)); |
|
104
|
|
|
} |
|
105
|
|
|
} |
|
106
|
|
|
|
|
107
|
|
|
protected function processProxy(SchemaManager $sm) |
|
|
|
|
|
|
108
|
|
|
{ |
|
109
|
|
|
$classes = array_filter($this->getMetadataFactory()->getAllMetadata(), function (ClassMetadata $classMetadata) { |
|
110
|
|
|
return !$classMetadata->isEmbeddedDocument && !$classMetadata->isMappedSuperclass && !$classMetadata->isQueryResultDocument; |
|
111
|
|
|
}); |
|
112
|
|
|
|
|
113
|
|
|
$this->getDocumentManager()->getProxyFactory()->generateProxyClasses($classes); |
|
114
|
|
|
} |
|
115
|
|
|
} |
|
116
|
|
|
|
doctrine /
mongodb-odm
Loading history...
Duplicated code is one of the most pungent code smells. If you need to duplicate the same code in three or more different places, we strongly encourage you to look into extracting the code into a single class or operation.
You can also find more detailed suggestions in the “Code” section of your repository.