Graze\Queue\Handler\AbstractAcknowledgementHandler

Complexity

Total Complexity

5

Size/Duplication

Total Lines	72
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
B	__invoke()	0	25	5

<?php

/**
 * This file is part of graze/queue.
 *
 * Copyright (c) 2015 Nature Delivered Ltd. <https://www.graze.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @license https://github.com/graze/queue/blob/master/LICENSE MIT
 *
 * @link    https://github.com/graze/queue
 */

namespace Graze\Queue\Handler;

use Exception;
use Graze\Queue\Adapter\AdapterInterface;
use Graze\Queue\Message\MessageInterface;
use Iterator;

abstract class AbstractAcknowledgementHandler
{
    /**
     * @param MessageInterface $message
     * @param AdapterInterface $adapter
     * @param mixed            $result
     */
    abstract protected function acknowledge(
        MessageInterface $message,
        AdapterInterface $adapter,
        $result = null
    );

    /**
     * @param MessageInterface $message
     * @param AdapterInterface $adapter
     * @param int              $duration Number of seconds to ensure that this message is not seen by any other clients
     */
    abstract protected function extend(
        MessageInterface $message,
        AdapterInterface $adapter,
        $duration
    );

    /**
     * @param MessageInterface $message
     * @param AdapterInterface $adapter
     * @param mixed            $result
     */
    abstract protected function reject(
        MessageInterface $message,
        AdapterInterface $adapter,
        $result = null
    );

    /**
     * @param AdapterInterface $adapter
     */
    abstract protected function flush(AdapterInterface $adapter);

    /**
     * @param Iterator         $messages
     * @param AdapterInterface $adapter
     * @param callable         $worker
     *
     * @throws Exception
     */
    public function __invoke(Iterator $messages, AdapterInterface $adapter, callable $worker)
    {
        // Used to break from polling consumer
        $break = false;
        $done = function () use (&$break) {
            $break = true;
        };

        try {
            foreach ($messages as $message) {
                if ($message->isValid()) {
                    $result = call_user_func($worker, $message, $done);
                    $this->acknowledge($message, $adapter, $result);
                }

                if ($break) {
                    break;
                }
            }
        } catch (Exception $e) {
            $this->flush($adapter);
            throw $e;
        }

        $this->flush($adapter);
    }
}