home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / sql / sql_udf.h < prev    next >
C/C++ Source or Header  |  2000-08-31  |  4KB  |  145 lines
  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.  
  17.  
  18. /* This file defines structures needed by udf functions */
  19.  
  20. #ifdef __GNUC__
  21. #pragma interface
  22. #endif
  23.  
  24. enum Item_udftype {UDFTYPE_FUNCTION=1,UDFTYPE_AGGREGATE};
  25.  
  26. typedef struct st_udf_func
  27. {
  28.   char *name;
  29.   int name_length;
  30.   Item_result returns;
  31.   Item_udftype type;
  32.   char *dl;
  33.   void *dlhandle;
  34.   void *func;
  35.   void *func_init;
  36.   void *func_deinit;
  37.   void *func_reset;
  38.   void *func_add;
  39.   ulong usage_count;
  40. } udf_func;
  41.  
  42. class Item_result_field;
  43. struct st_table_list;
  44.  
  45. class udf_handler :public Sql_alloc
  46. {
  47.  protected:
  48.   udf_func *u_d;
  49.   String *buffers;
  50.   UDF_ARGS f_args;
  51.   UDF_INIT initid;
  52.   char *num_buffer;
  53.   uchar error;
  54.   bool initialized;
  55.   Item **args;
  56.  
  57.  public:
  58.   table_map used_tables_cache;
  59.   bool const_item_cache;
  60.   udf_handler(udf_func *udf_arg) :u_d(udf_arg), buffers(0), error(0),
  61.     initialized(0)
  62.   {}
  63.   ~udf_handler();
  64.   const char *name() const { return u_d ? u_d->name : "?"; }
  65.   Item_result result_type () const
  66.   { return u_d    ? u_d->returns : STRING_RESULT;}
  67.   bool get_arguments();
  68.   bool fix_fields(THD *thd,struct st_table_list *tlist,Item_result_field *item,
  69.           uint arg_count,Item **args);
  70.   double val(my_bool *null_value)
  71.   {
  72.     if (get_arguments())
  73.     {
  74.       *null_value=1;
  75.       return 0.0;
  76.     }
  77.     uchar is_null=0;
  78.     double (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  79.       (double (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func;
  80.     double tmp=func(&initid, &f_args, &is_null, &error);
  81.     if (is_null || error)
  82.     {
  83.       *null_value=1;
  84.       return 0.0;
  85.     }
  86.     *null_value=0;
  87.     return tmp;
  88.   }
  89.   longlong val_int(my_bool *null_value)
  90.   {
  91.     if (get_arguments())
  92.     {
  93.       *null_value=1;
  94.       return LL(0);
  95.     }
  96.     uchar is_null=0;
  97.     longlong (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  98.       (longlong (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func;
  99.     longlong tmp=func(&initid, &f_args, &is_null, &error);
  100.     if (is_null || error)
  101.     {
  102.       *null_value=1;
  103.       return LL(0);
  104.     }
  105.     *null_value=0;
  106.     return tmp;
  107.   }
  108.   void reset(my_bool *null_value)
  109.   {
  110.     uchar is_null=0;
  111.     if (get_arguments())
  112.     {
  113.       *null_value=1;
  114.       return;
  115.     }
  116.     void (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  117.     (void (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func_reset;
  118.     func(&initid, &f_args, &is_null, &error);
  119.     *null_value= (my_bool) (is_null || error);
  120.   }
  121.   void add(my_bool *null_value)
  122.   {
  123.     uchar is_null=0;
  124.     if (get_arguments())
  125.     {
  126.       *null_value=1;
  127.       return;
  128.     }
  129.     void (*func)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)=
  130.     (void (*)(UDF_INIT *, UDF_ARGS *, uchar *, uchar *)) u_d->func_add;
  131.     func(&initid, &f_args, &is_null, &error);
  132.     *null_value= (my_bool) (is_null || error);
  133.   }
  134.   String *val_str(String *str,String *save_str);
  135. };
  136.  
  137.  
  138. #ifdef HAVE_DLOPEN
  139. void udf_init(void),udf_free(void);
  140. udf_func *find_udf(const char *name, uint len=0,bool mark_used=0);
  141. void free_udf(udf_func *udf);
  142. int mysql_create_function(THD *thd,udf_func *udf);
  143. int mysql_drop_function(THD *thd,const char *name);
  144. #endif
  145.