mariadb-columnstore-engine/dbcon/execplan/pseudocolumn.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: pseudocolumn.cpp 9576 2013-05-29 21:02:11Z zzhu $
 *
 *
 ***********************************************************************/

#include <iostream>
#include <string>
#include <sstream>
using namespace std;
#include "basic/string_utils.h"

#include "bytestream.h"
#include "collation.h"

using namespace messageqcpp;

#include "objectreader.h"
#include "pseudocolumn.h"

namespace execplan
{
/**
 * Constructors/Destructors
 */
PseudoColumn::PseudoColumn() : SimpleColumn(), fPseudoType(PSEUDO_UNKNOWN)
{
}

PseudoColumn::PseudoColumn(const uint32_t pseudoType) : SimpleColumn(), fPseudoType(pseudoType)
{
}

PseudoColumn::PseudoColumn(const string& token, const uint32_t pseudoType, const uint32_t sessionID)
 : SimpleColumn(token, sessionID), fPseudoType(pseudoType)
{
  adjustResultType();
}

PseudoColumn::PseudoColumn(const string& schemaName, const string& tableName, const string& columnName,
                           const uint32_t pseudoType, const uint32_t sessionID)
 : SimpleColumn(schemaName, tableName, columnName, sessionID), fPseudoType(pseudoType)

{
}

PseudoColumn::PseudoColumn(const string& schemaName, const string& tableName, const string& columnName,
                           const bool isColumnStore, const uint32_t pseudoType, const uint32_t sessionID)
 : SimpleColumn(schemaName, tableName, columnName, isColumnStore, sessionID), fPseudoType(pseudoType)
{
  adjustResultType();
}

PseudoColumn::PseudoColumn(const SimpleColumn& rhs, const uint32_t pseudoType, const uint32_t sessionID)
 : SimpleColumn(rhs, sessionID), fPseudoType(pseudoType)
{
  adjustResultType();
}

PseudoColumn::PseudoColumn(const PseudoColumn& rhs, const uint32_t sessionID)
 : SimpleColumn(rhs, sessionID), fPseudoType(rhs.pseudoType())
{
  adjustResultType();
}

PseudoColumn::~PseudoColumn()
{
}

/**
 * Methods
 */

PseudoColumn& PseudoColumn::operator=(const PseudoColumn& rhs)
{
  if (this != &rhs)
  {
    fTableName = rhs.tableName();
    fColumnName = rhs.columnName();
    fOid = rhs.oid();
    fSchemaName = rhs.schemaName();
    fAlias = rhs.alias();
    fTableAlias = rhs.tableAlias();
    fAsc = rhs.asc();
    fIndexName = rhs.indexName();
    fViewName = rhs.viewName();
    fData = rhs.data();
    fSequence = rhs.sequence();
    fDistinct = rhs.distinct();
    fisColumnStore = rhs.isColumnStore();
    fPseudoType = rhs.pseudoType();
  }

  return *this;
}

ostream& operator<<(ostream& output, const PseudoColumn& rhs)
{
  output << rhs.toString();

  return output;
}

const string PseudoColumn::toString() const
{
  ostringstream output;
  output << "PseudoColumn " << data() << endl;

  output << "SimpleColumn " << data() << endl;
  output << "  s/t/c/v/o/ct/TA/CA/RA/#/card/join/source/engine: " << schemaName() << '/' << tableName() << '/'
         << columnName() << '/' << viewName() << '/' << oid() << '/'
         << colDataTypeToString(fResultType.colDataType) << '/' << tableAlias() << '/' << alias() << '/'
         << returnAll() << '/' << sequence() << '/' << cardinality() << '/' << joinInfo() << '/'
         << colSource() << '/' << (isColumnStore() ? "ColumnStore" : "ForeignEngine") << endl;

  output << "Pseudotype=" << fPseudoType << endl;
  return output.str();
}

string PseudoColumn::toCppCode(IncludeSet& includes) const
{
  includes.insert("pseudocolumn.h");
  stringstream ss;
  ss << "PseudoColumn(" << std::quoted(fSchemaName) << ", " << std::quoted(fTableName) << ", "
     << fisColumnStore << ", " << std::quoted(fColumnName) << ", " << fPseudoType << ", " << sessionID()
     << ")";

  return ss.str();
}

void PseudoColumn::serialize(messageqcpp::ByteStream& b) const
{
  b << (ObjectReader::id_t)ObjectReader::PSEUDOCOLUMN;
  SimpleColumn::serialize(b);
  b << static_cast<ByteStream::quadbyte>(fPseudoType);
}

void PseudoColumn::unserialize(messageqcpp::ByteStream& b)
{
  ObjectReader::checkType(b, ObjectReader::PSEUDOCOLUMN);
  SimpleColumn::unserialize(b);
  b >> reinterpret_cast<ByteStream::quadbyte&>(fPseudoType);
}

bool PseudoColumn::operator==(const PseudoColumn& t) const
{
  const SimpleColumn *rc1, *rc2;

  rc1 = static_cast<const SimpleColumn*>(this);
  rc2 = static_cast<const SimpleColumn*>(&t);

  if (*rc1 != *rc2)
    return false;

  if (fPseudoType != t.fPseudoType)
    return false;

  return true;
}

bool PseudoColumn::operator==(const TreeNode* t) const
{
  const PseudoColumn* pc;

  pc = dynamic_cast<const PseudoColumn*>(t);

  if (pc == NULL)
    return false;

  return *this == *pc;
}

bool PseudoColumn::operator!=(const PseudoColumn& t) const
{
  return !(*this == t);
}

bool PseudoColumn::operator!=(const TreeNode* t) const
{
  return !(*this == t);
}

uint32_t PseudoColumn::pseudoNameToType(string& name)
{
  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbpm"))
    return PSEUDO_PM;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbdbroot"))
    return PSEUDO_DBROOT;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbextentrelativerid"))
    return PSEUDO_EXTENTRELATIVERID;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbsegment"))
    return PSEUDO_SEGMENT;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbsegmentdir"))
    return PSEUDO_SEGMENTDIR;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbextentmin"))
    return PSEUDO_EXTENTMIN;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbextentmax"))
    return PSEUDO_EXTENTMAX;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbblockid"))
    return PSEUDO_BLOCKID;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbextentid"))
    return PSEUDO_EXTENTID;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idbpartition"))
    return PSEUDO_PARTITION;

  if (datatypes::ASCIIStringCaseInsensetiveEquals(name, "idblocalpm"))
    return PSEUDO_LOCALPM;

  return PSEUDO_UNKNOWN;
}

void PseudoColumn::adjustResultType()
{
  switch (fPseudoType)
  {
    case PSEUDO_EXTENTRELATIVERID:
    case PSEUDO_DBROOT:
    case PSEUDO_PM:
    case PSEUDO_SEGMENT:
    case PSEUDO_SEGMENTDIR:
    case PSEUDO_BLOCKID:
    case PSEUDO_EXTENTID:
    {
      fResultType.colDataType = CalpontSystemCatalog::BIGINT;
      fResultType.colWidth = 8;
      break;
    }

    case PSEUDO_PARTITION:
    {
      fResultType.colDataType = CalpontSystemCatalog::VARCHAR;
      fResultType.colWidth = 256;
      break;
    }

    default:
    {
      // same type of pseudocolumn arg
    }
  }
}

}  // namespace execplan