|
|
@@ -44,6 +44,10 @@ import org.elasticsearch.xpack.esql.expression.function.scalar.date.DateTrunc;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.date.Now;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.ip.CIDRMatch;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.math.Abs;
|
|
|
+import org.elasticsearch.xpack.esql.expression.function.scalar.math.Acos;
|
|
|
+import org.elasticsearch.xpack.esql.expression.function.scalar.math.Asin;
|
|
|
+import org.elasticsearch.xpack.esql.expression.function.scalar.math.Atan;
|
|
|
+import org.elasticsearch.xpack.esql.expression.function.scalar.math.Atan2;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.math.AutoBucket;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.math.Cos;
|
|
|
import org.elasticsearch.xpack.esql.expression.function.scalar.math.Cosh;
|
|
|
@@ -279,6 +283,9 @@ public final class PlanNamedTypes {
|
|
|
of(QL_UNARY_SCLR_CLS, IsNull.class, PlanNamedTypes::writeQLUnaryScalar, PlanNamedTypes::readQLUnaryScalar),
|
|
|
of(QL_UNARY_SCLR_CLS, Not.class, PlanNamedTypes::writeQLUnaryScalar, PlanNamedTypes::readQLUnaryScalar),
|
|
|
of(ESQL_UNARY_SCLR_CLS, Abs.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
+ of(ESQL_UNARY_SCLR_CLS, Acos.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
+ of(ESQL_UNARY_SCLR_CLS, Asin.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
+ of(ESQL_UNARY_SCLR_CLS, Atan.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
of(ESQL_UNARY_SCLR_CLS, Cos.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
of(ESQL_UNARY_SCLR_CLS, Cosh.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
of(ESQL_UNARY_SCLR_CLS, Floor.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
@@ -302,6 +309,7 @@ public final class PlanNamedTypes {
|
|
|
of(ESQL_UNARY_SCLR_CLS, ToVersion.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
of(ESQL_UNARY_SCLR_CLS, Trim.class, PlanNamedTypes::writeESQLUnaryScalar, PlanNamedTypes::readESQLUnaryScalar),
|
|
|
// ScalarFunction
|
|
|
+ of(ScalarFunction.class, Atan2.class, PlanNamedTypes::writeAtan2, PlanNamedTypes::readAtan2),
|
|
|
of(ScalarFunction.class, AutoBucket.class, PlanNamedTypes::writeAutoBucket, PlanNamedTypes::readAutoBucket),
|
|
|
of(ScalarFunction.class, Case.class, PlanNamedTypes::writeCase, PlanNamedTypes::readCase),
|
|
|
of(ScalarFunction.class, CIDRMatch.class, PlanNamedTypes::writeCIDRMatch, PlanNamedTypes::readCIDRMatch),
|
|
|
@@ -995,6 +1003,9 @@ public final class PlanNamedTypes {
|
|
|
|
|
|
static final Map<String, BiFunction<Source, Expression, UnaryScalarFunction>> ESQL_UNARY_SCALAR_CTRS = Map.ofEntries(
|
|
|
entry(name(Abs.class), Abs::new),
|
|
|
+ entry(name(Acos.class), Acos::new),
|
|
|
+ entry(name(Asin.class), Asin::new),
|
|
|
+ entry(name(Atan.class), Atan::new),
|
|
|
entry(name(Cos.class), Cos::new),
|
|
|
entry(name(Cosh.class), Cosh::new),
|
|
|
entry(name(Floor.class), Floor::new),
|
|
|
@@ -1072,6 +1083,15 @@ public final class PlanNamedTypes {
|
|
|
|
|
|
// -- ScalarFunction
|
|
|
|
|
|
+ static Atan2 readAtan2(PlanStreamInput in) throws IOException {
|
|
|
+ return new Atan2(Source.EMPTY, in.readExpression(), in.readExpression());
|
|
|
+ }
|
|
|
+
|
|
|
+ static void writeAtan2(PlanStreamOutput out, Atan2 atan2) throws IOException {
|
|
|
+ out.writeExpression(atan2.y());
|
|
|
+ out.writeExpression(atan2.x());
|
|
|
+ }
|
|
|
+
|
|
|
static AutoBucket readAutoBucket(PlanStreamInput in) throws IOException {
|
|
|
return new AutoBucket(Source.EMPTY, in.readExpression(), in.readExpression(), in.readExpression(), in.readExpression());
|
|
|
}
|