some more commands, multi.exec<'typed'>

2025-08-06 02:15:48 +03:00 · 2023-04-24 19:50:58 -04:00
parent 39bcac6484
commit b272f18818
13 changed files with 171 additions and 132 deletions
--- a/packages/client/lib/commands/FUNCTION_STATS.ts
+++ b/packages/client/lib/commands/FUNCTION_STATS.ts
@@ -1,87 +1,87 @@
-// import { RedisCommandArguments } from '.';
+// // import { RedisCommandArguments } from '.';

-// export function transformArguments(): RedisCommandArguments {
+// // export function transformArguments(): RedisCommandArguments {
+// //     return ['FUNCTION', 'STATS'];
+// // }
+
+// // type FunctionStatsRawReply = [
+// //     'running_script',
+// //     null | [
+// //         'name',
+// //         string,
+// //         'command',
+// //         string,
+// //         'duration_ms',
+// //         number
+// //     ],
+// //     'engines',
+// //     Array<any> // "flat tuples" (there is no way to type that)
+// //     // ...[string, [
+// //     //     'libraries_count',
+// //     //     number,
+// //     //     'functions_count',
+// //     //     number
+// //     // ]]
+// // ];
+
+// // interface FunctionStatsReply {
+// //     runningScript: null | {
+// //         name: string;
+// //         command: string;
+// //         durationMs: number;
+// //     };
+// //     engines: Record<string, {
+// //         librariesCount: number;
+// //         functionsCount: number;
+// //     }>;
+// // }
+
+// // export function transformReply(reply: FunctionStatsRawReply): FunctionStatsReply {
+// //     const engines = Object.create(null);
+// //     for (let i = 0; i < reply[3].length; i++) {
+// //         engines[reply[3][i]] = {
+// //             librariesCount: reply[3][++i][1],
+// //             functionsCount: reply[3][i][3]
+// //         };
+// //     }
+
+// //     return {
+// //         runningScript: reply[1] === null ? null : {
+// //             name: reply[1][1],
+// //             command: reply[1][3],
+// //             durationMs: reply[1][5]
+// //         },
+// //         engines
+// //     };
+// // }
+
+
+// // #!LUA name=math \n redis.register_function{ function_name = "square", callback = function(keys, args) return args[1] * args[1] end, flags = { "no-writes" } }
+
+// import { Command, TuplesToMapReply, BlobStringReply, NullReply, NumberReply, MapReply } from '../RESP/types';
+
+// type FunctionStatsReply = TuplesToMapReply<[
+//   [BlobStringReply<'running_script'>, NullReply | TuplesToMapReply<[
+//     [BlobStringReply<'name'>, BlobStringReply],
+//     [BlobStringReply<'command'>, BlobStringReply],
+//     [BlobStringReply<'duration_ms'>, NumberReply]
+//   ]>],
+//   [BlobStringReply<'engines'>, MapReply<BlobStringReply, TuplesToMapReply<[
+//     [BlobStringReply<'libraries_count'>, NumberReply],
+//     [BlobStringReply<'functions_count'>, NumberReply]
+//   ]>>]
+// ]>;
+
+// export default {
+//   IS_READ_ONLY: true,
+//   FIRST_KEY_INDEX: undefined,
+//   transformArguments() {
 //     return ['FUNCTION', 'STATS'];
-// }
+//   },
+//   transformReply: {
+//     2: (reply) => {

-// type FunctionStatsRawReply = [
-//     'running_script',
-//     null | [
-//         'name',
-//         string,
-//         'command',
-//         string,
-//         'duration_ms',
-//         number
-//     ],
-//     'engines',
-//     Array<any> // "flat tuples" (there is no way to type that)
-//     // ...[string, [
-//     //     'libraries_count',
-//     //     number,
-//     //     'functions_count',
-//     //     number
-//     // ]]
-// ];
-
-// interface FunctionStatsReply {
-//     runningScript: null | {
-//         name: string;
-//         command: string;
-//         durationMs: number;
-//     };
-//     engines: Record<string, {
-//         librariesCount: number;
-//         functionsCount: number;
-//     }>;
-// }
-
-// export function transformReply(reply: FunctionStatsRawReply): FunctionStatsReply {
-//     const engines = Object.create(null);
-//     for (let i = 0; i < reply[3].length; i++) {
-//         engines[reply[3][i]] = {
-//             librariesCount: reply[3][++i][1],
-//             functionsCount: reply[3][i][3]
-//         };
-//     }
-
-//     return {
-//         runningScript: reply[1] === null ? null : {
-//             name: reply[1][1],
-//             command: reply[1][3],
-//             durationMs: reply[1][5]
-//         },
-//         engines
-//     };
-// }
-
-
-// #!LUA name=math \n redis.register_function{ function_name = "square", callback = function(keys, args) return args[1] * args[1] end, flags = { "no-writes" } }
-
-import { Command, TuplesToMapReply, BlobStringReply, NullReply, NumberReply, MapReply } from '../RESP/types';
-
-type FunctionStatsReply = TuplesToMapReply<[
-  [BlobStringReply<'running_script'>, NullReply | TuplesToMapReply<[
-    [BlobStringReply<'name'>, BlobStringReply],
-    [BlobStringReply<'command'>, BlobStringReply],
-    [BlobStringReply<'duration_ms'>, NumberReply]
-  ]>],
-  [BlobStringReply<'engines'>, MapReply<BlobStringReply, TuplesToMapReply<[
-    [BlobStringReply<'libraries_count'>, NumberReply],
-    [BlobStringReply<'functions_count'>, NumberReply]
-  ]>>]
-]>;
-
-export default {
-  IS_READ_ONLY: true,
-  FIRST_KEY_INDEX: undefined,
-  transformArguments() {
-    return ['FUNCTION', 'STATS'];
-  },
-  transformReply: {
-    2: (reply) => {
-
-    },
-    3: undefined as unknown as () => 
-  }
-} as const satisfies Command;
+//     },
+//     3: undefined as unknown as () => 
+//   }
+// } as const satisfies Command;
--- a/packages/client/lib/commands/RESTORE-ASKING.ts
+++ b/packages/client/lib/commands/RESTORE-ASKING.ts
@@ -1,5 +1,10 @@
-export function transformArguments(): Array<string> {
-    return ['RESTORE-ASKING'];
-}
+import { SimpleStringReply, Command } from '../RESP/types';

-export declare function transformReply(): string;
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments() {
+    return ['RESTORE-ASKING'];
+  },
+  transformReply: undefined as unknown as () => SimpleStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SAVE.ts
+++ b/packages/client/lib/commands/SAVE.ts
@@ -1,7 +1,10 @@
-import { RedisCommandArgument } from '.';
+import { SimpleStringReply, Command } from '../RESP/types';

-export function transformArguments(): Array<string> {
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments() {
    return ['SAVE'];
-}
-
-export declare function transformReply(): RedisCommandArgument;
+  },
+  transformReply: undefined as unknown as () => SimpleStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SCRIPT_DEBUG.ts
+++ b/packages/client/lib/commands/SCRIPT_DEBUG.ts
@@ -1,5 +1,10 @@
-export function transformArguments(mode: 'YES' | 'SYNC' | 'NO'): Array<string> {
-    return ['SCRIPT', 'DEBUG', mode];
-}
+import { SimpleStringReply, Command } from '../RESP/types';

-export declare function transformReply(): string;
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments(mode: 'YES' | 'SYNC' | 'NO') {
+    return ['SCRIPT', 'DEBUG', mode];
+  },
+  transformReply: undefined as unknown as () => SimpleStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SCRIPT_EXISTS.ts
+++ b/packages/client/lib/commands/SCRIPT_EXISTS.ts
@@ -1,8 +1,11 @@
-import { RedisCommandArguments } from '.';
+import { RedisArgument, ArrayReply, NumberReply, Command } from '../RESP/types';
 import { pushVariadicArguments } from './generic-transformers';

-export function transformArguments(sha1: string | Array<string>): RedisCommandArguments {
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments(sha1: RedisArgument | Array<RedisArgument>) {
    return pushVariadicArguments(['SCRIPT', 'EXISTS'], sha1);
-}
-
-export { transformBooleanArrayReply as transformReply } from './generic-transformers';
+  },
+  transformReply: undefined as unknown as () => ArrayReply<NumberReply>
+} as const satisfies Command;
--- a/packages/client/lib/commands/SCRIPT_FLUSH.ts
+++ b/packages/client/lib/commands/SCRIPT_FLUSH.ts
@@ -1,11 +1,16 @@
-export function transformArguments(mode?: 'ASYNC' | 'SYNC'): Array<string> {
+import { SimpleStringReply, Command } from '../RESP/types';
+
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments(mode?: 'ASYNC' | 'SYNC') {
    const args = ['SCRIPT', 'FLUSH'];

    if (mode) {
-        args.push(mode);
+      args.push(mode);
    }

    return args;
-}
-
-export declare function transformReply(): string;
+  },
+  transformReply: undefined as unknown as () => SimpleStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SCRIPT_KILL.ts
+++ b/packages/client/lib/commands/SCRIPT_KILL.ts
@@ -1,5 +1,10 @@
-export function transformArguments(): Array<string> {
-    return ['SCRIPT', 'KILL'];
-}
+import { SimpleStringReply, Command } from '../RESP/types';

-export declare function transformReply(): string;
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments() {
+    return ['SCRIPT', 'KILL'];
+  },
+  transformReply: undefined as unknown as () => SimpleStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SCRIPT_LOAD.ts
+++ b/packages/client/lib/commands/SCRIPT_LOAD.ts
@@ -1,5 +1,10 @@
-export function transformArguments(script: string): Array<string> {
-    return ['SCRIPT', 'LOAD', script];
-}
+import { BlobStringReply, Command, RedisArgument } from '../RESP/types';

-export declare function transformReply(): string;
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments(script: RedisArgument) {
+    return ['SCRIPT', 'LOAD', script];
+  },
+  transformReply: undefined as unknown as () => BlobStringReply
+} as const satisfies Command;
--- a/packages/client/lib/commands/SET.ts
+++ b/packages/client/lib/commands/SET.ts
@@ -1,6 +1,6 @@
 import { RedisArgument, SimpleStringReply, BlobStringReply, NullReply, Command } from '../RESP/types';

-interface SetOptions {
+export interface SetOptions {
  expiration?: {
    type: 'EX' | 'PX' | 'EXAT' | 'PXAT';
    value: number;
--- a/packages/client/lib/commands/SETBIT.ts
+++ b/packages/client/lib/commands/SETBIT.ts
@@ -1,14 +1,15 @@
-import { RedisCommandArgument, RedisCommandArguments } from '.';
+import { RedisArgument, NumberReply, Command } from '../RESP/types';
 import { BitValue } from './generic-transformers';

-export const FIRST_KEY_INDEX = 1;
-
-export function transformArguments(
-    key: RedisCommandArgument,
+export default {
+  IS_READ_ONLY: false,
+  FIRST_KEY_INDEX: 1,
+  transformArguments(
+    key: RedisArgument,
    offset: number,
    value: BitValue
-): RedisCommandArguments {
+  ) {
    return ['SETBIT', key, offset.toString(), value.toString()];
-}
-
-export declare function transformReply(): BitValue;
+  },
+  transformReply: undefined as unknown as () => NumberReply<BitValue>
+} as const satisfies Command;