如何扩展ifstream？实现带目录的ifstreamExt类及>>运算符重载

我希望通过定义ifstreamExt类来扩展ifstream，要求如下：

1. 拥有一个名为directory的静态字符串变量，用于存储文件打开路径；
2. 重写open方法，使其使用directory变量拼接文件路径；
3. >>运算符至少支持string类型。

我已完成前两点，但在实现第三点时遇到问题。我尝试过用getline，但它无法实现>>默认的跳过空白、读取连续非空白字符的行为——getline会读取整行而非单个子串。附上我的代码：

class ifstreamExt : ifstream {
  public:

    void open(string f) {
      string s = h + "/" + f;
      name = s;
      if (!is_open()) {
        cerr << "cannot open " << s << endl;
        exit(1);
      } else {
        cerr << "open file " << s << endl;
      }
    }
    friend ifstreamExt & operator >> (ifstreamExt & is, string &str) {
    }

    static void set_directory(string d) {
      cerr << "set home directory : " << d << endl;
      h = d;
      DIR* dir = opendir(h.c_str());
      if (dir) {
        cerr << "directory " << h << " exists " << endl;
      } else {
        cerr << "directory " << h << " does not exist " << endl;
        exit(1);
      }
    }
  private :
    static string h;
};

你的代码存在几个关键问题需要修正，再处理>>运算符的实现：

1. 修正继承方式 #

原代码中ifstreamExt是私有继承ifstream，这会导致基类的所有成员在外部无法访问，也无法正常使用流的核心功能。必须改成公有继承：

class ifstreamExt : public ifstream {

2. 修复open方法的逻辑 #

你的open方法没有实际调用基类的open函数，导致文件根本没有被打开，is_open()永远返回false。需要在拼接路径后调用ifstream::open：

void open(const string& f) {
  string s = h + "/" + f;
  // 调用基类的open方法打开文件
  ifstream::open(s);
  if (!is_open()) {
    cerr << "cannot open " << s << endl;
    exit(1);
  } else {
    cerr << "open file " << s << endl;
  }
}

另外，参数建议用const string&避免不必要的拷贝。

3. 实现>>运算符 #

你不需要自己重新实现跳过空白的逻辑，直接复用基类ifstream已经实现的operator>>即可——它本来就支持读取连续非空白的string。

方式一：直接依赖公有继承的特性 #

因为是公有继承，ifstreamExt已经继承了基类的operator>>，外部可以直接使用，不需要额外定义友元函数。比如：

ifstreamExt in;
in.set_directory("/path/to/dir");
in.open("test.txt");
string str;
in >> str; // 直接继承基类的operator>>，自动跳过空白读取子串

方式二：显式定义友元函数（如果需要扩展逻辑） #

如果之后需要在读取前后添加自定义逻辑，可以显式调用基类的运算符：

friend ifstreamExt& operator>>(ifstreamExt& is, string& str) {
  // 调用基类的operator>>处理string
  static_cast<ifstream&>(is) >> str;
  // 这里可以添加自定义逻辑，比如日志、数据校验等
  return is;
}

完整修正后的代码 #

#include <fstream>
#include <string>
#include <cerrno>
#include <cstdlib>
#include <dirent.h>

using namespace std;

class ifstreamExt : public ifstream {
public:
    void open(const string& f) {
        string s = h + "/" + f;
        ifstream::open(s);
        if (!is_open()) {
            cerr << "cannot open " << s << endl;
            exit(1);
        } else {
            cerr << "open file " << s << endl;
        }
    }

    // 可选：如果需要扩展operator>>才定义，否则直接用继承的即可
    friend ifstreamExt& operator>>(ifstreamExt& is, string& str) {
        static_cast<ifstream&>(is) >> str;
        return is;
    }

    static void set_directory(const string& d) {
        cerr << "set home directory : " << d << endl;
        h = d;
        DIR* dir = opendir(h.c_str());
        if (dir) {
            cerr << "directory " << h << " exists " << endl;
            closedir(dir); // 记得关闭目录句柄，避免资源泄漏
        } else {
            cerr << "directory " << h << " does not exist " << endl;
            exit(1);
        }
    }

private:
    static string h;
};

// 静态成员变量必须在类外初始化
string ifstreamExt::h;

补充说明 #

• 静态成员h需要在类外部初始化，否则会出现链接错误。
• opendir后记得调用closedir关闭目录句柄，避免资源泄漏。
• 公有继承后，你还可以直接使用基类的其他成员函数（比如close、eof等），不需要重复实现。

内容的提问来源于stack exchange，提问作者Pierre G.