3 $Date: 2010/10/07 19:45:52 $
10 #include <mysql/mysql.h>
16 #include "stg/module_settings.h"
17 #include "stg/store.h"
18 #include "stg/user_traff.h"
19 #include "stg/logger.h"
21 //-----------------------------------------------------------------------------
22 class MYSQL_STORE_SETTINGS
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; }
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; }
37 MYSQL_STORE_SETTINGS(const MYSQL_STORE_SETTINGS & rvalue);
38 MYSQL_STORE_SETTINGS & operator=(const MYSQL_STORE_SETTINGS & rvalue);
40 const MODULE_SETTINGS * settings;
42 int ParseParam(const std::vector<PARAM_VALUE> & moduleParams,
43 const std::string & name, std::string & result);
53 //-----------------------------------------------------------------------------
54 class MYSQL_STORE: public STORE
58 virtual ~MYSQL_STORE() {}
59 virtual const std::string & GetStrError() const { return errorStr; }
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,
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,
79 const DIR_TRAFF & down,
80 const DIR_TRAFF & sessionUp,
81 const DIR_TRAFF & sessionDown,
84 const std::string & reason) const;
86 virtual int WriteDetailedStat(const std::map<IP_DIR_PAIR, STAT_NODE> & statTree,
88 const std::string & login) const;
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;
96 virtual int SaveMonthStat(const USER_STAT & stat, int month, int year, const std::string & login) const;
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;
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;
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;}
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;}
126 virtual void SetSettings(const MODULE_SETTINGS & s) { settings = s; }
127 virtual int ParseSettings();
128 virtual const std::string & GetVersion() const { return version; }
131 MYSQL_STORE(const MYSQL_STORE & rvalue);
132 MYSQL_STORE & operator=(const MYSQL_STORE & rvalue);
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 ;
145 MYSQL_STORE_SETTINGS storeSettings;
146 MODULE_SETTINGS settings;
149 PLUGIN_LOGGER logger;
151 //-----------------------------------------------------------------------------