如何能够根据联合类型区分两个对象?

时间:2020-04-24 23:54:23

标签: typescript type-inference

我想实现tryInline,它应该尝试调用一个函数并根据调用的成功或失败返回两种对象。

如果成功调用了fn参数,则tryInline应该返回:

{
  ok: true,
  data: <The value returned by `fn`>,
  error: null
}

如果fn抛出错误,则tryInline应该返回:

{
  ok: false,
  data: null,
  error: Error
}

我设法执行以下操作:

type Result<D> =
  | { ok: true; data: D; error: null }
  | { ok: false; data: null; error: Error };

function tryInline<Fn extends (...args: any[]) => any>(
  fn: Fn,
  ...args: Parameters<Fn>
): Result<ReturnType<Fn>> {
  try {
    const data = fn(...args);

    return { ok: true, data, error: null };
  } catch (error) {
    return { ok: false, data: null, error: error };
  }
}

const { ok, data, error } = tryInline((a: number, b: number) => a + b, 1, 3);

if (ok) {
  console.log(ok); // inferred type: `true` ?
  console.log(data); // inferred type: `number | null` ? Should be `number`
  console.log(error); // inferred type: `Error | null` ? Should be `null`
}

if (!ok) {
  console.log(ok); // inferred type: `boolean` ? Should be `false`
  console.log(data); // inferred type: `number | null` ? Should be `null`
  console.log(error); // inferred type: `Error | null` ? Should be `Error`
}

Playground link.

但是,我希望能够正确推断if块中的类型,这不是当前情况。

有没有办法解决这个推理问题?

1 个答案:

答案 0 :(得分:4)

您需要在条件分支后使用 解构,因为TS链接的类型是对象,而不是普通变量。分解它们时,okdata之间的链接会丢失。此外,我认为这样比较干净,因为您正在破坏真正需要的东西。


const result = tryInline((a: number, b: number) => a + b, 1, 3);

if (result.ok) {
  const { data } = result;
  console.log(data); // inferred type: `number` 
}

if (!result.ok) {
  const { error } = result;
  console.log(error); // inferred type: `Error`
}

如果您只是想在不分支的情况下进行分解,那么对我来说,只需const { data, error } = tryInline(...)并继续进行if (data) { ... }

Playground Link