← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/select_dumpvar.h

merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/select_dumpvar.h
22
22
 
23
23
#include "drizzled/error.h"
24
24
#include "drizzled/function/set_user_var.h"
 
25
#include "drizzled/select_result_interceptor.h"
 
26
#include <drizzled/base.h>
25
27
 
26
28
#include <vector>
27
29
 
28
30
namespace drizzled
29
31
{
30
32
 
 
33
class var;
 
34
 
31
35
class select_dumpvar :public select_result_interceptor {
32
36
  ha_rows row_count;
 
37
 
33
38
public:
34
39
  std::vector<var *> var_list;
35
40
  select_dumpvar()  { var_list.clear(); row_count= 0;}
36
41
  ~select_dumpvar() {}
37
42
 
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
 
  }
 
43
  int prepare(List<Item> &list, Select_Lex_Unit *u);
50
44
 
51
45
  void cleanup()
52
46
  {
54
48
  }
55
49
 
56
50
 
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
 
  }
 
51
  bool send_data(List<Item> &items);
 
52
 
 
53
  bool send_eof();
104
54
 
105
55
};
106
56