1 | <?php |
||||
2 | |||||
3 | namespace Mdiyakov\DoctrineSolrBundle\Query; |
||||
4 | |||||
5 | use Mdiyakov\DoctrineSolrBundle\Exception\SchemaNotFoundException; |
||||
6 | use Mdiyakov\DoctrineSolrBundle\Query\Select\ClassSelectQuery; |
||||
7 | use Mdiyakov\DoctrineSolrBundle\Query\Select\MultiClassSelectQuery; |
||||
8 | use Nelmio\SolariumBundle\ClientRegistry; |
||||
9 | use Mdiyakov\DoctrineSolrBundle\Config\Config; |
||||
10 | use Mdiyakov\DoctrineSolrBundle\Exception\EntityNotIndexedException; |
||||
11 | use Mdiyakov\DoctrineSolrBundle\Exception\SchemaConfigException; |
||||
12 | use Mdiyakov\DoctrineSolrBundle\Query\Hydrator\HydratorBuilder; |
||||
13 | use Mdiyakov\DoctrineSolrBundle\Query\Hydrator\SelectQueryHydrator; |
||||
14 | use Mdiyakov\DoctrineSolrBundle\Schema\Schema; |
||||
15 | |||||
16 | class SelectQueryBuilder |
||||
17 | { |
||||
18 | /** |
||||
19 | * @var Config |
||||
20 | */ |
||||
21 | private $config; |
||||
22 | |||||
23 | /** |
||||
24 | * @var ClientRegistry |
||||
25 | */ |
||||
26 | private $clientRegistry; |
||||
27 | |||||
28 | /** |
||||
29 | * @var SelectQueryHydrator[] |
||||
30 | */ |
||||
31 | private $hydrators; |
||||
32 | |||||
33 | public function __construct( |
||||
34 | Config $config, |
||||
35 | ClientRegistry $clientRegistry, |
||||
36 | HydratorBuilder $hydratorBuilder |
||||
37 | ) |
||||
38 | { |
||||
39 | $this->config = $config; |
||||
40 | $this->clientRegistry = $clientRegistry; |
||||
41 | |||||
42 | foreach($config->getIndexedEntities() as $entityConfig) { |
||||
43 | $this->hydrators[$entityConfig['class']] = $hydratorBuilder->buildSelectQueryHydratorByClass($entityConfig['class']); |
||||
44 | } |
||||
45 | } |
||||
46 | |||||
47 | |||||
48 | /** |
||||
49 | * @param string $class |
||||
50 | * @return ClassSelectQuery |
||||
51 | */ |
||||
52 | public function buildClassSelectQuery($class) |
||||
53 | { |
||||
54 | $entityConfig = $this->config->getEntityConfig($class); |
||||
55 | if (!$entityConfig) { |
||||
56 | throw new EntityNotIndexedException( |
||||
57 | sprintf('""%s" is not indexed. Check bundle config in config.yml', $class) |
||||
58 | ); |
||||
59 | } |
||||
60 | $schema = $this->config->getSchemaByEntityClass($class); |
||||
61 | |||||
62 | return new ClassSelectQuery( |
||||
63 | $this->clientRegistry->getClient( |
||||
64 | $this->config->getSolariumClient($schema->getClient()) |
||||
65 | ), |
||||
66 | $schema, |
||||
67 | $entityConfig, |
||||
0 ignored issues
–
show
Bug
introduced
by
![]() |
|||||
68 | $this->hydrators[$class] |
||||
69 | ); |
||||
70 | } |
||||
71 | |||||
72 | /** |
||||
73 | * @param string $schemaName |
||||
74 | * @param string[] $classes |
||||
75 | * @return MultiClassSelectQuery |
||||
76 | * @throws \InvalidArgumentException |
||||
77 | * @throws SchemaNotFoundException |
||||
78 | */ |
||||
79 | public function buildMultiClassSelectQueryBySchemaName($schemaName, $classes) |
||||
80 | { |
||||
81 | if (!is_string($schemaName)) { |
||||
0 ignored issues
–
show
|
|||||
82 | throw new \InvalidArgumentException('Argument $schemaName must be a string'); |
||||
83 | } |
||||
84 | |||||
85 | $schema = $this->config->getSchemaByName($schemaName); |
||||
86 | if (!$schema) { |
||||
87 | throw new SchemaNotFoundException( |
||||
88 | sprintf('Schema "%s" is not found', $schemaName) |
||||
89 | ); |
||||
90 | } |
||||
91 | |||||
92 | return $this->buildMultiClassSelectQuery($schema, $classes); |
||||
93 | } |
||||
94 | |||||
95 | |||||
96 | /** |
||||
97 | * @param Schema $schema |
||||
98 | * @param string[] $classes |
||||
99 | * @return MultiClassSelectQuery |
||||
100 | */ |
||||
101 | public function buildMultiClassSelectQuery(Schema $schema, $classes) |
||||
102 | { |
||||
103 | if (!is_array($classes) || empty($classes)) { |
||||
0 ignored issues
–
show
|
|||||
104 | throw new \InvalidArgumentException(sprintf('Argument must be an array of classes')); |
||||
105 | } |
||||
106 | |||||
107 | $multiClassQueryConfig = [ |
||||
108 | 'entityConfigs' => [], |
||||
109 | 'hydrators' => [] |
||||
110 | ]; |
||||
111 | |||||
112 | $discriminatorConfigField = $schema->getDiscriminatorConfigField(); |
||||
113 | |||||
114 | foreach ($classes as $class) { |
||||
115 | $entityConfig = $this->config->getEntityConfig($class); |
||||
116 | if (!$entityConfig) { |
||||
117 | throw new EntityNotIndexedException( |
||||
118 | sprintf('""%s" is not indexed. Check bundle config in config.yml', $class) |
||||
119 | ); |
||||
120 | } |
||||
121 | |||||
122 | if ($schema->getName() != $entityConfig['schema']) |
||||
123 | { |
||||
124 | throw new SchemaConfigException( |
||||
125 | 'Entity class %s doesn\'t support scheme %s', |
||||
126 | $entityConfig['class'], |
||||
0 ignored issues
–
show
$entityConfig['class'] of type string[] is incompatible with the type integer expected by parameter $code of Mdiyakov\DoctrineSolrBun...xception::__construct() .
(
Ignorable by Annotation
)
If this is a false-positive, you can also ignore this issue in your code via the
![]() |
|||||
127 | $schema->getName() |
||||
0 ignored issues
–
show
$schema->getName() of type string is incompatible with the type Throwable|null expected by parameter $previous of Mdiyakov\DoctrineSolrBun...xception::__construct() .
(
Ignorable by Annotation
)
If this is a false-positive, you can also ignore this issue in your code via the
![]() |
|||||
128 | ); |
||||
129 | } |
||||
130 | |||||
131 | $multiClassQueryConfig['entityConfigs'][] = $entityConfig; |
||||
132 | $multiClassQueryConfig['hydrators'][$discriminatorConfigField->getValue($entityConfig)] = $this->hydrators[$class]; |
||||
133 | } |
||||
134 | |||||
135 | return new MultiClassSelectQuery( |
||||
136 | $schema, |
||||
137 | $this->clientRegistry->getClient( |
||||
138 | $this->config->getSolariumClient($schema->getClient()) |
||||
139 | ), |
||||
140 | $multiClassQueryConfig['entityConfigs'], |
||||
141 | $multiClassQueryConfig['hydrators'] |
||||
142 | ); |
||||
143 | } |
||||
144 | |||||
145 | } |