Kotlin中如何限制枚举取值，实现原枚举的子集类型约束？

问题场景 #

开发中常定义包含全量取值的枚举类（典型如错误码、官方标准编码表），但不同业务逻辑下的函数仅允许传入该枚举的部分取值，核心诉求包括：

• 保留受限取值与全量枚举的关联，无需重复定义枚举值
• 既支持接收全量枚举的函数编写覆盖所有分支的穷尽式when模式匹配，也支持向这类函数传入仅包含部分取值的受限枚举类型参数

以全量错误码枚举定义为例：

enum class ApiError(val errorCode: Int) {
    INCORRECT_CHARACTER(1),
    MISSING_VALUE(2),
    TOO_SMALL(3),
    TOO_LARGE(4)
}

实际业务中，一类函数仅可能返回TOO_SMALL、TOO_LARGE两类错误，另一类函数仅可能返回INCORRECT_CHARACTER、MISSING_VALUE两类错误。直接通过子类化原枚举类收缩取值范围的方式在Kotlin中无法实现。枚举取值扩展方向的技术限制，和收缩取值范围的场景并不通用。

可行实现方案 #

方案1：密封接口标记枚举子集 #

这是Kotlin下最适配需求的实现方式，无需重复维护枚举值，所有校验在编译期完成，零额外运行时开销：

1. 为每类受限取值定义专属的密封接口作为类型标记

// 数值范围类错误子集，仅包含TOO_SMALL、TOO_LARGE
sealed interface RangeError
// 参数校验类错误子集，仅包含INCORRECT_CHARACTER、MISSING_VALUE
sealed interface ValidationError

2. 让全量枚举中归属对应子集的枚举值实现对应标记接口

enum class ApiError(val errorCode: Int) {
    INCORRECT_CHARACTER(1) : ApiError, ValidationError,
    MISSING_VALUE(2) : ApiError, ValidationError,
    TOO_SMALL(3) : ApiError, RangeError,
    TOO_LARGE(4) : ApiError, RangeError
}

3. 业务代码中直接将对应密封接口作为函数入参类型即可：编译器会自动校验传入值的范围，对受限类型做when匹配时会自动校验分支穷尽性，不需要写else分支；同时子集类型的实例本身就是ApiError的实例，可以直接传给接收全量ApiError的函数，无需额外转换。

代码示例：

// 仅允许传入范围类错误
fun handleRangeError(error: RangeError) {
    // 编译器要求穷尽两个分支，遗漏会直接报错
    when(error) {
        ApiError.TOO_SMALL -> println("数值小于允许最小值")
        ApiError.TOO_LARGE -> println("数值大于允许最大值")
    }
    // 直接传入接收全量错误的函数，无需转换
    handleAllError(error)
}

// 接收全量ApiError
fun handleAllError(error: ApiError) {
    when(error) {
        ApiError.INCORRECT_CHARACTER -> println("包含非法字符")
        ApiError.MISSING_VALUE -> println("必填值缺失")
        ApiError.TOO_SMALL -> println("数值小于允许最小值")
        ApiError.TOO_LARGE -> println("数值大于允许最大值")
    }
}

方案2：值类包装+构造约束 #

如果无法修改原枚举类的定义，可以使用JVM内联值类包装枚举值，通过私有构造函数收敛允许的取值范围：

// 范围类错误受限类型
@JvmInline
value class RangeErrorVal private constructor(val error: ApiError) {
    companion object {
        // 仅允许指定的两个枚举值构造实例
        fun from(error: ApiError): RangeErrorVal? = when(error) {
            ApiError.TOO_SMALL, ApiError.TOO_LARGE -> RangeErrorVal(error)
            else -> null
        }
        val TOO_SMALL get() = RangeErrorVal(ApiError.TOO_SMALL)
        val TOO_LARGE get() = RangeErrorVal(ApiError.TOO_LARGE)
    }
}

该方案的局限是无法自动获得when分支的穷尽性校验，需要自行覆盖所有可能取值，仅适合无法修改原枚举定义的场景。

对应类型论参考 #

该需求对应的类型论专业概念是精化类型（Refinement Type，也叫细化类型），属于子类型多态的分支方向：

• 精化类型是在原有基础类型之上，通过附加谓词约束收缩类型的合法取值范围，得到原类型的一个取值子集作为新类型
• 尝试通过子类化收缩枚举取值的思路，本质就是希望定义原枚举类型的精化子类型。上述密封接口实现方案，本质是通过声明式的子类型实现关系，在编译期固化了子集的取值约束，模拟了精化类型的效果
• 与扩展枚举取值（给原类型增加新的取值，属于代数数据类型的扩展，对应开放代数数据类型相关研究）的方向完全相反：枚举扩展是做类型的并集扩充，而收缩取值是做类型的子集约束，二者面临的技术限制没有关联。JVM层面枚举类默认被修饰为final不可继承，核心目的是保证枚举取值集合的封闭性以支持穷尽匹配，并不影响通过实现接口的方式定义子类型来划分取值子集。

内容的提问来源于stack exchange，提问作者Ulrich Schuster