|
851
by Brian Aker
Class rewrite of Session (aka get all of the junk out) |
1 |
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
|
2 |
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
|
|
3 |
*
|
|
4 |
* Copyright (C) 2008 Sun Microsystems
|
|
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; version 2 of the License.
|
|
9 |
*
|
|
10 |
* This program is distributed in the hope that it will be useful,
|
|
11 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
12 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
13 |
* GNU General Public License for more details.
|
|
14 |
*
|
|
15 |
* You should have received a copy of the GNU General Public License
|
|
16 |
* along with this program; if not, write to the Free Software
|
|
17 |
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
18 |
*/
|
|
19 |
||
20 |
||
21 |
#ifndef DRIZZLED_SELECT_DUMPVAR_H
|
|
22 |
#define DRIZZLED_SELECT_DUMPVAR_H
|
|
23 |
||
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
24 |
#include <drizzled/error.h> |
|
851
by Brian Aker
Class rewrite of Session (aka get all of the junk out) |
25 |
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
26 |
#include <vector> |
|
851
by Brian Aker
Class rewrite of Session (aka get all of the junk out) |
27 |
|
28 |
class select_dumpvar :public select_result_interceptor { |
|
29 |
ha_rows row_count; |
|
30 |
public: |
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
31 |
std::vector<my_var *> var_list; |
32 |
select_dumpvar() { var_list.clear(); row_count= 0;} |
|
|
851
by Brian Aker
Class rewrite of Session (aka get all of the junk out) |
33 |
~select_dumpvar() {} |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
34 |
|
35 |
int prepare(List<Item> &list, Select_Lex_Unit *u) |
|
36 |
{
|
|
37 |
unit= u; |
|
38 |
||
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
39 |
if (var_list.size() != list.elements) |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
40 |
{
|
41 |
my_message(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT, |
|
42 |
ER(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT), MYF(0)); |
|
43 |
return 1; |
|
44 |
}
|
|
45 |
return 0; |
|
46 |
}
|
|
47 |
||
48 |
void cleanup() |
|
49 |
{
|
|
50 |
row_count= 0; |
|
51 |
}
|
|
52 |
||
53 |
||
54 |
bool send_data(List<Item> &items) |
|
55 |
{
|
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
56 |
|
57 |
std::vector<my_var *>::const_iterator iter= var_list.begin(); |
|
58 |
||
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
59 |
List_iterator<Item> it(items); |
60 |
Item *item; |
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
61 |
my_var *current_var; |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
62 |
|
63 |
if (unit->offset_limit_cnt) |
|
64 |
{ // using limit offset,count |
|
65 |
unit->offset_limit_cnt--; |
|
66 |
return(0); |
|
67 |
}
|
|
68 |
if (row_count++) |
|
69 |
{
|
|
70 |
my_message(ER_TOO_MANY_ROWS, ER(ER_TOO_MANY_ROWS), MYF(0)); |
|
71 |
return(1); |
|
72 |
}
|
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
73 |
while ((iter != var_list.end()) && (item= it++)) |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
74 |
{
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
75 |
current_var= *iter; |
76 |
if (current_var->local == 0) |
|
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
77 |
{
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
78 |
Item_func_set_user_var *suv= new Item_func_set_user_var(current_var->s, item); |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
79 |
suv->fix_fields(session, 0); |
80 |
suv->check(0); |
|
81 |
suv->update(); |
|
82 |
}
|
|
|
897.2.2
by Jay Pipes
Replaced custom List in select_dumpvar.h with std::vector<> |
83 |
++iter; |
|
882
by Brian Aker
Minor refactoring (we will need to disconnect the code from the include |
84 |
}
|
85 |
return(session->is_error()); |
|
86 |
}
|
|
87 |
||
88 |
bool send_eof() |
|
89 |
{
|
|
90 |
if (! row_count) |
|
91 |
push_warning(session, DRIZZLE_ERROR::WARN_LEVEL_WARN, |
|
92 |
ER_SP_FETCH_NO_DATA, ER(ER_SP_FETCH_NO_DATA)); |
|
93 |
/*
|
|
94 |
In order to remember the value of affected rows for ROW_COUNT()
|
|
95 |
function, SELECT INTO has to have an own SQLCOM.
|
|
96 |
TODO: split from SQLCOM_SELECT
|
|
97 |
*/
|
|
98 |
session->my_ok(row_count); |
|
99 |
return 0; |
|
100 |
}
|
|
101 |
||
|
851
by Brian Aker
Class rewrite of Session (aka get all of the junk out) |
102 |
};
|
103 |
||
104 |
#endif /* DRIZZLED_SELECT_DUMPVAR_H */ |