SequenceApplier.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\EntityManager\Sequence;

class SequenceApplier
{
    private $metadataPool;

    private $typeResolver;

    private $sequenceManager;

    private $sequenceRegistry;

    private $hydratorPool;

    public function __construct(
        \Magento\Framework\EntityManager\MetadataPool $metadataPool,
        \Magento\Framework\EntityManager\TypeResolver $typeResolver,
        \Magento\Framework\EntityManager\Sequence\SequenceManager $sequenceManager,
        \Magento\Framework\EntityManager\Sequence\SequenceRegistry $sequenceRegistry,
        \Magento\Framework\EntityManager\HydratorPool $hydratorPool
    ) {
        $this->metadataPool = $metadataPool;
        $this->typeResolver = $typeResolver;
        $this->sequenceManager = $sequenceManager;
        $this->sequenceRegistry = $sequenceRegistry;
        $this->hydratorPool = $hydratorPool;
    }

    public function apply($entity)
    {
        $entityType = $this->typeResolver->resolve($entity);

        $sequence = $this->sequenceRegistry->retrieve($entityType)['sequence'];

        if ($sequence) {
            $metadata = $this->metadataPool->getMetadata($entityType);
            $hydrator = $this->hydratorPool->getHydrator($entityType);

            $entityData = $hydrator->extract($entity);

            // Object already has identifier.
            if (isset($entityData[$metadata->getIdentifierField()]) && $entityData[$metadata->getIdentifierField()]) {
                $this->sequenceManager->force($entityType, $entityData[$metadata->getIdentifierField()]);
            } else {
                $entityData[$metadata->getIdentifierField()] = $sequence->getNextValue();

                $entity = $hydrator->hydrate($entity, $entityData);
            }
        }

        return $entity;
    }
}