Graviton\DocumentBundle\Listener\DocumentVersionListener

Complexity

Total Complexity

5

Size/Duplication

Total Lines	64
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

0%

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	modelUpdate()	0	8	1
A	modelInsert()	0	8	1
A	updateCounter()	0	15	2

<?php
/**
 * Custom Model Document Version listener
 */

namespace Graviton\DocumentBundle\Listener;

use Graviton\RestBundle\Event\ModelEvent;
use Doctrine\ODM\MongoDB\DocumentManager;

/**
 * Class DocumentVersionListener
 * @package Graviton\DocumentBundle\Listener
 *
 * @author   List of contributors <https://github.com/libgraviton/graviton/graphs/contributors>
 * @license  http://opensource.org/licenses/gpl-license.php GNU Public License
 * @link     http://swisscom.ch
 */
class DocumentVersionListener
{
    /** @var DocumentManager Document manager */
    private $documentManager;

    /**
     * constructor.
     * @param DocumentManager $documentManager Db Connection document manager
     */
    public function __construct(DocumentManager $documentManager)
    {
        $this->documentManager = $documentManager;
    }

    /**
     * Updating a Model
     * @param ModelEvent $event Mongo.odm event argument
     * @return void
     */
    public function modelUpdate(ModelEvent $event)
    {
        $this->updateCounter(
            $event->getCollectionClass(),
            $event->getCollectionId(),
            'update'
        );
    }
    /**
     * Insert a Model
     * @param ModelEvent $event Mongo.odm event argument
     * @return void
     */
    public function modelInsert(ModelEvent $event)
    {
        $this->updateCounter(
            $event->getCollectionClass(),
            $event->getCollectionId(),
            'insert'
        );
    }

    /**
     * Update Counter for all new saved items
     *
     * @param string $collection Qualified Object class
     * @param string $id         Id to be updated
     * @param string $action     What is to be done
     */
    private function updateCounter($collection, $id, $action)
    {
        $qb = $this->documentManager->createQueryBuilder($collection);
        if ('update'==$action) {
            $qb->findAndUpdate()
                ->field('id')->equals($id)
                ->field('version')->inc(1)
                ->getQuery()->execute();
        } else {
            $qb->findAndUpdate()
                ->field('id')->equals($id)
                ->field('version')->set(1)
                ->getQuery()->execute();
        }
    }
}