projects/stargazer/plugins/store/mysql/mysql_store.h

   1  /*
   2  $Revision: 1.5 $
   3  $Date: 2010/10/07 19:45:52 $
   4  */
   5
   6
   7 #ifndef MYSQL_STORE_H
   8 #define MYSQL_STORE_H
   9
  10 #include <mysql/mysql.h>
  11
  12 #include <string>
  13 #include <vector>
  14 #include <map>
  15
  16 #include "stg/module_settings.h"
  17 #include "stg/store.h"
  18 #include "stg/user_traff.h"
  19 #include "stg/logger.h"
  20
  21 //-----------------------------------------------------------------------------
  22 class MYSQL_STORE_SETTINGS
  23 {
  24 public:
  25     MYSQL_STORE_SETTINGS();
  26     virtual ~MYSQL_STORE_SETTINGS() {}
  27     virtual int ParseSettings(const MODULE_SETTINGS & s);
  28     virtual const std::string & GetStrError() const { return errorStr; }
  29
  30     const std::string & GetDBUser() const { return dbUser; }
  31     const std::string & GetDBPassword() const { return dbPass; }
  32     const std::string & GetDBHost() const { return dbHost; }
  33     const std::string & GetDBName() const { return dbName; }
  34     unsigned int GetDBPort() const { return dbPort; }
  35
  36 private:
  37     MYSQL_STORE_SETTINGS(const MYSQL_STORE_SETTINGS & rvalue);
  38     MYSQL_STORE_SETTINGS & operator=(const MYSQL_STORE_SETTINGS & rvalue);
  39
  40     const MODULE_SETTINGS * settings;
  41
  42     int     ParseParam(const std::vector<PARAM_VALUE> & moduleParams,
  43                        const std::string & name, std::string & result);
  44
  45     std::string  errorStr;
  46
  47     std::string  dbUser;
  48     std::string  dbPass;
  49     std::string  dbName;
  50     std::string  dbHost;
  51     unsigned int dbPort;
  52 };
  53 //-----------------------------------------------------------------------------
  54 class MYSQL_STORE: public STORE
  55 {
  56 public:
  57     MYSQL_STORE();
  58     virtual ~MYSQL_STORE() {}
  59     virtual const std::string & GetStrError() const { return errorStr; }
  60
  61     //User
  62     virtual int GetUsersList(std::vector<std::string> * usersList) const;
  63     virtual int AddUser(const std::string & login) const;
  64     virtual int DelUser(const std::string & login) const;
  65     virtual int SaveUserStat(const USER_STAT & stat, const std::string & login) const;
  66     virtual int SaveUserConf(const USER_CONF & conf, const std::string & login) const;
  67     virtual int RestoreUserStat(USER_STAT * stat, const std::string & login) const;
  68     virtual int RestoreUserConf(USER_CONF * conf, const std::string & login) const;
  69     virtual int WriteUserChgLog(const std::string & login,
  70                                 const std::string & admLogin,
  71                                 uint32_t       admIP,
  72                                 const std::string & paramName,
  73                                 const std::string & oldValue,
  74                                 const std::string & newValue,
  75                                 const std::string & message = "") const;
  76     virtual int WriteUserConnect(const std::string & login, uint32_t ip) const;
  77     virtual int WriteUserDisconnect(const std::string & login,
  78                                     const DIR_TRAFF & up,
  79                                     const DIR_TRAFF & down,
  80                                     const DIR_TRAFF & sessionUp,
  81                                     const DIR_TRAFF & sessionDown,
  82                                     double cash,
  83                                     double freeMb,
  84                                     const std::string & reason) const;
  85
  86     virtual int WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
  87                                   time_t lastStat,
  88                                   const std::string & login) const;
  89
  90     virtual int AddMessage(STG_MSG * msg, const std::string & login) const;
  91     virtual int EditMessage(const STG_MSG & msg, const std::string & login) const;
  92     virtual int GetMessage(uint64_t id, STG_MSG * msg, const std::string & login) const;
  93     virtual int DelMessage(uint64_t id, const std::string & login) const;
  94     virtual int GetMessageHdrs(std::vector<STG_MSG_HDR> * hdrsList, const std::string & login) const;
  95
  96     virtual int SaveMonthStat(const USER_STAT & stat, int month, int year, const std::string & login) const;
  97
  98     //Admin
  99     virtual int GetAdminsList(std::vector<std::string> * adminsList) const;
 100     virtual int AddAdmin(const std::string & login) const;
 101     virtual int DelAdmin(const std::string & login) const;
 102     virtual int RestoreAdmin(ADMIN_CONF * ac, const std::string & login) const;
 103     virtual int SaveAdmin(const ADMIN_CONF & ac) const;
 104
 105     //Tariff
 106     virtual int GetTariffsList(std::vector<std::string> * tariffsList) const;
 107     virtual int AddTariff(const std::string & name) const;
 108     virtual int DelTariff(const std::string & name) const;
 109     virtual int SaveTariff(const TARIFF_DATA & td, const std::string & tariffName) const;
 110     virtual int RestoreTariff(TARIFF_DATA * td, const std::string & tariffName) const;
 111
 112     //Corparation
 113     virtual int GetCorpsList(std::vector<std::string> *) const {return 0;}
 114     virtual int SaveCorp(const CORP_CONF &) const {return 0;}
 115     virtual int RestoreCorp(CORP_CONF *, const std::string &) const {return 0;}
 116     virtual int AddCorp(const std::string &) const {return 0;}
 117     virtual int DelCorp(const std::string &) const {return 0;}
 118
 119     // Services
 120     virtual int GetServicesList(std::vector<std::string> *) const {return 0;}
 121     virtual int SaveService(const SERVICE_CONF &) const {return 0;}
 122     virtual int RestoreService(SERVICE_CONF *, const std::string &) const {return 0;}
 123     virtual int AddService(const std::string &) const {return 0;}
 124     virtual int DelService(const std::string &) const {return 0;}
 125
 126     virtual void            SetSettings(const MODULE_SETTINGS & s) { settings = s; }
 127     virtual int             ParseSettings();
 128     virtual const std::string &  GetVersion() const { return version; }
 129
 130 private:
 131     MYSQL_STORE(const MYSQL_STORE & rvalue);
 132     MYSQL_STORE & operator=(const MYSQL_STORE & rvalue);
 133
 134     virtual int WriteLogString(const std::string & str, const std::string & login) const;
 135     int GetAllParams(std::vector<std::string> * ParamList, const std::string & table, const std::string & name) const;
 136     int CheckAllTables(MYSQL * sock);
 137     int MakeUpdates(MYSQL * sock);
 138     bool IsTablePresent(const std::string & str,MYSQL * sock);
 139     mutable std::string          errorStr;
 140     int                        MysqlQuery(const char* sQuery,MYSQL * sock) const;
 141     int                     MysqlGetQuery(const char * Query,MYSQL * & sock) const;
 142     int                     MysqlSetQuery(const char * Query) const;
 143     MYSQL  *                MysqlConnect() const ;
 144     std::string                  version;
 145     MYSQL_STORE_SETTINGS    storeSettings;
 146     MODULE_SETTINGS         settings;
 147     int                     schemaVersion;
 148
 149     PLUGIN_LOGGER           logger;
 150 };
 151 //-----------------------------------------------------------------------------
 152
 153 #endif