sql_udf.h
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  7.    
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  16. /* This file defines structures needed by udf functions */
  17. #ifdef __GNUC__
  18. #pragma interface
  19. #endif
  20. enum Item_udftype {UDFTYPE_FUNCTION=1,UDFTYPE_AGGREGATE};
  21. typedef struct st_udf_func
  22. {
  23.   char *name;
  24.   int name_length;
  25.   Item_result returns;
  26.   Item_udftype type;
  27.   char *dl;
  28.   void *dlhandle;
  29.   void *func;
  30.   void *func_init;
  31.   void *func_deinit;
  32.   void *func_reset;
  33.   void *func_add;
  34.   ulong usage_count;
  35. } udf_func;
  36. class Item_result_field;
  37. struct st_table_list;
  38. class udf_handler :public Sql_alloc
  39. {
  40.  protected:
  41.   udf_func *u_d;
  42.   String *buffers;
  43.   UDF_ARGS f_args;
  44.   UDF_INIT initid;
  45.   char *num_buffer;
  46.   uchar error;
  47.   bool initialized;
  48.   Item **args;
  49.  public:
  50.   table_map used_tables_cache;
  51.   bool const_item_cache;
  52.   udf_handler(udf_func *udf_arg) :u_d(udf_arg), buffers(0), error(0),
  53.     initialized(0)
  54.   {}
  55.   ~udf_handler();
  56.   const char *name() const { return u_d ? u_d->name : "?"; }
  57.   Item_result result_type () const
  58.   { return u_d ? u_d->returns : STRING_RESULT;}
  59.   bool get_arguments();
  60.   bool fix_fields(THD *thd,struct st_table_list *tlist,Item_result_field *item,
  61.   uint arg_count,Item **args);
  62.   double val(my_bool *null_value)
  63.   {
  64.     if (get_arguments())
  65.     {
  66.       *null_value=1;
  67.       return 0.0;
  68.     }
  69.     uchar is_null=0;
  70.     double (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  71.       (double (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func;
  72.     double tmp=func(&initid, &f_args, &is_null, &error);
  73.     if (is_null || error)
  74.     {
  75.       *null_value=1;
  76.       return 0.0;
  77.     }
  78.     *null_value=0;
  79.     return tmp;
  80.   }
  81.   longlong val_int(my_bool *null_value)
  82.   {
  83.     if (get_arguments())
  84.     {
  85.       *null_value=1;
  86.       return LL(0);
  87.     }
  88.     uchar is_null=0;
  89.     longlong (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  90.       (longlong (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func;
  91.     longlong tmp=func(&initid, &f_args, &is_null, &error);
  92.     if (is_null || error)
  93.     {
  94.       *null_value=1;
  95.       return LL(0);
  96.     }
  97.     *null_value=0;
  98.     return tmp;
  99.   }
  100.   void reset(my_bool *null_value)
  101.   {
  102.     uchar is_null=0;
  103.     if (get_arguments())
  104.     {
  105.       *null_value=1;
  106.       return;
  107.     }
  108.     void (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  109.     (void (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func_reset;
  110.     func(&initid, &f_args, &is_null, &error);
  111.     *null_value= (my_bool) (is_null || error);
  112.   }
  113.   void add(my_bool *null_value)
  114.   {
  115.     uchar is_null=0;
  116.     if (get_arguments())
  117.     {
  118.       *null_value=1;
  119.       return;
  120.     }
  121.     void (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  122.     (void (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func_add;
  123.     func(&initid, &f_args, &is_null, &error);
  124.     *null_value= (my_bool) (is_null || error);
  125.   }
  126.   String *val_str(String *str,String *save_str);
  127. };
  128. #ifdef HAVE_DLOPEN
  129. void udf_init(void),udf_free(void);
  130. udf_func *find_udf(const char *name, uint len=0,bool mark_used=0);
  131. void free_udf(udf_func *udf);
  132. int mysql_create_function(THD *thd,udf_func *udf);
  133. int mysql_drop_function(THD *thd,const char *name);
  134. #endif