flagset.cpp

天之炼狱1服务器端源文件游戏服务端不完整

//////////////////////////////////////////////////////////////////////////////
// Filename    : FlagSet.cpp
// Written by  : excel96
// Description : 
//////////////////////////////////////////////////////////////////////////////

#include "FlagSet.h"
#include "DB.h"
#include "StringStream.h"

//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////

string FlagSet::m_pLookup[256];
bool   FlagSet::m_bInit   = false;

//////////////////////////////////////////////////////////////////////////////
// class Flag member methods
//////////////////////////////////////////////////////////////////////////////

FlagSet::FlagSet()
	throw()
{
	__BEGIN_TRY

	//if (!FlagSet::m_bInit) FlagSet::initialize();

	memset(m_pData, 0, sizeof(BYTE)*FLAG_SIZE_MAX/8);

	__END_CATCH
}

FlagSet::~FlagSet()
	throw()
{
	__BEGIN_TRY
	__END_CATCH
}

void FlagSet::create(const string& owner) 
	throw()
{
	__BEGIN_TRY

	Statement* pStmt = NULL;

	BEGIN_DB
	{
		pStmt = g_pDatabaseManager->getConnection("DARKEDEN")->createStatement();

		StringStream sql;
		sql << "INSERT INTO FlagSet ("
			<< "OwnerID, FlagData"
			<< ") VALUES ("
			<< "'" << owner      << "', "
			<< "'" << toString() << "'  "
			<< ")";

		pStmt->executeQuery(sql.toString());

		SAFE_DELETE(pStmt);
	}
	END_DB(pStmt)

	__END_CATCH
}

void FlagSet::load(const string& owner) 
	throw()
{
	__BEGIN_TRY

	Statement* pStmt = NULL;

	BEGIN_DB
	{
		pStmt = g_pDatabaseManager->getConnection("DARKEDEN")->createStatement();
		Result* pResult = pStmt->executeQuery("SELECT FlagData FROM FlagSet WHERE OwnerID = '%s'", 
												owner.c_str());

		if (pResult->getRowCount() == 0)
		{
			StringStream msg;
			msg << "FlagSet::load() " 
				<< "There is no flag data which owner is "
				<< "[" << owner << "]";

			filelog("flagSetError.txt", "%s", msg.toString().c_str());

			// flagSet捞 绝绰 版快俊绰 老窜 default肺 持绢辑 悼累窍霸 茄促.
			pStmt->executeQuery("INSERT IGNORE INTO FlagSet (OwnerID, FlagData) VALUES ('%s','')", owner.c_str());

			//cerr << msg.toString() << endl;
			//throw (msg.toString());
		}
		else
		{
			pResult->next();

			string text = pResult->getString(1);

			for (uint i=0; i<text.size() && i<(FLAG_SIZE_MAX/8); i++)
			{
				if (text[i] == '0') turnOff(i);
				else turnOn(i);
			}
		}

		SAFE_DELETE(pStmt);
	}
	END_DB(pStmt)

	__END_CATCH
}

void FlagSet::save(const string& owner) 
	throw()
{
	__BEGIN_TRY

	Statement* pStmt = NULL;

	BEGIN_DB
	{
		StringStream sql;
		sql << "UPDATE FlagSet SET FlagData='" << toString()
			<< "' WHERE OwnerID='" << owner << "'";

		pStmt = g_pDatabaseManager->getConnection("DARKEDEN")->createStatement();

		pStmt->executeQuery(sql.toString());

		SAFE_DELETE(pStmt);
	}
	END_DB(pStmt)

	__END_CATCH
}

void FlagSet::destroy(const string& owner) 
	throw()
{
	__BEGIN_TRY

	Statement* pStmt = NULL;

	BEGIN_DB
	{
		StringStream sql;
		sql << "DELETE FROM FlagSet WHERE OwnerID = '" << owner << "'";

		pStmt = g_pDatabaseManager->getConnection("DARKEDEN")->createStatement();
		pStmt->executeQuery(sql.toString());
		SAFE_DELETE(pStmt);
	}
	END_DB(pStmt)

	__END_CATCH
}

bool FlagSet::isOn(int index)
	throw()
{
	__BEGIN_TRY

	if (!isValidIndex(index)) return false;
	if (m_pData[index/8] & (1<<(index%8))) return true;
	return false;

	__END_CATCH
}

bool FlagSet::turnOn(int index) 
	throw()
{
	__BEGIN_TRY

	if (!isValidIndex(index)) return false;
	if (isOn(index)) return false;
	
	m_pData[index>>3] |= 1<<(index%8);
	return true;

	__END_CATCH
}

bool FlagSet::turnOff(int index) 
	throw()
{
	__BEGIN_TRY

	if (!isValidIndex(index)) return false;
	if (!isOn(index)) return false;
	
	m_pData[index>>3] &= ~(1<<(index%8));
	return true;

	__END_CATCH
}

bool FlagSet::isValidIndex(int index) 
	throw()
{
	__BEGIN_TRY

	if (index < 0 || index >= FLAG_SIZE_MAX) return false;
	return true;

	__END_CATCH
}

string FlagSet::toString(void) 
	throw()
{
	__BEGIN_TRY

	string result = "";

	for (int i=0; i<(FLAG_SIZE_MAX/8); i++)
	{
		result += m_pLookup[m_pData[i]];
	}

	return result;

	__END_CATCH
}

FlagSet FlagSet::fromString(const string& text) 
	throw()
{
	__BEGIN_TRY

	FlagSet result;

	for (uint i=0; i<text.size() && i<(FLAG_SIZE_MAX/8); i++)
	{
		if (text[i] == '0') result.turnOff(i);
		else                result.turnOn(i);
	}

	return result;

	__END_CATCH
}

void FlagSet::initialize(void)
	throw()
{
	__BEGIN_TRY

	if (m_bInit) return;

	for (int i=0; i<256; i++)
	{
		string result = "";

		if (i & 0x01) result += "1";
		else          result += "0";

		if (i & 0x02) result += "1";
		else          result += "0";

		if (i & 0x04) result += "1";
		else          result += "0";

		if (i & 0x08) result += "1";
		else          result += "0";

		if (i & 0x10) result += "1";
		else          result += "0";

		if (i & 0x20) result += "1";
		else          result += "0";

		if (i & 0x40) result += "1";
		else          result += "0";

		if (i & 0x80) result += "1";
		else          result += "0";

		m_pLookup[i] = result;
	}
	
	m_bInit = true;

	__END_CATCH
}