MysqlDatabaseLayer.cpp

Go to the documentation of this file.

#include "../include/MysqlDatabaseLayer.h"
#include "../include/MysqlPreparedStatement.h"
#include "../include/MysqlPreparedStatementResultSet.h"
#include "../include/DatabaseErrorCodes.h"

#include <wx/tokenzr.h>

// ctor
MysqlDatabaseLayer::MysqlDatabaseLayer()
 : DatabaseLayer()
{
  InitDatabase();
  m_strServer = _("localhost");
  m_iPort = 3306; // default
  m_strDatabase = _("");
  m_strUser = _("");
  m_strPassword = _("");
}

MysqlDatabaseLayer::MysqlDatabaseLayer(const wxString& strDatabase)
 : DatabaseLayer()
{
  InitDatabase();
  m_strServer = _("localhost");
  m_iPort = 3306; // default
  m_strUser = _("");
  m_strPassword = _("");
  Open(strDatabase);
}

MysqlDatabaseLayer::MysqlDatabaseLayer(const wxString& strServer, const wxString& strDatabase)
 : DatabaseLayer()
{
  InitDatabase();
  ParseServerAndPort(strServer);
  m_strUser = _("");
  m_strPassword = _("");
  Open(strDatabase);
}

MysqlDatabaseLayer::MysqlDatabaseLayer(const wxString& strDatabase, const wxString& strUser, const wxString& strPassword)
 : DatabaseLayer()
{
  InitDatabase();
  m_strServer = _("localhost");
  m_iPort = 3306; // default
  m_strUser = strUser;
  m_strPassword = strPassword;
  Open(strDatabase);
}

MysqlDatabaseLayer::MysqlDatabaseLayer(const wxString& strServer, const wxString& strDatabase, const wxString& strUser, const wxString& strPassword)
 : DatabaseLayer()
{
  InitDatabase();
  ParseServerAndPort(strServer);
  m_strUser = strUser;
  m_strPassword = strPassword;
  Open(strDatabase);
}

// dtor
MysqlDatabaseLayer::~MysqlDatabaseLayer()
{
  Close();
  //mysql_close(m_pDatabase);
  //delete m_pDatabase;
  //mysql_library_end();
  mysql_server_end();
}

// open database
void MysqlDatabaseLayer::InitDatabase()
{
  //char *server_options[] = { "mysql_test", "--defaults-file=my.cnf" };
  //int num_elements = sizeof(server_options)/ sizeof(char *);

  //char *server_groups[] = { "libmysqld_server", "libmysqld_client" };
  //mysql_server_init(num_elements, server_options, server_groups);
  //m_pDatabase = new MYSQL();
  //mysql_library_init();
  //mysql_init(m_pDatabase);
  mysql_server_init( 0, NULL, NULL );
  m_pDatabase = mysql_init(NULL);
}

// open database
bool MysqlDatabaseLayer::Open(const wxString& strServer, const wxString& strDatabase)
{
  ParseServerAndPort(strServer);
  return Open(strDatabase);
}

bool MysqlDatabaseLayer::Open(const wxString& strDatabase, const wxString& strUser, const wxString& strPassword)
{
  m_strUser = strUser;
  m_strPassword = strPassword;
  return Open(strDatabase);
}

bool MysqlDatabaseLayer::Open(const wxString& strServer, const wxString& strDatabase, const wxString& strUser, const wxString& strPassword)
{
  ParseServerAndPort(strServer);
  m_strUser = strUser;
  m_strPassword = strPassword;
  return Open(strDatabase);
}

bool MysqlDatabaseLayer::Open(const wxString& strDatabase)
{
  m_strDatabase = strDatabase;
  
  wxCharBuffer serverCharBuffer = ConvertToUnicodeStream(m_strServer);
  wxCharBuffer userCharBuffer = ConvertToUnicodeStream(m_strUser);
  wxCharBuffer passwordCharBuffer = ConvertToUnicodeStream(m_strPassword);
  wxCharBuffer databaseNameCharBuffer = ConvertToUnicodeStream(m_strDatabase);
  if (mysql_real_connect(m_pDatabase, serverCharBuffer, userCharBuffer, passwordCharBuffer, databaseNameCharBuffer, m_iPort, NULL/*socket*/, 0) != NULL)
  {
#if wxUSE_UNICODE
    const char* sqlStatement = "SET CHARACTER_SET_CLIENT=utf8, "
                 "CHARACTER_SET_CONNECTION=utf8, "
                 "CHARACTER_SET_RESULTS=utf8;";

    mysql_real_query(m_pDatabase, sqlStatement, strlen(sqlStatement));
    wxCSConv conv(_("UTF-8"));
    SetEncoding(&conv);
#endif

    return true;
  }
  else
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
    return false;
  }
}

void MysqlDatabaseLayer::ParseServerAndPort(const wxString& strServer)
{
  int portIndicator = strServer.Find(_(":"));
  if (portIndicator > -1)
  {
    m_strServer = strServer.SubString(0, portIndicator-1);
    m_iPort = wxAtoi(strServer.SubString(portIndicator+1, strServer.Length()-1));
  }
  else
  {
    m_strServer = strServer;
    m_iPort = 3306; // default
  }
}

// close database
bool MysqlDatabaseLayer::Close()
{
  CloseResultSets();
  CloseStatements();

  ResetErrorCodes();
/*
#if wxUSE_UNICODE
  // Iterate through all the result sets and close any open ones
  PointerLookupMap::iterator ResultSetStart = m_ResultSets.begin();
  PointerLookupMap::iterator ResultSetStop = m_ResultSets.end();

  while (ResultSetStart != ResultSetStop)
  {
    delete (MysqlPreparedStatementResultSet*)((*ResultSetStart).first);
    if ((*ResultSetStart).second != NULL)
    {
        MYSQL_STMT* pStatement = (MYSQL_STMT*)((*ResultSetStart).second);
        wxPrintf(_("Closing statement %d\n"), pStatement);
        mysql_stmt_close(pStatement);
    }
    ResultSetStart++;  
  }
  m_ResultSets.clear();
#endif
*/
  mysql_close(m_pDatabase);
  mysql_server_end();
  //delete m_pDatabase;
  //m_pDatabase = NULL;
  return true;
}

  
// transaction support
void MysqlDatabaseLayer::BeginTransaction()
{
  ResetErrorCodes();

  int nReturn = mysql_autocommit(m_pDatabase, 0);
  if (nReturn != 0)
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
  }
}

void MysqlDatabaseLayer::Commit()
{
  ResetErrorCodes();

  int nReturn = mysql_commit(m_pDatabase);
  if (nReturn != 0)
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
  }
  nReturn = mysql_autocommit(m_pDatabase, 1);
  if (nReturn != 0)
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
  }
}

void MysqlDatabaseLayer::RollBack()
{
  ResetErrorCodes();

  int nReturn = mysql_rollback(m_pDatabase);
  if (nReturn != 0)
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
  }
  nReturn = mysql_autocommit(m_pDatabase, 1);
  if (nReturn != 0)
  {
    SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
    SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
    ThrowDatabaseException();
  }
}

  
// query database
bool MysqlDatabaseLayer::RunQuery(const wxString& strQuery, bool bParseQuery)
{
  ResetErrorCodes();

  wxArrayString QueryArray;
  if (bParseQuery)
    QueryArray = ParseQueries(strQuery);
  else
    QueryArray.push_back(strQuery);

  wxArrayString::iterator start = QueryArray.begin();
  wxArrayString::iterator stop = QueryArray.end();

  while (start != stop)
  {
    wxCharBuffer sqlBuffer = ConvertToUnicodeStream(*start);
    //puts(sqlBuffer);
    int nReturn = mysql_query(m_pDatabase, sqlBuffer);
    if (nReturn != 0)
    {
      SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
      SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
      ThrowDatabaseException();
      return false;
    }
    start++;
  }
  return true;
}

DatabaseResultSet* MysqlDatabaseLayer::RunQueryWithResults(const wxString& strQuery)
{
  ResetErrorCodes();

  wxArrayString QueryArray = ParseQueries(strQuery);

  int nArraySize = QueryArray.size();
  MysqlPreparedStatementResultSet* pResultSet = NULL;
  for (int i=0; i<nArraySize; i++)
  {
    wxString strCurrentQuery = QueryArray[i];
    MYSQL_STMT* pMysqlStatement = mysql_stmt_init(m_pDatabase);
    if (pMysqlStatement != NULL)
    {
      wxCharBuffer sqlBuffer = ConvertToUnicodeStream(strCurrentQuery);
      //puts(sqlBuffer);
      wxString sqlUTF8((const char*)sqlBuffer, wxConvUTF8);
      if (mysql_stmt_prepare(pMysqlStatement, sqlBuffer, sqlUTF8.Length()) == 0)
      {
        int nReturn = mysql_stmt_execute(pMysqlStatement);
        if (nReturn != 0)
        {
          SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_stmt_errno(pMysqlStatement)));
          SetErrorMessage(ConvertFromUnicodeStream(mysql_stmt_error(pMysqlStatement)));
          ThrowDatabaseException();
          return NULL;
        }
      }
      else
      {
        SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
        SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
        ThrowDatabaseException();
      }
      if (i == nArraySize-1)
      {
        pResultSet = new MysqlPreparedStatementResultSet(pMysqlStatement);
        if (pResultSet)
          pResultSet->SetEncoding(GetEncoding());
#if wxUSE_UNICODE
        //wxPrintf(_("Allocating statement at %d\n"), pMysqlStatement);
       // m_ResultSets[pResultSet] = pMysqlStatement;
#endif
        LogResultSetForCleanup(pResultSet);
        return pResultSet;
      }

      mysql_stmt_free_result(pMysqlStatement);
      mysql_stmt_close(pMysqlStatement);
    }
    else
    {
      SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
      SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
      ThrowDatabaseException();
      return NULL;
    }
  }
  LogResultSetForCleanup(pResultSet);
  return pResultSet;
}
/*
void MysqlDatabaseLayer::CloseResultSet(DatabaseResultSet* pResultSet)
{
  if (pResultSet != NULL)
  {
    //pResultSet->Close();
    delete pResultSet;

    // If the result set is registered in the map, then delete any associated mysql statements
    //  and remove from the map
#if wxUSE_UNICODE
    PointerLookupMap::iterator finder = m_ResultSets.find(pResultSet);
    if (finder != m_ResultSets.end())
    {
      if ((*finder).second != NULL)
      {
        MYSQL_STMT* pStatement = (MYSQL_STMT*)((*finder).second);
        //wxPrintf(_("deleting statement at %d\n"), m_ResultSets[pResultSet]);
        mysql_stmt_close((MYSQL_STMT*)(m_ResultSets[pResultSet]));
      }
      m_ResultSets.erase(pResultSet);
    }
#endif
  }
}
*/

// PreparedStatement support
PreparedStatement* MysqlDatabaseLayer::PrepareStatement(const wxString& strQuery)
{
  ResetErrorCodes();

  wxArrayString QueryArray = ParseQueries(strQuery);

  wxArrayString::iterator start = QueryArray.begin();
  wxArrayString::iterator stop = QueryArray.end();

  MysqlPreparedStatement* pStatement = new MysqlPreparedStatement();
  if (pStatement)
    pStatement->SetEncoding(GetEncoding());
  while (start != stop)
  {
    MYSQL_STMT* pMysqlStatement = mysql_stmt_init(m_pDatabase);
    if (pMysqlStatement != NULL)
    {
      wxCharBuffer sqlBuffer = ConvertToUnicodeStream((*start));
      //puts(sqlBuffer);
      if (mysql_stmt_prepare(pMysqlStatement, sqlBuffer, GetEncodedStreamLength((*start))) == 0)
      {
        pStatement->AddPreparedStatement(pMysqlStatement);
      }
      else
      {
        SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
        SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
        ThrowDatabaseException();
      }
    }
    else
    {
      SetErrorCode(MysqlDatabaseLayer::TranslateErrorCode(mysql_errno(m_pDatabase)));
      SetErrorMessage(ConvertFromUnicodeStream(mysql_error(m_pDatabase)));
      ThrowDatabaseException();
      return NULL;
    }
    start++;
  }
  LogStatementForCleanup(pStatement);
  return pStatement;
}

int MysqlDatabaseLayer::TranslateErrorCode(int nCode)
{
  // Ultimately, this will probably be a map of Mysql database error code values to DatabaseLayer values
  // For now though, we'll just return error
  return nCode;
  //return DATABASE_LAYER_ERROR;
}

---