]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/store/postgresql/postgresql_store_services.cpp
Merge remote-tracking branch 'github/master'
[stg.git] / projects / stargazer / plugins / store / postgresql / postgresql_store_services.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 /*
23  *  Services manipulation methods
24  *
25  *  $Revision: 1.2 $
26  *  $Date: 2009/06/09 12:32:40 $
27  *
28  */
29
30 #include "postgresql_store.h"
31
32 #include "stg/service_conf.h"
33 #include "stg/common.h"
34 #include "stg/locker.h"
35
36 #include <string>
37 #include <vector>
38 #include <sstream>
39
40 #include <libpq-fe.h>
41
42 //-----------------------------------------------------------------------------
43 int POSTGRESQL_STORE::GetServicesList(std::vector<std::string> * servicesList) const
44 {
45 STG_LOCKER lock(&mutex);
46
47 if (PQstatus(connection) != CONNECTION_OK)
48     {
49     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
50     if (Reset())
51         {
52         strError = "Connection lost";
53         printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): '%s'\n", strError.c_str());
54         return -1;
55         }
56     }
57
58 PGresult * result;
59
60 if (StartTransaction())
61     {
62     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to start transaction'\n");
63     return -1;
64     }
65
66 result = PQexec(connection, "SELECT name FROM tb_services");
67
68 if (PQresultStatus(result) != PGRES_TUPLES_OK)
69     {
70     strError = PQresultErrorMessage(result);
71     PQclear(result);
72     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): '%s'\n", strError.c_str());
73     if (RollbackTransaction())
74         {
75         printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to rollback transaction'\n");
76         }
77     return -1;
78     }
79
80 int tuples = PQntuples(result);
81
82 for (int i = 0; i < tuples; ++i)
83     {
84     servicesList->push_back(PQgetvalue(result, i, 0));
85     }
86
87 PQclear(result);
88
89 if (CommitTransaction())
90     {
91     printfd(__FILE__, "POSTGRESQL_STORE::GetServicesList(): 'Failed to commit transaction'\n");
92     return -1;
93     }
94
95 return 0;
96 }
97
98 //-----------------------------------------------------------------------------
99 int POSTGRESQL_STORE::SaveService(const STG::ServiceConf & sc) const
100 {
101 STG_LOCKER lock(&mutex);
102
103 if (PQstatus(connection) != CONNECTION_OK)
104     {
105     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
106     if (Reset())
107         {
108         strError = "Connection lost";
109         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): '%s'\n", strError.c_str());
110         return -1;
111         }
112     }
113
114 PGresult * result;
115
116 if (StartTransaction())
117     {
118     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to start transaction'\n");
119     return -1;
120     }
121
122 std::string ename = sc.name;
123 std::string ecomment = sc.comment;
124
125 if (EscapeString(ename))
126     {
127     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to escape name'\n");
128     if (RollbackTransaction())
129         {
130         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
131         }
132     return -1;
133     }
134
135 if (EscapeString(ecomment))
136     {
137     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to escape comment'\n");
138     if (RollbackTransaction())
139         {
140         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
141         }
142     return -1;
143     }
144
145 std::ostringstream query;
146 query << "UPDATE tb_services SET "
147           << "comment = '" << ecomment << "', "
148           << "cost = " << sc.cost << ", "
149           << "pay_day = " << sc.payDay << " "
150       << "WHERE name = '" << ename << "'";
151
152 result = PQexec(connection, query.str().c_str());
153
154 if (PQresultStatus(result) != PGRES_COMMAND_OK)
155     {
156     strError = PQresultErrorMessage(result);
157     PQclear(result);
158     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): '%s'\n", strError.c_str());
159     if (RollbackTransaction())
160         {
161         printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to rollback transaction'\n");
162         }
163     return -1;
164     }
165
166 PQclear(result);
167
168 if (CommitTransaction())
169     {
170     printfd(__FILE__, "POSTGRESQL_STORE::SaveService(): 'Failed to commit transaction'\n");
171     return -1;
172     }
173
174 return 0;
175 }
176
177 //-----------------------------------------------------------------------------
178 int POSTGRESQL_STORE::RestoreService(STG::ServiceConf * sc,
179                                      const std::string & name) const
180 {
181 STG_LOCKER lock(&mutex);
182
183 if (PQstatus(connection) != CONNECTION_OK)
184     {
185     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
186     if (Reset())
187         {
188         strError = "Connection lost";
189         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): '%s'\n", strError.c_str());
190         return -1;
191         }
192     }
193
194 PGresult * result;
195
196 if (StartTransaction())
197     {
198     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to start transaction'\n");
199     return -1;
200     }
201
202 std::string ename = name;
203
204 if (EscapeString(ename))
205     {
206     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to escape name'\n");
207     if (RollbackTransaction())
208         {
209         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
210         }
211     return -1;
212     }
213
214 std::ostringstream query;
215 query << "SELECT comment, cost, pay_day FROM tb_services WHERE name = '" << ename << "'";
216
217 result = PQexec(connection, query.str().c_str());
218
219 if (PQresultStatus(result) != PGRES_TUPLES_OK)
220     {
221     strError = PQresultErrorMessage(result);
222     PQclear(result);
223     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): '%s'\n", strError.c_str());
224     if (RollbackTransaction())
225         {
226         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
227         }
228     return -1;
229     }
230
231 int tuples = PQntuples(result);
232
233 if (tuples != 1)
234     {
235     strError = "Failed to fetch service's data";
236     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Invalid number of tuples. Wanted 1, actulally %d'\n", tuples);
237     PQclear(result);
238     if (RollbackTransaction())
239         {
240         printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to rollback transaction'\n");
241         }
242     return -1;
243     }
244
245 std::stringstream tuple;
246 tuple << PQgetvalue(result, 0, 0) << " "
247       << PQgetvalue(result, 0, 1) << " "
248       << PQgetvalue(result, 0, 2);
249
250 PQclear(result);
251
252 tuple >> sc->comment
253       >> sc->cost
254       >> sc->payDay;
255
256 if (CommitTransaction())
257     {
258     printfd(__FILE__, "POSTGRESQL_STORE::RestoreService(): 'Failed to commit transaction'\n");
259     return -1;
260     }
261
262 return 0;
263 }
264
265 //-----------------------------------------------------------------------------
266 int POSTGRESQL_STORE::AddService(const std::string & name) const
267 {
268 STG_LOCKER lock(&mutex);
269
270 if (PQstatus(connection) != CONNECTION_OK)
271     {
272     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
273     if (Reset())
274         {
275         strError = "Connection lost";
276         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): '%s'\n", strError.c_str());
277         return -1;
278         }
279     }
280
281 PGresult * result;
282
283 if (StartTransaction())
284     {
285     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to start transaction'\n");
286     return -1;
287     }
288
289 std::string ename = name;
290
291 if (EscapeString(ename))
292     {
293     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to escape name'\n");
294     if (RollbackTransaction())
295         {
296         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to rollback transaction'\n");
297         }
298     return -1;
299     }
300
301 std::ostringstream query;
302 query << "INSERT INTO tb_services \
303               (name, comment, cost, pay_day) \
304           VALUES \
305               ('" << ename << "', '', 0, 0)";
306
307 result = PQexec(connection, query.str().c_str());
308
309 if (PQresultStatus(result) != PGRES_COMMAND_OK)
310     {
311     strError = PQresultErrorMessage(result);
312     PQclear(result);
313     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): '%s'\n", strError.c_str());
314     if (RollbackTransaction())
315         {
316         printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to rollback transaction'\n");
317         }
318     return -1;
319     }
320
321 PQclear(result);
322
323 if (CommitTransaction())
324     {
325     printfd(__FILE__, "POSTGRESQL_STORE::AddService(): 'Failed to commit transaction'\n");
326     return -1;
327     }
328
329 return 0;
330 }
331
332 //-----------------------------------------------------------------------------
333 int POSTGRESQL_STORE::DelService(const std::string & name) const
334 {
335 STG_LOCKER lock(&mutex);
336
337 if (PQstatus(connection) != CONNECTION_OK)
338     {
339     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Connection lost. Trying to reconnect...'\n", strError.c_str());
340     if (Reset())
341         {
342         strError = "Connection lost";
343         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): '%s'\n", strError.c_str());
344         return -1;
345         }
346     }
347
348 PGresult * result;
349
350 if (StartTransaction())
351     {
352     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to start transaction'\n");
353     return -1;
354     }
355
356 std::string ename = name;
357
358 if (EscapeString(ename))
359     {
360     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to escape name'\n");
361     if (RollbackTransaction())
362         {
363         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to rollback transaction'\n");
364         }
365     return -1;
366     }
367
368 std::ostringstream query;
369 query << "DELETE FROM tb_services WHERE name = '" << ename << "'";
370
371 result = PQexec(connection, query.str().c_str());
372
373 if (PQresultStatus(result) != PGRES_COMMAND_OK)
374     {
375     strError = PQresultErrorMessage(result);
376     PQclear(result);
377     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): '%s'\n", strError.c_str());
378     if (RollbackTransaction())
379         {
380         printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to rollback transaction'\n");
381         }
382     return -1;
383     }
384
385 PQclear(result);
386
387 if (CommitTransaction())
388     {
389     printfd(__FILE__, "POSTGRESQL_STORE::DelService(): 'Failed to commit transaction'\n");
390     return -1;
391     }
392
393 return 0;
394 }
395 //-----------------------------------------------------------------------------
396