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 {