将宏转换为函数但无法用std::stringstream拼接输入的解决方案咨询

问题描述 #

我原本有一个宏MY_MACRO，用于检查布尔条件，当条件不满足时输出指定消息并调用std::abort()终止程序。调用该宏时，可直接传入类似"test (" << p << ")"这样的拼接表达式。但尝试将其替换为接收const std::stringstream&参数的静态内联函数后，原调用方式出现编译错误：

:39:30: error: invalid operands to binary expression ('const char[7]' and 'int')

请问有没有简单的实现方法？

核心原因 #

宏是预处理阶段展开的，"test (" << p << ")"会被直接插入到宏的代码中，与宏内的std::stringstream对象拼接后能正常编译。但换成函数后，"test (" << p << ")"是独立表达式——const char[]和int无法直接用<<操作，必须以stringstream对象作为左操作数，因此无法直接转换成const std::stringstream&。

简单实现方案 #

方案1：模板函数兼容原调用方式（推荐） #

通过C17折叠表达式或C11/14递归模板，自动构建字符串流，完全兼容原宏的调用语法：

C++17版本

#include <sstream>
#include <cstdlib>
#include <iostream>

template<typename... Args>
static inline void MY_CHECK(bool condition, Args&&... args) {
    if (!condition) {
        std::stringstream ss;
        (ss << ... << std::forward<Args>(args)); // 折叠表达式展开参数
        std::cerr << ss.str() << '\n';
        std::abort();
    }
}

// 调用示例
int main() {
    int p = 42;
    MY_CHECK(p == 0, "test (" << p << ")"); // 和原宏调用方式一致
    return 0;
}

C++11/14兼容版本

#include <sstream>
#include <cstdlib>
#include <iostream>

// 递归终止函数
static inline void stream_args(std::stringstream&) {}

// 递归展开参数并写入流
template<typename T, typename... Args>
static inline void stream_args(std::stringstream& ss, T&& val, Args&&... args) {
    ss << std::forward<T>(val);
    stream_args(ss, std::forward<Args>(args)...);
}

template<typename... Args>
static inline void MY_CHECK(bool condition, Args&&... args) {
    if (!condition) {
        std::stringstream ss;
        stream_args(ss, std::forward<Args>(args)...);
        std::cerr << ss.str() << '\n';
        std::abort();
    }
}

// 调用示例
int main() {
    int p = 42;
    MY_CHECK(p == 0, "test (" << p << ")");
    return 0;
}

方案2：Lambda延迟构建消息 #

如果不想使用模板，可让函数接收一个返回字符串的lambda，将拼接逻辑放在lambda内部：

#include <sstream>
#include <cstdlib>
#include <iostream>
#include <functional>

static inline void MY_CHECK(bool condition, const std::function<std::string()>& msg_provider) {
    if (!condition) {
        std::cerr << msg_provider() << '\n';
        std::abort();
    }
}

// 调用示例
int main() {
    int p = 42;
    MY_CHECK(p == 0, [&](){
        std::stringstream ss;
        ss << "test (" << p << ")";
        return ss.str();
    });
    return 0;
}

这种方式需要修改调用代码，但逻辑清晰，无需依赖高阶模板特性。

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