<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\MessageQueue\Model;

use Magento\Framework\MessageQueue\CountableQueueInterface;
use Magento\Framework\MessageQueue\QueueRepository;

/**
 * Class CheckIsAvailableMessagesInQueue for checking messages available in queue
 */
class CheckIsAvailableMessagesInQueue
{
    /**
     * @var QueueRepository
     */
    private $queueRepository;

    /**
     * Initialize dependencies.
     *
     * @param QueueRepository $queueRepository
     */
    public function __construct(QueueRepository $queueRepository)
    {
        $this->queueRepository = $queueRepository;
    }

    /**
     * Checks if there is available messages in the queue
     *
     * @param string $connectionName connection name
     * @param string $queueName queue name
     * @return bool
     * @throws \LogicException if queue is not available
     */
    public function execute($connectionName, $queueName)
    {
        $queue = $this->queueRepository->get($connectionName, $queueName);
        if ($queue instanceof CountableQueueInterface) {
            return $queue->count() > 0;
        }
        $message = $queue->dequeue();
        if ($message) {
            $queue->reject($message);
            return true;
        }
        return false;
    }
}