重定向和时间戳std :: cout

我需要将一个ofstream重定向到一个文件并为每一行打印时间戳。（这是测井系统的一部分）。重定向和时间戳std :: cout

我有一个工作类，它可以做到这一点，但它拒绝在std :: endl被刷新时刷新文件。我会对此提供任何帮助。（如果有更简单的方法可以做到这一点）。

#include <iostream> 
#include <streambuf> 
#include <fstream> 
#include <sys/time.h> 
#include <cstring> 
#include <memory> 

class TimeStampBuf: public std::streambuf { 
public: 
    explicit TimeStampBuf(std::streambuf* dest) : 
      _dest(dest), 
      _isAtStartOfLine(true), 
      _owner(NULL) { 
    } 
    explicit TimeStampBuf(std::ostream& dest) : 
      _dest(dest.rdbuf()), 
      _isAtStartOfLine(true), 
      _owner(&dest) { 
     _owner->rdbuf(this); 
    } 
    virtual ~TimeStampBuf() { 
     if (_owner != NULL) { 
      _owner->rdbuf(_dest); 
     } 
    } 
protected: 
    virtual int overflow(int ch) { 
     if (_isAtStartOfLine) { 
      char timebuff[30]; 
      timeval curTime; 
      gettimeofday(&curTime, NULL); 
      strftime(timebuff, sizeof(timebuff), "%Y-%m-%d %H:%M:%S:", 
        localtime(&curTime.tv_sec)); 
      sprintf(timebuff + strlen(timebuff), "%03u\t", 
        (unsigned int) curTime.tv_usec/1000); 
      _dest->sputn(timebuff, strlen(timebuff)); 
     } 
     _isAtStartOfLine = ch == '\n'; 
     return _dest->sputc(ch); 
    } 
private: 
    std::streambuf *_dest; 
    bool _isAtStartOfLine; 
    std::ostream *_owner; 
}; 
class OutputRedirectAndStamp { 
public: 
    OutputRedirectAndStamp(std::string file, std::ostream &s = std::cout, std::ios::openmode mode = std::ios::out){ 
     _s=&s; 
     _file=file; 
     if(_file.size()){ 
      _mode=mode; 
      _buf.open(file.c_str(),mode); 
      _coutBuf = s.rdbuf((std::streambuf*)&_buf); 
     } 
     _tsb.reset(new TimeStampBuf(s)); 
    } 
    void reopen(void){ 
     _tsb.reset(); 
     if(_file.size()){ 
      _s->rdbuf(_coutBuf); //reset to previous output 
      _buf.close(); 
      _buf.open(_file.c_str(),_mode); 
      _coutBuf = _s->rdbuf((std::streambuf*)&_buf); 
     } 
     _tsb.reset(new TimeStampBuf(*_s)); 
    } 
    ~OutputRedirectAndStamp() { 
     _tsb.reset(); 
     if(_file.size()){ 
      _s->rdbuf(_coutBuf); //reset to previous output 
     } 
    } 
private: 
    std::string _file; 
    std::ios::openmode _mode; 
    std::ostream *_s; 
    std::filebuf _buf; 
    std::streambuf *_coutBuf; 
    std::unique_ptr<TimeStampBuf> _tsb; 
}; 
int main() //example main 
{ 
    std::unique_ptr<OutputRedirectAndStamp> a; 
    a.reset(new OutputRedirectAndStamp("test.txt",std::cout,std::ios::app | std::ios::out)); 

    std::cout<<"this is written to file"<<2<<std::endl; 
    a->reopen(); 
    std::cout<<"this is written to file also"<<std::endl; 
    a.reset(); 
    std::cout<<"this is written to stdout"<<std::endl; 
    return 0; 
}

来源

2013-10-10 user1783395

你可能会得到更多的帮助，如果你能减少你的代码，只是需要什么重现该问题。尽可能地尽可能地切断，这将使我们更容易帮助您找到解决方案。 –

如果Boost.Iostreams可供您使用，它会严重减少您的工作量。 – GManNickG

我之前用过boost。我会看看Iostreams库，但不幸的是它不适合这个特定的项目。谢谢你的消化。 – user1783395

当你刷新的std::ostream，流缓冲的pubsync()被称为这反过来又调用virtual功能sync()。如果您不覆盖sync()它只会声称它成功返回0。从你sync()覆盖你应该叫召开流缓冲区的pubsync()：

int TimeStampBuf::sync() { 
    return _dest->pubsync(); 
}

来源

2013-10-10 21:45:48

这完全解决了这个问题。谢谢！ – user1783395

重定向和时间戳std :: cout

回答

相关问题