1
0
mirror of https://github.com/redis/node-redis.git synced 2025-08-06 02:15:48 +03:00
This commit is contained in:
Leibale
2023-07-10 12:20:40 -04:00
parent a4aa903987
commit 3fa7717a7d
20 changed files with 648 additions and 711 deletions

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import CONFIG_GET from './CONFIG_GET';
describe('CONFIG GET', () => {
describe('GRAPH.CONFIG GET', () => {
it('transformArguments', () => {
assert.deepEqual(
CONFIG_GET.transformArguments('TIMEOUT'),

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import CONFIG_SET from './CONFIG_SET';
describe('CONFIG SET', () => {
describe('GRAPH.CONFIG SET', () => {
it('transformArguments', () => {
assert.deepEqual(
CONFIG_SET.transformArguments('TIMEOUT', 0),

View File

@@ -11,5 +11,5 @@ export default {
value.toString()
];
},
transformReply: undefined as unknown as () => SimpleStringReply
transformReply: undefined as unknown as () => SimpleStringReply<'OK'>
} as const satisfies Command;

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import DELETE from './DELETE';
describe('', () => {
describe('GRAPH.DELETE', () => {
it('transformArguments', () => {
assert.deepEqual(
DELETE.transformArguments('key'),
@@ -11,11 +11,11 @@ describe('', () => {
});
testUtils.testWithClient('client.graph.delete', async client => {
await client.graph.query('key', 'RETURN 1');
const [, reply] = await Promise.all([
client.graph.query('key', 'RETURN 1'),
client.graph.delete('key')
]);
assert.equal(
typeof await client.graph.delete('key'),
'string'
);
assert.equal(reply, 'OK');
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -6,5 +6,5 @@ export default {
transformArguments(key: RedisArgument) {
return ['GRAPH.DELETE', key];
},
transformReply: undefined as unknown as () => SimpleStringReply
transformReply: undefined as unknown as () => SimpleStringReply<'OK'>
} as const satisfies Command;

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import EXPLAIN from './EXPLAIN';
describe('EXPLAIN', () => {
describe('GRAPH.EXPLAIN', () => {
it('transformArguments', () => {
assert.deepEqual(
EXPLAIN.transformArguments('key', 'RETURN 0'),
@@ -16,6 +16,8 @@ describe('EXPLAIN', () => {
client.graph.explain('key', 'RETURN 0')
]);
assert.ok(Array.isArray(reply));
assert.ok(!reply.find(x => typeof x !== 'string'));
for (const item of reply) {
assert.equal(typeof item, 'string');
}
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import LIST from './LIST';
describe('LIST', () => {
describe('GRAPH.LIST', () => {
it('transformArguments', () => {
assert.deepEqual(
LIST.transformArguments(),

View File

@@ -2,7 +2,7 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import PROFILE from './PROFILE';
describe('PROFILE', () => {
describe('GRAPH.PROFILE', () => {
it('transformArguments', () => {
assert.deepEqual(
PROFILE.transformArguments('key', 'RETURN 0'),
@@ -13,6 +13,8 @@ describe('PROFILE', () => {
testUtils.testWithClient('client.graph.profile', async client => {
const reply = await client.graph.profile('key', 'RETURN 0');
assert.ok(Array.isArray(reply));
assert.ok(!reply.find(x => typeof x !== 'string'));
for (const item of reply) {
assert.equal(typeof item, 'string');
}
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -2,14 +2,60 @@ import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import QUERY from './QUERY';
describe('QUERY', () => {
it('transformArguments', () => {
assert.deepEqual(
QUERY.transformArguments('key', 'query'),
['GRAPH.QUERY', 'key', 'query']
);
describe('GRAPH.QUERY', () => {
describe('transformArguments', () => {
it('simple', () => {
assert.deepEqual(
QUERY.transformArguments('key', 'query'),
['GRAPH.QUERY', 'key', 'query']
);
});
describe('params', () => {
it('all types', () => {
assert.deepEqual(
QUERY.transformArguments('key', 'query', {
params: {
null: null,
string: '"\\',
number: 0,
boolean: false,
array: [0],
object: {a: 0}
}
}),
['GRAPH.QUERY', 'key', 'CYPHER null=null string="\\"\\\\" number=0 boolean=false array=[0] object={a:0} query']
);
});
it('TypeError', () => {
assert.throws(() => {
QUERY.transformArguments('key', 'query', {
params: {
a: Buffer.from('a')
}
})
}, TypeError);
});
});
it('TIMEOUT', () => {
assert.deepEqual(
QUERY.transformArguments('key', 'query', {
TIMEOUT: 1
}),
['GRAPH.QUERY', 'key', 'query', 'TIMEOUT', '1']
);
});
it('compact', () => {
assert.deepEqual(
QUERY.transformArguments('key', 'query', undefined, true),
['GRAPH.QUERY', 'key', 'query', '--compact']
);
});
});
testUtils.testWithClient('client.graph.query', async client => {
const { data } = await client.graph.query('key', 'RETURN 0');
assert.deepEqual(data, [[0]]);

View File

@@ -1,55 +1,103 @@
import { RedisCommandArgument, RedisCommandArguments } from '@redis/client/dist/lib/commands/index';
import { pushQueryArguments, QueryOptionsBackwardCompatible } from '.';
import { RedisArgument, Command, ArrayReply, BlobStringReply, NumberReply, NullReply, TuplesReply } from '@redis/client/dist/lib/RESP/types';
export const FIRST_KEY_INDEX = 1;
type Headers = ArrayReply<BlobStringReply>;
export function transformArguments(
graph: RedisCommandArgument,
query: RedisCommandArgument,
options?: QueryOptionsBackwardCompatible,
compact?: boolean
): RedisCommandArguments {
return pushQueryArguments(
['GRAPH.QUERY'],
graph,
query,
options,
compact
);
}
// TODO: cannot use `ArrayReply` due to circular reference
type Data = Array<BlobStringReply | NumberReply | NullReply | Data>;
type Headers = Array<string>;
type Metadata = ArrayReply<BlobStringReply>;
type Data = Array<string | number | null | Data>;
type Metadata = Array<string>;
type QueryRawReply = [
type QueryRawReply = TuplesReply<[
headers: Headers,
data: Data,
metadata: Metadata
] | [
metadata: Metadata
];
]>;
export type QueryReply = {
headers: undefined;
data: undefined;
metadata: Metadata;
} | {
headers: Headers;
data: Data;
metadata: Metadata;
type QueryParam = null | string | number | boolean | QueryParams | Array<QueryParam>;
type QueryParams = {
[key: string]: QueryParam;
};
export function transformReply(reply: QueryRawReply): QueryReply {
return reply.length === 1 ? {
headers: undefined,
data: undefined,
metadata: reply[0]
} : {
headers: reply[0],
data: reply[1],
metadata: reply[2]
};
export interface QueryOptions {
params?: QueryParams;
TIMEOUT?: number;
}
export function transformQueryArguments(
command: RedisArgument,
graph: RedisArgument,
query: RedisArgument,
options?: QueryOptions,
compact?: boolean
) {
const args = [
command,
graph,
options?.params ?
`CYPHER ${queryParamsToString(options.params)} ${query}` :
query
];
if (options?.TIMEOUT !== undefined) {
args.push('TIMEOUT', options.TIMEOUT.toString());
}
if (compact) {
args.push('--compact');
}
return args;
}
function queryParamsToString(params: QueryParams) {
return Object.entries(params)
.map(([key, value]) => `${key}=${queryParamToString(value)}`)
.join(' ');
}
function queryParamToString(param: QueryParam): string {
if (param === null) {
return 'null';
}
switch (typeof param) {
case 'string':
return `"${param.replace(/["\\]/g, '\\$&')}"`;
case 'number':
case 'boolean':
return param.toString();
}
if (Array.isArray(param)) {
return `[${param.map(queryParamToString).join(',')}]`;
} else if (typeof param === 'object') {
const body = [];
for (const [key, value] of Object.entries(param)) {
body.push(`${key}:${queryParamToString(value)}`);
}
return `{${body.join(',')}}`;
} else {
throw new TypeError(`Unexpected param type ${typeof param} ${param}`)
}
}
export default {
FIRST_KEY_INDEX: 1,
IS_READ_ONLY: false,
transformArguments: transformQueryArguments.bind(undefined, 'GRAPH.QUERY'),
transformReply(reply: QueryRawReply) {
return reply.length === 1 ? {
headers: undefined,
data: undefined,
metadata: reply[0]
} : {
headers: reply[0],
data: reply[1],
metadata: reply[2]
};
}
} as const satisfies Command;

View File

@@ -1,20 +1,20 @@
import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import { transformArguments } from './RO_QUERY';
import RO_QUERY from './RO_QUERY';
describe('RO_QUERY', () => {
it('transformArguments', () => {
assert.deepEqual(
transformArguments('key', 'query'),
['GRAPH.RO_QUERY', 'key', 'query']
);
});
describe('GRAPH.RO_QUERY', () => {
it('transformArguments', () => {
assert.deepEqual(
RO_QUERY.transformArguments('key', 'query'),
['GRAPH.RO_QUERY', 'key', 'query']
);
});
testUtils.testWithClient('client.graph.roQuery', async client => {
const [, { data }] = await Promise.all([
client.graph.query('key', 'RETURN 0'), // make sure to create a graph first
client.graph.roQuery('key', 'RETURN 0')
]);
assert.deepEqual(data, [[0]]);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('client.graph.roQuery', async client => {
const [, { data }] = await Promise.all([
client.graph.query('key', 'RETURN 0'), // make sure to create a graph first
client.graph.roQuery('key', 'RETURN 0')
]);
assert.deepEqual(data, [[0]]);
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -1,23 +1,9 @@
import { RedisCommandArgument, RedisCommandArguments } from '@redis/client/dist/lib/commands';
import { pushQueryArguments, QueryOptionsBackwardCompatible } from '.';
import { Command } from '@redis/client/dist/lib/RESP/types';
import QUERY, { transformQueryArguments } from './QUERY';
export { FIRST_KEY_INDEX } from './QUERY';
export const IS_READ_ONLY = true;
export function transformArguments(
graph: RedisCommandArgument,
query: RedisCommandArgument,
options?: QueryOptionsBackwardCompatible,
compact?: boolean
): RedisCommandArguments {
return pushQueryArguments(
['GRAPH.RO_QUERY'],
graph,
query,
options,
compact
);
}
export { transformReply } from './QUERY';
export default {
FIRST_KEY_INDEX: QUERY.FIRST_KEY_INDEX,
IS_READ_ONLY: true,
transformArguments: transformQueryArguments.bind(undefined, 'GRAPH.RO_QUERY'),
transformReply: QUERY.transformReply
} as const satisfies Command;

View File

@@ -1,8 +1,8 @@
import { strict as assert } from 'assert';
import testUtils, { GLOBAL } from '../test-utils';
import { transformArguments } from './SLOWLOG';
import SLOWLOG from './SLOWLOG';
describe('SLOWLOG', () => {
describe('GRAPH.SLOWLOG', () => {
it('transformArguments', () => {
assert.deepEqual(
transformArguments('key'),
@@ -11,8 +11,10 @@ describe('SLOWLOG', () => {
});
testUtils.testWithClient('client.graph.slowLog', async client => {
await client.graph.query('key', 'RETURN 1');
const reply = await client.graph.slowLog('key');
const [, reply] = await Promise.all([
client.graph.query('key', 'RETURN 1'),
client.graph.slowLog('key')
]);
assert.equal(reply.length, 1);
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -1,30 +1,24 @@
export const IS_READ_ONLY = true;
import { RedisArgument, ArrayReply, TuplesReply, BlobStringReply, Command } from '@redis/client/dist/lib/RESP/types';
export const FIRST_KEY_INDEX = 1;
type SlowLogRawReply = ArrayReply<TuplesReply<[
timestamp: BlobStringReply,
command: BlobStringReply,
query: BlobStringReply,
took: BlobStringReply
]>>;
export function transformArguments(key: string) {
export default {
FIRST_KEY_INDEX: 1,
IS_READ_ONLY: true,
transformArguments(key: RedisArgument) {
return ['GRAPH.SLOWLOG', key];
}
type SlowLogRawReply = Array<[
timestamp: string,
command: string,
query: string,
took: string
]>;
type SlowLogReply = Array<{
timestamp: Date;
command: string;
query: string;
took: number;
}>;
export function transformReply(logs: SlowLogRawReply): SlowLogReply {
return logs.map(([timestamp, command, query, took]) => ({
timestamp: new Date(Number(timestamp) * 1000),
command,
query,
took: Number(took)
},
transformReply(reply: SlowLogRawReply) {
return reply.map(([timestamp, command, query, took]) => ({
timestamp: Number(timestamp),
command,
query,
took: Number(took)
}));
}
}
} as const satisfies Command;

View File

@@ -1,62 +0,0 @@
import { strict as assert } from 'assert';
import { pushQueryArguments } from '.';
describe('pushQueryArguments', () => {
it('simple', () => {
assert.deepEqual(
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query'),
['GRAPH.QUERY', 'graph', 'query']
);
});
describe('params', () => {
it('all types', () => {
assert.deepEqual(
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query', {
params: {
null: null,
string: '"\\',
number: 0,
boolean: false,
array: [0],
object: {a: 0}
}
}),
['GRAPH.QUERY', 'graph', 'CYPHER null=null string="\\"\\\\" number=0 boolean=false array=[0] object={a:0} query']
);
});
it('TypeError', () => {
assert.throws(() => {
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query', {
params: {
a: undefined as any
}
})
}, TypeError);
});
});
it('TIMEOUT backward compatible', () => {
assert.deepEqual(
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query', 1),
['GRAPH.QUERY', 'graph', 'query', 'TIMEOUT', '1']
);
});
it('TIMEOUT', () => {
assert.deepEqual(
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query', {
TIMEOUT: 1
}),
['GRAPH.QUERY', 'graph', 'query', 'TIMEOUT', '1']
);
});
it('compact', () => {
assert.deepEqual(
pushQueryArguments(['GRAPH.QUERY'], 'graph', 'query', undefined, true),
['GRAPH.QUERY', 'graph', 'query', '--compact']
);
});
});

View File

@@ -1,114 +1,31 @@
import * as CONFIG_GET from './CONFIG_GET';
import * as CONFIG_SET from './CONFIG_SET';;
import * as DELETE from './DELETE';
import * as EXPLAIN from './EXPLAIN';
import * as LIST from './LIST';
import * as PROFILE from './PROFILE';
import * as QUERY from './QUERY';
import * as RO_QUERY from './RO_QUERY';
import * as SLOWLOG from './SLOWLOG';
import { RedisCommandArgument, RedisCommandArguments } from '@redis/client/dist/lib/commands';
import type { RedisCommands } from '@redis/client/dist/lib/RESP/types';
import CONFIG_GET from './CONFIG_GET';
import CONFIG_SET from './CONFIG_SET';;
import DELETE from './DELETE';
import EXPLAIN from './EXPLAIN';
import LIST from './LIST';
import PROFILE from './PROFILE';
import QUERY from './QUERY';
import RO_QUERY from './RO_QUERY';
// import SLOWLOG from './SLOWLOG';
export default {
CONFIG_GET,
configGet: CONFIG_GET,
CONFIG_SET,
configSet: CONFIG_SET,
DELETE,
delete: DELETE,
EXPLAIN,
explain: EXPLAIN,
LIST,
list: LIST,
PROFILE,
profile: PROFILE,
QUERY,
query: QUERY,
RO_QUERY,
roQuery: RO_QUERY,
SLOWLOG,
slowLog: SLOWLOG
};
type QueryParam = null | string | number | boolean | QueryParams | Array<QueryParam>;
type QueryParams = {
[key: string]: QueryParam;
};
export interface QueryOptions {
params?: QueryParams;
TIMEOUT?: number;
}
export type QueryOptionsBackwardCompatible = QueryOptions | number;
export function pushQueryArguments(
args: RedisCommandArguments,
graph: RedisCommandArgument,
query: RedisCommandArgument,
options?: QueryOptionsBackwardCompatible,
compact?: boolean
): RedisCommandArguments {
args.push(graph);
if (typeof options === 'number') {
args.push(query);
pushTimeout(args, options);
} else {
args.push(
options?.params ?
`CYPHER ${queryParamsToString(options.params)} ${query}` :
query
);
if (options?.TIMEOUT !== undefined) {
pushTimeout(args, options.TIMEOUT);
}
}
if (compact) {
args.push('--compact');
}
return args;
}
function pushTimeout(args: RedisCommandArguments, timeout: number): void {
args.push('TIMEOUT', timeout.toString());
}
function queryParamsToString(params: QueryParams): string {
const parts = [];
for (const [key, value] of Object.entries(params)) {
parts.push(`${key}=${queryParamToString(value)}`);
}
return parts.join(' ');
}
function queryParamToString(param: QueryParam): string {
if (param === null) {
return 'null';
}
switch (typeof param) {
case 'string':
return `"${param.replace(/["\\]/g, '\\$&')}"`;
case 'number':
case 'boolean':
return param.toString();
}
if (Array.isArray(param)) {
return `[${param.map(queryParamToString).join(',')}]`;
} else if (typeof param === 'object') {
const body = [];
for (const [key, value] of Object.entries(param)) {
body.push(`${key}:${queryParamToString(value)}`);
}
return `{${body.join(',')}}`;
} else {
throw new TypeError(`Unexpected param type ${typeof param} ${param}`)
}
}
CONFIG_GET,
configGet: CONFIG_GET,
CONFIG_SET,
configSet: CONFIG_SET,
DELETE,
delete: DELETE,
EXPLAIN,
explain: EXPLAIN,
LIST,
list: LIST,
PROFILE,
profile: PROFILE,
QUERY,
query: QUERY,
RO_QUERY,
roQuery: RO_QUERY,
// SLOWLOG,
// slowLog: SLOWLOG
} as const satisfies RedisCommands;

View File

@@ -3,146 +3,146 @@ import testUtils, { GLOBAL } from './test-utils';
import Graph from './graph';
describe('Graph', () => {
testUtils.testWithClient('null', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN null AS key');
testUtils.testWithClient('null', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN null AS key');
assert.deepEqual(
data,
[{ key: null }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: null }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('string', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN "string" AS key');
testUtils.testWithClient('string', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN "string" AS key');
assert.deepEqual(
data,
[{ key: 'string' }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: 'string' }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('integer', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN 0 AS key');
testUtils.testWithClient('integer', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN 0 AS key');
assert.deepEqual(
data,
[{ key: 0 }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: 0 }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('boolean', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN false AS key');
testUtils.testWithClient('boolean', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN false AS key');
assert.deepEqual(
data,
[{ key: false }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: false }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('double', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN 0.1 AS key');
testUtils.testWithClient('double', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN 0.1 AS key');
assert.deepEqual(
data,
[{ key: 0.1 }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: 0.1 }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('array', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN [null] AS key');
testUtils.testWithClient('array', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN [null] AS key');
assert.deepEqual(
data,
[{ key: [null] }]
);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(
data,
[{ key: [null] }]
);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('edge', async client => {
const graph = new Graph(client as any, 'graph');
testUtils.testWithClient('edge', async client => {
const graph = new Graph(client as any, 'graph');
// check with and without metadata cache
for (let i = 0; i < 2; i++) {
const { data } = await graph.query<any>('CREATE ()-[edge :edge]->() RETURN edge');
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.equal(typeof data[0].edge.id, 'number');
assert.equal(data[0].edge.relationshipType, 'edge');
assert.equal(typeof data[0].edge.sourceId, 'number');
assert.equal(typeof data[0].edge.destinationId, 'number');
assert.deepEqual(data[0].edge.properties, {});
}
// check with and without metadata cache
for (let i = 0; i < 2; i++) {
const { data } = await graph.query<any>('CREATE ()-[edge :edge]->() RETURN edge');
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.equal(typeof data[0].edge.id, 'number');
assert.equal(data[0].edge.relationshipType, 'edge');
assert.equal(typeof data[0].edge.sourceId, 'number');
assert.equal(typeof data[0].edge.destinationId, 'number');
assert.deepEqual(data[0].edge.properties, {});
}
}, GLOBAL.SERVERS.OPEN);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('node', async client => {
const graph = new Graph(client as any, 'graph');
testUtils.testWithClient('node', async client => {
const graph = new Graph(client as any, 'graph');
// check with and without metadata cache
for (let i = 0; i < 2; i++) {
const { data } = await graph.query<any>('CREATE (node :node { p: 0 }) RETURN node');
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.equal(typeof data[0].node.id, 'number');
assert.deepEqual(data[0].node.labels, ['node']);
assert.deepEqual(data[0].node.properties, { p: 0 });
}
}, GLOBAL.SERVERS.OPEN);
// check with and without metadata cache
for (let i = 0; i < 2; i++) {
const { data } = await graph.query<any>('CREATE (node :node { p: 0 }) RETURN node');
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.equal(typeof data[0].node.id, 'number');
assert.deepEqual(data[0].node.labels, ['node']);
assert.deepEqual(data[0].node.properties, { p: 0 });
}
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('path', async client => {
const graph = new Graph(client as any, 'graph'),
[, { data }] = await Promise.all([
await graph.query('CREATE ()-[:edge]->()'),
await graph.roQuery<any>('MATCH path = ()-[:edge]->() RETURN path')
]);
testUtils.testWithClient('path', async client => {
const graph = new Graph(client as any, 'graph'),
[, { data }] = await Promise.all([
await graph.query('CREATE ()-[:edge]->()'),
await graph.roQuery<any>('MATCH path = ()-[:edge]->() RETURN path')
]);
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.ok(Array.isArray(data));
assert.equal(data.length, 1);
assert.ok(Array.isArray(data[0].path.nodes));
assert.equal(data[0].path.nodes.length, 2);
for (const node of data[0].path.nodes) {
assert.equal(typeof node.id, 'number');
assert.deepEqual(node.labels, []);
assert.deepEqual(node.properties, {});
}
assert.ok(Array.isArray(data[0].path.nodes));
assert.equal(data[0].path.nodes.length, 2);
for (const node of data[0].path.nodes) {
assert.equal(typeof node.id, 'number');
assert.deepEqual(node.labels, []);
assert.deepEqual(node.properties, {});
}
assert.ok(Array.isArray(data[0].path.edges));
assert.equal(data[0].path.edges.length, 1);
for (const edge of data[0].path.edges) {
assert.equal(typeof edge.id, 'number');
assert.equal(edge.relationshipType, 'edge');
assert.equal(typeof edge.sourceId, 'number');
assert.equal(typeof edge.destinationId, 'number');
assert.deepEqual(edge.properties, {});
}
}, GLOBAL.SERVERS.OPEN);
assert.ok(Array.isArray(data[0].path.edges));
assert.equal(data[0].path.edges.length, 1);
for (const edge of data[0].path.edges) {
assert.equal(typeof edge.id, 'number');
assert.equal(edge.relationshipType, 'edge');
assert.equal(typeof edge.sourceId, 'number');
assert.equal(typeof edge.destinationId, 'number');
assert.deepEqual(edge.properties, {});
}
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('map', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN { key: "value" } AS map');
testUtils.testWithClient('map', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN { key: "value" } AS map');
assert.deepEqual(data, [{
map: {
key: 'value'
}
}]);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(data, [{
map: {
key: 'value'
}
}]);
}, GLOBAL.SERVERS.OPEN);
testUtils.testWithClient('point', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN point({ latitude: 1, longitude: 2 }) AS point');
testUtils.testWithClient('point', async client => {
const graph = new Graph(client as any, 'graph'),
{ data } = await graph.query('RETURN point({ latitude: 1, longitude: 2 }) AS point');
assert.deepEqual(data, [{
point: {
latitude: 1,
longitude: 2
}
}]);
}, GLOBAL.SERVERS.OPEN);
assert.deepEqual(data, [{
point: {
latitude: 1,
longitude: 2
}
}]);
}, GLOBAL.SERVERS.OPEN);
});

View File

@@ -1,359 +1,359 @@
import { RedisClientType } from '@redis/client/dist/lib/client/index';
import { RedisCommandArgument, RedisFunctions, RedisScripts } from '@redis/client/dist/lib/commands';
import { QueryOptions } from './commands';
import { QueryReply } from './commands/QUERY';
import { RedisArgument, RedisFunctions, RedisScripts } from '@redis/client/dist/lib/RESP/types';
import QUERY, { QueryOptions } from './commands/QUERY';
interface GraphMetadata {
labels: Array<string>;
relationshipTypes: Array<string>;
propertyKeys: Array<string>;
labels: Array<string>;
relationshipTypes: Array<string>;
propertyKeys: Array<string>;
}
// https://github.com/RedisGraph/RedisGraph/blob/master/src/resultset/formatters/resultset_formatter.h#L20
enum GraphValueTypes {
UNKNOWN = 0,
NULL = 1,
STRING = 2,
INTEGER = 3,
BOOLEAN = 4,
DOUBLE = 5,
ARRAY = 6,
EDGE = 7,
NODE = 8,
PATH = 9,
MAP = 10,
POINT = 11
UNKNOWN = 0,
NULL = 1,
STRING = 2,
INTEGER = 3,
BOOLEAN = 4,
DOUBLE = 5,
ARRAY = 6,
EDGE = 7,
NODE = 8,
PATH = 9,
MAP = 10,
POINT = 11
}
type GraphEntityRawProperties = Array<[
id: number,
...value: GraphRawValue
id: number,
...value: GraphRawValue
]>;
type GraphEdgeRawValue = [
GraphValueTypes.EDGE,
[
id: number,
relationshipTypeId: number,
sourceId: number,
destinationId: number,
properties: GraphEntityRawProperties
]
GraphValueTypes.EDGE,
[
id: number,
relationshipTypeId: number,
sourceId: number,
destinationId: number,
properties: GraphEntityRawProperties
]
];
type GraphNodeRawValue = [
GraphValueTypes.NODE,
[
id: number,
labelIds: Array<number>,
properties: GraphEntityRawProperties
]
GraphValueTypes.NODE,
[
id: number,
labelIds: Array<number>,
properties: GraphEntityRawProperties
]
];
type GraphPathRawValue = [
GraphValueTypes.PATH,
[
nodes: [
GraphValueTypes.ARRAY,
Array<GraphNodeRawValue>
],
edges: [
GraphValueTypes.ARRAY,
Array<GraphEdgeRawValue>
]
GraphValueTypes.PATH,
[
nodes: [
GraphValueTypes.ARRAY,
Array<GraphNodeRawValue>
],
edges: [
GraphValueTypes.ARRAY,
Array<GraphEdgeRawValue>
]
]
];
type GraphMapRawValue = [
GraphValueTypes.MAP,
Array<string | GraphRawValue>
GraphValueTypes.MAP,
Array<string | GraphRawValue>
];
type GraphRawValue = [
GraphValueTypes.NULL,
null
GraphValueTypes.NULL,
null
] | [
GraphValueTypes.STRING,
string
GraphValueTypes.STRING,
string
] | [
GraphValueTypes.INTEGER,
number
GraphValueTypes.INTEGER,
number
] | [
GraphValueTypes.BOOLEAN,
string
GraphValueTypes.BOOLEAN,
string
] | [
GraphValueTypes.DOUBLE,
string
GraphValueTypes.DOUBLE,
string
] | [
GraphValueTypes.ARRAY,
Array<GraphRawValue>
GraphValueTypes.ARRAY,
Array<GraphRawValue>
] | GraphEdgeRawValue | GraphNodeRawValue | GraphPathRawValue | GraphMapRawValue | [
GraphValueTypes.POINT,
[
latitude: string,
longitude: string
]
GraphValueTypes.POINT,
[
latitude: string,
longitude: string
]
];
type GraphEntityProperties = Record<string, GraphValue>;
interface GraphEdge {
id: number;
relationshipType: string;
sourceId: number;
destinationId: number;
properties: GraphEntityProperties;
id: number;
relationshipType: string;
sourceId: number;
destinationId: number;
properties: GraphEntityProperties;
}
interface GraphNode {
id: number;
labels: Array<string>;
properties: GraphEntityProperties;
id: number;
labels: Array<string>;
properties: GraphEntityProperties;
}
interface GraphPath {
nodes: Array<GraphNode>;
edges: Array<GraphEdge>;
nodes: Array<GraphNode>;
edges: Array<GraphEdge>;
}
type GraphMap = {
[key: string]: GraphValue;
[key: string]: GraphValue;
};
type GraphValue = null | string | number | boolean | Array<GraphValue> | {
} | GraphEdge | GraphNode | GraphPath | GraphMap | {
latitude: string;
longitude: string;
latitude: string;
longitude: string;
};
type GraphReply<T> = Omit<QueryReply, 'headers' | 'data'> & {
data?: Array<T>;
type GraphReply<T> = {
data?: Array<T>;
};
type GraphClientType = RedisClientType<{
graph: {
query: typeof import('./commands/QUERY'),
roQuery: typeof import('./commands/RO_QUERY')
}
graph: {
query: typeof QUERY,
roQuery: typeof import('./commands/RO_QUERY').default
}
}, RedisFunctions, RedisScripts>;
export default class Graph {
#client: GraphClientType;
#name: string;
#metadata?: GraphMetadata;
#client: GraphClientType;
#name: string;
#metadata?: GraphMetadata;
constructor(
client: GraphClientType,
name: string
) {
this.#client = client;
this.#name = name;
}
constructor(
client: GraphClientType,
name: string
) {
this.#client = client;
this.#name = name;
}
async query<T>(
query: RedisCommandArgument,
options?: QueryOptions
) {
return this.#parseReply<T>(
await this.#client.graph.query(
this.#name,
query,
options,
true
)
);
}
async query<T>(
query: RedisArgument,
options?: QueryOptions
) {
return this.#parseReply<T>(
await this.#client.graph.query(
this.#name,
query,
options,
true
)
);
}
async roQuery<T>(
query: RedisCommandArgument,
options?: QueryOptions
) {
return this.#parseReply<T>(
await this.#client.graph.roQuery(
this.#name,
query,
options,
true
)
);
}
async roQuery<T>(
query: RedisArgument,
options?: QueryOptions
) {
return this.#parseReply<T>(
await this.#client.graph.roQuery(
this.#name,
query,
options,
true
)
);
}
#setMetadataPromise?: Promise<GraphMetadata>;
#setMetadataPromise?: Promise<GraphMetadata>;
#updateMetadata(): Promise<GraphMetadata> {
this.#setMetadataPromise ??= this.#setMetadata()
.finally(() => this.#setMetadataPromise = undefined);
return this.#setMetadataPromise;
}
#updateMetadata(): Promise<GraphMetadata> {
this.#setMetadataPromise ??= this.#setMetadata()
.finally(() => this.#setMetadataPromise = undefined);
return this.#setMetadataPromise;
}
// DO NOT use directly, use #updateMetadata instead
async #setMetadata(): Promise<GraphMetadata> {
const [labels, relationshipTypes, propertyKeys] = await Promise.all([
this.#client.graph.roQuery(this.#name, 'CALL db.labels()'),
this.#client.graph.roQuery(this.#name, 'CALL db.relationshipTypes()'),
this.#client.graph.roQuery(this.#name, 'CALL db.propertyKeys()')
]);
// DO NOT use directly, use #updateMetadata instead
async #setMetadata(): Promise<GraphMetadata> {
const [labels, relationshipTypes, propertyKeys] = await Promise.all([
this.#client.graph.roQuery(this.#name, 'CALL db.labels()'),
this.#client.graph.roQuery(this.#name, 'CALL db.relationshipTypes()'),
this.#client.graph.roQuery(this.#name, 'CALL db.propertyKeys()')
]);
this.#metadata = {
labels: this.#cleanMetadataArray(labels.data as Array<[string]>),
relationshipTypes: this.#cleanMetadataArray(relationshipTypes.data as Array<[string]>),
propertyKeys: this.#cleanMetadataArray(propertyKeys.data as Array<[string]>)
};
this.#metadata = {
labels: this.#cleanMetadataArray(labels.data as Array<[string]>),
relationshipTypes: this.#cleanMetadataArray(relationshipTypes.data as Array<[string]>),
propertyKeys: this.#cleanMetadataArray(propertyKeys.data as Array<[string]>)
};
return this.#metadata;
}
return this.#metadata;
}
#cleanMetadataArray(arr: Array<[string]>): Array<string> {
return arr.map(([value]) => value);
}
#cleanMetadataArray(arr: Array<[string]>): Array<string> {
return arr.map(([value]) => value);
}
#getMetadata<T extends keyof GraphMetadata>(
key: T,
id: number
): GraphMetadata[T][number] | Promise<GraphMetadata[T][number]> {
return this.#metadata?.[key][id] ?? this.#getMetadataAsync(key, id);
}
#getMetadata<T extends keyof GraphMetadata>(
key: T,
id: number
): GraphMetadata[T][number] | Promise<GraphMetadata[T][number]> {
return this.#metadata?.[key][id] ?? this.#getMetadataAsync(key, id);
}
// DO NOT use directly, use #getMetadata instead
async #getMetadataAsync<T extends keyof GraphMetadata>(
key: T,
id: number
): Promise<GraphMetadata[T][number]> {
const value = (await this.#updateMetadata())[key][id];
if (value === undefined) throw new Error(`Cannot find value from ${key}[${id}]`);
// DO NOT use directly, use #getMetadata instead
async #getMetadataAsync<T extends keyof GraphMetadata>(
key: T,
id: number
): Promise<GraphMetadata[T][number]> {
const value = (await this.#updateMetadata())[key][id];
if (value === undefined) throw new Error(`Cannot find value from ${key}[${id}]`);
return value;
}
// TODO: reply type
async #parseReply<T>(reply: any): Promise<GraphReply<T>> {
if (!reply.data) return reply;
const promises: Array<Promise<unknown>> = [],
parsed = {
metadata: reply.metadata,
data: reply.data!.map((row: any) => {
const data: Record<string, GraphValue> = {};
for (let i = 0; i < row.length; i++) {
data[reply.headers[i][1]] = this.#parseValue(row[i], promises);
}
return data as unknown as T;
})
};
if (promises.length) await Promise.all(promises);
return parsed;
}
#parseValue([valueType, value]: GraphRawValue, promises: Array<Promise<unknown>>): GraphValue {
switch (valueType) {
case GraphValueTypes.NULL:
return null;
case GraphValueTypes.STRING:
case GraphValueTypes.INTEGER:
return value;
}
async #parseReply<T>(reply: QueryReply): Promise<GraphReply<T>> {
if (!reply.data) return reply;
case GraphValueTypes.BOOLEAN:
return value === 'true';
const promises: Array<Promise<unknown>> = [],
parsed = {
metadata: reply.metadata,
data: reply.data!.map((row: any) => {
const data: Record<string, GraphValue> = {};
for (let i = 0; i < row.length; i++) {
data[reply.headers[i][1]] = this.#parseValue(row[i], promises);
}
case GraphValueTypes.DOUBLE:
return parseFloat(value);
return data as unknown as T;
})
};
case GraphValueTypes.ARRAY:
return value.map(x => this.#parseValue(x, promises));
if (promises.length) await Promise.all(promises);
case GraphValueTypes.EDGE:
return this.#parseEdge(value, promises);
return parsed;
}
#parseValue([valueType, value]: GraphRawValue, promises: Array<Promise<unknown>>): GraphValue {
switch (valueType) {
case GraphValueTypes.NULL:
return null;
case GraphValueTypes.STRING:
case GraphValueTypes.INTEGER:
return value;
case GraphValueTypes.BOOLEAN:
return value === 'true';
case GraphValueTypes.DOUBLE:
return parseFloat(value);
case GraphValueTypes.ARRAY:
return value.map(x => this.#parseValue(x, promises));
case GraphValueTypes.EDGE:
return this.#parseEdge(value, promises);
case GraphValueTypes.NODE:
return this.#parseNode(value, promises);
case GraphValueTypes.PATH:
return {
nodes: value[0][1].map(([, node]) => this.#parseNode(node, promises)),
edges: value[1][1].map(([, edge]) => this.#parseEdge(edge, promises))
};
case GraphValueTypes.MAP:
const map: GraphMap = {};
for (let i = 0; i < value.length; i++) {
map[value[i++] as string] = this.#parseValue(value[i] as GraphRawValue, promises);
}
return map;
case GraphValueTypes.POINT:
return {
latitude: parseFloat(value[0]),
longitude: parseFloat(value[1])
};
default:
throw new Error(`unknown scalar type: ${valueType}`);
}
}
#parseEdge([
id,
relationshipTypeId,
sourceId,
destinationId,
properties
]: GraphEdgeRawValue[1], promises: Array<Promise<unknown>>): GraphEdge {
const edge = {
id,
sourceId,
destinationId,
properties: this.#parseProperties(properties, promises)
} as GraphEdge;
const relationshipType = this.#getMetadata('relationshipTypes', relationshipTypeId);
if (relationshipType instanceof Promise) {
promises.push(
relationshipType.then(value => edge.relationshipType = value)
);
} else {
edge.relationshipType = relationshipType;
}
return edge;
}
#parseNode([
id,
labelIds,
properties
]: GraphNodeRawValue[1], promises: Array<Promise<unknown>>): GraphNode {
const labels = new Array<string>(labelIds.length);
for (let i = 0; i < labelIds.length; i++) {
const value = this.#getMetadata('labels', labelIds[i]);
if (value instanceof Promise) {
promises.push(value.then(value => labels[i] = value));
} else {
labels[i] = value;
}
}
case GraphValueTypes.NODE:
return this.#parseNode(value, promises);
case GraphValueTypes.PATH:
return {
id,
labels,
properties: this.#parseProperties(properties, promises)
nodes: value[0][1].map(([, node]) => this.#parseNode(node, promises)),
edges: value[1][1].map(([, edge]) => this.#parseEdge(edge, promises))
};
}
#parseProperties(raw: GraphEntityRawProperties, promises: Array<Promise<unknown>>): GraphEntityProperties {
const parsed: GraphEntityProperties = {};
for (const [id, type, value] of raw) {
const parsedValue = this.#parseValue([type, value] as GraphRawValue, promises),
key = this.#getMetadata('propertyKeys', id);
if (key instanceof Promise) {
promises.push(key.then(key => parsed[key] = parsedValue));
} else {
parsed[key] = parsedValue;
}
case GraphValueTypes.MAP:
const map: GraphMap = {};
for (let i = 0; i < value.length; i++) {
map[value[i++] as string] = this.#parseValue(value[i] as GraphRawValue, promises);
}
return parsed;
return map;
case GraphValueTypes.POINT:
return {
latitude: parseFloat(value[0]),
longitude: parseFloat(value[1])
};
default:
throw new Error(`unknown scalar type: ${valueType}`);
}
}
#parseEdge([
id,
relationshipTypeId,
sourceId,
destinationId,
properties
]: GraphEdgeRawValue[1], promises: Array<Promise<unknown>>): GraphEdge {
const edge = {
id,
sourceId,
destinationId,
properties: this.#parseProperties(properties, promises)
} as GraphEdge;
const relationshipType = this.#getMetadata('relationshipTypes', relationshipTypeId);
if (relationshipType instanceof Promise) {
promises.push(
relationshipType.then(value => edge.relationshipType = value)
);
} else {
edge.relationshipType = relationshipType;
}
return edge;
}
#parseNode([
id,
labelIds,
properties
]: GraphNodeRawValue[1], promises: Array<Promise<unknown>>): GraphNode {
const labels = new Array<string>(labelIds.length);
for (let i = 0; i < labelIds.length; i++) {
const value = this.#getMetadata('labels', labelIds[i]);
if (value instanceof Promise) {
promises.push(value.then(value => labels[i] = value));
} else {
labels[i] = value;
}
}
return {
id,
labels,
properties: this.#parseProperties(properties, promises)
};
}
#parseProperties(raw: GraphEntityRawProperties, promises: Array<Promise<unknown>>): GraphEntityProperties {
const parsed: GraphEntityProperties = {};
for (const [id, type, value] of raw) {
const parsedValue = this.#parseValue([type, value] as GraphRawValue, promises),
key = this.#getMetadata('propertyKeys', id);
if (key instanceof Promise) {
promises.push(key.then(key => parsed[key] = parsedValue));
} else {
parsed[key] = parsedValue;
}
}
return parsed;
}
}

View File

@@ -2,19 +2,19 @@ import TestUtils from '@redis/test-utils';
import RedisGraph from '.';
export default new TestUtils({
dockerImageName: 'redislabs/redisgraph',
dockerImageVersionArgument: 'redisgraph-version'
dockerImageName: 'redislabs/redisgraph',
dockerImageVersionArgument: 'redisgraph-version'
});
export const GLOBAL = {
SERVERS: {
OPEN: {
serverArguments: ['--loadmodule /usr/lib/redis/modules/redisgraph.so'],
clientOptions: {
modules: {
graph: RedisGraph
}
}
SERVERS: {
OPEN: {
serverArguments: ['--loadmodule /usr/lib/redis/modules/redisgraph.so'],
clientOptions: {
modules: {
graph: RedisGraph
}
}
}
}
};

View File

@@ -5,10 +5,12 @@
"main": "./dist/index.js",
"types": "./dist/index.d.ts",
"files": [
"dist/"
"dist/",
"!dist/tsconfig.tsbuildinfo"
],
"scripts": {
"test": "nyc -r text-summary -r lcov mocha -r source-map-support/register -r ts-node/register './lib/**/*.spec.ts'" },
"test": "nyc -r text-summary -r lcov mocha -r source-map-support/register -r ts-node/register './lib/**/*.spec.ts'"
},
"peerDependencies": {
"@redis/client": "*"
},