php-yaoi/php-yaoi

<?php

namespace Yaoi\Storage\Driver;

use Yaoi\Storage\Contract\Driver;
use Yaoi\Storage\Contract\Expire;
use Yaoi\Storage\Exception;
use Yaoi\Storage\Settings;

class Memcache implements Driver, Expire
{
    /**
     * @var Settings
     */
    protected $dsn;

    public function __construct(Settings $dsn = null)
    {
        $this->dsn = $dsn;
    }

    /**
     * @var \Memcache
     */
    protected $memcache;

    protected function connect()
    {
        $this->memcache = new \Memcache();
        $hostname = $this->dsn->unixSocket
            ? 'unix://' . $this->dsn->unixSocket
            : ($this->dsn->hostname
                ? $this->dsn->hostname
                : '127.0.0.1');
        if ($this->dsn->unixSocket) {
            $port = 0;
        } else {
            $port = $this->dsn->port
                ? $this->dsn->port
                : ini_get(' memcache.default_port');
            if (!$port) {
                $port = 11211;
            }
        }
        $timeout = $this->dsn->connectionTimeout;
        $result = $this->memcache->connect($hostname, $port, $timeout);
        if (!$result) {
            throw new Exception("Connection failed ($hostname, $port, $timeout)", Exception::CONNECTION_FAILED);
        }
        return $result;
    }

    public function set($key, $value, $ttl)
    {
        if (null === $this->memcache) {
            $this->connect();
        }
        $result = $this->memcache->set($key, $value, 0, $ttl);
        if (!$result) {
            throw new Exception('Set failed', Exception::SET_FAILED);
        }
        return $result;
    }

    function keyExists($key)
    {
        if (null === $this->memcache) {
            $this->connect();
        }
        if ($this->memcache->add($key, null)) {
            $this->memcache->delete($key);
            return false;
        } else {
            return true;
        }
    }

    public function get($key)
    {
        if (null === $this->memcache) {
            $this->connect();
        }
        return $this->memcache->get($key);
    }

    function delete($key)
    {
        if (null === $this->memcache) {
            $this->connect();
        }
        return $this->memcache->delete($key);
    }

    function deleteAll()
    {
        if (null === $this->memcache) {
            $this->connect();
        }
        $this->memcache->flush();
    }

    public function __destruct()
    {
        if (null !== $this->memcache) {
            $this->memcache->close();
        }
    }

}