mariadb-columnstore-engine/writeengine/shared/we_bulkrollbackfilecompressedhdfs.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. */

/** @file
 * Class to restore compressed hdfs db files on behalf of BulkRollBackMgr.
 */

#ifndef WE_BULKROLLBACKFILECOMPRESSEDHDFS_H_
#define WE_BULKROLLBACKFILECOMPRESSEDHDFS_H_

#include <cstdio>
#include <cstring>

#include "we_define.h"
#include "we_type.h"
#include "we_bulkrollbackfile.h"

namespace WriteEngine
{
class BulkRollbackMgr;

//------------------------------------------------------------------------------
/** @brief Class used by BulkRollbackMgr to restore compressed hdfs db files.
 *
 *  BulkRollbackFileCompressed is used by non-hdfs system to restore a backup
 *  of the hwm compressed chunk.
 *  BulkRollbackFileCompressedHdfs is used by hdfs system to restore a backup
 *  of the entire hwm compressed db file.
 */
//------------------------------------------------------------------------------
class BulkRollbackFileCompressedHdfs : public BulkRollbackFile
{
public:

    /** @brief BulkRollbackFile constructor
     * @param mgr The controlling BulkRollbackMgr object.
     */
    BulkRollbackFileCompressedHdfs(BulkRollbackMgr* mgr);

    /** @brief BulkRollbackFile destructor
     */
    virtual     ~BulkRollbackFileCompressedHdfs();

    /** @brief Do we reinit trailing blocks in the HWM extent for the specified
     * segment file
     *
     * @param columnOID OID of the segment file in question
     * @param dbRoot DBRoot for the segment file in question
     * @param partNum Partition number for the segment file in question
     * @param segNum Segment number for the segment file in question
     */
    virtual bool doWeReInitExtent(OID columnOID,
                                  uint32_t  dbRoot,
                                  uint32_t  partNum,
                                  uint32_t  segNum) const;

    /** @brief Reinitialize the specified column segment file starting at
     * startOffsetBlk, and truncate trailing extents.
     * @param columnOID OID of the relevant segment file
     * @param dbRoot DBRoot of the relevant segment file
     * @param partNum Partition number of the relevant segment file
     * @param segNum Segment number of the relevant segment file
     * @param startOffsetBlk Starting block offset where file is to be
     *        reinitialized
     * @param nBlocks Number of blocks to be reinitialized
     * @param colType Column type of the relevant segment file
     * @param colWidth Width in bytes of column.
     * @param restoreHwmChk Restore HWM chunk
     */
    virtual void reInitTruncColumnExtent(OID columnOID,
                                         uint32_t  dbRoot,
                                         uint32_t  partNum,
                                         uint32_t  segNum,
                                         long long  startOffsetBlk,
                                         int        nBlocks,
                                         execplan::CalpontSystemCatalog::ColDataType colType,
                                         uint32_t  colWidth,
                                         bool       restoreHwmChk );

    /** @brief Reinitialize the specified dictionary store segment file starting
     * at startOffsetBlk, and truncate trailing extents.
     * @param columnOID OID of the relevant segment file
     * @param dbRoot DBRoot of the relevant segment file
     * @param partNum Partition number of the relevant segment file
     * @param segNum Segment number of the relevant segment file
     * @param startOffsetBlk Starting block offset where file is to be
     *        reinitialized
     * @param nBlocks Number of blocks to be reinitialized
     */
    virtual void reInitTruncDctnryExtent(OID columnOID,
                                         uint32_t  dbRoot,
                                         uint32_t  partNum,
                                         uint32_t  segNum,
                                         long long  startOffsetBlk,
                                         int        nBlocks );

    /** @brief Truncate the specified segment file to a specified num of bytes
     * @param columnOID OID of the relevant segment file
     * @param dbRoot DBRoot of the relevant segment file
     * @param partNum Partition number of the relevant segment file
     * @param segNum Segment number of the relevant segment file
     * @param fileSizeBlocks Number of blocks to retain in the file
     */
    virtual void truncateSegmentFile(OID columnOID,
                                     uint32_t  dbRoot,
                                     uint32_t  partNum,
                                     uint32_t  segNum,
                                     long long  filesSizeBlocks );

private:
    // Disable unnecessary copy constructor and assignment operator
    BulkRollbackFileCompressedHdfs(const BulkRollbackFileCompressedHdfs& rhs);
    BulkRollbackFileCompressedHdfs& operator=(
        const BulkRollbackFileCompressedHdfs& rhs);

    void restoreFromBackup(const char* colType,
                           OID        columnOID,
                           uint32_t  dbRoot,
                           uint32_t  partNum,
                           uint32_t  segNum );
};

} //end of namespace

#endif // WE_BULKROLLBACKFILECOMPRESSEDHDFS_H_