如何基于条件推导模板实参类型，实现可扩展的std::variant初始化

问题场景 #

现有类C持有std::variant<A, B>类型成员，需要实现一个模板构造函数，接收任意类型参数并按规则初始化variant：

• 若参数类型与B完全相同，则用参数构造B并初始化variant
• 否则若参数可构造A，则用参数构造A并初始化variant
• 要求未来扩展variant的类型时，无需大幅修改构造逻辑

原实现使用if constexpr分支，但会先默认初始化variant再赋值，希望改用模板元编程直接推导目标类型，通过初始化列表完成构造，避免额外初始化。

原代码示例：

struct A {
    // ...
};

struct B {
    // ...
};

class C {
public:
    template <typename T>
    C(T arg) {
        if constexpr (std::same_v<B, T>) {
            var = B{arg};
        }
        else if constexpr (std::is_constructible<A, T>) {
            var = A{arg};
        }
    }

private:
    std::variant<A, B> var;
};

解决方案 #

核心思路 #

实现一个模板元工具，按优先级从variant的备选类型列表中，匹配第一个满足条件的类型：

1. 优先匹配与参数类型完全相同的备选类型
2. 其次匹配可由参数构造的备选类型
3. 自动适配variant的类型扩展，无需手动修改推导逻辑

完整代码实现 #

#include <variant>
#include <type_traits>
#include <tuple>

struct A {
    // 示例构造函数，可根据实际需求调整
    A(int) {}
};

struct B {
    // 示例构造函数，可根据实际需求调整
    B(const char*) {}
};

// 辅助元编程工具：按优先级推导variant的目标类型
template <typename T, typename... Candidates>
struct VariantDeductor;

// 递归终止：无匹配类型时触发编译错误
template <typename T>
struct VariantDeductor<T> {
    static_assert(sizeof(T) == 0, "No matching type found for variant initialization");
};

// 按顺序检查候选类型：先匹配完全相同类型，再匹配可构造类型
template <typename T, typename Candidate, typename... Rest>
struct VariantDeductor<T, Candidate, Rest...> {
    using type = std::conditional_t<
        std::same_v<T, Candidate>,
        Candidate,
        typename std::conditional_t<
            std::is_constructible_v<Candidate, T>,
            std::type_identity<Candidate>,
            VariantDeductor<T, Rest...>
        >::type
    >;
};

class C {
private:
    // 定义variant类型，未来扩展仅需修改此处
    using VariantType = std::variant<A, B>;
    VariantType var;

    // 从variant中提取所有备选类型的工具
    template <typename V>
    struct ExtractVariantTypes;

    template <typename... Ts>
    struct ExtractVariantTypes<std::variant<Ts...>> {
        using types = std::tuple<Ts...>;
    };

public:
    // 对外暴露的类型推导模板
    template <typename T>
    struct DeduceAOrB {
        // 自动从VariantType中获取所有备选类型进行推导
        using type = typename VariantDeductor<
            T, 
            typename ExtractVariantTypes<VariantType>::types::value_type...
        >::type;
    };

    // 模板构造函数：直接推导目标类型并初始化variant
    template <typename T>
    C(T&& arg) : var(DeduceAOrB<std::decay_t<T>>::type{std::forward<T>(arg)}) {}
};

代码说明 #

1. VariantDeductor：递归遍历variant的备选类型，按优先级匹配目标类型。完全相同类型的优先级高于可构造类型，确保原逻辑的一致性。
2. ExtractVariantTypes：从std::variant中提取所有备选类型，避免硬编码类型列表，未来扩展variant时只需修改VariantType定义即可。
3. 构造函数优化：使用std::forward保持参数的值类别，通过初始化列表直接构造variant，避免默认初始化后的赋值操作，提升效率。
4. 错误处理：当无匹配类型时，静态断言会触发编译错误，便于排查问题。

扩展方式 #

若未来需要给variant添加新类型（如struct D），仅需修改C类中的VariantType定义：

using VariantType = std::variant<A, B, D>;

推导逻辑会自动按A→B→D的顺序检查匹配条件，无需修改其他代码。

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