// ============================================================================
//  Copyright (C) 2001-2013 by Jozef Starosczyk
//  ixen@copyhandler.com
//
//  This program is free software; you can redistribute it and/or modify
//  it under the terms of the GNU Library General Public License
//  (version 2) as published by the Free Software Foundation;
//
//  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 Library General Public
//  License along with this program; if not, write to the
//  Free Software Foundation, Inc.,
//  59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
// ============================================================================
#include "stdafx.h"
#include "TSQLiteSerializerContainer.h"
#include "TSQLiteSerializerRowData.h"
#include "ErrorCodes.h"
#include "TCoreException.h"
#include <boost/format.hpp>
#include "TSQLiteStatement.h"
#include "TSQLiteSerializerRowReader.h"
#include <atltrace.h>
#include "TRemovedObjects.h"

BEGIN_CHCORE_NAMESPACE

using namespace sqlite;

TSQLiteSerializerContainer::TSQLiteSerializerContainer(const TString& strName, const sqlite::TSQLiteDatabasePtr& spDB) :
	m_spColumns(new TSQLiteColumnsDefinition),
	m_strName(strName),
	m_spDB(spDB)
{
}

TSQLiteSerializerContainer::TSQLiteSerializerContainer(const TString& strName, size_t stParentID, const sqlite::TSQLiteDatabasePtr& spDB) :
	m_stParentID(stParentID),
	m_spColumns(new TSQLiteColumnsDefinition),
	m_strName(strName),
	m_spDB(spDB)
{
}

TSQLiteSerializerContainer::~TSQLiteSerializerContainer()
{
}

chcore::ISerializerRowDataPtr TSQLiteSerializerContainer::AddRow(size_t stRowID)
{
	RowMap::iterator iterInsert = m_mapRows.insert(
			std::make_pair(stRowID, TSQLiteSerializerRowDataPtr(new TSQLiteSerializerRowData(stRowID, m_spColumns, true)))
		).first;
	return (*iterInsert).second;
}

ISerializerRowDataPtr TSQLiteSerializerContainer::GetRow(size_t stRowID)
{
	RowMap::iterator iterFnd = m_mapRows.find(stRowID);
	if(iterFnd == m_mapRows.end())
		iterFnd = m_mapRows.insert(std::make_pair(stRowID, TSQLiteSerializerRowDataPtr(new TSQLiteSerializerRowData(stRowID, m_spColumns, false)))).first;

	return (*iterFnd).second;
}

void TSQLiteSerializerContainer::DeleteRow(size_t stRowID)
{
	RowMap::iterator iterFnd = m_mapRows.find(stRowID);
	if(iterFnd != m_mapRows.end())
		m_mapRows.erase(iterFnd);

	m_setDeleteItems.insert(stRowID);
}

void TSQLiteSerializerContainer::DeleteRows(const TRemovedObjects& setObjects)
{
	size_t stCount = setObjects.GetCount();
	while(stCount-- != 0)
	{
		DeleteRow(setObjects.GetAt(stCount));
	}
}

ISerializerRowReaderPtr TSQLiteSerializerContainer::GetRowReader()
{
	TSQLiteSerializerRowReaderPtr spRowReader(new TSQLiteSerializerRowReader(m_spDB, m_spColumns, m_strName));
	return spRowReader;
}

chcore::IColumnsDefinitionPtr TSQLiteSerializerContainer::GetColumnsDefinition() const
{
	return m_spColumns;
}

void TSQLiteSerializerContainer::Flush()
{
	FlushDeletions();

	for(RowMap::iterator iterRows = m_mapRows.begin(); iterRows != m_mapRows.end(); ++iterRows)
	{
		iterRows->second->Flush(m_spDB, m_strName);
	}
}

void TSQLiteSerializerContainer::FlushDeletions()
{
	// delete from m_strName WHERE id IN (???)
	TSQLiteStatement tStatement(m_spDB);

	const size_t stMaxToRemoveAtOnce = 10;

	// delete items in chunks
	std::set<size_t>::const_iterator iterToDelete = m_setDeleteItems.begin();
	while(iterToDelete != m_setDeleteItems.end())
	{
		TString strItemsToRemove;
		size_t stToRemove = stMaxToRemoveAtOnce;
		while(iterToDelete != m_setDeleteItems.end() && (--stToRemove) != 0)
		{
			strItemsToRemove += boost::str(boost::wformat(L"%1%,") % *iterToDelete).c_str();
			++iterToDelete;
		}
		strItemsToRemove.TrimRightSelf(_T(","));

		TString strQuery = boost::str(boost::wformat(L"DELETE FROM %1% WHERE id IN (%2%)") % m_strName % strItemsToRemove).c_str();
		tStatement.Prepare(strQuery);

		ATLTRACE(_T("Executing query: %s\n"), (PCTSTR)strQuery);
		tStatement.Step();
	}
}

END_CHCORE_NAMESPACE