OTS/OTSCPP/OTSSQLiteDll/SQLiteDB.cpp

#pragma once

#include "stdafx.h"
#include "SQLiteDB.h"
#include <wtypes.h>
#include "DBStoreBase.h"
#include "DBTableBase.h"

namespace OTSSQLITE
{

	CSQLiteDB::CSQLiteDB(CDBStoreBasePtr a_datastore,CDBTableBasePtr a_tableInfo)
	{
		m_datastorePtr = a_datastore;
		m_tableInfo = a_tableInfo;
	}


	CSQLiteDB::~CSQLiteDB(void)
	{
	}

	BOOL CSQLiteDB::Init(const BOOL a_bClean /*= FALSE*/)
	{
		if (IsDBExist())
		{
			if (a_bClean)
			{
				DeleteTable();
			}
			else
			{
				return TRUE;
			}
		}

		return CreateTable();
	}

	BOOL CSQLiteDB::IsDBExist()
	{
		auto datastorePtr = GetDatastore();
		ASSERT(datastorePtr);
		if (!datastorePtr)
		{
			return FALSE;
		}
	
		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr)
		{
			return FALSE;
		}

		return datastorePtr->IsTableExists(tableInfoPtr->GetTableName());
	}

    CDBQueryBasePtr	CSQLiteDB::GetTableQuery(LPCTSTR a_sOrderColumnName /*= nullptr*/)
	{
		auto datastorePtr = GetDatastore();

		ASSERT(datastorePtr);
		if (!datastorePtr)
		{
			return CDBQueryBasePtr();
		}

		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr) { return CDBQueryBasePtr(); }

		CString sTableName = tableInfoPtr->GetTableName();
		if (!datastorePtr->IsTableExists(sTableName))
		{
			return CDBQueryBasePtr();
		}

		auto query = datastorePtr->QueryByTableName(sTableName, a_sOrderColumnName);
		ASSERT(query);
		if (!query || !query->IsValid())
		{
			LogErrorTrace(__FILE__, __LINE__, _T("Invalid table(%s) query command."), (LPCTSTR)tableInfoPtr->GetTableName());
			return CDBQueryBasePtr();
		}

		// do the table related valid checking
		if (query->GetColCount() != tableInfoPtr->GetColumnCount())
		{
			LogErrorTrace(__FILE__, __LINE__, _T("Invalid table columns(%d): should be %d"), query->GetColCount(), tableInfoPtr->GetColumnCount());
		}

		return query;
	}

	OTSSQLITE::CDBTableBasePtr CSQLiteDB::GetTableInfo()
	{
		return m_tableInfo;
		
	}

	BOOL CSQLiteDB::CreateTable(const BOOL a_bForce /*= FALSE*/)
	{
		auto datastorePtr = GetDatastore();
		ASSERT(datastorePtr);
		if (!datastorePtr)
		{
			return FALSE;
		}

		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr)
		{
			return FALSE;
		}

		if (datastorePtr->IsTableExists(tableInfoPtr->GetTableName()))
		{
			if (a_bForce)
			{
				if (!DeleteTable())
				{
					LogErrorTrace(__FILE__, __LINE__, _T("CreateTable:delete table (%s) failed."), (LPCTSTR)tableInfoPtr->GetTableName());
					ASSERT(FALSE);
					return FALSE;
				}
			}
			else
			{
				return TRUE;
			}
		}

		// create table
		CString sSQLCommand = tableInfoPtr->GetCreateTableCommandString();

		return datastorePtr->RunCommand(sSQLCommand);
	}

	BOOL CSQLiteDB::DeleteTable()
	{
		auto datastorePtr = GetDatastore();
		ASSERT(datastorePtr);
		if (!datastorePtr)
		{
			return FALSE;
		}

		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr)
		{
			return FALSE;
		}

		if (!datastorePtr->IsTableExists(tableInfoPtr->GetTableName()))
		{
			return TRUE;
		}

		// create table
		CString sSQLCommand = tableInfoPtr->GetDeleteTableCommandString();

		return datastorePtr->RunCommand(sSQLCommand);
	}

	// Removes all rows.
	BOOL CSQLiteDB::RemoveAllRows()
	{
		auto datastorePtr = GetDatastore();
		ASSERT(datastorePtr);
		if (!datastorePtr)
		{
			return FALSE;
		}

		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr)
		{
			return FALSE;
		}

		if (!datastorePtr->IsTableExists(tableInfoPtr->GetTableName()))
		{
			LogInfoTrace(__FILE__,__LINE__,_T("Didn't remove all rows from table(%s): table not exist"), tableInfoPtr->GetTableName());
			return TRUE;
		}

		// create table
		CString sSQLCommand = tableInfoPtr->GetRemoveAllRowsCommandString();

		return datastorePtr->RunCommand(sSQLCommand);

	}

	CString CSQLiteDB::GetReadColumnError(const int a_nColId)
	{
		auto tableInfoPtr = GetTableInfo();
		ASSERT(tableInfoPtr);
		if (!tableInfoPtr)
		{
			return _T("");
		}
		CString sColumnName = tableInfoPtr->GetColumnName(a_nColId);
		CString sErrorMessage;
		sErrorMessage.Format(_T("Read column(%s) from table(%s) failed"), (LPCTSTR)sColumnName, (LPCTSTR)tableInfoPtr->GetTableName());

		return sErrorMessage;
	}

	OTSSQLITE::CDBBasePtr CreateNewSQLiteDB(CDBStoreBasePtr a_datastore, CDBTableBasePtr a_tableInfo)
	{
		return CDBBasePtr(new CSQLiteDB(a_datastore,a_tableInfo));
	}
}