~drizzle-trunk/drizzle/development

/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2010 Andrew Hutchings

 *

 *  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 St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include "drizzledump_data.h"

#include "drizzledump_drizzle.h"

#include "client_priv.h"

#include <string>

#include <iostream>

#include <drizzled/gettext.h>

#include <boost/lexical_cast.hpp>

extern bool verbose;

extern bool ignore_errors;

bool DrizzleDumpDatabaseDrizzle::populateTables()

{

  drizzle_result_st *result;

  drizzle_row_t row;

  std::string query;

  if (not dcon->setDB(databaseName))

    return false;

  if (verbose)

    std::cerr << _("-- Retrieving table structures for ") << databaseName << "..." << std::endl;

  query="SELECT TABLE_NAME, TABLE_COLLATION, ENGINE, AUTO_INCREMENT, TABLE_COMMENT FROM DATA_DICTIONARY.TABLES WHERE TABLE_SCHEMA='";

  query.append(databaseName);

  query.append("' ORDER BY TABLE_NAME");

  result= dcon->query(query);

  if (result == NULL)

    return false;

  while ((row= drizzle_row_next(result)))

  {

    size_t* row_sizes= drizzle_row_field_sizes(result);

    std::string tableName(row[0]);

    std::string displayName(tableName);

    cleanTableName(displayName);

    if (not ignoreTable(displayName))

      continue;

    DrizzleDumpTable *table = new DrizzleDumpTableDrizzle(tableName, dcon);

    table->displayName= displayName;

    table->collate= row[1];

    table->engineName= row[2];

    table->autoIncrement= boost::lexical_cast<uint64_t>(row[3]);

    if (row[4])

      table->comment= DrizzleDumpData::escape(row[4], row_sizes[4]);

    else

      table->comment= "";

    table->database= this;

    if ((not table->populateFields()) or (not table->populateIndexes()) or

      (not table->populateFkeys()))

    {

      delete table;

      if (not ignore_errors)

        return false;

      else

        continue;

    }

    tables.push_back(table);

  }

  dcon->freeResult(result);

  return true;

}

bool DrizzleDumpDatabaseDrizzle::populateTables(const std::vector<std::string> &table_names)

{

  drizzle_result_st *result;

  drizzle_row_t row;

  std::string query;

  if (not dcon->setDB(databaseName))

    return false;

  if (verbose)

    std::cerr << _("-- Retrieving table structures for ") << databaseName << "..." << std::endl;

  for (std::vector<std::string>::const_iterator it= table_names.begin(); it != table_names.end(); ++it)

  {

    std::string tableName= *it;

    std::string displayName(tableName);

    cleanTableName(displayName);

    if (not ignoreTable(displayName))

      continue;

    query="SELECT TABLE_NAME, TABLE_COLLATION, ENGINE FROM DATA_DICTIONARY.TABLES WHERE TABLE_SCHEMA='";

    query.append(databaseName);

    query.append("' AND TABLE_NAME = '");

    query.append(tableName);

    query.append("'");

    result= dcon->query(query);

    if (result == NULL)

    {

      std::cerr << "Error: Could not obtain schema for table " << displayName << std::endl;

      return false;

    }

    if ((row= drizzle_row_next(result)))

    {

      DrizzleDumpTableDrizzle *table = new DrizzleDumpTableDrizzle(tableName, dcon);

      table->displayName= displayName;

      table->collate= row[1];

      table->engineName= row[2];

      table->autoIncrement= 0;

      table->database= this;

      if ((not table->populateFields()) or (not table->populateIndexes()))

      {

        std::cerr  << "Error: Could not get fields and/ot indexes for table " << displayName << std::endl;

        delete table;

        dcon->freeResult(result);

        if (not ignore_errors)

          return false;

        else

          continue;

      }

      tables.push_back(table);

      dcon->freeResult(result);

    }

    else

    {

      std::cerr << "Error: Table " << displayName << " not found." << std::endl;

      dcon->freeResult(result);

      if (not ignore_errors)

        return false;

      else

        continue;

    }

  }

  return true;

}

void DrizzleDumpDatabaseDrizzle::setCollate(const char* newCollate)

{

  if (newCollate)

    collate= newCollate;

  else

    collate= "utf8_general_ci";

}

bool DrizzleDumpTableDrizzle::populateFields()

{

  drizzle_result_st *result;

  drizzle_row_t row;

  std::string query;

  if (verbose)

    std::cerr << _("-- Retrieving fields for ") << tableName << "..." << std::endl;

  query= "SELECT COLUMN_NAME, DATA_TYPE, COLUMN_DEFAULT, COLUMN_DEFAULT_IS_NULL, IS_NULLABLE, CHARACTER_MAXIMUM_LENGTH, NUMERIC_PRECISION, NUMERIC_SCALE, COLLATION_NAME, IS_AUTO_INCREMENT, ENUM_VALUES, COLUMN_COMMENT FROM DATA_DICTIONARY.COLUMNS WHERE TABLE_SCHEMA='";

  query.append(database->databaseName);

  query.append("' AND TABLE_NAME='");

  query.append(tableName);

  query.append("'");

  result= dcon->query(query);

  if (result == NULL)

    return false;

  while ((row= drizzle_row_next(result)))

  {

    std::string fieldName(row[0]);

    DrizzleDumpField *field = new DrizzleDumpFieldDrizzle(fieldName, dcon);

    /* Stop valgrind warning */

    field->convertDateTime= false;

    /* Also sets collation */

    field->setType(row[1], row[8]);

    if (row[2])

      field->defaultValue= row[2];

    else

      field->defaultValue= "";

    field->isNull= (strcmp(row[4], "YES") == 0) ? true : false;

    field->isAutoIncrement= (strcmp(row[9], "YES") == 0) ? true : false;

    field->defaultIsNull= (strcmp(row[3], "YES") == 0) ? true : false;

    field->enumValues= (row[10]) ? row[10] : "";

    field->length= (row[5]) ? boost::lexical_cast<uint32_t>(row[5]) : 0;

    field->decimalPrecision= (row[6]) ? boost::lexical_cast<uint32_t>(row[6]) : 0;

    field->decimalScale= (row[7]) ? boost::lexical_cast<uint32_t>(row[7]) : 0;

    field->comment= (row[11]) ? row[11] : "";

    fields.push_back(field);

  }

  dcon->freeResult(result);

  return true;

}

bool DrizzleDumpTableDrizzle::populateIndexes()

{

  drizzle_result_st *result;

  drizzle_row_t row;

  std::string query;

  std::string lastKey;

  bool firstIndex= true;

  DrizzleDumpIndex *index;

  if (verbose)

    std::cerr << _("-- Retrieving indexes for ") << tableName << "..." << std::endl;

  query= "SELECT INDEX_NAME, COLUMN_NAME, IS_USED_IN_PRIMARY, IS_UNIQUE, COMPARE_LENGTH FROM DATA_DICTIONARY.INDEX_PARTS WHERE TABLE_SCHEMA='";

  query.append(database->databaseName);

  query.append("' AND TABLE_NAME='");

  query.append(tableName);

  query.append("'");

  result= dcon->query(query);

  if (result == NULL)

    return false;

  while ((row= drizzle_row_next(result)))

  {

    std::string indexName(row[0]);

    if (indexName.compare(lastKey) != 0)

    {

      if (!firstIndex)

        indexes.push_back(index);

      index = new DrizzleDumpIndexDrizzle(indexName, dcon);

      index->isPrimary= (strcmp(row[0], "PRIMARY") == 0);

      index->isUnique= (strcmp(row[3], "YES") == 0);

      index->isHash= 0;

      index->length= (row[4]) ? boost::lexical_cast<uint32_t>(row[4]) : 0;

      lastKey= row[0];

      firstIndex= false;

    }

    index->columns.push_back(row[1]);

  }

  if (!firstIndex)

    indexes.push_back(index);

  dcon->freeResult(result);

  return true;

}

bool DrizzleDumpTableDrizzle::populateFkeys()

{

  drizzle_result_st *result;

  drizzle_row_t row;

  std::string query;

  DrizzleDumpForeignKey *fkey;

  if (verbose)

    std::cerr << _("-- Retrieving foreign keys for ") << tableName << "..." << std::endl;

  query= "SELECT CONSTRAINT_NAME, CONSTRAINT_COLUMNS, REFERENCED_TABLE_NAME, REFERENCED_TABLE_COLUMNS, MATCH_OPTION, DELETE_RULE, UPDATE_RULE FROM DATA_DICTIONARY.FOREIGN_KEYS WHERE CONSTRAINT_SCHEMA='";

  query.append(database->databaseName);

  query.append("' AND CONSTRAINT_TABLE='");

  query.append(tableName);

  query.append("'");

  result= dcon->query(query);

  if (result == NULL)

    return false;

  while ((row= drizzle_row_next(result)))

  {

    fkey= new DrizzleDumpForeignKey(row[0], dcon);

    fkey->parentColumns= row[1];

    fkey->childTable= row[2];

    fkey->childColumns= row[3];

    fkey->matchOption= (strcmp(row[4], "NONE") != 0) ? row[4] : "";

    fkey->deleteRule= (strcmp(row[5], "UNDEFINED") != 0) ? row[5] : "";

    fkey->updateRule= (strcmp(row[6], "UNDEFINED") != 0) ? row[6] : "";

    fkeys.push_back(fkey);

  }

  dcon->freeResult(result);

  return true;

}

DrizzleDumpData* DrizzleDumpTableDrizzle::getData(void)

{

  try

  {

    return new DrizzleDumpDataDrizzle(this, dcon);

  }

  catch(...)

  {

    return NULL;

  }

}

void DrizzleDumpFieldDrizzle::setType(const char* raw_type, const char* raw_collation)

{

  collation= raw_collation;

  if (strcmp(raw_type, "BLOB") == 0)

  {

    if (strcmp(raw_collation, "binary") != 0)

      type= "TEXT";

    else

      type= raw_type;

    return;

  }

  if (strcmp(raw_type, "VARCHAR") == 0)

  {

    if (strcmp(raw_collation, "binary") != 0)

      type= "VARCHAR";

    else

      type= "VARBINARY";

    return;

  }

  if (strcmp(raw_type, "INTEGER") == 0)

  {

    type= "INT";

    return;

  }

  type= raw_type;

}

DrizzleDumpDataDrizzle::DrizzleDumpDataDrizzle(DrizzleDumpTable *dataTable,

  DrizzleDumpConnection *connection)

  : DrizzleDumpData(dataTable, connection)

{

  std::string query;

  query= "SELECT * FROM `";

  query.append(table->displayName);

  query.append("`");

  result= dcon->query(query);

  if (result == NULL)

    throw std::exception();

}

DrizzleDumpDataDrizzle::~DrizzleDumpDataDrizzle()

{

  dcon->freeResult(result);

}