Iterator模式让代码更简化

struct MemTable {
char* buf;
size_t len;
};
 
struct DiskTable {
char* filename;
};

class DataReader {
public:
    int Init(const MemTable& memtable);
    int Init(const char* filename);
    int GetOneRecord(Record* record);
};

class DataMerger {
public:
    void Init(MemTable& m1, MemTable& m2);
    void Init(DiskTable& f1, DiskTable& f2);
    void Init(MemTable& m, MemTable& f);
    int DoMerge();
private:
    enum MergeType{MEM_TO_MEM, DISK_TO_DISK, FILE_TO_FILE};
    int MergeMemoryMemory();
    int MergeDiskDisk();
    int MergeMemoryDisk();
};

class Iterator {
public:
	virtual bool Valid() const = 0;
	virtual void SeekToFirst() const = 0;
	virtual void Next() = 0;
	virtual const Record& GetRecord() = 0;
};
 
class MemTable {
public:
        // 由MemTable生成Iterator，来做具体的数据读取工作
        // 去掉DataReader 类
	Iterator* NewIterator() const;
	// 其他的数据操作
	// Append Delete Find
private:
	// 这时我们已经不关注MemTable的具体实现方式了
	char* buf;
	size_t len;
};
 
class DiskTable {
public:
        // 由DiskTable管理具体的数据读取工作，去掉DataReader类
	Iterator* NewIterator() const;
	int Init(char* filename);
private:
	char* filename;
};
 
// 这里我们去掉了DataMerger类，数据的Merge操作全部完全不依赖于Table具体的实现方式
// 非常容易扩展，增加任意多的Table类型对Merge算法都没有影响
Iterator* NewMergingIterator(Iterator** list, int n);