ClientRegister.h

Go to the documentation of this file.

/*
* Client register - Stores active clients
*
* @Author Michael Matzen <mim@informatik.uni-kiel.de>
*
* 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

#define _WIN32_WINNT 0x0600 

#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 "SraNetworkShared.h"


namespace SraNetwork{
    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 a.binaryAddress;
            }
            //Since the SystemAddress class has a compare function, use this to compare addresses
            static bool equal( const SystemAddress& a, const SystemAddress& b)
            {
                return a == 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( wchar_t* data, SystemAddress address, int userLen);
    private:
        int nextID;
    };
}