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