PublisherPool.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\MessageQueue;

use Magento\Framework\MessageQueue\ConfigInterface as QueueConfig;
use Magento\Framework\Communication\ConfigInterface as CommunicationConfig;
use Magento\Framework\MessageQueue\Publisher\ConfigInterface as PublisherConfig;

class PublisherPool implements PublisherInterface, BulkPublisherInterface
{
    const MODE_SYNC = 'sync';
    const MODE_ASYNC = 'async';

    const TYPE = 'type';

    const CONNECTION_NAME = 'connectionName';

    protected $publishers = [];

    protected $communicationConfig;

    private $publisherConfig;

    private $connectionTypeResolver;

    public function __construct(
        CommunicationConfig $communicationConfig,
        QueueConfig $queueConfig,
        array $publishers
    ) {
        $this->communicationConfig = $communicationConfig;
        $this->initializePublishers($publishers);
    }

    public function publish($topicName, $data)
    {
        $publisherType = $this->communicationConfig->getTopic($topicName)[CommunicationConfig::TOPIC_IS_SYNCHRONOUS]
            ? self::MODE_SYNC
            : self::MODE_ASYNC;
        $connectionName = $this->getPublisherConfig()->getPublisher($topicName)->getConnection()->getName();
        $publisher = $this->getPublisherForConnectionNameAndType($publisherType, $connectionName);
        return $publisher->publish($topicName, $data);
    }

    private function initializePublishers(array $publishers)
    {
        $asyncPublishers = isset($publishers[self::MODE_ASYNC]) ? $publishers[self::MODE_ASYNC] : [];
        $syncPublishers = isset($publishers[self::MODE_SYNC]) ? $publishers[self::MODE_SYNC] : [];
        foreach ($asyncPublishers as $connectionType => $publisher) {
            $this->addPublisherToPool(
                self::MODE_ASYNC,
                $connectionType,
                $publisher
            );
        }
        foreach ($syncPublishers as $connectionType => $publisher) {
            $this->addPublisherToPool(
                self::MODE_SYNC,
                $connectionType,
                $publisher
            );
        }
    }

    private function addPublisherToPool($type, $connectionType, PublisherInterface $publisher)
    {
        $this->publishers[$type][$connectionType] = $publisher;
        return $this;
    }

    private function getPublisherForConnectionNameAndType($type, $connectionName)
    {
        $connectionType = $this->getConnectionTypeResolver()->getConnectionType($connectionName);
        if (!isset($this->publishers[$type])) {
            throw new \InvalidArgumentException('Unknown publisher type ' . $type);
        }

        if (!isset($this->publishers[$type][$connectionType])) {
            throw new \LogicException(
                sprintf(
                    'Could not find an implementation type for type "%s" and connection "%s".',
                    $type,
                    $connectionName
                )
            );
        }
        return $this->publishers[$type][$connectionType];
    }

    private function getPublisherConfig()
    {
        if ($this->publisherConfig === null) {
            $this->publisherConfig = \Magento\Framework\App\ObjectManager::getInstance()->get(PublisherConfig::class);
        }
        return $this->publisherConfig;
    }

    private function getConnectionTypeResolver()
    {
        if ($this->connectionTypeResolver === null) {
            $this->connectionTypeResolver = \Magento\Framework\App\ObjectManager::getInstance()
                ->get(ConnectionTypeResolver::class);
        }
        return $this->connectionTypeResolver;
    }
}