]> git.stg.codes - stg.git/blob - stargazer/plugins/store/firebird/firebird_store_tariffs.cpp
10c45ab9b6a79698185aa6b47eb558ad8e07a0c9
[stg.git] / 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