typescript - 根据可选参数指定返回类型

Question

function f<T>(defaultValue?: T) { return defaultValue; }

const definitelyUndefined = f<string>();      // type: string | undefined
const definitelyString = f<string>('foobar'); // type: string | undefined

是否可以定义 f()使得 definitelyUndefined隐含地是 undefined和 definitelyString隐含地是 string ?

背景

我的真实世界用例是我正在使用并希望改进的功能。是 function f<T>(o: { [key: string]: T }, key: string, defaultValue?: T)它返回 key[o]如果存在，否则 defaultValue .当我提供它时 defaultValue ，我保证得到 T回来，但 Typescript 认为它 T | undefined .

Answer 1

对于您的第一种情况，我立即想到使用重载，例如:

function f<T>(): undefined;
function f<T>(value: T): T;
function f<T>(value?: T) { return value; }

const definitelyUndefined = f();      // type: undefined
const definitelyString = f('foobar'); // type: "foobar"

但是，对于更复杂的用例，我认为您可以使用重载和更复杂的泛型来解决它，例如:

function f<T, K extends string & keyof T>(o: T, key: K, defaultValue?: T[K]): T[K];
function f<T, V = undefined>(o: T, key: string, defaultValue?: V): V;
function f<T, V = undefined>(o: T, key: string, defaultValue?: V) {
  return o[key] || defaultValue;
}

const obj = { foo: "123" };

const definitelyUndefined = f(obj, "bar");    // type: undefined
const definitelyNumber = f(obj, "bar", 123);  // type: number
const definitelyString = f(obj, "foo");       // type: string

我不知道这是否适用于所有可能的场景(因为返回类型是基于泛型类型参数确定的，而不是实际的函数参数)，但我认为它非常接近。