HIVE-28076: Selecting data from a bucketed table with decimal column …

…type throwing NPE. (Dayakar M, reviewed by Krisztian Kasa)

(cherry picked from commit 3e48a01)

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

@@ -202,7 +202,7 @@ protected void generatePredicate(NodeProcessorCtx procCtx,
       BitSet bs = new BitSet(numBuckets);
       bs.clear();
       PrimitiveObjectInspector bucketOI = (PrimitiveObjectInspector)bucketField.getFieldObjectInspector();
-      PrimitiveObjectInspector constOI = PrimitiveObjectInspectorFactory.getPrimitiveWritableObjectInspector(bucketOI.getPrimitiveCategory());
+      PrimitiveObjectInspector constOI = PrimitiveObjectInspectorFactory.getPrimitiveWritableObjectInspector(bucketOI.getTypeInfo());
       // Fetch the bucketing version from table scan operator
       int bucketingVersion = top.getConf().getTableMetadata().getBucketingVersion();
 

ql/src/test/queries/clientpositive/bucket_decimal_col_select.q

@@ -0,0 +1,6 @@
+set hive.tez.bucket.pruning=true;
+
+create table bucket_table(id decimal(38,0), name string) clustered by(id) into 3 buckets;
+insert into bucket_table values(5000000000000999640711, 'Cloud');
+
+select * from bucket_table bt where id = 5000000000000999640711;

ql/src/test/results/clientpositive/llap/bucket_decimal_col_select.q.out

@@ -0,0 +1,27 @@
+PREHOOK: query: create table bucket_table(id decimal(38,0), name string) clustered by(id) into 3 buckets
+PREHOOK: type: CREATETABLE
+PREHOOK: Output: database:default
+PREHOOK: Output: default@bucket_table
+POSTHOOK: query: create table bucket_table(id decimal(38,0), name string) clustered by(id) into 3 buckets
+POSTHOOK: type: CREATETABLE
+POSTHOOK: Output: database:default
+POSTHOOK: Output: default@bucket_table
+PREHOOK: query: insert into bucket_table values(5000000000000999640711, 'Cloud')
+PREHOOK: type: QUERY
+PREHOOK: Input: _dummy_database@_dummy_table
+PREHOOK: Output: default@bucket_table
+POSTHOOK: query: insert into bucket_table values(5000000000000999640711, 'Cloud')
+POSTHOOK: type: QUERY
+POSTHOOK: Input: _dummy_database@_dummy_table
+POSTHOOK: Output: default@bucket_table
+POSTHOOK: Lineage: bucket_table.id SCRIPT []
+POSTHOOK: Lineage: bucket_table.name SCRIPT []
+PREHOOK: query: select * from bucket_table bt where id = 5000000000000999640711
+PREHOOK: type: QUERY
+PREHOOK: Input: default@bucket_table
+#### A masked pattern was here ####
+POSTHOOK: query: select * from bucket_table bt where id = 5000000000000999640711
+POSTHOOK: type: QUERY
+POSTHOOK: Input: default@bucket_table
+#### A masked pattern was here ####
+5000000000000999640711	Cloud