TempFile.hxx

Go to the documentation of this file.

 
#ifndef _OS_TEMPFILE_HXX_
#define _OS_TEMPFILE_HXX_
 
#include <string>
 
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include "utils/logging.h"
 
class TempDir {
public:
#ifndef __FreeRTOS__
  TempDir();
#endif
 
  ~TempDir() {
      if (::rmdir(dirName_.c_str()) != 0)
      {
          LOG(WARNING, "Error deleting temporary directory %s: %s",
              dirName_.c_str(), strerror(errno));
    }
  }
 
  static TempDir* instance() {
    static TempDir me;
    return &me;
  }
 
  const string& name() const {
    return dirName_;
  }
 
private:
  string dirName_;
};
 
class TempFile {
public:
  TempFile(const TempDir& dir, const string& basename);
 
  ~TempFile() {
    ::close(fd_);
    ::unlink(fileName_.c_str());
  }
 
  const string& name() const {
    return fileName_;
  }
 
  int fd()
  {
      return fd_;
  }
 
  void write(const uint8_t byte) {
    string s;
    s.push_back(byte);
    write(s);
  }
 
  void rewrite(const string& s);
    
  void write(const string& s) {
    size_t ofs = 0;
    while (ofs < s.size()) {
      int ret = ::write(fd_, s.data() + ofs, s.size() - ofs);
      HASSERT(ret >= 0);
      ofs += ret;
    }
#ifndef __WINNT__    
    fsync(fd_);
#endif    
  }
 
private:
  string fileName_;
  int fd_;
};
 
#endif