在TypeScript[重复]中实现条件返回类型的函数

Question

假设我想写一个函数，它可以接受一个参数，这个参数可以是null类型，也可以是非null类型。如果参数不可为null，则返回类型也应不可为null。类似地，如果参数是可空类型，那么返回类型也应该是可空类型。

以下是我为实现这一职能所做的尝试：

type TransformKey = 'one' | 'two' | 'three'
function transform<T extends TransformKey | null>(key: T): T extends TransformKey ? string : string | null {
    if (key === null)
        return null as any
    else
        return key.toUpperCase() as any
}

函数签名似乎是正确的，因为它在调用站点上实现了所需的行为，例如：

const neverNullArg = 'two'
const neverNullResult: string = transform(neverNullArg)

const maybeNullArg: TransformKey | null = 'three'
const maybeNullResult: string | null = transform(maybeNullArg)

即返回类型确实由参数类型以正确的方式确定。

然而，我的问题在于函数的实现。如果不将返回值强制转换为any，两个返回语句都会导致TS2322错误：类型“null”/“string”不可分配给类型“T extends TransformKey”？字符串：字符串| null'。因此，我想知道该函数如何以满足签名的方式实现，而不会像任何类型转换或/@ts ignore那样破坏键入系统

Answer 1

可以为函数创建重载：

type TransformKey = 'one' | 'two' | 'three'

function transform(key: TransformKey): string;
function transform(key: TransformKey | null): string | null;
function transform(key: any): string | null {
    if (key === null)
        return null;
    else
        return key.toUpperCase();
}

const neverNullArg = 'two'
const neverNullResult: string = transform(neverNullArg)

const maybeNullArg: TransformKey | null = 'three'
const maybeNullResult: string | null = transform(maybeNullArg)

另请参见：编写良好的重载