NovikovViktor/RequestLimitBundle

<?php

namespace NW\RequestLimitBundle\Storage\Provider;

use Doctrine\ORM\EntityManager;
use Doctrine\DBAL\Driver\Exception as DriverException;
use \Doctrine\DBAL\Exception;
use \PDO;

/**
 * Class MySQLProvider
 * @package NW\RequestLimitBundle\Storage\Provider
 * @author Novikov Viktor
 */
class MySQLProvider implements ProviderInterface
{
    /**
     * @var EntityManager $_em
     */
    private $_em;

    /**
     * @param EntityManager $em
     */
    public function __construct(EntityManager $em)
    {
        $this->_em = $em;
    }

    /**
     * @param $configuration
     */
    public function configure($configuration)
    {
        // no additional configuration needed, as we use Doctrine entity manager
    }

    /**
     * {@inheritdoc}
     *
     * @throws DriverException
     * @throws Exception
     */
    public function get($key)
    {
        $connection = $this->_em->getConnection();
        $statement = $connection->prepare('SELECT expires_at FROM nw_request_limit_items WHERE item_key = :item_key');
        $statement->bindValue('item_key', $key);
        $statement->execute();
        $result = $statement->fetchAll(PDO::FETCH_COLUMN);

        if (!$result) {
            return null;
        }

        $connection->close();
        $date = \DateTime::createFromFormat('Y-m-d H:i:s', $result[0]);

        return $date->getTimestamp();
    }

    /**
     * {@inheritdoc}
     *
     * @throws DriverException
     * @throws Exception
     */
    public function set($key, $expiresAt)
    {
        $connection = $this->_em->getConnection();
        $statement = $connection->prepare(
            'INSERT INTO nw_request_limit_items (item_key, expires_at) VALUES (:item_key, :expires_at);'
        );
        $statement->bindValue('item_key', $key);
        $statement->bindValue('expires_at', date('Y-m-d H:i:s', $expiresAt));
        $statement->execute();
        $connection->close();
    }

    /**
     * {@inheritdoc}
     *
     * @throws DriverException
     * @throws Exception
     */
    public function remove($key)
    {
        $connection = $this->_em->getConnection();
        $statement = $connection->prepare('DELETE FROM nw_request_limit_items WHERE item_key = :item_key');
        $statement->bindValue('item_key', $key);
        $statement->execute();
        $connection->close();
    }

    /**
     * {@inheritdoc}
     *
     * @throws DriverException
     * @throws Exception
     */
    public function fetchAllItems()
    {
        $connection = $this->_em->getConnection();
        $statement = $connection->prepare('SELECT * FROM nw_request_limit_items');
        $statement->execute();
        $result = $statement->fetchAll();
        $connection->close();

        return $result;
    }

    /**
     * {@inheritdoc}
     *
     * @throws DriverException
     * @throws Exception
     */
    public function getItemsCount()
    {
        $connection = $this->_em->getConnection();
        $statement = $connection->prepare('SELECT COUNT(*) FROM nw_request_limit_items');
        $statement->execute();
        $result = $statement->fetchColumn(0);
        $connection->close();

        return $result;
    }
}