使用std::variant在编译期根据参数类型确定返回值类型

0x00 | 演示代码

#include <variant>
#include <iostream>

std::variant<float, bool> get_result(std::variant<int, char> arg) {
    return std::visit([](auto&& value) -> std::variant<float, bool> {
        using T = std::decay_t<decltype(value)>;

        if constexpr (std::is_same_v<T, int>) {
            return 3.14f;
        } else if constexpr (std::is_same_v<T, char>) {
            return true;
        }
    }, arg);
}

int main() {
    auto result1 = get_result(42);
    std::cout << "Result for int input: " << std::get<float>(result1) << '\n';

    auto result2 = get_result('A');
    std::cout << "Result for char input: " << std::get<bool>(result2) << '\n';

    return 0;
}

get_result函数根据参数的类型在编译期确定了其返回值的类型：当参数为int时返回float，当参数为char时返回bool。

0x01 | 原理解读

这个函数完全是依靠std::variant和lambda实现的，前者类似于Union，以不同的类型解析自身内存，其大小由支持的类型中的最大者决定，后者则简化了代码，将std::variant直接交由std::visit，以访问其中的值。

由于lambda获取的参数value的类型无法直接确定（可能带有引用，数组，const等等），我们使用std::decay_t获取其底层类型（underlying type），即移除其可能存在的引用，数组，const等修饰，获得其底层的类型。

接着我们使用constexpr配合type_traits库所提供的std::is_same_v<T,U>在编译期判断参数类型并跳转到对应代码。

最后一路return，我们的函数返回了一个std::variant<float,bool>，接着我们用std::get<T>从中获取指定类型的数据，并输出它。

0x02 | 实际应用

最近我迷上了一种写法，为类的每个属性创建一个空struct，然后用constexpr和is_same在编译期确定其类型，再加上重载[]运算符...效果类似于这样：

// set
myobj[attrib] = val;
// get
std::cout << myobj[attrib] << std::endl;

这样很好，除了编译时间稍微变长了点外。

但是很明显，不可能所有属性都是同一个类型，遂有此文。