ClientRegister.h

Go to the documentation of this file.

/*
* Client register - Stores active clients
*
* @Author Michael <Tharbas> Matzen
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the Free
* Software Foundation; either version 2 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 General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along with
* this program; if not, write to the Free Software Foundation, Inc., 51
* Franklin St, Fifth Floor, Boston, MA 02110, USA
* 
*/
#pragma once

#include "WindowsIncludes.h"
#include <tbb_thread.h>
#include <task_scheduler_init.h>
#include <concurrent_hash_map.h>
#include <concurrent_vector.h>
#include <task.h>
#include <stdio.h>
#include <string>

#include "Utils.h"
#include "SraNetworkShared.h"
#include "BasicCharacterInfo.h"

// SQL includes:
#include "DbConnectionFactory.h"
#include "DbProcedures.h"
#include "DbStream.h"

using namespace RakNet;
using namespace std;

namespace SraNetwork{

    // The client register class, used as an interface between the database
    // and the world server. Encapsulates all character related functions.
    // TODO: Auto-generate method stubs.
    class ClientRegister
    {
        //Hash-Map compare function for system addresses.
        struct SystemAddressCompare {
            //Simple 'hash', we're simply using the binaryaddress, maybe we need somthing better.
            static size_t hash( const SystemAddress& a)
            {
                return SystemAddress::ToInteger(a);
            }
            //Since the SystemAddress class has a compare function, use this to compare addresses
            static bool equal( const SystemAddress& a, const SystemAddress& b)
            {
                return a.EqualsExcludingPort(b);
            }
        };

    private:
        ClientRegister(void);
        ~ClientRegister(void);
        //tbb::concurrent_vector<SraNetwork::SraClientData> clients;
        tbb::concurrent_hash_map<SystemAddress, SraClientData, SystemAddressCompare>  clientAddressRegister;
        // Singleton-instance yea yea singletons are bad and evil but easy to use ;)
        static ClientRegister* instance;


        //NOTE: Naively removed to ensure it builds in test
        //SraData::DbProcedureManager dbManager;
    public:
        static ClientRegister* getInstance();

        void addClient( SystemAddress address, SraClientData client);

        const SraClientData* getClientFromAddress( SystemAddress address);

        const SystemAddress* getAddressFromClient( const SraClientData* data);

        const SystemAddress* getAddressFromClientID( const int clientID);

        void removeClient(const SystemAddress address);

        void removeClient( const SraClientData* data);

        int loginClient( const string& username, const string& password, SystemAddress address);

        int createCharacterForPlayer(int raceId, SystemAddress address, int clientID );

        void getPlayersForClient( const int clientID, const int* outPlayerIDs);

        bool checkForValidCharacterInfo( const BasicCharacterInfo* info, RakString* resultMessage);

    private:
        int nextID;
    };
}