stargazer/plugins/store/firebird/firebird_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.5 $
  25  *  $Date: 2007/12/23 13:39:59 $
  26  *
  27  */
  28
  29 #include "firebird_store.h"
  30
  31 #include "stg/ibpp.h"
  32 #include "stg/tariff.h"
  33 #include "stg/tariff_conf.h"
  34 #include "stg/common.h"
  35
  36 #include <cmath>
  37
  38 namespace
  39 {
  40
  41 const int pt_mega = 1024 * 1024;
  42
  43 }
  44
  45 //-----------------------------------------------------------------------------
  46 int FIREBIRD_STORE::GetTariffsList(std::vector<std::string> * tariffsList) const
  47 {
  48 STG_LOCKER lock(&mutex);
  49
  50 IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amRead, til, tlr);
  51 IBPP::Statement st = IBPP::StatementFactory(db, tr);
  52
  53 std::string name;
  54
  55 try
  56     {
  57     tr->Start();
  58     st->Execute("select name from tb_tariffs");
  59     while (st->Fetch())
  60         {
  61         st->Get(1, name);
  62         tariffsList->push_back(name);
  63         }
  64     tr->Commit();
  65     }
  66
  67 catch (IBPP::Exception & ex)
  68     {
  69     tr->Rollback();
  70     strError = "IBPP exception";
  71     printfd(__FILE__, ex.what());
  72     return -1;
  73     }
  74
  75 return 0;
  76 }
  77 //-----------------------------------------------------------------------------
  78 int FIREBIRD_STORE::AddTariff(const std::string & name) const
  79 {
  80 STG_LOCKER lock(&mutex);
  81
  82 IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amWrite, til, tlr);
  83 IBPP::Statement st = IBPP::StatementFactory(db, tr);
  84
  85 try
  86     {
  87     tr->Start();
  88     st->Prepare("execute procedure sp_add_tariff(?, ?)");
  89     st->Set(1, name);
  90     st->Set(2, DIR_NUM);
  91     st->Execute();
  92     tr->Commit();
  93     }
  94
  95 catch (IBPP::Exception & ex)
  96     {
  97     tr->Rollback();
  98     strError = "IBPP exception";
  99     printfd(__FILE__, ex.what());
 100     return -1;
 101     }
 102
 103 return 0;
 104 }
 105 //-----------------------------------------------------------------------------
 106 int FIREBIRD_STORE::DelTariff(const std::string & name) const
 107 {
 108 STG_LOCKER lock(&mutex);
 109
 110 IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amWrite, til, tlr);
 111 IBPP::Statement st = IBPP::StatementFactory(db, tr);
 112
 113 try
 114     {
 115     tr->Start();
 116     st->Prepare("execute procedure sp_delete_tariff(?)");
 117     st->Set(1, name);
 118     st->Execute();
 119     tr->Commit();
 120     }
 121
 122 catch (IBPP::Exception & ex)
 123     {
 124     tr->Rollback();
 125     strError = "IBPP exception";
 126     printfd(__FILE__, ex.what());
 127     return -1;
 128     }
 129
 130 return 0;
 131 }
 132 //-----------------------------------------------------------------------------
 133 int FIREBIRD_STORE::SaveTariff(const STG::TariffData & td,
 134                                const std::string & tariffName) const
 135 {
 136 STG_LOCKER lock(&mutex);
 137
 138 IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amWrite, til, tlr);
 139 IBPP::Statement st = IBPP::StatementFactory(db, tr);
 140
 141 try
 142     {
 143     tr->Start();
 144     st->Prepare("select pk_tariff from tb_tariffs where name = ?");
 145     st->Set(1, tariffName);
 146     st->Execute();
 147     if (!st->Fetch())
 148     {
 149     tr->Rollback();
 150     strprintf(&strError, "Tariff \"%s\" not found in database", tariffName.c_str());
 151     printfd(__FILE__, "Tariff '%s' not found in database\n", tariffName.c_str());
 152     return -1;
 153     }
 154     int32_t id;
 155     st->Get(1, id);
 156     st->Close();
 157
 158     std::string query = "update tb_tariffs set \
 159                             fee = ?, \
 160                             free = ?, \
 161                             passive_cost = ?, \
 162                             traff_type = ?";
 163
 164     if (schemaVersion > 0)
 165         query += ", period = ?";
 166     if (schemaVersion > 1)
 167         query += ", change_policy = ?, \
 168                     change_policy_timeout = ?";
 169
 170     query += " where pk_tariff = ?";
 171
 172     unsigned num = 5;
 173     st->Prepare(query);
 174     st->Set(1, td.tariffConf.fee);
 175     st->Set(2, td.tariffConf.free);
 176     st->Set(3, td.tariffConf.passiveCost);
 177     st->Set(4, td.tariffConf.traffType);
 178
 179     if (schemaVersion > 0)
 180         {
 181         st->Set(5, STG::Tariff::toString(td.tariffConf.period));
 182         ++num;
 183         }
 184
 185     if (schemaVersion > 1)
 186         {
 187         st->Set(6, STG::Tariff::toString(td.tariffConf.changePolicy));
 188         IBPP::Timestamp policyTimeout;
 189         time_t2ts(td.tariffConf.changePolicyTimeout, &policyTimeout);
 190         st->Set(7, policyTimeout);
 191         num += 2;
 192         }
 193
 194     st->Set(num, id);
 195     st->Execute();
 196     st->Close();
 197
 198     IBPP::Time tb;
 199     IBPP::Time te;
 200
 201     for(int i = 0; i < DIR_NUM; i++)
 202         {
 203
 204         tb.SetTime(td.dirPrice[i].hDay, td.dirPrice[i].mDay, 0);
 205         te.SetTime(td.dirPrice[i].hNight, td.dirPrice[i].mNight, 0);
 206
 207         double pda = td.dirPrice[i].priceDayA * 1024 * 1024;
 208         double pdb = td.dirPrice[i].priceDayB * 1024 * 1024;
 209         double pna = 0;
 210         double pnb = 0;
 211
 212         if (td.dirPrice[i].singlePrice)
 213             {
 214             pna = pda;
 215             pnb = pdb;
 216             }
 217         else
 218             {
 219             pna = td.dirPrice[i].priceNightA;
 220             pnb = td.dirPrice[i].priceNightB;
 221             }
 222
 223         int threshold = 0;
 224         if (td.dirPrice[i].noDiscount)
 225             {
 226             threshold = 0xffFFffFF;
 227             }
 228         else
 229             {
 230             threshold = td.dirPrice[i].threshold;
 231             }
 232
 233         st->Prepare("update tb_tariffs_params set \
 234                 price_day_a = ?, \
 235                 price_day_b = ?, \
 236                 price_night_a = ?, \
 237                 price_night_b = ?, \
 238                 threshold = ?, \
 239                 time_day_begins = ?, \
 240                 time_day_ends = ? \
 241                 where fk_tariff = ? and dir_num = ?");
 242         st->Set(1, pda);
 243         st->Set(2, pdb);
 244         st->Set(3, pna);
 245         st->Set(4, pnb);
 246         st->Set(5, threshold);
 247         st->Set(6, tb);
 248         st->Set(7, te);
 249         st->Set(8, id);
 250         st->Set(9, i);
 251         st->Execute();
 252         st->Close();
 253         }
 254     tr->Commit();
 255     }
 256
 257 catch (IBPP::Exception & ex)
 258     {
 259     tr->Rollback();
 260     strError = "IBPP exception";
 261     printfd(__FILE__, ex.what());
 262     return -1;
 263     }
 264
 265 return 0;
 266 }
 267 //-----------------------------------------------------------------------------
 268 int FIREBIRD_STORE::RestoreTariff(STG::TariffData * td,
 269                                   const std::string & tariffName) const
 270 {
 271 STG_LOCKER lock(&mutex);
 272
 273 IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amRead, til, tlr);
 274 IBPP::Statement st = IBPP::StatementFactory(db, tr);
 275
 276
 277 td->tariffConf.name = tariffName;
 278
 279 try
 280     {
 281     tr->Start();
 282     st->Prepare("select * from tb_tariffs where name = ?"); // TODO: explicit field order!
 283     st->Set(1, tariffName);
 284     st->Execute();
 285     if (!st->Fetch())
 286         {
 287         strError = "Tariff \"" + tariffName + "\" not found in database";
 288     printfd(__FILE__, "Tariff '%s' not found in database\n", tariffName.c_str());
 289         tr->Rollback();
 290         return -1;
 291         }
 292     int32_t id;
 293     st->Get(1, id);
 294     st->Get(3, td->tariffConf.fee);
 295     st->Get(4, td->tariffConf.free);
 296     st->Get(5, td->tariffConf.passiveCost);
 297     td->tariffConf.traffType = STG::Tariff::fromInt(Get<int>(st, 6));
 298     if (schemaVersion > 0)
 299         td->tariffConf.period = STG::Tariff::parsePeriod(Get<std::string>(st, 7));
 300     if (schemaVersion > 1)
 301         {
 302         td->tariffConf.changePolicy = STG::Tariff::parseChangePolicy(Get<std::string>(st, 8));
 303         td->tariffConf.changePolicyTimeout = ts2time_t(Get<IBPP::Timestamp>(st, 9));
 304         }
 305     st->Close();
 306     st->Prepare("select * from tb_tariffs_params where fk_tariff = ?");
 307     st->Set(1, id);
 308     st->Execute();
 309     int i = 0;
 310     while (st->Fetch())
 311     {
 312     i++;
 313     if (i > DIR_NUM)
 314         {
 315         strError = "Too mach params for tariff \"" + tariffName + "\"";
 316         printfd(__FILE__, "Too mach params for tariff '%s'\n", tariffName.c_str());
 317         tr->Rollback();
 318         return -1;
 319         }
 320     int16_t dir;
 321     st->Get(3, dir);
 322     st->Get(4, td->dirPrice[dir].priceDayA);
 323     td->dirPrice[dir].priceDayA /= 1024*1024;
 324     st->Get(5, td->dirPrice[dir].priceDayB);
 325     td->dirPrice[dir].priceDayB /= 1024*1024;
 326     st->Get(6, td->dirPrice[dir].priceNightA);
 327     td->dirPrice[dir].priceNightA /= 1024*1024;
 328     st->Get(7, td->dirPrice[dir].priceNightB);
 329     td->dirPrice[dir].priceNightB /= 1024*1024;
 330     st->Get(8, td->dirPrice[dir].threshold);
 331     if (std::fabs(td->dirPrice[dir].priceDayA - td->dirPrice[dir].priceNightA) < 1.0e-3 / pt_mega &&
 332         std::fabs(td->dirPrice[dir].priceDayB - td->dirPrice[dir].priceNightB) < 1.0e-3 / pt_mega)
 333         {
 334         td->dirPrice[dir].singlePrice = true;
 335         }
 336     else
 337         {
 338         td->dirPrice[dir].singlePrice = false;
 339         }
 340     if (td->dirPrice[dir].threshold == (int)0xffFFffFF)
 341         {
 342         td->dirPrice[dir].noDiscount = true;
 343         }
 344     else
 345         {
 346
 347         td->dirPrice[dir].noDiscount = false;
 348         }
 349     IBPP::Time tb;
 350     st->Get(9, tb);
 351     IBPP::Time te;
 352     st->Get(10, te);
 353     int h, m, s;
 354     tb.GetTime(h, m, s);
 355     td->dirPrice[dir].hDay = h;
 356     td->dirPrice[dir].mDay = m;
 357     te.GetTime(h, m, s);
 358     td->dirPrice[dir].hNight = h;
 359     td->dirPrice[dir].mNight = m;
 360     }
 361     tr->Commit();
 362     }
 363
 364 catch (IBPP::Exception & ex)
 365     {
 366     tr->Rollback();
 367     strError = "IBPP exception";
 368     printfd(__FILE__, ex.what());
 369     return -1;
 370     }
 371
 372 return 0;
 373 }
 374 //-----------------------------------------------------------------------------
 375