1
|
|
|
<?php declare(strict_types=1); |
2
|
|
|
|
3
|
|
|
namespace Shopware\Core\Framework\DataAbstractionLayer\Dbal; |
4
|
|
|
|
5
|
|
|
use Doctrine\DBAL\Connection; |
6
|
|
|
use Doctrine\DBAL\Exception; |
7
|
|
|
use Doctrine\DBAL\Query\QueryBuilder as DbalQueryBuilderAlias; |
8
|
|
|
use Doctrine\DBAL\Types\Types; |
9
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\DefinitionInstanceRegistry; |
10
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\MultiInsertQueryQueue; |
11
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\RetryableQuery; |
12
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\RetryableTransaction; |
13
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\EntityDefinition; |
14
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\EntityTranslationDefinition; |
15
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Event\BeforeDeleteEvent; |
16
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityDeleteEvent; |
17
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWriteEvent; |
18
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\CanNotFindParentStorageFieldException; |
19
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\InvalidParentAssociationException; |
20
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\ParentFieldForeignKeyConstraintMissingException; |
21
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\ParentFieldNotFoundException; |
22
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\PrimaryKeyNotProvidedException; |
23
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Exception\UnsupportedCommandTypeException; |
24
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Field\Field; |
25
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Field\FkField; |
26
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Field\ManyToOneAssociationField; |
27
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Field\StorageAware; |
28
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Field\VersionField; |
29
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\MappingEntityDefinition; |
30
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\ChangeSet; |
31
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\ChangeSetAware; |
32
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\DeleteCommand; |
33
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\InsertCommand; |
34
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\JsonUpdateCommand; |
35
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\UpdateCommand; |
36
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\WriteCommand; |
37
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\WriteCommandQueue; |
38
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\DataStack\KeyValuePair; |
39
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\EntityExistence; |
40
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\EntityWriteGatewayInterface; |
41
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\PrimaryKeyBag; |
42
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Validation\PostWriteValidationEvent; |
43
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Validation\PreWriteValidationEvent; |
44
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\Validation\WriteCommandExceptionEvent; |
45
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\WriteContext; |
46
|
|
|
use Shopware\Core\Framework\DataAbstractionLayer\Write\WriteParameterBag; |
47
|
|
|
use Shopware\Core\Framework\Feature; |
48
|
|
|
use Shopware\Core\Framework\Log\Package; |
49
|
|
|
use Shopware\Core\Framework\Uuid\Uuid; |
50
|
|
|
use Symfony\Contracts\EventDispatcher\EventDispatcherInterface; |
51
|
|
|
|
52
|
|
|
/** |
53
|
|
|
* @internal |
54
|
|
|
*/ |
55
|
|
|
#[Package('core')] |
56
|
|
|
class EntityWriteGateway implements EntityWriteGatewayInterface |
57
|
|
|
{ |
58
|
|
|
private ?PrimaryKeyBag $primaryKeyBag = null; |
59
|
|
|
|
60
|
|
|
public function __construct( |
61
|
|
|
private readonly int $batchSize, |
62
|
|
|
private readonly Connection $connection, |
63
|
|
|
private readonly EventDispatcherInterface $eventDispatcher, |
64
|
|
|
private readonly ExceptionHandlerRegistry $exceptionHandlerRegistry, |
65
|
|
|
private readonly DefinitionInstanceRegistry $definitionInstanceRegistry |
66
|
|
|
) { |
67
|
|
|
} |
68
|
|
|
|
69
|
|
|
public function prefetchExistences(WriteParameterBag $parameters): void |
70
|
|
|
{ |
71
|
|
|
$primaryKeyBag = $this->primaryKeyBag = $parameters->getPrimaryKeyBag(); |
72
|
|
|
|
73
|
|
|
if ($primaryKeyBag->isPrefetchingCompleted()) { |
74
|
|
|
return; |
75
|
|
|
} |
76
|
|
|
|
77
|
|
|
foreach ($primaryKeyBag->getPrimaryKeys() as $entity => $pks) { |
78
|
|
|
$this->prefetch($this->definitionInstanceRegistry->getByEntityName($entity), $pks, $parameters); |
79
|
|
|
} |
80
|
|
|
|
81
|
|
|
$primaryKeyBag->setPrefetchingCompleted(true); |
82
|
|
|
} |
83
|
|
|
|
84
|
|
|
/** |
85
|
|
|
* {@inheritdoc} |
86
|
|
|
*/ |
87
|
|
|
public function getExistence(EntityDefinition $definition, array $primaryKey, array $data, WriteCommandQueue $commandQueue): EntityExistence |
88
|
|
|
{ |
89
|
|
|
$state = $this->getCurrentState($definition, $primaryKey, $commandQueue); |
90
|
|
|
|
91
|
|
|
$exists = !empty($state); |
92
|
|
|
|
93
|
|
|
$isChild = $this->isChild($definition, $data, $state, $primaryKey, $commandQueue); |
94
|
|
|
|
95
|
|
|
$wasChild = $this->wasChild($definition, $state); |
96
|
|
|
|
97
|
|
|
$decodedPrimaryKey = []; |
98
|
|
|
foreach ($primaryKey as $fieldStorageName => $fieldValue) { |
99
|
|
|
$field = $definition->getFields()->getByStorageName($fieldStorageName); |
100
|
|
|
$decodedPrimaryKey[$fieldStorageName] = $field ? $field->getSerializer()->decode($field, $fieldValue) : $fieldValue; |
101
|
|
|
} |
102
|
|
|
|
103
|
|
|
return new EntityExistence($definition->getEntityName(), $decodedPrimaryKey, $exists, $isChild, $wasChild, $state); |
104
|
|
|
} |
105
|
|
|
|
106
|
|
|
/** |
107
|
|
|
* {@inheritdoc} |
108
|
|
|
*/ |
109
|
|
|
public function execute(array $commands, WriteContext $context): void |
110
|
|
|
{ |
111
|
|
|
$beforeWriteEvent = EntityWriteEvent::create($context, $commands); |
112
|
|
|
|
113
|
|
|
$this->eventDispatcher->dispatch($beforeWriteEvent); |
114
|
|
|
|
115
|
|
|
try { |
116
|
|
|
RetryableTransaction::retryable($this->connection, function () use ($commands, $context): void { |
117
|
|
|
$this->executeCommands($commands, $context); |
118
|
|
|
}); |
119
|
|
|
|
120
|
|
|
$beforeWriteEvent->success(); |
121
|
|
|
} catch (\Throwable $e) { |
122
|
|
|
$event = new WriteCommandExceptionEvent($e, $commands, $context->getContext()); |
123
|
|
|
$this->eventDispatcher->dispatch($event); |
124
|
|
|
|
125
|
|
|
$beforeWriteEvent->error(); |
126
|
|
|
|
127
|
|
|
throw $e; |
128
|
|
|
} |
129
|
|
|
} |
130
|
|
|
|
131
|
|
|
/** |
132
|
|
|
* @param list<WriteCommand> $commands |
133
|
|
|
*/ |
134
|
|
|
private function executeCommands(array $commands, WriteContext $context): void |
135
|
|
|
{ |
136
|
|
|
$entityDeleteEvent = EntityDeleteEvent::create($context, $commands); |
137
|
|
|
$entityDeleteEventLegacy = BeforeDeleteEvent::create($context, $commands); |
138
|
|
|
if ($entityDeleteEvent->filled()) { |
139
|
|
|
$this->eventDispatcher->dispatch($entityDeleteEvent); |
140
|
|
|
|
141
|
|
|
Feature::ifNotActive('v6.6.0.0', fn () => $this->eventDispatcher->dispatch($entityDeleteEventLegacy)); |
142
|
|
|
} |
143
|
|
|
|
144
|
|
|
// throws exception on violation and then aborts/rollbacks this transaction |
145
|
|
|
$event = new PreWriteValidationEvent($context, $commands); |
146
|
|
|
$this->eventDispatcher->dispatch($event); |
147
|
|
|
/** @var list<WriteCommand> $commands */ |
148
|
|
|
$commands = $event->getCommands(); |
149
|
|
|
|
150
|
|
|
$this->generateChangeSets($commands); |
151
|
|
|
|
152
|
|
|
$context->getExceptions()->tryToThrow(); |
153
|
|
|
|
154
|
|
|
$previous = null; |
155
|
|
|
$mappings = new MultiInsertQueryQueue($this->connection, $this->batchSize, false, true); |
156
|
|
|
$inserts = new MultiInsertQueryQueue($this->connection, $this->batchSize); |
157
|
|
|
|
158
|
|
|
$executeInserts = function () use ($mappings, $inserts): void { |
159
|
|
|
$mappings->execute(); |
160
|
|
|
$inserts->execute(); |
161
|
|
|
}; |
162
|
|
|
|
163
|
|
|
try { |
164
|
|
|
foreach ($commands as $command) { |
165
|
|
|
if (!$command->isValid()) { |
166
|
|
|
continue; |
167
|
|
|
} |
168
|
|
|
$command->setFailed(false); |
169
|
|
|
$current = $command->getDefinition()->getEntityName(); |
170
|
|
|
|
171
|
|
|
if ($current !== $previous) { |
172
|
|
|
$executeInserts(); |
173
|
|
|
} |
174
|
|
|
$previous = $current; |
175
|
|
|
|
176
|
|
|
try { |
177
|
|
|
$definition = $command->getDefinition(); |
178
|
|
|
$table = $definition->getEntityName(); |
179
|
|
|
|
180
|
|
|
if ($command instanceof DeleteCommand) { |
181
|
|
|
$executeInserts(); |
182
|
|
|
|
183
|
|
|
RetryableQuery::retryable($this->connection, function () use ($command, $table): void { |
184
|
|
|
$this->connection->delete(EntityDefinitionQueryHelper::escape($table), $command->getPrimaryKey()); |
185
|
|
|
}); |
186
|
|
|
|
187
|
|
|
continue; |
188
|
|
|
} |
189
|
|
|
|
190
|
|
|
if ($command instanceof JsonUpdateCommand) { |
191
|
|
|
$executeInserts(); |
192
|
|
|
$this->executeJsonUpdate($command); |
193
|
|
|
|
194
|
|
|
continue; |
195
|
|
|
} |
196
|
|
|
|
197
|
|
|
if ($definition instanceof MappingEntityDefinition && $command instanceof InsertCommand) { |
198
|
|
|
$mappings->addInsert($definition->getEntityName(), $command->getPayload()); |
199
|
|
|
|
200
|
|
|
continue; |
201
|
|
|
} |
202
|
|
|
|
203
|
|
|
if ($command instanceof UpdateCommand) { |
204
|
|
|
$executeInserts(); |
205
|
|
|
|
206
|
|
|
RetryableQuery::retryable($this->connection, function () use ($command, $table): void { |
207
|
|
|
$this->connection->update( |
208
|
|
|
EntityDefinitionQueryHelper::escape($table), |
209
|
|
|
$this->escapeColumnKeys($command->getPayload()), |
210
|
|
|
$command->getPrimaryKey() |
211
|
|
|
); |
212
|
|
|
}); |
213
|
|
|
|
214
|
|
|
continue; |
215
|
|
|
} |
216
|
|
|
|
217
|
|
|
if ($command instanceof InsertCommand) { |
218
|
|
|
$inserts->addInsert($definition->getEntityName(), $command->getPayload()); |
219
|
|
|
|
220
|
|
|
continue; |
221
|
|
|
} |
222
|
|
|
|
223
|
|
|
throw new UnsupportedCommandTypeException($command); |
224
|
|
|
} catch (\Exception $e) { |
225
|
|
|
$command->setFailed(true); |
226
|
|
|
|
227
|
|
|
$innerException = $this->exceptionHandlerRegistry->matchException($e); |
228
|
|
|
|
229
|
|
|
if ($innerException instanceof \Exception) { |
230
|
|
|
$e = $innerException; |
231
|
|
|
} |
232
|
|
|
$context->getExceptions()->add($e); |
233
|
|
|
|
234
|
|
|
throw $e; |
235
|
|
|
} |
236
|
|
|
} |
237
|
|
|
|
238
|
|
|
$mappings->execute(); |
239
|
|
|
$inserts->execute(); |
240
|
|
|
$entityDeleteEvent->success(); |
241
|
|
|
Feature::ifNotActive('v6.6.0.0', fn () => $entityDeleteEventLegacy->success()); |
242
|
|
|
} catch (Exception $e) { |
243
|
|
|
// Match exception without passing a specific command when feature-flag 16640 is active |
244
|
|
|
$innerException = $this->exceptionHandlerRegistry->matchException($e); |
245
|
|
|
if ($innerException instanceof \Exception) { |
246
|
|
|
$e = $innerException; |
247
|
|
|
} |
248
|
|
|
$context->getExceptions()->add($e); |
249
|
|
|
|
250
|
|
|
$entityDeleteEvent->error(); |
251
|
|
|
Feature::ifNotActive('v6.6.0.0', fn () => $entityDeleteEventLegacy->error()); |
252
|
|
|
|
253
|
|
|
throw $e; |
254
|
|
|
} |
255
|
|
|
|
256
|
|
|
// throws exception on violation and then aborts/rollbacks this transaction |
257
|
|
|
$event = new PostWriteValidationEvent($context, $commands); |
258
|
|
|
$this->eventDispatcher->dispatch($event); |
259
|
|
|
$context->getExceptions()->tryToThrow(); |
260
|
|
|
} |
261
|
|
|
|
262
|
|
|
/** |
263
|
|
|
* @param list<array<string, string>> $pks |
264
|
|
|
*/ |
265
|
|
|
private function prefetch(EntityDefinition $definition, array $pks, WriteParameterBag $parameters): void |
266
|
|
|
{ |
267
|
|
|
$pkFields = []; |
268
|
|
|
$versionField = null; |
269
|
|
|
foreach ($definition->getPrimaryKeys() as $field) { |
270
|
|
|
/** @var StorageAware&Field $field */ |
271
|
|
|
if ($field instanceof VersionField) { |
272
|
|
|
$versionField = $field; |
273
|
|
|
|
274
|
|
|
continue; |
275
|
|
|
} |
276
|
|
|
if ($field instanceof StorageAware) { |
277
|
|
|
$pkFields[$field->getStorageName()] = $field; |
278
|
|
|
} |
279
|
|
|
} |
280
|
|
|
|
281
|
|
|
$query = $this->connection->createQueryBuilder(); |
282
|
|
|
$query->from(EntityDefinitionQueryHelper::escape($definition->getEntityName())); |
283
|
|
|
$query->addSelect('1 as `exists`'); |
284
|
|
|
|
285
|
|
|
if ($definition->isChildrenAware()) { |
286
|
|
|
$query->addSelect('parent_id'); |
287
|
|
|
} elseif ($definition->isInheritanceAware()) { |
288
|
|
|
$parent = $this->getParentField($definition); |
289
|
|
|
|
290
|
|
|
if ($parent !== null) { |
291
|
|
|
$query->addSelect( |
292
|
|
|
EntityDefinitionQueryHelper::escape($parent->getStorageName()) |
293
|
|
|
. ' as `parent`' |
294
|
|
|
); |
295
|
|
|
} |
296
|
|
|
} |
297
|
|
|
|
298
|
|
|
foreach ($pkFields as $storageName => $_) { |
299
|
|
|
$query->addSelect(EntityDefinitionQueryHelper::escape($storageName)); |
300
|
|
|
} |
301
|
|
|
if ($versionField) { |
302
|
|
|
$query->addSelect(EntityDefinitionQueryHelper::escape($versionField->getStorageName())); |
303
|
|
|
} |
304
|
|
|
|
305
|
|
|
$chunks = array_chunk($pks, 500, true); |
306
|
|
|
|
307
|
|
|
foreach ($chunks as $pks) { |
308
|
|
|
$query->resetQueryPart('where'); |
309
|
|
|
|
310
|
|
|
$params = []; |
311
|
|
|
$tupleCount = 0; |
312
|
|
|
|
313
|
|
|
foreach ($pks as $pk) { |
314
|
|
|
$newIds = []; |
315
|
|
|
foreach ($pkFields as $field) { |
316
|
|
|
$id = $pk[$field->getPropertyName()] ?? null; |
317
|
|
|
if ($id === null) { |
318
|
|
|
continue 2; |
319
|
|
|
} |
320
|
|
|
$newIds[] = $field->getSerializer()->encode( |
321
|
|
|
$field, |
322
|
|
|
EntityExistence::createForEntity($definition->getEntityName(), [$field->getPropertyName() => $id]), |
323
|
|
|
new KeyValuePair($field->getPropertyName(), $id, true), |
324
|
|
|
$parameters, |
325
|
|
|
)->current(); |
326
|
|
|
} |
327
|
|
|
|
328
|
|
|
foreach ($newIds as $newId) { |
329
|
|
|
$params[] = $newId; |
330
|
|
|
} |
331
|
|
|
|
332
|
|
|
++$tupleCount; |
333
|
|
|
} |
334
|
|
|
|
335
|
|
|
if ($tupleCount <= 0) { |
336
|
|
|
continue; |
337
|
|
|
} |
338
|
|
|
|
339
|
|
|
$placeholders = $this->getPlaceholders(\count($pkFields), $tupleCount); |
340
|
|
|
$columns = '`' . implode('`,`', array_keys($pkFields)) . '`'; |
341
|
|
|
if (\count($pkFields) > 1) { |
342
|
|
|
$columns = '(' . $columns . ')'; |
343
|
|
|
} |
344
|
|
|
|
345
|
|
|
$query->andWhere($columns . ' IN (' . $placeholders . ')'); |
346
|
|
|
if ($versionField) { |
347
|
|
|
$query->andWhere('version_id = ?'); |
348
|
|
|
$params[] = Uuid::fromHexToBytes($parameters->getContext()->getContext()->getVersionId()); |
349
|
|
|
} |
350
|
|
|
|
351
|
|
|
$query->setParameters($params); |
352
|
|
|
|
353
|
|
|
$result = $query->executeQuery()->fetchAllAssociative(); |
354
|
|
|
|
355
|
|
|
$primaryKeyBag = $parameters->getPrimaryKeyBag(); |
356
|
|
|
|
357
|
|
|
foreach ($result as $state) { |
358
|
|
|
$values = []; |
359
|
|
|
foreach ($pkFields as $storageKey => $field) { |
360
|
|
|
$values[$field->getPropertyName()] = $field->getSerializer()->decode($field, $state[$storageKey]); |
361
|
|
|
} |
362
|
|
|
if ($versionField) { |
363
|
|
|
$values[$versionField->getPropertyName()] = $parameters->getContext()->getContext()->getVersionId(); |
364
|
|
|
} |
365
|
|
|
|
366
|
|
|
$primaryKeyBag->addExistenceState($definition, $values, $state); |
367
|
|
|
} |
368
|
|
|
|
369
|
|
|
foreach ($pks as $pk) { |
370
|
|
|
if (!$primaryKeyBag->hasExistence($definition, $pk)) { |
371
|
|
|
$primaryKeyBag->addExistenceState($definition, $pk, []); |
372
|
|
|
} |
373
|
|
|
} |
374
|
|
|
} |
375
|
|
|
} |
376
|
|
|
|
377
|
|
|
/** |
378
|
|
|
* @param array<mixed> $array |
379
|
|
|
*/ |
380
|
|
|
private static function isAssociative(array $array): bool |
381
|
|
|
{ |
382
|
|
|
foreach ($array as $key => $_value) { |
383
|
|
|
if (!\is_int($key)) { |
384
|
|
|
return true; |
385
|
|
|
} |
386
|
|
|
} |
387
|
|
|
|
388
|
|
|
return false; |
389
|
|
|
} |
390
|
|
|
|
391
|
|
|
private function executeJsonUpdate(JsonUpdateCommand $command): void |
392
|
|
|
{ |
393
|
|
|
/* |
394
|
|
|
* mysql json functions are tricky. |
395
|
|
|
* |
396
|
|
|
* TL;DR: cast objects and arrays to json |
397
|
|
|
* |
398
|
|
|
* This works: |
399
|
|
|
* |
400
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', 7) |
401
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', "str") |
402
|
|
|
* |
403
|
|
|
* This does NOT work: |
404
|
|
|
* |
405
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', '{"foo": "bar"}') |
406
|
|
|
* |
407
|
|
|
* Instead, you have to do this, because mysql cannot differentiate between a string and a json string: |
408
|
|
|
* |
409
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', CAST('{"foo": "bar"}' AS json)) |
410
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', CAST('["foo", "bar"]' AS json)) |
411
|
|
|
* |
412
|
|
|
* Yet this does NOT work: |
413
|
|
|
* |
414
|
|
|
* SELECT JSON_SET('{"a": "b"}', '$.a', CAST("str" AS json)) |
415
|
|
|
* |
416
|
|
|
*/ |
417
|
|
|
|
418
|
|
|
$values = []; |
419
|
|
|
$sets = []; |
420
|
|
|
$types = []; |
421
|
|
|
|
422
|
|
|
$query = new QueryBuilder($this->connection); |
423
|
|
|
$query->update('`' . $command->getDefinition()->getEntityName() . '`'); |
424
|
|
|
|
425
|
|
|
foreach ($command->getPayload() as $attribute => $value) { |
426
|
|
|
// add path and value for each attribute value pair |
427
|
|
|
$values[] = '$."' . $attribute . '"'; |
428
|
|
|
$types[] = Types::STRING; |
429
|
|
|
if (\is_array($value) || \is_object($value)) { |
430
|
|
|
$types[] = Types::STRING; |
431
|
|
|
$values[] = json_encode($value, \JSON_PRESERVE_ZERO_FRACTION | \JSON_UNESCAPED_UNICODE); |
432
|
|
|
// does the same thing as CAST(?, json) but works on mariadb |
433
|
|
|
$identityValue = \is_object($value) || self::isAssociative($value) ? '{}' : '[]'; |
434
|
|
|
$sets[] = '?, JSON_MERGE("' . $identityValue . '", ?)'; |
435
|
|
|
} else { |
436
|
|
|
if (!\is_bool($value)) { |
437
|
|
|
$values[] = $value; |
438
|
|
|
} |
439
|
|
|
|
440
|
|
|
$set = '?, ?'; |
441
|
|
|
|
442
|
|
|
if (\is_float($value)) { |
443
|
|
|
$types[] = \PDO::PARAM_STR; |
444
|
|
|
$set = '?, ? + 0.0'; |
445
|
|
|
} elseif (\is_int($value)) { |
446
|
|
|
$types[] = \PDO::PARAM_INT; |
447
|
|
|
} elseif (\is_bool($value)) { |
448
|
|
|
$set = '?, ' . ($value ? 'true' : 'false'); |
449
|
|
|
} else { |
450
|
|
|
$types[] = \PDO::PARAM_STR; |
451
|
|
|
} |
452
|
|
|
|
453
|
|
|
$sets[] = $set; |
454
|
|
|
} |
455
|
|
|
} |
456
|
|
|
|
457
|
|
|
$storageName = $command->getStorageName(); |
458
|
|
|
$query->set( |
459
|
|
|
$storageName, |
460
|
|
|
sprintf( |
461
|
|
|
'JSON_SET(IFNULL(%s, "{}"), %s)', |
462
|
|
|
EntityDefinitionQueryHelper::escape($storageName), |
463
|
|
|
implode(', ', $sets) |
464
|
|
|
) |
465
|
|
|
); |
466
|
|
|
|
467
|
|
|
$identifier = $command->getPrimaryKey(); |
468
|
|
|
foreach ($identifier as $key => $_value) { |
469
|
|
|
$query->andWhere(EntityDefinitionQueryHelper::escape($key) . ' = ?'); |
470
|
|
|
} |
471
|
|
|
$query->setParameters([...$values, ...array_values($identifier)], $types); |
472
|
|
|
|
473
|
|
|
RetryableQuery::retryable($this->connection, function () use ($query): void { |
474
|
|
|
$query->executeStatement(); |
475
|
|
|
}); |
476
|
|
|
} |
477
|
|
|
|
478
|
|
|
/** |
479
|
|
|
* @param array<string, mixed> $payload |
480
|
|
|
* |
481
|
|
|
* @return array<string, mixed> |
482
|
|
|
*/ |
483
|
|
|
private function escapeColumnKeys(array $payload): array |
484
|
|
|
{ |
485
|
|
|
$escaped = []; |
486
|
|
|
foreach ($payload as $key => $value) { |
487
|
|
|
$escaped[EntityDefinitionQueryHelper::escape($key)] = $value; |
488
|
|
|
} |
489
|
|
|
|
490
|
|
|
return $escaped; |
491
|
|
|
} |
492
|
|
|
|
493
|
|
|
/** |
494
|
|
|
* @param list<WriteCommand> $commands |
495
|
|
|
*/ |
496
|
|
|
private function generateChangeSets(array $commands): void |
497
|
|
|
{ |
498
|
|
|
$primaryKeys = []; |
499
|
|
|
$definitions = []; |
500
|
|
|
|
501
|
|
|
foreach ($commands as $command) { |
502
|
|
|
if (!$command instanceof ChangeSetAware || !$command instanceof WriteCommand) { |
503
|
|
|
continue; |
504
|
|
|
} |
505
|
|
|
|
506
|
|
|
if (!$command->requiresChangeSet()) { |
507
|
|
|
continue; |
508
|
|
|
} |
509
|
|
|
|
510
|
|
|
$entity = $command->getDefinition()->getEntityName(); |
511
|
|
|
|
512
|
|
|
$primaryKeys[$entity][] = $command->getPrimaryKey(); |
513
|
|
|
$definitions[$entity] = $command->getDefinition(); |
514
|
|
|
} |
515
|
|
|
|
516
|
|
|
if (empty($primaryKeys)) { |
517
|
|
|
return; |
518
|
|
|
} |
519
|
|
|
|
520
|
|
|
$states = []; |
521
|
|
|
foreach ($primaryKeys as $entity => $ids) { |
522
|
|
|
$query = $this->connection->createQueryBuilder(); |
523
|
|
|
|
524
|
|
|
$definition = $definitions[$entity]; |
525
|
|
|
|
526
|
|
|
$query->addSelect('*'); |
527
|
|
|
$query->from(EntityDefinitionQueryHelper::escape($definition->getEntityName())); |
528
|
|
|
|
529
|
|
|
$this->addPrimaryCondition($query, $ids); |
530
|
|
|
|
531
|
|
|
$states[$entity] = $query->executeQuery()->fetchAllAssociative(); |
532
|
|
|
} |
533
|
|
|
|
534
|
|
|
foreach ($commands as $command) { |
535
|
|
|
if (!$command instanceof ChangeSetAware || !$command instanceof WriteCommand) { |
536
|
|
|
continue; |
537
|
|
|
} |
538
|
|
|
|
539
|
|
|
if (!$command->requiresChangeSet()) { |
540
|
|
|
continue; |
541
|
|
|
} |
542
|
|
|
|
543
|
|
|
$entity = $command->getDefinition()->getEntityName(); |
544
|
|
|
|
545
|
|
|
$command->setChangeSet( |
546
|
|
|
$this->calculateChangeSet($command, $states[$entity]) |
547
|
|
|
); |
548
|
|
|
} |
549
|
|
|
} |
550
|
|
|
|
551
|
|
|
/** |
552
|
|
|
* @param list<array<string, string>> $primaryKeys |
553
|
|
|
*/ |
554
|
|
|
private function addPrimaryCondition(DbalQueryBuilderAlias $query, array $primaryKeys): void |
555
|
|
|
{ |
556
|
|
|
$all = []; |
557
|
|
|
$i = 0; |
558
|
|
|
foreach ($primaryKeys as $primaryKey) { |
559
|
|
|
$where = []; |
560
|
|
|
|
561
|
|
|
foreach ($primaryKey as $field => $value) { |
562
|
|
|
++$i; |
563
|
|
|
$field = EntityDefinitionQueryHelper::escape($field); |
564
|
|
|
$where[] = $field . ' = :param' . $i; |
565
|
|
|
$query->setParameter('param' . $i, $value); |
566
|
|
|
} |
567
|
|
|
|
568
|
|
|
$all[] = implode(' AND ', $where); |
569
|
|
|
} |
570
|
|
|
|
571
|
|
|
$query->andWhere(implode(' OR ', $all)); |
572
|
|
|
} |
573
|
|
|
|
574
|
|
|
/** |
575
|
|
|
* @param list<array<string, mixed>> $states |
576
|
|
|
*/ |
577
|
|
|
private function calculateChangeSet(WriteCommand $command, array $states): ChangeSet |
578
|
|
|
{ |
579
|
|
|
foreach ($states as $state) { |
580
|
|
|
// check if current loop matches the command primary key |
581
|
|
|
$primaryKey = array_intersect($command->getPrimaryKey(), $state); |
582
|
|
|
|
583
|
|
|
if (\count(array_diff_assoc($command->getPrimaryKey(), $primaryKey)) === 0) { |
584
|
|
|
return new ChangeSet($state, $command->getPayload(), $command instanceof DeleteCommand); |
585
|
|
|
} |
586
|
|
|
} |
587
|
|
|
|
588
|
|
|
return new ChangeSet([], [], $command instanceof DeleteCommand); |
589
|
|
|
} |
590
|
|
|
|
591
|
|
|
private function getPlaceholders(int $columnCount, int $tupleCount): string |
592
|
|
|
{ |
593
|
|
|
if ($columnCount > 1) { |
594
|
|
|
// multi column pk. Example: (product_id, language_id) IN ((p1, l1), (p2, l2), (px,lx),...) |
595
|
|
|
$tupleStr = '(?' . str_repeat(',?', $columnCount - 1) . ')'; |
596
|
|
|
} else { |
597
|
|
|
// single column pk. Example: category_id IN (c1, c2, c3,...) |
598
|
|
|
$tupleStr = '?'; |
599
|
|
|
} |
600
|
|
|
|
601
|
|
|
return $tupleStr . str_repeat(',' . $tupleStr, $tupleCount - 1); |
602
|
|
|
} |
603
|
|
|
|
604
|
|
|
private function getParentField(EntityDefinition $definition): ?FkField |
605
|
|
|
{ |
606
|
|
|
if (!$definition->isInheritanceAware()) { |
607
|
|
|
return null; |
608
|
|
|
} |
609
|
|
|
|
610
|
|
|
/** @var ManyToOneAssociationField|null $parent */ |
611
|
|
|
$parent = $definition->getFields()->get('parent'); |
612
|
|
|
|
613
|
|
|
if (!$parent) { |
614
|
|
|
throw new ParentFieldNotFoundException($definition); |
615
|
|
|
} |
616
|
|
|
|
617
|
|
|
if (!$parent instanceof ManyToOneAssociationField) { |
618
|
|
|
throw new InvalidParentAssociationException($definition, $parent); |
619
|
|
|
} |
620
|
|
|
|
621
|
|
|
$fk = $definition->getFields()->getByStorageName($parent->getStorageName()); |
622
|
|
|
|
623
|
|
|
if (!$fk) { |
624
|
|
|
throw new CanNotFindParentStorageFieldException($definition); |
625
|
|
|
} |
626
|
|
|
if (!$fk instanceof FkField) { |
627
|
|
|
throw new ParentFieldForeignKeyConstraintMissingException($definition, $fk); |
628
|
|
|
} |
629
|
|
|
|
630
|
|
|
return $fk; |
631
|
|
|
} |
632
|
|
|
|
633
|
|
|
/** |
634
|
|
|
* @param array<string, string> $primaryKey |
635
|
|
|
* |
636
|
|
|
* @return array<string, mixed> |
637
|
|
|
*/ |
638
|
|
|
private function getCurrentState(EntityDefinition $definition, array $primaryKey, WriteCommandQueue $commandQueue): array |
639
|
|
|
{ |
640
|
|
|
$commands = $commandQueue->getCommandsForEntity($definition, $primaryKey); |
641
|
|
|
|
642
|
|
|
$useDatabase = true; |
643
|
|
|
|
644
|
|
|
$state = []; |
645
|
|
|
|
646
|
|
|
foreach ($commands as $command) { |
647
|
|
|
if ($command instanceof DeleteCommand) { |
648
|
|
|
$state = []; |
649
|
|
|
$useDatabase = false; |
650
|
|
|
|
651
|
|
|
continue; |
652
|
|
|
} |
653
|
|
|
|
654
|
|
|
if (!$command instanceof InsertCommand && !$command instanceof UpdateCommand) { |
655
|
|
|
continue; |
656
|
|
|
} |
657
|
|
|
|
658
|
|
|
$state = array_replace_recursive($state, $command->getPayload()); |
659
|
|
|
|
660
|
|
|
if ($command instanceof InsertCommand) { |
661
|
|
|
$useDatabase = false; |
662
|
|
|
} |
663
|
|
|
} |
664
|
|
|
|
665
|
|
|
if (!$useDatabase) { |
666
|
|
|
return $state; |
667
|
|
|
} |
668
|
|
|
|
669
|
|
|
$decodedPrimaryKey = []; |
670
|
|
|
foreach ($primaryKey as $fieldName => $fieldValue) { |
671
|
|
|
$field = $definition->getField($fieldName); |
672
|
|
|
$decodedPrimaryKey[$fieldName] = $field ? $field->getSerializer()->decode($field, $fieldValue) : $fieldValue; |
673
|
|
|
} |
674
|
|
|
|
675
|
|
|
$currentState = $this->primaryKeyBag === null ? null : $this->primaryKeyBag->getExistenceState($definition, $decodedPrimaryKey); |
676
|
|
|
if ($currentState === null) { |
677
|
|
|
$currentState = $this->fetchFromDatabase($definition, $primaryKey); |
678
|
|
|
} |
679
|
|
|
|
680
|
|
|
$parent = $this->getParentField($definition); |
681
|
|
|
|
682
|
|
|
if ($parent && \array_key_exists('parent', $currentState)) { |
683
|
|
|
$currentState[$parent->getStorageName()] = $currentState['parent']; |
684
|
|
|
unset($currentState['parent']); |
685
|
|
|
} |
686
|
|
|
|
687
|
|
|
return array_replace_recursive($currentState, $state); |
688
|
|
|
} |
689
|
|
|
|
690
|
|
|
/** |
691
|
|
|
* @param array<string, string> $primaryKey |
692
|
|
|
* |
693
|
|
|
* @return array<string, mixed> |
694
|
|
|
*/ |
695
|
|
|
private function fetchFromDatabase(EntityDefinition $definition, array $primaryKey): array |
696
|
|
|
{ |
697
|
|
|
$query = $this->connection->createQueryBuilder(); |
698
|
|
|
$query->from(EntityDefinitionQueryHelper::escape($definition->getEntityName())); |
699
|
|
|
|
700
|
|
|
$fields = $definition->getPrimaryKeys(); |
701
|
|
|
|
702
|
|
|
/** @var Field&StorageAware $field */ |
703
|
|
|
foreach ($fields as $field) { |
704
|
|
|
if (!\array_key_exists($field->getStorageName(), $primaryKey)) { |
705
|
|
|
if (!\array_key_exists($field->getPropertyName(), $primaryKey)) { |
706
|
|
|
throw new PrimaryKeyNotProvidedException($definition, $field); |
707
|
|
|
} |
708
|
|
|
|
709
|
|
|
$primaryKey[$field->getStorageName()] = $primaryKey[$field->getPropertyName()]; |
710
|
|
|
unset($primaryKey[$field->getPropertyName()]); |
711
|
|
|
} |
712
|
|
|
|
713
|
|
|
$param = 'param_' . Uuid::randomHex(); |
714
|
|
|
$query->andWhere(EntityDefinitionQueryHelper::escape($field->getStorageName()) . ' = :' . $param); |
715
|
|
|
$query->setParameter($param, $primaryKey[$field->getStorageName()]); |
716
|
|
|
} |
717
|
|
|
|
718
|
|
|
$query->addSelect('1 as `exists`'); |
719
|
|
|
|
720
|
|
|
if ($definition->isChildrenAware()) { |
721
|
|
|
$query->addSelect('parent_id'); |
722
|
|
|
} elseif ($definition->isInheritanceAware()) { |
723
|
|
|
$parent = $this->getParentField($definition); |
724
|
|
|
|
725
|
|
|
if ($parent !== null) { |
726
|
|
|
$query->addSelect( |
727
|
|
|
EntityDefinitionQueryHelper::escape($parent->getStorageName()) |
728
|
|
|
. ' as `parent`' |
729
|
|
|
); |
730
|
|
|
} |
731
|
|
|
} |
732
|
|
|
|
733
|
|
|
$exists = $query->executeQuery()->fetchAssociative(); |
734
|
|
|
if (!$exists) { |
735
|
|
|
$exists = []; |
736
|
|
|
} |
737
|
|
|
|
738
|
|
|
return $exists; |
739
|
|
|
} |
740
|
|
|
|
741
|
|
|
/** |
742
|
|
|
* @param array<string, mixed> $data |
743
|
|
|
* @param array<string, mixed> $state |
744
|
|
|
* @param array<string, string> $primaryKey |
745
|
|
|
*/ |
746
|
|
|
private function isChild(EntityDefinition $definition, array $data, array $state, array $primaryKey, WriteCommandQueue $commandQueue): bool |
747
|
|
|
{ |
748
|
|
|
if ($definition instanceof EntityTranslationDefinition) { |
749
|
|
|
return $this->isTranslationChild($definition, $primaryKey, $commandQueue); |
750
|
|
|
} |
751
|
|
|
|
752
|
|
|
if (!$definition->isInheritanceAware()) { |
753
|
|
|
return false; |
754
|
|
|
} |
755
|
|
|
|
756
|
|
|
$fk = $this->getParentField($definition); |
757
|
|
|
|
758
|
|
|
\assert($fk instanceof FkField); |
759
|
|
|
|
760
|
|
|
// foreign key provided, !== null has parent otherwise not |
761
|
|
|
if (\array_key_exists($fk->getPropertyName(), $data)) { |
762
|
|
|
return isset($data[$fk->getPropertyName()]); |
763
|
|
|
} |
764
|
|
|
|
765
|
|
|
/** @var Field $association */ |
766
|
|
|
$association = $definition->getFields()->get('parent'); |
767
|
|
|
if (isset($data[$association->getPropertyName()])) { |
768
|
|
|
return true; |
769
|
|
|
} |
770
|
|
|
|
771
|
|
|
return isset($state[$fk->getStorageName()]); |
772
|
|
|
} |
773
|
|
|
|
774
|
|
|
/** |
775
|
|
|
* @param array<string, mixed> $state |
776
|
|
|
*/ |
777
|
|
|
private function wasChild(EntityDefinition $definition, array $state): bool |
778
|
|
|
{ |
779
|
|
|
if (!$definition->isInheritanceAware()) { |
780
|
|
|
return false; |
781
|
|
|
} |
782
|
|
|
|
783
|
|
|
$fk = $this->getParentField($definition); |
784
|
|
|
|
785
|
|
|
return $fk !== null && isset($state[$fk->getStorageName()]); |
786
|
|
|
} |
787
|
|
|
|
788
|
|
|
/** |
789
|
|
|
* @param array<string, string> $primaryKey |
790
|
|
|
*/ |
791
|
|
|
private function isTranslationChild(EntityTranslationDefinition $definition, array $primaryKey, WriteCommandQueue $commandQueue): bool |
792
|
|
|
{ |
793
|
|
|
$parent = $definition->getParentDefinition(); |
794
|
|
|
|
795
|
|
|
if (!$parent->isInheritanceAware()) { |
796
|
|
|
return false; |
797
|
|
|
} |
798
|
|
|
|
799
|
|
|
/** @var FkField $fkField */ |
800
|
|
|
$fkField = $definition->getFields()->getByStorageName( |
801
|
|
|
$parent->getEntityName() . '_id' |
802
|
|
|
); |
803
|
|
|
$parentPrimaryKey = [ |
804
|
|
|
'id' => $primaryKey[$fkField->getStorageName()], |
805
|
|
|
]; |
806
|
|
|
|
807
|
|
|
if ($parent->isVersionAware()) { |
808
|
|
|
$parentPrimaryKey['versionId'] = $primaryKey[$parent->getEntityName() . '_version_id']; |
809
|
|
|
} |
810
|
|
|
|
811
|
|
|
$existence = $this->getExistence($parent, $parentPrimaryKey, [], $commandQueue); |
812
|
|
|
|
813
|
|
|
return $existence->isChild(); |
814
|
|
|
} |
815
|
|
|
} |
816
|
|
|
|