DatabaseLayer.cpp

Go to the documentation of this file.

#include "../include/DatabaseLayer.h"

#include <wx/tokenzr.h>

// ctor()
DatabaseLayer::DatabaseLayer()
 : DatabaseErrorReporter()
{
}

// dtor()
DatabaseLayer::~DatabaseLayer()
{
  CloseResultSets();
  CloseStatements();
}

bool DatabaseLayer::RunQuery(const wxString& strQuery)
{
  return RunQuery(strQuery, true);
}

void DatabaseLayer::CloseResultSets()
{
  // Iterate through all of the result sets and close them all
  DatabaseResultSetHashSet::iterator start = m_ResultSets.begin();
  DatabaseResultSetHashSet::iterator stop = m_ResultSets.end();
  while (start != stop)
  {
    delete (*start);
    start++;
  }
  m_ResultSets.clear();
}

void DatabaseLayer::CloseStatements()
{
  // Iterate through all of the statements and close them all
  DatabaseStatementHashSet::iterator start = m_Statements.begin();
  DatabaseStatementHashSet::iterator stop = m_Statements.end();
  while (start != stop)
  {
    delete (*start);
    start++;
  }
  m_Statements.clear();
}

bool DatabaseLayer::CloseResultSet(DatabaseResultSet* pResultSet)
{
  if (pResultSet != NULL)
  {
    // Check if we have this result set in our list
    if (m_ResultSets.find(pResultSet) != m_ResultSets.end())
    {
      // Remove the result set pointer from the list and delete the pointer
      delete pResultSet;
      m_ResultSets.erase(pResultSet);
      return true;
    }

    // If not then iterate through all of the statements and see
    //  if any of them have the result set in their lists
    DatabaseStatementHashSet::iterator it;
    for( it = m_Statements.begin(); it != m_Statements.end(); ++it )
    {
      // If the statement knows about the result set then it will close the 
      //  result set and return true, otherwise it will return false
      PreparedStatement* pStatement = *it;
      if (pStatement != NULL)
      {
        if (pStatement->CloseResultSet(pResultSet))
        {
          return true;
        }
      }
    }

    // If we don't know about the result set and the statements don't
    //  know about it, the just delete it
    delete pResultSet;
    return true;
  }
  else
  {
    // Return false on NULL pointer
    return false;
  }

}

bool DatabaseLayer::CloseStatement(PreparedStatement* pStatement)
{
  if (pStatement != NULL)
  {
    // See if we know about this pointer, if so then remove it from the list
    if (m_Statements.find(pStatement) != m_Statements.end())
    {
      // Remove the statement pointer from the list and delete the pointer
      delete pStatement;
      m_Statements.erase(pStatement);
      return true;
    }

    // Otherwise just delete it
    delete pStatement;
    return true;
  }
  else
  {
    // Return false on NULL pointer
    return false;
  }
}

---