fix sql for round mathematical function

Author: fupelaqu

es6/sql-bridge/src/test/scala/app/softnetwork/elastic/sql/SQLQuerySpec.scala

@@ -2219,7 +2219,7 @@ class SQLQuerySpec extends AnyFlatSpec with Matchers {
       |        "source": "(def arg0 = (!doc.containsKey('identifier') || doc['identifier'].empty ? null : doc['identifier'].value); (arg0 == null) ? null : Math.pow(arg0, 3))"
       |      }
       |    },
-      |    "round_identifier_0": {
+      |    "round_identifier": {
       |      "script": {
       |        "lang": "painless",
       |        "source": "(def arg0 = (!doc.containsKey('identifier') || doc['identifier'].empty ? null : doc['identifier'].value); (arg0 == null) ? null : (def p = Math.pow(10, 0); Math.round((arg0 * p) / p)))"

sql/bridge/src/test/scala/app/softnetwork/elastic/sql/SQLQuerySpec.scala

@@ -2208,7 +2208,7 @@ class SQLQuerySpec extends AnyFlatSpec with Matchers {
         |        "source": "(def arg0 = (!doc.containsKey('identifier') || doc['identifier'].empty ? null : doc['identifier'].value); (arg0 == null) ? null : Math.pow(arg0, 3))"
         |      }
         |    },
-        |    "round_identifier_0": {
+        |    "round_identifier": {
         |      "script": {
         |        "lang": "painless",
         |        "source": "(def arg0 = (!doc.containsKey('identifier') || doc['identifier'].empty ? null : doc['identifier'].value); (arg0 == null) ? null : (def p = Math.pow(10, 0); Math.round((arg0 * p) / p)))"

sql/src/main/scala/app/softnetwork/elastic/sql/SQLFunction.scala

@@ -994,18 +994,14 @@ case class SQLPow(arg: PainlessScript, exponent: Int) extends MathematicalFuncti
   override def nullable: Boolean = arg.nullable
 }
 
-case class SQLRound(arg: PainlessScript, scale: Int = 0) extends MathematicalFunction {
+case class SQLRound(arg: PainlessScript, scale: Option[Int]) extends MathematicalFunction {
   override def operator: UnaryArithmeticOperator = Round
 
   override def args: List[PainlessScript] =
-    List(arg) :+ SQLIntValue(scale)
+    List(arg) ++ scale.map(SQLIntValue(_)).toList
 
   override def toPainlessCall(callArgs: List[String]): String =
-    callArgs match {
-      case List(arg0, arg1) =>
-        s"(def p = ${SQLPow(SQLIntValue(10), arg1.toInt).painless}; ${operator.painless}(($arg0 * p) / p))"
-      case _ => throw new IllegalArgumentException("ROUND requires one or two arguments")
-    }
+    s"(def p = ${SQLPow(SQLIntValue(10), scale.getOrElse(0)).painless}; ${operator.painless}((${callArgs.head} * p) / p))"
 }
 
 case class SQLSign(arg: PainlessScript) extends MathematicalFunction {

sql/src/main/scala/app/softnetwork/elastic/sql/SQLParser.scala

@@ -465,7 +465,7 @@ trait SQLParser extends RegexParsers with PackratParsers { _: SQLWhereParser =>
 
   def roundFunction: PackratParser[MathematicalFunction] =
     round ~ start ~ valueExpr ~ separator.? ~ long.? ~ end ^^ { case _ ~ _ ~ v ~ _ ~ s ~ _ =>
-      SQLRound(v, s.map(_.value.toInt).getOrElse(0))
+      SQLRound(v, s.map(_.value.toInt))
     }
 
   private[this] def pow: PackratParser[UnaryArithmeticOperator] = Pow.regex ^^ (_ => Pow)