使用 TypeScript,我可以键入 getProperty<T, K extends keyof T> 的柯里化版本吗

With TypeScript, can I type a curried version of getProperty<T, K extends keyof T>

示例来自 https://www.typescriptlang.org/docs/handbook/advanced-types.html

function getProperty<T, K extends keyof T>(o: T, name: K): T[K] {
    return o[name]; // o[name] is of type T[K]
}

柯里化版本:

function curriedGetProperty<T, K extends keyof T>(name: K): (o: T) => T[K] {
    return (o: T) => o[name]; // o[name] is of type T[K]
}

const record = { id: 4, label: 'hello' }

const getId = curriedGetProperty('id') // Argument of type '"id"' is not assignable to parameter of type 'never'.

const id = getId(record)
const getProperty = <P extends string>(prop: P) => <O extends any>(obj: O) => obj[prop]

const record = { id: 4, label: 'hello' }

const getId = getProperty('id')

const id = getId(record)

这似乎有效。 id 的类型被正确推断为数字。唯一的问题是,如果传递给 getId 的对象没有 id 属性,您将收到 any,因此它并不严格,但总体上是一个优雅的解决方案.

编辑:写完这个答案后,我了解到 Record 类型可用于指定需要特定键的对象类型。利用这些知识,我们可以编写一个类型安全、简洁、可读的解决方案:

// implementation
const get = <K extends string>(key: K) => <V>(obj: Record<K, V>) => obj[key]

// usage
const person = {
  name: "kingdaro",
  age: 21,
}

const fruit = {
  type: "apple",
  color: "red",
}

const nameGetter = get("name")

nameGetter(person) // return type inferred as string
nameGetter(fruit) // fails, fruit has no key "name"

// minor caveat: when passing an object literal, the extra key will raise an error
// you can declare the object separately to sidestep around this
// but this wouldn't come up often anyway
nameGetter({ name: "kingdaro", age: 21 })

如果将其拆分为两步过程,它可以尽可能不冗长,同时完全类型安全:

interface recordType {
   id: number,
   label: string
}

const record = { id: 4, label: 'hello' };

const getPropertyBuilder = function <T>() {
   return <K extends keyof T>(key: K) => (o: T) => o[key];
};

const propertyBuilder = getPropertyBuilder<recordType>();
const getId = propertyBuilder('id'); // getId is (o: recordType) => number
const id = getId(record); // id is number

// or in one go
const label = getPropertyBuilder<recordType>()('label')(record); // label is string

也适用于 Partial,如前所述:

const propertyBuilder = getPropertyBuilder<Partial<typeof record>>();
const getId = propertyBuilder('id');
const id = getId(record); // id is number
const id2 = getId({ id: 3 }); // also number
type WithProp<T extends any, K extends string> = { [P in K]: T[P] }

function curriedGetProperty <P extends string>(prop: P) {
  return <T, O extends WithProp<T, typeof prop>>(o: O) => {
    return o[prop]
  }
}

似乎打字更安全。

const getId = curriedGetProperty('id')
getId({id: 'foo'}) // returns string
getId({label: 'hello'}) // fails

使用 TypeScript 3.0.3 我能够做到这一点:

function composeGetter<K extends string>(prop: K) {
    function getter<T extends { [P in K]?: any }>(object: T): T[typeof prop]
    function getter<T extends { [P in K]: any }>(object: T) {
        return object[prop]
    }

    return getter
}