You've already forked node-redis
mirror of
https://github.com/redis/node-redis.git
synced 2025-08-06 02:15:48 +03:00
use dockers for tests, use npm workspaces, add rejson & redisearch modules, fix some bugs
This commit is contained in:
83
packages/client/lib/multi-command.ts
Normal file
83
packages/client/lib/multi-command.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
import { RedisCommand, RedisCommandArguments, RedisCommandRawReply, RedisScript } from './commands';
|
||||
import { WatchError } from './errors';
|
||||
|
||||
export interface RedisMultiQueuedCommand {
|
||||
args: RedisCommandArguments;
|
||||
transformReply?: RedisCommand['transformReply'];
|
||||
}
|
||||
|
||||
export default class RedisMultiCommand {
|
||||
static generateChainId(): symbol {
|
||||
return Symbol('RedisMultiCommand Chain Id');
|
||||
}
|
||||
|
||||
readonly queue: Array<RedisMultiQueuedCommand> = [];
|
||||
|
||||
readonly scriptsInUse = new Set<string>();
|
||||
|
||||
addCommand(args: RedisCommandArguments, transformReply?: RedisCommand['transformReply']): void {
|
||||
this.queue.push({
|
||||
args,
|
||||
transformReply
|
||||
});
|
||||
}
|
||||
|
||||
addScript(script: RedisScript, args: Array<unknown>): RedisCommandArguments {
|
||||
const transformedArguments: RedisCommandArguments = [];
|
||||
if (this.scriptsInUse.has(script.SHA1)) {
|
||||
transformedArguments.push(
|
||||
'EVALSHA',
|
||||
script.SHA1
|
||||
);
|
||||
} else {
|
||||
this.scriptsInUse.add(script.SHA1);
|
||||
transformedArguments.push(
|
||||
'EVAL',
|
||||
script.SCRIPT
|
||||
);
|
||||
}
|
||||
|
||||
transformedArguments.push(script.NUMBER_OF_KEYS.toString());
|
||||
|
||||
const scriptArguments = script.transformArguments(...args);
|
||||
transformedArguments.push(...scriptArguments);
|
||||
if (scriptArguments.preserve) {
|
||||
transformedArguments.preserve = scriptArguments.preserve;
|
||||
}
|
||||
|
||||
this.addCommand(
|
||||
transformedArguments,
|
||||
script.transformReply
|
||||
);
|
||||
|
||||
return transformedArguments;
|
||||
}
|
||||
|
||||
exec(): undefined | Array<RedisMultiQueuedCommand> {
|
||||
if (!this.queue.length) {
|
||||
return;
|
||||
}
|
||||
|
||||
return [
|
||||
{ args: ['MULTI'] },
|
||||
...this.queue,
|
||||
{ args: ['EXEC'] }
|
||||
];
|
||||
}
|
||||
|
||||
handleExecReplies(rawReplies: Array<RedisCommandRawReply>): Array<RedisCommandRawReply> {
|
||||
const execReply = rawReplies[rawReplies.length - 1] as (null | Array<RedisCommandRawReply>);
|
||||
if (execReply === null) {
|
||||
throw new WatchError();
|
||||
}
|
||||
|
||||
return this.transformReplies(execReply);
|
||||
}
|
||||
|
||||
transformReplies(rawReplies: Array<RedisCommandRawReply>): Array<RedisCommandRawReply> {
|
||||
return rawReplies.map((reply, i) => {
|
||||
const { transformReply, args } = this.queue[i];
|
||||
return transformReply ? transformReply(reply, args.preserve) : reply;
|
||||
});
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user