• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #5036

28 Apr 2026 02:11PM UTC coverage: 73.055% (-0.005%) from 73.06%
#5036

push

travis-ci

web-flow
func/regexp_extract: new scalar func and test cases (#35191)

46 of 59 new or added lines in 1 file covered. (77.97%)

5784 existing lines in 151 files now uncovered.

276105 of 377940 relevant lines covered (73.06%)

133708549.2 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

82.45
/source/libs/function/src/functionMgt.c
1
/*
2
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3
 *
4
 * This program is free software: you can use, redistribute, and/or modify
5
 * it under the terms of the GNU Affero General Public License, version 3
6
 * or later ("AGPL"), as published by the Free Software Foundation.
7
 *
8
 * This program is distributed in the hope that it will be useful, but WITHOUT
9
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10
 * FITNESS FOR A PARTICULAR PURPOSE.
11
 *
12
 * You should have received a copy of the GNU Affero General Public License
13
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14
 */
15

16
#include "functionMgt.h"
17

18
#include "builtins.h"
19
#include "builtinsimpl.h"
20
#include "functionMgtInt.h"
21
#include "nodes.h"
22
#include "querynodes.h"
23
#include "taos.h"
24
#include "taoserror.h"
25
#include "thash.h"
26
#include "tudf.h"
27

28
typedef struct SFuncMgtService {
29
  SHashObj* pFuncNameHashTable;
30
} SFuncMgtService;
31

32
static SFuncMgtService gFunMgtService;
33
static TdThreadOnce    functionHashTableInit = PTHREAD_ONCE_INIT;
34
static int32_t         initFunctionCode = 0;
35

36
static void InitSpecialFuncId() {
1,555,625✔
37
  // just to make sure the funcId of these functions are initialized before used.
38
  (void)fmGetTwstartFuncId();
1,555,625✔
39
  (void)fmGetTwendFuncId();
1,555,625✔
40
  (void)fmGetTwdurationFuncId();
1,555,625✔
41
  (void)fmGetExternalWindowColumnFuncId();
1,555,625✔
42
}
1,555,625✔
43

44
static void doInitFunctionTable() {
1,555,625✔
45
  gFunMgtService.pFuncNameHashTable =
1,555,625✔
46
      taosHashInit(funcMgtBuiltinsNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
1,555,625✔
47
  if (NULL == gFunMgtService.pFuncNameHashTable) {
1,555,625✔
48
    initFunctionCode = terrno;
×
49
    return;
×
50
  }
51

52
  for (int32_t i = 0; i < funcMgtBuiltinsNum; ++i) {
362,460,625✔
53
    if (TSDB_CODE_SUCCESS != taosHashPut(gFunMgtService.pFuncNameHashTable, funcMgtBuiltins[i].name,
360,905,000✔
54
                                         strlen(funcMgtBuiltins[i].name), &i, sizeof(int32_t))) {
360,905,000✔
55
      initFunctionCode = terrno;
×
56
      return;
×
57
    }
58
  }
59
  InitSpecialFuncId();
1,555,625✔
60
}
61

62
static bool isSpecificClassifyFunc(int32_t funcId, uint64_t classification) {
2,147,483,647✔
63
  if (fmIsUserDefinedFunc(funcId)) {
2,147,483,647✔
64
    return FUNC_MGT_AGG_FUNC == classification
65
               ? FUNC_AGGREGATE_UDF_ID == funcId
66
               : (FUNC_MGT_SCALAR_FUNC == classification ? FUNC_SCALAR_UDF_ID == funcId : false);
8,908,648✔
67
  }
68
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
2,147,483,647✔
69
    return false;
2,147,483,647✔
70
  }
71
  return FUNC_MGT_TEST_MASK(funcMgtBuiltins[funcId].classification, classification);
2,147,483,647✔
72
}
73

74
int32_t fmFuncMgtInit() {
1,555,625✔
75
  (void)taosThreadOnce(&functionHashTableInit, doInitFunctionTable);
1,555,625✔
76
  return initFunctionCode;
1,555,625✔
77
}
78

79
int32_t fmGetFuncInfo(SFunctionNode* pFunc, char* pMsg, int32_t msgLen) {
858,100,516✔
80
  if (NULL != gFunMgtService.pFuncNameHashTable) {
858,100,516✔
81
    void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, pFunc->functionName, strlen(pFunc->functionName));
857,933,073✔
82
    if (NULL != pVal) {
857,954,306✔
83
      pFunc->funcId = *(int32_t*)pVal;
857,748,483✔
84
      pFunc->funcType = funcMgtBuiltins[pFunc->funcId].type;
857,748,418✔
85
      return funcMgtBuiltins[pFunc->funcId].translateFunc(pFunc, pMsg, msgLen);
857,748,880✔
86
    }
87
    return TSDB_CODE_FUNC_NOT_BUILTIN_FUNTION;
205,823✔
88
  }
89
  for (int32_t i = 0; i < funcMgtBuiltinsNum; ++i) {
3,301,194✔
90
    if (0 == strcmp(funcMgtBuiltins[i].name, pFunc->functionName)) {
3,301,056✔
91
      pFunc->funcId = i;
177,408✔
92
      pFunc->funcType = funcMgtBuiltins[pFunc->funcId].type;
177,408✔
93
      return funcMgtBuiltins[pFunc->funcId].translateFunc(pFunc, pMsg, msgLen);
177,408✔
94
    }
95
  }
96
  return TSDB_CODE_FUNC_NOT_BUILTIN_FUNTION;
138✔
97
}
98

99
EFuncReturnRows fmGetFuncReturnRows(SFunctionNode* pFunc) {
19,127,214✔
100
  if (NULL != funcMgtBuiltins[pFunc->funcId].estimateReturnRowsFunc) {
19,127,214✔
101
    return funcMgtBuiltins[pFunc->funcId].estimateReturnRowsFunc(pFunc);
13,750,688✔
102
  }
103
  return (fmIsIndefiniteRowsFunc(pFunc->funcId) || fmIsMultiRowsFunc(pFunc->funcId)) ? FUNC_RETURN_ROWS_INDEFINITE
5,376,526✔
104
                                                                                     : FUNC_RETURN_ROWS_NORMAL;
5,376,526✔
105
}
106

107
bool canCoexistIndefiniteRowsFunc(int32_t funcId1, int32_t funcId2) {
181,400✔
108
  if (fmIsUserDefinedFunc(funcId1) || funcId1 < 0 || funcId1 >= funcMgtBuiltinsNum ||
362,800✔
109
      fmIsUserDefinedFunc(funcId2) || funcId2 < 0 || funcId2 >= funcMgtBuiltinsNum) {
362,800✔
110
    return false;
×
111
  }
112
  if (funcId1 == funcId2) {
181,400✔
113
    return true;
165,670✔
114
  }
115
  if ((funcMgtBuiltins[funcId1].type == FUNCTION_TYPE_LAG || funcMgtBuiltins[funcId1].type == FUNCTION_TYPE_LEAD) &&
15,730✔
116
      (funcMgtBuiltins[funcId2].type == FUNCTION_TYPE_LAG || funcMgtBuiltins[funcId2].type == FUNCTION_TYPE_LEAD)) {
15,282✔
117
    return true;
15,282✔
118
  }
119
  return false;
448✔
120
}
121

122
bool fmIsBuiltinFunc(const char* pFunc) {
16,632✔
123
  return NULL != taosHashGet(gFunMgtService.pFuncNameHashTable, pFunc, strlen(pFunc));
16,632✔
124
}
125

126
EFunctionType fmGetFuncType(const char* pFunc) {
625,144,994✔
127
  void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, pFunc, strlen(pFunc));
625,144,994✔
128
  if (NULL != pVal) {
625,162,494✔
129
    return funcMgtBuiltins[*(int32_t*)pVal].type;
624,953,343✔
130
  }
131
  return FUNCTION_TYPE_UDF;
209,151✔
132
}
133

134
EFunctionType fmGetFuncTypeFromId(int32_t funcId) {
24,207,326✔
135
  if (funcId >= 0 && funcId < funcMgtBuiltinsNum) {
24,207,326✔
136
    return funcMgtBuiltins[funcId].type;
24,207,326✔
137
  }
UNCOV
138
  return FUNCTION_TYPE_UDF;
×
139
}
140

141
EFuncDataRequired fmFuncDataRequired(SFunctionNode* pFunc, STimeWindow* pTimeWindow) {
42,012,471✔
142
  if (fmIsUserDefinedFunc(pFunc->funcId) || pFunc->funcId < 0 || pFunc->funcId >= funcMgtBuiltinsNum) {
42,012,471✔
143
    return FUNC_DATA_REQUIRED_DATA_LOAD;
×
144
  }
145
  if (NULL == funcMgtBuiltins[pFunc->funcId].dataRequiredFunc) {
42,017,910✔
146
    return FUNC_DATA_REQUIRED_DATA_LOAD;
×
147
  }
148
  return funcMgtBuiltins[pFunc->funcId].dataRequiredFunc(pFunc, pTimeWindow);
42,018,137✔
149
}
150

151
EFuncDataRequired fmFuncDynDataRequired(int32_t funcId, void* pRes, SDataBlockInfo* pBlockInfo) {
2,147,483,647✔
152
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
2,147,483,647✔
153
    return FUNC_DATA_REQUIRED_DATA_LOAD;
80✔
154
  }
155

156
  const char* name = funcMgtBuiltins[funcId].name;
2,147,483,647✔
157
  if ((strcmp(name, "_group_key") == 0) || (strcmp(name, "_select_value") == 0)) {
2,147,483,647✔
158
    return FUNC_DATA_REQUIRED_NOT_LOAD;
2,131,546✔
159
    ;
160
  }
161

162
  if (funcMgtBuiltins[funcId].dynDataRequiredFunc == NULL) {
2,147,483,647✔
163
    return FUNC_DATA_REQUIRED_DATA_LOAD;
175,072,143✔
164
  } else {
165
    return funcMgtBuiltins[funcId].dynDataRequiredFunc(pRes, pBlockInfo);
2,147,483,647✔
166
  }
167
}
168

169
int32_t fmGetFuncExecFuncs(int32_t funcId, SFuncExecFuncs* pFpSet) {
347,602,362✔
170
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
347,602,362✔
171
    return TSDB_CODE_FAILED;
14,118✔
172
  }
173
  pFpSet->getEnv = funcMgtBuiltins[funcId].getEnvFunc;
347,589,019✔
174
  pFpSet->init = funcMgtBuiltins[funcId].initFunc;
347,560,699✔
175
  pFpSet->process = funcMgtBuiltins[funcId].processFunc;
347,583,449✔
176
  pFpSet->finalize = funcMgtBuiltins[funcId].finalizeFunc;
347,557,790✔
177
  pFpSet->combine = funcMgtBuiltins[funcId].combineFunc;
347,576,765✔
178
  pFpSet->processFuncByRow = funcMgtBuiltins[funcId].processFuncByRow;
347,563,584✔
179
  pFpSet->cleanup = funcMgtBuiltins[funcId].cleanupFunc;
347,596,836✔
180
  return TSDB_CODE_SUCCESS;
347,565,444✔
181
}
182

183
int32_t fmGetUdafExecFuncs(int32_t funcId, SFuncExecFuncs* pFpSet) {
37,398✔
184
#ifdef USE_UDF
185
  if (!fmIsUserDefinedFunc(funcId)) {
37,398✔
186
    return TSDB_CODE_FAILED;
×
187
  }
188
  pFpSet->getEnv = udfAggGetEnv;
37,398✔
189
  pFpSet->init = udfAggInit;
37,398✔
190
  pFpSet->process = udfAggProcess;
37,398✔
191
  pFpSet->finalize = udfAggFinalize;
37,398✔
192
  return TSDB_CODE_SUCCESS;
37,398✔
193
#else
194
  TAOS_RETURN(TSDB_CODE_OPS_NOT_SUPPORT);
195
#endif
196
}
197

198
int32_t fmGetScalarFuncExecFuncs(int32_t funcId, SScalarFuncExecFuncs* pFpSet) {
978,936,512✔
199
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
978,936,512✔
200
    return TSDB_CODE_FAILED;
112,508✔
201
  }
202
  pFpSet->process = funcMgtBuiltins[funcId].sprocessFunc;
978,928,367✔
203
  pFpSet->getEnv = funcMgtBuiltins[funcId].getEnvFunc;
978,917,969✔
204
  return TSDB_CODE_SUCCESS;
978,945,089✔
205
}
206

207
bool fmIsAggFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_AGG_FUNC); }
2,147,483,647✔
208

209
bool fmIsScalarFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SCALAR_FUNC); }
2,147,483,647✔
210

211
bool fmIsVectorFunc(int32_t funcId) { return !fmIsScalarFunc(funcId) && !fmIsPseudoColumnFunc(funcId); }
1,334,451,703✔
212

213
bool fmIsSelectColsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SELECT_COLS_FUNC); }
157,115,055✔
214

215
bool fmIsSelectFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SELECT_FUNC); }
2,036,402,312✔
216

217
bool fmIsTimelineFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_TIMELINE_FUNC); }
1,205,398,691✔
218

219
bool fmIsDateTimeFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_DATETIME_FUNC); }
609,907,422✔
220

221
bool fmIsPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_PSEUDO_COLUMN_FUNC); }
2,147,483,647✔
222

223
bool fmIsScanPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SCAN_PC_FUNC); }
2,147,483,647✔
224

225
bool fmIsWindowPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_WINDOW_PC_FUNC); }
1,541,704,823✔
226

227
bool fmIsWindowClauseFunc(int32_t funcId) { return fmIsAggFunc(funcId) || fmIsWindowPseudoColumnFunc(funcId); }
60,938,840✔
228

229
bool fmIsWindowIndefRowsFunc(int32_t funcId) { return fmIsVectorFunc(funcId) || fmIsWindowPseudoColumnFunc(funcId); }
400,930✔
230

231
bool fmIsStreamWindowClauseFunc(int32_t funcId) { return fmIsWindowClauseFunc(funcId) || fmIsPlaceHolderFunc(funcId); }
1,117,556✔
232

233
bool fmIsStreamVectorFunc(int32_t funcId) { return fmIsVectorFunc(funcId) || fmIsPlaceHolderFunc(funcId); }
10,692✔
234

235
bool fmIsIndefiniteRowsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INDEFINITE_ROWS_FUNC); }
1,653,522,054✔
236

237
bool fmIsSpecialDataRequiredFunc(int32_t funcId) {
94,680,963✔
238
  return isSpecificClassifyFunc(funcId, FUNC_MGT_SPECIAL_DATA_REQUIRED);
94,680,963✔
239
}
240

241
bool fmIsDynamicScanOptimizedFunc(int32_t funcId) {
45,217,813✔
242
  return isSpecificClassifyFunc(funcId, FUNC_MGT_DYNAMIC_SCAN_OPTIMIZED);
45,217,813✔
243
}
244

245
bool fmIsMultiResFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_MULTI_RES_FUNC); }
1,240,750,877✔
246

247
bool fmIsRepeatScanFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_REPEAT_SCAN_FUNC); }
1,218,287,948✔
248

249
bool fmIsUserDefinedFunc(int32_t funcId) { return funcId > FUNC_UDF_ID_START; }
2,147,483,647✔
250

251
bool fmIsForbidFillFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_FORBID_FILL_FUNC); }
610,190,192✔
252

253
bool fmIsIntervalInterpoFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INTERVAL_INTERPO_FUNC); }
65,307,034✔
254

255
bool fmIsSystemInfoFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SYSTEM_INFO_FUNC); }
613,247,672✔
256

257
bool fmIsImplicitTsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_IMPLICIT_TS_FUNC); }
2,147,483,647✔
258

259
bool fmIsClientPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_CLIENT_PC_FUNC); }
622,899,355✔
260

261
bool fmIsMultiRowsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_MULTI_ROWS_FUNC); }
1,215,674,205✔
262

263
static bool fmIsKeepOrderFuncId(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_KEEP_ORDER_FUNC); }
734,447,190✔
264

265
bool fmIsKeepOrderFunc(SFunctionNode* pFunc) {
740,981,606✔
266
  if (pFunc->funcType == FUNCTION_TYPE_TOP || pFunc->funcType == FUNCTION_TYPE_BOTTOM ||
740,981,606✔
267
      pFunc->funcType == FUNCTION_TYPE_SAMPLE) {
736,842,604✔
268
    if (pFunc->pParameterList != NULL && pFunc->pParameterList->length >= 2) {
6,551,021✔
269
      SNode* pParam = nodesListGetNode(pFunc->pParameterList, 1);
6,569,215✔
270
      if (pParam != NULL && nodeType(pParam) == QUERY_NODE_VALUE) {
6,569,215✔
271
        SValueNode* pConst = (SValueNode*)pParam;
6,569,215✔
272
        if (pConst->datum.i == 1) {
6,569,215✔
273
          return true;
764,677✔
274
        }
275
      }
276
    }
277
    return false;
5,794,866✔
278
  }
279
  return fmIsKeepOrderFuncId(pFunc->funcId);
734,458,633✔
280
}
281

282
bool fmIsCumulativeFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_CUMULATIVE_FUNC); }
159,659,218✔
283

284
bool fmIsForbidSysTableFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_FORBID_SYSTABLE_FUNC); }
610,113,617✔
285

286
bool fmIsInterpFunc(int32_t funcId) {
1,251,081,821✔
287
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
1,251,081,821✔
288
    return false;
384,523✔
289
  }
290
  return FUNCTION_TYPE_INTERP == funcMgtBuiltins[funcId].type;
1,250,700,564✔
291
}
292

293
bool fmIsInterpPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INTERP_PC_FUNC); }
1,229,913,459✔
294

295
bool fmIsForecastFunc(int32_t funcId) {
1,199,174,984✔
296
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
1,199,174,984✔
297
    return false;
384,717✔
298
  }
299
  return FUNCTION_TYPE_FORECAST == funcMgtBuiltins[funcId].type;
1,198,793,853✔
300
}
301

302
bool fmIsAnalysisPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_ANALYTICS_PC_FUNC); }
1,233,034,619✔
303

304
bool fmIsImputationCorrelationFunc(int32_t funcId) {
×
305
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
×
306
    return false;
×
307
  }
308

309
  int32_t type = funcMgtBuiltins[funcId].type;
×
310
  return FUNCTION_TYPE_IMPUTATION == type || FUNCTION_TYPE_DTW == type || FUNCTION_TYPE_DTW_PATH == type ||
×
311
         FUNCTION_TYPE_TLCC == type;
312
}
313

314
bool fmIsLastRowFunc(int32_t funcId) {
71,914,920✔
315
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
71,914,920✔
UNCOV
316
    return false;
×
317
  }
318
  return FUNCTION_TYPE_LAST_ROW == funcMgtBuiltins[funcId].type;
71,916,068✔
319
}
320

321
bool fmIsLastFunc(int32_t funcId) {
12,890✔
322
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
12,890✔
323
    return false;
×
324
  }
325
  return FUNCTION_TYPE_LAST == funcMgtBuiltins[funcId].type;
12,890✔
326
}
327

328
bool fmIsNotNullOutputFunc(int32_t funcId) {
520,143,906✔
329
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
520,143,906✔
330
    return false;
94,079✔
331
  }
332
  return FUNCTION_TYPE_LAST == funcMgtBuiltins[funcId].type ||
520,055,711✔
333
         FUNCTION_TYPE_CACHE_LAST == funcMgtBuiltins[funcId].type ||
499,945,777✔
334
         FUNCTION_TYPE_LAST_PARTIAL == funcMgtBuiltins[funcId].type ||
498,707,442✔
335
         FUNCTION_TYPE_LAST_MERGE == funcMgtBuiltins[funcId].type ||
489,799,681✔
336
         FUNCTION_TYPE_FIRST == funcMgtBuiltins[funcId].type ||
486,168,632✔
337
         FUNCTION_TYPE_FIRST_PARTIAL == funcMgtBuiltins[funcId].type ||
472,016,834✔
338
         FUNCTION_TYPE_FIRST_MERGE == funcMgtBuiltins[funcId].type ||
466,840,856✔
339
         FUNCTION_TYPE_COUNT == funcMgtBuiltins[funcId].type ||
464,853,154✔
340
         FUNCTION_TYPE_HYPERLOGLOG == funcMgtBuiltins[funcId].type ||
409,487,425✔
341
         FUNCTION_TYPE_HYPERLOGLOG_PARTIAL == funcMgtBuiltins[funcId].type ||
1,425,549,914✔
342
         FUNCTION_TYPE_HYPERLOGLOG_MERGE == funcMgtBuiltins[funcId].type;
405,460,652✔
343
}
344

345
bool fmIsSelectValueFunc(int32_t funcId) {
351,976,672✔
346
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
351,976,672✔
347
    return false;
148✔
348
  }
349
  return FUNCTION_TYPE_SELECT_VALUE == funcMgtBuiltins[funcId].type;
351,976,976✔
350
}
351

352
bool fmIsGroupKeyFunc(int32_t funcId) {
627,922,989✔
353
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
627,922,989✔
354
    return false;
×
355
  }
356
  return FUNCTION_TYPE_GROUP_KEY == funcMgtBuiltins[funcId].type;
627,924,549✔
357
}
358

359
bool fmIsHasNullFunc(int32_t funcId) {
×
360
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
×
361
    return false;
×
362
  }
363
  return FUNCTION_TYPE_HAS_NULL == funcMgtBuiltins[funcId].type;
×
364
}
365

366
bool fmisSelectGroupConstValueFunc(int32_t funcId) {
376,850,407✔
367
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
376,850,407✔
368
    return false;
×
369
  }
370
  return FUNCTION_TYPE_GROUP_CONST_VALUE == funcMgtBuiltins[funcId].type;
376,850,407✔
371
}
372

373
bool fmIsElapsedFunc(int32_t funcId) {
11,066,622✔
374
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
11,066,622✔
375
    return false;
×
376
  }
377
  return FUNCTION_TYPE_ELAPSED == funcMgtBuiltins[funcId].type;
11,066,622✔
378
}
379

380
bool fmIsBlockDistFunc(int32_t funcId) {
609,908,132✔
381
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
609,908,132✔
382
    return false;
196,370✔
383
  }
384
  return FUNCTION_TYPE_BLOCK_DIST == funcMgtBuiltins[funcId].type;
609,712,323✔
385
}
386

387
bool fmIsDBUsageFunc(int32_t funcId) {
609,906,047✔
388
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
609,906,047✔
389
    return false;
190,172✔
390
  }
391
  return FUNCTION_TYPE_DB_USAGE == funcMgtBuiltins[funcId].type;
609,718,227✔
392
}
393

394
bool fmIsProcessByRowFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_PROCESS_BY_ROW); }
839,034,624✔
395

396
bool fmIsVolatileFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_VOLATILE_FUNC); }
908,424,285✔
397

398
bool fmIsNoPushdownFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_NO_PUSHDOWN_FUNC); }
52,862,423✔
399

400
bool fmIsIgnoreNullFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_IGNORE_NULL_FUNC); }
254,597✔
401

402
void fmFuncMgtDestroy() {
1,555,672✔
403
  void* m = gFunMgtService.pFuncNameHashTable;
1,555,672✔
404
  if (m != NULL && atomic_val_compare_exchange_ptr((void**)&gFunMgtService.pFuncNameHashTable, m, 0) == m) {
1,555,672✔
405
    taosHashCleanup(m);
1,555,625✔
406
  }
407
}
1,555,672✔
408

409
#ifdef BUILD_NO_CALL
410
int32_t fmSetNormalFunc(int32_t funcId, SFuncExecFuncs* pFpSet) {
411
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
412
    return TSDB_CODE_FAILED;
413
  }
414
  pFpSet->process = funcMgtBuiltins[funcId].processFunc;
415
  return TSDB_CODE_SUCCESS;
416
}
417
#endif
418

419
// function has same input/output type
420
bool fmIsSameInOutType(int32_t funcId) {
49,579,057✔
421
  bool res = false;
49,579,057✔
422
  switch (funcMgtBuiltins[funcId].type) {
49,579,057✔
423
    case FUNCTION_TYPE_MAX:
19,772,118✔
424
    case FUNCTION_TYPE_MIN:
425
    case FUNCTION_TYPE_TOP:
426
    case FUNCTION_TYPE_BOTTOM:
427
    case FUNCTION_TYPE_FIRST:
428
    case FUNCTION_TYPE_LAST:
429
    case FUNCTION_TYPE_SAMPLE:
430
    case FUNCTION_TYPE_TAIL:
431
    case FUNCTION_TYPE_UNIQUE:
432
      res = true;
19,772,118✔
433
      break;
19,772,118✔
434
    default:
29,806,939✔
435
      break;
29,806,939✔
436
  }
437
  return res;
49,579,057✔
438
}
439

440
bool fmIsConstantResFunc(SFunctionNode* pFunc) {
15,091,421✔
441
  SNode* pNode;
442
  FOREACH(pNode, pFunc->pParameterList) {
15,179,093✔
443
    if (nodeType(pNode) != QUERY_NODE_VALUE) {
157,310✔
444
      return false;
69,638✔
445
    }
446
  }
447
  return true;
15,021,783✔
448
}
449

450
bool fmIsSkipScanCheckFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SKIP_SCAN_CHECK_FUNC); }
45,212,555✔
451

452
bool fmIsPrimaryKeyFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_PRIMARY_KEY_FUNC); }
529,547,282✔
453

454
bool fmIsPlaceHolderFunc(int32_t funcId) {return isSpecificClassifyFunc(funcId, FUNC_MGT_PLACE_HOLDER_FUNC); }
2,147,483,647✔
455

456
bool fmIsPlaceHolderFuncForExternalWin(int32_t funcId) {
687,781✔
457
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
687,781✔
458
    return false;
×
459
  }
460
  return (funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WSTART || funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WEND ||
1,154,552✔
461
          funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WDURATION);
466,771✔
462
}
463

464
void getLastCacheDataType(SDataType* pType, int32_t pkBytes) {
×
465
  // TODO: do it later.
466
  pType->bytes = getFirstLastInfoSize(pType->bytes, pkBytes) + VARSTR_HEADER_SIZE;
×
467
  pType->type = TSDB_DATA_TYPE_BINARY;
×
468
}
×
469

470
static int32_t getFuncInfo(SFunctionNode* pFunc) {
158,608,376✔
471
  char msg[128] = {0};
158,608,376✔
472
  return fmGetFuncInfo(pFunc, msg, sizeof(msg));
158,608,376✔
473
}
474

475
int32_t createFunction(const char* pName, SNodeList* pParameterList, SFunctionNode** ppFunc) {
10,362,081✔
476
  int32_t code = nodesMakeNode(QUERY_NODE_FUNCTION, (SNode**)ppFunc);
10,362,081✔
477
  if (NULL == *ppFunc) {
10,362,081✔
478
    return code;
×
479
  }
480
  (void)snprintf((*ppFunc)->functionName, sizeof((*ppFunc)->functionName), "%s", pName);
10,362,081✔
481
  (*ppFunc)->pParameterList = pParameterList;
10,362,081✔
482
  code = getFuncInfo((*ppFunc));
10,362,081✔
483
  if (TSDB_CODE_SUCCESS != code) {
10,362,081✔
484
    (*ppFunc)->pParameterList = NULL;
×
485
    nodesDestroyNode((SNode*)*ppFunc);
×
486
    *ppFunc = NULL;
×
487
    return code;
×
488
  }
489
  return code;
10,362,081✔
490
}
491

492
static void resetOutputChangedFunc(SFunctionNode *pFunc, const SFunctionNode* pSrcFunc) {
148,248,433✔
493
  if (funcMgtBuiltins[pFunc->funcId].type == FUNCTION_TYPE_LAST_MERGE) {
148,248,433✔
494
    pFunc->node.resType = pSrcFunc->node.resType;
23,697,796✔
495
    return;
23,697,796✔
496
  }
497
}
498

499
int32_t createFunctionWithSrcFunc(const char* pName, const SFunctionNode* pSrcFunc, SNodeList* pParameterList, SFunctionNode** ppFunc) {
148,248,372✔
500
  int32_t code = nodesMakeNode(QUERY_NODE_FUNCTION, (SNode**)ppFunc);
148,248,372✔
501
  if (NULL == *ppFunc) {
148,251,774✔
502
    return code;
×
503
  }
504

505
  (*ppFunc)->hasPk = pSrcFunc->hasPk;
148,251,774✔
506
  (*ppFunc)->pkBytes = pSrcFunc->pkBytes;
148,251,774✔
507
  (*ppFunc)->pSrcFuncRef = pSrcFunc;
148,251,774✔
508

509
  (void)snprintf((*ppFunc)->functionName, sizeof((*ppFunc)->functionName), "%s", pName);
148,251,774✔
510
  (*ppFunc)->pParameterList = pParameterList;
148,251,774✔
511
  code = getFuncInfo((*ppFunc));
148,251,730✔
512
  if (TSDB_CODE_SUCCESS != code) {
148,250,003✔
513
    (*ppFunc)->pParameterList = NULL;
×
514
    nodesDestroyNode((SNode*)*ppFunc);
×
515
    *ppFunc = NULL;
×
516
    return code;
×
517
  }
518
  resetOutputChangedFunc(*ppFunc, pSrcFunc);
148,250,003✔
519
  (*ppFunc)->node.relatedTo = pSrcFunc->node.relatedTo;
148,249,543✔
520
  (*ppFunc)->node.bindExprID = pSrcFunc->node.bindExprID;
148,249,543✔
521
  return code;
148,249,587✔
522
}
523

524
static int32_t createColumnByFunc(const SFunctionNode* pFunc, SColumnNode** ppCol) {
98,662,999✔
525
  int32_t code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)ppCol);
98,662,999✔
526
  if (NULL == *ppCol) {
98,668,038✔
527
    return code;
×
528
  }
529
  tstrncpy((*ppCol)->colName, pFunc->node.aliasName, TSDB_COL_NAME_LEN);
98,668,038✔
530
  (*ppCol)->node.resType = pFunc->node.resType;
98,668,038✔
531
  return TSDB_CODE_SUCCESS;
98,668,038✔
532
}
533

534
bool fmIsDistExecFunc(int32_t funcId) {
382,800,313✔
535
  if (fmIsUserDefinedFunc(funcId)) {
382,800,313✔
536
    return false;
105,219✔
537
  }
538
  if (!fmIsVectorFunc(funcId)) {
382,720,005✔
539
    return true;
495,028✔
540
  }
541
  return (NULL != funcMgtBuiltins[funcId].pPartialFunc && NULL != funcMgtBuiltins[funcId].pMergeFunc);
382,231,013✔
542
}
543

544
static int32_t createPartialFunction(const SFunctionNode* pSrcFunc, SFunctionNode** pPartialFunc) {
49,580,042✔
545
  SNodeList* pParameterList = NULL;
49,580,042✔
546
  int32_t    code = nodesCloneList(pSrcFunc->pParameterList, &pParameterList);
49,580,042✔
547
  if (NULL == pParameterList) {
49,582,668✔
548
    return code;
×
549
  }
550
  code =
551
      createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pPartialFunc, pSrcFunc, pParameterList, pPartialFunc);
49,582,668✔
552
  if (TSDB_CODE_SUCCESS != code) {
49,580,763✔
UNCOV
553
    nodesDestroyList(pParameterList);
×
554
    return code;
×
555
  }
556
  (*pPartialFunc)->hasOriginalFunc = true;
49,581,133✔
557
  (*pPartialFunc)->originalFuncId = pSrcFunc->hasOriginalFunc ? pSrcFunc->originalFuncId : pSrcFunc->funcId;
49,581,133✔
558
  char name[TSDB_FUNC_NAME_LEN + TSDB_NAME_DELIMITER_LEN + TSDB_POINTER_PRINT_BYTES + 1] = {0};
49,581,133✔
559

560
  int32_t len = snprintf(name, sizeof(name), "%s.%p", (*pPartialFunc)->functionName, pSrcFunc);
49,581,133✔
561
  if (taosHashBinary(name, len, sizeof(name)) < 0) {
49,581,934✔
562
    return TSDB_CODE_FAILED;
×
563
  }
564
  tstrncpy((*pPartialFunc)->node.aliasName, name, TSDB_COL_NAME_LEN);
49,581,934✔
565
  return TSDB_CODE_SUCCESS;
49,581,934✔
566
}
567

568
static int32_t createMergeFuncPara(const SFunctionNode* pSrcFunc, const SFunctionNode* pPartialFunc,
98,663,068✔
569
                                   SNodeList** pParameterList) {
570
  SNode*  pRes = NULL;
98,663,068✔
571
  int32_t code = createColumnByFunc(pPartialFunc, (SColumnNode**)&pRes);
98,663,068✔
572
  if (TSDB_CODE_SUCCESS != code) {
98,667,091✔
573
    return code;
×
574
  }
575
  if (NULL != funcMgtBuiltins[pSrcFunc->funcId].createMergeParaFuc) {
98,667,091✔
576
    return funcMgtBuiltins[pSrcFunc->funcId].createMergeParaFuc(pSrcFunc->pParameterList, pRes, pParameterList);
801,528✔
577
  } else {
578
    return nodesListMakeStrictAppend(pParameterList, pRes);
97,865,519✔
579
  }
580
}
581

582
static int32_t createMidFunction(const SFunctionNode* pSrcFunc, const SFunctionNode* pPartialFunc,
49,081,866✔
583
                                 SFunctionNode** pMidFunc) {
584
  SNodeList*     pParameterList = NULL;
49,081,866✔
585
  SFunctionNode* pFunc = NULL;
49,081,866✔
586

587
  int32_t code = createMergeFuncPara(pSrcFunc, pPartialFunc, &pParameterList);
49,081,866✔
588
  if (TSDB_CODE_SUCCESS == code) {
49,086,399✔
589
    if(funcMgtBuiltins[pSrcFunc->funcId].pMiddleFunc != NULL){
49,086,399✔
590
      code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMiddleFunc, pSrcFunc, pParameterList, &pFunc);
5,551,626✔
591
    }else{
592
      code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMergeFunc, pSrcFunc, pParameterList, &pFunc);
43,534,773✔
593
    }
594
  }
595
  if (TSDB_CODE_SUCCESS == code) {
49,085,874✔
596
    tstrncpy(pFunc->node.aliasName, pPartialFunc->node.aliasName, TSDB_COL_NAME_LEN);
49,086,148✔
597
  }
598

599
  if (TSDB_CODE_SUCCESS == code) {
49,085,830✔
600
    *pMidFunc = pFunc;
49,085,830✔
601
  } else {
602
    nodesDestroyList(pParameterList);
×
603
  }
604
  return code;
49,085,899✔
605
}
606

607
static int32_t createMergeFunction(const SFunctionNode* pSrcFunc, const SFunctionNode* pPartialFunc,
49,580,719✔
608
                                   SFunctionNode** pMergeFunc) {
609
  SNodeList*     pParameterList = NULL;
49,580,719✔
610
  SFunctionNode* pFunc = NULL;
49,580,719✔
611

612
  int32_t code = createMergeFuncPara(pSrcFunc, pPartialFunc, &pParameterList);
49,580,719✔
613
  if (TSDB_CODE_SUCCESS == code) {
49,582,831✔
614
    code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMergeFunc, pSrcFunc, pParameterList, &pFunc);
49,582,831✔
615
  }
616
  if (TSDB_CODE_SUCCESS == code) {
49,584,330✔
617
    pFunc->hasOriginalFunc = true;
49,582,668✔
618
    pFunc->originalFuncId = pSrcFunc->hasOriginalFunc ? pSrcFunc->originalFuncId : pSrcFunc->funcId;
49,582,668✔
619
    // overwrite function restype set by translate function
620
    if (fmIsSameInOutType(pSrcFunc->funcId)) {
49,582,668✔
621
      pFunc->node.resType = pSrcFunc->node.resType;
19,772,325✔
622
    }
623
    tstrncpy(pFunc->node.aliasName, pSrcFunc->node.aliasName, TSDB_COL_NAME_LEN);
49,579,979✔
624
  }
625

626
  if (TSDB_CODE_SUCCESS == code) {
49,581,641✔
627
    *pMergeFunc = pFunc;
49,581,641✔
628
  } else {
629
    nodesDestroyList(pParameterList);
×
630
  }
631
  return code;
49,581,831✔
632
}
633

634
int32_t fmGetDistMethod(const SFunctionNode* pFunc, SFunctionNode** pPartialFunc, SFunctionNode** pMidFunc,
49,578,024✔
635
                        SFunctionNode** pMergeFunc) {
636
  if (!fmIsDistExecFunc(pFunc->funcId)) {
49,578,024✔
637
    return TSDB_CODE_FAILED;
×
638
  }
639

640
  int32_t code = createPartialFunction(pFunc, pPartialFunc);
49,582,486✔
641
  if (TSDB_CODE_SUCCESS == code && pMidFunc) {
49,580,669✔
642
    code = createMidFunction(pFunc, *pPartialFunc, pMidFunc);
49,084,538✔
643
  }
644
  if (TSDB_CODE_SUCCESS == code) {
49,582,417✔
645
    code = createMergeFunction(pFunc, *pPartialFunc, pMergeFunc);
49,582,486✔
646
  }
647

648
  if (TSDB_CODE_SUCCESS != code) {
49,579,592✔
649
    nodesDestroyNode((SNode*)*pPartialFunc);
×
650
    if (pMidFunc) nodesDestroyNode((SNode*)*pMidFunc);
×
651
    nodesDestroyNode((SNode*)*pMergeFunc);
×
652
  }
653

654
  return code;
49,580,771✔
655
}
656

657
const char* fmGetFuncName(int32_t funcId) {
347,680✔
658
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
347,680✔
659
    return "invalid function";
×
660
  }
661
  return funcMgtBuiltins[funcId].name;
347,680✔
662
}
663

664
/// @param [out] pStateFunc, not changed if error occured or no need to create state func
665
/// @retval 0 for succ, otherwise err occured
666
static int32_t fmCreateStateFunc(const SFunctionNode* pFunc, SFunctionNode** pStateFunc) {
7,508✔
667
  if (funcMgtBuiltins[pFunc->funcId].pStateFunc) {
7,508✔
668
    SNodeList* pParams = NULL;
7,508✔
669
    int32_t    code = nodesCloneList(pFunc->pParameterList, &pParams);
7,508✔
670
    if (!pParams) return code;
7,508✔
671
    code = createFunction(funcMgtBuiltins[pFunc->funcId].pStateFunc, pParams, pStateFunc);
7,508✔
672
    if (TSDB_CODE_SUCCESS != code) {
7,508✔
673
      nodesDestroyList(pParams);
×
674
      return code;
×
675
    }
676
    tstrncpy((*pStateFunc)->node.aliasName, pFunc->node.aliasName, TSDB_COL_NAME_LEN);
7,508✔
677
    tstrncpy((*pStateFunc)->node.userAlias, pFunc->node.userAlias, TSDB_COL_NAME_LEN);
7,508✔
678
  }
679
  return TSDB_CODE_SUCCESS;
7,508✔
680
}
681

682
bool fmIsTSMASupportedFunc(func_id_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_TSMA_FUNC); }
1,274,584✔
683

684
bool fmIsRsmaSupportedFunc(func_id_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_RSMA_FUNC); }
139,480✔
685

686
int32_t fmCreateStateFuncs(SNodeList* pFuncs) {
3,306✔
687
  int32_t code;
688
  SNode*  pNode;
689
  char    buf[128] = {0};
3,306✔
690
  FOREACH(pNode, pFuncs) {
10,814✔
691
    SFunctionNode* pFunc = (SFunctionNode*)pNode;
7,508✔
692
    code = fmGetFuncInfo(pFunc, buf, 128);
7,508✔
693
    if (code) break;
7,508✔
694
    if (fmIsTSMASupportedFunc(pFunc->funcId)) {
7,508✔
695
      SFunctionNode* pNewFunc = NULL;
7,508✔
696
      code = fmCreateStateFunc(pFunc, &pNewFunc);
7,508✔
697
      if (code) {
7,508✔
698
        // error
699
        break;
×
700
      } else if (!pNewFunc) {
7,508✔
701
        // no need state func
702
        continue;
×
703
      } else {
704
        REPLACE_NODE(pNewFunc);
7,508✔
705
        nodesDestroyNode(pNode);
7,508✔
706
      }
707
    }
708
  }
709
  return code;
3,306✔
710
}
711

712
static int32_t fmCreateStateMergeFunc(SFunctionNode* pFunc, SFunctionNode** pStateMergeFunc) {
1,650✔
713
  if (funcMgtBuiltins[pFunc->funcId].pMergeFunc) {
1,650✔
714
    SNodeList* pParams = NULL;
1,350✔
715
    int32_t    code = nodesCloneList(pFunc->pParameterList, &pParams);
1,350✔
716
    if (!pParams) return code;
1,350✔
717
    code = createFunctionWithSrcFunc(funcMgtBuiltins[pFunc->funcId].pMergeFunc, pFunc, pParams, pStateMergeFunc);
1,350✔
718
    if (TSDB_CODE_SUCCESS != code) {
1,350✔
719
      nodesDestroyList(pParams);
×
720
      return code;
×
721
    }
722
    tstrncpy((*pStateMergeFunc)->node.aliasName, pFunc->node.aliasName, TSDB_COL_NAME_LEN);
1,350✔
723
    tstrncpy((*pStateMergeFunc)->node.userAlias, pFunc->node.userAlias, TSDB_COL_NAME_LEN);
1,350✔
724
  }
725
  return TSDB_CODE_SUCCESS;
1,650✔
726
}
727

728
int32_t fmCreateStateMergeFuncs(SNodeList* pFuncs) {
750✔
729
  int32_t code;
730
  SNode*  pNode;
731
  char    buf[128] = {0};
750✔
732
  FOREACH(pNode, pFuncs) {
2,400✔
733
    SFunctionNode* pFunc = (SFunctionNode*)pNode;
1,650✔
734
    if (fmIsTSMASupportedFunc(pFunc->funcId)) {
1,650✔
735
      SFunctionNode* pNewFunc = NULL;
1,650✔
736
      code = fmCreateStateMergeFunc(pFunc, &pNewFunc);
1,650✔
737
      if (code) {
1,650✔
738
        // error
739
        break;
×
740
      } else if (!pNewFunc) {
1,650✔
741
        // no state merge func
742
        continue;
300✔
743
      } else {
744
        REPLACE_NODE(pNewFunc);
1,350✔
745
        nodesDestroyNode(pNode);
1,350✔
746
      }
747
    }
748
  }
749
  return code;
750✔
750
}
751

752
int32_t fmGetFuncId(const char* name) {
8,168,702✔
753
  if (NULL != gFunMgtService.pFuncNameHashTable) {
8,168,702✔
754
    void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, name, strlen(name));
8,168,702✔
755
    if (NULL != pVal) {
8,168,702✔
756
      return *(int32_t*)pVal;
8,168,702✔
757
    }
758
    return -1;
×
759
  }
760
  for (int32_t i = 0; i < funcMgtBuiltinsNum; ++i) {
×
761
    if (0 == strcmp(funcMgtBuiltins[i].name, name)) {
×
762
      return i;
×
763
    }
764
  }
765
  return -1;
×
766
}
767

768
int32_t fmGetTwstartFuncId() {
1,743,794✔
769
  static int32_t twstartFuncId = -2;
770
  if (twstartFuncId < 0) {
1,743,794✔
771
    twstartFuncId = fmGetFuncId("_twstart");
1,555,625✔
772
  }
773
  return twstartFuncId;
1,743,794✔
774
}
775

776
int32_t fmGetTwendFuncId() {
1,588,466✔
777
  static int32_t twendFuncId = -2;
778
  if (twendFuncId < 0) {
1,588,466✔
779
    twendFuncId = fmGetFuncId("_twend");
1,555,625✔
780
  }
781
  return twendFuncId;
1,588,466✔
782
}
783

784
int32_t fmGetTwdurationFuncId() {
1,556,050✔
785
  static int32_t twdurationFuncId = -2;
786
  if (twdurationFuncId < 0) {
1,556,050✔
787
    twdurationFuncId = fmGetFuncId("_twduration");
1,555,625✔
788
  }
789
  return twdurationFuncId;
1,556,050✔
790
}
791

792
int32_t fmGetExternalWindowColumnFuncId() {
1,600,576✔
793
  static int32_t externalWindowColumnFuncId = -2;
794
  if (externalWindowColumnFuncId < 0) {
1,600,576✔
795
    externalWindowColumnFuncId = fmGetFuncId("_external_window_column");
1,555,625✔
796
  }
797
  return externalWindowColumnFuncId;
1,600,576✔
798
}
799

800
bool fmIsMyStateFunc(int32_t funcId, int32_t stateFuncId) {
2,134,200✔
801
  const SBuiltinFuncDefinition* pFunc = &funcMgtBuiltins[funcId];
2,134,200✔
802
  const SBuiltinFuncDefinition* pStateFunc = &funcMgtBuiltins[stateFuncId];
2,134,200✔
803
  if (!pFunc->pStateFunc) {
2,134,200✔
804
    return false;
×
805
  }
806
  if (strcmp(pFunc->pStateFunc, pStateFunc->name) == 0) return true;
2,134,200✔
807
  int32_t stateMergeFuncId = fmGetFuncId(pFunc->pStateFunc);
775,350✔
808
  if (stateMergeFuncId == -1) {
775,350✔
809
    return false;
×
810
  }
811
  const SBuiltinFuncDefinition* pStateMergeFunc = &funcMgtBuiltins[stateMergeFuncId];
775,350✔
812
  return strcmp(pStateFunc->name, pStateMergeFunc->pMergeFunc) == 0;
775,350✔
813
}
814

815
bool fmIsCountLikeFunc(int32_t funcId) {
610,980,934✔
816
  return isSpecificClassifyFunc(funcId, FUNC_MGT_COUNT_LIKE_FUNC);
610,980,934✔
817
}
818

819
bool fmIsRowTsOriginFunc(int32_t funcId) {
6,072,969✔
820
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
6,072,969✔
821
    return false;
×
822
  }
823
  return FUNCTION_TYPE_IROWTS_ORIGIN == funcMgtBuiltins[funcId].type;
6,073,503✔
824
}
825

826
bool fmIsGroupIdFunc(int32_t funcId) {
2,375,200✔
827
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
2,375,200✔
828
    return false;
×
829
  }
830
  return FUNCTION_TYPE_GROUP_ID == funcMgtBuiltins[funcId].type;
2,375,200✔
831
}
832

833
const void* fmGetExternalWindowColumnFuncVal(const SStreamRuntimeFuncInfo* pStreamRuntimeFuncInfo, int32_t index) {
10,543,244✔
834
  SSTriggerCalcParam *pParams = taosArrayGet(pStreamRuntimeFuncInfo->pStreamPesudoFuncVals, pStreamRuntimeFuncInfo->curIdx);
10,543,244✔
835
  if (pParams == NULL || pParams->pExternalWindowData == NULL || index < 0) {
10,549,454✔
836
    return NULL;
×
837
  }
838
  return taosArrayGet(pParams->pExternalWindowData, index);
10,549,454✔
839
}
840

841
const void* fmGetStreamPesudoFuncVal(int32_t funcId, const SStreamRuntimeFuncInfo* pStreamRuntimeFuncInfo) {
267,117,674✔
842
  switch (funcMgtBuiltins[funcId].type) {
267,117,674✔
843
    case FUNCTION_TYPE_TGRPID:
136,830✔
844
      return &pStreamRuntimeFuncInfo->groupId;
136,830✔
845
    case FUNCTION_TYPE_PLACEHOLDER_COLUMN:
857,696✔
846
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
857,696✔
847
    case FUNCTION_TYPE_PLACEHOLDER_TBNAME:
2,168,291✔
848
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
2,168,291✔
849
    default:
263,986,968✔
850
      break;
263,986,968✔
851
  }
852

853
  SSTriggerCalcParam *pParams = taosArrayGet(pStreamRuntimeFuncInfo->pStreamPesudoFuncVals, pStreamRuntimeFuncInfo->curIdx);
263,986,968✔
854
  switch (funcMgtBuiltins[funcId].type) {
263,999,489✔
855
    case FUNCTION_TYPE_TPREV_TS:
×
856
      return &pParams->prevTs;
×
857
    case FUNCTION_TYPE_TCURRENT_TS:
2,390,220✔
858
      return &pParams->currentTs;
2,390,220✔
859
    case FUNCTION_TYPE_TNEXT_TS:
×
860
      return &pParams->nextTs;
×
861
    case FUNCTION_TYPE_TWSTART:
209,285,353✔
862
      return &pParams->wstart;
209,285,353✔
863
    case FUNCTION_TYPE_TWEND:
17,969,462✔
864
      return &pParams->wend;
17,969,462✔
865
    case FUNCTION_TYPE_TWDURATION:
1,077,339✔
866
      return &pParams->wduration;
1,077,339✔
867
    case FUNCTION_TYPE_TWROWNUM:
31,422,408✔
868
      return &pParams->wrownum;
31,422,408✔
869
    case FUNCTION_TYPE_TPREV_LOCALTIME:
355,410✔
870
      return &pParams->prevLocalTime;
355,410✔
871
    case FUNCTION_TYPE_TLOCALTIME:
939,655✔
872
      return &pParams->triggerTime;
939,655✔
873
    case FUNCTION_TYPE_TNEXT_LOCALTIME:
544,459✔
874
      return &pParams->nextLocalTime;
544,459✔
875
    case FUNCTION_TYPE_TGRPID:
×
876
      return &pStreamRuntimeFuncInfo->groupId;
×
877
    case FUNCTION_TYPE_PLACEHOLDER_COLUMN:
×
878
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
×
879
    case FUNCTION_TYPE_PLACEHOLDER_TBNAME:
×
880
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
×
881
    case FUNCTION_TYPE_TIDLESTART:
6,264✔
882
      return &pParams->idlestart;
6,264✔
883
    case FUNCTION_TYPE_TIDLEEND:
6,264✔
884
      return &pParams->idleend;
6,264✔
885
    default:
×
886
      break;
×
887
  }
888
  return NULL;
×
889
}
890

891
bool fmIsStreamPesudoColVal(int32_t funcId) {
537,161✔
892
  return funcMgtBuiltins[funcId].type == FUNCTION_TYPE_PLACEHOLDER_COLUMN
537,161✔
893
         || funcMgtBuiltins[funcId].type == FUNCTION_TYPE_PLACEHOLDER_TBNAME;
537,161✔
894
}
895

896
int32_t fmGetStreamPesudoFuncEnv(int32_t funcId, SNodeList* pParamNodes, SFuncExecEnv *pEnv) {
7,305,799✔
897
  if (NULL == pParamNodes || pParamNodes->length < 1) {
7,305,799✔
898
    uError("invalid stream pesudo func param list %p", pParamNodes);
414✔
899
    return TSDB_CODE_INTERNAL_ERROR;
×
900
  }
901

902
  int32_t firstParamType = nodeType(nodesListGetNode(pParamNodes, 0));
7,305,385✔
903
  if (QUERY_NODE_VALUE != firstParamType) {
7,305,799✔
904
    uError("invalid stream pesudo func first param type %d", firstParamType);
×
905
    return TSDB_CODE_INTERNAL_ERROR;
×
906
  }
907

908
  SValueNode* pResDesc = (SValueNode*)nodesListGetNode(pParamNodes, 0);
7,305,799✔
909
  pEnv->calcMemSize = pResDesc->node.resType.bytes;
7,305,799✔
910

911
  return TSDB_CODE_SUCCESS;
7,305,520✔
912
}
913

914

915
int32_t fmSetStreamPseudoFuncParamVal(int32_t funcId, SNodeList* pParamNodes, const SStreamRuntimeFuncInfo* pStreamRuntimeInfo) {
277,631,602✔
916
  if (!pStreamRuntimeInfo) {
277,631,602✔
917
    uError("internal error, should have pVals for stream pseudo funcs");
×
918
    return TSDB_CODE_INTERNAL_ERROR;
×
919
  }
920
  int32_t code = 0;
277,631,602✔
921
  SArray *pVals1 = NULL;
277,631,602✔
922
  SNode* pFirstParam = nodesListGetNode(pParamNodes, 0);
277,631,602✔
923
  if (nodeType(pFirstParam) != QUERY_NODE_VALUE) {
277,569,388✔
924
    uError("invalid param node type: %d for func: %d", nodeType(pFirstParam), funcId);
×
925
    return TSDB_CODE_INTERNAL_ERROR;
×
926
  }
927

928
  int32_t t = funcMgtBuiltins[funcId].type;
277,621,136✔
929
  uDebug("set stream pseudo func param val, functype: %d, pStreamRuntimeInfo: %p", t, pStreamRuntimeInfo);
277,697,512✔
930
  if (FUNCTION_TYPE_TGRPID == t) {
277,641,482✔
931
    SValue v = {0};
136,830✔
932
    v.type = TSDB_DATA_TYPE_BIGINT;
136,830✔
933
    v.val = *(int64_t*)fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
136,830✔
934
    
935
    code = nodesSetValueNodeValue((SValueNode*)pFirstParam, VALUE_GET_DATUM(&v, pFirstParam->type));
136,830✔
936
    if (code != 0) {
136,830✔
937
      uError("failed to set value node value: %s", tstrerror(code));
×
938
      return code;
×
939
    }
940
  } else if (FUNCTION_TYPE_PLACEHOLDER_COLUMN == t) {
277,504,652✔
941
    SNode* pSecondParam = nodesListGetNode(pParamNodes, 1);
857,845✔
942
    if (nodeType(pSecondParam) != QUERY_NODE_VALUE) {
857,895✔
943
      uError("invalid param node type: %d for func: %d", nodeType(pSecondParam), funcId);
×
944
      return TSDB_CODE_INTERNAL_ERROR;
×
945
    }
946
    SArray* pVal = (SArray*)fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
857,895✔
947
    int32_t idx = ((SValueNode*)pSecondParam)->datum.i;
857,895✔
948
    if (idx - 1 < 0 || idx - 1 >= taosArrayGetSize(pVal)) {
857,895✔
949
      uError("invalid idx: %d for func: %d, should be in [1, %d]", idx, funcId, (int32_t)taosArrayGetSize(pVal));
×
950
      return TSDB_CODE_INTERNAL_ERROR;
×
951
    }
952
    SStreamGroupValue* pValue = taosArrayGet(pVal, idx - 1);
858,146✔
953
    if (pValue == NULL) {
858,146✔
954
      uError("invalid idx: %d for func: %d, should be in [1, %d]", idx, funcId, (int32_t)taosArrayGetSize(pVal));
×
955
      return TSDB_CODE_INTERNAL_ERROR;
×
956
    }
957
    if (!pValue->isNull){
858,146✔
958
      if (pValue->data.type != ((SValueNode*)pFirstParam)->node.resType.type){
851,505✔
959
        uError("invalid value type: %d for func: %d, should be: %d", pValue->data.type, funcId, ((SValueNode*)pFirstParam)->node.resType.type);
×
960
        return TSDB_CODE_INTERNAL_ERROR;
×
961
      }
962
      if (IS_VAR_DATA_TYPE(((SValueNode*)pFirstParam)->node.resType.type)) {
851,306✔
963
        char* tmp = taosMemoryCalloc(1, pValue->data.nData + VARSTR_HEADER_SIZE); 
600,518✔
964
        if (tmp == NULL) {
600,717✔
965
          return TSDB_CODE_OUT_OF_MEMORY;
×
966
        }
967
        taosMemoryFree(((SValueNode*)pFirstParam)->datum.p);
600,717✔
968
        ((SValueNode*)pFirstParam)->datum.p = tmp;
600,518✔
969
        memcpy(varDataVal(((SValueNode*)pFirstParam)->datum.p), pValue->data.pData, pValue->data.nData);
600,518✔
970
        varDataLen(((SValueNode*)pFirstParam)->datum.p) = pValue->data.nData;
600,518✔
971
      } else {
972
        code = nodesSetValueNodeValue((SValueNode*)pFirstParam, VALUE_GET_DATUM(&pValue->data, pValue->data.type));
250,788✔
973
      }
974
    }
975
    ((SValueNode*)pFirstParam)->isNull = pValue->isNull;
857,748✔
976
  } else if (FUNCTION_TYPE_PLACEHOLDER_TBNAME == t) {
276,646,807✔
977
    SArray* pVal = (SArray*)fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
2,152,627✔
978
    for (int32_t i = 0; i < taosArrayGetSize(pVal); ++i) {
2,582,597✔
979
      SStreamGroupValue* pValue = taosArrayGet(pVal, i);
2,582,318✔
980
      if (pValue != NULL && pValue->isTbname) {
2,582,318✔
981
        taosMemoryFreeClear(((SValueNode*)pFirstParam)->datum.p);
2,153,477✔
982
        ((SValueNode*)pFirstParam)->datum.p = taosMemoryCalloc(pValue->data.nData + VARSTR_HEADER_SIZE, 1);
2,152,909✔
983
        if (NULL == ((SValueNode*)pFirstParam)->datum.p ) {
2,153,198✔
984
          return terrno;
×
985
        }
986

987
        (void)memcpy(varDataVal(((SValueNode*)pFirstParam)->datum.p), pValue->data.pData, pValue->data.nData);
2,152,627✔
988
        varDataLen(((SValueNode*)pFirstParam)->datum.p) = pValue->data.nData;
2,152,630✔
989
        break;
2,152,627✔
990
      }
991
    }
992
  } else if(FUNCTION_TYPE_EXTERNAL_WINDOW_COLUMN == t) {
274,494,180✔
993
    if (NULL == pParamNodes || LIST_LENGTH(pParamNodes) < 2) {
10,544,072✔
UNCOV
994
      uError("invalid stream external window column param list %p, len: %d", pParamNodes,
×
995
             pParamNodes ? LIST_LENGTH(pParamNodes) : 0);
996
      return TSDB_CODE_INTERNAL_ERROR;
×
997
    }
998

999
    SNode* pParamNode = nodesListGetNode(pParamNodes, 1);
10,547,384✔
1000
    if (NULL == pParamNode || QUERY_NODE_VALUE != nodeType(pParamNode)) {
10,546,556✔
1001
      uError("invalid stream external window column param node %p type %d for func: %d", pParamNode,
3,726✔
1002
             pParamNode ? nodeType(pParamNode) : -1, funcId);
1003
      return TSDB_CODE_INTERNAL_ERROR;
×
1004
    }
1005

1006
    const SStreamGroupValue* pVal =
1007
        fmGetExternalWindowColumnFuncVal(pStreamRuntimeInfo, ((SValueNode*)pParamNode)->placeholderNo);
10,544,900✔
1008
    if (!pVal) {
10,549,868✔
1009
      uError("failed to set stream pseudo func param val, NULL val for funcId: %d", funcId);
×
1010
      return TSDB_CODE_INTERNAL_ERROR;
×
1011
    }
1012
    if (pVal->isNull) {
10,549,868✔
1013
      ((SValueNode*)pFirstParam)->isNull = true;
828✔
1014
    } else {
1015
      ((SValueNode*)pFirstParam)->isNull = false;
10,549,040✔
1016
      if (IS_VAR_DATA_TYPE(pVal->data.type)) {
10,548,626✔
1017
        // pVal->data.pData already includes the VARSTR header from colDataGetData,
1018
        // so copy it directly as datum.p expects VARSTR format [header][raw data].
1019
        char* tmp = taosMemoryMalloc(pVal->data.nData);
9,128✔
1020
        if (tmp == NULL) {
9,128✔
1021
          return TSDB_CODE_OUT_OF_MEMORY;
×
1022
        }
1023
        taosMemoryFree(((SValueNode*)pFirstParam)->datum.p);
9,128✔
1024
        ((SValueNode*)pFirstParam)->datum.p = tmp;
9,128✔
1025
        memcpy(tmp, pVal->data.pData, pVal->data.nData);
9,128✔
1026
      } else {
1027
        code = nodesSetValueNodeValue((SValueNode*)pFirstParam, (void*)&pVal->data.val);
10,539,498✔
1028
      }
1029
    }
1030
    if (code != 0) {
10,547,798✔
1031
      uError("failed to set value node value: %s", tstrerror(code));
×
1032
      return code;
×
1033
    }
1034
  } else if (LIST_LENGTH(pParamNodes) == 1) {
527,944,920✔
1035
    // twstart, twend
1036
    const void* pVal = fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
264,001,161✔
1037
    if (!pVal) {
263,975,517✔
1038
      uError("failed to set stream pseudo func param val, NULL val for funcId: %d", funcId);
×
1039
      return TSDB_CODE_INTERNAL_ERROR;
×
1040
    }
1041
    code = nodesSetValueNodeValue((SValueNode*)pFirstParam, (void*)pVal);
263,975,517✔
1042
    if (code != 0) {
264,003,537✔
1043
      uError("failed to set value node value: %s", tstrerror(code));
8,725✔
1044
      return code;
×
1045
    }
1046
  } else {
1047
    uError("invalid placeholder function type: %d", t);
1,519✔
1048
    return TSDB_CODE_INTERNAL_ERROR;
×
1049
  }
1050
  
1051
  return code;
277,671,981✔
1052
}
1053

STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc