force merge/finalize function exists

c2c1a247 · dapan1121 · 48444692 · c2c1a247
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

src/query/src/qExecutor.c src/query/src/qExecutor.c +3 -0

未找到文件。
--- a/src/query/src/qExecutor.c
+++ b/src/query/src/qExecutor.c
@@ -8107,6 +8107,9 @@ int32_t initUdfInfo(SUdfInfo* pUdfInfo) {
    if (pUdfInfo->funcType == TSDB_UDF_TYPE_AGGREGATE) {
      pUdfInfo->funcs[TSDB_UDF_FUNC_FINALIZE] = taosLoadSym(pUdfInfo->handle, getUdfFuncName(funcname, pUdfInfo->name, TSDB_UDF_FUNC_FINALIZE));
      pUdfInfo->funcs[TSDB_UDF_FUNC_MERGE] = taosLoadSym(pUdfInfo->handle, getUdfFuncName(funcname, pUdfInfo->name, TSDB_UDF_FUNC_MERGE));
+      if (NULL == pUdfInfo->funcs[TSDB_UDF_FUNC_MERGE] || NULL == pUdfInfo->funcs[TSDB_UDF_FUNC_FINALIZE]) {
+        return TSDB_CODE_QRY_SYS_ERROR;
+      }
    }

    pUdfInfo->funcs[TSDB_UDF_FUNC_DESTROY] = taosLoadSym(pUdfInfo->handle, getUdfFuncName(funcname, pUdfInfo->name, TSDB_UDF_FUNC_DESTROY));