Add type checker for cidrmatch

Signed-off-by: Yuanchun Shen <yuanchu@amazon.com>

Author: yuancu

core/src/main/java/org/opensearch/sql/calcite/utils/UserDefinedFunctionUtils.java

@@ -97,6 +97,8 @@ public static SqlTypeName convertRelDataTypeToSqlTypeName(RelDataType type) {
         case EXPR_DATE -> SqlTypeName.DATE;
         case EXPR_TIME -> SqlTypeName.TIME;
         case EXPR_TIMESTAMP -> SqlTypeName.TIMESTAMP;
+        case EXPR_IP -> SqlTypeName.VARCHAR;
+        case EXPR_BINARY -> SqlTypeName.VARBINARY;
         default -> type.getSqlTypeName();
       };
     }

core/src/main/java/org/opensearch/sql/expression/function/udf/ip/CidrMatchFunction.java

@@ -12,12 +12,14 @@
 import org.apache.calcite.linq4j.tree.Expression;
 import org.apache.calcite.linq4j.tree.Expressions;
 import org.apache.calcite.rex.RexCall;
+import org.apache.calcite.sql.type.OperandTypes;
 import org.apache.calcite.sql.type.ReturnTypes;
 import org.apache.calcite.sql.type.SqlReturnTypeInference;
 import org.opensearch.sql.data.model.ExprIpValue;
 import org.opensearch.sql.data.model.ExprValue;
 import org.opensearch.sql.data.model.ExprValueUtils;
 import org.opensearch.sql.expression.function.ImplementorUDF;
+import org.opensearch.sql.expression.function.UDFOperandMetadata;
 import org.opensearch.sql.expression.ip.IPFunctions;
 
 /**
@@ -40,6 +42,13 @@ public SqlReturnTypeInference getReturnTypeInference() {
     return ReturnTypes.BOOLEAN_FORCE_NULLABLE;
   }
 
+  @Override
+  public UDFOperandMetadata getOperandMetadata() {
+    // EXPR_IP is mapped to SqlTypeFamily.VARCHAR in
+    // UserDefinedFunctionUtils.convertRelDataTypeToSqlTypeName
+    return UDFOperandMetadata.wrap(OperandTypes.STRING_STRING);
+  }
+
   public static class CidrMatchImplementor implements NotNullImplementor {
     @Override
     public Expression implement(