← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/select_dumpvar.h

  • Committer: Muhammad Umair
  • Date: 2011-08-16 11:47:29 UTC
  • mto: This revision was merged to the branch mainline in revision 2402.
  • Revision ID: umair@remotedesk-20110816114729-w6x88fj0sow4g3z9
merge lp:~mumair/drizzle/drizzle-IPv6Address

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/select_dumpvar.h
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#ifndef DRIZZLED_SELECT_DUMPVAR_H
21
 
#define DRIZZLED_SELECT_DUMPVAR_H
 
20
#pragma once
22
21
 
23
 
#include "drizzled/error.h"
24
 
#include "drizzled/function/set_user_var.h"
 
22
#include <drizzled/error.h>
 
23
#include <drizzled/function/set_user_var.h>
 
24
#include <drizzled/select_result_interceptor.h>
 
25
#include <drizzled/base.h>
25
26
 
26
27
#include <vector>
27
28
 
28
 
namespace drizzled
29
 
{
 
29
namespace drizzled {
30
30
 
31
31
class select_dumpvar :public select_result_interceptor {
32
32
  ha_rows row_count;
 
33
 
33
34
public:
34
35
  std::vector<var *> var_list;
35
36
  select_dumpvar()  { var_list.clear(); row_count= 0;}
36
37
  ~select_dumpvar() {}
37
38
 
38
 
  int prepare(List<Item> &list, Select_Lex_Unit *u)
39
 
  {
40
 
    unit= u;
41
 
 
42
 
    if (var_list.size() != list.elements)
43
 
    {
44
 
      my_message(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT,
45
 
                 ER(ER_WRONG_NUMBER_OF_COLUMNS_IN_SELECT), MYF(0));
46
 
      return 1;
47
 
    }
48
 
    return 0;
49
 
  }
 
39
  int prepare(List<Item> &list, Select_Lex_Unit *u);
50
40
 
51
41
  void cleanup()
52
42
  {
54
44
  }
55
45
 
56
46
 
57
 
  bool send_data(List<Item> &items)
58
 
  {
59
 
    
60
 
    std::vector<var *>::const_iterator iter= var_list.begin();
61
 
 
62
 
    List_iterator<Item> it(items);
63
 
    Item *item= NULL;
64
 
    var *current_var;
65
 
 
66
 
    if (unit->offset_limit_cnt)
67
 
    {                                           // using limit offset,count
68
 
      unit->offset_limit_cnt--;
69
 
      return(0);
70
 
    }
71
 
    if (row_count++)
72
 
    {
73
 
      my_message(ER_TOO_MANY_ROWS, ER(ER_TOO_MANY_ROWS), MYF(0));
74
 
      return(1);
75
 
    }
76
 
    while ((iter != var_list.end()) && (item= it++))
77
 
    {
78
 
      current_var= *iter;
79
 
      if (current_var->local == 0)
80
 
      {
81
 
        Item_func_set_user_var *suv= new Item_func_set_user_var(current_var->s, item);
82
 
        suv->fix_fields(session, 0);
83
 
        suv->check(0);
84
 
        suv->update();
85
 
      }
86
 
      ++iter;
87
 
    }
88
 
    return(session->is_error());
89
 
  }
90
 
 
91
 
  bool send_eof()
92
 
  {
93
 
    if (! row_count)
94
 
      push_warning(session, DRIZZLE_ERROR::WARN_LEVEL_WARN,
95
 
                   ER_SP_FETCH_NO_DATA, ER(ER_SP_FETCH_NO_DATA));
96
 
    /*
97
 
      In order to remember the value of affected rows for ROW_COUNT()
98
 
      function, SELECT INTO has to have an own SQLCOM.
99
 
TODO: split from SQLCOM_SELECT
100
 
  */
101
 
    session->my_ok(row_count);
102
 
    return 0;
103
 
  }
 
47
  bool send_data(List<Item> &items);
 
48
 
 
49
  bool send_eof();
104
50
 
105
51
};
106
52
 
107
53
} /* namespace drizzled */
108
54
 
109
 
#endif /* DRIZZLED_SELECT_DUMPVAR_H */