Skip to content

Commit 843ce1b

Browse files
committed
Changes based on upstream SQL PR comments
Signed-off-by: Margarit Hakobyan <margarith@bitquilltech.com>
1 parent e63e4a3 commit 843ce1b

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

ppl/src/main/java/org/opensearch/sql/ppl/parser/AstExpressionBuilder.java

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import static org.opensearch.sql.ast.dsl.AstDSL.qualifiedName;
1010
import static org.opensearch.sql.expression.function.BuiltinFunctionName.IS_NOT_NULL;
1111
import static org.opensearch.sql.expression.function.BuiltinFunctionName.IS_NULL;
12+
import static org.opensearch.sql.expression.function.BuiltinFunctionName.POSITION;
1213
import static org.opensearch.sql.ppl.antlr.parser.OpenSearchPPLParser.BinaryArithmeticContext;
1314
import static org.opensearch.sql.ppl.antlr.parser.OpenSearchPPLParser.BooleanFunctionCallContext;
1415
import static org.opensearch.sql.ppl.antlr.parser.OpenSearchPPLParser.BooleanLiteralContext;
@@ -69,7 +70,6 @@
6970
import org.opensearch.sql.ast.expression.Literal;
7071
import org.opensearch.sql.ast.expression.Not;
7172
import org.opensearch.sql.ast.expression.Or;
72-
import org.opensearch.sql.ast.expression.PositionFunction;
7373
import org.opensearch.sql.ast.expression.QualifiedName;
7474
import org.opensearch.sql.ast.expression.RelevanceFieldList;
7575
import org.opensearch.sql.ast.expression.Span;
@@ -294,8 +294,10 @@ public UnresolvedExpression visitTableSource(TableSourceContext ctx) {
294294
@Override
295295
public UnresolvedExpression visitPositionFunction(
296296
OpenSearchPPLParser.PositionFunctionContext ctx) {
297-
return new PositionFunction(visitFunctionArg(ctx.functionArg(0)),
298-
visitFunctionArg(ctx.functionArg(1)));
297+
return new Function(
298+
POSITION.getName().getFunctionName(),
299+
Arrays.asList(visitFunctionArg(ctx.functionArg(0)),
300+
visitFunctionArg(ctx.functionArg(1))));
299301
}
300302

301303
/**

ppl/src/test/java/org/opensearch/sql/ppl/parser/AstExpressionBuilderTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@
3535
import static org.opensearch.sql.ast.dsl.AstDSL.not;
3636
import static org.opensearch.sql.ast.dsl.AstDSL.nullLiteral;
3737
import static org.opensearch.sql.ast.dsl.AstDSL.or;
38-
import static org.opensearch.sql.ast.dsl.AstDSL.position;
3938
import static org.opensearch.sql.ast.dsl.AstDSL.projectWithArg;
4039
import static org.opensearch.sql.ast.dsl.AstDSL.qualifiedName;
4140
import static org.opensearch.sql.ast.dsl.AstDSL.relation;
@@ -184,12 +183,13 @@ public void testEvalFunctionExprNoArgs() {
184183

185184
@Test
186185
public void testPositionFunctionExpr() {
187-
assertEqual("source=t | eval f=position(fieldA IN fieldB)",
186+
assertEqual("source=t | eval f=position('substr' IN 'str')",
188187
eval(
189188
relation("t"),
190189
let(
191190
field("f"),
192-
position(field("fieldA"), field("fieldB"))
191+
function("position",
192+
stringLiteral("substr"), stringLiteral("str"))
193193
)
194194
));
195195
}

0 commit comments

Comments
 (0)