predis/lib/Predis/Commands/Preprocessors/KeyPrefixPreprocessor.php

<?php

namespace Predis\Commands\Preprocessors;

use Predis\ClientException;
use Predis\Profiles\IServerProfile;

class KeyPrefixPreprocessor implements ICommandPreprocessor {
    private $_prefix;
    private $_strategies;

    public function __construct($prefix, IServerProfile $profile = null) {
        if (isset($profile)) {
            $this->checkProfile($profile);
        }
        $this->_prefix = $prefix;
        $this->_strategies = $this->getPrefixStrategies();
    }

    protected function checkProfile(IServerProfile $profile) {
        if (!in_array($profile, $this->getSupportedProfiles())) {
            throw new ClientException("Unsupported profile: $profile");
        }
    }

    protected function getSupportedProfiles() {
        return array('1.2', '2.0', '2.2');
    }

    protected function getPrefixStrategies() {
        $singleKey = function(&$arguments, $prefix) {
            $arguments[0] = "$prefix{$arguments[0]}";
        };

        $multiKeys = function(&$arguments, $prefix) {
            if (count($arguments) === 1 && is_array($arguments[0])) {
                $arguments = &$arguments[0];
            }
            foreach ($arguments as &$key) {
                $key = "$prefix$key";
            }
        };

        $skipLast = function(&$arguments, $prefix) {
            $length = count($arguments);
            for ($i = 0; $i < $length - 1; $i++) {
                $arguments[$i] = "$prefix{$arguments[$i]}";
            }
        };

        $interleavedKeys = function(&$arguments, $prefix) {
            $length = count($arguments);
            if ($length === 1 && is_array($arguments[0])) {
                $oldKvs = &$arguments[0];
                $newKvs = array();
                foreach ($oldKvs as $key => $value) {
                    $newKvs["$prefix$key"] = $value;
                    unset($oldKvs[$key]);
                }
                $arguments[0] = $newKvs;
            }
            else {
                for ($i = 0; $i < $length; $i += 2) {
                    $arguments[$i] = "$prefix{$arguments[$i]}";
                }
            }
        };

        $zunionstore = function(&$arguments, $prefix) {
            $arguments[0] = "$prefix{$arguments[0]}";
            if (is_array($arguments[1])) {
                foreach ($arguments[1] as &$destinationKey) {
                    $destinationKey = "$prefix$destinationKey";
                }
                $args = &$arguments[1];
                $length = count($args);
                for ($i = 0; $i < $length; $i++) {
                    $arguments[1][$i] = "$prefix{$args[$i]}";
                }
            }
            else {
                $length = (int)$arguments[1];
                for ($i = 2; $i < $length; $i++) {
                    $arguments[$i] = "$prefix{$arguments[$i]}";
                }
            }
        };

        $sort = function(&$arguments, $prefix) {
            $arguments[0] = "$prefix{$arguments[0]}";
            if (count($arguments) === 1) {
                return;
            }
            foreach ($arguments[1] as $modifier => &$value) {
                switch (strtoupper($modifier)) {
                    case 'BY':
                    case 'STORE':
                        $value = "$prefix$value";
                        break;
                    case 'GET':
                        if (is_array($value)) {
                            foreach ($value as &$getItem) {
                                $getItem = "$prefix$getItem";
                            }
                        }
                        else {
                            $value = "$prefix$value";
                        }
                        break;
                }
            }
        };

        $debug = function(&$arguments, $prefix) {
            if (count($arguments) === 3 && strtoupper($arguments[1]) == 'OBJECT') {
                $arguments[2] = "$prefix{$arguments[2]}";
            }
        };

        $cmdSingleKey = array(
            'type', 'exists', 'move', 'expire', 'persist', 'sort', 'expireat', 'ttl', 'append',
            'getrange', 'setnx', 'decr', 'getset', 'setrange', 'decrby', 'incr', 'set', 'strlen',
            'get', 'incrby', 'setbit', 'getbit', 'setex', 'hdel', 'hgetall', 'hlen', 'hset',
            'hexists', 'hincrby', 'hmget', 'hsetnx', 'hget', 'hkeys', 'hmset', 'hvals', 'lindex', 
            'linsert', 'llen', 'lpop', 'lpush', 'lpushx', 'rpushx', 'lrange', 'lrem', 'lset',
            'ltrim', 'rpop', 'rpush', 'rpushx', 'sadd', 'scard', 'sismember', 'smembers', 'spop', 
            'srandmember', 'srem', 'zadd', 'zcard', 'zcount', 'zincrby', 'zrange', 'zrangebyscore', 
            'zrank', 'zrem', 'zremrangebyrank', 'zremrangebyscore', 'zrevrange', 'zrevrangebyscore', 
            'zrevrank', 'zscore', 'publish', 'keys',
        );
        $cmdMultiKeys = array(
            'del', 'rename', 'renamenx', 'mget', 'rpoplpush', 'sdiff', 'sdiffstore', 'sinter', 
            'sinterstore', 'sunion', 'sunionstore', 'subscribe', 'punsubscribe', 'subscribe', 
            'unsubscribe', 'watch', 
        );

        return array_merge(
            array_fill_keys($cmdSingleKey, $singleKey),
            array_fill_keys($cmdMultiKeys, $multiKeys),
            array(
                'blpop' => $skipLast, 'blpop' => $skipLast, 'brpoplpush' => $skipLast, 'smove' => $skipLast,
                'mset' => $interleavedKeys, 'msetnx' => $interleavedKeys,
                'zinterstore' => $zunionstore, 'zunionstore' => $zunionstore,
                'sort' => $sort, 'debug' => $debug
            )
        );
    }

    public function setPrefixStrategy($command, $strategy) {
        if (!is_callable($callable)) {
            throw new \InvalidArgumentException(
                'The command preprocessor strategy must be a callable object'
            );
        }
        $this->_strategies[$command] = $strategy;
    }

    public function getPrefixStrategy($command) {
        if (isset($this->_strategies[$command])) {
            return $this->_strategies[$command];
        }
    }

    public function setPrefix($prefix) {
        $this->_prefix = $prefix;
    }

    public function getPrefix() {
        return $this->_prefix;
    }

    public function process($method, &$arguments) {
        if (isset($this->_strategies[$method])) {
            $this->_strategies[$method]($arguments, $this->_prefix);
        }
    }
}