feat(query): support timezone() function

TD-14243

include/common/ttokendef.h

@@ -183,48 +183,49 @@
 #define TK_LAST                           165
 #define TK_NOW                            166
 #define TK_TODAY                          167
-#define TK_CAST                           168
-#define TK_ROWTS                          169
-#define TK_TBNAME                         170
-#define TK_QSTARTTS                       171
-#define TK_QENDTS                         172
-#define TK_WSTARTTS                       173
-#define TK_WENDTS                         174
-#define TK_WDURATION                      175
-#define TK_BETWEEN                        176
-#define TK_IS                             177
-#define TK_NK_LT                          178
-#define TK_NK_GT                          179
-#define TK_NK_LE                          180
-#define TK_NK_GE                          181
-#define TK_NK_NE                          182
-#define TK_MATCH                          183
-#define TK_NMATCH                         184
-#define TK_JOIN                           185
-#define TK_INNER                          186
-#define TK_SELECT                         187
-#define TK_DISTINCT                       188
-#define TK_WHERE                          189
-#define TK_PARTITION                      190
-#define TK_BY                             191
-#define TK_SESSION                        192
-#define TK_STATE_WINDOW                   193
-#define TK_SLIDING                        194
-#define TK_FILL                           195
-#define TK_VALUE                          196
-#define TK_NONE                           197
-#define TK_PREV                           198
-#define TK_LINEAR                         199
-#define TK_NEXT                           200
-#define TK_GROUP                          201
-#define TK_HAVING                         202
-#define TK_ORDER                          203
-#define TK_SLIMIT                         204
-#define TK_SOFFSET                        205
-#define TK_LIMIT                          206
-#define TK_OFFSET                         207
-#define TK_ASC                            208
-#define TK_NULLS                          209
+#define TK_TIMEZONE                       168
+#define TK_CAST                           169
+#define TK_ROWTS                          170
+#define TK_TBNAME                         171
+#define TK_QSTARTTS                       172
+#define TK_QENDTS                         173
+#define TK_WSTARTTS                       174
+#define TK_WENDTS                         175
+#define TK_WDURATION                      176
+#define TK_BETWEEN                        177
+#define TK_IS                             178
+#define TK_NK_LT                          179
+#define TK_NK_GT                          180
+#define TK_NK_LE                          181
+#define TK_NK_GE                          182
+#define TK_NK_NE                          183
+#define TK_MATCH                          184
+#define TK_NMATCH                         185
+#define TK_JOIN                           186
+#define TK_INNER                          187
+#define TK_SELECT                         188
+#define TK_DISTINCT                       189
+#define TK_WHERE                          190
+#define TK_PARTITION                      191
+#define TK_BY                             192
+#define TK_SESSION                        193
+#define TK_STATE_WINDOW                   194
+#define TK_SLIDING                        195
+#define TK_FILL                           196
+#define TK_VALUE                          197
+#define TK_NONE                           198
+#define TK_PREV                           199
+#define TK_LINEAR                         200
+#define TK_NEXT                           201
+#define TK_GROUP                          202
+#define TK_HAVING                         203
+#define TK_ORDER                          204
+#define TK_SLIMIT                         205
+#define TK_SOFFSET                        206
+#define TK_LIMIT                          207
+#define TK_OFFSET                         208
+#define TK_ASC                            209
+#define TK_NULLS                          210
 
 #define TK_NK_SPACE                       300
 #define TK_NK_COMMENT                     301

include/libs/scalar/scalar.h

@@ -80,6 +80,7 @@ int32_t timeTruncateFunction(SScalarParam *pInput, int32_t inputNum, SScalarPara
 int32_t timeDiffFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput);
 int32_t nowFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput);
 int32_t todayFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput);
+int32_t timezoneFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput);
 
 bool getTimePseudoFuncEnv(struct SFunctionNode* pFunc, SFuncExecEnv* pEnv);
 

source/libs/function/src/builtins.c

@@ -138,6 +138,12 @@ static int32_t translateTimePseudoColumn(SFunctionNode* pFunc, char* pErrBuf, in
   return TSDB_CODE_SUCCESS;
 }
 
+static int32_t translateTimezone(SFunctionNode* pFunc, char* pErrBuf, int32_t len) {
+  // pseudo column do not need to check parameters
+  pFunc->node.resType = (SDataType){.bytes = TD_TIMEZONE_LEN, .type = TSDB_DATA_TYPE_BINARY};
+  return TSDB_CODE_SUCCESS;
+}
+
 static int32_t translatePercentile(SFunctionNode* pFunc, char* pErrBuf, int32_t len) {
   if (2 != LIST_LENGTH(pFunc->pParameterList)) {
     return invaildFuncParaNumErrMsg(pErrBuf, len, pFunc->functionName);
@@ -802,6 +808,16 @@ const SBuiltinFuncDefinition funcMgtBuiltins[] = {
     .sprocessFunc = todayFunction,
     .finalizeFunc = NULL
   },
+  {
+    .name = "timezone",
+    .type = FUNCTION_TYPE_TIMEZONE,
+    .classification = FUNC_MGT_SCALAR_FUNC,
+    .translateFunc = translateTimezone,
+    .getEnvFunc   = NULL,
+    .initFunc     = NULL,
+    .sprocessFunc = timezoneFunction,
+    .finalizeFunc = NULL
+  },
   {
     .name = "_rowts",
     .type = FUNCTION_TYPE_ROWTS,

source/libs/parser/inc/sql.y

@@ -505,6 +505,7 @@ function_name(A) ::= FIRST(B).
 function_name(A) ::= LAST(B).                                                     { A = B; }
 function_name(A) ::= NOW(B).                                                      { A = B; }
 function_name(A) ::= TODAY(B).                                                    { A = B; }
+function_name(A) ::= TIMEZONE(B).                                                 { A = B; }
 
 %type table_alias                                                                 { SToken }
 %destructor table_alias                                                           { }

source/libs/parser/src/parAstCreater.c

@@ -260,7 +260,8 @@ SNode* createValueNode(SAstCreateContext* pCxt, int32_t dataType, const SToken*
   CHECK_OUT_OF_MEM(val);
   if (NULL != pLiteral) {
     val->literal = strndup(pLiteral->z, pLiteral->n);
-    if (TK_NK_ID != pLiteral->type && (IS_VAR_DATA_TYPE(dataType) || TSDB_DATA_TYPE_TIMESTAMP == dataType)) {
+    if (TK_NK_ID != pLiteral->type && TK_TIMEZONE != pLiteral->type &&
+       (IS_VAR_DATA_TYPE(dataType) || TSDB_DATA_TYPE_TIMESTAMP == dataType)) {
       trimString(pLiteral->z, pLiteral->n, val->literal, pLiteral->n);
     }
     CHECK_OUT_OF_MEM(val->literal);
@@ -379,11 +380,11 @@ SNode* createFunctionNodeNoParam(SAstCreateContext* pCxt, const SToken* pFuncNam
       dataType = TSDB_DATA_TYPE_BIGINT;
       break;
     }
-    //case TK_TIMEZONE: {
-    //  strncpy(buf, tsTimezoneStr, strlen(tsTimezoneStr));
-    //  dataType = TSDB_DATA_TYPE_BINARY;
-    //  break;
-    //}
+    case TK_TIMEZONE: {
+      strncpy(buf, tsTimezoneStr, strlen(tsTimezoneStr));
+      dataType = TSDB_DATA_TYPE_BINARY;
+      break;
+    }
   }
   SToken token = {.type = pFuncName->type, .n = strlen(buf), .z = buf};
 

source/libs/parser/src/parTokenizer.c

@@ -174,6 +174,7 @@ static SKeyword keywordTable[] = {
     {"TAGS",          TK_TAGS},
     {"TBNAME",        TK_TBNAME},
     {"TIMESTAMP",     TK_TIMESTAMP},
+    {"TIMEZONE",      TK_TIMEZONE},
     {"TINYINT",       TK_TINYINT},
     {"TODAY",         TK_TODAY},
     {"TOPIC",         TK_TOPIC},

source/libs/scalar/src/sclfunc.c

@@ -1275,6 +1275,14 @@ int32_t todayFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOut
   return TSDB_CODE_SUCCESS;
 }
 
+int32_t timezoneFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput) {
+  if (inputNum != 1) {
+    return TSDB_CODE_FAILED;
+  }
+  colDataAppend(pOutput->columnData, pOutput->numOfRows, (char *)colDataGetData(pInput->columnData, 0), false);
+  return TSDB_CODE_SUCCESS;
+}
+
 int32_t atanFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutput) {
   return doScalarFunctionUnique(pInput, inputNum, pOutput, atan);
 }