Skip to content

Commit c539056

Browse files
committed
Refactor ip comparator to comparator
Signed-off-by: Yuanchun Shen <yuanchu@amazon.com>
1 parent bd9f3bb commit c539056

File tree

2 files changed

+25
-26
lines changed

2 files changed

+25
-26
lines changed

core/src/main/java/org/opensearch/sql/expression/function/PPLBuiltinOperators.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@
7171
import org.opensearch.sql.expression.function.udf.datetime.WeekdayFunction;
7272
import org.opensearch.sql.expression.function.udf.datetime.YearweekFunction;
7373
import org.opensearch.sql.expression.function.udf.ip.CidrMatchFunction;
74-
import org.opensearch.sql.expression.function.udf.ip.CompareIpFunction;
74+
import org.opensearch.sql.expression.function.udf.CompareFunctions;
7575
import org.opensearch.sql.expression.function.udf.math.CRC32Function;
7676
import org.opensearch.sql.expression.function.udf.math.ConvFunction;
7777
import org.opensearch.sql.expression.function.udf.math.DivideFunction;
@@ -107,12 +107,12 @@ public class PPLBuiltinOperators extends ReflectiveSqlOperatorTable {
107107
public static final SqlOperator CIDRMATCH = new CidrMatchFunction().toUDF("CIDRMATCH");
108108

109109
// Comparing functions
110-
public static final SqlOperator NOT_EQUALS = CompareIpFunction.notEquals().toUDF("NOT_EQUALS");
111-
public static final SqlOperator EQUALS = CompareIpFunction.equals().toUDF("EQUALS");
112-
public static final SqlOperator GREATER = CompareIpFunction.greater().toUDF("GREATER");
113-
public static final SqlOperator GTE = CompareIpFunction.greaterOrEquals().toUDF("GTE");
114-
public static final SqlOperator LESS = CompareIpFunction.less().toUDF("LESS");
115-
public static final SqlOperator LTE = CompareIpFunction.lessOrEquals().toUDF("LTE");
110+
public static final SqlOperator NOT_EQUALS = CompareFunctions.notEquals().toUDF("NOT_EQUALS");
111+
public static final SqlOperator EQUALS = CompareFunctions.equals().toUDF("EQUALS");
112+
public static final SqlOperator GREATER = CompareFunctions.greater().toUDF("GREATER");
113+
public static final SqlOperator GTE = CompareFunctions.greaterOrEquals().toUDF("GTE");
114+
public static final SqlOperator LESS = CompareFunctions.less().toUDF("LESS");
115+
public static final SqlOperator LTE = CompareFunctions.lessOrEquals().toUDF("LTE");
116116

117117
// Condition function
118118
public static final SqlOperator EARLIEST = new EarliestFunction().toUDF("EARLIEST");
Lines changed: 18 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
* SPDX-License-Identifier: Apache-2.0
44
*/
55

6-
package org.opensearch.sql.expression.function.udf.ip;
6+
package org.opensearch.sql.expression.function.udf;
77

88
import java.util.List;
99
import org.apache.calcite.adapter.enumerable.NotNullImplementor;
@@ -22,44 +22,43 @@
2222
import org.opensearch.sql.expression.function.UDFOperandMetadata;
2323

2424
/**
25-
* {@code compare(ip1, ip2)} compares two IP addresses using a provided op.
25+
* {@code compare(value1, value2)} compares two values. It extends Calcite's built-in comparators by
26+
* supporting IP comparison.
2627
*
2728
* <p>Signature:
2829
*
2930
* <ul>
30-
* <li>(IP, STRING) -> BOOLEAN
31-
* <li>(STRING, IP) -> BOOLEAN
32-
* <li>(IP, IP) -> BOOLEAN
31+
* <li>(COMPARABLE, COMPARABLE) -> BOOLEAN
3332
* </ul>
3433
*/
35-
public class CompareIpFunction extends ImplementorUDF {
34+
public class CompareFunctions extends ImplementorUDF {
3635

37-
private CompareIpFunction(ComparisonType comparisonType) {
36+
private CompareFunctions(ComparisonType comparisonType) {
3837
super(new CompareImplementor(comparisonType), NullPolicy.ANY);
3938
}
4039

41-
public static CompareIpFunction less() {
42-
return new CompareIpFunction(ComparisonType.LESS);
40+
public static CompareFunctions less() {
41+
return new CompareFunctions(ComparisonType.LESS);
4342
}
4443

45-
public static CompareIpFunction greater() {
46-
return new CompareIpFunction(ComparisonType.GREATER);
44+
public static CompareFunctions greater() {
45+
return new CompareFunctions(ComparisonType.GREATER);
4746
}
4847

49-
public static CompareIpFunction lessOrEquals() {
50-
return new CompareIpFunction(ComparisonType.LESS_OR_EQUAL);
48+
public static CompareFunctions lessOrEquals() {
49+
return new CompareFunctions(ComparisonType.LESS_OR_EQUAL);
5150
}
5251

53-
public static CompareIpFunction greaterOrEquals() {
54-
return new CompareIpFunction(ComparisonType.GREATER_OR_EQUAL);
52+
public static CompareFunctions greaterOrEquals() {
53+
return new CompareFunctions(ComparisonType.GREATER_OR_EQUAL);
5554
}
5655

57-
public static CompareIpFunction equals() {
58-
return new CompareIpFunction(ComparisonType.EQUALS);
56+
public static CompareFunctions equals() {
57+
return new CompareFunctions(ComparisonType.EQUALS);
5958
}
6059

61-
public static CompareIpFunction notEquals() {
62-
return new CompareIpFunction(ComparisonType.NOT_EQUALS);
60+
public static CompareFunctions notEquals() {
61+
return new CompareFunctions(ComparisonType.NOT_EQUALS);
6362
}
6463

6564
@Override

0 commit comments

Comments
 (0)