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

taosdata / TDengine / #5035

24 Apr 2026 11:25AM UTC coverage: 73.06% (+0.002%) from 73.058%
#5035

push

travis-ci

web-flow
merge: from main to 3.0 branch #35224

merge: from main to 3.0 branch[manual-only]

1344 of 1975 new or added lines in 48 files covered. (68.05%)

14127 existing lines in 142 files now uncovered.

275902 of 377640 relevant lines covered (73.06%)

132208813.58 hits per line

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

82.92
/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,522,042✔
37
  // just to make sure the funcId of these functions are initialized before used.
38
  (void)fmGetTwstartFuncId();
1,522,042✔
39
  (void)fmGetTwendFuncId();
1,522,042✔
40
  (void)fmGetTwdurationFuncId();
1,522,042✔
41
  (void)fmGetExternalWindowColumnFuncId();
1,522,042✔
42
}
1,522,042✔
43

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

52
  for (int32_t i = 0; i < funcMgtBuiltinsNum; ++i) {
353,113,744✔
53
    if (TSDB_CODE_SUCCESS != taosHashPut(gFunMgtService.pFuncNameHashTable, funcMgtBuiltins[i].name,
351,591,702✔
54
                                         strlen(funcMgtBuiltins[i].name), &i, sizeof(int32_t))) {
351,591,702✔
55
      initFunctionCode = terrno;
×
56
      return;
×
57
    }
58
  }
59
  InitSpecialFuncId();
1,522,042✔
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,971,180✔
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,522,042✔
75
  (void)taosThreadOnce(&functionHashTableInit, doInitFunctionTable);
1,522,042✔
76
  return initFunctionCode;
1,522,042✔
77
}
78

79
int32_t fmGetFuncInfo(SFunctionNode* pFunc, char* pMsg, int32_t msgLen) {
851,025,456✔
80
  if (NULL != gFunMgtService.pFuncNameHashTable) {
851,025,456✔
81
    void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, pFunc->functionName, strlen(pFunc->functionName));
850,863,818✔
82
    if (NULL != pVal) {
850,898,780✔
83
      pFunc->funcId = *(int32_t*)pVal;
850,691,483✔
84
      pFunc->funcType = funcMgtBuiltins[pFunc->funcId].type;
850,691,491✔
85
      return funcMgtBuiltins[pFunc->funcId].translateFunc(pFunc, pMsg, msgLen);
850,691,231✔
86
    }
87
    return TSDB_CODE_FUNC_NOT_BUILTIN_FUNTION;
207,297✔
88
  }
89
  for (int32_t i = 0; i < funcMgtBuiltinsNum; ++i) {
3,284,436✔
90
    if (0 == strcmp(funcMgtBuiltins[i].name, pFunc->functionName)) {
3,284,384✔
91
      pFunc->funcId = i;
176,512✔
92
      pFunc->funcType = funcMgtBuiltins[pFunc->funcId].type;
176,512✔
93
      return funcMgtBuiltins[pFunc->funcId].translateFunc(pFunc, pMsg, msgLen);
176,512✔
94
    }
95
  }
96
  return TSDB_CODE_FUNC_NOT_BUILTIN_FUNTION;
52✔
97
}
98

99
EFuncReturnRows fmGetFuncReturnRows(SFunctionNode* pFunc) {
18,164,068✔
100
  if (NULL != funcMgtBuiltins[pFunc->funcId].estimateReturnRowsFunc) {
18,164,068✔
101
    return funcMgtBuiltins[pFunc->funcId].estimateReturnRowsFunc(pFunc);
12,818,420✔
102
  }
103
  return (fmIsIndefiniteRowsFunc(pFunc->funcId) || fmIsMultiRowsFunc(pFunc->funcId)) ? FUNC_RETURN_ROWS_INDEFINITE
5,345,124✔
104
                                                                                     : FUNC_RETURN_ROWS_NORMAL;
5,345,124✔
105
}
106

107
bool canCoexistIndefiniteRowsFunc(int32_t funcId1, int32_t funcId2) {
182,584✔
108
  if (fmIsUserDefinedFunc(funcId1) || funcId1 < 0 || funcId1 >= funcMgtBuiltinsNum ||
365,168✔
109
      fmIsUserDefinedFunc(funcId2) || funcId2 < 0 || funcId2 >= funcMgtBuiltinsNum) {
365,168✔
110
    return false;
×
111
  }
112
  if (funcId1 == funcId2) {
182,584✔
113
    return true;
166,790✔
114
  }
115
  if ((funcMgtBuiltins[funcId1].type == FUNCTION_TYPE_LAG || funcMgtBuiltins[funcId1].type == FUNCTION_TYPE_LEAD) &&
15,794✔
116
      (funcMgtBuiltins[funcId2].type == FUNCTION_TYPE_LAG || funcMgtBuiltins[funcId2].type == FUNCTION_TYPE_LEAD)) {
15,342✔
117
    return true;
15,342✔
118
  }
119
  return false;
452✔
120
}
121

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

126
EFunctionType fmGetFuncType(const char* pFunc) {
619,862,417✔
127
  void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, pFunc, strlen(pFunc));
619,862,417✔
128
  if (NULL != pVal) {
619,881,949✔
129
    return funcMgtBuiltins[*(int32_t*)pVal].type;
619,671,350✔
130
  }
131
  return FUNCTION_TYPE_UDF;
210,599✔
132
}
133

134
EFunctionType fmGetFuncTypeFromId(int32_t funcId) {
23,802,517✔
135
  if (funcId >= 0 && funcId < funcMgtBuiltinsNum) {
23,802,517✔
136
    return funcMgtBuiltins[funcId].type;
23,802,360✔
137
  }
138
  return FUNCTION_TYPE_UDF;
157✔
139
}
140

141
EFuncDataRequired fmFuncDataRequired(SFunctionNode* pFunc, STimeWindow* pTimeWindow) {
41,719,553✔
142
  if (fmIsUserDefinedFunc(pFunc->funcId) || pFunc->funcId < 0 || pFunc->funcId >= funcMgtBuiltinsNum) {
41,719,553✔
UNCOV
143
    return FUNC_DATA_REQUIRED_DATA_LOAD;
×
144
  }
145
  if (NULL == funcMgtBuiltins[pFunc->funcId].dataRequiredFunc) {
41,726,626✔
146
    return FUNC_DATA_REQUIRED_DATA_LOAD;
×
147
  }
148
  return funcMgtBuiltins[pFunc->funcId].dataRequiredFunc(pFunc, pTimeWindow);
41,726,482✔
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;
242,629✔
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;
1,065,579✔
159
    ;
160
  }
161

162
  if (funcMgtBuiltins[funcId].dynDataRequiredFunc == NULL) {
2,147,483,647✔
163
    return FUNC_DATA_REQUIRED_DATA_LOAD;
172,959,247✔
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) {
335,196,936✔
170
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
335,196,936✔
171
    return TSDB_CODE_FAILED;
196✔
172
  }
173
  pFpSet->getEnv = funcMgtBuiltins[funcId].getEnvFunc;
335,201,850✔
174
  pFpSet->init = funcMgtBuiltins[funcId].initFunc;
335,127,075✔
175
  pFpSet->process = funcMgtBuiltins[funcId].processFunc;
335,168,221✔
176
  pFpSet->finalize = funcMgtBuiltins[funcId].finalizeFunc;
335,147,017✔
177
  pFpSet->combine = funcMgtBuiltins[funcId].combineFunc;
335,137,745✔
178
  pFpSet->processFuncByRow = funcMgtBuiltins[funcId].processFuncByRow;
335,137,464✔
179
  pFpSet->cleanup = funcMgtBuiltins[funcId].cleanupFunc;
335,149,603✔
180
  return TSDB_CODE_SUCCESS;
335,161,457✔
181
}
182

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

198
int32_t fmGetScalarFuncExecFuncs(int32_t funcId, SScalarFuncExecFuncs* pFpSet) {
961,895,314✔
199
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
961,895,314✔
200
    return TSDB_CODE_FAILED;
96,963✔
201
  }
202
  pFpSet->process = funcMgtBuiltins[funcId].sprocessFunc;
962,171,311✔
203
  pFpSet->getEnv = funcMgtBuiltins[funcId].getEnvFunc;
962,134,617✔
204
  return TSDB_CODE_SUCCESS;
962,175,065✔
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,322,462,468✔
212

213
bool fmIsSelectColsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SELECT_COLS_FUNC); }
155,939,704✔
214

215
bool fmIsSelectFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SELECT_FUNC); }
2,002,466,801✔
216

217
bool fmIsTimelineFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_TIMELINE_FUNC); }
1,190,814,333✔
218

219
bool fmIsDateTimeFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_DATETIME_FUNC); }
602,300,958✔
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,511,630,297✔
226

227
bool fmIsWindowClauseFunc(int32_t funcId) { return fmIsAggFunc(funcId) || fmIsWindowPseudoColumnFunc(funcId); }
61,109,546✔
228

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

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

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

235
bool fmIsIndefiniteRowsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INDEFINITE_ROWS_FUNC); }
1,629,876,128✔
236

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

241
bool fmIsDynamicScanOptimizedFunc(int32_t funcId) {
44,999,923✔
242
  return isSpecificClassifyFunc(funcId, FUNC_MGT_DYNAMIC_SCAN_OPTIMIZED);
44,999,923✔
243
}
244

245
bool fmIsMultiResFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_MULTI_RES_FUNC); }
1,227,136,167✔
246

247
bool fmIsRepeatScanFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_REPEAT_SCAN_FUNC); }
1,203,855,066✔
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); }
602,568,673✔
252

253
bool fmIsIntervalInterpoFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INTERVAL_INTERPO_FUNC); }
64,772,894✔
254

255
bool fmIsSystemInfoFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SYSTEM_INFO_FUNC); }
605,635,618✔
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); }
613,350,189✔
260

261
bool fmIsMultiRowsFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_MULTI_ROWS_FUNC); }
1,200,534,041✔
262

263
static bool fmIsKeepOrderFuncId(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_KEEP_ORDER_FUNC); }
727,162,031✔
264

265
bool fmIsKeepOrderFunc(SFunctionNode* pFunc) {
733,654,377✔
266
  if (pFunc->funcType == FUNCTION_TYPE_TOP || pFunc->funcType == FUNCTION_TYPE_BOTTOM ||
733,654,377✔
267
      pFunc->funcType == FUNCTION_TYPE_SAMPLE) {
729,566,047✔
268
    if (pFunc->pParameterList != NULL && pFunc->pParameterList->length >= 2) {
6,480,823✔
269
      SNode* pParam = nodesListGetNode(pFunc->pParameterList, 1);
6,498,640✔
270
      if (pParam != NULL && nodeType(pParam) == QUERY_NODE_VALUE) {
6,498,640✔
271
        SValueNode* pConst = (SValueNode*)pParam;
6,498,640✔
272
        if (pConst->datum.i == 1) {
6,498,640✔
273
          return true;
765,582✔
274
        }
275
      }
276
    }
277
    return false;
5,723,326✔
278
  }
279
  return fmIsKeepOrderFuncId(pFunc->funcId);
727,189,510✔
280
}
281

282
bool fmIsCumulativeFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_CUMULATIVE_FUNC); }
158,467,104✔
283

284
bool fmIsForbidSysTableFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_FORBID_SYSTABLE_FUNC); }
602,492,716✔
285

286
bool fmIsInterpFunc(int32_t funcId) {
1,234,549,219✔
287
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
1,234,549,219✔
288
    return false;
373,202✔
289
  }
290
  return FUNCTION_TYPE_INTERP == funcMgtBuiltins[funcId].type;
1,234,183,879✔
291
}
292

293
bool fmIsInterpPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_INTERP_PC_FUNC); }
1,214,993,178✔
294

295
bool fmIsForecastFunc(int32_t funcId) {
1,184,718,702✔
296
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
1,184,718,702✔
297
    return false;
388,236✔
298
  }
299
  return FUNCTION_TYPE_FORECAST == funcMgtBuiltins[funcId].type;
1,184,333,682✔
300
}
301

302
bool fmIsAnalysisPseudoColumnFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_ANALYTICS_PC_FUNC); }
1,217,914,899✔
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,648,708✔
315
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
71,648,708✔
316
    return false;
1✔
317
  }
318
  return FUNCTION_TYPE_LAST_ROW == funcMgtBuiltins[funcId].type;
71,648,924✔
319
}
320

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

328
bool fmIsNotNullOutputFunc(int32_t funcId) {
502,642,150✔
329
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
502,642,150✔
330
    return false;
90,685✔
331
  }
332
  return FUNCTION_TYPE_LAST == funcMgtBuiltins[funcId].type ||
502,557,679✔
333
         FUNCTION_TYPE_CACHE_LAST == funcMgtBuiltins[funcId].type ||
483,314,647✔
334
         FUNCTION_TYPE_LAST_PARTIAL == funcMgtBuiltins[funcId].type ||
482,004,831✔
335
         FUNCTION_TYPE_LAST_MERGE == funcMgtBuiltins[funcId].type ||
473,005,269✔
336
         FUNCTION_TYPE_FIRST == funcMgtBuiltins[funcId].type ||
469,389,377✔
337
         FUNCTION_TYPE_FIRST_PARTIAL == funcMgtBuiltins[funcId].type ||
456,024,064✔
338
         FUNCTION_TYPE_FIRST_MERGE == funcMgtBuiltins[funcId].type ||
450,794,799✔
339
         FUNCTION_TYPE_COUNT == funcMgtBuiltins[funcId].type ||
448,693,400✔
340
         FUNCTION_TYPE_HYPERLOGLOG == funcMgtBuiltins[funcId].type ||
393,984,331✔
341
         FUNCTION_TYPE_HYPERLOGLOG_PARTIAL == funcMgtBuiltins[funcId].type ||
1,375,998,683✔
342
         FUNCTION_TYPE_HYPERLOGLOG_MERGE == funcMgtBuiltins[funcId].type;
390,233,546✔
343
}
344

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

352
bool fmIsGroupKeyFunc(int32_t funcId) {
626,058,996✔
353
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
626,058,996✔
UNCOV
354
    return false;
×
355
  }
356
  return FUNCTION_TYPE_GROUP_KEY == funcMgtBuiltins[funcId].type;
626,062,472✔
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) {
374,295,435✔
367
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
374,295,435✔
368
    return false;
×
369
  }
370
  return FUNCTION_TYPE_GROUP_CONST_VALUE == funcMgtBuiltins[funcId].type;
374,295,435✔
371
}
372

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

380
bool fmIsBlockDistFunc(int32_t funcId) {
602,297,755✔
381
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
602,297,755✔
382
    return false;
176,550✔
383
  }
384
  return FUNCTION_TYPE_BLOCK_DIST == funcMgtBuiltins[funcId].type;
602,128,428✔
385
}
386

387
bool fmIsDBUsageFunc(int32_t funcId) {
602,300,276✔
388
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
602,300,276✔
389
    return false;
183,930✔
390
  }
391
  return FUNCTION_TYPE_DB_USAGE == funcMgtBuiltins[funcId].type;
602,120,947✔
392
}
393

394
bool fmIsProcessByRowFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_PROCESS_BY_ROW); }
825,378,894✔
395

396
bool fmIsVolatileFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_VOLATILE_FUNC); }
892,774,091✔
397

398
bool fmIsNoPushdownFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_NO_PUSHDOWN_FUNC); }
49,151,664✔
399

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

402
void fmFuncMgtDestroy() {
1,522,090✔
403
  void* m = gFunMgtService.pFuncNameHashTable;
1,522,090✔
404
  if (m != NULL && atomic_val_compare_exchange_ptr((void**)&gFunMgtService.pFuncNameHashTable, m, 0) == m) {
1,522,090✔
405
    taosHashCleanup(m);
1,522,042✔
406
  }
407
}
1,522,090✔
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,880,970✔
421
  bool res = false;
49,880,970✔
422
  switch (funcMgtBuiltins[funcId].type) {
49,880,970✔
423
    case FUNCTION_TYPE_MAX:
19,843,567✔
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,843,567✔
433
      break;
19,843,567✔
434
    default:
30,037,359✔
435
      break;
30,037,359✔
436
  }
437
  return res;
49,880,926✔
438
}
439

440
bool fmIsConstantResFunc(SFunctionNode* pFunc) {
15,149,041✔
441
  SNode* pNode;
442
  FOREACH(pNode, pFunc->pParameterList) {
15,236,245✔
443
    if (nodeType(pNode) != QUERY_NODE_VALUE) {
199,088✔
444
      return false;
111,884✔
445
    }
446
  }
447
  return true;
15,037,157✔
448
}
449

450
bool fmIsSkipScanCheckFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_SKIP_SCAN_CHECK_FUNC); }
44,993,409✔
451

452
bool fmIsPrimaryKeyFunc(int32_t funcId) { return isSpecificClassifyFunc(funcId, FUNC_MGT_PRIMARY_KEY_FUNC); }
526,219,259✔
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) {
668,559✔
457
  if (funcId < 0 || funcId >= funcMgtBuiltinsNum) {
668,559✔
458
    return false;
×
459
  }
460
  return (funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WSTART || funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WEND ||
1,121,170✔
461
          funcMgtBuiltins[funcId].type == FUNCTION_TYPE_WDURATION);
452,611✔
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) {
159,470,743✔
471
  char msg[128] = {0};
159,470,743✔
472
  return fmGetFuncInfo(pFunc, msg, sizeof(msg));
159,470,743✔
473
}
474

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

492
static void resetOutputChangedFunc(SFunctionNode *pFunc, const SFunctionNode* pSrcFunc) {
149,217,153✔
493
  if (funcMgtBuiltins[pFunc->funcId].type == FUNCTION_TYPE_LAST_MERGE) {
149,217,153✔
494
    pFunc->node.resType = pSrcFunc->node.resType;
23,574,768✔
495
    return;
23,574,722✔
496
  }
497
}
498

499
int32_t createFunctionWithSrcFunc(const char* pName, const SFunctionNode* pSrcFunc, SNodeList* pParameterList, SFunctionNode** ppFunc) {
149,217,206✔
500
  int32_t code = nodesMakeNode(QUERY_NODE_FUNCTION, (SNode**)ppFunc);
149,217,206✔
501
  if (NULL == *ppFunc) {
149,223,517✔
502
    return code;
×
503
  }
504

505
  (*ppFunc)->hasPk = pSrcFunc->hasPk;
149,223,517✔
506
  (*ppFunc)->pkBytes = pSrcFunc->pkBytes;
149,223,473✔
507
  (*ppFunc)->pSrcFuncRef = pSrcFunc;
149,223,473✔
508

509
  (void)snprintf((*ppFunc)->functionName, sizeof((*ppFunc)->functionName), "%s", pName);
149,223,517✔
510
  (*ppFunc)->pParameterList = pParameterList;
149,223,517✔
511
  code = getFuncInfo((*ppFunc));
149,223,473✔
512
  if (TSDB_CODE_SUCCESS != code) {
149,219,398✔
513
    (*ppFunc)->pParameterList = NULL;
×
514
    nodesDestroyNode((SNode*)*ppFunc);
×
515
    *ppFunc = NULL;
×
516
    return code;
×
517
  }
518
  resetOutputChangedFunc(*ppFunc, pSrcFunc);
149,219,398✔
519
  (*ppFunc)->node.relatedTo = pSrcFunc->node.relatedTo;
149,219,127✔
520
  (*ppFunc)->node.bindExprID = pSrcFunc->node.bindExprID;
149,219,219✔
521
  return code;
149,219,219✔
522
}
523

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

534
bool fmIsDistExecFunc(int32_t funcId) {
379,445,693✔
535
  if (fmIsUserDefinedFunc(funcId)) {
379,445,693✔
536
    return false;
106,267✔
537
  }
538
  if (!fmIsVectorFunc(funcId)) {
379,371,800✔
539
    return true;
482,986✔
540
  }
541
  return (NULL != funcMgtBuiltins[funcId].pPartialFunc && NULL != funcMgtBuiltins[funcId].pMergeFunc);
378,898,373✔
542
}
543

544
static int32_t createPartialFunction(const SFunctionNode* pSrcFunc, SFunctionNode** pPartialFunc) {
49,882,435✔
545
  SNodeList* pParameterList = NULL;
49,882,435✔
546
  int32_t    code = nodesCloneList(pSrcFunc->pParameterList, &pParameterList);
49,882,435✔
547
  if (NULL == pParameterList) {
49,886,691✔
548
    return code;
×
549
  }
550
  code =
551
      createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pPartialFunc, pSrcFunc, pParameterList, pPartialFunc);
49,886,691✔
552
  if (TSDB_CODE_SUCCESS != code) {
49,883,445✔
553
    nodesDestroyList(pParameterList);
1,638✔
554
    return code;
×
555
  }
556
  (*pPartialFunc)->hasOriginalFunc = true;
49,881,807✔
557
  (*pPartialFunc)->originalFuncId = pSrcFunc->hasOriginalFunc ? pSrcFunc->originalFuncId : pSrcFunc->funcId;
49,881,853✔
558
  char name[TSDB_FUNC_NAME_LEN + TSDB_NAME_DELIMITER_LEN + TSDB_POINTER_PRINT_BYTES + 1] = {0};
49,881,807✔
559

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

568
static int32_t createMergeFuncPara(const SFunctionNode* pSrcFunc, const SFunctionNode* pPartialFunc,
99,329,187✔
569
                                   SNodeList** pParameterList) {
570
  SNode*  pRes = NULL;
99,329,187✔
571
  int32_t code = createColumnByFunc(pPartialFunc, (SColumnNode**)&pRes);
99,329,187✔
572
  if (TSDB_CODE_SUCCESS != code) {
99,333,539✔
573
    return code;
×
574
  }
575
  if (NULL != funcMgtBuiltins[pSrcFunc->funcId].createMergeParaFuc) {
99,333,539✔
576
    return funcMgtBuiltins[pSrcFunc->funcId].createMergeParaFuc(pSrcFunc->pParameterList, pRes, pParameterList);
846,414✔
577
  } else {
578
    return nodesListMakeStrictAppend(pParameterList, pRes);
98,487,081✔
579
  }
580
}
581

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

587
  int32_t code = createMergeFuncPara(pSrcFunc, pPartialFunc, &pParameterList);
49,443,397✔
588
  if (TSDB_CODE_SUCCESS == code) {
49,449,771✔
589
    if(funcMgtBuiltins[pSrcFunc->funcId].pMiddleFunc != NULL){
49,449,771✔
590
      code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMiddleFunc, pSrcFunc, pParameterList, &pFunc);
5,582,931✔
591
    }else{
592
      code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMergeFunc, pSrcFunc, pParameterList, &pFunc);
43,866,840✔
593
    }
594
  }
595
  if (TSDB_CODE_SUCCESS == code) {
49,449,763✔
596
    tstrncpy(pFunc->node.aliasName, pPartialFunc->node.aliasName, TSDB_COL_NAME_LEN);
49,449,841✔
597
  }
598

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

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

612
  int32_t code = createMergeFuncPara(pSrcFunc, pPartialFunc, &pParameterList);
49,886,626✔
613
  if (TSDB_CODE_SUCCESS == code) {
49,886,816✔
614
    code = createFunctionWithSrcFunc(funcMgtBuiltins[pSrcFunc->funcId].pMergeFunc, pSrcFunc, pParameterList, &pFunc);
49,886,816✔
615
  }
616
  if (TSDB_CODE_SUCCESS == code) {
49,890,551✔
617
    pFunc->hasOriginalFunc = true;
49,886,357✔
618
    pFunc->originalFuncId = pSrcFunc->hasOriginalFunc ? pSrcFunc->originalFuncId : pSrcFunc->funcId;
49,886,357✔
619
    // overwrite function restype set by translate function
620
    if (fmIsSameInOutType(pSrcFunc->funcId)) {
49,886,357✔
621
      pFunc->node.resType = pSrcFunc->node.resType;
19,843,859✔
622
    }
623
    tstrncpy(pFunc->node.aliasName, pSrcFunc->node.aliasName, TSDB_COL_NAME_LEN);
49,881,479✔
624
  }
625

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

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

640
  int32_t code = createPartialFunction(pFunc, pPartialFunc);
49,886,274✔
641
  if (TSDB_CODE_SUCCESS == code && pMidFunc) {
49,884,027✔
642
    code = createMidFunction(pFunc, *pPartialFunc, pMidFunc);
49,446,502✔
643
  }
644
  if (TSDB_CODE_SUCCESS == code) {
49,887,904✔
645
    code = createMergeFunction(pFunc, *pPartialFunc, pMergeFunc);
49,886,770✔
646
  }
647

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

654
  return code;
49,886,559✔
655
}
656

657
const char* fmGetFuncName(int32_t funcId) {
345,866✔
658
  if (fmIsUserDefinedFunc(funcId) || funcId < 0 || funcId >= funcMgtBuiltinsNum) {
345,866✔
659
    return "invalid function";
×
660
  }
661
  return funcMgtBuiltins[funcId].name;
345,866✔
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) {
6,760✔
667
  if (funcMgtBuiltins[pFunc->funcId].pStateFunc) {
6,760✔
668
    SNodeList* pParams = NULL;
6,760✔
669
    int32_t    code = nodesCloneList(pFunc->pParameterList, &pParams);
6,760✔
670
    if (!pParams) return code;
6,760✔
671
    code = createFunction(funcMgtBuiltins[pFunc->funcId].pStateFunc, pParams, pStateFunc);
6,760✔
672
    if (TSDB_CODE_SUCCESS != code) {
6,760✔
673
      nodesDestroyList(pParams);
×
674
      return code;
×
675
    }
676
    tstrncpy((*pStateFunc)->node.aliasName, pFunc->node.aliasName, TSDB_COL_NAME_LEN);
6,760✔
677
    tstrncpy((*pStateFunc)->node.userAlias, pFunc->node.userAlias, TSDB_COL_NAME_LEN);
6,760✔
678
  }
679
  return TSDB_CODE_SUCCESS;
6,760✔
680
}
681

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

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

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

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

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

752
int32_t fmGetFuncId(const char* name) {
7,938,360✔
753
  if (NULL != gFunMgtService.pFuncNameHashTable) {
7,938,360✔
754
    void* pVal = taosHashGet(gFunMgtService.pFuncNameHashTable, name, strlen(name));
7,938,360✔
755
    if (NULL != pVal) {
7,938,360✔
756
      return *(int32_t*)pVal;
7,938,360✔
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,704,973✔
769
  static int32_t twstartFuncId = -2;
770
  if (twstartFuncId < 0) {
1,704,973✔
771
    twstartFuncId = fmGetFuncId("_twstart");
1,522,042✔
772
  }
773
  return twstartFuncId;
1,704,973✔
774
}
775

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

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

792
int32_t fmGetExternalWindowColumnFuncId() {
1,566,471✔
793
  static int32_t externalWindowColumnFuncId = -2;
794
  if (externalWindowColumnFuncId < 0) {
1,566,471✔
795
    externalWindowColumnFuncId = fmGetFuncId("_external_window_column");
1,522,042✔
796
  }
797
  return externalWindowColumnFuncId;
1,566,471✔
798
}
799

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

815
bool fmIsCountLikeFunc(int32_t funcId) {
586,334,113✔
816
  return isSpecificClassifyFunc(funcId, FUNC_MGT_COUNT_LIKE_FUNC);
586,334,113✔
817
}
818

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

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

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

841
const void* fmGetStreamPesudoFuncVal(int32_t funcId, const SStreamRuntimeFuncInfo* pStreamRuntimeFuncInfo) {
394,907,079✔
842
  switch (funcMgtBuiltins[funcId].type) {
394,907,079✔
843
    case FUNCTION_TYPE_TGRPID:
139,197✔
844
      return &pStreamRuntimeFuncInfo->groupId;
139,197✔
845
    case FUNCTION_TYPE_PLACEHOLDER_COLUMN:
887,357✔
846
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
887,357✔
847
    case FUNCTION_TYPE_PLACEHOLDER_TBNAME:
2,171,392✔
848
      return pStreamRuntimeFuncInfo->pStreamPartColVals;
2,171,392✔
849
    default:
391,795,106✔
850
      break;
391,795,106✔
851
  }
852

853
  SSTriggerCalcParam *pParams = taosArrayGet(pStreamRuntimeFuncInfo->pStreamPesudoFuncVals, pStreamRuntimeFuncInfo->curIdx);
391,795,106✔
854
  switch (funcMgtBuiltins[funcId].type) {
391,875,615✔
855
    case FUNCTION_TYPE_TPREV_TS:
×
856
      return &pParams->prevTs;
×
857
    case FUNCTION_TYPE_TCURRENT_TS:
2,530,530✔
858
      return &pParams->currentTs;
2,530,530✔
859
    case FUNCTION_TYPE_TNEXT_TS:
×
860
      return &pParams->nextTs;
×
861
    case FUNCTION_TYPE_TWSTART:
338,057,779✔
862
      return &pParams->wstart;
338,057,779✔
863
    case FUNCTION_TYPE_TWEND:
17,290,609✔
864
      return &pParams->wend;
17,290,609✔
865
    case FUNCTION_TYPE_TWDURATION:
959,815✔
866
      return &pParams->wduration;
959,815✔
867
    case FUNCTION_TYPE_TWROWNUM:
31,542,073✔
868
      return &pParams->wrownum;
31,542,073✔
869
    case FUNCTION_TYPE_TPREV_LOCALTIME:
281,822✔
870
      return &pParams->prevLocalTime;
281,822✔
871
    case FUNCTION_TYPE_TLOCALTIME:
756,780✔
872
      return &pParams->triggerTime;
756,780✔
873
    case FUNCTION_TYPE_TNEXT_LOCALTIME:
433,118✔
874
      return &pParams->nextLocalTime;
433,118✔
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,102✔
882
      return &pParams->idlestart;
6,102✔
883
    case FUNCTION_TYPE_TIDLEEND:
6,102✔
884
      return &pParams->idleend;
6,102✔
885
    default:
×
886
      break;
×
887
  }
888
  return NULL;
×
889
}
890

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

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

902
  int32_t firstParamType = nodeType(nodesListGetNode(pParamNodes, 0));
7,248,494✔
903
  if (QUERY_NODE_VALUE != firstParamType) {
7,250,314✔
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,250,314✔
909
  pEnv->calcMemSize = pResDesc->node.resType.bytes;
7,249,844✔
910

911
  return TSDB_CODE_SUCCESS;
7,249,361✔
912
}
913

914

915
int32_t fmSetStreamPseudoFuncParamVal(int32_t funcId, SNodeList* pParamNodes, const SStreamRuntimeFuncInfo* pStreamRuntimeInfo) {
405,462,742✔
916
  if (!pStreamRuntimeInfo) {
405,462,742✔
917
    uError("internal error, should have pVals for stream pseudo funcs");
×
918
    return TSDB_CODE_INTERNAL_ERROR;
×
919
  }
920
  int32_t code = 0;
405,462,742✔
921
  SArray *pVals1 = NULL;
405,462,742✔
922
  SNode* pFirstParam = nodesListGetNode(pParamNodes, 0);
405,462,742✔
923
  if (nodeType(pFirstParam) != QUERY_NODE_VALUE) {
405,651,902✔
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;
405,611,454✔
929
  uDebug("set stream pseudo func param val, functype: %d, pStreamRuntimeInfo: %p", t, pStreamRuntimeInfo);
405,654,207✔
930
  if (FUNCTION_TYPE_TGRPID == t) {
405,620,427✔
931
    SValue v = {0};
139,197✔
932
    v.type = TSDB_DATA_TYPE_BIGINT;
139,197✔
933
    v.val = *(int64_t*)fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
139,197✔
934
    
935
    code = nodesSetValueNodeValue((SValueNode*)pFirstParam, VALUE_GET_DATUM(&v, pFirstParam->type));
139,197✔
936
    if (code != 0) {
139,197✔
937
      uError("failed to set value node value: %s", tstrerror(code));
×
938
      return code;
×
939
    }
940
  } else if (FUNCTION_TYPE_PLACEHOLDER_COLUMN == t) {
405,481,230✔
941
    SNode* pSecondParam = nodesListGetNode(pParamNodes, 1);
887,357✔
942
    if (nodeType(pSecondParam) != QUERY_NODE_VALUE) {
887,108✔
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);
887,108✔
947
    int32_t idx = ((SValueNode*)pSecondParam)->datum.i;
887,108✔
948
    if (idx - 1 < 0 || idx - 1 >= taosArrayGetSize(pVal)) {
887,357✔
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);
887,357✔
953
    if (pValue == NULL) {
887,108✔
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){
887,108✔
958
      if (pValue->data.type != ((SValueNode*)pFirstParam)->node.resType.type){
880,637✔
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)) {
880,637✔
963
        char* tmp = taosMemoryCalloc(1, pValue->data.nData + VARSTR_HEADER_SIZE); 
611,996✔
964
        if (tmp == NULL) {
611,996✔
965
          return TSDB_CODE_OUT_OF_MEMORY;
×
966
        }
967
        taosMemoryFree(((SValueNode*)pFirstParam)->datum.p);
611,996✔
968
        ((SValueNode*)pFirstParam)->datum.p = tmp;
611,996✔
969
        memcpy(varDataVal(((SValueNode*)pFirstParam)->datum.p), pValue->data.pData, pValue->data.nData);
611,996✔
970
        varDataLen(((SValueNode*)pFirstParam)->datum.p) = pValue->data.nData;
611,996✔
971
      } else {
972
        code = nodesSetValueNodeValue((SValueNode*)pFirstParam, VALUE_GET_DATUM(&pValue->data, pValue->data.type));
268,641✔
973
      }
974
    }
975
    ((SValueNode*)pFirstParam)->isNull = pValue->isNull;
886,859✔
976
  } else if (FUNCTION_TYPE_PLACEHOLDER_TBNAME == t) {
404,593,873✔
977
    SArray* pVal = (SArray*)fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
2,084,938✔
978
    for (int32_t i = 0; i < taosArrayGetSize(pVal); ++i) {
2,465,955✔
979
      SStreamGroupValue* pValue = taosArrayGet(pVal, i);
2,465,949✔
980
      if (pValue != NULL && pValue->isTbname) {
2,465,387✔
981
        taosMemoryFreeClear(((SValueNode*)pFirstParam)->datum.p);
2,085,789✔
982
        ((SValueNode*)pFirstParam)->datum.p = taosMemoryCalloc(pValue->data.nData + VARSTR_HEADER_SIZE, 1);
2,085,780✔
983
        if (NULL == ((SValueNode*)pFirstParam)->datum.p ) {
2,085,237✔
984
          return terrno;
×
985
        }
986

987
        (void)memcpy(varDataVal(((SValueNode*)pFirstParam)->datum.p), pValue->data.pData, pValue->data.nData);
2,084,948✔
988
        varDataLen(((SValueNode*)pFirstParam)->datum.p) = pValue->data.nData;
2,084,669✔
989
        break;
2,084,390✔
990
      }
991
    }
992
  } else if(FUNCTION_TYPE_EXTERNAL_WINDOW_COLUMN == t) {
402,508,935✔
993
    if (NULL == pParamNodes || LIST_LENGTH(pParamNodes) < 2) {
10,626,870✔
994
      uError("invalid stream external window column param list %p, len: %d", pParamNodes,
12,570✔
995
             pParamNodes ? LIST_LENGTH(pParamNodes) : 0);
996
      return TSDB_CODE_INTERNAL_ERROR;
×
997
    }
998

999
    SNode* pParamNode = nodesListGetNode(pParamNodes, 1);
10,613,881✔
1000
    if (NULL == pParamNode || QUERY_NODE_VALUE != nodeType(pParamNode)) {
10,665,418✔
1001
      uError("invalid stream external window column param node %p type %d for func: %d", pParamNode,
48,185✔
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,618,909✔
1008
    if (!pVal) {
10,672,541✔
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,672,541✔
1013
      ((SValueNode*)pFirstParam)->isNull = true;
838✔
1014
    } else {
1015
      ((SValueNode*)pFirstParam)->isNull = false;
10,671,703✔
1016
      if (IS_VAR_DATA_TYPE(pVal->data.type)) {
10,672,122✔
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);
11,744✔
1020
        if (tmp == NULL) {
9,230✔
1021
          return TSDB_CODE_OUT_OF_MEMORY;
×
1022
        }
1023
        taosMemoryFree(((SValueNode*)pFirstParam)->datum.p);
9,230✔
1024
        ((SValueNode*)pFirstParam)->datum.p = tmp;
9,230✔
1025
        memcpy(tmp, pVal->data.pData, pVal->data.nData);
9,230✔
1026
      } else {
1027
        code = nodesSetValueNodeValue((SValueNode*)pFirstParam, (void*)&pVal->data.val);
10,658,702✔
1028
      }
1029
    }
1030
    if (code != 0) {
10,648,239✔
1031
      uError("failed to set value node value: %s", tstrerror(code));
×
1032
      return code;
×
1033
    }
1034
  } else if (LIST_LENGTH(pParamNodes) == 1) {
783,723,046✔
1035
    // twstart, twend
1036
    const void* pVal = fmGetStreamPesudoFuncVal(funcId, pStreamRuntimeInfo);
391,859,785✔
1037
    if (!pVal) {
391,862,299✔
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);
391,862,299✔
1042
    if (code != 0) {
391,857,721✔
1043
      uError("failed to set value node value: %s", tstrerror(code));
16,740✔
1044
      return code;
×
1045
    }
1046
  } else {
1047
    uError("invalid placeholder function type: %d", t);
142,971✔
1048
    return TSDB_CODE_INTERNAL_ERROR;
×
1049
  }
1050
  
1051
  return code;
405,585,895✔
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