database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[opt](Nereids) add two args signature to trim family functions (#25169)

Browse Source
This commit is contained in:
morrySnow
2023-10-09 20:17:52 +08:00
committed by GitHub
parent 08e7a7b932
commit 37247ac449
9 changed files with 253 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/Ltrim.java
View File

@ -37,11 +37,18 @@ import java.util.List;
public class Ltrim extends ScalarFunction
implements UnaryExpression, ExplicitlyCastableSignature, PropagateNullable {
public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
private static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT)
.args(VarcharType.SYSTEM_DEFAULT, VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE, StringType.INSTANCE),
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT).args(VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE)
);
private Ltrim(List<Expression> args) {
super("ltrim", args);
}
/**
* constructor with 1 argument.
*/
@ -49,13 +56,20 @@ public class Ltrim extends ScalarFunction
super("ltrim", arg);
}
/**
* constructor with 2 argument.
*/
public Ltrim(Expression arg0, Expression arg1) {
super("ltrim", arg0, arg1);
}
/**
* withChildren.
*/
@Override
public Ltrim withChildren(List<Expression> children) {
Preconditions.checkArgument(children.size() == 1);
return new Ltrim(children.get(0));
Preconditions.checkArgument(children.size() == 1 || children.size() == 2);
return new Ltrim(children);
}
@Override

20
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/Rtrim.java
View File

@ -37,11 +37,18 @@ import java.util.List;
public class Rtrim extends ScalarFunction
implements UnaryExpression, ExplicitlyCastableSignature, PropagateNullable {
public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
private static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT)
.args(VarcharType.SYSTEM_DEFAULT, VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE, StringType.INSTANCE),
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT).args(VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE)
);
private Rtrim(List<Expression> args) {
super("rtrim", args);
}
/**
* constructor with 1 argument.
*/
@ -49,13 +56,20 @@ public class Rtrim extends ScalarFunction
super("rtrim", arg);
}
/**
* constructor with 2 argument.
*/
public Rtrim(Expression arg0, Expression arg1) {
super("rtrim", arg0, arg1);
}
/**
* withChildren.
*/
@Override
public Rtrim withChildren(List<Expression> children) {
Preconditions.checkArgument(children.size() == 1);
return new Rtrim(children.get(0));
Preconditions.checkArgument(children.size() == 1 || children.size() == 2);
return new Rtrim(children);
}
@Override

20
fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/Trim.java
View File

@ -37,11 +37,18 @@ import java.util.List;
public class Trim extends ScalarFunction
implements UnaryExpression, ExplicitlyCastableSignature, PropagateNullable {
public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
private static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT)
.args(VarcharType.SYSTEM_DEFAULT, VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE, StringType.INSTANCE),
FunctionSignature.ret(VarcharType.SYSTEM_DEFAULT).args(VarcharType.SYSTEM_DEFAULT),
FunctionSignature.ret(StringType.INSTANCE).args(StringType.INSTANCE)
);
private Trim(List<Expression> args) {
super("trim", args);
}
/**
* constructor with 1 argument.
*/
@ -49,13 +56,20 @@ public class Trim extends ScalarFunction
super("trim", arg);
}
/**
* constructor with 2 argument.
*/
public Trim(Expression arg0, Expression arg1) {
super("trim", arg0, arg1);
}
/**
* withChildren.
*/
@Override
public Trim withChildren(List<Expression> children) {
Preconditions.checkArgument(children.size() == 1);
return new Trim(children.get(0));
Preconditions.checkArgument(children.size() == 1 || children.size() == 2);
return new Trim(children);
}
@Override