Drizzled Public API Documentation

sql_executor.cc
1 /* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2  * vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3  *
4  * Copyright (C) 2011 David Shrewsbury
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #include <config.h>
22 #include <plugin/slave/sql_executor.h>
23 #include <drizzled/plugin/listen.h>
24 #include <drizzled/plugin/client.h>
25 #include <drizzled/catalog/local.h>
26 #include <drizzled/execute.h>
27 #include <drizzled/sql/result_set.h>
28 #include <drizzled/errmsg_print.h>
29 #include <drizzled/plugin.h>
30 
31 using namespace std;
32 using namespace drizzled;
33 
34 namespace slave
35 {
36 
37 SQLExecutor::SQLExecutor(const string &user, const string &schema)
38  : _in_error_state(false)
39 {
40  /* setup a Session object */
41  _session= Session::make_shared(plugin::Listen::getNullClient(), catalog::local());
42  identifier::user::mptr user_id= identifier::User::make_shared();
43  user_id->setUser(user);
44  _session->setUser(user_id);
45  _session->set_schema(schema);
46 }
47 
48 
49 bool SQLExecutor::executeSQL(vector<string> &sql)
50 {
51  if (not _in_error_state)
52  _error_message.clear();
53 
54  Execute execute(*(_session.get()), true);
55 
56  string combined_sql;
57  BOOST_FOREACH(string& it, sql)
58  {
59  combined_sql.append(it);
60  combined_sql.append("; ");
61  }
62 
63  sql::ResultSet result_set(1);
64 
65  /* Execute wraps the SQL to run within a transaction */
66  execute.run(combined_sql, result_set);
67 
68  sql::Exception exception= result_set.getException();
69 
70  drizzled::error_t err= exception.getErrorCode();
71 
72  if ((err != drizzled::EE_OK) && (err != drizzled::ER_EMPTY_QUERY))
73  {
74  /* avoid recursive errors */
75  if (_in_error_state)
76  {
77  return true;
78  }
79 
80  _in_error_state= true;
81  _error_message= "(SQLSTATE ";
82  _error_message.append(exception.getSQLState());
83  _error_message.append(") ");
84  _error_message.append(exception.getErrorMessage());
85 
86  string bad_sql("Failure while executing:\n");
87  for (size_t y= 0; y < sql.size(); y++)
88  {
89  bad_sql.append(sql[y]);
90  bad_sql.append("\n");
91  }
92 
93  errmsg_printf(error::ERROR, _("%s\n%s\n"), _error_message.c_str(), bad_sql.c_str());
94  return false;
95  }
96 
97  return true;
98 }
99 
100 } /* namespace slave */
TODO: Rename this file - func.h is stupid.
drizzled::Session::shared_ptr _session
Definition: sql_executor.h:129
Definition: module.cc:36