[Typescript] Function overload: For default value arguement

import { Equal, Expect } from "../helpers/type-utils";

const obj = {
  a: 1,
  b: 2,
  c: 3,
} as const;

type ObjKey = keyof typeof obj;

// If don't pass in arguement, default value is "a", return value will be 1
function getObjKey(): 1;
function getObjKey<TKey extends ObjKey>(key: TKey): typeof obj[TKey];
function getObjKey(key: ObjKey = "a") {
  return obj[key];
}

const one = getObjKey("a");
const oneByDefault = getObjKey();
const two = getObjKey("b");
const three = getObjKey("c");

type tests = [
  Expect<Equal<typeof one, 1>>,
  Expect<Equal<typeof oneByDefault, 1>>,
  Expect<Equal<typeof two, 2>>,
  Expect<Equal<typeof three, 3>>,
];

 

posted @ 2023-02-02 16:04  Zhentiw  阅读(18)  评论(0编辑  收藏  举报