← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/sql_udf.cc

  • Committer: Brian Aker
  • Date: 2009-03-20 18:52:05 UTC
  • mfrom: (950.1.1 mordred)
  • Revision ID: brian@tangent.org-20090320185205-g7o6kq17r25b6odf
Merge Monty

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/sql_udf.cc
16
16
/* This implements 'user defined functions' */
17
17
#include <drizzled/server_includes.h>
18
18
#include <drizzled/gettext.h>
19
 
#include <mysys/hash.h>
20
19
#include <drizzled/sql_udf.h>
21
20
 
22
21
#include <map>
25
24
using namespace std;
26
25
 
27
26
static bool udf_startup= false; /* We do not lock because startup is single threaded */
28
 
static MEM_ROOT mem;
29
 
static map<string, udf_func *> udf_map;
30
 
 
31
 
extern "C" unsigned char* get_hash_key(const unsigned char *buff, size_t *length,
32
 
                               bool )
33
 
{
34
 
  udf_func *udf= (udf_func*) buff;
35
 
  *length= (uint32_t) udf->name.length;
36
 
  return (unsigned char*) udf->name.str;
37
 
}
38
 
 
39
 
 
40
 
void udf_init()
41
 
{
42
 
  init_sql_alloc(&mem, UDF_ALLOC_BLOCK_SIZE, 0);
43
 
}
44
 
 
45
 
/* called by mysqld.cc clean_up() */
46
 
void udf_free()
47
 
{
48
 
  free_root(&mem, MYF(0));
49
 
}
 
27
static map<string, Function_builder *> udf_map;
 
28
 
50
29
 
51
30
/* This is only called if using_udf_functions != 0 */
52
 
udf_func *find_udf(const char *name, uint32_t length)
 
31
Function_builder *find_udf(const char *name, uint32_t length)
53
32
{
54
 
  udf_func *udf= NULL;
 
33
 
 
34
  /**
 
35
   * @todo: check without transform, then check with transform if needed
 
36
   */
 
37
  Function_builder *udf= NULL;
55
38
 
56
39
  if (udf_startup == false)
57
40
    return NULL;
60
43
  transform(find_str.begin(), find_str.end(),
61
44
            find_str.begin(), ::tolower);
62
45
 
63
 
  map<string, udf_func *>::iterator find_iter;
 
46
  map<string, Function_builder *>::iterator find_iter;
64
47
  find_iter=  udf_map.find(find_str);
65
48
  if (find_iter != udf_map.end())
66
49
    udf= (*find_iter).second;
68
51
  return (udf);
69
52
}
70
53
 
71
 
static bool add_udf(udf_func *udf)
 
54
static bool add_udf(Function_builder *udf)
72
55
{
73
 
  string add_str(udf->name.str, udf->name.length);
 
56
  /**
 
57
   * @todo: add all lower and all upper version
 
58
   */
 
59
  string add_str= udf->get_name();
74
60
  transform(add_str.begin(), add_str.end(),
75
61
            add_str.begin(), ::tolower);
76
62
 
83
69
 
84
70
int initialize_udf(st_plugin_int *plugin)
85
71
{
86
 
  udf_func *f;
 
72
  Function_builder *f;
87
73
 
88
 
  if (udf_startup == false)
89
 
  {
90
 
    udf_init();
91
 
    udf_startup= true;
92
 
  }
 
74
  udf_startup= true;
93
75
 
94
76
  if (plugin->plugin->init)
95
77
  {
96
78
    int r;
97
79
    if ((r= plugin->plugin->init((void *)&f)))
98
80
    {
99
 
      errmsg_printf(ERRMSG_LVL_ERROR, "Plugin '%s' init function returned error %d.",
100
 
                    plugin->name.str, r);
 
81
      errmsg_printf(ERRMSG_LVL_ERROR,
 
82
                    _("Plugin '%s' init function returned error %d."),
 
83
                    plugin->name.str, r);
101
84
      return r;
102
85
    }
103
86
  }
115
98
 
116
99
int finalize_udf(st_plugin_int *plugin)
117
100
{
118
 
  udf_func *udff = (udf_func *)plugin->data;
 
101
  Function_builder *udff = (Function_builder *)plugin->data;
119
102
 
120
103
  /* TODO: Issue warning on failure */
121
104
  if (udff && plugin->plugin->deinit)