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