mariadb-columnstore-engine/writeengine/server/we_dmlcommandproc.h

/* 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_ddlcommandproc.h 3043 2011-08-29 22:03:03Z chao $
 *
 *******************************************************************************/
#pragma once

#include <unistd.h>
#include <boost/scoped_ptr.hpp>
#include "bytestream.h"
#include "we_messages.h"
#include "dbrm.h"
#include "we_message_handlers.h"
#include "calpontdmlpackage.h"
#include "updatedmlpackage.h"
#include "calpontsystemcatalog.h"
#include "insertdmlpackage.h"
#include "liboamcpp.h"
#include "dataconvert.h"
#include "writeengine.h"
#include "we_convertor.h"
#include "we_dbrootextenttracker.h"
#include "we_rbmetawriter.h"
#include "rowgroup.h"
#include "we_log.h"
#include "nullstring.h"

#define EXPORT

namespace WriteEngine
{
class WE_DMLCommandProc
{
 public:
  typedef std::vector<std::string> ColValues;

  EXPORT WE_DMLCommandProc();
  EXPORT WE_DMLCommandProc(const WE_DMLCommandProc& rhs);
  EXPORT ~WE_DMLCommandProc();
  inline void isFirstBatchPm(bool firstBatch)
  {
    fIsFirstBatchPm = firstBatch;
  }

  inline bool isFirstBatchPm()
  {
    return fIsFirstBatchPm;
  }

  // Convert rid from logical block rid to file relative rid
  inline void convertToRelativeRid(uint64_t& rid, const uint8_t extentNum, const uint16_t blockNum)
  {
    rid = rid + extentNum * extentRows + blockNum * 8192;
  }

  EXPORT uint8_t processSingleInsert(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t commitVersion(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t rollbackBlocks(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t rollbackVersion(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processBatchInsert(messageqcpp::ByteStream& bs, std::string& err,
                                    ByteStream::quadbyte& PMId);
  EXPORT uint8_t processBatchInsertBinary(messageqcpp::ByteStream& bs, std::string& err,
                                          ByteStream::quadbyte& PMId);
  EXPORT uint8_t commitBatchAutoOn(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t commitBatchAutoOff(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t rollbackBatchAutoOn(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t rollbackBatchAutoOff(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processBatchInsertHwm(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processUpdate(messageqcpp::ByteStream& bs, std::string& err, ByteStream::quadbyte& PMId,
                               uint64_t& blocksChanged);
  EXPORT uint8_t processUpdate1(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processFlushFiles(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processDelete(messageqcpp::ByteStream& bs, std::string& err, ByteStream::quadbyte& PMId,
                               uint64_t& blocksChanged);
  EXPORT uint8_t processRemoveMeta(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processBulkRollback(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t processBulkRollbackCleanup(messageqcpp::ByteStream& bs, std::string& err);
  EXPORT uint8_t updateSyscolumnNextval(ByteStream& bs, std::string& err);
  EXPORT uint8_t processPurgeFDCache(ByteStream& bs, std::string& err);
  EXPORT uint8_t processEndTransaction(ByteStream& bs, std::string& err);
  EXPORT uint8_t processFixRows(ByteStream& bs, std::string& err, ByteStream::quadbyte& PMId);
  EXPORT uint8_t getWrittenLbids(messageqcpp::ByteStream& bs, std::string& err, ByteStream::quadbyte& PMId);
  int validateColumnHWMs(execplan::CalpontSystemCatalog::RIDList& ridList,
                         boost::shared_ptr<execplan::CalpontSystemCatalog> systemCatalogPtr,
                         const std::vector<DBRootExtentInfo>& segFileInfo, const char* stage,
                         bool hasAuxCol);

 private:
  WriteEngineWrapper fWEWrapper;
  boost::scoped_ptr<RBMetaWriter> fRBMetaWriter;
  std::vector<boost::shared_ptr<DBRootExtentTracker> > dbRootExtTrackerVec;
  inline bool isDictCol(execplan::CalpontSystemCatalog::ColType& colType)
  {
    if (((colType.colDataType == execplan::CalpontSystemCatalog::CHAR) && (colType.colWidth > 8)) ||
        ((colType.colDataType == execplan::CalpontSystemCatalog::VARCHAR) && (colType.colWidth > 7)) ||
        ((colType.colDataType == execplan::CalpontSystemCatalog::DECIMAL) && (colType.precision > 38)) ||
        ((colType.colDataType == execplan::CalpontSystemCatalog::UDECIMAL) && (colType.precision > 38)) ||
        (colType.colDataType == execplan::CalpontSystemCatalog::VARBINARY) ||
        (colType.colDataType == execplan::CalpontSystemCatalog::BLOB) ||
        (colType.colDataType == execplan::CalpontSystemCatalog::TEXT))
    {
      return true;
    }
    else
      return false;
  }
  uint8_t processBatchInsertHwmFlushChunks(uint32_t tableOID, int txnID,
                                           const std::vector<BRM::FileInfo>& files,
                                           const std::vector<BRM::OID_t>& oidsToFlush, std::string& err);

  void processAuxCol(const std::vector<utils::NullString>& origVals,
                     WriteEngine::ColValueList& colValuesList,
                     WriteEngine::DictStrList& dicStringList);

  bool fIsFirstBatchPm;
  std::map<uint32_t, rowgroup::RowGroup*> rowGroups;
  std::map<uint32_t, dmlpackage::UpdateDMLPackage> cpackages;
  BRM::DBRM fDbrm;
  unsigned extentsPerSegmentFile, extentRows, filesPerColumnPartition, dbrootCnt;
  Log fLog;
  static const uint32_t DEFAULT_EXTENTS_PER_SEGMENT_FILE = 2;
};

}  // namespace WriteEngine
#undef EXPORT