如何让TypeScript识别我的对象赋值操作合法？

问题描述 #

我尝试编写以下TypeScript代码：

type A = {
    b: false,
} | {
    b: true,
    p: string;
}

function createA(b: boolean, p: string | undefined): A {

    if (b && p === undefined) {
        throw 'Error';
    }
    const a: A = {
        b,
        p,
    }
    return a;
}

但在创建a时出现类型错误：

Type 'string | undefined' is not assignable to type 'string'.
Type 'undefined' is not assignable to type 'string'.

我理解此时TypeScript无法推断p在b为true时一定已定义，于是写出了修复版本：

function createAFixed(b: boolean, p: string | undefined): A {
    let a: A;
    if (!b) {
        a = { b };
    } else {
        if (p) {
            a = { b, p };
        } else {
            throw 'Error';
        }
    }
    return a;
}

但想知道是否存在更优的实现方式？能否通过单一条件判断和单次赋值来创建a？

优化方案 #

1. 非空断言 + 三元表达式单次赋值 #

既然已经通过前置判断排除了b为true但p为undefined的非法情况，就可以用非空断言（p!）告诉TypeScript：此时p不可能是undefined。结合三元表达式就能实现单次赋值：

function createA(b: boolean, p: string | undefined): A {
    if (b && p === undefined) {
        throw 'Error';
    }
    const a: A = b ? { b, p: p! } : { b };
    return a;
}

这种写法保留了原有的单一前置判断，同时用一行代码完成a的赋值，简洁且符合逻辑。

2. 自定义类型守卫（更严谨） #

如果不想依赖非空断言，可以写一个类型守卫函数帮助TypeScript自动推导类型：

type A = { b: false } | { b: true; p: string };

// 类型守卫：验证输入是否符合A的构造条件
function isValidAConfig(b: boolean, p: string | undefined): p is string {
    return !b || p !== undefined;
}

function createA(b: boolean, p: string | undefined): A {
    if (!isValidAConfig(b, p)) {
        throw 'Error';
    }
    return b ? { b, p } : { b };
}

当isValidAConfig返回true时，TypeScript会自动知道：如果b为true，则p必然是string类型，因此直接赋值不会出现类型错误。

3. 条件类型约束参数（编译期校验） #

还可以通过条件类型优化函数参数，让非法输入在编译阶段就被拦截，无需运行时判断：

type A = { b: false } | { b: true; p: string };

function createA<T extends boolean>(
    b: T, 
    p: T extends true ? string : undefined
): A {
    return b ? { b, p } : { b };
}

这种方式下，调用createA(true, undefined)会直接触发编译错误，从根源避免了非法输入，更符合TypeScript的类型安全设计理念。

内容的提问来源于stack exchange，提问作者heapOverflow