mariadb-columnstore-engine/dbcon/execplan/groupconcatcolumn.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. */

#include <sstream>
#include <cstring>

using namespace std;

#include "bytestream.h"
using namespace messageqcpp;

#include "rowgroup.h"
using namespace rowgroup;

#include "joblisttypes.h"
using namespace joblist;

#include "simplefilter.h"
#include "constantfilter.h"
#include "arithmeticcolumn.h"
#include "functioncolumn.h"
#include "objectreader.h"
#include "groupconcatcolumn.h"

namespace execplan
{
/**
 * Constructors/Destructors
 */
GroupConcatColumn::GroupConcatColumn(bool isJsonArrayAgg) : AggregateColumn(), fIsJsonArrayAgg(isJsonArrayAgg)
{
}

GroupConcatColumn::GroupConcatColumn(const uint32_t sessionID, bool isJsonArrayAgg)
 : AggregateColumn(sessionID), fIsJsonArrayAgg(isJsonArrayAgg)
{
}

GroupConcatColumn::GroupConcatColumn(const GroupConcatColumn& rhs, const uint32_t /*sessionID*/)
 : AggregateColumn(dynamic_cast<const AggregateColumn&>(rhs))
 , fOrderCols(rhs.fOrderCols)
 , fSeparator(rhs.fSeparator)
 , fIsJsonArrayAgg(rhs.fIsJsonArrayAgg)
{
}

/**
 * Methods
 */

const string GroupConcatColumn::toString() const
{
  ostringstream output;
  if (fIsJsonArrayAgg)
  {
    output << "JsonArrayAggColumn " << data() << endl;
    output << AggregateColumn::toString() << endl;
    output << "Json Array Order Columns: " << endl;
  }
  else
  {
    output << "GroupConcatColumn " << data() << endl;
    output << AggregateColumn::toString() << endl;
    output << "Group Concat Order Columns: " << endl;
  }

  for (uint32_t i = 0; i < fOrderCols.size(); i++)
  {
    output << *fOrderCols[i];
  }

  if (!fIsJsonArrayAgg)
    output << "\nSeparator: " << fSeparator << endl;
  return output.str();
}

string GroupConcatColumn::toCppCode(IncludeSet& includes) const
{
  includes.insert("groupconcatcolumn.h");
  stringstream ss;
  ss << "GroupConcatColumn(" << sessionID() << "," << std::boolalpha << fIsJsonArrayAgg << ")";

  return ss.str();
}

ostream& operator<<(ostream& output, const GroupConcatColumn& rhs)
{
  output << rhs.toString();
  return output;
}

void GroupConcatColumn::serialize(messageqcpp::ByteStream& b) const
{
  b << (uint8_t)ObjectReader::GROUPCONCATCOLUMN;
  AggregateColumn::serialize(b);

  b << static_cast<uint32_t>(fOrderCols.size());

  for (const auto& col : fOrderCols)
    col->serialize(b);

  b << fSeparator;
  b << (uint8_t)fIsJsonArrayAgg;
}

void GroupConcatColumn::unserialize(messageqcpp::ByteStream& b)
{
  ObjectReader::checkType(b, ObjectReader::GROUPCONCATCOLUMN);
  AggregateColumn::unserialize(b);
  fOrderCols.erase(fOrderCols.begin(), fOrderCols.end());

  uint32_t size, i;
  ReturnedColumn* rc;
  b >> size;

  for (i = 0; i < size; i++)
  {
    rc = dynamic_cast<ReturnedColumn*>(ObjectReader::createTreeNode(b));
    SRCP srcp(rc);
    fOrderCols.push_back(srcp);
  }

  b >> fSeparator;
  uint8_t tmp8;
  b >> tmp8;
  fIsJsonArrayAgg = tmp8;
}

bool GroupConcatColumn::operator==(const GroupConcatColumn& t) const
{
  const AggregateColumn *rc1, *rc2;

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

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

  for (uint32_t i = 0; i < fOrderCols.size(); i++)
  {
    if (fOrderCols[i].get() != NULL)
    {
      if (t.fOrderCols[i] == NULL)
        return false;

      if (*(fOrderCols[i].get()) != t.fOrderCols[i].get())
        return false;
    }
    else if (t.fOrderCols[i].get() != NULL)
      return false;
  }

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

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

  return true;
}

bool GroupConcatColumn::operator==(const TreeNode* t) const
{
  const GroupConcatColumn* ac;

  ac = dynamic_cast<const GroupConcatColumn*>(t);

  if (ac == NULL)
    return false;

  return *this == *ac;
}

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

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

}  // namespace execplan