1 | <?php |
||
24 | use Symfony\Component\DependencyInjection\Definition; |
||
25 | |||
26 | class MappingPass implements CompilerPassInterface |
||
27 | { |
||
28 | /** |
||
29 | * @var array |
||
30 | */ |
||
31 | private $indexes = []; |
||
32 | |||
33 | /** |
||
34 | * @var string |
||
35 | */ |
||
36 | private $defaultIndex = null; |
||
37 | |||
38 | public function process(ContainerBuilder $container) |
||
39 | { |
||
40 | $kernelDir = $container->getParameter('kernel.project_dir'); |
||
41 | $parser = $container->get(DocumentParser::class); |
||
42 | |||
43 | $indexClasses = []; |
||
44 | $indexSettingsArray = []; |
||
45 | foreach ($container->getParameter(Configuration::ONGR_SOURCE_DIR) as $dir) { |
||
46 | foreach ($this->getNamespaces($kernelDir . $dir) as $namespace) { |
||
47 | $indexClasses[$namespace] = $namespace; |
||
48 | } |
||
49 | } |
||
50 | |||
51 | $overwrittenClasses = []; |
||
52 | $indexOverrides = $container->getParameter(Configuration::ONGR_INDEXES_OVERRIDE); |
||
53 | |||
54 | foreach ($indexOverrides as $name => $indexOverride) { |
||
55 | $class = isset($indexOverride['class']) ? $indexOverride['class'] : $name; |
||
56 | |||
57 | if (!isset($indexClasses[$class])) { |
||
58 | throw new \RuntimeException( |
||
59 | sprintf( |
||
60 | 'Document `%s` defined in ongr_elasticsearch.indexes config could not been found', |
||
61 | $class, |
||
62 | ) |
||
|
|||
63 | ); |
||
64 | } |
||
65 | |||
66 | $indexSettings = $this->parseIndexSettingsFromClass($parser, $class); |
||
67 | |||
68 | if ($class !== $name) { |
||
69 | $indexSettings->setIndexName('ongr.es.index.'.$name); |
||
70 | } |
||
71 | |||
72 | if (isset($indexOverride['alias'])) { |
||
73 | $indexSettings->setAlias($indexOverride['alias']); |
||
74 | } |
||
75 | |||
76 | if (isset($indexOverride['settings'])) { |
||
77 | $indexSettings->setIndexMetadata($indexOverride['settings']); |
||
78 | } |
||
79 | |||
80 | if (isset($indexOverride['hosts'])) { |
||
81 | $indexSettings->setHosts($indexOverride['hosts']); |
||
82 | } |
||
83 | |||
84 | if (isset($indexOverride['default'])) { |
||
85 | $indexSettings->setDefaultIndex($indexOverride['default']); |
||
86 | } |
||
87 | |||
88 | $indexSettingsArray[$name] = $indexSettings; |
||
89 | $overwrittenClasses[$class] = $class; |
||
90 | } |
||
91 | |||
92 | foreach (array_diff($indexClasses, $overwrittenClasses) as $indexClass) { |
||
93 | try { |
||
94 | $indexSettingsArray[$indexClass] = $this->parseIndexSettingsFromClass($parser, $indexClass); |
||
95 | } catch (DocumentIndexParserException $e) {} |
||
96 | } |
||
97 | |||
98 | foreach($indexSettingsArray as $indexSettings) { |
||
99 | $this->createIndex($container, $indexSettings); |
||
100 | } |
||
101 | |||
102 | $container->setParameter(Configuration::ONGR_INDEXES, $this->indexes); |
||
103 | $container->setParameter( |
||
104 | Configuration::ONGR_DEFAULT_INDEX, |
||
105 | $this->defaultIndex ?? current(array_keys($this->indexes)) |
||
106 | ); |
||
107 | } |
||
108 | |||
109 | private function parseIndexSettingsFromClass(DocumentParser $parser, string $className) : IndexSettings |
||
110 | { |
||
111 | $class = new \ReflectionClass($className); |
||
112 | |||
113 | /** @var Index $document */ |
||
114 | $document = $parser->getIndexAnnotation($class); |
||
115 | |||
116 | if ($document === null) { |
||
117 | throw new DocumentIndexParserException(); |
||
118 | } |
||
119 | |||
120 | $indexSettings = new IndexSettings( |
||
121 | $className, |
||
122 | $className, |
||
123 | $parser->getIndexAliasName($class), |
||
124 | $parser->getIndexMetadata($class), |
||
125 | $parser->getPropertyMetadata($class), |
||
126 | $document->hosts, |
||
127 | $parser->isDefaultIndex($class) |
||
128 | ); |
||
129 | |||
130 | $indexSettings->setIndexMetadata(['settings' => [ |
||
131 | 'number_of_replicas' => $document->numberOfReplicas, |
||
132 | 'number_of_shards' => $document->numberOfShards, |
||
133 | ]]); |
||
134 | |||
135 | return $indexSettings; |
||
136 | } |
||
137 | |||
138 | private function createIndex(Container $container, IndexSettings $indexSettings) { |
||
139 | $converterDefinition = $container->getDefinition(Converter::class); |
||
140 | |||
141 | $indexSettingsDefinition = new Definition( |
||
142 | IndexSettings::class, |
||
143 | [ |
||
144 | $indexSettings->getNamespace(), |
||
145 | $indexSettings->getAlias(), |
||
146 | $indexSettings->getAlias(), |
||
147 | $indexSettings->getIndexMetadata(), |
||
148 | $indexSettings->getPropertyMetadata(), |
||
149 | $indexSettings->getHosts(), |
||
150 | $indexSettings->isDefaultIndex(), |
||
151 | ] |
||
152 | ); |
||
153 | |||
154 | $indexServiceDefinition = new Definition(IndexService::class, [ |
||
155 | $indexSettings->getNamespace(), |
||
156 | $converterDefinition, |
||
157 | $container->getDefinition('event_dispatcher'), |
||
158 | $indexSettingsDefinition, |
||
159 | $container->getParameter(Configuration::ONGR_PROFILER_CONFIG) |
||
160 | ? $container->getDefinition('ongr.esb.tracer') : null |
||
161 | ]); |
||
162 | |||
163 | $indexServiceDefinition->setPublic(true); |
||
164 | $converterDefinition->addMethodCall( |
||
165 | 'addClassMetadata', |
||
166 | [ |
||
167 | $indexSettings->getNamespace(), |
||
168 | $indexSettings->getPropertyMetadata() |
||
169 | ] |
||
170 | ); |
||
171 | |||
172 | $container->setDefinition($indexSettings->getIndexName(), $indexServiceDefinition); |
||
173 | $this->indexes[$indexSettings->getAlias()] = $indexSettings->getIndexName(); |
||
174 | $isCurrentIndexDefault = $indexSettings->isDefaultIndex(); |
||
175 | if ($this->defaultIndex && $isCurrentIndexDefault) { |
||
176 | throw new \RuntimeException( |
||
229 |