Sokil\Mongo\Queue

Complexity

Total Complexity

8

Size/Duplication

Total Lines	96
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	enqueue()	0	12	1
A	dequeuePlain()	0	16	2
A	dequeue()	0	12	2
A	count()	0	4	1
A	clear()	0	5	1

<?php

/**
 * This file is part of the PHPMongo package.
 *
 * (c) Dmytro Sokil <dmytro.sokil@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Sokil\Mongo;

class Queue implements \Countable
{
    /**
     *
     * @var \Sokil\Mongo\Collection
     */
    private $collection;
    
    public function __construct(Database $database, $channel)
    {
        $this->collection = $database
            ->getCollection($channel)
            ->disableDocumentPool();
    }
    
    /**
     * Add item to queue
     *
     * @param mixed $payload data to send
     * @param int $priority more priority num give quicker getting from queue
     * @return \Sokil\Mongo\Queue
     */
    public function enqueue($payload, $priority = 0)
    {
        $this->collection
            ->createDocument(array(
                'payload'   => $payload,
                'priority'  => (int) $priority,
                'datetime'  => new \MongoDate,
            ))
            ->save();
        
        return $this;
    }
    
    /**
     * Get item from queue as is
     *
     * @return mixed
     */
    public function dequeuePlain()
    {
        $document = $this->collection
            ->find()
            ->sort(array(
                'priority' => -1,
                'datetime' => 1,
            ))
            ->findAndRemove();
        
        if (!$document) {
            return null;
        }
        
        return $document->get('payload');
    }
    
    /**
     * Get item from queue as Structure if array put into queue
     *
     * @return mixed|\Sokil\Mongo\Structure
     */
    public function dequeue()
    {
        $value = $this->dequeuePlain();
        if (!is_array($value)) {
            return $value;
        }

        $structure = new Structure();
        $structure->mergeUnmodified($value);

        return $structure;
    }
    
    /**
     * Get number of elements in queue
     *
     * @return int
     */
    public function count()
    {
        return count($this->collection);
    }
    
    /**
     * Clear queue
     *
     * @return \Sokil\Mongo\Queue
     */
    public function clear()
    {
        $this->collection->delete();
        return $this;
    }
}