pandorafms/extras/anytermd/libpbe/include/atomic_ofstream.hh
slerena 8594b9936c 012-05-05 Sancho Lerena <slerena@artica.es>
* anytermd: Added anyterm to extras. Included modifications on original
	anytermd project source code. Added a new spec file for centos/fedora/rhel.
	Tested on FC16/i386 and centos6/x86_84.




git-svn-id: https://svn.code.sf.net/p/pandora/code/trunk@6257 c3f86ba8-e40f-0410-aaad-9ba5e7f4b01f
2012-05-04 19:14:27 +00:00

93 lines
2.3 KiB
C++

// include/atomic_ofstream.hh
// This file is part of libpbe; see http://svn.chezphil.org/libpbe/
// (C) 2008 Philip Endecott
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#ifndef libpbe_atomic_ofstream_hh
#define libpbe_atomic_ofstream_hh
#include "Exception.hh"
#include <unistd.h>
#include <stdio.h>
#include <fstream>
#include <string>
namespace pbe {
// atomic_ofstream is like std::ofstream except that the stream writes to
// a temporary file. When the file has been completely written the client
// code should call commit() which removes any existing file and moves the
// new one into its place; this should happen somewhat atomically. If
// commit() has not been called by the time the destructor is reached
// the temporary file is removed and the original file is left unchanged.
struct atomic_ofstream_base {
const std::string orig_fn;
const std::string tmp_fn;
atomic_ofstream_base(std::string orig_fn_):
orig_fn(orig_fn_),
tmp_fn(orig_fn+".tmp")
{}
};
class atomic_ofstream:
private atomic_ofstream_base,
public std::ofstream {
bool committed;
public:
atomic_ofstream(std::string orig_fn_):
atomic_ofstream_base(orig_fn_),
std::ofstream(tmp_fn.c_str()),
committed(false)
{}
void commit() {
if (!committed) {
std::ofstream::close();
// We don't need to remove the old file; rename() does this.
int r = ::rename(tmp_fn.c_str(),orig_fn.c_str());
if (r==-1) {
throw_ErrnoException("rename()");
}
committed = true;
}
}
~atomic_ofstream() {
if (!committed) {
std::ofstream::close();
::unlink(tmp_fn.c_str());
}
}
};
};
#endif