OperationPool.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\EntityManager;

use Magento\Framework\ObjectManagerInterface as ObjectManager;
use Magento\Framework\EntityManager\Operation\CheckIfExists;
use Magento\Framework\EntityManager\Operation\Read;
use Magento\Framework\EntityManager\Operation\Create;
use Magento\Framework\EntityManager\Operation\Update;
use Magento\Framework\EntityManager\Operation\Delete;

class OperationPool
{
    private $defaultOperations = [
        'checkIfExists' => CheckIfExists::class,
        'read' => Read::class,
        'create' => Create::class,
        'update' => Update::class,
        'delete' => Delete::class,
    ];

    private $operations;

    private $objectManager;

    public function __construct(
        ObjectManager $objectManager,
        $operations = []
    ) {
        $this->objectManager = $objectManager;
        $this->operations = array_replace_recursive(
            ['default' => $this->defaultOperations],
            $operations
        );
    }

    public function getOperation($entityType, $operationName)
    {
        if (!isset($this->operations[$entityType][$operationName])) {
            return $this->objectManager->get($this->operations['default'][$operationName]);
        }
        return $this->objectManager->get($this->operations[$entityType][$operationName]);
    }
}