redis/node-redis
1
0
Fork 0
mirror of https://github.com/redis/node-redis.git synced 2025-08-07 13:22:56 +03:00
Code Activity

WIP

Browse Source
This commit is contained in:
Leibale
2023-04-30 10:46:43 -04:00
parent 88333c01de
commit e04609cc3a
5 changed files with 77 additions and 414 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
packages/client/lib/RESP/decoder.ts
View File

@@ -4,7 +4,7 @@ import { SimpleError, BlobError, ErrorReply } from '../errors';
import { Flags } from './types';
// https://github.com/redis/redis-specifications/blob/master/protocol/RESP3.md
export const TYPES = {
export const RESP_TYPES = {
NULL: 95, // _
BOOLEAN: 35, // #
NUMBER: 58, // :
@@ -35,7 +35,7 @@ const ASCII = {
} as const;
export const PUSH_FLAGS = {
[TYPES.BLOB_STRING]: Buffer
[RESP_TYPES.BLOB_STRING]: Buffer
};
// this was written with performance in mind, so it's not very readable... sorry :(
@@ -98,98 +98,98 @@ export class Decoder {
private _decodeTypeValue(type, chunk) {
switch (type) {
case TYPES.NULL:
case RESP_TYPES.NULL:
this._config.onReply(this._decodeNull());
return false;
case TYPES.BOOLEAN:
case RESP_TYPES.BOOLEAN:
return this._handleDecodedValue(
this._config.onReply,
this._decodeBoolean(chunk)
);
case TYPES.NUMBER:
case RESP_TYPES.NUMBER:
return this._handleDecodedValue(
this._config.onReply,
this._decodeNumber(chunk)
);
case TYPES.BIG_NUMBER:
case RESP_TYPES.BIG_NUMBER:
return this._handleDecodedValue(
this._config.onReply,
this._decodeBigNumber(
this._config.getFlags()[TYPES.BIG_NUMBER],
this._config.getFlags()[RESP_TYPES.BIG_NUMBER],
chunk
)
);
case TYPES.DOUBLE:
case RESP_TYPES.DOUBLE:
return this._handleDecodedValue(
this._config.onReply,
this._decodeDouble(
this._config.getFlags()[TYPES.DOUBLE],
this._config.getFlags()[RESP_TYPES.DOUBLE],
chunk
)
);
case TYPES.SIMPLE_STRING:
case RESP_TYPES.SIMPLE_STRING:
return this._handleDecodedValue(
this._config.onReply,
this._decodeSimpleString(
this._config.getFlags()[TYPES.SIMPLE_STRING],
this._config.getFlags()[RESP_TYPES.SIMPLE_STRING],
chunk
)
);
case TYPES.BLOB_STRING:
case RESP_TYPES.BLOB_STRING:
return this._handleDecodedValue(
this._config.onReply,
this._decodeBlobString(
this._config.getFlags()[TYPES.BLOB_STRING],
this._config.getFlags()[RESP_TYPES.BLOB_STRING],
chunk
)
);
case TYPES.VERBATIM_STRING:
case RESP_TYPES.VERBATIM_STRING:
return this._handleDecodedValue(
this._config.onReply,
this._decodeVerbatimString(
this._config.getFlags()[TYPES.VERBATIM_STRING],
this._config.getFlags()[RESP_TYPES.VERBATIM_STRING],
chunk
)
);
case TYPES.SIMPLE_ERROR:
case RESP_TYPES.SIMPLE_ERROR:
return this._handleDecodedValue(
this._config.onErrorReply,
this._decodeSimpleError(chunk)
);
case TYPES.BLOB_ERROR:
case RESP_TYPES.BLOB_ERROR:
return this._handleDecodedValue(
this._config.onErrorReply,
this._decodeBlobError(chunk)
);
case TYPES.ARRAY:
case RESP_TYPES.ARRAY:
return this._handleDecodedValue(
this._config.onReply,
this._decodeArray(this._config.getFlags(), chunk)
);
case TYPES.SET:
case RESP_TYPES.SET:
return this._handleDecodedValue(
this._config.onReply,
this._decodeSet(this._config.getFlags(), chunk)
);
case TYPES.MAP:
case RESP_TYPES.MAP:
return this._handleDecodedValue(
this._config.onReply,
this._decodeMap(this._config.getFlags(), chunk)
);
case TYPES.PUSH:
case RESP_TYPES.PUSH:
return this._handleDecodedValue(
this._config.onPush,
this._decodeArray(PUSH_FLAGS, chunk)
@@ -664,43 +664,43 @@ export class Decoder {
private _decodeNestedTypeValue(type, flags, chunk) {
switch (type) {
case TYPES.NULL:
case RESP_TYPES.NULL:
return this._decodeNull();
case TYPES.BOOLEAN:
case RESP_TYPES.BOOLEAN:
return this._decodeBoolean(chunk);
case TYPES.NUMBER:
case RESP_TYPES.NUMBER:
return this._decodeNumber(chunk);
case TYPES.BIG_NUMBER:
return this._decodeBigNumber(flags[TYPES.BIG_NUMBER], chunk);
case RESP_TYPES.BIG_NUMBER:
return this._decodeBigNumber(flags[RESP_TYPES.BIG_NUMBER], chunk);
case TYPES.DOUBLE:
return this._decodeDouble(flags[TYPES.DOUBLE], chunk);
case RESP_TYPES.DOUBLE:
return this._decodeDouble(flags[RESP_TYPES.DOUBLE], chunk);
case TYPES.SIMPLE_STRING:
return this._decodeSimpleString(flags[TYPES.SIMPLE_STRING], chunk);
case RESP_TYPES.SIMPLE_STRING:
return this._decodeSimpleString(flags[RESP_TYPES.SIMPLE_STRING], chunk);
case TYPES.BLOB_STRING:
return this._decodeBlobString(flags[TYPES.BLOB_STRING], chunk);
case RESP_TYPES.BLOB_STRING:
return this._decodeBlobString(flags[RESP_TYPES.BLOB_STRING], chunk);
case TYPES.VERBATIM_STRING:
return this._decodeVerbatimString(flags[TYPES.VERBATIM_STRING], chunk);
case RESP_TYPES.VERBATIM_STRING:
return this._decodeVerbatimString(flags[RESP_TYPES.VERBATIM_STRING], chunk);
case TYPES.SIMPLE_ERROR:
case RESP_TYPES.SIMPLE_ERROR:
return this._decodeSimpleError(chunk);
case TYPES.BLOB_ERROR:
case RESP_TYPES.BLOB_ERROR:
return this._decodeBlobError(chunk);
case TYPES.ARRAY:
case RESP_TYPES.ARRAY:
return this._decodeArray(flags, chunk);
case TYPES.SET:
case RESP_TYPES.SET:
return this._decodeSet(flags, chunk);
case TYPES.MAP:
case RESP_TYPES.MAP:
return this._decodeMap(flags, chunk);
}
}
@@ -805,7 +805,7 @@ export class Decoder {
}
private _decodeSetItems(length, flags, chunk) {
return flags[TYPES.SET] === Set ?
return flags[RESP_TYPES.SET] === Set ?
this._decodeSetAsSet(
new Set(),
length,
@@ -888,7 +888,7 @@ export class Decoder {
}
private _decodeMapItems(length, flags, chunk) {
switch (flags[TYPES.MAP]) {
switch (flags[RESP_TYPES.MAP]) {
case Map:
return this._decodeMapAsMap(
new Map(),
@@ -978,11 +978,11 @@ export class Decoder {
private _decodeMapKeyValue(type, flags, chunk) {
switch (type) {
// decode simple string map key as string (and not as buffer)
case TYPES.SIMPLE_STRING:
case RESP_TYPES.SIMPLE_STRING:
return this._decodeSimpleString(String, chunk);
// decode blob string map key as string (and not as buffer)
case TYPES.BLOB_STRING:
case RESP_TYPES.BLOB_STRING:
return this._decodeBlobString(String, chunk);
default:

60
packages/client/lib/RESP/types.ts
View File

@@ -1,13 +1,13 @@
import { RedisScriptConfig, SHA1 } from '../lua-script';
import { TYPES } from './decoder';
import { RESP_TYPES } from './decoder';
import { VerbatimString } from './verbatim-string';
export type RespTypes = typeof TYPES;
export type RESP_TYPES = typeof RESP_TYPES;
export type RespTypesUnion = RespTypes[keyof RespTypes];
export type RespTypes = RESP_TYPES[keyof RESP_TYPES];
type RespType<
RESP_TYPE extends RespTypesUnion,
RESP_TYPE extends RespTypes,
DEFAULT,
TYPES = never,
FLAG_TYPES = DEFAULT | TYPES
@@ -19,19 +19,19 @@ type RespType<
};
export type NullReply = RespType<
RespTypes['NULL'],
RESP_TYPES['NULL'],
null
>;
export type BooleanReply<
T extends boolean = boolean
> = RespType<
RespTypes['BOOLEAN'],
RESP_TYPES['BOOLEAN'],
T
>;
export type NumberReply<
T extends number = number
> = RespType<
RespTypes['NUMBER'],
RESP_TYPES['NUMBER'],
T,
`${T}`,
number | string
@@ -39,7 +39,7 @@ export type NumberReply<
export type BigNumberReply<
T extends bigint = bigint
> = RespType<
RespTypes['BIG_NUMBER'],
RESP_TYPES['BIG_NUMBER'],
T,
number | `${T}`,
bigint | number | string
@@ -47,7 +47,7 @@ export type BigNumberReply<
export type DoubleReply<
T extends number = number
> = RespType<
RespTypes['DOUBLE'],
RESP_TYPES['DOUBLE'],
T,
`${T}`,
number | string
@@ -55,7 +55,7 @@ export type DoubleReply<
export type SimpleStringReply<
T extends string = string
> = RespType<
RespTypes['SIMPLE_STRING'],
RESP_TYPES['SIMPLE_STRING'],
T,
Buffer,
string | Buffer
@@ -63,7 +63,7 @@ export type SimpleStringReply<
export type BlobStringReply<
T extends string = string
> = RespType<
RespTypes['BLOB_STRING'],
RESP_TYPES['BLOB_STRING'],
T,
Buffer,
string | Buffer
@@ -71,39 +71,39 @@ export type BlobStringReply<
export type VerbatimStringReply<
T extends string = string
> = RespType<
RespTypes['VERBATIM_STRING'],
RESP_TYPES['VERBATIM_STRING'],
T,
Buffer | VerbatimString,
string | Buffer | VerbatimString
>;
export type SimpleErrorReply = RespType<
RespTypes['SIMPLE_ERROR'],
RESP_TYPES['SIMPLE_ERROR'],
Buffer
>;
export type BlobErrorReply = RespType<
RespTypes['BLOB_ERROR'],
RESP_TYPES['BLOB_ERROR'],
Buffer
>;
export type ArrayReply<T> = RespType<
RespTypes['ARRAY'],
RESP_TYPES['ARRAY'],
Array<T>,
never,
Array<any>
>;
export type TuplesReply<T extends [...Array<unknown>]> = RespType<
RespTypes['ARRAY'],
RESP_TYPES['ARRAY'],
T,
never,
Array<any>
>;
export type SetReply<T> = RespType<
RespTypes['SET'],
RESP_TYPES['SET'],
Array<T>,
Set<T>,
Array<any> | Set<any>
>;
export type MapReply<K, V> = RespType<
RespTypes['MAP'],
RESP_TYPES['MAP'],
{ [key: string]: V },
Map<K, V> | Array<K | V>,
Map<any, any> | Array<any>
@@ -114,7 +114,7 @@ type MapKeyValue = [key: BlobStringReply, value: unknown];
type MapTuples = Array<MapKeyValue>;
export type TuplesToMapReply<T extends MapTuples> = RespType<
RespTypes['MAP'],
RESP_TYPES['MAP'],
{
[P in T[number] as P[0] extends BlobStringReply<infer S> ? S : never]: P[1];
},
@@ -134,16 +134,16 @@ type FlattenTuples<T> = (
export type ReplyUnion = NullReply | BooleanReply | NumberReply | BigNumberReply | DoubleReply | SimpleStringReply | BlobStringReply | VerbatimStringReply | SimpleErrorReply | BlobErrorReply |
// cannot reuse ArrayReply, SetReply and MapReply because of circular reference
RespType<
RespTypes['ARRAY'],
RESP_TYPES['ARRAY'],
Array<ReplyUnion>
> |
RespType<
RespTypes['SET'],
RESP_TYPES['SET'],
Array<ReplyUnion>,
Set<ReplyUnion>
> |
RespType<
RespTypes['MAP'],
RESP_TYPES['MAP'],
{ [key: string]: ReplyUnion },
Map<ReplyUnion, ReplyUnion> | Array<ReplyUnion | ReplyUnion>
>;
@@ -152,10 +152,10 @@ export type Reply = ReplyWithFlags<ReplyUnion, {}>;
export type Flag<T> = ((...args: any) => T) | (new (...args: any) => T);
type RespTypeUnion<T> = T extends RespType<RespTypesUnion, unknown, unknown, infer FLAG_TYPES> ? FLAG_TYPES : never;
type RespTypeUnion<T> = T extends RespType<RespTypes, unknown, unknown, infer FLAG_TYPES> ? FLAG_TYPES : never;
export type Flags = {
[P in RespTypesUnion]?: Flag<RespTypeUnion<Extract<ReplyUnion, RespType<P, any, any, any>>>>;
[P in RespTypes]?: Flag<RespTypeUnion<Extract<ReplyUnion, RespType<P, any, any, any>>>>;
};
type MapKey<
@@ -163,8 +163,8 @@ type MapKey<
FLAGS extends Flags
> = ReplyWithFlags<T, FLAGS & {
// simple and blob strings as map keys decoded as strings
[TYPES.SIMPLE_STRING]: StringConstructor;
[TYPES.BLOB_STRING]: StringConstructor;
[RESP_TYPES.SIMPLE_STRING]: StringConstructor;
[RESP_TYPES.BLOB_STRING]: StringConstructor;
}>;
export type ReplyWithFlags<
@@ -299,13 +299,13 @@ type Resp2Array<T> = (
export type Resp2Reply<RESP3REPLY> = (
RESP3REPLY extends RespType<infer RESP_TYPE, infer DEFAULT, infer TYPES, unknown> ?
// TODO: RESP3 only scalar types
RESP_TYPE extends RespTypes['DOUBLE'] ? BlobStringReply :
RESP_TYPE extends RespTypes['ARRAY'] | RespTypes['SET'] ? RespType<
RESP_TYPE extends RESP_TYPES['DOUBLE'] ? BlobStringReply :
RESP_TYPE extends RESP_TYPES['ARRAY'] | RESP_TYPES['SET'] ? RespType<
RESP_TYPE,
Resp2Array<DEFAULT>
> :
RESP_TYPE extends RespTypes['MAP'] ? RespType<
RespTypes['ARRAY'],
RESP_TYPE extends RESP_TYPES['MAP'] ? RespType<
RESP_TYPES['ARRAY'],
Resp2Array<Extract<TYPES, Array<any>>>
> :
RespType<