khanat-code-old/code/ryzom/server/src/pd_lib/pd_server_utils.cpp
2010-05-06 02:08:41 +02:00

444 lines
7.7 KiB
C++

// Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
// Copyright (C) 2010 Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "pd_server_utils.h"
#include "pd_lib.h"
#include <nel/misc/path.h>
#include <nel/misc/debug.h>
#include <nel/net/service.h>
#include <nel/misc/i_xml.h>
#include <nel/misc/o_xml.h>
#include <nel/misc/file.h>
#include <time.h>
#include "db_reference_file.h"
using namespace NLMISC;
using namespace NLNET;
using namespace std;
/*
* Load a Reference index file
*/
bool CRefIndex::load(const string& filename)
{
CIFile reffile;
CIXml ixml;
if (!reffile.open(filename) || !ixml.init(reffile))
return false;
try
{
serial(ixml);
}
catch (Exception&)
{
return false;
}
return true;
}
/*
* Load a Reference index file
*/
bool CRefIndex::load()
{
string rootpath = getRootPath();
if (rootpath.empty())
return false;
return load(rootpath+"ref");
}
/*
* Save a Reference index file
*/
bool CRefIndex::save(const string& filename)
{
COFile reffile;
COXml oxml;
if (!reffile.open(filename) || !oxml.init(&reffile))
return false;
try
{
serial(oxml);
}
catch (Exception&)
{
return false;
}
return true;
}
/*
* Save a Reference index file
*/
bool CRefIndex::save()
{
string path = getPath();
if (path.empty())
return false;
return checkDirectory(path) && save(path+"ref");
}
/*
* Set As Valid Reference
*/
bool CRefIndex::setAsValidRef()
{
string rootpath = getRootPath();
if (rootpath.empty())
return false;
return save(rootpath+"ref");
}
/*
* Build next Reference index file
*/
bool CRefIndex::buildNext()
{
if (!load())
Index = 0;
else
++Index;
setup();
if (!setupDirectory())
return false;
if (!save())
return false;
return true;
}
/*
* Get next Reference index file
*/
void CRefIndex::getNext()
{
++Index;
setup();
}
/*
* Get (and setup if needed) database root path
*/
std::string CRefIndex::getRootPath()
{
string path = getNominalRootPath();
if (!CFile::isExists(path) || !CFile::isDirectory(path))
{
if (!CFile::createDirectoryTree(path))
{
nlwarning("getRootPath(): unable to create save path '%s'", path.c_str());
return string("");
}
if (!CFile::setRWAccess(path))
{
nlwarning("getRootPath(): failure, can't set RW access to path '%s', can't start.", path.c_str());
return string("");
}
}
return path;
}
/*
* Get Nominal Root Path
*/
std::string CRefIndex::getNominalRootPath()
{
return RY_PDS::CPDSLib::getRootDirectory(DatabaseId);
}
/*
* Get reference path
*/
std::string CRefIndex::getPath()
{
return NLMISC::CPath::standardizePath(getRootPath() + Path);
}
/*
* Setup reference directory
*/
bool CRefIndex::setupDirectory()
{
string rootpath = getRootPath();
if (rootpath.empty())
return false;
string path = NLMISC::CPath::standardizePath(rootpath + Path);
if (!checkDirectory(path))
{
nlwarning("CRefIndex::setupDirectory(): failed, can't check directory '%s'", path.c_str());
return false;
}
if (!checkDirectory(getHoursUpdatePath()))
{
nlwarning("CRefIndex::setupDirectory(): failed, can't check subdirectory '%s'", getHoursUpdatePath().c_str());
return false;
}
if (!checkDirectory(getMinutesUpdatePath()))
{
nlwarning("CRefIndex::setupDirectory(): failed, can't check subdirectory '%s'", getMinutesUpdatePath().c_str());
return false;
}
if (!checkDirectory(getSecondsUpdatePath()))
{
nlwarning("CRefIndex::setupDirectory(): failed, can't check subdirectory '%s'", getSecondsUpdatePath().c_str());
return false;
}
if (!checkDirectory(getLogPath()))
{
nlwarning("CRefIndex::setupDirectory(): failed, can't check subdirectory '%s'", getLogPath().c_str());
return false;
}
return true;
}
/*
* Check directory
*/
bool CRefIndex::checkDirectory(const std::string& path)
{
if (!CFile::isExists(path))
{
if (!CFile::createDirectoryTree(path))
{
nlwarning("CRefIndex::checkDirectory(): failed, can't create directory '%s'", path.c_str());
return false;
}
if (!CFile::setRWAccess(path))
{
nlwarning("CRefIndex::checkDirectory(): failed, can't set RW access to directory '%s'", path.c_str());
return false;
}
}
else if (!CFile::isDirectory(path))
{
nlwarning("CRefIndex::checkDirectory(): failed, directory '%s' is already a file", path.c_str());
return false;
}
return true;
}
/*
* Set Time stamp
*/
void CRefIndex::setTimestamp()
{
Timestamp.setToCurrent();
}
/*
* Get Seconds update path
*/
std::string CRefIndex::getSecondsUpdatePath()
{
return getRootPath() + "seconds/";
}
/*
* Get Minutes update path
*/
std::string CRefIndex::getMinutesUpdatePath()
{
return getRootPath() + "minutes/";
}
/*
* Get Hours update path
*/
std::string CRefIndex::getHoursUpdatePath()
{
return getRootPath() + "hours/";
}
/*
* Get Log path
*/
std::string CRefIndex::getLogPath()
{
return getRootPath() + "logs/";
}
/*
* Constructor
*/
CDatabaseState::CDatabaseState()
{
Id = 0xffffffff;
LastUpdateId = (0-1);
CurrentIndex = 0;
}
/*
* Serial method
*/
void CDatabaseState::serial(NLMISC::IStream& s)
{
s.xmlPush("database_state");
s.serialCheck((uint32)'DBST');
uint version = s.serialVersion(0);
s.xmlPush("name");
s.serial(Name);
s.xmlPop();
s.xmlPush("id");
s.serial(Id);
s.xmlPop();
s.xmlPush("lastupdateid");
s.serial(LastUpdateId);
s.xmlPop();
s.xmlPush("currentindex");
s.serial(CurrentIndex);
s.xmlPop();
s.xmlPush("endtimestamp");
if (s.isReading())
{
std::string ts;
s.serial(ts);
EndTimestamp.fromString(ts.c_str());
}
else
{
std::string ts = EndTimestamp.toString();
s.serial(ts);
}
s.xmlPop();
s.xmlPop();
}
/*
* Save State
*/
bool CDatabaseState::save(CRefIndex& ref)
{
COFile f;
COXml oxml;
string filename = fileName(ref);
if (!f.open(filename) || !oxml.init(&f))
return false;
try
{
serial(oxml);
}
catch (Exception&)
{
return false;
}
return true;
}
/*
* Load State
*/
bool CDatabaseState::load(CRefIndex& ref, bool usePrevious)
{
return load(ref.getRootPath(), usePrevious);
}
/*
* Load State
*/
bool CDatabaseState::load(const std::string& rootpath, bool usePrevious)
{
CIFile f;
CIXml ixml;
string filename = CPath::standardizePath(rootpath);
if (usePrevious)
filename += "previous_state";
else
filename += "state";
if (!f.open(filename) || !ixml.init(f))
return false;
try
{
serial(ixml);
}
catch (Exception&)
{
return false;
}
return true;
}
/*
* State exists in path
*/
bool CDatabaseState::exists(const std::string& rootpath)
{
string filename = CPath::standardizePath(rootpath) + "state";
return CFile::fileExists(filename);
}
uint64 CMixedStreamFile::_ReadBytes = 0;
uint64 CMixedStreamFile::_WrittenBytes = 0;