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.
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.
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
18 * Author : Maxim Mamontov <faust@stargazer.dp.ua>
22 * This file contains a realization of a base postgresql-storage plugin class
25 * FreeMb logging on disconnects added
28 * Reconnection on faults added
34 * Initial implementation
37 * $Date: 2010/01/06 10:43:48 $
47 #include "postgresql_store.h"
48 #include "postgresql_store_utils.h"
49 #include "stg/module_settings.h"
51 class POSTGRESQL_STORE_CREATOR
54 POSTGRESQL_STORE_CREATOR()
55 : pqStore(new POSTGRESQL_STORE())
58 ~POSTGRESQL_STORE_CREATOR()
62 POSTGRESQL_STORE * GetStore() { return pqStore; };
64 POSTGRESQL_STORE * pqStore;
67 //-----------------------------------------------------------------------------
70 return pqStoreeCreator.GetStore();
73 //-----------------------------------------------------------------------------
74 POSTGRESQL_STORE::POSTGRESQL_STORE()
75 : versionString("postgresql_store v.1.3"),
77 database("stargazer"),
84 pthread_mutex_init(&mutex, NULL);
86 //-----------------------------------------------------------------------------
87 POSTGRESQL_STORE::~POSTGRESQL_STORE()
93 pthread_mutex_destroy(&mutex);
95 //-----------------------------------------------------------------------------
96 int POSTGRESQL_STORE::ParseSettings()
98 std::vector<PARAM_VALUE>::iterator i;
101 for(i = settings.moduleParams.begin(); i != settings.moduleParams.end(); ++i)
104 std::transform(s.begin(), s.end(), s.begin(), ToLower());
107 server = *(i->value.begin());
111 database = *(i->value.begin());
115 user = *(i->value.begin());
119 password = *(i->value.begin());
123 if (str2x(*(i->value.begin()), retries))
125 strError = "Invalid 'retries' value";
126 printfd(__FILE__, "POSTGRESQL_STORE::ParseSettings(): '%s'\n", strError.c_str());
132 clientEncoding = "KOI8";
136 //-----------------------------------------------------------------------------
137 int POSTGRESQL_STORE::Connect()
140 params = "host=" + server + " "
141 + "dbname=" + database + " "
142 + "user=" + user + " "
143 + "password=" + password;
145 connection = PQconnectdb(params.c_str());
147 if (PQstatus(connection) != CONNECTION_OK)
149 strError = PQerrorMessage(connection);
150 printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());
151 // Will try to connect later
155 if (PQsetClientEncoding(connection, clientEncoding.c_str()))
157 strError = PQerrorMessage(connection);
158 printfd(__FILE__, "POSTGRESQL_STORE::Connect(): '%s'\n", strError.c_str());
162 return CheckVersion();
164 //-----------------------------------------------------------------------------
165 int POSTGRESQL_STORE::Reset() const
167 for (int i = 0; i < retries && PQstatus(connection) != CONNECTION_OK; ++i)
169 struct timespec ts = {1, 0};
170 nanosleep(&ts, NULL);
174 if (PQstatus(connection) != CONNECTION_OK)
176 strError = PQerrorMessage(connection);
177 printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());
181 if (PQsetClientEncoding(connection, clientEncoding.c_str()))
183 strError = PQerrorMessage(connection);
184 printfd(__FILE__, "POSTGRESQL_STORE::Reset(): '%s'\n", strError.c_str());
188 return CheckVersion();
190 //-----------------------------------------------------------------------------
191 int POSTGRESQL_STORE::CheckVersion() const
194 if (StartTransaction())
196 strError = "Failed to start transaction";
197 printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
201 PGresult * result = PQexec(connection, "SELECT MAX(version) FROM tb_info");
203 if (PQresultStatus(result) != PGRES_TUPLES_OK)
205 strError = PQresultErrorMessage(result);
207 printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n");
208 RollbackTransaction();
212 if (str2x(PQgetvalue(result, 0, 0), version))
214 strError = "Invalid DB version";
216 RollbackTransaction();
217 printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
223 if (version < DB_MIN_VERSION)
225 strError = "DB version too old";
226 RollbackTransaction();
227 printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
233 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);
236 if (CommitTransaction())
238 strError = "Failed to commit transaction";
239 printfd(__FILE__, "POSTGRESQL_STORE::CheckVersion(): '%s'\n", strError.c_str());
245 //-----------------------------------------------------------------------------