vendor/shopware/elasticsearch/Product/CustomFieldUpdater.php line 38

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Elasticsearch\Product;
  5. use Elasticsearch\Client;
  6. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenContainerEvent;
  7. use Shopware\Core\System\CustomField\CustomFieldDefinition;
  8. use Shopware\Core\System\CustomField\CustomFieldTypes;
  9. use Shopware\Elasticsearch\Framework\ElasticsearchHelper;
  10. use Shopware\Elasticsearch\Framework\ElasticsearchOutdatedIndexDetector;
  11. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  13. class CustomFieldUpdater implements EventSubscriberInterface
  14. {
  15.     private ElasticsearchOutdatedIndexDetector $indexDetector;
  17.     private Client $client;
  19.     private ElasticsearchHelper $elasticsearchHelper;
  21.     /**
  22.      * @internal
  23.      */
  24.     public function __construct(ElasticsearchOutdatedIndexDetector $indexDetector, Client $client, ElasticsearchHelper $elasticsearchHelper)
  25.     {
  26.         $this->indexDetector = $indexDetector;
  27.         $this->client = $client;
  28.         $this->elasticsearchHelper = $elasticsearchHelper;
  29.     }
  31.     public static function getSubscribedEvents(): array
  32.     {
  33.         return [
  34.             EntityWrittenContainerEvent::class => 'onNewCustomFieldCreated',
  35.         ];
  36.     }
  38.     public function onNewCustomFieldCreated(EntityWrittenContainerEvent $containerEvent): void
  39.     {
  40.         $event = $containerEvent->getEventByEntityName(CustomFieldDefinition::ENTITY_NAME);
  42.         if ($event === null) {
  43.             return;
  44.         }
  46.         if (!$this->elasticsearchHelper->allowIndexing()) {
  47.             return;
  48.         }
  50.         $newCreatedFields = [];
  52.         foreach ($event->getWriteResults() as $writeResult) {
  53.             $existence = $writeResult->getExistence();
  55.             if ($existence && $existence->exists()) {
  56.                 continue;
  57.             }
  59.             $esType = self::getTypeFromCustomFieldType($writeResult->getProperty('type'));
  61.             if ($esType === null) {
  62.                 continue;
  63.             }
  65.             $newCreatedFields[$writeResult->getProperty('name')] = $esType;
  66.         }
  68.         if (\count($newCreatedFields) === 0) {
  69.             return;
  70.         }
  72.         $this->createNewFieldsInIndices($newCreatedFields);
  73.     }
  75.     public static function getTypeFromCustomFieldType(string $type): ?array
  76.     {
  77.         switch ($type) {
  78.             case CustomFieldTypes::INT:
  79.                 return [
  80.                     'type' => 'long',
  81.                 ];
  82.             case CustomFieldTypes::FLOAT:
  83.                 return [
  84.                     'type' => 'double',
  85.                 ];
  86.             case CustomFieldTypes::BOOL:
  87.                 return [
  88.                     'type' => 'boolean',
  89.                 ];
  90.             case CustomFieldTypes::DATETIME:
  91.                 return [
  92.                     'type' => 'date',
  93.                     'format' => 'yyyy-MM-dd HH:mm:ss.000',
  94.                     'ignore_malformed' => true,
  95.                 ];
  96.             case CustomFieldTypes::JSON:
  97.                 return [
  98.                     'type' => 'object',
  99.                     'dynamic' => true,
  100.                 ];
  101.             case CustomFieldTypes::HTML:
  102.             case CustomFieldTypes::TEXT:
  103.                 return [
  104.                     'type' => 'text',
  105.                 ];
  106.             case CustomFieldTypes::COLORPICKER:
  107.             case CustomFieldTypes::ENTITY:
  108.             case CustomFieldTypes::MEDIA:
  109.             case CustomFieldTypes::SELECT:
  110.             case CustomFieldTypes::SWITCH:
  111.             default:
  112.                 return [
  113.                     'type' => 'keyword',
  114.                 ];
  115.         }
  116.     }
  118.     private function createNewFieldsInIndices(array $newCreatedFields): void
  119.     {
  120.         $indices = $this->indexDetector->getAllUsedIndices();
  122.         foreach ($indices as $indexName) {
  123.             $body = [
  124.                 'properties' => [
  125.                     'customFields' => [
  126.                         'properties' => $newCreatedFields,
  127.                     ],
  128.                 ],
  129.             ];
  131.             // For some reason, we need to include the includes to prevent merge conflicts.
  132.             // This error can happen for example after updating from version <6.4.
  133.             $current = $this->client->indices()->get(['index' => $indexName]);
  134.             $includes = $current[$indexName]['mappings']['_source']['includes'] ?? [];
  135.             if ($includes !== []) {
  136.                 $body['_source'] = [
  137.                     'includes' => $includes,
  138.                 ];
  139.             }
  141.             $this->client->indices()->putMapping([
  142.                 'index' => $indexName,
  143.                 'body' => $body,
  144.             ]);
  145.         }
  146.     }
  147. }