mirror of
https://github.com/predis/predis.git
synced 2026-08-19 01:41:46 +00:00
messikiller
7bf1f6eb29
Co-authored-by: heqiming <heqiming@kingsoft.com> Co-authored-by: Till Krüss <tillkruss@users.noreply.github.com> Co-authored-by: Vladyslav Vildanov <117659936+vladvildanov@users.noreply.github.com>
474 lines
12 KiB
PHP
474 lines
12 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Predis package.
|
|
*
|
|
* (c) 2009-2020 Daniele Alessandri
|
|
* (c) 2021-2025 Till Krüss
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Predis\Cluster;
|
|
|
|
use Predis\Cluster\SimpleSlotMap as SlotMap;
|
|
use PredisTestCase;
|
|
|
|
class SimpleSlotMapTest extends PredisTestCase
|
|
{
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsValidReturnsTrueOnValidSlot(): void
|
|
{
|
|
$this->assertTrue(SlotMap::isValid(0));
|
|
$this->assertTrue(SlotMap::isValid(16383));
|
|
|
|
$this->assertTrue(SlotMap::isValid(5000));
|
|
$this->assertTrue(SlotMap::isValid('5000'));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsValidReturnsFalseOnInvalidSlot(): void
|
|
{
|
|
$this->assertFalse(SlotMap::isValid(-1));
|
|
$this->assertFalse(SlotMap::isValid(16384));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsValidRangeReturnsTrueOnValidSlotRange(): void
|
|
{
|
|
$this->assertTrue(SlotMap::isValidRange(0, 16383));
|
|
$this->assertTrue(SlotMap::isValidRange(2000, 2999));
|
|
$this->assertTrue(SlotMap::isValidRange(3000, 3000));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsValidRangeReturnsFalseOnInvalidSlotRange(): void
|
|
{
|
|
$this->assertFalse(SlotMap::isValidRange(0, 16384));
|
|
$this->assertFalse(SlotMap::isValidRange(-1, 16383));
|
|
$this->assertFalse(SlotMap::isValidRange(-1, 16384));
|
|
$this->assertFalse(SlotMap::isValidRange(2999, 2000));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testToArrayReturnsEmptyArrayOnEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$this->assertEmpty($slotmap->toArray());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSetSlotsAssignsSpecifiedNodeToSlotRange(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
|
|
$expectedMap = array_merge(
|
|
array_fill(0, 5461, '127.0.0.1:6379'),
|
|
array_fill(5461, 5462, '127.0.0.1:6380'),
|
|
array_fill(10923, 5461, '127.0.0.1:6381')
|
|
);
|
|
|
|
$this->assertSame($expectedMap, $slotmap->toArray());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSetSlotsOverwritesSlotRange(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(1000, 2000, '127.0.0.1:6380');
|
|
|
|
$expectedMap =
|
|
array_fill(0, 5461, '127.0.0.1:6379') +
|
|
array_fill(1000, 2000, '127.0.0.1:6380');
|
|
|
|
$this->assertSame($expectedMap, $slotmap->toArray());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSetSlotsAssignsSingleSlotWhenFirstAndLastSlotMatch(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(10, 10, '127.0.0.1:6379');
|
|
|
|
$this->assertSame([10 => '127.0.0.1:6379'], $slotmap->toArray());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSetSlotsCastsValueToString(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$connection = $this->getMockConnection();
|
|
$connection
|
|
->expects($this->once())
|
|
->method('__toString')
|
|
->willReturn('127.0.0.1:6379');
|
|
|
|
$slotmap->setSlots(10, 10, $connection);
|
|
|
|
$this->assertSame([10 => '127.0.0.1:6379'], $slotmap->toArray());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testSetSlotsThrowsExceptionOnInvalidSlotRange(): void
|
|
{
|
|
$this->expectException('OutOfBoundsException');
|
|
$this->expectExceptionMessage('Invalid slot range 0-16384 for `127.0.0.1:6379`');
|
|
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 16384, '127.0.0.1:6379');
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetSlotsReturnsEmptyArrayOnEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$this->assertEmpty($slotmap->getSlots(3, 11));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetSlotsReturnsDictionaryOfSlotsWithAssignedNodes(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5, '127.0.0.1:6379');
|
|
$slotmap->setSlots(10, 13, '127.0.0.1:6380');
|
|
|
|
$expectedMap = [
|
|
3 => '127.0.0.1:6379',
|
|
4 => '127.0.0.1:6379',
|
|
5 => '127.0.0.1:6379',
|
|
10 => '127.0.0.1:6380',
|
|
11 => '127.0.0.1:6380',
|
|
];
|
|
|
|
$this->assertSame($expectedMap, $slotmap->getSlots(3, 11));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetSlotsReturnsEmptyArrayOnEmptySlotRange(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5, '127.0.0.1:6379');
|
|
$slotmap->setSlots(10, 13, '127.0.0.1:6380');
|
|
|
|
$this->assertEmpty($slotmap->getSlots(100, 200));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetSlotsThrowsExceptionOnInvalidSlotRange(): void
|
|
{
|
|
$this->expectException('OutOfBoundsException');
|
|
$this->expectExceptionMessage('Invalid slot range 0-16384');
|
|
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->getSlots(0, 16384);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsEmptyReturnsTrueOnEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$this->assertTrue($slotmap->isEmpty());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testIsEmptyReturnsFalseOnNonEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertFalse($slotmap->isEmpty());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testCountReturnsZeroOnEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$this->assertCount(0, $slotmap);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testCountReturnsAssignedSlotsInSlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$this->assertCount(5461, $slotmap);
|
|
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$this->assertCount(10923, $slotmap);
|
|
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
$this->assertCount(16384, $slotmap);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testResetEmptiesSlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
|
|
$this->assertFalse($slotmap->isEmpty());
|
|
|
|
$slotmap->reset();
|
|
|
|
$this->assertTrue($slotmap->isEmpty());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetNodesReturnsEmptyArrayOnEmptySlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$this->assertEmpty($slotmap->getNodes());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetNodesReturnsArrayOfNodesInSlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
|
|
$this->assertSame(['127.0.0.1:6379', '127.0.0.1:6380', '127.0.0.1:6381'], $slotmap->getNodes());
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetExistsReturnsTrueOnAssignedSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertTrue(isset($slotmap[0]));
|
|
$this->assertTrue(isset($slotmap[2000]));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetExistsReturnsFalseOnAssignedSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertFalse(isset($slotmap[6000]));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetExistsReturnsFalseOnInvalidSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertFalse(isset($slotmap[-100]));
|
|
$this->assertFalse(isset($slotmap[16384]));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetGetReturnsNodeOfAssignedSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
|
|
$this->assertSame('127.0.0.1:6379', $slotmap[0]);
|
|
$this->assertSame('127.0.0.1:6380', $slotmap[5461]);
|
|
$this->assertSame('127.0.0.1:6381', $slotmap[10923]);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetGetReturnsNullOnUnassignedSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertNull($slotmap[5461]);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetGetReturnsNullOnInvalidSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertNull($slotmap[-100]);
|
|
$this->assertNull($slotmap[16384]);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetUnsetRemovesSlotAssignment(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertTrue(isset($slotmap[100]));
|
|
unset($slotmap[100]);
|
|
$this->assertFalse(isset($slotmap[100]));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetUnsetDoesNotDoAnythingOnUnassignedSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertFalse(isset($slotmap[5461]));
|
|
unset($slotmap[5461]);
|
|
$this->assertFalse(isset($slotmap[5461]));
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetSetAssignsNodeToSlot(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
|
|
$this->assertSame('127.0.0.1:6380', $slotmap[100] = '127.0.0.1:6380');
|
|
$this->assertSame('127.0.0.1:6380', $slotmap[100]);
|
|
|
|
$this->assertNull($slotmap[5461]);
|
|
$this->assertSame('127.0.0.1:6380', $slotmap[5461] = '127.0.0.1:6380');
|
|
$this->assertSame('127.0.0.1:6380', $slotmap[5461]);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetSetCastsValueToString(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$connection = $this->getMockConnection();
|
|
$connection
|
|
->expects($this->once())
|
|
->method('__toString')
|
|
->willReturn('127.0.0.1:6379');
|
|
|
|
$this->assertSame($connection, $slotmap[0] = $connection);
|
|
$this->assertSame('127.0.0.1:6379', $slotmap[0]);
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testOffsetSetThrowsExceptionOnInvalidSlot(): void
|
|
{
|
|
$this->expectException('OutOfBoundsException');
|
|
$this->expectExceptionMessage('Invalid slot 16384 for `127.0.0.1:6379`');
|
|
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap[16384] = '127.0.0.1:6379';
|
|
}
|
|
|
|
/**
|
|
* @group disconnected
|
|
*/
|
|
public function testGetIteratorReturnsIteratorOverSlotMap(): void
|
|
{
|
|
$slotmap = new SlotMap();
|
|
|
|
$slotmap->setSlots(0, 5460, '127.0.0.1:6379');
|
|
$slotmap->setSlots(5461, 10922, '127.0.0.1:6380');
|
|
$slotmap->setSlots(10923, 16383, '127.0.0.1:6381');
|
|
|
|
$expectedMap = array_merge(
|
|
array_fill(0, 5461, '127.0.0.1:6379'),
|
|
array_fill(5461, 5462, '127.0.0.1:6380'),
|
|
array_fill(10923, 5461, '127.0.0.1:6381')
|
|
);
|
|
|
|
$this->assertSame($expectedMap, iterator_to_array($slotmap));
|
|
}
|
|
}
|