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