redis/node-redis
1
0
Fork 0
mirror of https://github.com/redis/node-redis.git synced 2025-08-06 02:15:48 +03:00
Code Activity

feat: add EPSILON parameter support to VSIM command (#3041)

Browse Source
This commit is contained in:
Pavel Pashov
2025-07-31 15:11:51 +03:00
committed by GitHub
parent d8e14fa4fe
commit 5f09e4a8a5
2 changed files with 31 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
packages/client/lib/commands/VSIM.spec.ts
View File

@@ -31,14 +31,15 @@ describe('VSIM', () => {
FILTER: '.price > 20', FILTER: '.price > 20',
'FILTER-EF': 50, 'FILTER-EF': 50,
TRUTH: true, TRUTH: true,
NOTHREAD: true NOTHREAD: true,
EPSILON: 0.1
}); });
assert.deepEqual( assert.deepEqual(
parser.redisArgs, parser.redisArgs,
[ [
'VSIM', 'key', 'ELE', 'element', 'VSIM', 'key', 'ELE', 'element', 'COUNT', '5',
'COUNT', '5', 'EF', '100', 'FILTER', '.price > 20', 'EPSILON', '0.1', 'EF', '100', 'FILTER', '.price > 20',
'FILTER-EF', '50', 'TRUTH', 'NOTHREAD' 'FILTER-EF', '50', 'TRUTH', 'NOTHREAD',
] ]
); );
}); });
@@ -56,6 +57,27 @@ describe('VSIM', () => {
cluster: { ...GLOBAL.CLUSTERS.OPEN, minimumDockerVersion: [8, 0] } cluster: { ...GLOBAL.CLUSTERS.OPEN, minimumDockerVersion: [8, 0] }
}); });
testUtils.testAll('vSim with options', async client => {
await client.vAdd('key', [1.0, 2.0, 3.0], 'element1');
await client.vAdd('key', [1.1, 2.1, 3.1], 'element2');
const result = await client.vSim('key', 'element1', {
EPSILON: 0.1,
COUNT: 1,
EF: 100,
FILTER: '.year == 8',
'FILTER-EF': 50,
TRUTH: true,
NOTHREAD: true
});
assert.ok(Array.isArray(result));
}, {
client: { ...GLOBAL.SERVERS.OPEN, minimumDockerVersion: [8, 0] },
cluster: { ...GLOBAL.CLUSTERS.OPEN, minimumDockerVersion: [8, 0] }
});
testUtils.testWithClient('vSim with RESP3', async client => { testUtils.testWithClient('vSim with RESP3', async client => {
await client.vAdd('resp3-key', [1.0, 2.0, 3.0], 'element1'); await client.vAdd('resp3-key', [1.0, 2.0, 3.0], 'element1');
await client.vAdd('resp3-key', [1.1, 2.1, 3.1], 'element2'); await client.vAdd('resp3-key', [1.1, 2.1, 3.1], 'element2');

5
packages/client/lib/commands/VSIM.ts
View File

@@ -4,6 +4,7 @@ import { transformDoubleArgument } from './generic-transformers';
export interface VSimOptions { export interface VSimOptions {
COUNT?: number; COUNT?: number;
EPSILON?: number;
EF?: number; EF?: number;
FILTER?: string; FILTER?: string;
'FILTER-EF'?: number; 'FILTER-EF'?: number;
@@ -44,6 +45,10 @@ export default {
parser.push('COUNT', options.COUNT.toString()); parser.push('COUNT', options.COUNT.toString());
} }
if (options?.EPSILON !== undefined) {
parser.push('EPSILON', options.EPSILON.toString());
}
if (options?.EF !== undefined) { if (options?.EF !== undefined) {
parser.push('EF', options.EF.toString()); parser.push('EF', options.EF.toString());
} }