CallbackHandler.php 2.06 KB
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Magento\Framework\EntityManager;

use Magento\Framework\Model\CallbackPool;
use Psr\Log\LoggerInterface;

/**
 * Class CallbackHandler
 */
class CallbackHandler
{
    /**
     * @var MetadataPool
     */
    protected $metadataPool;

    /**
     * @var LoggerInterface
     */
    protected $logger;

    /**
     * CallbackHandler constructor.
     *
     * @param MetadataPool $metadataPool
     * @param LoggerInterface $logger
     */
    public function __construct(
        MetadataPool $metadataPool,
        LoggerInterface $logger
    ) {
        $this->metadataPool = $metadataPool;
        $this->logger = $logger;
    }

    /**
     * @param string $entityType
     * @throws \Exception
     * @return void
     */
    public function process($entityType)
    {
        $metadata = $this->metadataPool->getMetadata($entityType);
        $connection = $metadata->getEntityConnection();
        $hash = spl_object_hash($connection);
        if ($connection->getTransactionLevel() === 0) {
            $callbacks = CallbackPool::get($hash);
            try {
                foreach ($callbacks as $callback) {
                    call_user_func($callback);
                }
            } catch (\Exception $e) {
                $this->logger->error($e->getMessage(), $e->getTrace());
                throw $e;
            }
        }
    }

    /**
     * @param string $entityType
     * @param array $callback
     * @throws \Exception
     * @return void
     */
    public function attach($entityType, $callback)
    {
        $metadata = $this->metadataPool->getMetadata($entityType);
        CallbackPool::attach(spl_object_hash($metadata->getEntityConnection()), $callback);
    }

    /**
     * @param string $entityType
     * @throws \Exception
     * @return void
     */
    public function clear($entityType)
    {
        $metadata = $this->metadataPool->getMetadata($entityType);
        CallbackPool::clear(spl_object_hash($metadata->getEntityConnection()));
    }
}