94 lines
4.4 KiB
Markdown
94 lines
4.4 KiB
Markdown
# User Define Function
|
|
|
|
用户可以通过UDF机制来扩展Doris的能力。通过这篇文档，用户能够创建自己的UDF。
|
|
|
|
## 编写UDF函数
|
|
|
|
在使用UDF之前，用户需要先在Doris的UDF框架下，编写自己的UDF函数。在`be/src/udf_samples/udf_sample.h|cpp`文件中是一个简单的UDF Demo。
|
|
|
|
编写一个UDF函数需要以下几个步骤。
|
|
|
|
### 编写函数
|
|
|
|
创建对应的头文件、CPP文件，在CPP文件中实现你需要的逻辑。CPP文件中的实现函数格式与UDF的对应关系。
|
|
|
|
#### 非可变参数
|
|
|
|
对于非可变参数的UDF，那么两者之间的对应关系很直接。
|
|
比如`INT MyADD(INT, INT)`的UDF就会对应`IntVal AddUdf(FunctionContext* context, const IntVal& arg1, const IntVal& arg2)`。
|
|
|
|
1. `AddUdf`可以为任意的名字，只要创建UDF的时候指定即可。
|
|
2. 实现函数中的第一个参数永远是`FunctionContext*`。实现者可以通过这个结构体获得一些查询相关的内容，以及申请一些需要使用的内存。具体使用的接口可以参考`udf/udf.h`中的定义。
|
|
3. 实现函数中从第二个参数开始需要与UDF的参数一一对应，比如`IntVal`对应`INT`类型。这部分的类型都要使用`const`引用。
|
|
4. 返回参数与UDF的参数的类型要相对应。
|
|
|
|
#### 可变参数
|
|
|
|
对于可变参数，可以参见以下例子，UDF`String md5sum(String, ...)`对应的
|
|
实现函数是`StringVal md5sumUdf(FunctionContext* ctx, int num_args, const StringVal* args)`
|
|
|
|
1. `md5sumUdf`这个也是可以任意改变的，创建的时候指定即可。
|
|
2. 第一个参数与非可变参数函数一样，传入的是一个`FunctionContext*`。
|
|
3. 可变参数部分由两部分组成，首先会传入一个整数，说明后面还有几个参数。后面传入的是一个可变参数部分的数组。
|
|
|
|
#### 类型对应关系
|
|
|
|
|UDF Type|Argument Type|
|
|
|----|---------|
|
|
|TinyInt|TinyIntVal|
|
|
|SmallInt|SmallIntVal|
|
|
|Int|IntVal|
|
|
|BigInt|BigIntVal|
|
|
|LargeInt|LargeIntVal|
|
|
|Float|FloatVal|
|
|
|Double|DoubleVal|
|
|
|Date|DateTimeVal|
|
|
|Datetime|DateTimeVal|
|
|
|Char|StringVal|
|
|
|Varchar|StringVal|
|
|
|Decimal|DecimalVal|
|
|
|
|
## 编译UDF函数
|
|
|
|
### 编译Doris
|
|
|
|
在Doris根目录下执行`sh build.sh`就会在`output/udf/`生成对应`headers|libs`
|
|
|
|
### 编写CMakeLists.txt
|
|
|
|
基于上一步生成的`headers|libs`，用户可以使用`CMakeLists`等工具引入该依赖；在`CMakeLists`中，可以通过向`CMAKE_CXX_FLAGS`添加`-I|L`分别指定`headers|libs`路径；然后使用`add_library`添加动态库。例如，在`be/src/udf_samples/CMakeLists.txt`中，使用`add_library(udfsample SHARED udf_sample.cpp)`增加了一个`udfsample`动态库。后面需要写上涉及的所有源文件（不包含头文件）。
|
|
|
|
### 执行编译
|
|
|
|
在该目录下创建一个`build`目录并在`build`下执行`cmake ../`生成`Makefile`，并执行`make`就会生成对应动态库。
|
|
|
|
## 创建UDF函数
|
|
|
|
通过上述的步骤后，你可以得到一个动态库。你需要将这个动态库放到一个能够通过HTTP协议访问到的位置。然后执行创建UDF函数在Doris系统内部创建一个UDF，你需要拥有AMDIN权限才能够完成这个操作。
|
|
|
|
```
|
|
CREATE [AGGREGATE] FUNCTION
|
|
name ([argtype][,...])
|
|
[RETURNS] rettype
|
|
PROPERTIES (["key"="value"][,...])
|
|
```
|
|
说明：
|
|
|
|
1. PROPERTIES中`symbol`表示的是，执行入口函数的对应symbol，这个参数是必须设定。你可以通过`nm`命令来获得对应的symbol，比如`nm libudfsample.so | grep AddUdf`获得到的`_ZN9doris_udf6AddUdfEPNS_15FunctionContextERKNS_6IntValES4_`就是对应的symbol。
|
|
2. PROPERTIES中`object_file`表示的是从哪里能够下载到对应的动态库，这个参数是必须设定的。
|
|
3. name: 一个function是要归属于某个DB的，name的形式为`dbName`.`funcName`。当`dbName`没有明确指定的时候，就是使用当前session所在的db作为`dbName`。
|
|
|
|
具体使用可以参见 `CREATE FUNCTION` 获取更详细信息。
|
|
|
|
## 使用UDF
|
|
|
|
用户使用UDF/UDAF必须拥有对应数据库的 `SELECT` 权限。
|
|
|
|
UDF的使用与普通的函数方式一致，唯一的区别在于，内置函数的作用域是全局的，而UDF的作用域是DB内部。当链接session位于数据内部时，直接使用UDF名字会在当前DB内部查找对应的UDF。否则用户需要显示的指定UDF的数据库名字，例如`dbName`.`funcName`。
|
|
|
|
|
|
## 删除UDF函数
|
|
|
|
当你不再需要UDF函数时，你可以通过下述命令来删除一个UDF函数, 可以参考 `DROP FUNCTION`。
|
|
|