source: trunk/mysql_pool.cpp

Last change on this file was 7, checked in by willem, 9 years ago

willem

File size: 6.5 KB
Line 
1
2/*
3   Copyright 2013 Willem Vermin, SURFsara
4
5   Licensed under the Apache License, Version 2.0 (the "License");
6   you may not use this file except in compliance with the License.
7   You may obtain a copy of the License at
8
9       http://www.apache.org/licenses/LICENSE-2.0
10
11   Unless required by applicable law or agreed to in writing, software
12   distributed under the License is distributed on an "AS IS" BASIS,
13   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   See the License for the specific language governing permissions and
15   limitations under the License.
16 */
17#include <iostream>
18#include "wutils.h"
19#include "stopos.h"
20
21#include "mysql_connection.h"
22#include <cppconn/driver.h>
23#include <cppconn/exception.h>
24#include <cppconn/resultset.h>
25#include <cppconn/statement.h>
26#include <cppconn/prepared_statement.h>
27
28#include "stopos_pool.h"
29#include "mysql_pool.h"
30// construcor
31mysql_pool::mysql_pool()
32{
33  this->whoami = "mysql";
34  std::string schema = "stopos";
35  sql::Driver *driver;
36  driver = get_driver_instance();
37  //  connection, mysqlusername, mysqlpassword
38  this->connection = driver->connect("tcp://127.0.0.1:3307",schema,"x");
39  this->connection->setSchema(schema);
40  this->stmt = this->connection->createStatement();
41}
42
43mysql_pool::~mysql_pool(void)
44{
45  delete this->stmt;
46  delete this->connection;
47}
48
49int mysql_pool::create_db(const std::string &dbname)
50{
51  int rc;
52  rc = this->purge_db();
53  if (rc != 0)
54    return rc;
55
56  try
57  {
58    int rc;
59
60    rc = this->open_db();
61    if (rc != 0)
62      return rc;
63    this->close_db();
64
65    return 0;
66  }
67  catch (sql::SQLException &e)
68  {
69    std::cerr << "# ERR: SQLException in " << __FILE__;
70    std::cerr << "(" << __FUNCTION__ << ") on line "
71       << __LINE__ << std::endl;
72    std::cerr << "# ERR: " << e.what();
73    std::cerr << " (MySQL error code: " << e.getErrorCode();
74    std::cerr << ", SQLState: " << e.getSQLState() <<
75   " )" << std::endl;
76    return OPENERROR;
77  }
78}
79
80int mysql_pool::purge_db(void)
81{
82  if (this->db_name == 0)
83    return NOFILENAME;
84  try
85  {
86    std::string cmd = "DROP TABLE IF EXISTS " + this->sdb_name;
87    this->stmt->execute(cmd);
88
89    cmd = "CREATE TABLE " + this->sdb_name +
90          "(pkey CHAR(20) NOT NULL,"
91          " PRIMARY KEY (pkey),"
92          " INDEX(pkey),"
93          " data VARCHAR(" +
94            NumberToString(this->reclen) +
95          +") NOT NULL)";
96    this->stmt->execute(cmd);
97  }
98  catch (sql::SQLException &e)
99  {
100    std::cerr << "# ERR: SQLException in " << __FILE__;
101    std::cerr << "(" << __FUNCTION__ << ") on line "
102       << __LINE__ << std::endl;
103    std::cerr << "# ERR: " << e.what();
104    std::cerr << " (MySQL error code: " << e.getErrorCode();
105    std::cerr << ", SQLState: " << e.getSQLState() <<
106   " )" << std::endl;
107    return PURGEERROR;
108  }
109    return 0;
110}
111
112int mysql_pool::open_db(void)
113{
114  if (! this->db_name)
115    return OPENERROR;
116
117  if (this->db_open)
118    return 0;
119
120  try
121  {
122    sql::ResultSet *res;
123    std::string cmd;
124    cmd="select count(*) as count "
125        "from information_schema.tables "
126        "where table_schema='stopos' "
127        "and table_name = '"+this->sdb_name+"'";
128    res = this->stmt->executeQuery(cmd);
129    res->beforeFirst();
130    if (res->next())
131      if (res->getInt("count") == 0)
132        return OPENERROR;
133    delete res;
134
135    cmd = "lock tables " + this->sdb_name + " write";
136    this->stmt->execute(cmd);
137  }
138  catch (sql::SQLException &e)
139  {
140    std::cerr << "# ERR: SQLException in " << __FILE__;
141    std::cerr << "(" << __FUNCTION__ << ") on line "
142       << __LINE__ << std::endl;
143    std::cerr << "# ERR: " << e.what();
144    std::cerr << " (MySQL error code: " << e.getErrorCode();
145    std::cerr << ", SQLState: " << e.getSQLState() <<
146   " )" << std::endl;
147    return OPENERROR;
148  }
149
150  this->db_open = 1;
151
152  return 0;
153}
154
155int mysql_pool::close_db(void)
156{
157  if (!this->db_open)
158    return 0;
159
160  std::string cmd;
161  try
162  {
163    cmd = "unlock tables";
164    this->stmt->execute(cmd);
165  }
166  catch (sql::SQLException &e)
167  {
168    std::cerr << "# ERR: SQLException in " << __FILE__;
169    std::cerr << "(" << __FUNCTION__ << ") on line "
170       << __LINE__ << std::endl;
171    std::cerr << "# ERR: " << e.what();
172    std::cerr << " (MySQL error code: " << e.getErrorCode();
173    std::cerr << ", SQLState: " << e.getSQLState() <<
174   " )" << std::endl;
175    return OPENERROR;
176  }
177
178  this->db_open = 0;
179
180  return 0;
181}
182
183int mysql_pool::get_record(std::string &r, const std::string &k, const std::string &l)
184{
185  try
186  {
187    std::string cmd;
188    sql::ResultSet *res;
189    cmd = "select data from " + sdb_name +
190        " where pkey = '" + k + "'";
191    res = this->stmt->executeQuery(cmd);
192    res->beforeFirst();
193    if (res->next() == 0)
194      return NOTFOUND;
195    r = res->getString("data"); 
196    delete res;
197  }
198  catch (sql::SQLException &e)
199  {
200    std::cerr << "# ERR: SQLException in " << __FILE__;
201    std::cerr << "(" << __FUNCTION__ << ") on line "
202       << __LINE__ << std::endl;
203    std::cerr << "# ERR: " << e.what();
204    std::cerr << " (MySQL error code: " << e.getErrorCode();
205    std::cerr << ", SQLState: " << e.getSQLState() <<
206   " )" << std::endl;
207    return NOTFOUND;
208  }
209
210  return 0;
211}
212
213int mysql_pool::put_record(const std::string &r, 
214                           const std::string &k,
215                           const std::string &l)
216{
217  bool rectoolong=0;
218  try
219  {
220    std::string cmd;
221    std::string rr;
222    rr = r;
223
224    if (rr.size() >= reclen)
225    {
226      rectoolong = 1;
227      rr.resize(reclen-1);
228    }
229
230    cmd = "replace " + sdb_name +
231      " values('" + k + "','" +
232      rr + "')";
233
234    this->stmt->execute(cmd);
235  }
236  catch (sql::SQLException &e)
237  {
238    std::cerr << "# ERR: SQLException in " << __FILE__;
239    std::cerr << "(" << __FUNCTION__ << ") on line "
240       << __LINE__ << std::endl;
241    std::cerr << "# ERR: " << e.what();
242    std::cerr << " (MySQL error code: " << e.getErrorCode();
243    std::cerr << ", SQLState: " << e.getSQLState() <<
244   " )" << std::endl;
245    return STOREERROR;
246  }
247
248  if (rectoolong)
249    return RECORDTOOLONG;
250
251  return 0;
252}
253
254int mysql_pool::remove_record(const std::string &k, const std::string &l)
255{
256  try
257  {
258    std::string cmd;
259
260    cmd = "delete from " + sdb_name +
261          " where pkey ='" + k + "'";
262
263    this->stmt->execute(cmd);
264  }
265  catch (sql::SQLException &e)
266  {
267    std::cerr << "# ERR: SQLException in " << __FILE__;
268    std::cerr << "(" << __FUNCTION__ << ") on line "
269       << __LINE__ << std::endl;
270    std::cerr << "# ERR: " << e.what();
271    std::cerr << " (MySQL error code: " << e.getErrorCode();
272    std::cerr << ", SQLState: " << e.getSQLState() <<
273   " )" << std::endl;
274    return STOREERROR;
275  }
276
277  return 0;
278}
279
280
281
Note: See TracBrowser for help on using the repository browser.