You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-11-03 17:13:17 +03:00
128 lines
3.6 KiB
C++
128 lines
3.6 KiB
C++
/* 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_filereadthread.h
|
|
*
|
|
* Created on: Oct 25, 2011
|
|
* Author: bpaul
|
|
*/
|
|
|
|
#ifndef WE_FILEREADTHREAD_H_
|
|
#define WE_FILEREADTHREAD_H_
|
|
|
|
namespace WriteEngine
|
|
{
|
|
|
|
class WESDHandler;
|
|
class WEFileReadThread;
|
|
|
|
class WEReadThreadRunner
|
|
{
|
|
public:
|
|
WEReadThreadRunner(WEFileReadThread& Owner):fRef(Owner)
|
|
{ // ctor
|
|
}
|
|
~WEReadThreadRunner()
|
|
{
|
|
}
|
|
|
|
void operator()(); // Thread function
|
|
|
|
private:
|
|
WEFileReadThread& fRef;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
|
|
class WEFileReadThread
|
|
{
|
|
public:
|
|
WEFileReadThread(WESDHandler& aSdh);
|
|
virtual ~WEFileReadThread();
|
|
|
|
void reset();
|
|
void setup(std::string FileName);
|
|
void shutdown();
|
|
void feedData();
|
|
unsigned int readDataFile(messageqcpp::SBS& Sbs);
|
|
unsigned int readBinaryDataFile(messageqcpp::SBS& Sbs, unsigned int recLen);
|
|
void openInFile();
|
|
|
|
int getNextRow(std::istream& ifs, char*pBuf, int MaxLen);
|
|
|
|
boost::thread* getFpThread() const { return fpThread; }
|
|
bool isContinue() const { return fContinue; }
|
|
void setContinue(bool fContinue) { this->fContinue = fContinue; }
|
|
std::string getInFileName() const { return fInFileName; }
|
|
unsigned int getTgtPmId() const { return fTgtPmId; }
|
|
unsigned int getBatchQty() const { return fBatchQty; }
|
|
void setFpThread(boost::thread* fpThread) { this->fpThread = fpThread; }
|
|
void setInFileName(std::string fInFileName)
|
|
{
|
|
if((0==fInFileName.compare("STDIN"))||(0==fInFileName.compare("stdin")))
|
|
this->fInFileName = "/dev/stdin";
|
|
else
|
|
this->fInFileName = fInFileName;
|
|
}
|
|
//@BUG 4326
|
|
const std::istream& getInFile() const { return fInFile; }
|
|
void setBatchQty(unsigned int BatchQty) { fBatchQty = BatchQty; }
|
|
|
|
bool chkForListOfFiles(std::string& FileName);
|
|
std::string getNextInputDataFile();
|
|
void add2InputDataFileList(std::string& FileName);
|
|
|
|
private:
|
|
enum { MAXBUFFSIZE=1024*1024 };
|
|
|
|
// don't allow anyone else to set
|
|
void setTgtPmId(unsigned int fTgtPmId) { this->fTgtPmId = fTgtPmId; }
|
|
|
|
WESDHandler & fSdh;
|
|
boost::thread *fpThread;
|
|
boost::mutex fFileMutex;
|
|
bool fContinue;
|
|
std::string fInFileName;
|
|
std::istream fInFile; //@BUG 4326
|
|
std::ifstream fIfFile; //@BUG 4326
|
|
|
|
typedef std::list<std::string> strList;
|
|
strList fInfileList;
|
|
|
|
unsigned int fTgtPmId;
|
|
unsigned int fBatchQty;
|
|
bool fEnclEsc; // Encl/Esc char is set
|
|
char fEncl; // Encl char
|
|
char fEsc; // Esc char
|
|
char fDelim; // Column Delimit char
|
|
char fBuff[MAXBUFFSIZE]; // main data buffer
|
|
};
|
|
|
|
} /* namespace WriteEngine */
|
|
#endif /* WE_FILEREADTHREAD_H_ */
|