projects/stargazer/plugins/store/postgresql/postgresql_store_tariffs.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  *  Tariffs manipulation methods
  23  *
  24  *  $Revision: 1.2 $
  25  *  $Date: 2009/06/09 12:32:40 $
  26  *
  27  */
  28
  29 #include <string>
  30 #include <vector>
  31 #include <sstream>
  32 #include <cmath>
  33
  34 #include <libpq-fe.h>
  35
  36 #include "postgresql_store.h"
  37 #include "stg/locker.h"
  38
  39 //-----------------------------------------------------------------------------
  40 int POSTGRESQL_STORE::GetTariffsList(std::vector<std::string> * tariffsList) const
  41 {
  42 STG_LOCKER lock(&mutex);
  43
  44 if (PQstatus(connection) != CONNECTION_OK)
  45     {
  46     printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
  47     if (Reset())
  48         {
  49         strError = "Connection lost";
  50         printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): '%s'\n", strError.c_str());
  51         return -1;
  52         }
  53     }
  54
  55 PGresult * result;
  56
  57 if (StartTransaction())
  58     {
  59     printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): 'Failed to start transaction'\n");
  60     return -1;
  61     }
  62
  63 result = PQexec(connection, "SELECT name FROM tb_tariffs");
  64
  65 if (PQresultStatus(result) != PGRES_TUPLES_OK)
  66     {
  67     strError = PQresultErrorMessage(result);
  68     PQclear(result);
  69     printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): '%s'\n", strError.c_str());
  70     if (RollbackTransaction())
  71         {
  72         printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): '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     tariffsList->push_back(PQgetvalue(result, i, 0));
  82     }
  83
  84 PQclear(result);
  85
  86 if (CommitTransaction())
  87     {
  88     printfd(__FILE__, "POSTGRESQL_STORE::GetTariffsList(): 'Failed to commit transaction'\n");
  89     return -1;
  90     }
  91
  92 return 0;
  93 }
  94
  95 //-----------------------------------------------------------------------------
  96 int POSTGRESQL_STORE::AddTariff(const std::string & name) const
  97 {
  98 STG_LOCKER lock(&mutex);
  99
 100 if (PQstatus(connection) != CONNECTION_OK)
 101     {
 102     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 103     if (Reset())
 104         {
 105         strError = "Connection lost";
 106         printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): '%s'\n", strError.c_str());
 107         return -1;
 108         }
 109     }
 110
 111 PGresult * result;
 112
 113 if (StartTransaction())
 114     {
 115     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to start transaction'\n");
 116     return -1;
 117     }
 118
 119 std::string ename = name;
 120
 121 if (EscapeString(ename))
 122     {
 123     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to escape name'\n");
 124     if (RollbackTransaction())
 125         {
 126         printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to rollback transaction'\n");
 127         }
 128     return -1;
 129     }
 130
 131 std::ostringstream query;
 132 query << "SELECT sp_add_tariff('" << ename << "', " << DIR_NUM << ")";
 133
 134 result = PQexec(connection, query.str().c_str());
 135
 136 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 137     {
 138     strError = PQresultErrorMessage(result);
 139     PQclear(result);
 140     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): '%s'\n", strError.c_str());
 141     if (RollbackTransaction())
 142         {
 143         printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to rollback transaction'\n");
 144         }
 145     return -1;
 146     }
 147
 148 PQclear(result);
 149
 150 if (CommitTransaction())
 151     {
 152     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to commit transaction'\n");
 153     return -1;
 154     }
 155
 156 return 0;
 157 }
 158 //-----------------------------------------------------------------------------
 159 int POSTGRESQL_STORE::DelTariff(const std::string & name) const
 160 {
 161 STG_LOCKER lock(&mutex);
 162
 163 if (PQstatus(connection) != CONNECTION_OK)
 164     {
 165     printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 166     if (Reset())
 167         {
 168         strError = "Connection lost";
 169         printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): '%s'\n", strError.c_str());
 170         return -1;
 171         }
 172     }
 173
 174 PGresult * result;
 175
 176 if (StartTransaction())
 177     {
 178     printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): 'Failed to start transaction'\n");
 179     return -1;
 180     }
 181
 182 std::string ename = name;
 183
 184 if (EscapeString(ename))
 185     {
 186     printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to escape name'\n");
 187     if (RollbackTransaction())
 188         {
 189         printfd(__FILE__, "POSTGRESQL_STORE::AddTariff(): 'Failed to rollback transaction'\n");
 190         }
 191     return -1;
 192     }
 193
 194 std::ostringstream query;
 195 query << "DELETE FROM tb_tariffs WHERE name = '" << ename << "'";
 196
 197 result = PQexec(connection, query.str().c_str());
 198
 199 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 200     {
 201     strError = PQresultErrorMessage(result);
 202     PQclear(result);
 203     printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): '%s'\n", strError.c_str());
 204     if (RollbackTransaction())
 205         {
 206         printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): 'Failed to rollback transaction'\n");
 207         }
 208     return -1;
 209     }
 210
 211 PQclear(result);
 212
 213 if (CommitTransaction())
 214     {
 215     printfd(__FILE__, "POSTGRESQL_STORE::DelTariff(): 'Failed to commit transaction'\n");
 216     return -1;
 217     }
 218
 219 return 0;
 220 }
 221 //-----------------------------------------------------------------------------
 222 int POSTGRESQL_STORE::SaveTariff(const TARIFF_DATA & td,
 223                                  const std::string & tariffName) const
 224 {
 225 STG_LOCKER lock(&mutex);
 226
 227 if (PQstatus(connection) != CONNECTION_OK)
 228     {
 229     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 230     if (Reset())
 231         {
 232         strError = "Connection lost";
 233         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): '%s'\n", strError.c_str());
 234         return -1;
 235         }
 236     }
 237
 238 PGresult * result;
 239
 240 if (StartTransaction())
 241     {
 242     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to start transaction'\n");
 243     return -1;
 244     }
 245
 246 std::string ename = tariffName;
 247
 248 if (EscapeString(ename))
 249     {
 250     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to escape name'\n");
 251     if (RollbackTransaction())
 252         {
 253         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to rollback transaction'\n");
 254         }
 255     return -1;
 256     }
 257
 258     {
 259     std::ostringstream query;
 260     query << "SELECT pk_tariff FROM tb_tariffs WHERE name = '" << ename << "'";
 261
 262     result = PQexec(connection, query.str().c_str());
 263     }
 264
 265 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 266     {
 267     strError = PQresultErrorMessage(result);
 268     PQclear(result);
 269     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): '%s'\n", strError.c_str());
 270     if (RollbackTransaction())
 271         {
 272         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to rollback transaction'\n");
 273         }
 274     return -1;
 275     }
 276
 277 int tuples = PQntuples(result);
 278
 279 if (tuples != 1)
 280     {
 281     strError = "Failed to fetch tariff ID";
 282     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 283     PQclear(result);
 284     if (RollbackTransaction())
 285         {
 286         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to rollback transaction'\n");
 287         }
 288     return -1;
 289     }
 290
 291 int32_t id;
 292
 293     {
 294     std::stringstream tuple;
 295     tuple << PQgetvalue(result, 0, 0);
 296
 297     PQclear(result);
 298
 299     tuple >> id;
 300     }
 301
 302     {
 303     std::ostringstream query;
 304     query << "UPDATE tb_tariffs SET \
 305                   fee = " << td.tariffConf.fee << ", \
 306                   free = " << td.tariffConf.free << ", \
 307                   passive_cost = " << td.tariffConf.passiveCost << ", \
 308                   traff_type = " << td.tariffConf.traffType;
 309
 310     if (version > 6)
 311         query << ", period = '" << TARIFF::PeriodToString(td.tariffConf.period) << "'";
 312
 313     query << " WHERE pk_tariff = " << id;
 314
 315     result = PQexec(connection, query.str().c_str());
 316     }
 317
 318 if (PQresultStatus(result) != PGRES_COMMAND_OK)
 319     {
 320     strError = PQresultErrorMessage(result);
 321     PQclear(result);
 322     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): '%s'\n", strError.c_str());
 323     if (RollbackTransaction())
 324         {
 325         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to rollback transaction'\n");
 326         }
 327     return -1;
 328     }
 329
 330 PQclear(result);
 331
 332 for(int i = 0; i < DIR_NUM; i++)
 333     {
 334     double pda = td.dirPrice[i].priceDayA * 1024 * 1024;
 335     double pdb = td.dirPrice[i].priceDayB * 1024 * 1024;
 336     double pna = 0;
 337     double pnb = 0;
 338
 339     if (td.dirPrice[i].singlePrice)
 340         {
 341         pna = pda;
 342         pnb = pdb;
 343         }
 344     else
 345         {
 346         pna = td.dirPrice[i].priceNightA * 1024 * 1024;
 347         pnb = td.dirPrice[i].priceNightB * 1024 * 1024;
 348         }
 349
 350     int threshold = 0;
 351     if (td.dirPrice[i].noDiscount)
 352         {
 353         threshold = 0xffFFffFF;
 354         }
 355     else
 356         {
 357         threshold = td.dirPrice[i].threshold;
 358         }
 359
 360         {
 361         std::ostringstream query;
 362         query << "UPDATE tb_tariffs_params SET \
 363                       price_day_a = " << pda << ", \
 364                       price_day_b = " << pdb << ", \
 365                       price_night_a = " << pna << ", \
 366                       price_night_b = " << pnb << ", \
 367                       threshold = " << threshold << ", \
 368                       time_day_begins = CAST('" << td.dirPrice[i].hDay
 369                                                 << ":"
 370                                                 << td.dirPrice[i].mDay << "' AS TIME), \
 371                       time_day_ends = CAST('" << td.dirPrice[i].hNight
 372                                               << ":"
 373                                               << td.dirPrice[i].mNight << "' AS TIME) \
 374                  WHERE fk_tariff = " << id << " AND dir_num = " << i;
 375
 376         result = PQexec(connection, query.str().c_str());
 377         }
 378
 379     if (PQresultStatus(result) != PGRES_COMMAND_OK)
 380         {
 381         strError = PQresultErrorMessage(result);
 382         PQclear(result);
 383         printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): '%s'\n", strError.c_str());
 384         if (RollbackTransaction())
 385             {
 386             printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to rollback transaction'\n");
 387             }
 388         return -1;
 389         }
 390
 391     PQclear(result);
 392     }
 393
 394 if (CommitTransaction())
 395     {
 396     printfd(__FILE__, "POSTGRESQL_STORE::SaveTariff(): 'Failed to commit transaction'\n");
 397     return -1;
 398     }
 399
 400 return 0;
 401 }
 402 //-----------------------------------------------------------------------------
 403 int POSTGRESQL_STORE::RestoreTariff(TARIFF_DATA * td,
 404                                   const std::string & tariffName) const
 405 {
 406 STG_LOCKER lock(&mutex);
 407
 408 if (PQstatus(connection) != CONNECTION_OK)
 409     {
 410     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
 411     if (Reset())
 412         {
 413         strError = "Connection lost";
 414         printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): '%s'\n", strError.c_str());
 415         return -1;
 416         }
 417     }
 418
 419 PGresult * result;
 420
 421 if (StartTransaction())
 422     {
 423     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to start transaction'\n");
 424     return -1;
 425     }
 426
 427 std::string ename = tariffName;
 428
 429 if (EscapeString(ename))
 430     {
 431     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to escape name'\n");
 432     if (RollbackTransaction())
 433         {
 434         printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to rollback transaction'\n");
 435         }
 436     return -1;
 437     }
 438
 439 td->tariffConf.name = tariffName;
 440
 441 std::ostringstream query;
 442 query << "SELECT pk_tariff, \
 443                  fee, \
 444                  free, \
 445                  passive_cost, \
 446                  traff_type";
 447
 448 if (version > 6)
 449     query << ", period";
 450
 451 query << " FROM tb_tariffs WHERE name = '" << ename << "'";
 452
 453 result = PQexec(connection, query.str().c_str());
 454
 455 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 456     {
 457     strError = PQresultErrorMessage(result);
 458     PQclear(result);
 459     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): '%s'\n", strError.c_str());
 460     if (RollbackTransaction())
 461         {
 462         printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to rollback transaction'\n");
 463         }
 464     return -1;
 465     }
 466
 467 int tuples = PQntuples(result);
 468
 469 if (tuples != 1)
 470     {
 471     strError = "Failed to fetch tariff data";
 472     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
 473     PQclear(result);
 474     if (RollbackTransaction())
 475         {
 476         printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to rollback transaction'\n");
 477         }
 478     return -1;
 479     }
 480
 481 int id;
 482
 483     {
 484     std::stringstream tuple;
 485     tuple << PQgetvalue(result, 0, 0) << " ";
 486     tuple << PQgetvalue(result, 0, 1) << " ";
 487     tuple << PQgetvalue(result, 0, 2) << " ";
 488     tuple << PQgetvalue(result, 0, 3) << " ";
 489     tuple << PQgetvalue(result, 0, 4) << " ";
 490
 491     tuple >> id;
 492     tuple >> td->tariffConf.fee;
 493     tuple >> td->tariffConf.free;
 494     tuple >> td->tariffConf.passiveCost;
 495     tuple >> td->tariffConf.traffType;
 496     }
 497
 498 if (version > 6)
 499     td->tariffConf.period = TARIFF::StringToPeriod(PQgetvalue(result, 0, 5));
 500
 501 PQclear(result);
 502
 503 query.str("");
 504 query << "SELECT dir_num, \
 505                  price_day_a, \
 506                  price_day_b, \
 507                  price_night_a, \
 508                  price_night_b, \
 509                  threshold, \
 510                  EXTRACT(hour FROM time_day_begins), \
 511                  EXTRACT(minute FROM time_day_begins), \
 512                  EXTRACT(hour FROM time_day_ends), \
 513                  EXTRACT(minute FROM time_day_ends) \
 514           FROM tb_tariffs_params \
 515           WHERE fk_tariff = " << id;
 516
 517 result = PQexec(connection, query.str().c_str());
 518
 519 if (PQresultStatus(result) != PGRES_TUPLES_OK)
 520     {
 521     strError = PQresultErrorMessage(result);
 522     PQclear(result);
 523     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): '%s'\n", strError.c_str());
 524     if (RollbackTransaction())
 525         {
 526         printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to rollback transaction'\n");
 527         }
 528     return -1;
 529     }
 530
 531 tuples = PQntuples(result);
 532
 533 if (tuples != DIR_NUM)
 534     {
 535     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Tariff params count and DIR_NUM does not feet (wanted %d, actually %d)'\n", DIR_NUM, tuples);
 536     }
 537
 538 for (int i = 0; i < std::min(tuples, DIR_NUM); ++i)
 539     {
 540     int dir;
 541
 542         {
 543         std::stringstream tuple;
 544         tuple << PQgetvalue(result, i, 0) << " ";
 545         tuple << PQgetvalue(result, i, 1) << " ";
 546         tuple << PQgetvalue(result, i, 2) << " ";
 547         tuple << PQgetvalue(result, i, 3) << " ";
 548         tuple << PQgetvalue(result, i, 4) << " ";
 549         tuple << PQgetvalue(result, i, 5) << " ";
 550         tuple << PQgetvalue(result, i, 6) << " ";
 551         tuple << PQgetvalue(result, i, 7) << " ";
 552         tuple << PQgetvalue(result, i, 8) << " ";
 553         tuple << PQgetvalue(result, i, 9) << " ";
 554
 555         tuple >> dir;
 556         tuple >> td->dirPrice[dir].priceDayA;
 557         td->dirPrice[dir].priceDayA /= 1024 * 1024;
 558         tuple >> td->dirPrice[dir].priceDayB;
 559         td->dirPrice[dir].priceDayB /= 1024 * 1024;
 560         tuple >> td->dirPrice[dir].priceNightA;
 561         td->dirPrice[dir].priceNightA /= 1024 * 1024;
 562         tuple >> td->dirPrice[dir].priceNightB;
 563         td->dirPrice[dir].priceNightB /= 1024 * 1024;
 564         tuple >> td->dirPrice[dir].threshold;
 565         tuple >> td->dirPrice[dir].hDay;
 566         tuple >> td->dirPrice[dir].mDay;
 567         tuple >> td->dirPrice[dir].hNight;
 568         tuple >> td->dirPrice[dir].mNight;
 569         }
 570
 571     if (std::fabs(td->dirPrice[dir].priceDayA - td->dirPrice[dir].priceNightA) > 1.0e-3 &&
 572         std::fabs(td->dirPrice[dir].priceDayB - td->dirPrice[dir].priceNightB) > 1.0e-3)
 573         {
 574         td->dirPrice[dir].singlePrice = true;
 575         }
 576     else
 577         {
 578         td->dirPrice[dir].singlePrice = false;
 579         }
 580     if (td->dirPrice[dir].threshold == (int)0xffFFffFF)
 581         {
 582         td->dirPrice[dir].noDiscount = true;
 583         }
 584     else
 585         {
 586
 587         td->dirPrice[dir].noDiscount = false;
 588         }
 589
 590     }
 591
 592 PQclear(result);
 593
 594 if (CommitTransaction())
 595     {
 596     printfd(__FILE__, "POSTGRESQL_STORE::RestoreTariff(): 'Failed to commit transaction'\n");
 597     return -1;
 598     }
 599
 600 return 0;
 601 }
 602 //-----------------------------------------------------------------------------
 603