mariadb-columnstore-engine/writeengine/server/we_cleartablelockcmd.cpp

/* 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: we_cleartablelockcmd.cpp 4450 2013-01-21 14:13:24Z rdempsey $
 *
 ****************************************************************************/
#include "we_cleartablelockcmd.h"

#include <iostream>
#include <stdexcept>

#include "we_bulkrollbackmgr.h"
#include "we_confirmhdfsdbfile.h"

#include "IDBPolicy.h"

namespace WriteEngine
{

//------------------------------------------------------------------------------
// Process a bulk rollback request based on input from the specified input
// bytestream object.
//------------------------------------------------------------------------------
int WE_ClearTableLockCmd::processRollback(
	messageqcpp::ByteStream& bs,
	std::string& errMsg)
{
	uint8_t rc = 0;
	errMsg.clear();

	try {
		uint32_t    tableOID;
		uint64_t    tableLockID;
		std::string tableName;
		std::string appName;

		// May want to eventually comment out this logging to stdout,
		// but it shouldn't hurt to keep in here.
		std::cout << "ClearTableLockCmd::processRollback for " << fUserDesc;
		bs >> tableLockID;
		std::cout << ": tableLock-"<< tableLockID;

		bs >> tableOID;
		std::cout << "; tableOID-" << tableOID;

		bs >> tableName;
		std::cout << "; table-"    << tableName;

		bs >> appName;
		std::cout << "; app-"      << appName << std::endl;

		int we_rc = fWEWrapper.bulkRollback(
			tableOID,
			tableLockID,
			tableName,
			appName,
			false, // no extra debug logging to the console
			errMsg );

		if (we_rc != NO_ERROR)
			rc = 2;
	}
	catch(std::exception& ex)
	{
		std::cout << "ClearTableLockCmd::Rollback exception-" << ex.what() <<
			std::endl;
		errMsg = ex.what();
		rc     = 1;
	}

	return rc;
}

//------------------------------------------------------------------------------
// Process a bulk rollback cleanup request based on input from the specified
// input bytestream object.
// Optionally, an additional flag is appended to the message to indicate
// whether HDFS db file *.orig db files are to be cleaned up as well.
// We only need this for the "successful" case, as the bulk rollback takes
// care of db file cleanup in the "unsuccessful" case.
//------------------------------------------------------------------------------
int WE_ClearTableLockCmd::processCleanup(
	messageqcpp::ByteStream& bs,
	std::string& errMsg)
{
	uint8_t rc = 0;
	errMsg.clear();

	try {
		uint32_t tableOID;

		// May want to eventually comment out this logging to stdout,
		// but it shouldn't hurt to keep in here.
		std::cout << "ClearTableLockCmd::processCleanup for " << fUserDesc;
		bs >> tableOID;
		std::cout << ": tableOID-" << tableOID << std::endl;

		// On an HDFS system, this is where we delete any DB files
		// (ex: *.orig) that we no longer need.
		if ((idbdatafile::IDBPolicy::useHdfs()) &&
			(bs.length() >= sizeof(messageqcpp::ByteStream::byte)))
		{
			messageqcpp::ByteStream::byte deleteHdfsTempDbFiles;
			bs >> deleteHdfsTempDbFiles;
			if (deleteHdfsTempDbFiles)
			{
				std::string endDbErrMsg;
				ConfirmHdfsDbFile confirmHdfs;

                // We always pass "true", as this only applies to the "success-
                // ful" case.  See comments that precede this function.
				int endRc = confirmHdfs.endDbFileListFromMetaFile(
					tableOID, true, endDbErrMsg);

				// In this case, a deletion error is not fatal.
				// so we don't propagate a bad return code.  We
				// may want to add syslog msg (TBD)
				if (endRc != NO_ERROR)
					std::cout << "Orig db file deletion error: " <<
					endDbErrMsg << std::endl;
			}
		}

		BulkRollbackMgr::deleteMetaFile( tableOID );
	}
	catch(std::exception& ex)
	{
		std::cout << "ClearTableLockCmd::Cleanup exception-" << ex.what() <<
			std::endl;
		errMsg = ex.what();
		rc     = 1;
	}

	return rc;
}

}