使用特定代码段将输出重定向到文件

std::cout为我们提供了一种非常方便的打印方式,使用起来也十分方便,易于扩展,并可全局访问。即使我们想打印对应的信息时,比如错误信息,我们可以使用错误输出std::cerr进行输出,其和cout的用法一样,只不过一个从标准通道进行输出,另一个从错误通道进行输出。

当我们要打印比较复杂的日志信息时。比如,要将函数的输出重定向到一个文件中,或者将函数的打印输出处于静默状态,而不需要对函数进行任何修改。或许这个函数为一个库函数,我们没有办法看到其源码。可能,这个函数并没有设计为写入到文件的函数,但是我们还是想将其输出输入到文件中。

这里可以重定向输出流对象的输出。本节中,我们将看到如何使用一种简单并且优雅地方式来完成输出流的重定向。

How to do it…

我们将实现一个辅助类,其能在构造和析构阶段,帮助我们完成流的重定向,以及对流的定向进行恢复。然后,我们来看其是怎么使用的:

  1. 包含必要的头文件,并声明所使用的命名空间:

    1. #include <iostream>
    2. #include <fstream>
    3. using namespace std;
  2. 我们实现了一个类,其具有一个文件输出流对象和一个指向流内部缓冲区的指针。cout作为流对象,其内部具有一个缓冲区,其可以用来进行数据交换,我们可以保存我们之前做过的事情,这样就很方便进行对后续修改的撤销。我们可以在C++手册中查询对其返回类型的解释,也可以使用decltypecout.rdbuf()所返回的类型进行查询。这并不是一个很好的体验,在我们的例子中,其就是一个指针类型:

    1. class redirect_cout_region
    2. {
    3. using buftype = decltype(cout.rdbuf());
    4. ofstream ofs;
    5. buftype buf_backup;
  3. 类的构造函数接受一个文件名字符串作为输入参数。这个字符串用来初始化文件流成员ofs。对其进行初始化后,可以将其输入到cout作为一个新的流缓冲区。rdbuf在接受一个新缓冲区的同时,会将旧缓冲区以指针的方式进行返回,这样当需要对缓冲区进行恢复时,就可以直接使用了:

    1. public:
    2. explicit
    3. redirect_cout_region (const string &filename)
    4. : ofs{filename}
    5. , buf_backup{cout.rdbuf(ofs.rdbuf())}
    6. {}
  4. 默认构造函数和其他构造函数做的事情几乎一样。其区别在于,默认构造函数不会打开任何文件。默认构造的文件流会直接替换cout的流缓冲,这样会导致cout的一些功能失效。其会丢弃一些要打印的东西。这在某些情况下是非常有用的:

    1. redirect_cout_region()
    2. : ofs{}
    3. ,buf_backup{cout.rdbuf(ofs.rdbuf())}
    4. {}
  5. 析构函数会对重定向进行恢复。当类在运行过程中超出了范围,可以使用原始的cout流缓冲区对其进行还原:

    1. ~redirect_cout_region() {
    2. cout.rdbuf(buf_backup);
    3. }
    4. };
  6. 让我们模拟一个有很多输出的函数:

    1. void my_output_heavy_function()
    2. {
    3. cout << "some output\n";
    4. cout << "this function does really heavy work\n";
    5. cout << "... and lots of it...\n";
    6. // ...
    7. }
  7. 主函数中,我们先会进行一次标准打印:

    1. int main()
    2. {
    3. cout << "Readable from normal stdout\n";
  8. 现在进行重定向,首先使用一个文本文件名对类进行实例化。文件流会使用读取和写入模式作为默认模式,所以其会创建一个文件。所以即便是后续使用cout进行打印,其输出将会重定向到这个文件中:

    1. {
    2. redirect_cout_region _ {"output.txt"};
    3. cout << "Only visible in output.txt\n";
    4. my_output_heavy_function();
    5. }
  9. 离开这段代码后,文件将会关闭,打印输出也会重归标准输出。我们再开启一个代码段,并使用默认构造函数对类进行构造。这样后续的打印信息将无法看到,都会被丢弃:

    1. {
    2. redirect_cout_region _;
    3. cout << "This output will "
    4. "completely vanish\n";
    5. }
  10. 离开这段代码后,我们的标准输出将再度恢复,并且将程序的最后一行打印出来:

    1. cout << "Readable from normal stdout again\n";
    2. }
  11. 编译并运行这个程序,其输出和我们期望的一致。我们只看到了第一行和最后一行输出:

    1. $ ./log_regions
    2. Readable from normal stdout
    3. Readable from normal stdout again
  12. 我们可以将新文件output.txt打开,其内容如下:

    1. $ cat output.txt
    2. Only visible in output.txt
    3. some output
    4. this function does really heavy work
    5. ... and lots of it...

How it works…

每个流对象都有一个内部缓冲区,这样的缓冲区可以进行交换。当我们有一个流对象s时,我们将其缓冲区存入到变量a中,并且为流对象换上了一个新的缓冲区b,这段代码就可以完成上述的过程:a = s.rdbuf(b)。需要恢复的时候只需要执行s.rdbuf(a)

这就如同我们在本节所做的。另一件很酷的事情是,可以将这些redirect_cout_region辅助函数放入堆栈中:

  1. {
  2. cout << "print to standard output\n";
  3. redirect_cout_region la {"a.txt"};
  4. cout << "print to a.txt\n";
  5. redirect_cout_region lb {"b.txt"};
  6. cout << "print to b.txt\n";
  7. }
  8. cout << "print to standard output again\n";

这也应该好理解,通常析构的顺序和构造的顺序是相反的。这种模式是将对象的构造和析构进行紧耦合,其也称作为资源获得即初始化(RAII)

这里有一个很重要的点需要注意——redirect_cout_region类中成员变量的初始化顺序:

  1. class redirect_cout_region {
  2. using buftype = decltype(cout.rdbuf());
  3. ofstream ofs;
  4. buftype buf_backup;
  5. public:
  6. explicit
  7. redirect_cout_region(const string &filename)
  8. : ofs{filename},
  9. buf_backup{cout.rdbuf(ofs.rdbuf())}
  10. {}
  11. ...

我们可以看到,成员buf_backup的初始化需要依赖成员ofs进行。有趣的是,这些成员初始化的顺序,不会按照初始化列表中给定元素的顺序进行初始化。这里初始化的顺序值与成员变量声明的顺序有关!

Note:

当一成员变量需要在另一个成员变量之后进行初始化,其需要在类声明的时候以相应的顺序进行声明。初始化列表中的顺序,对于构造函数来说没有任何影响。