In Expr bindings. (apache#10)

* Complete In Expression Support.

* Fixed lint issues.

* Address Review comments.

* Fix review comments.

e Please enter the commit message for your changes. Lines starting

* Fix lint issues.

cpp/src/gandiva/jni/jni_common.cc

@@ -313,6 +313,45 @@ NodePtr ProtoTypeToOrNode(const types::OrNode& node) {
  return TreeExprBuilder::MakeOr(children);
 }
 
+NodePtr ProtoTypeToInNode(const types::InNode& node) {
+ NodePtr field = ProtoTypeToFieldNode(node.field());
+
+ if (node.has_intvalues()) {
+ std::unordered_set<int32_t> int_values;
+ for (int i = 0; i < node.intvalues().intvalues_size(); i++) {
+ int_values.insert(node.intvalues().intvalues(i).value());
+ }
+ return TreeExprBuilder::MakeInExpressionInt32(field, int_values);
+ }
+
+ if (node.has_longvalues()) {
+ std::unordered_set<int64_t> long_values;
+ for (int i = 0; i < node.longvalues().longvalues_size(); i++) {
+ long_values.insert(node.longvalues().longvalues(i).value());
+ }
+ return TreeExprBuilder::MakeInExpressionInt64(field, long_values);
+ }
+
+ if (node.has_stringvalues()) {
+ std::unordered_set<std::string> stringvalues;
+ for (int i = 0; i < node.stringvalues().stringvalues_size(); i++) {
+ stringvalues.insert(node.stringvalues().stringvalues(i).value());
+ }
+ return TreeExprBuilder::MakeInExpressionString(field, stringvalues);
+ }
+
+ if (node.has_binaryvalues()) {
+ std::unordered_set<std::string> stringvalues;
+ for (int i = 0; i < node.binaryvalues().binaryvalues_size(); i++) {
+ stringvalues.insert(node.binaryvalues().binaryvalues(i).value());
+ }
+ return TreeExprBuilder::MakeInExpressionBinary(field, stringvalues);
+ }
+ // not supported yet.
+ std::cerr << "Unknown constant type for in expression.\n";
+ return nullptr;
+}
+
 NodePtr ProtoTypeToNullNode(const types::NullNode& node) {
  DataTypePtr data_type = ProtoTypeToDataType(node.type());
  if (data_type == nullptr) {
@@ -344,6 +383,10 @@ NodePtr ProtoTypeToNode(const types::TreeNode& node) {
  return ProtoTypeToOrNode(node.ornode());
  }
 
+ if (node.has_innode()) {
+ return ProtoTypeToInNode(node.innode());
+ }
+
  if (node.has_nullnode()) {
  return ProtoTypeToNullNode(node.nullnode());
  }

cpp/src/gandiva/proto/Types.proto

@@ -173,6 +173,9 @@ message TreeNode {
  optional StringNode stringNode = 17;
  optional BinaryNode binaryNode = 18;
  optional DecimalNode decimalNode = 19;
+
+ // in expr
+ optional InNode inNode = 21;
 }
 
 message ExpressionRoot {
@@ -205,3 +208,27 @@ message FunctionSignature {
  optional ExtGandivaType returnType = 2;
  repeated ExtGandivaType paramTypes = 3;
 }
+
+message InNode {
+ optional FieldNode field = 1;
+ optional IntConstants intValues = 2;
+ optional LongConstants longValues = 3;
+ optional StringConstants stringValues = 4;
+ optional BinaryConstants binaryValues = 5;
+}
+
+message IntConstants {
+ repeated IntNode intValues = 1;
+}
+
+message LongConstants {
+ repeated LongNode longValues = 1;
+}
+
+message StringConstants {
+ repeated StringNode stringValues = 1;
+}
+
+message BinaryConstants {
+ repeated BinaryNode binaryValues = 1;
+}

java/gandiva/src/main/java/org/apache/arrow/gandiva/expression/InNode.java

@@ -0,0 +1,102 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.arrow.gandiva.expression;
+
+import java.nio.charset.Charset;
+import java.util.Set;
+
+import org.apache.arrow.gandiva.exceptions.GandivaException;
+import org.apache.arrow.gandiva.ipc.GandivaTypes;
+import org.apache.arrow.vector.types.pojo.Field;
+
+import com.google.protobuf.ByteString;
+
+/**
+ * In Node representation in java.
+ */
+public class InNode implements TreeNode {
+ private static final Charset charset = Charset.forName("UTF-8");
+
+ private final Set<Integer> intValues;
+ private final Set<Long> longValues;
+ private final Set<String> stringValues;
+ private final Set<byte[]> binaryValues;
+ private final Field field;
+
+ private InNode(Set<Integer> values, Set<Long> longValues, Set<String> stringValues, Set<byte[]>
+ binaryValues, Field field) {
+ this.intValues = values;
+ this.longValues = longValues;
+ this.stringValues = stringValues;
+ this.binaryValues = binaryValues;
+ this.field = field;
+ }
+
+ public static InNode makeIntInExpr(Field field, Set<Integer> intValues) {
+ return new InNode(intValues, null, null, null ,field);
+ }
+
+ public static InNode makeLongInExpr(Field field, Set<Long> longValues) {
+ return new InNode(null, longValues, null, null ,field);
+ }
+
+ public static InNode makeStringInExpr(Field field, Set<String> stringValues) {
+ return new InNode(null, null, stringValues, null ,field);
+ }
+
+ public static InNode makeBinaryInExpr(Field field, Set<byte[]> binaryValues) {
+ return new InNode(null, null, null, binaryValues ,field);
+ }
+
+ @Override
+ public GandivaTypes.TreeNode toProtobuf() throws GandivaException {
+ GandivaTypes.InNode.Builder inNode = GandivaTypes.InNode.newBuilder();
+
+ GandivaTypes.FieldNode.Builder fieldNode = GandivaTypes.FieldNode.newBuilder();
+ fieldNode.setField(ArrowTypeHelper.arrowFieldToProtobuf(field));
+ inNode.setField(fieldNode);
+
+ if (intValues != null) {
+ GandivaTypes.IntConstants.Builder intConstants = GandivaTypes.IntConstants.newBuilder();
+ intValues.stream().forEach(val -> intConstants.addIntValues(GandivaTypes.IntNode.newBuilder()
+ .setValue(val).build()));
+ inNode.setIntValues(intConstants.build());
+ } else if (longValues != null) {
+ GandivaTypes.LongConstants.Builder longConstants = GandivaTypes.LongConstants.newBuilder();
+ longValues.stream().forEach(val -> longConstants.addLongValues(GandivaTypes.LongNode.newBuilder()
+ .setValue(val).build()));
+ inNode.setLongValues(longConstants.build());
+ } else if (stringValues != null) {
+ GandivaTypes.StringConstants.Builder stringConstants = GandivaTypes.StringConstants
+ .newBuilder();
+ stringValues.stream().forEach(val -> stringConstants.addStringValues(GandivaTypes.StringNode
+ .newBuilder().setValue(ByteString.copyFrom(val.getBytes(charset))).build()));
+ inNode.setStringValues(stringConstants.build());
+ } else if (binaryValues != null) {
+ GandivaTypes.BinaryConstants.Builder binaryConstants = GandivaTypes.BinaryConstants
+ .newBuilder();
+ binaryValues.stream().forEach(val -> binaryConstants.addBinaryValues(GandivaTypes.BinaryNode
+ .newBuilder().setValue(ByteString.copyFrom(val)).build()));
+ inNode.setBinaryValues(binaryConstants.build());
+ }
+ GandivaTypes.TreeNode.Builder builder = GandivaTypes.TreeNode.newBuilder();
+ builder.setInNode(inNode.build());
+ return builder.build();
+
+ }
+}

java/gandiva/src/main/java/org/apache/arrow/gandiva/expression/TreeBuilder.java

@@ -19,6 +19,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 
 import org.apache.arrow.vector.types.pojo.ArrowType;
 import org.apache.arrow.vector.types.pojo.Field;
@@ -185,4 +186,24 @@ public static Condition makeCondition(String function,
  TreeNode root = makeFunction(function, children, new ArrowType.Bool());
  return makeCondition(root);
  }
+
+ public static TreeNode makeInExpressionInt32(Field resultField,
+ Set<Integer> intValues) {
+ return InNode.makeIntInExpr(resultField, intValues);
+ }
+
+ public static TreeNode makeInExpressionBigInt(Field resultField,
+ Set<Long> longValues) {
+ return InNode.makeLongInExpr(resultField, longValues);
+ }
+
+ public static TreeNode makeInExpressionString(Field resultField,
+ Set<String> stringValues) {
+ return InNode.makeStringInExpr(resultField, stringValues);
+ }
+
+ public static TreeNode makeInExpressionBinary(Field resultField,
+ Set<byte[]> binaryValues) {
+ return InNode.makeBinaryInExpr(resultField, binaryValues);
+ }
 }

java/gandiva/src/test/java/org/apache/arrow/gandiva/evaluator/ProjectorTest.java

@@ -51,6 +51,7 @@
 import org.junit.Test;
 
 import com.google.common.collect.Lists;
+import com.google.common.collect.Sets;
 
 import io.netty.buffer.ArrowBuf;
 
@@ -1047,6 +1048,96 @@ public void testEquals() throws GandivaException, Exception {
  eval.close();
  }
 
+ @Test
+ public void testInExpr() throws GandivaException, Exception {
+ Field c1 = Field.nullable("c1", int32);
+
+ TreeNode inExpr =
+ TreeBuilder.makeInExpressionInt32(c1, Sets.newHashSet(1,2,3,4,5,15,16));
+ ExpressionTree expr = TreeBuilder.makeExpression(inExpr, Field.nullable("result", boolType));
+ Schema schema = new Schema(Lists.newArrayList(c1));
+ Projector eval = Projector.make(schema, Lists.newArrayList(expr));
+
+ int numRows = 16;
+ byte[] validity = new byte[]{(byte) 255, 0};
+ int[] c1Values = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16};
+
+ ArrowBuf c1Validity = buf(validity);
+ ArrowBuf c1Data = intBuf(c1Values);
+ ArrowBuf c2Validity = buf(validity);
+
+ ArrowFieldNode fieldNode = new ArrowFieldNode(numRows, 0);
+ ArrowRecordBatch batch =
+ new ArrowRecordBatch(
+ numRows,
+ Lists.newArrayList(fieldNode, fieldNode),
+ Lists.newArrayList(c1Validity, c1Data, c2Validity));
+
+ BitVector bitVector = new BitVector(EMPTY_SCHEMA_PATH, allocator);
+ bitVector.allocateNew(numRows);
+
+ List<ValueVector> output = new ArrayList<ValueVector>();
+ output.add(bitVector);
+ eval.evaluate(batch, output);
+
+ for (int i = 0; i < 5; i++) {
+ assertTrue(bitVector.getObject(i).booleanValue());
+ }
+ for (int i = 5; i < 16; i++) {
+ assertFalse(bitVector.getObject(i).booleanValue());
+ }
+
+ releaseRecordBatch(batch);
+ releaseValueVectors(output);
+ eval.close();
+ }
+
+ @Test
+ public void testInExprStrings() throws GandivaException, Exception {
+ Field c1 = Field.nullable("c1", new ArrowType.Utf8());
+
+ TreeNode inExpr =
+ TreeBuilder.makeInExpressionString(c1, Sets.newHashSet("one", "two", "three", "four"));
+ ExpressionTree expr = TreeBuilder.makeExpression(inExpr, Field.nullable("result", boolType));
+ Schema schema = new Schema(Lists.newArrayList(c1));
+ Projector eval = Projector.make(schema, Lists.newArrayList(expr));
+
+ int numRows = 16;
+ byte[] validity = new byte[]{(byte) 255, 0};
+ String[] c1Values = new String[]{"one", "two", "three", "four", "five", "six", "seven",
+ "eight", "nine", "ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen",
+ "sixteen"};
+
+ ArrowBuf c1Validity = buf(validity);
+ List<ArrowBuf> dataBufsX = stringBufs(c1Values);
+ ArrowBuf c2Validity = buf(validity);
+
+ ArrowFieldNode fieldNode = new ArrowFieldNode(numRows, 0);
+ ArrowRecordBatch batch =
+ new ArrowRecordBatch(
+ numRows,
+ Lists.newArrayList(fieldNode, fieldNode),
+ Lists.newArrayList(c1Validity, dataBufsX.get(0),dataBufsX.get(1), c2Validity));
+
+ BitVector bitVector = new BitVector(EMPTY_SCHEMA_PATH, allocator);
+ bitVector.allocateNew(numRows);
+
+ List<ValueVector> output = new ArrayList<ValueVector>();
+ output.add(bitVector);
+ eval.evaluate(batch, output);
+
+ for (int i = 0; i < 4; i++) {
+ assertTrue(bitVector.getObject(i).booleanValue());
+ }
+ for (int i = 5; i < 16; i++) {
+ assertFalse(bitVector.getObject(i).booleanValue());
+ }
+
+ releaseRecordBatch(batch);
+ releaseValueVectors(output);
+ eval.close();
+ }
+
  @Test
  public void testSmallOutputVectors() throws GandivaException, Exception {
  Field a = Field.nullable("a", int32);