[ts技巧] 利用枚举和位运算作标志

1,548 阅读1分钟

枚举的一个很好的用途是作为标志,用以检查一组条件中的某个条件是否为真。

以 animals 为例子,如下:


enum AnimalFlags {
    None        = 0,
    HasClaws    = 1 << 0,
    CanFly      = 1 << 1,
    EatsFish    = 1 << 2,
    Endangered  = 1 << 3,
}

我们在此使用了 左移 位运算符,分别得到二进制数字 0001001001001000(对应十进制 1,2,4, 8)。配合位运算符 |& 可以使得对标志的操作(添加、去掉、检查)变得很友好。

范例如下:

enum AnimalFlags {
    None        = 0,
    HasClaws    = 1 << 0,
    CanFly      = 1 << 1,
    EatsFish    = 1 << 2,
    Endangered  = 1 << 3,
}

interface Animal {
    flags: AnimalFlags;
}

function printAnimalAbilities(animal: Animal) {
  const animalFlags = animal.flags;
  // & 检查是否存在标志
  if (animalFlags & AnimalFlags.HasClaws) {
    console.log('animal has claws');
  }
  if (animalFlags & AnimalFlags.CanFly) {
    console.log('animal can fly');
  }
  if (animalFlags === AnimalFlags.None) {
    console.log('nothing');
  }
}

const animal: Animal = { flags: AnimalFlags.None };
printAnimalAbilities(animal); // nothing

// 添加 CanFly 标志
animal |= AnimalFlags.CanFly;
printAnimalAbilities(animal); // animal can fly

// 去掉 CanFly 标志
animal &= ~AnimalFlags.CanFly;
printAnimalAbilities(animal); // nothing

// 合并添加 Canfly 和 HasClwas 标志(本质上是两次 |= 操作)
animal |= AnimalFlags.CanFly | AnimalFlags.HasClaws;
printAnimalAbilities(animal); // animal has claws\n animal can fly


总结:

  1. 使用 左移 运算符生成二进制标志位;
  2. 使用 |= 添加一个标志;
  3. 使用 & 检查是否存在某个标志;
  4. 配合使用 &=~ 去掉一个标志;