mariadb-columnstore-engine/writeengine/server/we_brmrprtparser.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$
*
*******************************************************************************/

/*
 * we_brmrprtparser.cpp
 *
 *  Created on: Dec 13, 2011
 *      Author: bpaul
 */

#include <fstream>
#include <iostream>
using namespace std;

#include "we_brmrprtparser.h"

namespace WriteEngine
{

BrmReportParser::BrmReportParser(): fRptFile()
{

}

BrmReportParser::~BrmReportParser()
{

}

//------------------------------------------------------------------------------
// Serialize specified RptFileName into the ByteStream Bs.
// Serialization is done one record at a time.  File should be an ASCII file
// with newlines markers, because getline() is used.
// Function is limited to records that are no longer than 255 bytes.
// If a record is longer than 255 bytes, the function misbehaves.
//------------------------------------------------------------------------------
bool BrmReportParser::serialize(std::string RptFileName,
                                messageqcpp::ByteStream& Bs)
{
    try
    {
        fRptFile.open(RptFileName.c_str(), ifstream::in);
    }
    catch (std::exception& ex)
    {
        cout << "Failed to open BRMRptFile " << RptFileName << endl;
        cout << ex.what() << endl;
        throw runtime_error(ex.what());
    }

    if (fRptFile.good())
    {
        char aBuff[10240];
        unsigned int aLen = 0;

        while (fRptFile.good() && !fRptFile.eof())
        {
            fRptFile.getline(aBuff, sizeof(aBuff) - 1);
            aLen = fRptFile.gcount();

            if ((aLen != (sizeof(aBuff) - 1)) && (aLen > 0))
            {
                //aBuff[aLen-1] = '\n';
                //aBuff[aLen]=0;
                //cout << "Data Read " << aBuff <<endl;
                if (aBuff[0] != '#')	// do not serialize comments
                {
                    std::string strData = aBuff;
                    Bs << strData;
                }
            }
        }// while

        fRptFile.close();
        cout << "Closed File : " << RptFileName << " " << errno << endl;
    }
    else
    {
        cout << "Failed to open : " << RptFileName << " " << errno << endl;
        return false;
    }

    return true;
}

//------------------------------------------------------------------------------
// Serialize specified RptFileName into the ByteStream Bs.
// Serialization is done in 8192 byte blocks instead of by line or by record,
// so that this function will work for both character and binary files.
//------------------------------------------------------------------------------
bool BrmReportParser::serializeBlocks(std::string RptFileName,
                                      messageqcpp::ByteStream& Bs)
{
    try
    {
        fRptFile.open(RptFileName.c_str(), ifstream::in);
    }
    catch (std::exception& ex)
    {
        cout << "Failed to open Report File " << RptFileName << endl;
        cout << ex.what() << endl;
        return false;
    }

    if (fRptFile.good())
    {
        char aBuff[8192];
        unsigned int aLen = 0;
        std::string strBuff;

        while ( fRptFile.good() )
        {
            fRptFile.read(aBuff, sizeof(aBuff));
            aLen = fRptFile.gcount();

            if (aLen > 0)
            {
                strBuff.assign( aBuff, aLen );
                Bs << strBuff;
            }
        }

        fRptFile.close();
        cout << "Closed Report File : " << RptFileName << endl;
    }
    else
    {
        std::ostringstream oss;
        oss << "Failed to open Report File " << RptFileName << endl;
        cout << oss.str() << endl;
        return false;
    }

    return true;
}

void BrmReportParser::unserialize(messageqcpp::ByteStream& Bs)
{
    //TODO to be changed. left it here to understand how to implement
    /*
    ObjectReader::checkType(b, ObjectReader::SIMPLECOLUMN);
    ReturnedColumn::unserialize(b); // parent class unserialize
    b >> (uint32_t&) fOid;
    b >> fData;
    b >> reinterpret_cast<ByteStream::doublebyte&>(fReturnAll);
    b >> (uint32_t&) fSequence;
    */

    std::string aStrLine;

    while (Bs.length() > 0)
    {
        Bs >> aStrLine;
        cout << aStrLine;
    }

}





} /* namespace WriteEngine */