C#标记枚举实现原理及Scala同类功能实现方案问询

基础定义与核心逻辑 #

标记枚举本质是基于整数位运算的集合表示，核心要求有两个：

• 给枚举添加[Flags]特性：这是语法糖，主要优化ToString()的输出格式，不影响底层位运算逻辑
• 枚举成员的值必须是2的整数次幂（如1、2、4、8...），确保每个成员对应整数二进制中的唯一一位，避免位重叠导致的判断冲突

示例定义：

[Flags]
public enum Permission
{
    None = 0,
    Read = 1 << 0, // 二进制 0001
    Write = 1 << 1, // 二进制 0010
    Delete = 1 << 2, // 二进制 0100
    Admin = Read | Write | Delete // 二进制 0111
}

|运算符的工作原理 #

|在这里就是整数按位或运算：将两个枚举值对应的二进制位逐一进行或操作，只要某一位有1，结果的该位就会被设为1，以此实现多个枚举成员的组合。

示例：

var userPerms = Permission.Read | Permission.Write;
// 等价于二进制 0001 | 0010 = 0011，对应整数3

后续判断是否包含某个权限时，用&按位与运算验证：如果结果等于被判断的成员，说明该位已被设置：

bool canRead = (userPerms & Permission.Read) == Permission.Read; // 返回true

Scala没有内置标记枚举特性，但可以通过位运算+自定义类型封装实现等价功能，以下是两种实用方案：

方案1：Scala 3 枚举+位运算 #

利用Scala 3枚举支持成员方法的特性，直接封装位运算逻辑：

enum Permission(val value: Int):
  case None extends Permission(0)
  case Read extends Permission(1 << 0)
  case Write extends Permission(1 << 1)
  case Delete extends Permission(1 << 2)
  case Admin extends Permission(Read.value | Write.value | Delete.value)

  // 实现 | 运算符，组合多个权限
  def |(other: Permission): Permission = Permission.fromValue(this.value | other.value)
  // 实现 & 运算符，取权限交集
  def &(other: Permission): Permission = Permission.fromValue(this.value & other.value)
  // 判断是否包含指定权限
  def contains(other: Permission): Boolean = (this.value & other.value) == other.value

object Permission:
  // 根据整数值反向解析枚举实例
  private val valueMap = values.map(p => p.value -> p).toMap
  def fromValue(value: Int): Permission = valueMap.getOrElse(value, None)

使用示例：

val userPerms = Permission.Read | Permission.Write
println(userPerms.contains(Permission.Read)) // 输出true
println(userPerms.contains(Permission.Delete)) // 输出false

方案2：Case Object + 特质封装 #

如果使用Scala 2，或偏好更灵活的结构，可以用特质+case object实现：

sealed trait Permission:
  def value: Int
  def |(other: Permission): Permission = Permission.Combined(this.value | other.value)
  def &(other: Permission): Permission = Permission.Combined(this.value & other.value)
  def contains(other: Permission): Boolean = (this.value & other.value) == other.value

object Permission:
  case object None extends Permission { val value = 0 }
  case object Read extends Permission { val value = 1 << 0 }
  case object Write extends Permission { val value = 1 << 1 }
  case object Delete extends Permission { val value = 1 << 2 }
  case object Admin extends Permission { val value = Read.value | Write.value | Delete.value }
  // 用于包装组合结果的类
  case class Combined(value: Int) extends Permission

使用逻辑和方案1完全一致，这种方式的优势是可以灵活扩展Combined类的逻辑，比如重写toString输出更友好的权限组合描述。

内容的提问来源于stack exchange，提问作者Zvi Mints