C++如何为多个文件编写一个ifstream？

Question

所以我有几个文件，它们形成了一个随机的行数，所以让我们称它们为 file1、file2、file3 等等。 我想要做的是创建一个 istream 类，它将所有文件作为一个流。 我得到的一件事不是对 std::istream 进行子类化，而是重新实现 streambuf。 我的问题是：我将如何解决这个问题 - 我如何能够从多个文件中读取而不将它们全部存储在内存中？

Answer 1

是否可以制作一个读取多个文件的istream ？

是的，但你必须自己做。 您可以通过扩展std::istream来实现您自己的istream类，然后实现它定义的方法：

class custom_istream : public std::istream {
    std::ifstream underlying; 
    //... implement methods
};

std::istream的接口足够灵活，可以让你做你想做的事。 但是，实现std::istream需要大量工作，因为您必须实现整个接口。

有更简单的解决方案吗？

如果您只需要std::istream提供的功能的一个子集，您可以编写自己的类。

例如，如果您只需要能够从文件中读取行，下面的类将适用于多个文件：

class MultiFileReader {
    std::ifstream filestream; 
    std::ios_base::openmode mode;  
    size_t file_index = 0; 
    std::vector<std::string> filenames; 
    void open_next_file() {
        file_index++; 
        filestream = std::ifstream(filenames.at(file_index), mode); 
    }
   public:
    MultiFileReader(std::vector<std::string> const& files, std::ios_base::openmode mode)
      : filestream(files[0], mode), mode(mode) {}
    // Checks if there's no more files to open, and no more to read
    // in the current file
    bool hasMoreToRead() {
        if(file_index == filenames.size()) return false;
        if(file_index + 1 == filenames.size()) 
            return not filestream.eof(); 
        return true;
    }
    std::string read_line() {
        if(not hasMoreToRead()) {
            throw std::logic_error("No more to read"); 
        }
        // If at the end of the file, open the next file
        if(filestream.eof()) {
            open_next_file(); 
        }
        else {
            std::string line; 
            std::getline(filestream, line);
            return line; 
        }
    }
};