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 //-----------------------------------------------------------------------------