fix print error log

apache · Jul 25, 2023 · 7159178 · 7159178
1 parent 5241a5e
commit 7159178
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.../spark/src/main/scala/org/apache/linkis/engineplugin/spark/mdq/MDQPostExecutionHook.scala b/.../spark/src/main/scala/org/apache/linkis/engineplugin/spark/mdq/MDQPostExecutionHook.scala
@@ -56,7 +56,12 @@ class MDQPostExecutionHook extends SparkPostExecutionHook with Logging {
       case l: CodeLanguageLabel => l.getCodeType
       case _ => ""
     }
-    if (StringUtils.isEmpty(runType) || !SparkKind.FUNCTION_MDQ_TYPE.equalsIgnoreCase(runType)) {
+    val linkisVar = "val linkisVar=1"
+    if (
+        StringUtils.isEmpty(runType) || !SparkKind.FUNCTION_MDQ_TYPE.equalsIgnoreCase(
+          runType
+        ) || (code != null && code.contains(linkisVar))
+    ) {
       return
     }
     val sender = Sender.getSender(SparkConfiguration.MDQ_APPLICATION_NAME.getValue)

diff --git a/...s/spark/src/main/scala/org/apache/linkis/engineplugin/spark/mdq/MDQPreExecutionHook.scala b/...s/spark/src/main/scala/org/apache/linkis/engineplugin/spark/mdq/MDQPreExecutionHook.scala
@@ -60,7 +60,12 @@ class MDQPreExecutionHook extends SparkPreExecutionHook with Logging {
       case _ =>
         ""
     }
-    if (StringUtils.isEmpty(runType) || !SparkKind.FUNCTION_MDQ_TYPE.equalsIgnoreCase(runType)) {
+    val linkisVar = "val linkisVar=1"
+    if (
+        StringUtils.isEmpty(runType) || !SparkKind.FUNCTION_MDQ_TYPE.equalsIgnoreCase(
+          runType
+        ) || (code != null && code.contains(linkisVar))
+    ) {
       return code
     }
     val sender = Sender.getSender(SparkConfiguration.MDQ_APPLICATION_NAME.getValue)