mariadb-columnstore-engine/writeengine/wrapper/we_dctnrycompress.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_dctnrycompress.h 4726 2013-08-07 03:38:36Z bwilkinson $


/** @file */

#ifndef _WE_DCTNRY_COMPRESS_H_
#define _WE_DCTNRY_COMPRESS_H_

#include <stdlib.h>

#include "../dictionary/we_dctnry.h"
#include "we_chunkmanager.h"
#if defined(_MSC_VER) && defined(WRITEENGINE_DLLEXPORT)
#define EXPORT __declspec(dllexport)
#else
#define EXPORT
#endif

/** Namespace WriteEngine */
namespace WriteEngine
{

/** Class DctnryCompress */
class DctnryCompress0 : public Dctnry
{
public:
    /**
    * @brief Constructor
    */
    EXPORT DctnryCompress0();
    EXPORT DctnryCompress0(Log* logger);

    /**
    * @brief Default Destructor
    */
    EXPORT virtual ~DctnryCompress0();
};



/** Class DctnryCompress1 */
class DctnryCompress1 : public Dctnry
{
public:
    /**
    * @brief Constructor
    */
    EXPORT DctnryCompress1(Log* logger = 0);

    /**
    * @brief Default Destructor
    */
    EXPORT virtual ~DctnryCompress1();

    /**
    * @brief virtual method in FileOp
    */
    EXPORT int flushFile(int rc, std::map<FID, FID>& columnOids);

    /**
    * @brief virtual method in DBFileOp
    */
    EXPORT int readDBFile(IDBDataFile* pFile, unsigned char* readBuf, const uint64_t lbid,
                          const bool isFbo = false );

    /**
    * @brief virtual method in DBFileOp
    */
    EXPORT int writeDBFile(IDBDataFile* pFile, const unsigned char* writeBuf, const uint64_t lbid,
                           const int numOfBlock = 1);

    /**
    * @brief virtual method in DBFileOp
    */
    EXPORT int writeDBFileNoVBCache(IDBDataFile* pFile,
                                    const unsigned char* writeBuf, const int fbo,
                                    const int numOfBlock = 1);


    /**
    * @brief virtual method in Dctnry
    */
    IDBDataFile* createDctnryFile(const char* name, int width, const char* mode, int ioBuffSize);

    /**
    * @brief virtual method in Dctnry
    */
    IDBDataFile* openDctnryFile(bool useTmpSuffix);

    /**
    * @brief virtual method in Dctnry
    */
    void closeDctnryFile(bool doFlush, std::map<FID, FID>& columnOids);

    /**
    * @brief virtual method in Dctnry
    */
    int numOfBlocksInFile();

    /**
    * @brief For bulkload to use
    */
    void setMaxActiveChunkNum(unsigned int maxActiveChunkNum)
    {
        m_chunkManager->setMaxActiveChunkNum(maxActiveChunkNum);
    };
    void setBulkFlag(bool isBulkLoad)
    {
        m_chunkManager->setBulkFlag(isBulkLoad);
    };
    void setFixFlag(bool isFix)
    {
        m_chunkManager->setFixFlag(isFix);
    };
    int checkFixLastDictChunk ()
    {
        return m_chunkManager->checkFixLastDictChunk(m_dctnryOID, m_dbRoot, m_partition, m_segment);
    };
//   void chunkManager(ChunkManager* cm);

    /**
    * @brief virtual method in FileOp
    */
    void setTransId(const TxnID& transId)
    {
        Dctnry::setTransId(transId);

        if (m_chunkManager) m_chunkManager->setTransId(transId);
    }

    /**
    * @brief Set the IsInsert flag in the ChunkManager.
    * This forces flush at end of statement. Used only for bulk insert.
    */
    void setIsInsert(bool isInsert)
    {
        m_chunkManager->setIsInsert(isInsert);
    }

protected:

    /**
    * @brief virtual method in FileOp
    */
    int updateDctnryExtent(IDBDataFile* pFile, int nBlocks);

    /**
    * @brief convert lbid to fbo
    */
    int lbidToFbo(const uint64_t lbid, int& fbo);
};


} //end of namespace

#undef EXPORT

#endif // _WE_DCTNRY_COMPRESS_H_