(hive) branch master updated: Tez Progress bar is not displayed in Beeline upon setting session level execution engine to Tez (Shivangi Jha, reviewed by Attila Turoczy, Vikram Ahuja, Ayush Saxena, Laszlo Bodor)

[pravin]

This is an automated email from the ASF dual-hosted git repository.

pravin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/hive.git


The following commit(s) were added to refs/heads/master by this push:
     new 1787a08ac64 Tez Progress bar is not displayed in Beeline upon setting 
session level execution engine to Tez (Shivangi Jha, reviewed by Attila 
Turoczy, Vikram Ahuja, Ayush Saxena, Laszlo Bodor)
1787a08ac64 is described below

commit 1787a08ac642fd674629315f171ea5e79dfc50be
Author: Shivangi Jha <149884343+shivjh...@users.noreply.github.com>
AuthorDate: Mon Dec 4 20:17:22 2023 +0530

    Tez Progress bar is not displayed in Beeline upon setting session level 
execution engine to Tez (Shivangi Jha, reviewed by Attila Turoczy, Vikram 
Ahuja, Ayush Saxena, Laszlo Bodor)
---
 service/src/java/org/apache/hive/service/cli/CLIService.java          | 4 ++++
 .../src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java | 4 ++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/service/src/java/org/apache/hive/service/cli/CLIService.java 
b/service/src/java/org/apache/hive/service/cli/CLIService.java
index 151b2067576..a061536f857 100644
--- a/service/src/java/org/apache/hive/service/cli/CLIService.java
+++ b/service/src/java/org/apache/hive/service/cli/CLIService.java
@@ -510,6 +510,10 @@ public class CLIService extends CompositeService 
implements ICLIService {
     opStatus.setJobProgressUpdate(progressUpdateLog(getProgressUpdate, 
operation, conf));
     return opStatus;
   }
+  public HiveConf getHiveSessionConf(OperationHandle opHandle) throws 
HiveSQLException {
+    Operation operation = 
sessionManager.getOperationManager().getOperation(opHandle);
+    return operation.getParentSession().getHiveConf();
+  }
 
   public HiveConf getSessionConf(SessionHandle sessionHandle)
       throws HiveSQLException {
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
index d93d54b33b1..7d71881dc84 100644
--- a/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java
+++ b/service/src/java/org/apache/hive/service/cli/thrift/ThriftCLIService.java
@@ -794,7 +794,7 @@ public abstract class ThriftCLIService extends 
AbstractService implements TCLISe
     try {
       OperationStatus operationStatus =
           cliService.getOperationStatus(operationHandle, 
req.isGetProgressUpdate());
-
+      HiveConf sessionConf = cliService.getHiveSessionConf(operationHandle);
       if (operationStatus.getState().equals(OperationState.FINISHED)) {
         long numModifiedRows = operationStatus.getNumModifiedRows();
         resp.setNumModifiedRows(numModifiedRows);
@@ -810,7 +810,7 @@ public abstract class ThriftCLIService extends 
AbstractService implements TCLISe
       }
       JobProgressUpdate progressUpdate = operationStatus.jobProgressUpdate();
       ProgressMonitorStatusMapper mapper = ProgressMonitorStatusMapper.DEFAULT;
-      if ("tez".equals(hiveConf.getVar(ConfVars.HIVE_EXECUTION_ENGINE))) {
+      if ("tez".equals(sessionConf.getVar(ConfVars.HIVE_EXECUTION_ENGINE))) {
         mapper = new TezProgressMonitorStatusMapper();
       }
       TJobExecutionStatus executionStatus =