feat!: annotation support for LOCALTIME function (#6651)

* annotation support for LOCALTIME func

* handled mysql case

Author: AbhishekASLK

sqlglot/dialects/mysql.py

@@ -30,6 +30,7 @@
 from sqlglot.generator import unsupported_args
 from sqlglot.helper import seq_get
 from sqlglot.tokens import TokenType
+from sqlglot.typing.mysql import EXPRESSION_METADATA
 
 
 def _show_parser(*args: t.Any, **kwargs: t.Any) -> t.Callable[[MySQL.Parser], exp.Show]:
@@ -166,6 +167,8 @@ class MySQL(Dialect):
     SAFE_TO_ELIMINATE_DOUBLE_NEGATION = False
     LEAST_GREATEST_IGNORES_NULLS = False
 
+    EXPRESSION_METADATA = EXPRESSION_METADATA.copy()
+
     # https://prestodb.io/docs/current/functions/datetime.html#mysql-date-functions
     TIME_MAPPING = {
         "%M": "%B",

sqlglot/typing/__init__.py

@@ -144,6 +144,7 @@
         expr_type: {"returns": exp.DataType.Type.TIME}
         for expr_type in {
             exp.CurrentTime,
+            exp.Localtime,
             exp.Time,
             exp.TimeAdd,
             exp.TimeSub,

sqlglot/typing/mysql.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+from sqlglot import exp
+from sqlglot.typing import EXPRESSION_METADATA
+
+EXPRESSION_METADATA = {
+    **EXPRESSION_METADATA,
+    exp.Localtime: {"returns": exp.DataType.Type.DATETIME},
+}

tests/fixtures/optimizer/annotate_functions.sql

@@ -22,6 +22,9 @@ DOUBLE;
 CURRENT_TIME();
 TIME;
 
+LOCALTIME();
+TIME;
+
 TIME_ADD(CAST('09:05:03' AS TIME), INTERVAL 2 HOUR);
 TIME;
 
@@ -4831,3 +4834,11 @@ TIMESTAMPTZ;
 # dialect: tsql
 RADIANS(90);
 INT;
+
+--------------------------------------
+-- MySQL
+--------------------------------------
+
+# dialect: mysql
+LOCALTIME;
+DATETIME;