the begginning

versioning/BRM/blockresolutionmanager.cpp

@@ -0,0 +1,190 @@
+/* Copyright (C) 2014 InfiniDB, Inc.
+
+   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; version 2 of
+   the License.
+
+   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., 51 Franklin Street, Fifth Floor, Boston,
+   MA 02110-1301, USA. */
+
+/*****************************************************************************
+ * $Id: blockresolutionmanager.cpp 1910 2013-06-18 15:19:15Z rdempsey $
+ *
+ ****************************************************************************/
+
+#include <iostream>
+#include <sys/types.h>
+#include <vector>
+#ifdef __linux__
+#include <values.h>
+#endif
+#include <limits>
+#include <sys/stat.h>
+
+#include "brmtypes.h"
+#include "rwlock.h"
+#include "mastersegmenttable.h"
+#include "extentmap.h"
+#include "copylocks.h"
+#include "vss.h"
+#include "vbbm.h"
+#include "exceptclasses.h"
+#include "slavecomm.h"
+#define BLOCKRESOLUTIONMANAGER_DLLEXPORT
+#include "blockresolutionmanager.h"
+#undef BLOCKRESOLUTIONMANAGER_DLLEXPORT
+#include "IDBDataFile.h"
+#include "IDBPolicy.h"
+
+using namespace idbdatafile;
+using namespace logging;
+using namespace std;
+
+namespace BRM {
+
+BlockResolutionManager::BlockResolutionManager(bool ronly) throw()
+{
+ 	if (ronly) {
+		em.setReadOnly();
+		vss.setReadOnly();
+		vbbm.setReadOnly();
+		copylocks.setReadOnly();
+ 	}
+}
+
+BlockResolutionManager::BlockResolutionManager(const BlockResolutionManager& brm)
+{
+	throw logic_error("BRM: Don't use the copy constructor.");
+}
+
+BlockResolutionManager::~BlockResolutionManager() throw()
+{
+}
+
+BlockResolutionManager& BlockResolutionManager::operator=(const BlockResolutionManager& brm)
+{
+	throw logic_error("BRM: Don't use the = operator.");
+}
+
+int BlockResolutionManager::loadExtentMap(const string &filename, bool fixFL)
+{
+	em.load(filename, fixFL);
+	return 0;
+}
+
+int BlockResolutionManager::saveExtentMap(const string &filename)
+{
+	em.save(filename);
+	return 0;
+}
+
+int BlockResolutionManager::saveState(string filename) throw()
+{
+	string emFilename = filename + "_em";
+	string vssFilename = filename + "_vss";
+	string vbbmFilename = filename + "_vbbm";
+	string journalFilename = filename + "_journal";
+
+	bool locked[2] = { false, false };
+
+	try {
+		vbbm.lock(VBBM::READ);
+		locked[0] = true;
+		vss.lock(VSS::READ);
+		locked[1] = true;
+
+		saveExtentMap(emFilename);
+
+		// truncate teh file if already exists since no truncate in HDFS.
+		const char* filename = journalFilename.c_str();
+		if (IDBPolicy::useHdfs()) {
+			IDBDataFile* journal = IDBDataFile::open(
+					IDBPolicy::getType(filename, IDBPolicy::WRITEENG), filename, "wb", 0);
+			delete journal;
+		}
+		else {
+			ofstream journal;
+			uint32_t utmp = ::umask(0);
+			journal.open(filename, ios_base::out | ios_base::trunc | ios_base::binary);
+			journal.close();
+			::umask(utmp);
+		}
+
+		vbbm.save(vbbmFilename);
+		vss.save(vssFilename);
+
+		vss.release(VSS::READ);
+		locked[1] = false;
+		vbbm.release(VBBM::READ);
+		locked[0] = false;
+	}
+	catch (exception &e) {
+		if (locked[1])
+			vss.release(VSS::READ);
+		if (locked[0])
+			vbbm.release(VBBM::READ);
+		cout << e.what() << endl;
+		return -1;
+	}
+
+	return 0;
+}
+
+int BlockResolutionManager::loadState(string filename, bool fixFL) throw()
+{
+	string emFilename = filename + "_em";
+	string vssFilename = filename + "_vss";
+	string vbbmFilename = filename + "_vbbm";
+	bool locked[2] = { false, false};
+
+	try {
+		vbbm.lock(VBBM::WRITE);
+		locked[0] = true;
+		vss.lock(VSS::WRITE);
+		locked[1] = true;
+
+		loadExtentMap(emFilename, fixFL);
+		vbbm.load(vbbmFilename);
+		vss.load(vssFilename);
+
+		vss.release(VSS::WRITE);
+		locked[1] = false;
+		vbbm.release(VBBM::WRITE);
+		locked[0] = false;
+	}
+	catch (exception &e) {
+		if (locked[1])
+			vss.release(VSS::WRITE);
+		if (locked[0])
+			vbbm.release(VBBM::WRITE);
+		cout << e.what() << endl;
+		return -1;
+	}
+	return 0;
+}
+
+int BlockResolutionManager::replayJournal(string prefix) throw()
+{
+	SlaveComm sc;
+	int err = -1;
+
+	try {
+		err = sc.replayJournal(prefix);
+	}
+	catch(exception &e) {
+		cout << e.what();
+	}
+	return err;
+}
+
+
+}   //namespace
+