MunResolverDatabase.hpp

Go to the documentation of this file.

#ifndef __MUN_RESOLVER_DATABASE_HPP__
#define __MUN_RESOLVER_DATABASE_HPP__

#include <map>
#include <mrs.hpp>
#include <sqlite/SqliteAgent.hpp>
#include <sql/SQL.hpp>
#include <stream/MunStream.hpp>
#include <version/MunVersion.hpp>
#include "database/connection/MunResolverProxyConnectInfo.hpp"

class MunResolverDatabase
{
private:
    static  MunVersion                                  s_MunVersion;

    static  SqliteAgent                                 s_ResolverDB;
    
    static  std::map< uint64, ProxyServerConnectInfo >  s_ProxyServerConnectInfo;
    
    
    static  STREAM::ResolverResolveAddressResponse      s_ResolverResolveAddressResponse;
    
    
protected:
    static  ProxyServerConnectInfo* GetProxyServerConnectInfo( uint64 proxyUid ) {
        std::map<uint64, ProxyServerConnectInfo>::iterator itr = s_ProxyServerConnectInfo.find( proxyUid );
        return  ( s_ProxyServerConnectInfo.end() == itr ) ? NULL: &(itr->second);
    }
    
    static  void    AddProxyServerConnectInfo( uint64 proxyUid, ProxyServerConnectInfo& proxyServerConnectInfo ) {
        s_ProxyServerConnectInfo.insert( std::pair<uint64, ProxyServerConnectInfo>( proxyUid, proxyServerConnectInfo ) );
    }
    
    static  void    RemoveProxyServerConnectInfo( uint64 proxyUid ) {
        s_ProxyServerConnectInfo.erase( proxyUid );
    }
    
    static  bool    SetProxyWeight( uint64 proxyUid, int32 weight );
    
    static  void    DeleteProxyWeight( uint64 proxyUid );
    
    static  uint64      SelectProxyUid();
    
public:
    static  bool    InitDB();
    
    static  void    Recv_Resolver_ResolveAddress(MrsConnection pMunClient, STREAM::ResolverResolveAddressRequest& request);
    
    static  void    Recv_ProxyResolver_SetAddress(uint64 connectId, STREAM::ProxyResolverSetAddressRequest& request);
    
    static  void    Recv_ProxyResolver_SetWeight(uint64 connectId, STREAM::ProxyResolverSetWeightRequest& request);
};

#endif  /* __MUN_RESOLVER_DATABASE_HPP__ */