stargazer/plugins/store/postgresql/postgresql_store_services.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 /*
  23  *  Services manipulation methods
  24  *
  25  *  $Revision: 1.2 $
  26  *  $Date: 2009/06/09 12:32:40 $
  27  *
  28  */
  29
  30 #include <string>
  31 #include <vector>
  32 #include <sstream>
  33
  34 #include <libpq-fe.h>
  35
  36 #include "postgresql_store.h"
  37 #include "stg/locker.h"
  38
  39 //-----------------------------------------------------------------------------
  40 int POSTGRESQL_STORE::GetServicesList(std::vector<std::string> * servicesList) const
  41 {
  42 STG_LOCKER lock(&mutex);
  43
  44 if (PQstatus(connection) != CONNECTION_OK)
  45     {
  46     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
  47     if (Reset())
  48         {
  49         strError = "Connection lost";
  50         printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): '%s'\n", strError.c_str());
  51         return -1;
  52         }
  53     }
  54
  55 PGresult * result;
  56
  57 if (StartTransaction())
  58     {
  59     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to start transaction'\n");
  60     return -1;
  61     }
  62
  63 result = PQexec(connection, "SELECT name FROM tb_services");
  64
  65 if (PQresultStatus(result) != PGRES_TUPLES_OK)
  66     {
  67     strError = PQresultErrorMessage(result);
  68     PQclear(result);
  69     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): '%s'\n", strError.c_str());
  70     if (RollbackTransaction())
  71         {
  72         printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to rollback transaction'\n");
  73         }
  74     return -1;
  75     }
  76
  77 int tuples = PQntuples(result);
  78
  79 for (int i = 0; i < tuples; ++i)
  80     {
  81     servicesList->push_back(PQgetvalue(result, i, 0));
  82     }
  83
  84 PQclear(result);
  85
  86 if (CommitTransaction())
  87     {
  88     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to commit transaction'\n");
  89     return -1;
  90     }
  91
  92 return 0;
  93 }
  94
  95 //-----------------------------------------------------------------------------
  96 int POSTGRESQL_STORE::SaveService(const SERVICE_CONF & sc) const
  97 {
  98 STG_LOCKER lock(&mutex);
  99
 100 if (PQstatus(connection) != CONNECTION_OK)
 101     {
 102     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 103     if (Reset())
 104         {
 105         strError = "Connection lost";
 106         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): '%s'\n", strError.c_str());
 107         return -1;
 108         }
 109     }
 110
 111 PGresult * result;
 112
 113 if (StartTransaction())
 114     {
 115     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to start transaction'\n");
 116     return -1;
 117     }
 118
 119 std::string ename = sc.name;
 120 std::string ecomment = sc.comment;
 121
 122 if (EscapeString(ename))
 123     {
 124     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to escape name'\n");
 125     if (RollbackTransaction())
 126         {
 127         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
 128         }
 129     return -1;
 130     }
 131
 132 if (EscapeString(ecomment))
 133     {
 134     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to escape comment'\n");
 135     if (RollbackTransaction())
 136         {
 137         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
 138         }
 139     return -1;
 140     }
 141
 142 std::ostringstream query;
 143 query << "UPDATE tb_services SET "
 144           << "comment = '" << ecomment << "', "
 145           << "cost = " << sc.cost << ", "
 146           << "pay_day = " << sc.payDay << " "
 147       << "WHERE name = '" << ename << "'";
 148
 149 result = PQexec(connection, query.str().c_str());
 150
 151 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 152     {
 153     strError = PQresultErrorMessage(result);
 154     PQclear(result);
 155     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): '%s'\n", strError.c_str());
 156     if (RollbackTransaction())
 157         {
 158         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
 159         }
 160     return -1;
 161     }
 162
 163 PQclear(result);
 164
 165 if (CommitTransaction())
 166     {
 167     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to commit transaction'\n");
 168     return -1;
 169     }
 170
 171 return 0;
 172 }
 173
 174 //-----------------------------------------------------------------------------
 175 int POSTGRESQL_STORE::RestoreService(SERVICE_CONF * sc,
 176                                    const std::string & name) const
 177 {
 178 STG_LOCKER lock(&mutex);
 179
 180 if (PQstatus(connection) != CONNECTION_OK)
 181     {
 182     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 183     if (Reset())
 184         {
 185         strError = "Connection lost";
 186         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): '%s'\n", strError.c_str());
 187         return -1;
 188         }
 189     }
 190
 191 PGresult * result;
 192
 193 if (StartTransaction())
 194     {
 195     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to start transaction'\n");
 196     return -1;
 197     }
 198
 199 std::string ename = name;
 200
 201 if (EscapeString(ename))
 202     {
 203     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to escape name'\n");
 204     if (RollbackTransaction())
 205         {
 206         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
 207         }
 208     return -1;
 209     }
 210
 211 std::ostringstream query;
 212 query << "SELECT comment, cost, pay_day FROM tb_services WHERE name = '" << ename << "'";
 213
 214 result = PQexec(connection, query.str().c_str());
 215
 216 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 217     {
 218     strError = PQresultErrorMessage(result);
 219     PQclear(result);
 220     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): '%s'\n", strError.c_str());
 221     if (RollbackTransaction())
 222         {
 223         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
 224         }
 225     return -1;
 226     }
 227
 228 int tuples = PQntuples(result);
 229
 230 if (tuples != 1)
 231     {
 232     strError = "Failed to fetch service's data";
 233     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 234     PQclear(result);
 235     if (RollbackTransaction())
 236         {
 237         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
 238         }
 239     return -1;
 240     }
 241
 242 std::stringstream tuple;
 243 tuple << PQgetvalue(result, 0, 0) << " "
 244       << PQgetvalue(result, 0, 1) << " "
 245       << PQgetvalue(result, 0, 2);
 246
 247 PQclear(result);
 248
 249 tuple >> sc->comment
 250       >> sc->cost
 251       >> sc->payDay;
 252
 253 if (CommitTransaction())
 254     {
 255     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to commit transaction'\n");
 256     return -1;
 257     }
 258
 259 return 0;
 260 }
 261
 262 //-----------------------------------------------------------------------------
 263 int POSTGRESQL_STORE::AddService(const std::string & name) const
 264 {
 265 STG_LOCKER lock(&mutex);
 266
 267 if (PQstatus(connection) != CONNECTION_OK)
 268     {
 269     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 270     if (Reset())
 271         {
 272         strError = "Connection lost";
 273         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): '%s'\n", strError.c_str());
 274         return -1;
 275         }
 276     }
 277
 278 PGresult * result;
 279
 280 if (StartTransaction())
 281     {
 282     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to start transaction'\n");
 283     return -1;
 284     }
 285
 286 std::string ename = name;
 287
 288 if (EscapeString(ename))
 289     {
 290     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to escape name'\n");
 291     if (RollbackTransaction())
 292         {
 293         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to rollback transaction'\n");
 294         }
 295     return -1;
 296     }
 297
 298 std::ostringstream query;
 299 query << "INSERT INTO tb_services \
 300               (name, comment, cost, pay_day) \
 301           VALUES \
 302               ('" << ename << "', '', 0, 0)";
 303
 304 result = PQexec(connection, query.str().c_str());
 305
 306 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 307     {
 308     strError = PQresultErrorMessage(result);
 309     PQclear(result);
 310     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): '%s'\n", strError.c_str());
 311     if (RollbackTransaction())
 312         {
 313         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to rollback transaction'\n");
 314         }
 315     return -1;
 316     }
 317
 318 PQclear(result);
 319
 320 if (CommitTransaction())
 321     {
 322     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to commit transaction'\n");
 323     return -1;
 324     }
 325
 326 return 0;
 327 }
 328
 329 //-----------------------------------------------------------------------------
 330 int POSTGRESQL_STORE::DelService(const std::string & name) const
 331 {
 332 STG_LOCKER lock(&mutex);
 333
 334 if (PQstatus(connection) != CONNECTION_OK)
 335     {
 336     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 337     if (Reset())
 338         {
 339         strError = "Connection lost";
 340         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): '%s'\n", strError.c_str());
 341         return -1;
 342         }
 343     }
 344
 345 PGresult * result;
 346
 347 if (StartTransaction())
 348     {
 349     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to start transaction'\n");
 350     return -1;
 351     }
 352
 353 std::string ename = name;
 354
 355 if (EscapeString(ename))
 356     {
 357     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to escape name'\n");
 358     if (RollbackTransaction())
 359         {
 360         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to rollback transaction'\n");
 361         }
 362     return -1;
 363     }
 364
 365 std::ostringstream query;
 366 query << "DELETE FROM tb_services WHERE name = '" << ename << "'";
 367
 368 result = PQexec(connection, query.str().c_str());
 369
 370 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 371     {
 372     strError = PQresultErrorMessage(result);
 373     PQclear(result);
 374     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): '%s'\n", strError.c_str());
 375     if (RollbackTransaction())
 376         {
 377         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to rollback transaction'\n");
 378         }
 379     return -1;
 380     }
 381
 382 PQclear(result);
 383
 384 if (CommitTransaction())
 385     {
 386     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to commit transaction'\n");
 387     return -1;
 388     }
 389
 390 return 0;
 391 }
 392 //-----------------------------------------------------------------------------
 393