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 */