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