MYSQL用户定义函数（UDF）

UDF资源文件

UDF测试程序

这里解释什么是用户定义函数（UDF），是做什么的和怎样使用
1）用户定义函数（UDF）是什么？
基本用户定义函数是一类代码，对MYSQL服务器功能进行扩充，通过添加新函数，性质就象使用本地MYSQL函数abs()或concat().UDF是用C（或C++）写的。也许还可以用BASIC，.NET或其它什么虽然我还没见过有人这么做。
2）为什么和怎样使用UDF功能？
从字面上何以知道UDF是很有用的，尤其当你需要扩展MYSQL服务器功能时。下表给出了最佳解决方法的比较：

Method Speed Language Development
 方法 速度 语言 开发
Stored Procedures slow SQL ~minutes (for small functions)
 存储过程 慢 SQL ~分钟（对于小函数）
UDF fast C ~hour
 UDF 快 C ~小时
Native Function fast C major pain in the ***
本地函数 快 C 未知


慢的意思是和其它比较时。存储过程和一般SQL语句比仍然是很快的。
对本地函数的一点解释：本质上和UDF没太大区别。但是你必须用MYSQL的资源代码来写然后重新编译全部。这将是很大的工作量，你必须一边又一边的用最新版的MYSQL来完成这项工作。
3）怎样使用UDF？
这部分很简单。当你完成了一个UDF，你只是使用它就可以了。例如："SELECT MyFunction(data1, data2) FROM table"
4）编写UDF
现在我们开发写一个UDF：
建立一个新的shared-library项目（该例中我用的VC++ 6.0建立一个标准的DLL）
首先我需要一些头文件。这些头文件是标准的头文件和MYSQL服务器的包含目录里的文件
#ifdef STANDARD
/* STANDARD is defined, don't use any mysql functions */
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#ifdef __WIN__
typedef unsigned __int64 ulonglong; /* Microsofts 64 bit types */
typedef __int64 longlong;
#else
typedef unsigned long long ulonglong;
typedef long long longlong;
#endif /*__WIN__*/
#else
#include <my_global.h>
#include <my_sys.h>
#endif
#include <mysql.h>
#include <ctype.h>
static pthread_mutex_t LOCK_hostname;


 
现在我必须决定我们需要哪类函数。本质上有两种选择：
该函数是聚合函数吗？（后面我们将学习很多关于聚合函数的内容）
返回类型是什么？有4个选择：
类型 描述
STRING 一个合法的字符串，转换成char*类型
INTEGER 一个普通的整型变量，转换成64位的整型
REAL型 一个俘点数，转换成double型
DECIAML型 这个并没真正的结束，MYSQL将做字符串对待

 
现在开始讨论关于非聚合函数。必须声明并执行一些MYSQL使用UDF时用到的函数，但首先一些必要的结构必须并确：
UDF_INIT:
类型 名称 描述
<code> 
my_bool maybe_null 是1如果函数能返回NULL
unsigned int decimals 针对REAL函数
unsigned long max_length 针对字符串函数
char * ptr 自由指针针对函数的数据
my_bool const_item 0如果结果是独立的

UDF_ARGS:
类型 名称 描述
unsigned int arg_count 成员数量
enum Item_result * arg_type 成员类型的数组
char ** args 指向成员的指针的数组
unsigned long * lengths 成员长度的数组（针对字符串）
char * maybe_null "maybe_null"标记的数组
char ** attributes 指向成员属性的指针的数组
unsigned long * attribute_lengths 属性长度数组


现在看一下该函数：
De-/Initialization:

 Collapseextern "C" my_bool MyTest_init(UDF_INIT *initid, UDF_ARGS *args, 
  char *message)
{
   
  //非常重要的一件事是建立内存
  //你需要
  //我们需要一个很长的变量来保存检测数
  //虽然该例中我们不需要
  longlong* i = new longlong; // 建立变量
  *i = 0; // 设初值
   
   
  //指针变量中保存为一个字符指针
  //确认你不会遇到类型问题
  initid->ptr = (char*)i;
   

  //检测成员的格式
  if (args->arg_count != 1)
  {
  strcpy(message,"MyTest() requires one arguments");
  return 1;
  }

  if (args->arg_type[0] != INT_RESULT)
  {
  strcpy(message,"MyTest() requires an integer");
  return 1;
  }  
  return 0;  
}

extern "C" void MyTest_deinit(UDF_INIT *initid)
{
   
  //这里必须清空所分配的内存
  //引入函数
  delete (longlong*)initid->ptr;
}

The actual function:

extern "C" longlong MyTest(UDF_INIT *initid, UDF_ARGS *args,
  char *is_null, char *error)
{
   
  /*最后这是我们实际的工作部分。该函数为每个记录调用，返回值或指向当前值的指针保存在UDF_ARGS变量中。我们必须获得值，完成计算并返回值。注意你可以通过UDF_INIT变量进入MyTest_init中分配的内存，该例中我们将为每个值设置为5
  */
  return *((longlong*)args->args[0])+5;
}
全部完成！现在必须编译连接库，然后将其拷贝到操作系统可以加载的目录下。通常在WINDOWS里是系统变量的定义路径。我个人使用的是MYSQL服务器的bin目录。你必须确认该目录是其它MYSQL不能访问的。然后确认所有MYSQL需要的函数功能。
我们必须告诉MYSQL，这必须直接了当：执行以下SQL指令：
CREATE [AGGREGATE] FUNCTION MyTest 
RETURNS [INTEGER|STRING|REAL|DECIMAL] SONAME the_libraries_exact_name
现在你可以想使用其他函数一样使用它了。
5）成员函数：
现在说一下成员函数。当你的UDF是你个成员函数，你必须增加一些函数，一些函数在不同的方式中使用。调用次序是：
调用yTest_init来分配内存（就象一般的UDF一样）
MYSQL将表分类是通过GROUP BY
每组里的第一行调用MyTest_clear
每组里的第一列调用 MyTest_add
在组改变后或最后一列改变后调用MyTest得到结果
重复3到5直到所有列被处理。
调用MyTest_deinit清空内存
现在让我们看一下新的聚合函数所需的函数。该例中将简单的添加所有的值。（就象本地SUM函数）
void MyTest_clear(UDF_INIT *initid, char *is_null, char *error)
{
   
  /*为每个新组重新将总数设置为0，当然你必须分配一个longlong类型变量在在init函数中，并分配给指针
  */
  *((longlong*)initid->ptr) = 0;
}

void MyTest_add(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error)
{
   
  //为每列将当前值添加到总数
  *((longlong*)initid->ptr) = *((longlong*)initid->ptr) + 
  *((longlong*)args->args[0]);
}

longlong MyTest(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error)
{
   
  //最后返回总值
  return *((longlong*)initid->ptr);
}

6）更进一步的问题：
在你写一些复杂的UDF时需要注意几个问题：
一个字符串函数应该返回一个指向结果的指针并且设置*result和*length作为目录和返回值的长度值。例如：
memcpy(result, "result string", 13);
*length = 13;
MyTest建立的结果缓冲区是255字节。如果你的结果保存在里面。你不必担心结果的内存分配问题。
如果你的字符串函数需要返回一个大于255字节长度的字符串。你必须用malloc或新的MyTest_init或MyTest函数分配，然后用MyTest_deinit释放它。你能用UDF_INIT的指针保存分配的内存地址，并在MyTest中重用。
在主函数中指定一个错误返回，设置 *error为1：如果MyTest()为任何列将*error设置为1，则函数的值是NULL针对于当前列，以及对任何的通过MyTest()被调用的声明中并发的列请求。
想了解更多内容看一下MYSQL在线帮助。


7）一些指导方针：
如果你确实希望你的UDF运行良好，这里有一些建议：）
不要在UDF中调用任何其他的程序或进程
不要保存任何的本地信息。（这些在普通的库里已经共享）
不要分配任何的全局或静态的变量。
始终检测成员的类型。就象MYSQL将所有类型都转换为字符类型。如果你将一个字符类型转换成整型指针可能会出错。
特别注意内存的分配。如果你有内存泄漏问题你会使服务器彻底崩溃！


8）调式UDF
调试UDF需要勇气因为如果UDF有问题，每次都会使整个MYSQL服务器死掉。所以我写了一个命令行工具，来围绕这个问题工作。仅仅运行它，它会模仿"SELECT"调用指令将结果保存到库文件中，可以打印所有的结果行。所以当UDF存在一些错误只是该程序崩溃而不会是整个服务器。