]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/store/postgresql/postgresql_store.cpp
ba47896694ec5f61c9a7d7930614a117661524ad
[stg.git] / projects / stargazer / plugins / store / postgresql / postgresql_store.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
19  */
20
21 /*
22  *  This file contains a realization of a base postgresql-storage plugin class
23  *
24  *  v. 1.3
25  *  FreeMb logging on disconnects added
26  *
27  *  v. 1.2
28  *  Reconnection on faults added
29  *
30  *  v. 1.1
31  *  tb_stats removed
32  *
33  *  v. 1.0
34  *  Initial implementation
35  *
36  *  $Revision: 1.5 $
37  *  $Date: 2010/01/06 10:43:48 $
38  *
39  */
40
41 #include "postgresql_store.h"
42
43 #include "stg/common.h" // str2x, printfd
44
45 #include <string>
46 #include <vector>
47
48 #include <libpq-fe.h>
49
50 extern "C" STG::Store* GetStore()
51 {
52     static POSTGRESQL_STORE plugin;
53     return &plugin;
54 }
55
56 //-----------------------------------------------------------------------------
57 POSTGRESQL_STORE::POSTGRESQL_STORE()
58     : versionString("postgresql_store v.1.3"),
59       server("localhost"),
60       database("stargazer"),
61       user("stg"),
62       password("123456"),
63       clientEncoding("KOI8"),
64       version(0),
65       retries(3),
66       connection(NULL),
67       logger(STG::PluginLogger::get("store_postgresql"))
68 {
69 pthread_mutex_init(&mutex, NULL);
70 }
71 //-----------------------------------------------------------------------------
72 POSTGRESQL_STORE::~POSTGRESQL_STORE()
73 {
74 if (connection)
75     {
76     PQfinish(connection);
77     }
78 pthread_mutex_destroy(&mutex);
79 }
80 //-----------------------------------------------------------------------------
81 int POSTGRESQL_STORE::ParseSettings()
82 {
83 std::vector<STG::ParamValue>::iterator i;
84
85 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
86     {
87     if (i->value.empty())
88         continue;
89     std::string s = ToLower(i->param);
90     if (s == "server")
91         {
92         server = i->value.front();
93         }
94     if (s == "database")
95         {
96         database = i->value.front();
97         }
98     if (s == "user")
99         {
100         user = i->value.front();
101         }
102     if (s == "password")
103         {
104         password = i->value.front();
105         }
106     if (s == "retries")
107         {
108         if (str2x(i->value.front(), retries))
109             {
110             strError = "Invalid 'retries' value";
111             printfd(__FILE__, "POSTGRESQL_STORE::ParseSettings(): '%s'\n", strError.c_str());
112             return -1;
113             }
114         }
115     }
116
117 clientEncoding = "KOI8";
118
119 return Connect();
120 }
121 //-----------------------------------------------------------------------------
122 int POSTGRESQL_STORE::Connect()
123 {
124 std::string params;
125 params = "host=" + server + " "
126        + "dbname=" + database + " "
127        + "user=" + user + " "
128        + "password=" + password;
129
130 connection = PQconnectdb(params.c_str());
131
132 if (PQstatus(connection) != CONNECTION_OK)
133     {
134     strError = PQerrorMessage(connection);
135     printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());
136     // Will try to connect later
137     return 0;
138     }
139
140 if (PQsetClientEncoding(connection, clientEncoding.c_str()))
141     {
142     strError = PQerrorMessage(connection);
143     printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());
144     return 1;
145     }
146
147 return CheckVersion();
148 }
149 //-----------------------------------------------------------------------------
150 int POSTGRESQL_STORE::Reset() const
151 {
152 for (int i = 0; i < retries && PQstatus(connection) != CONNECTION_OK; ++i)
153     {
154     struct timespec ts = {1, 0};
155     nanosleep(&ts, NULL);
156     PQreset(connection);
157     }
158
159 if (PQstatus(connection) != CONNECTION_OK)
160     {
161     strError = PQerrorMessage(connection);
162     printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());
163     return 1;
164     }
165
166 if (PQsetClientEncoding(connection, clientEncoding.c_str()))
167     {
168     strError = PQerrorMessage(connection);
169     printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());
170     return -1;
171     }
172
173 return CheckVersion();
174 }
175 //-----------------------------------------------------------------------------
176 int POSTGRESQL_STORE::CheckVersion() const
177 {
178
179 if (StartTransaction())
180     {
181     strError = "Failed to start transaction";
182     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
183     return -1;
184     }
185
186 PGresult * result = PQexec(connection, "SELECT MAX(version) FROM tb_info");
187
188 if (PQresultStatus(result) != PGRES_TUPLES_OK)
189     {
190     strError = PQresultErrorMessage(result);
191     PQclear(result);
192     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n");
193     RollbackTransaction();
194     return -1;
195     }
196
197 if (str2x(PQgetvalue(result, 0, 0), version))
198     {
199     strError = "Invalid DB version";
200     PQclear(result);
201     RollbackTransaction();
202     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
203     return -1;
204     }
205
206 PQclear(result);
207
208 if (version < DB_MIN_VERSION)
209     {
210     strError = "DB version too old";
211     RollbackTransaction();
212     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
213     return -1;
214     }
215
216 if (version < 6)
217     {
218     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): I recommend you to upgrade your DB to higher version to support FreeMb logging on disconnect. Current version is %d\n", version);
219     }
220
221 if (CommitTransaction())
222     {
223     strError = "Failed to commit transaction";
224     printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
225     return -1;
226     }
227
228 logger("POSTGRESQL_STORE: Current DB schema version: %d", version);
229
230 return 0;
231 }
232 //-----------------------------------------------------------------------------