ts 如何根据可选参数返回不同的泛型成员，有简单的写法吗？

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 800 天前的主题，其中的信息可能已经有所发展或是发生改变。

我想实现一个 ts 版本的简版 usePromise

  // 期望类型推导出：data 的类型是 {list: FileNode[]} (即返回 T)
  const [data, isOk] = usePromise(
    () => {
      return Promise.resolve({list:[]}) as Promise<{ list: FileNode[] }>;
    },
    { initValue: {list:[]} }
  );

  // 期望类型推导出：fileNodes 的类型是一个 FileNode 数组 (即返回 T['list'])
  const [fileNodes, isOk] = usePromise(
    () => {
      return Promise.resolve({list:[]}) as Promise<{ list: FileNode[] }>;
    },
    { initValue: [] },
    "list",
  );

我的 usePromise 实现如下, 利用K extends keyof T 判断如果 K 属性存在就返回T[K], 否则返回T 测试发现 ts 判断可选参数时总会失败，只会返回T.

export function usePromise<T, K>(
  factory: () => Promise<T>,
  options: Options<K extends keyof T ? T[K] : T> = {},
  filterKey?: K,
): [K extends keyof T ? T[K] : T, boolean] {
  type R = K extends keyof T ? T[K] : T;
  const [state, setState] = useState<R>(
    options.initValue!,
  );
  const isLoadingRef = useRef(false);
  useEffect(() => {
    factory().then((r) => {
      if (filterKey) {
        setState((r as any)[filterKey] as unknown as R);
      } else {
        setState(r as unknown as R);
      }
    }).catch((res) => {
      if (options.onError) options.onError(res);
      else throw res;
    });
  }, []);
  return [state, isLoadingRef.current] as [R, boolean];
}

我知道可以增加两个 function 的类型重载解决这个问题，但是手写一遍类型重载感觉太麻烦。

请教一下有没有更简单的方法呢？

3 条回复 • 2022-09-15 11:55:17 +08:00