Update ThriftCLIService.java

HIVE-24944: When the default engine of the hiveserver is MR and the tez engine is set by the client, the client TEZ progress log cannot be printed normally（ZhangQidong reviewed by Zhihua Deng)
apache · Jun 17, 2021 · 0b2f11a · 0b2f11a
1 parent eb1a21d
commit 0b2f11a
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java b/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java
@@ -799,11 +799,15 @@ public TGetOperationStatusResp GetOperationStatus(TGetOperationStatusReq req) th
               .getParentSession()
               .getHiveConf()
               .getVar(ConfVars.HIVE_EXECUTION_ENGINE);
-      if ("tez".equals(engineInSessionConf)) {
-        mapper = new TezProgressMonitorStatusMapper();
-      }
-      if ("spark".equals(engineInSessionConf)) {
-        mapper = new SparkProgressMonitorStatusMapper();
+      switch (engineInSessionConf) {
+        case "tez":
+          mapper = new TezProgressMonitorStatusMapper();
+          break;
+        case "spark":
+          mapper = new SparkProgressMonitorStatusMapper();
+          break;
+        default:
+          break;
       }
       TJobExecutionStatus executionStatus =
           mapper.forStatus(progressUpdate.status);