← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/select_dumpvar.h

  • Committer: Brian Aker
  • Date: 2011-02-17 10:09:00 UTC
  • mfrom: (2173.2.1 clean-include-usuage)
  • Revision ID: brian@tangent.org-20110217100900-4tpuxxzdl1sj00sh
Merge Monty for headers.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/select_dumpvar.h
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems
 
4
 *  Copyright (C) 2008 Sun Microsystems, Inc.
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
20
20
#ifndef DRIZZLED_SELECT_DUMPVAR_H
21
21
#define DRIZZLED_SELECT_DUMPVAR_H
22
22
 
23
 
#include "drizzled/error.h"
24
 
#include "drizzled/function/set_user_var.h"
 
23
#include <drizzled/error.h>
 
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