HIVE_14343: HiveDriverRunHookContext's command is null in HS2 mode (Chao Sun, reviewed by Xuefu Zhang)
diff --git a/ql/src/java/org/apache/hadoop/hive/ql/Driver.java b/ql/src/java/org/apache/hadoop/hive/ql/Driver.java
index 14f221a..0278673 100644
--- a/ql/src/java/org/apache/hadoop/hive/ql/Driver.java
+++ b/ql/src/java/org/apache/hadoop/hive/ql/Driver.java
@@ -1262,7 +1262,8 @@
SQLState = null;
downstreamError = null;
- HiveDriverRunHookContext hookContext = new HiveDriverRunHookContextImpl(conf, command);
+ HiveDriverRunHookContext hookContext = new HiveDriverRunHookContextImpl(conf,
+ alreadyCompiled ? ctx.getCmd() : command);
// Get all the driver run hooks and pre-execute them.
List<HiveDriverRunHook> driverRunHooks;
try {