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

taosdata / TDengine / #3640

11 Mar 2025 12:59PM UTC coverage: 38.646% (+2.4%) from 36.248%
#3640

push

travis-ci

web-flow
Merge pull request #30118 from taosdata/wl30

udpate ci workflow

85317 of 295934 branches covered (28.83%)

Branch coverage included in aggregate %.

142965 of 294768 relevant lines covered (48.5%)

2201044.8 hits per line

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

63.33
/source/libs/function/src/detail/tavgfunction.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 "builtinsimpl.h"
17
#include "function.h"
18
#include "tdatablock.h"
19
#include "tfunctionInt.h"
20
#include "tglobal.h"
21

22
#define SET_VAL(_info, numOfElem, res) \
23
  do {                                 \
24
    if ((numOfElem) <= 0) {            \
25
      break;                           \
26
    }                                  \
27
    (_info)->numOfRes = (res);         \
28
  } while (0)
29

30
#define LIST_AVG_N(sumT, T)                                               \
31
  do {                                                                    \
32
    T* plist = (T*)pCol->pData;                                           \
33
    for (int32_t i = start; i < numOfRows + pInput->startRowIndex; ++i) { \
34
      if (colDataIsNull_f(pCol->nullbitmap, i)) {                         \
35
        continue;                                                         \
36
      }                                                                   \
37
                                                                          \
38
      numOfElem += 1;                                                     \
39
      pAvgRes->count -= 1;                                                \
40
      sumT -= plist[i];                                                   \
41
    }                                                                     \
42
  } while (0)
43

44
// define signed number sum with check overflow
45
#define CHECK_OVERFLOW_SUM_SIGNED(out, val)                                      \
46
  if (out->sum.overflow) {                                                       \
47
    out->sum.dsum += val;                                                        \
48
  } else if (out->sum.isum > 0 && val > 0 && INT64_MAX - out->sum.isum <= val || \
49
             out->sum.isum < 0 && val < 0 && INT64_MIN - out->sum.isum >= val) { \
50
    double dsum = (double)out->sum.isum;                                         \
51
    out->sum.overflow = true;                                                    \
52
    out->sum.dsum = dsum + val;                                                  \
53
  } else {                                                                       \
54
    out->sum.isum += val;                                                        \
55
  }
56

57
// val is big than INT64_MAX, val come from merge
58
#define CHECK_OVERFLOW_SUM_SIGNED_BIG(out, val, big)                             \
59
  if (out->sum.overflow) {                                                       \
60
    out->sum.dsum += val;                                                        \
61
  } else if (out->sum.isum > 0 && val > 0 && INT64_MAX - out->sum.isum <= val || \
62
             out->sum.isum < 0 && val < 0 && INT64_MIN - out->sum.isum >= val || \
63
             big) {                                                              \
64
    double dsum = (double)out->sum.isum;                                         \
65
    out->sum.overflow = true;                                                    \
66
    out->sum.dsum = dsum + val;                                                  \
67
  } else {                                                                       \
68
    out->sum.isum += val;                                                        \
69
  }
70

71
// define unsigned number sum with check overflow
72
#define CHECK_OVERFLOW_SUM_UNSIGNED(out, val)                 \
73
  if (out->sum.overflow) {                                    \
74
    out->sum.dsum += val;                                     \
75
  } else if (UINT64_MAX - out->sum.usum <= val) {             \
76
    double dsum = (double)out->sum.usum;                      \
77
    out->sum.overflow = true;                                 \
78
    out->sum.dsum = dsum + val;                               \
79
  } else {                                                    \
80
    out->sum.usum += val;                                     \
81
  }
82

83
// val is big than UINT64_MAX, val come from merge
84
#define CHECK_OVERFLOW_SUM_UNSIGNED_BIG(out, val, big)        \
85
  if (out->sum.overflow) {                                    \
86
    out->sum.dsum += val;                                     \
87
  } else if (UINT64_MAX - out->sum.usum <= val || big) {      \
88
    double dsum = (double)out->sum.usum;                      \
89
    out->sum.overflow = true;                                 \
90
    out->sum.dsum = dsum + val;                               \
91
  } else {                                                    \
92
    out->sum.usum += val;                                     \
93
  }
94

95
int32_t getAvgInfoSize() { return (int32_t)sizeof(SAvgRes); }
18,360,189✔
96

97
bool getAvgFuncEnv(SFunctionNode* UNUSED_PARAM(pFunc), SFuncExecEnv* pEnv) {
74,323✔
98
  pEnv->calcMemSize = sizeof(SAvgRes);
74,323✔
99
  return true;
74,323✔
100
}
101

102
int32_t avgFunctionSetup(SqlFunctionCtx* pCtx, SResultRowEntryInfo* pResultInfo) {
22,266,627✔
103
  if (pResultInfo->initialized) {
22,266,627!
104
    return TSDB_CODE_SUCCESS;
×
105
  }
106
  if (TSDB_CODE_SUCCESS != functionSetup(pCtx, pResultInfo)) {
22,266,627!
107
    return TSDB_CODE_FUNC_SETUP_ERROR;
×
108
  }
109

110
  SAvgRes* pRes = GET_ROWCELL_INTERBUF(pResultInfo);
22,554,549✔
111
  (void)memset(pRes, 0, sizeof(SAvgRes));
22,554,549✔
112
  return TSDB_CODE_SUCCESS;
22,554,549✔
113
}
114

115
static int32_t calculateAvgBySMAInfo(SAvgRes* pRes, int32_t numOfRows, int32_t type, const SColumnDataAgg* pAgg) {
×
116
  int32_t numOfElem = numOfRows - pAgg->numOfNull;
×
117

118
  pRes->count += numOfElem;
×
119
  if (IS_SIGNED_NUMERIC_TYPE(type)) {
×
120
    CHECK_OVERFLOW_SUM_SIGNED(pRes, pAgg->sum);
×
121
  } else if (IS_UNSIGNED_NUMERIC_TYPE(type)) {
×
122
    CHECK_OVERFLOW_SUM_UNSIGNED(pRes, pAgg->sum);
×
123
  } else if (IS_FLOAT_TYPE(type)) {
×
124
    pRes->sum.dsum += GET_DOUBLE_VAL((const char*)&(pAgg->sum));
×
125
  }
126

127
  return numOfElem;
×
128
}
129

130
static int32_t doAddNumericVector(SColumnInfoData* pCol, int32_t type, SInputColumnInfoData *pInput, SAvgRes* pRes) {
17,584,717✔
131
  int32_t start = pInput->startRowIndex;
17,584,717✔
132
  int32_t numOfRows = pInput->numOfRows;
17,584,717✔
133
  int32_t numOfElems = 0;
17,584,717✔
134

135
  switch (type) {
17,584,717!
136
    case TSDB_DATA_TYPE_TINYINT: {
4,695✔
137
      int8_t* plist = (int8_t*)pCol->pData;
4,695✔
138
      for (int32_t i = start; i < numOfRows + start; ++i) {
19,321✔
139
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
14,626✔
140
          continue;
13,149✔
141
        }
142

143
        numOfElems += 1;
1,477✔
144
        pRes->count += 1;
1,477✔
145
        CHECK_OVERFLOW_SUM_SIGNED(pRes, plist[i])
1,477!
146
      }
147

148
      break;
4,695✔
149
    }
150

151
    case TSDB_DATA_TYPE_SMALLINT: {
2,481✔
152
      int16_t* plist = (int16_t*)pCol->pData;
2,481✔
153
      for (int32_t i = start; i < numOfRows + start; ++i) {
20,232✔
154
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
17,751✔
155
          continue;
14,595✔
156
        }
157

158
        numOfElems += 1;
3,156✔
159
        pRes->count += 1;
3,156✔
160
        CHECK_OVERFLOW_SUM_SIGNED(pRes, plist[i])
3,156!
161
      }
162
      break;
2,481✔
163
    }
164

165
    case TSDB_DATA_TYPE_INT: {
100,769✔
166
      int32_t* plist = (int32_t*)pCol->pData;
100,769✔
167
      for (int32_t i = start; i < numOfRows + start; ++i) {
16,137,570✔
168
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
16,036,801✔
169
          continue;
15,372✔
170
        }
171

172
        numOfElems += 1;
16,021,429✔
173
        pRes->count += 1;
16,021,429✔
174
        CHECK_OVERFLOW_SUM_SIGNED(pRes, plist[i])
16,021,429!
175
      }
176

177
      break;
100,769✔
178
    }
179

180
    case TSDB_DATA_TYPE_BIGINT: {
17,471,325✔
181
      int64_t* plist = (int64_t*)pCol->pData;
17,471,325✔
182
      for (int32_t i = start; i < numOfRows + start; ++i) {
89,500,653✔
183
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
72,029,328✔
184
          continue;
64,447,668✔
185
        }
186

187
        numOfElems += 1;
7,581,660✔
188
        pRes->count += 1;
7,581,660✔
189
        CHECK_OVERFLOW_SUM_SIGNED(pRes, plist[i])
7,581,660!
190
      }
191
      break;
17,471,325✔
192
    }
193

194
    case TSDB_DATA_TYPE_UTINYINT: {
×
195
      uint8_t* plist = (uint8_t*)pCol->pData;
×
196
      for (int32_t i = start; i < numOfRows + start; ++i) {
×
197
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
×
198
          continue;
×
199
        }
200

201
        numOfElems += 1;
×
202
        pRes->count += 1;
×
203
        CHECK_OVERFLOW_SUM_UNSIGNED(pRes, plist[i])
×
204
      }
205

206
      break;
×
207
    }
208

209
    case TSDB_DATA_TYPE_USMALLINT: {
×
210
      uint16_t* plist = (uint16_t*)pCol->pData;
×
211
      for (int32_t i = start; i < numOfRows + start; ++i) {
×
212
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
×
213
          continue;
×
214
        }
215

216
        numOfElems += 1;
×
217
        pRes->count += 1;
×
218
        CHECK_OVERFLOW_SUM_UNSIGNED(pRes, plist[i])
×
219
      }
220
      break;
×
221
    }
222

223
    case TSDB_DATA_TYPE_UINT: {
18,403✔
224
      uint32_t* plist = (uint32_t*)pCol->pData;
18,403✔
225
      for (int32_t i = start; i < numOfRows + start; ++i) {
42,815✔
226
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
24,412✔
227
          continue;
8,624✔
228
        }
229

230
        numOfElems += 1;
15,788✔
231
        pRes->count += 1;
15,788✔
232
        CHECK_OVERFLOW_SUM_UNSIGNED(pRes, plist[i])
15,788!
233
      }
234

235
      break;
18,403✔
236
    }
237

238
    case TSDB_DATA_TYPE_UBIGINT: {
×
239
      uint64_t* plist = (uint64_t*)pCol->pData;
×
240
      for (int32_t i = start; i < numOfRows + start; ++i) {
×
241
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
×
242
          continue;
×
243
        }
244

245
        numOfElems += 1;
×
246
        pRes->count += 1;
×
247
        CHECK_OVERFLOW_SUM_UNSIGNED(pRes, plist[i])
×
248
        
249
      }
250
      break;
×
251
    }
252

253
    case TSDB_DATA_TYPE_FLOAT: {
2,373✔
254
      float* plist = (float*)pCol->pData;
2,373✔
255
      for (int32_t i = start; i < numOfRows + start; ++i) {
18,270✔
256
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
15,897✔
257
          continue;
13,869✔
258
        }
259

260
        numOfElems += 1;
2,028✔
261
        pRes->count += 1;
2,028✔
262
        pRes->sum.dsum += plist[i];
2,028✔
263
      }
264
      break;
2,373✔
265
    }
266

267
    case TSDB_DATA_TYPE_DOUBLE: {
2,550✔
268
      double* plist = (double*)pCol->pData;
2,550✔
269
      for (int32_t i = start; i < numOfRows + start; ++i) {
22,389✔
270
        if (colDataIsNull_f(pCol->nullbitmap, i)) {
19,839✔
271
          continue;
16,945✔
272
        }
273

274
        numOfElems += 1;
2,894✔
275
        pRes->count += 1;
2,894✔
276
        pRes->sum.dsum += plist[i];
2,894✔
277
      }
278
      break;
2,550✔
279
    }
280

281
    default:
×
282
      break;
×
283
  }
284

285
  return numOfElems;
17,584,717✔
286
}
287

288
int32_t avgFunction(SqlFunctionCtx* pCtx) {
37,049,517✔
289
  int32_t       numOfElem = 0;
37,049,517✔
290
  const int32_t THRESHOLD_SIZE = 8;
37,049,517✔
291

292
  SInputColumnInfoData* pInput = &pCtx->input;
37,049,517✔
293
  SColumnDataAgg*       pAgg = pInput->pColumnDataAgg[0];
37,049,517✔
294
  int32_t               type = pInput->pData[0]->info.type;
37,049,517✔
295

296
  SAvgRes* pAvgRes = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx));
37,049,517✔
297

298
  // computing based on the true data block
299
  SColumnInfoData* pCol = pInput->pData[0];
37,049,517✔
300

301
  int32_t start = pInput->startRowIndex;
37,049,517✔
302
  int32_t numOfRows = pInput->numOfRows;
37,049,517✔
303

304
  if (IS_NULL_TYPE(type)) {
37,049,517✔
305
    goto _over;
36✔
306
  }
307

308
  pAvgRes->type = type;
37,049,481✔
309

310
  if (pInput->colDataSMAIsSet) {  // try to use SMA if available
37,049,481!
311
    numOfElem = calculateAvgBySMAInfo(pAvgRes, numOfRows, type, pAgg);
×
312
  } else if (!pCol->hasNull) {  // try to employ the simd instructions to speed up the loop
37,049,481✔
313
    numOfElem = pInput->numOfRows;
20,507,673✔
314
    pAvgRes->count += pInput->numOfRows;
20,507,673✔
315

316
    switch(type) {
20,507,673!
317
      case TSDB_DATA_TYPE_UTINYINT:
6,000✔
318
      case TSDB_DATA_TYPE_TINYINT: {
319
        const int8_t* plist = (const int8_t*) pCol->pData;
6,000✔
320

321
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
22,416✔
322
          if (type == TSDB_DATA_TYPE_TINYINT) {
16,416!
323
            CHECK_OVERFLOW_SUM_SIGNED(pAvgRes, plist[i])
16,416!
324
          } else {
325
            CHECK_OVERFLOW_SUM_UNSIGNED(pAvgRes, (uint8_t)plist[i])
×
326
          }
327
        }
328
        break;
6,000✔
329
      }
330

331
      case TSDB_DATA_TYPE_USMALLINT:
1,758✔
332
      case TSDB_DATA_TYPE_SMALLINT: {
333
        const int16_t* plist = (const int16_t*)pCol->pData;
1,758✔
334

335
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
14,658✔
336
          if (type == TSDB_DATA_TYPE_SMALLINT) {
12,900!
337
            CHECK_OVERFLOW_SUM_SIGNED(pAvgRes, plist[i])
12,900!
338
          } else {
339
            CHECK_OVERFLOW_SUM_UNSIGNED(pAvgRes, (uint16_t)plist[i])
×
340
          }
341
        }
342
        break;
1,758✔
343
      }
344

345
      case TSDB_DATA_TYPE_UINT:
4,999,884✔
346
      case TSDB_DATA_TYPE_INT: {
347
        const int32_t* plist = (const int32_t*) pCol->pData;
4,999,884✔
348

349
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
92,574,749✔
350
          if (type == TSDB_DATA_TYPE_INT) {
87,574,865✔
351
            CHECK_OVERFLOW_SUM_SIGNED(pAvgRes, plist[i])
87,526,329!
352
          } else {
353
            CHECK_OVERFLOW_SUM_UNSIGNED(pAvgRes, (uint32_t)plist[i])
48,536!
354
          }
355
        }
356
        break;
4,999,884✔
357
      }
358

359
      case TSDB_DATA_TYPE_UBIGINT:
14,845,378✔
360
      case TSDB_DATA_TYPE_BIGINT: {
361
        const int64_t* plist = (const int64_t*) pCol->pData;
14,845,378✔
362

363
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
69,380,565✔
364
          if (type == TSDB_DATA_TYPE_BIGINT) {
54,535,187!
365
            CHECK_OVERFLOW_SUM_SIGNED(pAvgRes, plist[i])
54,535,187!
366
          } else {
367
            CHECK_OVERFLOW_SUM_UNSIGNED(pAvgRes, (uint64_t)plist[i])
×
368
          }
369
        }
370
        break;
14,845,378✔
371
      }
372

373
      case TSDB_DATA_TYPE_FLOAT: {
658,427✔
374
        const float* plist = (const float*) pCol->pData;
658,427✔
375

376
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
1,823,057✔
377
          pAvgRes->sum.dsum += plist[i];
1,164,630✔
378
        }
379
        break;
658,427✔
380
      }
381
      case TSDB_DATA_TYPE_DOUBLE: {
2,505✔
382
        const double* plist = (const double*)pCol->pData;
2,505✔
383

384
        for (int32_t i = pInput->startRowIndex; i < pInput->numOfRows + pInput->startRowIndex; ++i) {
31,302✔
385
          pAvgRes->sum.dsum += plist[i];
28,797✔
386
        }
387
        break;
2,505✔
388
      }
389
      default:
×
390
        return TSDB_CODE_FUNC_FUNTION_PARA_TYPE;
×
391
    }
392
  } else {
393
    numOfElem = doAddNumericVector(pCol, type, pInput, pAvgRes);
16,541,808✔
394
  }
395

396
_over:
38,237,282✔
397
  // data in the check operation are all null, not output
398
  SET_VAL(GET_RES_INFO(pCtx), numOfElem, 1);
38,237,282✔
399
  return TSDB_CODE_SUCCESS;
38,237,282✔
400
}
401

402
static void avgTransferInfo(SAvgRes* pInput, SAvgRes* pOutput) {
16,115,545✔
403
  if (IS_NULL_TYPE(pInput->type)) {
16,115,545✔
404
    return;
22✔
405
  }
406

407
  pOutput->type = pInput->type;
16,115,523✔
408
  if (IS_SIGNED_NUMERIC_TYPE(pOutput->type)) {
32,227,629!
409
    bool overflow = pInput->sum.overflow;
16,112,106✔
410
    CHECK_OVERFLOW_SUM_SIGNED_BIG(pOutput, (overflow ? pInput->sum.dsum : pInput->sum.isum), overflow);
16,112,106!
411
  } else if (IS_UNSIGNED_NUMERIC_TYPE(pOutput->type)) {
7,544!
412
    bool overflow = pInput->sum.overflow;
4,127✔
413
    CHECK_OVERFLOW_SUM_UNSIGNED_BIG(pOutput, (overflow ? pInput->sum.dsum : pInput->sum.usum), overflow);
4,127!
414
  } else {
415
    pOutput->sum.dsum += pInput->sum.dsum;
×
416
  }
417

418
  pOutput->count += pInput->count;
16,115,523✔
419
}
420

421
int32_t avgFunctionMerge(SqlFunctionCtx* pCtx) {
4,540,809✔
422
  SInputColumnInfoData* pInput = &pCtx->input;
4,540,809✔
423
  SColumnInfoData*      pCol = pInput->pData[0];
4,540,809✔
424

425
  if (IS_NULL_TYPE(pCol->info.type)) {
4,540,809✔
426
    SET_VAL(GET_RES_INFO(pCtx), 0, 1);
23✔
427
    return TSDB_CODE_SUCCESS;
23✔
428
  }
429

430
  if (pCol->info.type != TSDB_DATA_TYPE_BINARY) {
4,540,786!
431
    return TSDB_CODE_FUNC_FUNTION_PARA_TYPE;
×
432
  }
433

434
  SAvgRes* pInfo = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx));
4,540,786✔
435

436
  int32_t start = pInput->startRowIndex;
4,540,786✔
437

438
  for (int32_t i = start; i < start + pInput->numOfRows; ++i) {
20,656,890✔
439
    if(colDataIsNull_s(pCol, i)) continue;
32,229,634!
440
    char*    data = colDataGetData(pCol, i);
16,114,817!
441
    SAvgRes* pInputInfo = (SAvgRes*)varDataVal(data);
16,114,817✔
442
    avgTransferInfo(pInputInfo, pInfo);
16,114,817✔
443
  }
444

445
  SET_VAL(GET_RES_INFO(pCtx), 1, 1);
4,542,073✔
446

447
  return TSDB_CODE_SUCCESS;
4,542,073✔
448
}
449

450
#ifdef BUILD_NO_CALL
451
int32_t avgInvertFunction(SqlFunctionCtx* pCtx) {
452
  int32_t numOfElem = 0;
453

454
  // Only the pre-computing information loaded and actual data does not loaded
455
  SInputColumnInfoData* pInput = &pCtx->input;
456
  SAvgRes* pAvgRes = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx));
457

458
  // computing based on the true data block
459
  SColumnInfoData* pCol = pInput->pData[0];
460

461
  int32_t start = pInput->startRowIndex;
462
  int32_t numOfRows = pInput->numOfRows;
463

464
  switch (pCol->info.type) {
465
    case TSDB_DATA_TYPE_TINYINT: {
466
      LIST_AVG_N(pAvgRes->sum.isum, int8_t);
467
      break;
468
    }
469
    case TSDB_DATA_TYPE_SMALLINT: {
470
      LIST_AVG_N(pAvgRes->sum.isum, int16_t);
471
      break;
472
    }
473
    case TSDB_DATA_TYPE_INT: {
474
      LIST_AVG_N(pAvgRes->sum.isum, int32_t);
475
      break;
476
    }
477
    case TSDB_DATA_TYPE_BIGINT: {
478
      LIST_AVG_N(pAvgRes->sum.isum, int64_t);
479
      break;
480
    }
481
    case TSDB_DATA_TYPE_UTINYINT: {
482
      LIST_AVG_N(pAvgRes->sum.usum, uint8_t);
483
      break;
484
    }
485
    case TSDB_DATA_TYPE_USMALLINT: {
486
      LIST_AVG_N(pAvgRes->sum.usum, uint16_t);
487
      break;
488
    }
489
    case TSDB_DATA_TYPE_UINT: {
490
      LIST_AVG_N(pAvgRes->sum.usum, uint32_t);
491
      break;
492
    }
493
    case TSDB_DATA_TYPE_UBIGINT: {
494
      LIST_AVG_N(pAvgRes->sum.usum, uint64_t);
495
      break;
496
    }
497
    case TSDB_DATA_TYPE_FLOAT: {
498
      LIST_AVG_N(pAvgRes->sum.dsum, float);
499
      break;
500
    }
501
    case TSDB_DATA_TYPE_DOUBLE: {
502
      LIST_AVG_N(pAvgRes->sum.dsum, double);
503
      break;
504
    }
505
    default:
506
      break;
507
  }
508

509
  // data in the check operation are all null, not output
510
  SET_VAL(GET_RES_INFO(pCtx), numOfElem, 1);
511
  return TSDB_CODE_SUCCESS;
512
}
513
#endif
514

515
int32_t avgCombine(SqlFunctionCtx* pDestCtx, SqlFunctionCtx* pSourceCtx) {
×
516
  SResultRowEntryInfo* pDResInfo = GET_RES_INFO(pDestCtx);
×
517
  SAvgRes*             pDBuf = GET_ROWCELL_INTERBUF(pDResInfo);
×
518

519
  SResultRowEntryInfo* pSResInfo = GET_RES_INFO(pSourceCtx);
×
520
  SAvgRes*             pSBuf = GET_ROWCELL_INTERBUF(pSResInfo);
×
521
  int16_t              type = pDBuf->type == TSDB_DATA_TYPE_NULL ? pSBuf->type : pDBuf->type;
×
522

523
  if (IS_SIGNED_NUMERIC_TYPE(type)) {
×
524
    CHECK_OVERFLOW_SUM_SIGNED(pDBuf, pSBuf->sum.isum)
×
525
  } else if (IS_UNSIGNED_NUMERIC_TYPE(type)) {
×
526
    CHECK_OVERFLOW_SUM_UNSIGNED(pDBuf, pSBuf->sum.usum)
×
527
  } else {
528
    pDBuf->sum.dsum += pSBuf->sum.dsum;
×
529
  }
530
  pDBuf->count += pSBuf->count;
×
531

532
  return TSDB_CODE_SUCCESS;
×
533
}
534

535
int32_t avgFinalize(SqlFunctionCtx* pCtx, SSDataBlock* pBlock) {
4,461,595✔
536
  SResultRowEntryInfo* pEntryInfo = GET_RES_INFO(pCtx);
4,461,595✔
537

538
  SAvgRes* pRes = GET_ROWCELL_INTERBUF(pEntryInfo);
4,461,595✔
539
  int32_t  type = pRes->type;
4,461,595✔
540

541
  if (pRes->count > 0) {
4,461,595✔
542
    if(pRes->sum.overflow) {
4,431,955✔
543
      // overflow flag set , use dsum
544
      pRes->result = pRes->sum.dsum / ((double)pRes->count);
241✔
545
    }else if (IS_SIGNED_NUMERIC_TYPE(type)) {
4,431,714!
546
      pRes->result = pRes->sum.isum / ((double)pRes->count);
4,398,670✔
547
    } else if (IS_UNSIGNED_NUMERIC_TYPE(type)) {
33,044!
548
      pRes->result = pRes->sum.usum / ((double)pRes->count);
28,359✔
549
    } else {
550
      pRes->result = pRes->sum.dsum / ((double)pRes->count);
4,685✔
551
    }
552
  }
553

554
  if (pRes->count == 0 || isinf(pRes->result) || isnan(pRes->result)) {
4,461,595!
555
    pEntryInfo->numOfRes = 0;
29,618✔
556
  } else {
557
    pEntryInfo->numOfRes = 1;
4,431,977✔
558
  }
559

560
  return functionFinalize(pCtx, pBlock);
4,461,595✔
561
}
562

563
int32_t avgPartialFinalize(SqlFunctionCtx* pCtx, SSDataBlock* pBlock) {
18,300,258✔
564
  SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx);
18,300,258✔
565
  SAvgRes*             pInfo = GET_ROWCELL_INTERBUF(GET_RES_INFO(pCtx));
18,300,258✔
566
  int32_t              resultBytes = getAvgInfoSize();
18,300,258✔
567
  char*                res = taosMemoryCalloc(resultBytes + VARSTR_HEADER_SIZE, sizeof(char));
18,278,629!
568
  int32_t              code = TSDB_CODE_SUCCESS;
18,634,215✔
569
  if (NULL == res) {
18,634,215!
570
    return terrno;
×
571
  }
572
  (void)memcpy(varDataVal(res), pInfo, resultBytes);
18,634,215✔
573
  varDataSetLen(res, resultBytes);
18,634,215✔
574

575
  int32_t          slotId = pCtx->pExpr->base.resSchema.slotId;
18,634,215✔
576
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId);
18,634,215✔
577
  if(NULL == pCol) {
18,592,630!
578
    code = TSDB_CODE_OUT_OF_RANGE;
×
579
    goto _exit;
×
580
  }
581

582
  code = colDataSetVal(pCol, pBlock->info.rows, res, false);
18,592,630✔
583

584
_exit:
18,335,753✔
585
  taosMemoryFree(res);
18,335,753!
586
  return code;
18,712,643✔
587
}
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