Typescript 有没有可能为这样的函数写一个类型

我会采取这样的方法:

declare function useCache<T extends Params>(params: T): MyCache<T>;    

我们必须将Params定义为一个超类型，useCache()的所有可接受输入都可以赋值给它，我们还必须定义一个generic实用程序类型MyCache<T>，它将输入类型T、constrained到Params转换为预期的输出类型.

首先，您肯定需要从键的名称到预期类型的某种映射.您已经在示例中做到了这一点:

interface EntityMap {
    users: User,
    companies: Company,
}

由此我们可以将Params定义为mapped type:

type Params = {
    [K in keyof EntityMap]?: {
        ids: EntityMap[K]["id"][],
        fields?: (keyof EntityMap[K])[]
    }
}

请注意，这只是因为EntityMap中的每个属性类型都有一个id键. 我也没想到id会是string.相反，我使用indexed access type EntityMap[K]["id"]来说明，无论id属性是什么，您都需要在ids字段中使用该类型.

现在我们可以这样写MyCache<T>个字:

type MyCache<T extends Params> = {
    [K in keyof EntityMap & keyof T]: Record<string,
        T[K] extends {
            fields: (infer F extends keyof EntityMap[K])[]
        } ? Pick<EntityMap[K], F> : EntityMap[K]
    > };

这是另一个映射类型，尽管我们不是只映射keyof EntityMap，而是映射keyof EntityMap和keyof T的intersection.这样，输出类型只包含EntityMap的那些键，它们也作为T的键存在.

属性类型是Record<string, ⋯>(使用the Record utility type)，其中⋯取决于属性T[K]是否有fields条目. 如果没有，那么它只是EntityMap[K]，对应的实体类型.如果是这样，那么我们使用conditional type inference取出fields数组的元素类型F，constrain it作为EntityMap[K]的键，然后返回Pick<EntityMap[K], F>(使用the Pick utility type)而不仅仅是EntityMap[K].

我们就快做到了，但把useCache()个人中的T人再限制一点可能是个好主意:

declare function useCache<T extends Params &
    { [K in keyof T as Exclude<K, keyof EntityMap>]: never }>(
        params: T): MyCache<T>;

此约束将判断T是否有extra属性，如果有，则要求它们为the impossible never type. 这将阻止使用任何额外的密钥.额外的键不一定会造成任何伤害(尽管这取决于useCache()的实现，MyCache<T>的输出会忽略它们，但可以说它值得编译器错误.

好的，让我们来测试一下:

const { users: users1 } = useCache({
    users: {
        ids: ['1', '2'],
    },
});
//^? const users1: Record<string, User>
const userName = users1['1'].name; // okay

const { users: users2 } = useCache({
    users: {
        ids: ['1', '2'],
        fields: ['name', 'id']
    }
});
// const users2: Record<string, Pick<User, "id" | "name">>
users2['1'].name // okay
users2['1'].age // error
//          ~~~
// Property 'age' does not exist on type 'Pick<User, "id" | "name">'.

const { users, companies } = useCache({
    users: {
        ids: ['1', '2'],
    },
    companies: {
        ids: ['3', '4'],
        fields: ['title'],
    }
});
// const users: Record<string, User>
// const companies: Record<string, Pick<Company, "title">>

看上go 不错.基本用法可以达到预期效果.具有fields的属性返回Pick类型，而不具有fields的属性返回完整的实体类型.让我们确保禁止额外的密钥:

useCache({ oops: 123 }); // error!
//         ~~~~
// Type 'number' is not assignable to type 'never'

看起来也不错.

Playground link to code个