如何编写cout的包装函数，通过命令行参数控制程序输出？

这里提供两种简洁的实现方案，单份代码即可完成需求，无需修改代码，仅通过命令行参数控制是否打印：

方案1：使用模板包装函数（C++17+） #

这种方式最直观，用全局开关配合模板折叠表达式实现类似cout的链式输出：

#include <iostream>
#include <string>

// 全局开关，控制是否启用输出
bool enable_output = true;

// 包装函数，模拟cout的链式调用逻辑
template <typename... Args>
void print(Args&&... args) {
    if (enable_output) {
        // C++17折叠表达式，依次输出所有传入参数
        (std::cout << ... << args);
    }
}

int main(int argc, char* argv[]) {
    // 解析命令行参数，检测静默模式标识
    for (int i = 1; i < argc; ++i) {
        std::string arg = argv[i];
        if (arg == "--silent" || arg == "--no-print") {
            enable_output = false;
            break;
        }
    }

    // 业务代码中用print替代cout即可
    print("程序启动\n");
    print("当前计算结果：", 42, "\n");
    print("程序结束\n");

    return 0;
}

使用方式 #

• 编译（需支持C++17）：g++ -std=c++17 main.cpp -o program
• 正常输出模式：./program
• 静默无输出模式：./program --silent

方案2：自定义输出流包装类 #

如果习惯原生cout的写法，可以自定义包装流，重载<<运算符实现兼容：

#include <iostream>
#include <string>

class ControlledOStream {
public:
    ControlledOStream(bool enabled) : enabled_(enabled) {}

    // 重载<<运算符，支持所有cout可处理的类型
    template <typename T>
    ControlledOStream& operator<<(const T& value) {
        if (enabled_) {
            std::cout << value;
        }
        return *this;
    }

    // 支持endl等流操纵符
    ControlledOStream& operator<<(std::ostream& (*manip)(std::ostream&)) {
        if (enabled_) {
            std::cout << manip;
        }
        return *this;
    }

private:
    bool enabled_;
};

int main(int argc, char* argv[]) {
    bool enable_output = true;
    for (int i = 1; i < argc; ++i) {
        std::string arg = argv[i];
        if (arg == "--silent" || arg == "--no-print") {
            enable_output = false;
            break;
        }
    }

    // 创建自定义流对象，直接替代原生cout
    ControlledOStream cout(enable_output);

    // 业务代码写法和原生cout完全一致
    cout << "程序启动" << std::endl;
    cout << "当前计算结果：" << 42 << std::endl;
    cout << "程序结束" << std::endl;

    return 0;
}

注意事项 #

• 可以根据需求修改命令行参数的标识（比如改成-q）
• 如果使用C++11/14版本，方案1的折叠表达式无法使用，可以改用递归模板或初始列表方式实现包装函数
• 若涉及多线程场景，全局开关可改为局部变量传递，或用单例模式管理状态保证线程安全

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