diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/IsNull.java ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/IsNull.java index cb563d1..7f3f604 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/IsNull.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/IsNull.java @@ -84,7 +84,7 @@ public void evaluate(VectorizedRowBatch batch) { @Override public int getOutputColumn() { - return -1; + return outputColumn; } @Override diff --git ql/src/test/org/apache/hadoop/hive/ql/exec/vector/TestVectorizationContext.java ql/src/test/org/apache/hadoop/hive/ql/exec/vector/TestVectorizationContext.java index 74908e9..2329f52 100644 --- ql/src/test/org/apache/hadoop/hive/ql/exec/vector/TestVectorizationContext.java +++ ql/src/test/org/apache/hadoop/hive/ql/exec/vector/TestVectorizationContext.java @@ -545,6 +545,7 @@ public void testNullExpressions() throws HiveException { ve = vc.getVectorExpression(isNullExpr, VectorExpressionDescriptor.Mode.PROJECTION); assertEquals(ve.getClass(), IsNull.class); assertEquals(2, ((IsNull) ve).getColNum()); + assertEquals(3, ve.getOutputColumn()); assertEquals(ve.getChildExpressions()[0].getClass(), LongColGreaterLongScalar.class); }