mariadb-columnstore-engine/writeengine/server/we_readthread.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_readthread.h 4609 2013-04-19 15:32:02Z chao $
 *
 *******************************************************************************/
#pragma once

#include <unistd.h>
#include <boost/shared_ptr.hpp>
#include "messagequeue.h"
#include "threadpool.h"
#include "we_ddlcommandproc.h"

#include "we_ddlcommandproc.h"
#include "we_dmlcommandproc.h"
#include "we_cleartablelockcmd.h"
#include "we_dataloader.h"
#include "we_getfilesizes.h"
#include "dbrm.h"

namespace WriteEngine
{
class ReadThread
{
 public:
  explicit ReadThread(const messageqcpp::IOSocket& ios);
  virtual ~ReadThread();

  virtual void operator()();

  // protected:
 public:
  messageqcpp::IOSocket fIos;
  messageqcpp::ByteStream fIbs;

 private:
  // defaults okay
  // ReadThread(const ReadThread& rhs);
  // ReadThread& operator=(const ReadThread& rhs);
};

class DmlReadThread : public ReadThread
{
 public:
  explicit DmlReadThread(const messageqcpp::IOSocket& ios, ByteStream& ibs);
  virtual ~DmlReadThread();

  virtual void operator()();

 private:
  boost::shared_ptr<WE_DMLCommandProc> fWeDMLprocessor;
  boost::shared_ptr<WE_DDLCommandProc> fWeDDLprocessor;
};

class SplitterReadThread : public ReadThread
{
 public:
  SplitterReadThread(const messageqcpp::IOSocket& ios, ByteStream& Ibs);
  SplitterReadThread(const SplitterReadThread& rhs);
  virtual ~SplitterReadThread();

  virtual void operator()();

 private:
  // WEDataLoader* fpWeDataLoader;
  WEDataLoader fWeDataLoader;

  friend class ReadThreadFactory;
  // friend class WEDataLoader;
};

//------------------------------------------------------------------------------
// Thread object that receives a cleartablelock tool command.
//------------------------------------------------------------------------------
class ClearTableLockReadThread : public ReadThread
{
 public:
  ClearTableLockReadThread(const messageqcpp::IOSocket& ios, ByteStream& ibs);
  virtual ~ClearTableLockReadThread();

  virtual void operator()();

 private:
  boost::shared_ptr<WE_ClearTableLockCmd> fWeClearTableLockCmd;
};

//------------------------------------------------------------------------------
// Thread object that receives a redistributeDB tool command.
//------------------------------------------------------------------------------
class RedistributeReadThread : public ReadThread
{
 public:
  RedistributeReadThread(const messageqcpp::IOSocket& ios, ByteStream& ibs);
  virtual ~RedistributeReadThread();

  virtual void operator()();

 private:
};

//------------------------------------------------------------------------------
// Thread object that get file sizes for a table.
//------------------------------------------------------------------------------
class GetFileSizeThread : public ReadThread
{
 public:
  GetFileSizeThread(const messageqcpp::IOSocket& ios, ByteStream& ibs, BRM::DBRM& dbrm);
  virtual ~GetFileSizeThread();

  virtual void operator()();

 private:
  boost::shared_ptr<WE_GetFileSizes> fWeGetFileSizes;
  int key;
};

class ReadThreadFactory
{
 public:
  ReadThreadFactory()
  {
  }
  virtual ~ReadThreadFactory()
  {
  }

 public:
  static void CreateReadThread(threadpool::ThreadPool& Tp, IOSocket& ios, BRM::DBRM& dbrm);
};

}  // namespace WriteEngine