[feat](func) Support CURTIME with precision

Author: linrrzqqq

be/src/vec/functions/function_date_or_datetime_computation.h

@@ -838,10 +838,31 @@ struct CurrentTimeImpl {
     static Status execute(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
                           uint32_t result, size_t input_rows_count) {
         auto col_to = ColumnTimeV2::create();
-        VecDateTimeValue dtv;
+        DateV2Value<DateTimeV2ValueType> dtv;
         dtv.from_unixtime(context->state()->timestamp_ms() / 1000,
                           context->state()->timezone_obj());
-        auto time = TimeValue::make_time(dtv.hour(), dtv.minute(), dtv.second());
+        double time;
+        if (arguments.size() == 1) {
+            // the precision must be const, which is checked in fe.
+            const auto* col = assert_cast<const ColumnInt8*>(
+                    block.get_by_position(arguments[0]).column.get());
+            uint8_t precision = col->get_element(0);
+            if (precision >= 0 && precision <= 6) {
+                dtv.from_unixtime(context->state()->timestamp_ms() / 1000,
+                                  context->state()->nano_seconds(),
+                                  context->state()->timezone_obj(), precision);
+                time = TimeValue::make_time(dtv.hour(), dtv.minute(), dtv.second(),
+                                            dtv.microsecond());
+            } else {
+                return Status::InvalidArgument(
+                        "The precision in function CURTIME should be between 0 and 6, but got {}",
+                        precision);
+            }
+        } else {
+            dtv.from_unixtime(context->state()->timestamp_ms() / 1000,
+                              context->state()->timezone_obj());
+            time = TimeValue::make_time(dtv.hour(), dtv.minute(), dtv.second());
+        }
         col_to->insert_value(time);
         block.get_by_position(result).column =
                 ColumnConst::create(std::move(col_to), input_rows_count);

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/executable/DateTimeAcquire.java

@@ -24,6 +24,7 @@
 import org.apache.doris.nereids.trees.expressions.literal.DateV2Literal;
 import org.apache.doris.nereids.trees.expressions.literal.IntegerLiteral;
 import org.apache.doris.nereids.trees.expressions.literal.TimeV2Literal;
+import org.apache.doris.nereids.trees.expressions.literal.TinyIntLiteral;
 import org.apache.doris.nereids.util.DateUtils;
 
 import java.time.LocalDateTime;
@@ -95,11 +96,21 @@ public static Expression curTime() {
         return TimeV2Literal.fromJavaDateType(LocalDateTime.now(DateUtils.getTimeZone()));
     }
 
+    @ExecFunction(name = "curtime")
+    public static Expression curTime(TinyIntLiteral precision) {
+        return TimeV2Literal.fromJavaDateType(LocalDateTime.now(DateUtils.getTimeZone()), precision.getValue());
+    }
+
     @ExecFunction(name = "current_time")
     public static Expression currentTime() {
         return TimeV2Literal.fromJavaDateType(LocalDateTime.now(DateUtils.getTimeZone()));
     }
 
+    @ExecFunction(name = "current_time")
+    public static Expression currentTime(TinyIntLiteral precision) {
+        return TimeV2Literal.fromJavaDateType(LocalDateTime.now(DateUtils.getTimeZone()), precision.getValue());
+    }
+
     /**
      * date transformation function: unix_timestamp
      */

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/functions/scalar/CurrentTime.java

@@ -18,12 +18,16 @@
 package org.apache.doris.nereids.trees.expressions.functions.scalar;
 
 import org.apache.doris.catalog.FunctionSignature;
+import org.apache.doris.nereids.exceptions.AnalysisException;
 import org.apache.doris.nereids.trees.expressions.Expression;
 import org.apache.doris.nereids.trees.expressions.functions.AlwaysNotNullable;
-import org.apache.doris.nereids.trees.expressions.functions.ExplicitlyCastableSignature;
+import org.apache.doris.nereids.trees.expressions.functions.ComputeSignature;
+import org.apache.doris.nereids.trees.expressions.functions.ImplicitlyCastableSignature;
+import org.apache.doris.nereids.trees.expressions.literal.TinyIntLiteral;
 import org.apache.doris.nereids.trees.expressions.shape.LeafExpression;
 import org.apache.doris.nereids.trees.expressions.visitor.ExpressionVisitor;
 import org.apache.doris.nereids.types.TimeV2Type;
+import org.apache.doris.nereids.types.TinyIntType;
 
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableList;
@@ -34,10 +38,11 @@
  * ScalarFunction 'current_time'. This class is generated by GenerateFunction.
  */
 public class CurrentTime extends ScalarFunction
-        implements LeafExpression, ExplicitlyCastableSignature, AlwaysNotNullable {
+        implements LeafExpression, ImplicitlyCastableSignature, AlwaysNotNullable, ComputeSignature {
 
     public static final List<FunctionSignature> SIGNATURES = ImmutableList.of(
-            FunctionSignature.ret(TimeV2Type.INSTANCE).args()
+            FunctionSignature.ret(TimeV2Type.INSTANCE).args(),
+            FunctionSignature.ret(TimeV2Type.INSTANCE).args(TinyIntType.INSTANCE)
     );
 
     /**
@@ -47,17 +52,51 @@ public CurrentTime() {
         super("current_time");
     }
 
+    /**
+     * constructor with 1 argument.
+     */
+    public CurrentTime(Expression arg) {
+        super("current_time", arg);
+    }
+
     /** constructor for withChildren and reuse signature */
     private CurrentTime(ScalarFunctionParams functionParams) {
         super(functionParams);
     }
 
+    @Override
+    public FunctionSignature computeSignature(FunctionSignature signature) {
+        if (arity() == 1 && child(0) instanceof TinyIntLiteral) {
+            byte precision = ((TinyIntLiteral) child(0)).getValue();
+            if (precision < 0 || precision > 6) {
+                throw new IllegalArgumentException("The precision must be between 0 and 6");
+            }
+            return FunctionSignature.ret(TimeV2Type.of(precision)).args(TinyIntType.INSTANCE);
+        }
+
+        return super.computeSignature(signature);
+    }
+
     @Override
     public Expression withChildren(List<Expression> children) {
-        Preconditions.checkArgument(children.isEmpty());
+        Preconditions.checkArgument(children.isEmpty() || children.size() == 1);
         return new CurrentTime(getFunctionParams(children));
     }
 
+    @Override
+    public void checkLegalityAfterRewrite() {
+        if (arity() == 1) {
+            if (!child(0).isLiteral()) {
+                throw new AnalysisException("CURTIME only accepts literal as precision.");
+            }
+        }
+    }
+
+    @Override
+    public void checkLegalityBeforeTypeCoercion() {
+        checkLegalityAfterRewrite();
+    }
+
     @Override
     public List<FunctionSignature> getSignatures() {
         return SIGNATURES;

fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/literal/TimeV2Literal.java

@@ -301,6 +301,18 @@ public static Expression fromJavaDateType(LocalDateTime dateTime) {
         return new TimeV2Literal(dateTime.getHour(), dateTime.getMinute(), dateTime.getSecond(), 0, 0, false);
     }
 
+    /**
+     *  construct with precision
+     */
+    public static Expression fromJavaDateType(LocalDateTime dateTime, int precision) {
+        if (isDateOutOfRange(dateTime)) {
+            throw new AnalysisException("datetime out of range" + dateTime.toString());
+        }
+        int value = (int) Math.pow(10, TimeV2Type.MAX_SCALE - precision);
+        return new TimeV2Literal(dateTime.getHour(), dateTime.getMinute(), dateTime.getSecond(),
+                (dateTime.getNano() / 1000) / value * value, precision, false);
+    }
+
     public LocalDateTime toJavaDateType() {
         return LocalDateTime.of(0, 1, 1, ((int) getHour()), ((int) getMinute()), ((int) getSecond()),
                 (int) getMicroSecond() * 1000);

regression-test/suites/query_p0/sql_functions/datetime_functions/test_date_function.groovy

@@ -226,6 +226,16 @@ suite("test_date_function") {
     // TIME CURTIME()
     def curtime_result = sql """ SELECT CURTIME() """
     assertTrue(curtime_result[0].size() == 1)
+    curtime_result = sql """ SELECT CURTIME(3) """
+    assertTrue(curtime_result[0].size() == 1)
+    test {
+        sql """ SELECT CURTIME(114514);"""
+        exception "Can not find the compatibility function signature: current_time(INT)"
+    }
+    test {
+        sql """ SELECT CURTIME(7); """
+        exception "The precision must be between 0 and 6"
+    }
 
     sql """ insert into ${tableName} values ("2010-11-30 23:59:59") """
     // DATE_ADD