TypeScript中如何从泛型联合类型V | ((k:K)=>V)安全取值？

为什么dflt(k)会报错？ #

TypeScript对联合类型V | ((k: K) => V)的类型解析存在歧义：如果V本身是Function类型（比如V = () => number），联合类型会被拆解为((k: K) => V) | (V & Function)。后者虽属于函数类型，但它的调用签名不一定和(k: K) => V兼容——比如V是无参函数时，V & Function无法接受K类型参数，TypeScript因此判定不是所有分支都能安全调用，抛出错误。

为什么Exclude<V, Function> | ((k: K) => V)无法解决问题？ #

Exclude<V, Function>确实能排除V中属于Function的部分，但TypeScript的typeof dflt === 'function'类型守卫只能判断值是否为函数，无法区分它是((k: K) => V)还是其他函数类型（若V原本包含其他函数类型）。比如V = string | (() => number)时，Exclude<V, Function>得到string，联合((k: K) => V)后，typeof dflt === 'function'分支里，TypeScript仅知道dflt是函数，却无法确定它就是能接收K参数的(k: K) => V类型，仍会认为调用dflt(k)不安全。

可行解决方案 #

方案1：使用类型断言 #

直接在调用处断言dflt为目标函数类型，适合确定逻辑正确的场景：

class Hash<K, V> {
  get_or_default(k: K, dflt: V | ((k: K) => V)): V {
    return typeof dflt === 'function' ? (dflt as (k: K) => V)(k) : dflt;
  }
}

方案2：自定义类型守卫 #

通过自定义守卫让TypeScript精确缩小类型范围：

class Hash<K, V> {
  private isDefaultFn(dflt: V | ((k: K) => V)): dflt is (k: K) => V {
    return typeof dflt === 'function';
  }

  get_or_default(k: K, dflt: V | ((k: K) => V)): V {
    return this.isDefaultFn(dflt) ? dflt(k) : dflt;
  }
}

方案3：重构参数类型（限无需k参数场景） #

若默认值函数不需要接收k，可简化参数类型：

class Hash<K, V> {
  get_or_default(k: K, dflt: V | (() => V)): V {
    return typeof dflt === 'function' ? dflt() : dflt;
  }
}

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