← Back to branch summary

~jaypipes/drizzle/split-xa-resource-manager

« back to all changes in this revision

Viewing changes to drizzled/plugin/table_function.cc

  • Committer: Jay Pipes
  • Date: 2010-02-14 20:26:43 UTC
  • mfrom: (1273.1.27 staging)
  • Revision ID: jpipes@serialcoder-20100214202643-ahuqvc8rhn8u7y33
Merge trunk and resolve conflicts

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/plugin/table_function.cc
 
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2010 Monty Taylor
 
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
#include "config.h"
 
21
 
 
22
#include <drizzled/plugin/table_function.h>
 
23
#include <drizzled/table_function_container.h>
 
24
#include <drizzled/gettext.h>
 
25
#include "drizzled/plugin/registry.h"
 
26
#include "drizzled/global_charset_info.h"
 
27
#include "drizzled/session.h"
 
28
#include "drizzled/current_session.h"
 
29
 
 
30
#include <vector>
 
31
 
 
32
using namespace std;
 
33
 
 
34
namespace drizzled
 
35
{
 
36
 
 
37
static TableFunctionContainer table_functions;
 
38
 
 
39
void plugin::TableFunction::init()
 
40
{
 
41
  drizzled::message::Table::StorageEngine *engine;
 
42
  drizzled::message::Table::TableOptions *table_options;
 
43
 
 
44
  proto.set_name(identifier.getTableName());
 
45
  proto.set_type(drizzled::message::Table::FUNCTION);
 
46
 
 
47
  table_options= proto.mutable_options();
 
48
  table_options->set_collation_id(default_charset_info->number);
 
49
  table_options->set_collation(default_charset_info->name);
 
50
 
 
51
  engine= proto.mutable_engine();
 
52
  engine->set_name("FunctionEngine");
 
53
}
 
54
 
 
55
bool plugin::TableFunction::addPlugin(plugin::TableFunction *tool)
 
56
{
 
57
  assert(tool != NULL);
 
58
  table_functions.addFunction(tool); 
 
59
  return false;
 
60
}
 
61
 
 
62
plugin::TableFunction *plugin::TableFunction::getFunction(const string &arg)
 
63
{
 
64
  return table_functions.getFunction(arg);
 
65
}
 
66
 
 
67
void plugin::TableFunction::getNames(const string &arg,
 
68
                                     set<std::string> &set_of_names)
 
69
{
 
70
  table_functions.getNames(arg, set_of_names);
 
71
}
 
72
 
 
73
plugin::TableFunction::Generator *plugin::TableFunction::generator(Field **arg)
 
74
{
 
75
  return new Generator(arg);
 
76
}
 
77
 
 
78
void plugin::TableFunction::add_field(const char *label,
 
79
                                      uint32_t field_length)
 
80
{
 
81
  add_field(label, TableFunction::STRING, field_length);
 
82
}
 
83
 
 
84
void plugin::TableFunction::add_field(const char *label,
 
85
                              TableFunction::ColumnType type,
 
86
                              bool is_default_null)
 
87
{
 
88
  add_field(label, type, 5, is_default_null);
 
89
}
 
90
 
 
91
void plugin::TableFunction::add_field(const char *label,
 
92
                              TableFunction::ColumnType type,
 
93
                              uint32_t field_length,
 
94
                              bool is_default_null)
 
95
{
 
96
  drizzled::message::Table::Field *field;
 
97
  drizzled::message::Table::Field::FieldOptions *field_options;
 
98
  drizzled::message::Table::Field::FieldConstraints *field_constraints;
 
99
 
 
100
  field= proto.add_field();
 
101
  field->set_name(label);
 
102
 
 
103
  field_options= field->mutable_options();
 
104
  field_constraints= field->mutable_constraints();
 
105
  field_options->set_default_null(is_default_null);
 
106
  field_constraints->set_is_nullable(is_default_null);
 
107
 
 
108
  switch (type) 
 
109
  {
 
110
  default:
 
111
  case TableFunction::BOOLEAN: // Currently BOOLEAN always has a value
 
112
    field_length= 5;
 
113
    field_options->set_default_null(false);
 
114
    field_constraints->set_is_nullable(false);
 
115
  case TableFunction::STRING:
 
116
    drizzled::message::Table::Field::StringFieldOptions *string_field_options;
 
117
    field->set_type(drizzled::message::Table::Field::VARCHAR);
 
118
 
 
119
    string_field_options= field->mutable_string_options();
 
120
    string_field_options->set_length(field_length);
 
121
 
 
122
    break;
 
123
  case TableFunction::NUMBER: // Currently NUMBER always has a value
 
124
    field->set_type(drizzled::message::Table::Field::BIGINT);
 
125
    field_options->set_default_null(false);
 
126
    field_constraints->set_is_nullable(false);
 
127
    break;
 
128
  }
 
129
}
 
130
 
 
131
plugin::TableFunction::Generator::Generator(Field **arg) :
 
132
  columns(arg)
 
133
{
 
134
  scs= system_charset_info;
 
135
}
 
136
 
 
137
bool plugin::TableFunction::Generator::sub_populate(uint32_t field_size)
 
138
{
 
139
  bool ret;
 
140
  uint64_t difference;
 
141
 
 
142
  columns_iterator= columns;
 
143
  ret= populate();
 
144
  difference= columns_iterator - columns;
 
145
 
 
146
  if (ret == true)
 
147
  {
 
148
    assert(difference == field_size);
 
149
  }
 
150
 
 
151
  return ret;
 
152
}
 
153
 
 
154
void plugin::TableFunction::Generator::push(uint64_t arg)
 
155
{
 
156
  (*columns_iterator)->store(static_cast<int64_t>(arg), false);
 
157
  columns_iterator++;
 
158
}
 
159
 
 
160
void plugin::TableFunction::Generator::push(uint32_t arg)
 
161
{
 
162
  (*columns_iterator)->store(static_cast<int64_t>(arg), false);
 
163
  columns_iterator++;
 
164
}
 
165
 
 
166
void plugin::TableFunction::Generator::push(int64_t arg)
 
167
{
 
168
  (*columns_iterator)->store(arg, false);
 
169
  columns_iterator++;
 
170
}
 
171
 
 
172
void plugin::TableFunction::Generator::push(int32_t arg)
 
173
{
 
174
  (*columns_iterator)->store(arg, false);
 
175
  columns_iterator++;
 
176
}
 
177
 
 
178
void plugin::TableFunction::Generator::push(const char *arg, uint32_t length)
 
179
{
 
180
  assert(columns_iterator);
 
181
  assert(*columns_iterator);
 
182
  assert(arg);
 
183
  length= length ? length : strlen(arg);
 
184
 
 
185
  if (not length)
 
186
    return push();
 
187
 
 
188
  (*columns_iterator)->store(arg, length, scs);
 
189
  (*columns_iterator)->set_notnull();
 
190
  columns_iterator++;
 
191
}
 
192
 
 
193
void plugin::TableFunction::Generator::push()
 
194
{
 
195
  assert((*columns_iterator)->type()  == DRIZZLE_TYPE_VARCHAR);
 
196
  (*columns_iterator)->set_null();
 
197
  columns_iterator++;
 
198
}
 
199
 
 
200
void plugin::TableFunction::Generator::push(const std::string& arg)
 
201
{
 
202
  push(arg.c_str(), arg.length());
 
203
}
 
204
 
 
205
void plugin::TableFunction::Generator::push(bool arg)
 
206
{
 
207
  if (arg)
 
208
  {
 
209
    (*columns_iterator)->store("TRUE", 4, scs);
 
210
  }
 
211
  else
 
212
  {
 
213
    (*columns_iterator)->store("FALSE", 5, scs);
 
214
  }
 
215
 
 
216
  columns_iterator++;
 
217
}
 
218
 
 
219
bool plugin::TableFunction::Generator::isWild(const std::string &predicate)
 
220
{
 
221
  Session *session= current_session;
 
222
 
 
223
  if (not session->lex->wild)
 
224
    return false;
 
225
 
 
226
  bool match= wild_case_compare(system_charset_info,
 
227
                                predicate.c_str(),
 
228
                                session->lex->wild->ptr());
 
229
 
 
230
  return match;
 
231
}
 
232
 
 
233
} /* namespace drizzled */