难以重载运算符<<用于文件处理类

时间:2016-02-06 12:14:41

标签: c++ oop iostream raii

我必须:

  

使用带有字符串参数(文件名)的构造函数定义File_handle类,在构造函数中打开文件,然后在析构函数中将其关闭。

据我了解,这个类用于提供RAII,我正在尝试使用FILE*作为基本数据结构实现该类,其中我的目标主要是使FILE*成为智能指针:< / p>

fileHandler.h

// Class CFile_handler based on FILE*
class CFile_handler {
public:
    CFile_handler();                                           // default constructor   
    CFile_handler(const std::string& fileName,                 // constructor
                  const std::string& mode);     
    ~CFile_handler ();                                          // destructor

    // modifying member function
    void open_file(const std::string& fileName, 
                   const std::string& mode);

protected:
    typedef FILE* ptr;

private:
    CFile_handler(const CFile_handler&);                        // prevent copy creation
    CFile_handler& operator= (const CFile_handler&);            // prevent copy assignment

    ptr c_style_stream;                                         // data member
};

fileHandler.cpp

// Class CFile_handler member implementations
// default constuctor
CFile_handler::CFile_handler() {

}

// constructor
CFile_handler::CFile_handler(const std::string& fileName, const std::string& mode = "r")
    : c_style_stream( fopen( fileName.c_str(), mode.c_str() ) ) 
{

}

// destructor
CFile_handler::~CFile_handler() {
    if (c_style_stream) fclose(c_style_stream);
}

// Modifying member functions
void CFile_handler::open_file(const std::string& fileName, const std::string& mode) {
    c_style_stream = ( fopen( fileName.c_str(), mode.c_str() ) );
}

但是,我在重载I / O operators<< / >>时遇到了困难,因为我无法弄清楚如何实现其中任何一个。

如何重载operator<<以使该类与iostream对象一起使用?

编辑:

正如@LokiAstari提出的那样,继承istream并定义自己的streambuf会更好。

有人可以为处理streambuf的{​​{1}}提供示例或说明吗?

我想提供的是:

FILE*

或:

CFile_handler fh("filename.txt", "r");

std::string file_text;
fh >> file_text;

2 个答案:

答案 0 :(得分:1)

您的operator<<函数是将CFile_handler对象输出到C ++输出流,而不是将输出到 CFile_handler个对象。

要输出到CFile_handler对象,您有两个选择:

  1. 作为会员功能

    CFile_handler& CFile_handler::operator<<(int value)
    {
        // Output an integer to the contained file
        return *this;
    }
    
  2. 或者作为非成员函数,它将CFile_handler引用作为第一个参数:

    CFile_handler& operator<<(CFile_handler& file, int value)
    {
        // Output an integer to the file contained in `file`
        return file;
    }
    
  3. 对于上述两种变体,您都可以这样做。

    CFile_handler my_file(...);
    my_file << 1234;
    

答案 1 :(得分:1)

您可以使用std :: streambuf派生std :: streams的类型来处理FILE *

#include <iostream>
#include <stdio.h>

class OutputFilePointerStream: public std::ostream
{
    class OutputFilePointerStreamBuf: public std::streambuf
    {
        FILE*   buffer;
        public:
            OutputFilePointerStreamBuf(std::string const& fileName)
            {
                buffer  = fopen(fileName.c_str(), "w");
            }
            ~OutputFilePointerStreamBuf()
            {
                fclose(buffer);
            }
            virtual std::streamsize xsputn(const char* s, std::streamsize n) override
            {
                static char format[30];
                sprintf(format, "%%.%lds", n);
                fprintf(buffer, format, s);
                return n;
            }
    };
    OutputFilePointerStreamBuf       buffer;
    public:
        OutputFilePointerStream(std::string const& fileName)
            : std::ostream(nullptr)
            , buffer(fileName)
        {
            rdbuf(&buffer);
        }
};

int main()
{
    OutputFilePointerStream      fileStream("Test");

    fileStream << "Testing: " << 5 << "><\n";
    fileStream << "Line Again\n";
}