xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: dispatch to udf by language type and udf name

This commit is contained in:
slzhou 2023-02-14 13:08:32 +08:00
parent 45d42d8b1d
commit 2e266771b1
1 changed files with 29 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
source/libs/function/src/udfd.c
View File

@ -329,7 +329,7 @@ int32_t udfdLoadSharedLib(char *libPath, uv_lib_t *pLib, const char *funcName[],
void udfdInitializePythonPlugin(SUdfScriptPlugin *plugin) { void udfdInitializePythonPlugin(SUdfScriptPlugin *plugin) {
plugin->scriptType = TSDB_FUNC_SCRIPT_PYTHON; plugin->scriptType = TSDB_FUNC_SCRIPT_PYTHON;
sprintf("%s", plugin->libPath, "libtaosudf_py.so"); sprintf(plugin->libPath, "%s", "libtaosudf_py.so");
plugin->libLoaded = false; plugin->libLoaded = false;
const char *funcName[MAX_NUM_PLUGIN_FUNCS] = {"open", "close", "udfInit", const char *funcName[MAX_NUM_PLUGIN_FUNCS] = {"open", "close", "udfInit",
"udfDestroy", "udfScalarProc", "udfAggStart", "udfDestroy", "udfScalarProc", "udfAggStart",
@ -347,17 +347,41 @@ void udfdInitializePythonPlugin(SUdfScriptPlugin *plugin) {
return; return;
} }
void udfdDeinitCPlugin(SUdfScriptPlugin *plugin) {
return;
}
void udfdDeinitPythonPlugin(SUdfScriptPlugin *plugin) {
if (plugin->libLoaded) {
uv_dlclose(&plugin->lib);
plugin->libLoaded = false;
}
}
void udfdInitScriptPlugins() { void udfdInitScriptPlugins() {
SUdfScriptPlugin *plugins = taosMemoryCalloc(2, sizeof(SUdfScriptPlugin)); SUdfScriptPlugin *plugin = NULL;
// Initialize c language plugin
udfdInitializeCPlugin(plugins + 0); // Initialize c plugin
plugin = taosMemoryCalloc(1, sizeof(SUdfScriptPlugin));
udfdInitializeCPlugin(plugin);
global.scriptPlugins[TSDB_FUNC_SCRIPT_BIN_LIB] = plugin;
// Initialize python plugin // Initialize python plugin
udfdInitializePythonPlugin(plugins + 1); plugin = taosMemoryCalloc(1, sizeof(SUdfScriptPlugin));
udfdInitializePythonPlugin(plugin);
global.scriptPlugins[TSDB_FUNC_SCRIPT_PYTHON] = plugin;
return; return;
} }
void udfdDeinitScriptPlugins() { void udfdDeinitScriptPlugins() {
SUdfScriptPlugin *plugin = NULL;
plugin = global.scriptPlugins[TSDB_FUNC_SCRIPT_PYTHON];
udfdDeinitPythonPlugin(plugin);
taosMemoryFree(plugin);
plugin = global.scriptPlugins[TSDB_FUNC_SCRIPT_BIN_LIB];
udfdDeinitCPlugin(plugin);
taosMemoryFree(plugin);
return; return;
} }