vendor/pimcore/portal-engine/src/EventSubscriber/SizeEstimationStrategyListener.php line 46

Open in your IDE?
  1. <?php
  3. /**
  4.  * Pimcore
  5.  *
  6.  * This source file is available under following license:
  7.  * - Pimcore Commercial License (PCL)
  8.  *
  9.  *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
  10.  *  @license    http://www.pimcore.org/license     PCL
  11.  */
  13. namespace Pimcore\Bundle\PortalEngineBundle\EventSubscriber;
  15. use Pimcore\Bundle\PortalEngineBundle\Event\DataObject\ExtractMappingEvent;
  16. use Pimcore\Bundle\PortalEngineBundle\Event\DataObject\UpdateIndexDataEvent;
  17. use Pimcore\Bundle\PortalEngineBundle\Service\Download\SizeEstimation\SizeEstimationStrategyInterface;
  18. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  20. class SizeEstimationStrategyListener implements EventSubscriberInterface
  21. {
  22.     protected $sizeEstimationStrategy;
  24.     public function __construct(SizeEstimationStrategyInterface $sizeEstimationStrategy)
  25.     {
  26.         $this->sizeEstimationStrategy = $sizeEstimationStrategy;
  27.     }
  29.     public static function getSubscribedEvents()
  30.     {
  31.         return [
  32.             ExtractMappingEvent::class => 'onExtractMapping',
  33.             UpdateIndexDataEvent::class => 'onUpdateIndexData'
  34.         ];
  35.     }
  37.     public function onExtractMapping(ExtractMappingEvent $event)
  38.     {
  39.         $mappings = $event->getCustomFieldsMapping();
  41.         $mappings = array_merge($mappings, $this->sizeEstimationStrategy->getCustomDataObjectMappingForIndex($event->getClassDefinition()));
  43.         $event->setCustomFieldsMapping($mappings);
  44.     }
  46.     public function onUpdateIndexData(UpdateIndexDataEvent $event)
  47.     {
  48.         $data = $event->getCustomFields();
  50.         $event->setCustomFields(array_replace($data, $this->sizeEstimationStrategy->getCustomDataObjectDataForIndex($event->getDataObject())));
  51.     }
  52. }