mariadb-columnstore-engine/primitives/primproc/command.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 <unistd.h>

#include "bpp.h"
#include <typeinfo>

using namespace std;
using namespace messageqcpp;

namespace primitiveprocessor
{

Command::Command(CommandType c) : cmdType(c), fFilterFeeder(NOT_FEEDER) { }

Command::~Command() { };

void Command::createCommand(ByteStream& bs)
{
    bs >> OID;
    bs >> tupleKey;
    bs >> queryUuid;
    bs >> stepUuid;
}

void Command::resetCommand(ByteStream& bs) { };

void Command::setMakeAbsRids(bool) { }

Command* Command::makeCommand(ByteStream& bs, CommandType* type, vector<SCommand>& cmds)
{
    Command* ret;
    uint8_t tmp8;

    bs.peek(tmp8);
    *type = (CommandType) tmp8;

    switch (*type)
    {
        case COLUMN_COMMAND:
            ret = new ColumnCommand();
            break;

        case DICT_STEP:
            ret = new DictStep();
            break;

        case PASS_THRU:
            ret = new PassThruCommand();
            break;

        case RID_TO_STRING:
            ret = new RTSCommand();
            break;

        case FILTER_COMMAND:
            ret = FilterCommand::makeFilterCommand(bs, cmds);
            break;

        case PSEUDOCOLUMN:
            ret = new PseudoCC();
            break;

        default:
            throw logic_error("Command::makeCommand(): can't deserialize this bytestream");
    };

    ret->createCommand(bs);

    return ret;
}

void Command::setBatchPrimitiveProcessor(BatchPrimitiveProcessor* b)
{
    bpp = b;
}

void Command::copyRidsForFilterCmd()
{
    if (fFilterFeeder == LEFT_FEEDER)
    {
        bpp->fFiltRidCount[0] = bpp->ridCount;

        for (uint64_t i = 0; i < bpp->ridCount; i++)
            bpp->fFiltCmdRids[0][i] = bpp->relRids[i];
    }
    else // if (fFilterFeeder == RIGHT_FEEDER)
    {
        bpp->fFiltRidCount[1] = bpp->ridCount;

        for (uint64_t i = 0; i < bpp->ridCount; i++)
            bpp->fFiltCmdRids[1][i] = bpp->relRids[i];
    }
}

bool Command::operator==(const Command& c) const
{
    const type_info& cType = typeid(c);

    if (cType != typeid(*this))
        return false;

    if (cType == typeid(ColumnCommand))
    {
        const ColumnCommand* cc = dynamic_cast<const ColumnCommand*>(&c);
        const ColumnCommand* t = dynamic_cast<const ColumnCommand*>(this);

        if (*cc != *t)
            return false;
    }
    else if (cType == typeid(DictStep))
    {
        const DictStep* ds = dynamic_cast<const DictStep*>(&c);
        const DictStep* t = dynamic_cast<const DictStep*>(this);

        if (*ds != *t)
            return false;
    }
    else if (cType == typeid(PassThruCommand))
    {
        const PassThruCommand* pt = dynamic_cast<const PassThruCommand*>(&c);
        const PassThruCommand* t = dynamic_cast<const PassThruCommand*>(this);

        if (*pt != *t)
            return false;
    }
    else if (cType == typeid(RTSCommand))
    {
        const RTSCommand* rts = dynamic_cast<const RTSCommand*>(&c);
        const RTSCommand* t = dynamic_cast<const RTSCommand*>(this);

        if (*rts != *t)
            return false;
    }
    else if (cType == typeid(FilterCommand))
    {
        const FilterCommand* fc = dynamic_cast<const FilterCommand*>(&c);
        const FilterCommand* t = dynamic_cast<const FilterCommand*>(this);

        if (*fc != *t)
            return false;
    }
    else
        cerr << "unknown Command type\n";

    return true;
}

void Command::duplicate(Command* c)
{
    bpp = c->bpp;
    cmdType = c->cmdType;
    fFilterFeeder = c->fFilterFeeder;
    OID = c->OID;
    tupleKey = c->tupleKey;
    queryUuid = c->queryUuid;
    stepUuid = c->stepUuid;
}

}