TypeScript 中的類型保護

在 TypeScript 中使用聯合類型時，往往會碰到這種尴尬的情況：

interface Bird {
    // 獨有方法
    fly();
    // 共有方法
    layEggs();
}

interface Fish {
    // 獨有方法
    swim();
    // 共有方法
    layEggs();
}

function getSmallPet(): Fish | Bird {
    // ...
}

let pet = getSmallPet();
pet.layEggs(); // 正常
pet.swim();    // ts 報錯
      

如上所示，getSmallPet 函數中，既可以傳回 Fish 類型的對象，又可以傳回 Bird 類型的對象。由于傳回的對象類型不确定，是以使用聯合類型對象共有的方法時，一切正常，但是使用聯合類型對象各自獨有的方法時，ts 會報錯。

那麼如何解決這個問題呢？最粗暴的方法當然是将聯合類型轉換為 any，不過這種方法不值得提倡，畢竟我們寫的是 TypeScript 而不是 AnyScript。

此時，我們使用今天的主角——類型保護，閃亮登場，它可以完美的解決這個問題。

孔乙己說過，茴香豆有四種寫法，同理，實作類型保護，也有四種寫法。

類型斷言

類型斷言是最常用的一種類型保護方法，即直接指定類型。由于，TypeScript 中識别的類型大多是靠 TypeScript 的自動類型推算算出來的，是以會出現上面所說的那種問題，即 TypeScript 不知道具體對象類型是什麼，是以不确定有沒有聯合類型各自獨有的方法。

當使用類型斷言直接指定類型時，相當于你讓 TypeScript 開啟了上帝模式，可以直接知道具體類型是聯合類型中的那個，此時再使用對象的獨有方法就符合 TypeScript 的推斷了。

interface Bird {
  // 獨有方法
  fly();
  // 共有方法
  layEggs();
}

interface Fish {
  // 獨有方法
  swim();
  // 共有方法
  layEggs();
}

function getSmallPet(): Fish | Bird {
  // ...
}

let pet = getSmallPet();
pet.layEggs(); // 正常
// 通過鴨子類型來進行判斷
if ((pet as Bird).fly) {
  // 類型斷言
  (pet as Bird).fly()
} else {
  // 類型斷言
  (pet as Fish).swim()
}
      

如果嫌棄通過 as 來進行類型斷言不夠上流，還可以使用類泛型的寫法，即：

let pet = getSmallPet();
pet.layEggs(); // 正常
// 通過鴨子類型來進行判斷
if ((<Bird>pet).fly) {
  (<Bird>pet).fly()
} else {
  (<Fish>pet).swim()
}
      

tips：友情提示，雖然使用類泛型寫法進行類型斷言看起來高端一些，但是由于在 tsx 中文法存在歧義，是以為了統一起見，推薦使用 as 的方法進行類型斷言。

in 文法

在 js 中，我們經常使用 in 文法來判斷指定的屬性是否在指定的對象或其原型鍊中。

同理，在 TypeScript 中，我們可以通過這種方法确認對象類型。

interface Bird {
  // 獨有方法
  fly();
  // 共有方法
  layEggs();
}

interface Fish {
  // 獨有方法
  swim();
  // 共有方法
  layEggs();
}

function getSmallPet(): Fish | Bird {
  // ...
}

let pet = getSmallPet();
pet.layEggs(); // 正常
// 使用 in 文法進行類型保護
if ('fly' in pet) {
  pet.fly()
} else {
  pet.swim()
}
      

原理同類型斷言一樣，都是引導 TypeScript 的類型推斷，确定對象類型。

instanceof 文法

當聯合類型中使用的是 class 而不是 interface 時，instanceof 文法就派上用場了，通過 instanceof 文法可以區分不同的 class 類型。

class Bird {
  // 獨有方法
  fly() {};
  // 共有方法
  layEggs() {};
}

class Fish {
  // 獨有方法
  swim() {};
  // 共有方法
  layEggs() {};
}

function getSmallPet(): Fish | Bird {
  // ...
}

let pet = getSmallPet();
pet.layEggs(); // 正常
// 使用 in 文法進行
if (pet instanceof Bird) {
  pet.fly()
} else {
  pet.swim()
}      

typeof 文法

typeof 文法不同于 in 文法以及 instanceof 文法，in 文法以及 instanceof 文法都是用來引導類型推斷進行不同對象類型推斷，而 typeof 文法常用于基本類型的推斷（或者是聯合使用基本類型和對象類型）。

簡而言之，當使用 typeof 能夠區分聯合類型中的不同類型時，即可使用它。

function getSmallPet(): number | string {
  // ...
}

let pet = getSmallPet();
if (typeof pet === 'number') {
  pet++
} else {
  pet = Number(pet) + 1
}      

總結

就如茴香豆的四種寫法的本質依然是茴香豆一樣，類型保護的四種寫法的本質也是一樣的，即，引導 TypeScript 中的類型推斷将類型推斷的多選題變為單選題，這就是類型保護的本質。