fix #1758 - implement some CLIENT commands, add name to RedisClientOptions

2025-08-06 02:15:48 +03:00 · 2021-12-06 21:48:14 -05:00
parent 82920aef0b
commit ec7ccaf827
14 changed files with 323 additions and 9 deletions
--- a/packages/client/lib/commands/CLIENT_KILL.ts
+++ b/packages/client/lib/commands/CLIENT_KILL.ts
@@ -0,0 +1,95 @@
+import { RedisCommandArguments } from '.';
+
+export enum ClientKillFilters {
+    ADDRESS = 'ADDR',
+    LOCAL_ADDRESS = 'LADDR',
+    ID = 'ID',
+    TYPE = 'TYPE',
+    USER = 'USER',
+    SKIP_ME = 'SKIPME'
+}
+
+interface KillFilter<T extends ClientKillFilters> {
+    filter: T;
+}
+
+interface KillAddress extends KillFilter<ClientKillFilters.ADDRESS> {
+    address: `${string}:${number}`;
+}
+
+interface KillLocalAddress extends KillFilter<ClientKillFilters.LOCAL_ADDRESS> {
+    localAddress: `${string}:${number}`;
+}
+
+interface KillId extends KillFilter<ClientKillFilters.ID> {
+    id: number | `${number}`;
+}
+
+interface KillType extends KillFilter<ClientKillFilters.TYPE> {
+    type: 'normal' | 'master' | 'replica' | 'pubsub';
+}
+
+interface KillUser extends KillFilter<ClientKillFilters.USER> {
+    username: string;
+}
+
+type KillSkipMe = ClientKillFilters.SKIP_ME | (KillFilter<ClientKillFilters.SKIP_ME> & {
+    skipMe: boolean;
+});
+
+type KillFilters = KillAddress | KillLocalAddress | KillId | KillType | KillUser | KillSkipMe;
+
+export function transformArguments(filters: KillFilters | Array<KillFilters>): RedisCommandArguments {
+    const args = ['CLIENT', 'KILL'];
+
+    if (Array.isArray(filters)) {
+        for (const filter of filters) {
+            pushFilter(args, filter);
+        }
+    } else {
+        pushFilter(args, filters);
+    }
+
+    return args;
+}
+
+function pushFilter(args: RedisCommandArguments, filter: KillFilters): void {
+    if (filter === ClientKillFilters.SKIP_ME) {
+        args.push('SKIPME');
+        return;
+    }
+
+    args.push(filter.filter);
+
+    switch(filter.filter) {
+        case ClientKillFilters.ADDRESS:
+            args.push(filter.address);
+            break;
+
+        case ClientKillFilters.LOCAL_ADDRESS:
+            args.push(filter.localAddress);
+            break;
+
+        case ClientKillFilters.ID:
+            args.push(
+                typeof filter.id === 'number' ?
+                    filter.id.toString() :
+                    filter.id
+            );
+            break;
+
+        case ClientKillFilters.TYPE:
+            args.push(filter.type);
+            break;
+
+        case ClientKillFilters.USER:
+                args.push(filter.username);
+                break;
+
+        case ClientKillFilters.SKIP_ME:
+            args.push(filter.skipMe ? 'yes' : 'no');
+            break;
+    }
+}
+
+export declare function transformReply(): number;