mariadb-columnstore-engine/writeengine/xml/we_xmlgenproc.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_xmlgenproc.h 4450 2013-01-21 14:13:24Z rdempsey $
 *
 ***********************************************************************/
/** @file */
#pragma once

#include <libxml/parser.h>
#include <libxml/xmlwriter.h>
#include <string>
#include <vector>

#include "calpontsystemcatalog.h"
#include "we_log.h"
#include "we_xmlgendata.h"

#define EXPORT

namespace WriteEngine
{
struct SysCatColumn
{
  execplan::CalpontSystemCatalog::OID oid;
  execplan::CalpontSystemCatalog::ColType colType;
  execplan::CalpontSystemCatalog::TableColName tableColName;
};
typedef std::vector<SysCatColumn> SysCatColumnList;

class XMLGenData;

/** @brief Generates job_xxx.xml file (for cpimport.bin) using XMLGenData as
 *  input.  Can also be used to generate an XML-formatted report.
 *
 *  This class was moved from XMLProc, formerly in tools/dbloadxml.  It was
 *  moved to writeengine/xml so that both colxml and cpimport.bin could use it.
 */
class XMLGenProc
{
 public:
  /** @brief XMLGenProc constructor
   *
   * @param mgr The input data used to generate a Job XML file.
   * @param bUseXmlLogFile Log info/errors to Job XML log file.
   * @param bSysCatRpt Generating SysCat report (true) or XML file (false)
   */
  EXPORT XMLGenProc(XMLGenData* mgr, bool bUseXmlLogFile, bool bSysCatRpt);
  EXPORT ~XMLGenProc();

  /** @brief start constructing XML file document.
   */
  EXPORT void startXMLFile();

  /** @brief Creates table tag for the specified table.
   *
   * @param table Name of table for which the table tag is to be generated.
   */
  EXPORT void makeTableData(const execplan::CalpontSystemCatalog::TableName& table,
                            const std::string& tableOIDStr);

  EXPORT void makeTableData(const execplan::CalpontSystemCatalog::TableName& table);

  /** @brief Creates column tags for the specified table.
   *
   * @param table Name of table for which the column tags are to be generated.
   * @return true means column tags created; else false is returned
   */
  EXPORT bool makeColumnData(const execplan::CalpontSystemCatalog::TableName& table);

  /** @brief Generate Job XML file name
   */
  EXPORT std::string genJobXMLFileName() const;

  /** @brief Write xml file document to the destination Job XML file.
   *
   * @param xmlFileName Name of XML file to be generated.
   */
  EXPORT void writeXMLFile(const std::string& xmlFileName);

  /** @brief log a message.
   *
   * @param msg The message to be logged to the error log file.
   */
  EXPORT void logErrorMessage(const std::string& msg);
  std::string errorString()
  {
    return fErrorString;
  }

  /** @brief set debug level
   */
  void setDebugLevel(int dbg)
  {
    fDebugLevel = dbg;
  }

 protected:
 private:
  XMLGenProc(const XMLGenProc&);             // disable default copy ctor
  XMLGenProc& operator=(const XMLGenProc&);  // disable default assignment
  void getColumnsForTable(const std::string& schema, const std::string& table, SysCatColumnList& colList);
  void sortColumnsByPosition(SysCatColumnList& columns);

  Log fLog;
  xmlDocPtr fDoc;
  xmlTextWriterPtr fWriter;
  std::string fErrorString;
  int fDebugLevel;
  XMLGenData* fInputMgr;  // Input data used to generate Job XML file
  bool fSysCatRpt;        // True colxml output or a syscat report
  bool fUseXmlLogFile;    // Log info/errors to Job XML log file
};

}  // namespace WriteEngine

#undef EXPORT