enh: check level

This commit is contained in:
facetosea 2024-11-11 17:23:13 +08:00
parent bf9ea14f86
commit 61ccba3ce7
1 changed files with 15 additions and 17 deletions

View File

@ -543,10 +543,6 @@ void udfdDeinitScriptPlugins() {
void udfdProcessRequest(uv_work_t *req) { void udfdProcessRequest(uv_work_t *req) {
TAOS_UDF_CHECK_PTR_RVOID(req); TAOS_UDF_CHECK_PTR_RVOID(req);
if (req == NULL) {
fnError("udf request is NULL");
return;
}
SUvUdfWork *uvUdf = (SUvUdfWork *)(req->data); SUvUdfWork *uvUdf = (SUvUdfWork *)(req->data);
if (uvUdf == NULL) { if (uvUdf == NULL) {
fnError("udf work is NULL"); fnError("udf work is NULL");
@ -800,8 +796,8 @@ static int32_t checkUDFScalaResult(SSDataBlock *block, SUdfColumn *output) {
return TSDB_CODE_UDF_FUNC_EXEC_FAILURE; return TSDB_CODE_UDF_FUNC_EXEC_FAILURE;
} }
if (tsSafetyCheckLevel == TSDB_SAFETY_CHECK_LEVELL_BYROW) {
for (int32_t i = 0; i < output->colData.numOfRows; ++i) { for (int32_t i = 0; i < output->colData.numOfRows; ++i) {
if (tsSafetyCheckLevel < TSDB_SAFETY_CHECK_LEVELL_BYROW) break;
if (!udfColDataIsNull(output, i)) { if (!udfColDataIsNull(output, i)) {
if (IS_VAR_DATA_TYPE(output->colMeta.type)) { if (IS_VAR_DATA_TYPE(output->colMeta.type)) {
TAOS_UDF_CHECK_CONDITION(output->colData.varLenCol.payload != NULL, TSDB_CODE_UDF_FUNC_EXEC_FAILURE); TAOS_UDF_CHECK_CONDITION(output->colData.varLenCol.payload != NULL, TSDB_CODE_UDF_FUNC_EXEC_FAILURE);
@ -816,6 +812,8 @@ static int32_t checkUDFScalaResult(SSDataBlock *block, SUdfColumn *output) {
} }
} }
} }
}
return TSDB_CODE_SUCCESS; return TSDB_CODE_SUCCESS;
} }