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;
}

}  // namespace WriteEngine