Question

我正在尝试创建一个更新功能来更新记录中的深层值。对于不同深度的路径，我都有大量的变体。

我似乎无法弄清楚如何正确键入要更新的值上使用的回调函数。

interface Test {
    foo?: { bar: number }
}
const input: Test = { foo: { bar: 1 } }

update(input, 'foo', 'bar')(v => v + 1)

使用该函数时，它告诉我“ Object（v）类型未知”。

但是例如，我有类似的set函数，其定义几乎相同，但是在像这样使用时可以正确键入：

set(input, 'foo', 'bar')(2)

这是我的职能

type UpdateFn<T> = (value: T) => T
export function update<T extends Record<string, any>, K1 extends keyof T>(
    record: T | undefined,
    key1: K1
): (callback: UpdateFn<NonNullable<T[K1]>>) => T
export function update<
    T extends Record<string, any>,
    K1 extends keyof T,
    K2 extends keyof NonNullable<T[K1]>
>(
    record: T | undefined,
    key1: K1,
    key2: K2
): (callback: UpdateFn<NonNullable<T[K1][K2]>>) => T

export function update<
    T extends Record<string, any>,
    K1 extends keyof T,
    K2 extends keyof NonNullable<T[K1]>
>(
    record: T | undefined,
    key1: K1,
    key2?: K2
): (
    callback:
        | UpdateFn<NonNullable<T[K1]>>
        | UpdateFn<NonNullable<T[K1][K2]>>
) => T | undefined {
    return callback => {
        if (record === undefined) return record

        if (key2 === undefined) {
            const value = get(record, key1)
            if (value === undefined) return record
            return set(record, key1)(callback(value))
        } else {
            const value = get(record, key1, key2)
            if (value === undefined) return record
            return set(record, key1, key2)(callback(value))
        }
    }
}

设置（正常工作）：

export function set<
    T extends Record<string, any>,
    K1 extends keyof T,
    K2 extends keyof NonNullable<T[K1]>
>(record: T | undefined, key1: K1, key2: K2): (value: T[K1][K2]) => T

Answer 1

假设我只是想解决类型问题而不是实现问题，那么第二次重载可能应该是这样的：

export function update<
    T extends Record<string, any>,
    K1 extends keyof T,
    K2 extends keyof NonNullable<T[K1]>
>(
    record: T | undefined,
    key1: K1,
    key2: K2
): (callback: UpdateFn<NonNullable<NonNullable<T[K1]>[K2]>>) => T

如果NonNullable和record[key1][key2]都已定义/非空，那么其中额外的record可以确保您在谈论record[key1]的类型。可能还有其他更通用或更简洁的方式来为update()进行键入，但这至少可以解决您遇到的问题：

update(input, 'foo', 'bar')(v => v + 1); // okay

希望有所帮助；祝你好运！

Link to code

具有深层路径和回调的update Record函数的类型

1 个答案: