commit2

Author: ramitg254

ql/src/java/org/apache/hadoop/hive/ql/metadata/Table.java

@@ -644,7 +644,11 @@ public boolean hasNonNativePartitionSupport() {
   }
 
   public boolean isPartitionKey(String colName) {
-    return getPartColByName(colName) != null;
+    List<String> partKeys = getPartitionKeys().stream().map(FieldSchema::getName).toList();
+    if (partKeys.isEmpty()) {
+      return false;
+    }
+    return partKeys.contains(colName);
   }
 
   // TODO merge this with getBucketCols function

ql/src/java/org/apache/hadoop/hive/ql/optimizer/DynamicPartitionPruningOptimization.java

@@ -194,7 +194,7 @@ public Object process(Node nd, Stack<Node> stack, NodeProcessorCtx procCtx, Obje
         } else if (table.isNonNative() &&
           table.getStorageHandler().addDynamicSplitPruningEdge(table, ctx.parent)) {
           generateEventOperatorPlan(ctx, parseContext, ts, column,
-            table.getCols().stream().filter(e -> e.getName().equals(column)).
+            table.getAllCols().stream().filter(e -> e.getName().equals(column)).
           map(e -> e.getType()).findFirst().get(), ctx.parent);
         } else { // semijoin
           LOG.debug("Column " + column + " is not a partition column");