Print thread stack for udf execution error

Author: JackieTien97

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/fragment/FragmentInstanceManager.java

@@ -205,6 +205,7 @@ public FragmentInstanceInfo execDataQueryFragmentInstance(
                   } else if (t instanceof UDFTypeMismatchException) {
                     stateMachine.failed(new SemanticException(t.getMessage()));
                   } else if (t instanceof UDFException) {
+                    logger.warn("Exception happened when executing UDTF: ", t);
                     stateMachine.failed(
                         new IoTDBRuntimeException(
                             t.getMessage(), TSStatusCode.EXECUTE_UDF_ERROR.getStatusCode(), true));

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/function/TableFunctionLeafOperator.java

@@ -32,6 +32,8 @@
 import org.apache.tsfile.read.common.block.TsBlock;
 import org.apache.tsfile.read.common.block.TsBlockBuilder;
 import org.apache.tsfile.read.common.block.column.RunLengthEncodedColumn;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.util.Arrays;
 import java.util.List;
@@ -40,6 +42,7 @@
 
 // only one input source is supported now
 public class TableFunctionLeafOperator implements ProcessOperator {
+  private static final Logger LOGGER = LoggerFactory.getLogger(TableFunctionLeafOperator.class);
 
   private final OperatorContext operatorContext;
   private final TsBlockBuilder blockBuilder;
@@ -76,6 +79,7 @@ public TsBlock next() throws Exception {
     try {
       processor.process(columnBuilders);
     } catch (Exception e) {
+      LOGGER.warn("Exception happened when executing UDTF: ", e);
       throw new IoTDBRuntimeException(
           e.getMessage(), TSStatusCode.EXECUTE_UDF_ERROR.getStatusCode(), true);
     }
@@ -103,6 +107,7 @@ public void close() throws Exception {
     try {
       processor.beforeDestroy();
     } catch (Exception e) {
+      LOGGER.warn("Exception happened when executing UDTF: ", e);
       throw new IoTDBRuntimeException(
           e.getMessage(), TSStatusCode.EXECUTE_UDF_ERROR.getStatusCode(), true);
     }

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/utils/ErrorHandlingUtils.java

@@ -120,6 +120,7 @@ public static TSStatus onQueryException(Exception e, String operation, TSStatusC
             || status.getCode() == TSStatusCode.NO_AVAILABLE_REPLICA.getStatusCode()
             || status.getCode() == TSStatusCode.CANNOT_FETCH_FI_STATE.getStatusCode()
             || status.getCode() == TSStatusCode.QUERY_EXECUTION_MEMORY_NOT_ENOUGH.getStatusCode()
+            || status.getCode() == TSStatusCode.EXECUTE_UDF_ERROR.getStatusCode()
             || status.getCode() == TSStatusCode.QUERY_TIMEOUT.getStatusCode()) {
           LOGGER.info(message);
         } else {

iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/exception/IoTDBRuntimeException.java

@@ -40,6 +40,13 @@ public IoTDBRuntimeException(String message, int errorCode, boolean isUserExcept
     this.isUserException = isUserException;
   }
 
+  public IoTDBRuntimeException(
+      String message, int errorCode, Throwable cause, boolean isUserException) {
+    super(message, cause);
+    this.errorCode = errorCode;
+    this.isUserException = isUserException;
+  }
+
   public IoTDBRuntimeException(String message, Throwable cause, int errorCode) {
     super(message, cause);
     this.errorCode = errorCode;