khanat-opennel-code/code/ryzom/server/src/backup_service/backup_service.h

// 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/>.




#ifndef BACKUP_SERVICE_H
#define BACKUP_SERVICE_H


// net
#include "nel/misc/types_nl.h"
#include "nel/misc/log.h"
#include "nel/net/service.h"
#include "nel/misc/variable.h"

// stl
#include <string>

//
#include "backup_file_access.h"


/**
 * CBackupService
 *
 * \author Alain Saffray
 * \author Nevrax France
 * \date 2004
 */
class CBackupService : public NLNET::IService
{
public :
	/** 
	 * init the service
	 */
	void init();

	/**
	 * main loop
	 */
	bool update();
	
	/**
	 * release
	 */
	void release();

	// stall shard
	void stallShard(const std::string& fileName);

	/**
	 * get Instance
	 */
	static CBackupService * getInstance() { return (CBackupService *) NLNET::IService::getInstance(); }

	/**
	 * get stall, return true if we must not try to save file because previous pb are occurs
	 */
	bool getStall() { return _SaveStall; }

	/**
	 * set stall mode
	 */
	void setStall( bool stall ) { _SaveStall = stall; }

	NLNET::IModule		*getBackupModule()	{ return _BackupModule; };

	void				setBackupModule(NLNET::IModule *module)	{ _BackupModule = module; }

	void		onModuleDown(NLNET::IModuleProxy *proxy);

public:


	CFileAccessManager	FileManager;

private:

	bool		_SaveStall;

	NLNET::TModulePtr		_BackupModule;

	// A layer 3 server to serve synchronous file load
	NLNET::CCallbackServer	*_CallbackServer;
}; 


extern NLMISC::CVariable<std::string>	StatDirFilter;


class CDirectoryRateStat
{
public:

	CDirectoryRateStat()
	{
		clear();
	}

	void	clear();

	void	readFile(const std::string& filename, uint32 filesize);
	
	void	writeFile(const std::string& filename, uint32 filesize);

	uint	getMeanReadRate();

	uint	getMeanWriteRate();

	void	display(NLMISC::CLog& log);

private:

	struct CDirectory
	{
		struct CFileQueueEntry
		{
			CFileQueueEntry(NLMISC::TTime time, uint32 size, const std::string& filename = "", uint32 duration = 0) : ActionTime(time), Size(size), Filename(filename), Duration(duration)	{ }
			NLMISC::TTime	ActionTime;
			uint32			Size;
			std::string		Filename;
			uint32			Duration;
		};

		typedef std::deque<CFileQueueEntry>	TFileQueue;

		// read stats
		TFileQueue		ReadQueue;
		uint64			ReadBytes;
		uint32			ReadFiles;

		// write stats
		TFileQueue		WrittenQueue;
		uint64			WrittenBytes;
		uint32			WrittenFiles;

		CDirectory()
		{
			clear();
		}

		void	clear()
		{
			ReadBytes = 0;
			ReadFiles = 0;

			WrittenBytes = 0;
			WrittenFiles = 0;
		}

		void	read(NLMISC::TTime now, uint32 filesize)
		{
			updateTime(now - 60*1000);
			ReadBytes += filesize;
			++ReadFiles;
			ReadQueue.push_back(CFileQueueEntry(now, filesize));
		}

		void	write(NLMISC::TTime now, uint32 filesize)
		{
			updateTime(now - 60*1000);
			WrittenBytes += filesize;
			++WrittenFiles;
			WrittenQueue.push_back(CFileQueueEntry(now, filesize));
		}

		void	updateTime(NLMISC::TTime limit)
		{
			while (!WrittenQueue.empty() && WrittenQueue.front().ActionTime < limit)
			{
				WrittenBytes -= WrittenQueue.front().Size;
				--WrittenFiles;
				WrittenQueue.pop_front();
			}
			while (!ReadQueue.empty() && ReadQueue.front().ActionTime < limit)
			{
				ReadBytes -= ReadQueue.front().Size;
				--ReadFiles;
				ReadQueue.pop_front();
			}
		}
	};


	// mapping to filtered directories
	typedef std::map<std::string, CDirectory>	TDirectoryMap;
	TDirectoryMap						_DirectoryMap;

};



#endif // BACKUP_SERVICE_H

/* End of backup_service.h */