mariadb-columnstore-engine/tools/cleartablelock/cleartablelockthread.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: cleartablelockthread.cpp 2101 2013-01-21 14:12:52Z rdempsey $
 *
 ****************************************************************************/
#include "cleartablelockthread.h"

#include "bytestream.h"
#include "we_messages.h"

#include <stdexcept>
#include <sstream>

/*static*/ boost::mutex ClearTableLockThread::fStdOutLock;

//------------------------------------------------------------------------------
// ClearTableLockThread constructor
//------------------------------------------------------------------------------
ClearTableLockThread::ClearTableLockThread(
	BRM::DBRM*                brm,
	messageqcpp::MessageQueueClient* clt,
	const BRM::TableLockInfo& tInfo,
	const std::string&        tblName,
	CLRTBLLOCK_MSGTYPE        msgType, 
	ClearTableLockStatus*     pStatus) :
		fTableLockInfo(tInfo),
		fBrm(brm),
		fClt(clt),
		fTblName(tblName),
		fMsgType(msgType),    
		fStatus(pStatus)
{
}

//------------------------------------------------------------------------------
// Main entry point for a ClearTableLockThread object, used to forward a
// cleartablelock tool command to the WriteEngineServer specified to the ctor.
//------------------------------------------------------------------------------
void ClearTableLockThread::operator() ()
{
	try {
		if (fMsgType == CLRTBLLOCK_MSGTYPE_ROLLBACK)
			executeRollback( );
		else if (fMsgType == CLRTBLLOCK_MSGTYPE_CLEANUP)
			executeFileCleanup( );
	}
	catch (std::exception& ex)
	{
		setStatus( 101, ex.what() );
	}
	catch (...)
	{
		std::string errMsg("Unknown exception.");
		setStatus( 102, errMsg );
	}
}

//------------------------------------------------------------------------------
// Process bulk rollback portion of cleartablelock command.
//------------------------------------------------------------------------------
void ClearTableLockThread::executeRollback( )
{
	boost::shared_ptr<messageqcpp::ByteStream> bsIn;
	messageqcpp::ByteStream bsOut;

	const std::string APPLNAME("cleartablelock command");

	//--------------------------------------------------------------------------
	// Send rollback msg to the writeengine server connected to fClt.
	//--------------------------------------------------------------------------
	{
		boost::mutex::scoped_lock lk(fStdOutLock);
		std::cout << "Sending rollback request to PM" <<
			fStatus->moduleID() << "..." << std::endl;
//		std::cout << "cleartablelock rollback: tableLock-" <<fTableLockInfo.id<<
//			": oid-"  << fTableLockInfo.tableOID <<
//			"; name-" << fTblName                <<
//			"; app-"  << APPLNAME                << std::endl;
	}

	bsOut << (messageqcpp::ByteStream::byte)
		WriteEngine::WE_CLT_SRV_CLEAR_TABLE_LOCK;
	bsOut << fTableLockInfo.id;
	bsOut << fTableLockInfo.tableOID;
	bsOut << fTblName;
	bsOut << APPLNAME;
	fClt->write(bsOut);

	// Wait for the response, and check for any errors
	std::string rollbackErrMsg;
	bsIn.reset(new messageqcpp::ByteStream());
	bsIn = fClt->read();
	if (bsIn->length() == 0)
	{
		std::string errMsg("Network error, PM rollback");
		setStatus( 103, errMsg );

		boost::mutex::scoped_lock lk(fStdOutLock);
		std::cout << "No response from PM" << fStatus->moduleID() << std::endl;
		return;
	}
	else
	{
		messageqcpp::ByteStream::byte rc;
		*bsIn >> rc;
		*bsIn >> rollbackErrMsg;

		{
			boost::mutex::scoped_lock lk(fStdOutLock);
			if (rc == 0)
				std::cout << "Successful rollback response from PM" <<
					fStatus->moduleID() << std::endl;
			else
				std::cout << "Unsuccessful rollback response from PM" <<
					fStatus->moduleID() << "; " << rollbackErrMsg << std::endl;
//			std::cout << "cleartablelock rollback response; rc-" << (int)rc <<
//				"; retMsg: <" << rollbackErrMsg << '>' << std::endl;
		}

		if (rc != 0)
		{
			std::string errMsg("PM rollback error: ");
			errMsg +=  rollbackErrMsg;
			setStatus( 104, errMsg );
			return;
		}

	}
}

//------------------------------------------------------------------------------
// Process file cleanup portion of cleartablelock command.
//------------------------------------------------------------------------------
void ClearTableLockThread::executeFileCleanup( )
{
	boost::shared_ptr<messageqcpp::ByteStream> bsIn;
	messageqcpp::ByteStream bsOut;

	//--------------------------------------------------------------------------
	// Send cleanup msg (to delete rb files) to the we server connected to fClt.
	//--------------------------------------------------------------------------
	{
		boost::mutex::scoped_lock lk(fStdOutLock);
		std::cout << "Sending cleanup request to PM" <<
			fStatus->moduleID() << "..." << std::endl;
//		std::cout  << "cleartablelock cleanup: " <<
//			"oid-" << fTableLockInfo.tableOID    << std::endl;
	}

	bsOut << (messageqcpp::ByteStream::byte)
		WriteEngine::WE_CLT_SRV_CLEAR_TABLE_LOCK_CLEANUP;
	bsOut << fTableLockInfo.tableOID;
	fClt->write(bsOut);

	// Wait for the response, and check for any errors
	std::string fileDeleteErrMsg;
	bsIn.reset(new messageqcpp::ByteStream());
	bsIn = fClt->read();
	if (bsIn->length() == 0)
	{
		std::string errMsg("Network error; PM rollback cleanup");
		setStatus( 105, errMsg );

		boost::mutex::scoped_lock lk(fStdOutLock);
		std::cout << "No response from PM" << fStatus->moduleID() << std::endl;
		return;
	}
	else
	{
		messageqcpp::ByteStream::byte rc;
		*bsIn >> rc;
		*bsIn >> fileDeleteErrMsg;

		{
			boost::mutex::scoped_lock lk(fStdOutLock);
			if (rc == 0)
				std::cout << "Successful cleanup response from PM" <<
					fStatus->moduleID() << std::endl;
			else
				std::cout << "Unsuccessful cleanup response from PM" <<
					fStatus->moduleID() << "; " << fileDeleteErrMsg <<std::endl;
//			std::cout << "cleartablelock cleanup response; rc-" << (int)rc <<
//				"; retMsg: <" << fileDeleteErrMsg << '>' << std::endl;
		}

		if (rc != 0)
		{
			std::string errMsg("PM rollback cleanup error: ");
			errMsg +=  fileDeleteErrMsg;
			setStatus( 106, errMsg );
			return;
		}
	}
}