mirror of
https://github.com/predis/predis.git
synced 2026-08-22 08:40:56 +00:00
Daniele Alessandri
6e2fd181f1
Actually this was already the case for certain commands, but some of them was left unguarded for such cases. This commit also fixes #109. The behaviour of silently skipping key prefixing when a command has no arguments may change in the future so we added explicit tests as guards for future changes. Predis\Command\Processor\KeyPrefixProcessor will continue to skip key prefixing on empty arguments, regardless.
94 lines
2.2 KiB
PHP
94 lines
2.2 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Predis package.
|
|
*
|
|
* (c) Daniele Alessandri <suppakilla@gmail.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Predis\Command;
|
|
|
|
/**
|
|
* @link http://redis.io/commands/zunionstore
|
|
* @author Daniele Alessandri <suppakilla@gmail.com>
|
|
*/
|
|
class ZSetUnionStore extends PrefixableCommand
|
|
{
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function getId()
|
|
{
|
|
return 'ZUNIONSTORE';
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function filterArguments(Array $arguments)
|
|
{
|
|
$options = array();
|
|
$argc = count($arguments);
|
|
|
|
if ($argc > 2 && is_array($arguments[$argc - 1])) {
|
|
$options = $this->prepareOptions(array_pop($arguments));
|
|
}
|
|
|
|
if (is_array($arguments[1])) {
|
|
$arguments = array_merge(
|
|
array($arguments[0], count($arguments[1])),
|
|
$arguments[1]
|
|
);
|
|
}
|
|
|
|
return array_merge($arguments, $options);
|
|
}
|
|
|
|
/**
|
|
* Returns a list of options and modifiers compatible with Redis.
|
|
*
|
|
* @param array $options List of options.
|
|
* @return array
|
|
*/
|
|
private function prepareOptions($options)
|
|
{
|
|
$opts = array_change_key_case($options, CASE_UPPER);
|
|
$finalizedOpts = array();
|
|
|
|
if (isset($opts['WEIGHTS']) && is_array($opts['WEIGHTS'])) {
|
|
$finalizedOpts[] = 'WEIGHTS';
|
|
|
|
foreach ($opts['WEIGHTS'] as $weight) {
|
|
$finalizedOpts[] = $weight;
|
|
}
|
|
}
|
|
|
|
if (isset($opts['AGGREGATE'])) {
|
|
$finalizedOpts[] = 'AGGREGATE';
|
|
$finalizedOpts[] = $opts['AGGREGATE'];
|
|
}
|
|
|
|
return $finalizedOpts;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function prefixKeys($prefix)
|
|
{
|
|
if ($arguments = $this->getArguments()) {
|
|
$arguments[0] = "$prefix{$arguments[0]}";
|
|
$length = ((int) $arguments[1]) + 2;
|
|
|
|
for ($i = 2; $i < $length; $i++) {
|
|
$arguments[$i] = "$prefix{$arguments[$i]}";
|
|
}
|
|
|
|
$this->setRawArguments($arguments);
|
|
}
|
|
}
|
|
}
|