如何基于数组元素条件构建TypeScript联合类型并约束函数参数

可以通过泛型约束+条件类型+元组参数展开实现需求，无需依赖函数重载。以下是具体实现方案：

1. 定义类型映射与辅助类型 #

先统一管理VAL1/VAL2/null对应的目标类型，再通过辅助类型提取参数g的类型：

// 映射值到对应的参数类型
type ValTypeMap = {
  "VAL1": { simpleVal: string },
  "VAL2": { complex: string, val: number },
  null: undefined
};

// 根据dArg的元素类型，生成g的允许类型
type GetGType<T extends readonly (keyof ValTypeMap)[]> = 
  ValTypeMap[Exclude<T[number], null>] | (null extends T[number] ? undefined : never);

2. 实现泛型函数 #

利用元组展开的特性，根据dArg是否包含null来控制g的可选性：

const func = <T extends readonly (keyof ValTypeMap)[]>(
  dArg: T,
  ...(null extends T[number] 
    ? [g?: GetGType<T>] 
    : [g: GetGType<T>]
  )
): void => {
  // 这里编写函数逻辑
};

3. 测试示例（符合需求） #

// 情况1：dArg为[null] → g可选，仅允许undefined
func([null]);
func([null], undefined);

// 情况2：dArg包含null和VAL1 → g可选，允许undefined或{simpleVal: string}
func([null, "VAL1"]);
func([null, "VAL1"], undefined);
func([null, "VAL1"], { simpleVal: "hello" });

// 情况3：dArg为["VAL2"] → g必填，必须传入{complex: string, val: number}
func(["VAL2"], { complex: "test", val: 456 });

// 以下调用会触发TypeScript类型检查错误（符合预期）
// func(["VAL2"]); // 缺少必填参数g
// func(["VAL2"], { simpleVal: "wrong" }); // 类型不匹配

核心逻辑说明 #

• 泛型T约束为readonly (keyof ValTypeMap)[]，确保TypeScript能精准推断dArg的字面量元素类型（而非宽泛的联合数组类型）。
• 通过null extends T[number]判断dArg是否包含null，以此动态设置g的可选性。
• GetGType类型自动生成g的允许类型：排除null后映射到对应对象类型，若包含null则额外允许undefined。

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