如何为指定enum class启用标志位运算的模板运算符（Opt-in）

1. 核心思路 #

通过类型特征（Trait）+ Concept实现精准控制：先定义一个标记Trait，默认所有enum class都不启用标志位运算符；再用Concept约束运算符重载的适用范围，仅对主动标记的enum class生效，完全满足Opt-in需求。

2. 具体实现步骤 #

2.1 定义标记Trait与约束Concept #

在公共头文件（比如FlagOperators.h）中定义基础Trait和Concept：

#include <type_traits>

// 基础Trait：默认所有enum class都不是标志位枚举
template<typename T>
struct is_flag_enum : std::false_type {};

// 辅助变量模板，简化Trait调用
template<typename T>
constexpr bool is_flag_enum_v = is_flag_enum<T>::value;

// Concept：约束必须是带作用域的enum class，且被标记为标志位枚举
template<typename T>
concept FlagEnum = std::is_enum_v<T> && std::is_scoped_enum_v<T> && is_flag_enum_v<T>;

2.2 基于Concept重载标志位运算符 #

在同一个头文件中，为满足FlagEnum约束的类型重载所有需要的标志位运算符：

// 二元运算符：| & ^
template<FlagEnum T>
constexpr T operator|(T lhs, T rhs) noexcept {
    using Underlying = std::underlying_type_t<T>;
    return static_cast<T>(static_cast<Underlying>(lhs) | static_cast<Underlying>(rhs));
}

template<FlagEnum T>
constexpr T operator&(T lhs, T rhs) noexcept {
    using Underlying = std::underlying_type_t<T>;
    return static_cast<T>(static_cast<Underlying>(lhs) & static_cast<Underlying>(rhs));
}

template<FlagEnum T>
constexpr T operator^(T lhs, T rhs) noexcept {
    using Underlying = std::underlying_type_t<T>;
    return static_cast<T>(static_cast<Underlying>(lhs) ^ static_cast<Underlying>(rhs));
}

// 一元取反运算符：~
template<FlagEnum T>
constexpr T operator~(T rhs) noexcept {
    using Underlying = std::underlying_type_t<T>;
    return static_cast<T>(~static_cast<Underlying>(rhs));
}

// 复合赋值运算符：|= &= ^=
template<FlagEnum T>
constexpr T& operator|=(T& lhs, T rhs) noexcept {
    lhs = lhs | rhs;
    return lhs;
}

template<FlagEnum T>
constexpr T& operator&=(T& lhs, T rhs) noexcept {
    lhs = lhs & rhs;
    return lhs;
}

template<FlagEnum T>
constexpr T& operator^=(T& lhs, T rhs) noexcept {
    lhs = lhs ^ rhs;
    return lhs;
}

2.3 为目标enum class完成Opt-in配置 #

在你的MyFlags枚举定义的头文件中，通过特化is_flag_enumTrait来启用运算符：

// MyFlags.h
enum class MyFlags {
    None = 0,
    Read = 1 << 0,
    Write = 1 << 1,
    Execute = 1 << 2
};

// Opt-in核心：特化Trait，标记MyFlags为标志位枚举
template<>
struct is_flag_enum<MyFlags> : std::true_type {};

注：如果MyFlags在命名空间中，特化代码需要放在同一命名空间下，否则模板匹配会失败。

3. 使用示例 #

在业务代码中包含对应头文件后，即可像普通标志位一样使用MyFlags：

#include "MyFlags.h"
#include "FlagOperators.h"

int main() {
    auto flags = MyFlags::Read | MyFlags::Write;
    flags |= MyFlags::Execute;
    bool has_write_perm = (flags & MyFlags::Write) != MyFlags::None;
    auto inverted_flags = ~flags;
    return 0;
}

4. 关键注意事项 #

• 所有未特化is_flag_enum的enum class不会触发这些运算符重载，完全避免全局污染。
• 确保Trait特化与枚举定义的命名空间一致，否则模板无法匹配。
• 运算符重载仅依赖标准库，无需额外第三方依赖，兼容性强。

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