使用指向 igzstreams 的 STL 向量读取 gzip 压缩文件

Question

作为输入，我有一个 gzip 压缩文件列表。如此处所示，我使用 gzstream 来处理它们。出于实际原因，我想打开每个文件并将每个流记录到一个向量中。这看起来很简单，但我无法让它发挥作用。这是最小的代码：

#include <cstdlib>

#include <iostream>
#include <vector>
using namespace std;

#include <gzstream.h>

int main (int argc, char ** argv)
{
  size_t i;
  vector<string> vInFiles;
  vector<igzstream *> vStreams;
  string line;

  // create the dummy input files
  system ("rm -f infile*.gz; for i in {1..2}; do echo \"toto\"${i} | gzip > infile${i}.gz; done");
  vInFiles.push_back ("infile1.gz");
  vInFiles.push_back ("infile2.gz");

  // open each input file
  for (i = 0; i < vInFiles.size(); ++i)
  {
    igzstream inStream;
    inStream.open (vInFiles[i].c_str());
    if (! inStream.good())
    {
      cerr << "ERROR: can't open file " << vInFiles[i] << endl;
      exit (1);
    }
    vStreams.push_back (&inStream);
  }

  // manipulate each input file
  for (i = 0; i < vInFiles.size(); ++i)
  {
    cout << "read first line of file " << vInFiles[i] << endl;
    getline (*(vStreams[i]), line);
    if (line.empty())
    {
      cerr << "empty line" << endl;
      exit (1);
    }
    cout << line << endl;
  }

  // close each input file
  for (i = 0; i < vInFiles.size(); ++i)
  {
    vStreams[i]->close();
  }
  vStreams.clear();

  return 0;
}

该代码可以正确编译：

$ gcc -Wall test.cpp -lstdc++ -lgzstream -lz

尽管它运行顺利，但它无法正确读取文件：

$ ./a.out
read first line of file infile1.gz
empty line                

Answer 1

迭代结束后，您的流指针无效，因为自动流对象随后被销毁。如果您确实需要，则需要在免费商店上分配它们（或使 igzstream 可移动）。

// std::vector<boost::shared_ptr<igzstream>> for C++03 
std::vector<std::unique_ptr<igzstream>> vStreams;

// ...

for (size_t i = 0; i < vInFiles.size(); ++i) {
    // boost::shared_ptr<igzstream> inStream = boost::make_shared<igzstream>();
    auto inStream = std::unique_ptr<igzstream>(new igzstream);
    inStream->open(...);
    // ...
    vStreams.push_back(inStream);
}

// ...

Answer 2

这已破了;您存储指向流的指针向量，但使用指向流的本地作用域自动实例的指针（在 for 循环内）对其进行初始化。一旦循环的每次迭代完成，该实例就超出了范围，并且您有一个指向一些废话的指针。

然后你稍后再使用这些垃圾，然后你就会得到垃圾。

使用智能指针，例如

  std::vector<boost::shared_ptr<igzstream> > vStreams;
  // to initialize
  for (i = 0; i < vInFiles.size(); ++i)
  {
    boost::shared_ptr<igzstream> inStream(new igzstream(vInFiles[i].c_str());
    if (!inStream->good())
    {
      cerr << "ERROR: can't open file " << vInFiles[i] << endl;
      exit (1);
    }
    vStreams.push_back (inStream); // save the smart pointer
  }

Answer 3

正如评论中提到的，我不想使用 Boost 并且我只有 gcc 4.1.2。因此，感谢 Cat Plus Plus 的建议，这是使用免费商店的解决方案：

  // open each input file
  for (i = 0; i < vInFiles.size(); ++i)
  {
    igzstream * pt_inStream = new igzstream;
    pt_inStream->open (vInFiles[i].c_str());
    if (! pt_inStream->good())
    {
      cerr << "ERROR: can't open file " << vInFiles[i] << endl;
      exit (1);
    }
    vStreams.push_back (pt_inStream);
  }

并且：

  // close each input file                                                                                                                                                                                                                                           
  for (i = 0; i < vInFiles.size(); ++i)
  {
    vStreams[i]->close();
    delete vStreams[i];
  }