如何在 C++17 中使用 std::variant 处理多态数据？

作者

在 C++17 标准中，std::variant 成为一种强类型的联合体，能够安全地存储多种不同类型的值并在运行时动态地查询当前存储的类型。相比传统的空洞指针或 boost::variant，std::variant 提供了更好的类型安全性、易用的访问接口以及与现代 C++ 风格的完美兼容。下面我们通过一个完整的示例，演示如何使用 std::variant 来实现一个多态的数据容器，并演示访问、修改和组合操作。

1. 基本用法

#include <variant>
#include <string>
#include <iostream>

using MultiType = std::variant<int, double, std::string>;

void print(const MultiType& v)
{
    std::visit([](auto&& arg){
        std::cout << arg << '\n';
    }, v);
}

MultiType 可以容纳 int、double 或 std::string。使用 std::visit 可以根据当前存储的类型执行对应的操作。

2. 访问与转换

MultiType value = 42;
print(value);  // 输出 42

value = 3.14;
print(value);  // 输出 3.14

value = std::string{"hello"};
print(value);  // 输出 hello

要对 value 进行安全访问，可以使用 std::get、std::get_if 或 std::visit。

if (auto p = std::get_if<std::string>(&value)) {
    std::cout << "字符串: " << *p << '\n';
}

3. 访问错误处理

std::get 在类型不匹配时会抛出 std::bad_variant_access，因此如果不确定当前类型，最好先检查：

try {
    std::cout << std::get<int>(value) << '\n';
} catch (const std::bad_variant_access&) {
    std::cout << "当前不是 int 类型\n";
}

4. 组合使用

假设我们想存储一个可以是整数、浮点数或复杂对象的容器：

struct Complex { double real, imag; };
using Variant = std::variant<int, double, std::string, Complex>;

Variant v = Complex{1.0, 2.0};

std::visit([](auto&& arg){
    using T = std::decay_t<decltype(arg)>;
    if constexpr (std::is_same_v<T, Complex>) {
        std::cout << "Complex: " << arg.real << "+" << arg.imag << "i\n";
    } else {
        std::cout << arg << '\n';
    }
}, v);

5. 与模板函数结合

使用 std::variant 可以让模板函数更加灵活：

template <typename Variant>
void process(const Variant& v)
{
    std::visit([](auto&& arg){
        // 对所有类型都执行通用逻辑
        std::cout << "处理: " << arg << '\n';
    }, v);
}

int main()
{
    std::variant<int, std::string> v1 = 10;
    std::variant<double, std::string> v2 = "test";
    process(v1);
    process(v2);
}

6. 变体与结构体组合

如果需要在结构体里使用多种类型字段，可以直接使用 std::variant：

struct Record {
    std::string name;
    std::variant<int, double, std::string> value;
};

Record r1{"age", 30};
Record r2{"weight", 65.5};
Record r3{"nickname", std::string{"小明"}};

auto printRecord = [](const Record& r){
    std::cout << r.name << ": ";
    std::visit([](auto&& v){ std::cout << v; }, r.value);
    std::cout << '\n';
};

printRecord(r1);
printRecord(r2);
printRecord(r3);

7. 小结

  • 类型安全：std::variant 保证在编译期即可检查存储类型，避免了传统联合体的未定义行为。
  • 访问简洁：std::visit 与 lambda 组合可以一次性处理所有类型。
  • 与 STL 生态兼容：可以与 std::optional、std::any、std::tuple 等一起使用。
  • 适用场景：需要在运行时存储不同类型但同一接口的数据，如配置项、网络协议字段、解析器返回值等。

通过上述示例，你已经掌握了 std::variant 的基本使用方法，可以在自己的 C++ 项目中快速部署多态数据容器，实现更安全、更简洁的代码。祝你编码愉快！

发表评论