Compare commits

...

1 Commits

Author SHA1 Message Date
vladvildanov 59b60a403e Changed default dialect to 2 2025-02-24 17:13:57 +02:00
8 changed files with 152 additions and 48 deletions
+7 -1
View File
@@ -30,7 +30,13 @@ class FTAGGREGATE extends RedisCommand
public function setArguments(array $arguments)
{
[$index, $query] = $arguments;
$commandArguments = (!empty($arguments[2])) ? $arguments[2]->toArray() : [];
if (!empty($arguments[2]) && !in_array("DIALECT", $arguments[2]->toArray())) {
// Default dialect is 2
$arguments[2]->dialect(2);
}
$commandArguments = (!empty($arguments[2])) ? $arguments[2]->toArray() : ['DIALECT', 2];
parent::setArguments(array_merge(
[$index, $query],
+7 -1
View File
@@ -29,7 +29,13 @@ class FTEXPLAIN extends RedisCommand
public function setArguments(array $arguments)
{
[$index, $query] = $arguments;
$commandArguments = [];
if (!empty($arguments[2]) && !in_array("DIALECT", $arguments[2]->toArray())) {
// Default dialect is 2
$arguments[2]->dialect(2);
}
$commandArguments = ['DIALECT', 2];
if (!empty($arguments[2])) {
$commandArguments = $arguments[2]->toArray();
+7 -1
View File
@@ -29,7 +29,13 @@ class FTSEARCH extends RedisCommand
public function setArguments(array $arguments)
{
[$index, $query] = $arguments;
$commandArguments = (!empty($arguments[2])) ? $arguments[2]->toArray() : [];
if (!empty($arguments[2]) && !in_array("DIALECT", $arguments[2]->toArray())) {
// Default dialect is 2
$arguments[2]->dialect(2);
}
$commandArguments = (!empty($arguments[2])) ? $arguments[2]->toArray() : ['DIALECT', 2];
parent::setArguments(array_merge(
[$index, $query],
+7 -1
View File
@@ -24,7 +24,13 @@ class FTSPELLCHECK extends RedisCommand
public function setArguments(array $arguments)
{
[$index, $query] = $arguments;
$commandArguments = [];
if (!empty($arguments[2]) && !in_array("DIALECT", $arguments[2]->toArray())) {
// Default dialect is 2
$arguments[2]->dialect(2);
}
$commandArguments = ['DIALECT', 2];
if (!empty($arguments[2])) {
$commandArguments = $arguments[2]->toArray();
@@ -134,55 +134,55 @@ class FTAGGREGATE_Test extends PredisCommandTestCase
return [
'with default arguments' => [
['index', 'query'],
['index', 'query'],
['index', 'query', 'DIALECT', 2],
],
'with VERBATIM modifier' => [
['index', 'query', (new AggregateArguments())->verbatim()],
['index', 'query', 'VERBATIM'],
['index', 'query', 'VERBATIM', 'DIALECT', 2],
],
'with LOAD modifier - specified fields' => [
['index', 'query', (new AggregateArguments())->load('field1', 'field2')],
['index', 'query', 'LOAD', 2, 'field1', 'field2'],
['index', 'query', 'LOAD', 2, 'field1', 'field2', 'DIALECT', 2],
],
'with LOAD modifier - all fields' => [
['index', 'query', (new AggregateArguments())->load('*')],
['index', 'query', 'LOAD', '*'],
['index', 'query', 'LOAD', '*', 'DIALECT', 2],
],
'with TIMEOUT modifier' => [
['index', 'query', (new AggregateArguments())->timeout(2)],
['index', 'query', 'TIMEOUT', 2],
['index', 'query', 'TIMEOUT', 2, 'DIALECT', 2],
],
'with GROUPBY modifier' => [
['index', 'query', (new AggregateArguments())->groupBy('property1', 'property2')],
['index', 'query', 'GROUPBY', 2, 'property1', 'property2'],
['index', 'query', 'GROUPBY', 2, 'property1', 'property2', 'DIALECT', 2],
],
'with REDUCE modifier' => [
['index', 'query', (new AggregateArguments())->reduce('function', 'arg1', true, 'alias1', 'arg2')],
['index', 'query', 'REDUCE', 'function', 2, 'arg1', 'AS', 'alias1', 'arg2'],
['index', 'query', 'REDUCE', 'function', 2, 'arg1', 'AS', 'alias1', 'arg2', 'DIALECT', 2],
],
'with SORTBY modifier' => [
['index', 'query', (new AggregateArguments())->sortBy(2, 'property1', 'ASC', 'property2', 'DESC')],
['index', 'query', 'SORTBY', 2, 'property1', 'ASC', 'property2', 'DESC', 'MAX', 2],
['index', 'query', 'SORTBY', 2, 'property1', 'ASC', 'property2', 'DESC', 'MAX', 2, 'DIALECT', 2],
],
'with APPLY modifier' => [
['index', 'query', (new AggregateArguments())->apply('expression', 'name')],
['index', 'query', 'APPLY', 'expression', 'AS', 'name'],
['index', 'query', 'APPLY', 'expression', 'AS', 'name', 'DIALECT', 2],
],
'with LIMIT modifier' => [
['index', 'query', (new AggregateArguments())->limit(2, 3)],
['index', 'query', 'LIMIT', 2, 3],
['index', 'query', 'LIMIT', 2, 3, 'DIALECT', 2],
],
'with FILTER modifier' => [
['index', 'query', (new AggregateArguments())->filter('filter')],
['index', 'query', 'FILTER', 'filter'],
['index', 'query', 'FILTER', 'filter', 'DIALECT', 2],
],
'with WITHCURSOR modifier' => [
['index', 'query', (new AggregateArguments())->withCursor(10, 20)],
['index', 'query', 'WITHCURSOR', 'COUNT', 10, 'MAXIDLE', 20],
['index', 'query', 'WITHCURSOR', 'COUNT', 10, 'MAXIDLE', 20, 'DIALECT', 2],
],
'with PARAMS modifier' => [
['index', 'query', (new AggregateArguments())->params(['name1', 'value1', 'name2', 'value2'])],
['index', 'query', 'PARAMS', 4, 'name1', 'value1', 'name2', 'value2'],
['index', 'query', 'PARAMS', 4, 'name1', 'value1', 'name2', 'value2', 'DIALECT', 2],
],
'with DIALECT modifier' => [
['index', 'query', (new AggregateArguments())->dialect('dialect')],
@@ -200,7 +200,7 @@ class FTAGGREGATE_Test extends PredisCommandTestCase
],
[
'index', '@name: "test"', 'APPLY', 'year(@dob)', 'AS', 'birth', 'GROUPBY', 2, '@birth', '@country',
'REDUCE', 'COUNT', 0, 'AS', 'num_visits', 'SORTBY', 1, '@day',
'REDUCE', 'COUNT', 0, 'AS', 'num_visits', 'SORTBY', 1, '@day', 'DIALECT', 2,
],
],
];
@@ -105,7 +105,10 @@ EOT;
$this->assertEquals('OK', $redis->ftcreate('index', $schema));
$this->assertEquals(
$expectedResponse,
$redis->ftexplain('index', '(foo bar)|(hello world) @date:[100 200]|@date:[500 +inf]')
$redis->ftexplain(
'index', '(foo bar)|(hello world) @date:[100 200]|@date:[500 +inf]',
(new ExplainArguments())->dialect(1)
)
);
}
@@ -129,7 +132,7 @@ EOT;
return [
'with default arguments' => [
['index', 'query', null],
['index', 'query'],
['index', 'query', 'DIALECT', 2],
],
'with DIALECT' => [
['index', 'query', (new ExplainArguments())->dialect('dialect')],
@@ -18,6 +18,7 @@ use Predis\Command\Argument\Search\SchemaFields\GeoShapeField;
use Predis\Command\Argument\Search\SchemaFields\NumericField;
use Predis\Command\Argument\Search\SchemaFields\TagField;
use Predis\Command\Argument\Search\SchemaFields\TextField;
use Predis\Command\Argument\Search\SchemaFields\VectorField;
use Predis\Command\Argument\Search\SearchArguments;
use Predis\Command\Redis\PredisCommandTestCase;
use Predis\Response\ServerException;
@@ -409,96 +410,173 @@ class FTSEARCH_Test extends PredisCommandTestCase
);
}
/**
* @group connected
* @group relay-resp3
* @return void
* @requiresRedisVersion >= 7.9.0
*/
public function testVectorSearchWithDefaultDialect(): void
{
$redis = $this->getClient();
$this->assertEquals('OK', $redis->ftcreate('test', [
new VectorField(
'v', 'HNSW',
["TYPE", "FLOAT32", "DIM", 2, "DISTANCE_METRIC", "L2"]
)
]));
$this->sleep(0.1);
$redis->hset("a", "v", "aaaaaaaa");
$redis->hset("b", "v", "aaaabaaa");
$redis->hset("c", "v", "aaaaabaa");
$searchArguments = new SearchArguments();
$searchArguments->params(["vec", "aaaaaaaa"]);
$ftSearch = new FTSEARCH();
$ftSearch->setArguments(['test', '*=>[KNN 2 @v $vec]', $searchArguments]);
$this->assertContains('DIALECT', $ftSearch->getArguments());
$this->assertContains(2, $ftSearch->getArguments());
// Check if dialect applied via constructing custom command object.
$this->assertEquals(2, $redis->executeCommand($ftSearch)[0]);
// Check if it works via usual client interface.
$this->assertEquals(
2,
$redis->ftsearch('test', '*=>[KNN 2 @v $vec]', $searchArguments)[0]
);
}
/**
* @group connected
* @group relay-resp3
* @return void
* @requiresRedisVersion >= 7.9.0
*/
public function testSearchQueryWithDifferentDialects(): void
{
$redis = $this->getClient();
$createArguments = new CreateArguments();
$createArguments->prefix(['test:']);
$this->assertEquals('OK', $redis->ftcreate('test', [
new TextField("name"),
new TextField("lastname"),
], $createArguments));
$this->sleep(0.1);
$redis->hset("test:1", "name", "James");
$redis->hset("test:1", "lastname", "Brown");
# Query with default DIALECT 2
$this->assertEquals(1, $redis->ftsearch('test', '@name: James Brown')[0]);
$searchArguments = new SearchArguments();
$searchArguments->dialect(1);
# Query with explicit DIALECT 1
$this->assertEquals(
0,
$redis->ftsearch('test', '@name: James Brown', $searchArguments)[0]
);
}
public function argumentsProvider(): array
{
return [
'with NOCONTENT modifier' => [
['index', '*', (new SearchArguments())->noContent()],
['index', '*', 'NOCONTENT'],
['index', '*', 'NOCONTENT', 'DIALECT', 2],
],
'with VERBATIM modifier' => [
['index', '*', (new SearchArguments())->verbatim()],
['index', '*', 'VERBATIM'],
['index', '*', 'VERBATIM', 'DIALECT', 2],
],
'with WITHSCORES modifier' => [
['index', '*', (new SearchArguments())->withScores()],
['index', '*', 'WITHSCORES'],
['index', '*', 'WITHSCORES', 'DIALECT', 2],
],
'with WITHPAYLOADS modifier' => [
['index', '*', (new SearchArguments())->withPayloads()],
['index', '*', 'WITHPAYLOADS'],
['index', '*', 'WITHPAYLOADS', 'DIALECT', 2],
],
'with WITHSORTKEYS modifier' => [
['index', '*', (new SearchArguments())->withSortKeys()],
['index', '*', 'WITHSORTKEYS'],
['index', '*', 'WITHSORTKEYS', 'DIALECT', 2],
],
'with FILTER modifier' => [
['index', '*', (new SearchArguments())->searchFilter(['numeric_field', 1, 10])],
['index', '*', 'FILTER', 'numeric_field', 1, 10],
['index', '*', 'FILTER', 'numeric_field', 1, 10, 'DIALECT', 2],
],
'with GEOFILTER modifier' => [
['index', '*', (new SearchArguments())->geoFilter(['geo_field', 12.213, 14.212, 300, 'km'])],
['index', '*', 'GEOFILTER', 'geo_field', 12.213, 14.212, 300, 'km'],
['index', '*', 'GEOFILTER', 'geo_field', 12.213, 14.212, 300, 'km', 'DIALECT', 2],
],
'with INKEYS modifier' => [
['index', '*', (new SearchArguments())->inKeys(['key1', 'key2'])],
['index', '*', 'INKEYS', 2, 'key1', 'key2'],
['index', '*', 'INKEYS', 2, 'key1', 'key2', 'DIALECT', 2],
],
'with INFIELDS modifier' => [
['index', '*', (new SearchArguments())->inFields(['field1', 'field2'])],
['index', '*', 'INFIELDS', 2, 'field1', 'field2'],
['index', '*', 'INFIELDS', 2, 'field1', 'field2', 'DIALECT', 2],
],
'with RETURN modifier' => [
['index', '*', (new SearchArguments())->addReturn(2, 'identifier', true, 'property')],
['index', '*', 'RETURN', 2, 'identifier', 'AS', 'property'],
['index', '*', 'RETURN', 2, 'identifier', 'AS', 'property', 'DIALECT', 2],
],
'with SUMMARIZE modifier' => [
['index', '*', (new SearchArguments())->summarize(['field1', 'field2'], 2, 2, ',')],
['index', '*', 'SUMMARIZE', 'FIELDS', 2, 'field1', 'field2', 'FRAGS', 2, 'LEN', 2, 'SEPARATOR', ','],
['index', '*', 'SUMMARIZE', 'FIELDS', 2, 'field1', 'field2', 'FRAGS', 2, 'LEN', 2, 'SEPARATOR', ',', 'DIALECT', 2],
],
'with HIGHLIGHT modifier' => [
['index', '*', (new SearchArguments())->highlight(['field1', 'field2'], 'openTag', 'closeTag')],
['index', '*', 'HIGHLIGHT', 'FIELDS', 2, 'field1', 'field2', 'TAGS', 'openTag', 'closeTag'],
['index', '*', 'HIGHLIGHT', 'FIELDS', 2, 'field1', 'field2', 'TAGS', 'openTag', 'closeTag', 'DIALECT', 2],
],
'with SLOP modifier' => [
['index', '*', (new SearchArguments())->slop(2)],
['index', '*', 'SLOP', 2],
['index', '*', 'SLOP', 2, 'DIALECT', 2],
],
'with TIMEOUT modifier' => [
['index', '*', (new SearchArguments())->timeout(2)],
['index', '*', 'TIMEOUT', 2],
['index', '*', 'TIMEOUT', 2, 'DIALECT', 2],
],
'with INORDER modifier' => [
['index', '*', (new SearchArguments())->inOrder()],
['index', '*', 'INORDER'],
['index', '*', 'INORDER', 'DIALECT', 2],
],
'with EXPANDER modifier' => [
['index', '*', (new SearchArguments())->expander('expander')],
['index', '*', 'EXPANDER', 'expander'],
['index', '*', 'EXPANDER', 'expander', 'DIALECT', 2],
],
'with SCORER modifier' => [
['index', '*', (new SearchArguments())->scorer('scorer')],
['index', '*', 'SCORER', 'scorer'],
['index', '*', 'SCORER', 'scorer', 'DIALECT', 2],
],
'with EXPLAINSCORE modifier' => [
['index', '*', (new SearchArguments())->explainScore()],
['index', '*', 'EXPLAINSCORE'],
['index', '*', 'EXPLAINSCORE', 'DIALECT', 2],
],
'with PAYLOAD modifier' => [
['index', '*', (new SearchArguments())->payload('payload')],
['index', '*', 'PAYLOAD', 'payload'],
['index', '*', 'PAYLOAD', 'payload', 'DIALECT', 2],
],
'with SORTBY modifier' => [
['index', '*', (new SearchArguments())->sortBy('sort_attribute', 'desc')],
['index', '*', 'SORTBY', 'sort_attribute', 'DESC'],
['index', '*', 'SORTBY', 'sort_attribute', 'DESC', 'DIALECT', 2],
],
'with LIMIT modifier' => [
['index', '*', (new SearchArguments())->limit(2, 2)],
['index', '*', 'LIMIT', 2, 2],
['index', '*', 'LIMIT', 2, 2, 'DIALECT', 2],
],
'with PARAMS modifier' => [
['index', '*', (new SearchArguments())->params(['name1', 'value2', 'name2', 'value2'])],
['index', '*', 'PARAMS', 4, 'name1', 'value2', 'name2', 'value2'],
['index', '*', 'PARAMS', 4, 'name1', 'value2', 'name2', 'value2', 'DIALECT', 2],
],
'with DIALECT modifier' => [
['index', '*', (new SearchArguments())->dialect('dialect')],
@@ -506,7 +584,7 @@ class FTSEARCH_Test extends PredisCommandTestCase
],
'with chain of arguments' => [
['index', '*', (new SearchArguments())->withScores()->withPayloads()->searchFilter(['numeric_field', 1, 10])->addReturn(2, 'identifier', true, 'property')],
['index', '*', 'WITHSCORES', 'WITHPAYLOADS', 'FILTER', 'numeric_field', 1, 10, 'RETURN', 2, 'identifier', 'AS', 'property'],
['index', '*', 'WITHSCORES', 'WITHPAYLOADS', 'FILTER', 'numeric_field', 1, 10, 'RETURN', 2, 'identifier', 'AS', 'property', 'DIALECT', 2],
],
];
}
@@ -116,7 +116,6 @@ class FTSPELLCHECK_Test extends PredisCommandTestCase
$redis = $this->getClient();
$this->expectException(ServerException::class);
$this->expectExceptionMessage('Unknown Index name');
$redis->ftspellcheck(
'index',
@@ -130,19 +129,19 @@ class FTSPELLCHECK_Test extends PredisCommandTestCase
return [
'with default arguments' => [
['index', 'query'],
['index', 'query'],
['index', 'query', 'DIALECT', 2],
],
'with DISTANCE modifier' => [
['index', 'query', (new SpellcheckArguments())->distance(2)],
['index', 'query', 'DISTANCE', 2],
['index', 'query', 'DISTANCE', 2, 'DIALECT', 2],
],
'with TERMS modifier - INCLUDE' => [
['index', 'query', (new SpellcheckArguments())->terms('dict', 'INCLUDE', 'term')],
['index', 'query', 'TERMS', 'INCLUDE', 'dict', 'term'],
['index', 'query', 'TERMS', 'INCLUDE', 'dict', 'term', 'DIALECT', 2],
],
'with TERMS modifier - EXCLUDE' => [
['index', 'query', (new SpellcheckArguments())->terms('dict', 'EXCLUDE', 'term')],
['index', 'query', 'TERMS', 'EXCLUDE', 'dict', 'term'],
['index', 'query', 'TERMS', 'EXCLUDE', 'dict', 'term', 'DIALECT', 2],
],
'with DIALECT modifier' => [
['index', 'query', (new SpellcheckArguments())->dialect('dialect')],