聽說你熟悉 Pick,那就來手寫一個 MyPick!
你用過 TypeScript 內置的 Pick 工具類型么?你想知道它內部是如何實現的么?本期阿寶哥將帶你一起手寫一個 MyPick 工具類型。看完本文之后,你就能了解該工具類型涉及的相關知識點。你準備好了么?現在我們馬上進入正題。
這是一個 Pick 函數,它的作用是挑選 keys 屬性數組中每個 key 對應的值,組成一個新的對象。
function Pick(obj, keys) {
const ret = {};
for (const key of keys) {
ret[key] = obj[key];
}
return ret;
}
const user = {
id: 666,
name: "阿寶哥",
address: "廈門",
};
const PickedUser = MyPick(user, ["id", "name"]);
在 JavaScript 中我們操作的是對象,而在 TypeScript 中我們操作的是類型。同樣,在 TypeScript 中我們也可以把 User 類型轉換成只包含 id 和 name 屬性的 PickedUser 類型。
type User = {
id: number;
name: string;
address: string;
};
type PickedUser = {
id: number;
name: string;
};
要實現上述的功能,我們需要使用 TypeScript 提供的映射類型。它是一種泛型類型,可用于把原有的對象類型映射成新的對象類型。
映射類型的語法如下:
{ [ P in K ] : T }
其中 P in K 類似于 JavaScript 中的 for...in 語句,用于遍歷 K 類型中的所有類型,而 T 類型變量用于表示 TS 中的任意類型。了解完這些知識之后,我們來定義一個 MyPick 工具類型。
type MyPick<T, K extends keyof T> = {
[P in K]: T[P];
};
在以上代碼中,T、K 被稱為類型參數,與 Pick 函數中參數的作用類似。keyof 操作符用于獲取某種類型中的所有鍵,其返回類型是聯合類型。如果 keyof 操作的是對象類型,其功能與 Object.keys 方法類似。
而 K extends keyof T 是泛型約束,用于約束類型參數 K 對應的實際類型為對象類型所有鍵組成的聯合類型的子類型。T[P] 用于獲取 T 類型中 P 屬性對應的類型,其中類型變量 P 的值會在遍歷的過程中不斷改變。
下面我們來驗證一下 MyPick 工具類型,是否實現對應的功能。在實際工作中,若遇到對應的場景,我們可以直接使用 TypeScript 提供的 Pick 工具類型。
type PickedUser = MyPick<User, "id" | "name">;
// type PickedUser = {
// id: number;
// name: string;
// }
在 TypeScript 中映射類型非常重要,如果你想深入學習映射類型,可以觀看 “用了 TS 映射類型,同事直呼內行!