Typescript深度替换多种类型

flo*_*nkt 4 typescript mapped-types conditional-types

我将 mongodb 与 @types/mongodb 一起使用。这为我提供了一个很好的 FilterQuery 接口,用于我的 mogodb 查询形状文档集合。在我的域对象类中,我有一些额外的逻辑,例如将日期转换为时刻对象或将浮点数转换为 BigNumber 对象。

对于我的查询,我需要将它们转换回来,例如,需要将 Moment 对象转换为日期对象等等。为了避免重复和维护单独的接口(仅用于查询),我想到使用映射类型将所有 Moment 类型替换为 Date 类型

type DeepReplace<T, Conditon, Replacement> = {
  [P in keyof T]: T[P] extends Conditon
    ? Replacement
    : T[P] extends object
    ? DeepReplace<T[P], Conditon, Replacement>
    : T[P];
};


class MyDoaminClass {
  date: Moment;
  nested: {
    date: Moment;
  };
}

const query: DeepReplace<MyDoaminClass, Moment, Date> = {
  date: moment().toDate(),
  nested: {
    date: moment().toDate()
  }
};
Run Code Online (Sandbox Code Playgroud)

这基本上是可行的,但我有大约 4-5 个这样的类型需要更换。是否有一种优雅的方法来链接多个 DeepReplace 类型,甚至更好:在一个地方指定所有类型替换?我想避免类似的事情type ReplaceHell = DeepReplace<DeepReplace<DeepReplace<MyDoaminClass, Moment, Date>, BigNumber, number>, Something, string>

jca*_*alz 7

假设您想要“一次性”进行替换,而不是作为“链”进行替换(这意味着您不打算先替换为,然后X替换Y为),那么您可以重写以获取映射元组的并集对应于. 所以你的旧将是。定义如下:YZDeepReplaceM[Condition1, Replacement1] | [Condition2, Replacement2] | ...DeepReplace<T, C, R>DeepReplace<T, [C, R]>

type DeepReplace<T, M extends [any, any]> = {
    [P in keyof T]: T[P] extends M[0]
    ? Replacement<M, T[P]>
    : T[P] extends object
    ? DeepReplace<T[P], M>
    : T[P];
}
Run Code Online (Sandbox Code Playgroud)

where查找whereReplacement<M, T>中可分配给条件的映射元组并返回相应的替换,定义如下:MT

type Replacement<M extends [any, any], T> =
    M extends any ? [T] extends [M[0]] ? M[1] : never : never;
Run Code Online (Sandbox Code Playgroud)

让我们看看它是否适用于我将在这里弥补的某些类型。鉴于以下情况:

interface DateLike {
    v: Date;
}
interface StringLike {
    v: string;
}
interface NumberLike {
    v: number;
}

interface Original {
    a: {
        dat: DateLike;
        str: StringLike;
        num: NumberLike;
        boo: boolean
    },
    b: {
        arr: NumberLike[]
    },
    c: StringLike,
    d: number
}
Run Code Online (Sandbox Code Playgroud)

让我们替换...Like类型:

type Replaced = DeepReplace<Original, 
  [DateLike, Date] | [StringLike, string] | [NumberLike, number]
>

/* equivalent to
type Replaced = {
    a: {
        dat: Date;
        str: string;
        num: number;
        boo: boolean;
    };
    b: {
        arr: number[];
    };
    c: string;
    d: number;
}
*/
Run Code Online (Sandbox Code Playgroud)

这样就可以了。


请注意,调用新的DeepReplace<T, [C, R]>this 可能与原始的具有相同的边缘情况DeepReplace<T, C, R>。例如,像这样的联合{a: string | DateLike}不会被映射。我会认为对这些的任何调整都超出了问题的范围。


好的,希望有帮助;祝你好运!

Playground 代码链接

  • 如果在 `[P in keyof T]:` 之后,您将所有内容都包装在 `T[P] extends infer TP 中?... : never` 并将 `...` 内的任何 `T[P]` 替换为 `TP`,它应该将替换分布在联合体上。如果您想要更多信息,您可能应该提出一个新问题,因为我认为注释不是放置太多新代码的好地方。 (2认同)