[doris] 14/18: [fix](Hive bitmap udf) NPE error when mapreduce task only has map task (#19140)

[morningman]

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

morningman pushed a commit to branch branch-1.2-lts
in repository https://gitbox.apache.org/repos/asf/doris.git

commit 9c673f9bcec9b41abd09c3457f30b33170529c8b
Author: Liqf <109049295+lemonlit...@users.noreply.github.com>
AuthorDate: Thu Apr 27 14:52:26 2023 +0800

    [fix](Hive bitmap udf) NPE error when mapreduce task only has map task 
(#19140)
    
    Issue Number: close #17539
---
 fe/hive-udf/src/main/java/org/apache/doris/udf/BitmapUnionUDAF.java | 2 +-
 fe/hive-udf/src/main/java/org/apache/doris/udf/ToBitmapUDAF.java    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git 
a/fe/hive-udf/src/main/java/org/apache/doris/udf/BitmapUnionUDAF.java 
b/fe/hive-udf/src/main/java/org/apache/doris/udf/BitmapUnionUDAF.java
index 6f37925564..e28ccf613b 100644
--- a/fe/hive-udf/src/main/java/org/apache/doris/udf/BitmapUnionUDAF.java
+++ b/fe/hive-udf/src/main/java/org/apache/doris/udf/BitmapUnionUDAF.java
@@ -64,7 +64,7 @@ public class BitmapUnionUDAF extends 
AbstractGenericUDAFResolver {
             super.init(m, parameters);
             // init output object inspectors
             // The output of a partial aggregation is a binary
-            if (m == Mode.PARTIAL1) {
+            if (m == Mode.PARTIAL1 || m == Mode.COMPLETE) {
                 this.inputOI = (BinaryObjectInspector) parameters[0];
             } else {
                 this.internalMergeOI = (BinaryObjectInspector) parameters[0];
diff --git a/fe/hive-udf/src/main/java/org/apache/doris/udf/ToBitmapUDAF.java 
b/fe/hive-udf/src/main/java/org/apache/doris/udf/ToBitmapUDAF.java
index 39821a845b..9f402e68e9 100644
--- a/fe/hive-udf/src/main/java/org/apache/doris/udf/ToBitmapUDAF.java
+++ b/fe/hive-udf/src/main/java/org/apache/doris/udf/ToBitmapUDAF.java
@@ -70,7 +70,7 @@ public class ToBitmapUDAF extends AbstractGenericUDAFResolver 
{
             super.init(m, parameters);
             // init output object inspectors
             // The output of a partial aggregation is a binary
-            if (m == Mode.PARTIAL1) {
+            if (m == Mode.PARTIAL1 || m == Mode.COMPLETE) {
                 inputOI = (PrimitiveObjectInspector) parameters[0];
             } else {
                 this.internalMergeOI = (BinaryObjectInspector) parameters[0];


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org