Opensoft\RolloutBundle\Storage\Doctrine\DoctrineORMStorage

Complexity

Total Complexity

7

Size/Duplication

Total Lines	75
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	get()	0	10	2
A	set()	0	14	2
A	remove()	0	8	2

<?php

namespace Opensoft\RolloutBundle\Storage\Doctrine;

use Doctrine\ORM\EntityManagerInterface;
use Opensoft\Rollout\Storage\StorageInterface;
use Opensoft\RolloutBundle\Entity\Feature;

/**
 * @author Richard Fullmer <richard.fullmer@opensoftdev.com>
 */
class DoctrineORMStorage implements StorageInterface
{
    /**
     * @var \Doctrine\ORM\EntityManagerInterface
     */
    protected $em;

    /**
     * @var \Doctrine\ORM\EntityRepository
     */
    protected $repository;

    /**
     * @var string
     */
    protected $class;

    /**
     * @param EntityManagerInterface $em
     * @param string $class
     */
    public function __construct(EntityManagerInterface $em, $class)
    {
        $this->em = $em;
        $this->repository = $em->getRepository($class);
        $this->class = $class;
    }

    /**
     * @param string key
     *
     * @return mixed|null Null if the value is not found
     */
    public function get($key)
    {
        /** @var Feature $feature */
        $feature = $this->repository->findOneBy(array('name' => $key));
        if (!$feature) {
            return null;
        }

        return $feature->getSettings();
    }

    /**
     * @param string $key
     * @param mixed $value
     */
    public function set($key, $value)
    {
        /** @var Feature $feature */
        $feature = $this->repository->findOneBy(array('name' => $key));
        if (!$feature) {
            $feature = new Feature();
        }

        $feature->setName($key);
        $feature->setSettings($value);

        $this->em->persist($feature);
        $this->em->flush($feature);
    }

    /**
     * @param string $key
     */
    public function remove($key)
    {
        $feature = $this->repository->findOneBy(array('name' => $key));
        if ($feature) {
            $this->em->remove($feature);
            $this->em->flush($feature);
        }
    }
}