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

taosdata / TDengine / #3910

23 Apr 2025 02:47AM UTC coverage: 62.362% (-0.7%) from 63.063%
#3910

push

travis-ci

web-flow
docs(datain): add missing health status types (#30828)

155061 of 317305 branches covered (48.87%)

Branch coverage included in aggregate %.

240172 of 316469 relevant lines covered (75.89%)

6269478.46 hits per line

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

74.02
/source/libs/executor/src/streamfilloperator.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 "filter.h"
17
#include "os.h"
18
#include "query.h"
19
#include "taosdef.h"
20
#include "tmsg.h"
21
#include "ttypes.h"
22

23
#include "executorInt.h"
24
#include "streamexecutorInt.h"
25
#include "streamsession.h"
26
#include "streaminterval.h"
27
#include "tcommon.h"
28
#include "thash.h"
29
#include "ttime.h"
30

31
#include "function.h"
32
#include "operator.h"
33
#include "querynodes.h"
34
#include "querytask.h"
35
#include "tdatablock.h"
36
#include "tfill.h"
37

38
#define FILL_POS_INVALID 0
39
#define FILL_POS_START   1
40
#define FILL_POS_MID     2
41
#define FILL_POS_END     3
42

43
TSKEY getNextWindowTs(TSKEY ts, SInterval* pInterval) {
40✔
44
  STimeWindow win = {.skey = ts, .ekey = ts};
40✔
45
  getNextTimeWindow(pInterval, &win, TSDB_ORDER_ASC);
40✔
46
  return win.skey;
40✔
47
}
48

49
TSKEY getPrevWindowTs(TSKEY ts, SInterval* pInterval) {
15✔
50
  STimeWindow win = {.skey = ts, .ekey = ts};
15✔
51
  getNextTimeWindow(pInterval, &win, TSDB_ORDER_DESC);
15✔
52
  return win.skey;
15✔
53
}
54

55
int32_t setRowCell(SColumnInfoData* pCol, int32_t rowId, const SResultCellData* pCell) {
391,044✔
56
  return colDataSetVal(pCol, rowId, pCell->pData, pCell->isNull);
391,044✔
57
}
58

59
SResultCellData* getResultCell(SResultRowData* pRaw, int32_t index) {
124,689✔
60
  if (!pRaw || !pRaw->pRowVal) {
124,689!
61
    return NULL;
×
62
  }
63
  char*            pData = (char*)pRaw->pRowVal;
124,699✔
64
  SResultCellData* pCell = pRaw->pRowVal;
124,699✔
65
  for (int32_t i = 0; i < index; i++) {
723,999✔
66
    pData += (pCell->bytes + sizeof(SResultCellData));
599,300✔
67
    pCell = (SResultCellData*)pData;
599,300✔
68
  }
69
  return pCell;
124,699✔
70
}
71

72
void* destroyFillColumnInfo(SFillColInfo* pFillCol, int32_t start, int32_t end) {
495✔
73
  for (int32_t i = start; i < end; i++) {
789✔
74
    destroyExprInfo(pFillCol[i].pExpr, 1);
294✔
75
    taosVariantDestroy(&pFillCol[i].fillVal);
294✔
76
  }
77
  if (start < end) {
495✔
78
    taosMemoryFreeClear(pFillCol[start].pExpr);
253!
79
  }
80
  taosMemoryFree(pFillCol);
495!
81
  return NULL;
495✔
82
}
83

84
void destroyStreamFillSupporter(SStreamFillSupporter* pFillSup) {
495✔
85
  if (pFillSup == NULL) {
495!
86
    return;
×
87
  }
88
  pFillSup->pAllColInfo = destroyFillColumnInfo(pFillSup->pAllColInfo, pFillSup->numOfFillCols, pFillSup->numOfAllCols);
495✔
89
  tSimpleHashCleanup(pFillSup->pResMap);
495✔
90
  pFillSup->pResMap = NULL;
495✔
91
  cleanupExprSupp(&pFillSup->notFillExprSup);
495✔
92
  if (pFillSup->cur.pRowVal != pFillSup->prev.pRowVal && pFillSup->cur.pRowVal != pFillSup->next.pRowVal) {
495✔
93
    taosMemoryFree(pFillSup->cur.pRowVal);
112!
94
  }
95
  taosMemoryFree(pFillSup->prev.pRowVal);
495!
96
  taosMemoryFree(pFillSup->next.pRowVal);
495!
97
  taosMemoryFree(pFillSup->nextNext.pRowVal);
495!
98

99
  taosMemoryFree(pFillSup->pOffsetInfo);
495!
100
  taosArrayDestroy(pFillSup->pResultRange);
495✔
101
  pFillSup->pResultRange = NULL;
495✔
102

103
  taosMemoryFree(pFillSup);
495!
104
}
105

106
void destroySPoint(void* ptr) {
99,658✔
107
  SPoint* point = (SPoint*)ptr;
99,658✔
108
  taosMemoryFreeClear(point->val);
99,658!
109
}
99,658✔
110

111
void destroyStreamFillLinearInfo(SStreamFillLinearInfo* pFillLinear) {
495✔
112
  taosArrayDestroyEx(pFillLinear->pEndPoints, destroySPoint);
495✔
113
  taosArrayDestroyEx(pFillLinear->pNextEndPoints, destroySPoint);
495✔
114
  taosMemoryFree(pFillLinear);
495!
115
}
495✔
116

117
void destroyStreamFillInfo(SStreamFillInfo* pFillInfo) {
495✔
118
  if (pFillInfo == NULL) {
495!
119
    return;
×
120
  } 
121
  if (pFillInfo->type == TSDB_FILL_SET_VALUE || pFillInfo->type == TSDB_FILL_SET_VALUE_F ||
495✔
122
      pFillInfo->type == TSDB_FILL_NULL || pFillInfo->type == TSDB_FILL_NULL_F) {
392✔
123
    taosMemoryFreeClear(pFillInfo->pResRow->pRowVal);
215!
124
    taosMemoryFreeClear(pFillInfo->pResRow);
215!
125
    taosMemoryFreeClear(pFillInfo->pNonFillRow->pRowVal);
215!
126
    taosMemoryFreeClear(pFillInfo->pNonFillRow);
215!
127
  }
128
  destroyStreamFillLinearInfo(pFillInfo->pLinearInfo);
495✔
129
  pFillInfo->pLinearInfo = NULL;
495✔
130

131
  taosArrayDestroy(pFillInfo->delRanges);
495✔
132
  taosMemoryFreeClear(pFillInfo->pTempBuff);
495!
133
  taosMemoryFree(pFillInfo);
495!
134
}
135

136
void clearGroupResArray(SGroupResInfo* pGroupResInfo) {
495✔
137
  pGroupResInfo->freeItem = false;
495✔
138
  taosArrayDestroy(pGroupResInfo->pRows);
495✔
139
  pGroupResInfo->pRows = NULL;
495✔
140
  pGroupResInfo->index = 0;
495✔
141
}
495✔
142

143
void destroyStreamFillOperatorInfo(void* param) {
253✔
144
  SStreamFillOperatorInfo* pInfo = (SStreamFillOperatorInfo*)param;
253✔
145
  destroyStreamFillInfo(pInfo->pFillInfo);
253✔
146
  destroyStreamFillSupporter(pInfo->pFillSup);
253✔
147
  blockDataDestroy(pInfo->pRes);
253✔
148
  pInfo->pRes = NULL;
253✔
149
  blockDataDestroy(pInfo->pSrcBlock);
253✔
150
  pInfo->pSrcBlock = NULL;
253✔
151
  blockDataDestroy(pInfo->pDelRes);
253✔
152
  pInfo->pDelRes = NULL;
253✔
153
  taosArrayDestroy(pInfo->matchInfo.pList);
253✔
154
  pInfo->matchInfo.pList = NULL;
253✔
155
  taosArrayDestroy(pInfo->pUpdated);
253✔
156
  clearGroupResArray(&pInfo->groupResInfo);
253✔
157
  taosArrayDestroy(pInfo->pCloseTs);
253✔
158

159
  if (pInfo->stateStore.streamFileStateDestroy != NULL) {
253✔
160
    pInfo->stateStore.streamFileStateDestroy(pInfo->pState->pFileState);
44✔
161
  }
162

163
  if (pInfo->pState != NULL) {
253✔
164
    taosMemoryFreeClear(pInfo->pState);
44!
165
  }
166
  destroyStreamBasicInfo(&pInfo->basic);
253✔
167
  destroyNonBlockAggSupptor(&pInfo->nbSup);
253✔
168

169
  taosMemoryFree(pInfo);
253!
170
}
253✔
171

172
static void resetFillWindow(SResultRowData* pRowData) {
5,580✔
173
  pRowData->key = INT64_MIN;
5,580✔
174
  taosMemoryFreeClear(pRowData->pRowVal);
5,580!
175
}
5,580✔
176

177
static void resetPrevAndNextWindow(SStreamFillSupporter* pFillSup) {
1,258✔
178
  if (pFillSup->cur.pRowVal != pFillSup->prev.pRowVal && pFillSup->cur.pRowVal != pFillSup->next.pRowVal) {
1,258✔
179
    resetFillWindow(&pFillSup->cur);
962✔
180
  } else {
181
    pFillSup->cur.key = INT64_MIN;
296✔
182
    pFillSup->cur.pRowVal = NULL;
296✔
183
  }
184
  resetFillWindow(&pFillSup->prev);
1,258✔
185
  resetFillWindow(&pFillSup->next);
1,258✔
186
  resetFillWindow(&pFillSup->nextNext);
1,258✔
187
}
1,258✔
188

189
void getWindowFromDiscBuf(SOperatorInfo* pOperator, TSKEY ts, uint64_t groupId, SStreamFillSupporter* pFillSup) {
1,258✔
190
  SStorageAPI* pAPI = &pOperator->pTaskInfo->storageAPI;
1,258✔
191
  void*        pState = pOperator->pTaskInfo->streamInfo.pState;
1,258✔
192
  resetPrevAndNextWindow(pFillSup);
1,258✔
193

194
  SWinKey key = {.ts = ts, .groupId = groupId};
1,258✔
195
  void*   curVal = NULL;
1,258✔
196
  int32_t curVLen = 0;
1,258✔
197
  bool    hasCurKey = true;
1,258✔
198
  int32_t code = pAPI->stateStore.streamStateFillGet(pState, &key, (void**)&curVal, &curVLen, NULL);
1,258✔
199
  if (code == TSDB_CODE_SUCCESS) {
1,258✔
200
    pFillSup->cur.key = key.ts;
1,218✔
201
    pFillSup->cur.pRowVal = curVal;
1,218✔
202
  } else {
203
    qDebug("streamStateFillGet key failed, Data may be deleted. ts:%" PRId64 ", groupId:%" PRId64, ts, groupId);
40!
204
    pFillSup->cur.key = ts;
40✔
205
    pFillSup->cur.pRowVal = NULL;
40✔
206
    hasCurKey = false;
40✔
207
  }
208

209
  SStreamStateCur* pCur = pAPI->stateStore.streamStateFillSeekKeyPrev(pState, &key);
1,258✔
210
  SWinKey          preKey = {.ts = INT64_MIN, .groupId = groupId};
1,258✔
211
  void*            preVal = NULL;
1,258✔
212
  int32_t          preVLen = 0;
1,258✔
213
  code = pAPI->stateStore.streamStateFillGetGroupKVByCur(pCur, &preKey, (const void**)&preVal, &preVLen);
1,258✔
214

215
  if (code == TSDB_CODE_SUCCESS) {
1,258✔
216
    pFillSup->prev.key = preKey.ts;
1,021✔
217
    pFillSup->prev.pRowVal = preVal;
1,021✔
218

219
    if (hasCurKey) {
1,021✔
220
      pAPI->stateStore.streamStateCurNext(pState, pCur);
981✔
221
    }
222

223
    pAPI->stateStore.streamStateCurNext(pState, pCur);
1,021✔
224
  } else {
225
    pAPI->stateStore.streamStateFreeCur(pCur);
237✔
226
    pCur = pAPI->stateStore.streamStateFillSeekKeyNext(pState, &key);
237✔
227
  }
228

229
  SWinKey nextKey = {.ts = INT64_MIN, .groupId = groupId};
1,258✔
230
  void*   nextVal = NULL;
1,258✔
231
  int32_t nextVLen = 0;
1,258✔
232
  code = pAPI->stateStore.streamStateFillGetGroupKVByCur(pCur, &nextKey, (const void**)&nextVal, &nextVLen);
1,258✔
233
  if (code == TSDB_CODE_SUCCESS) {
1,258✔
234
    pFillSup->next.key = nextKey.ts;
716✔
235
    pFillSup->next.pRowVal = nextVal;
716✔
236
    if (pFillSup->type == TSDB_FILL_PREV || pFillSup->type == TSDB_FILL_NEXT) {
716✔
237
      pAPI->stateStore.streamStateCurNext(pState, pCur);
275✔
238
      SWinKey nextNextKey = {.groupId = groupId};
275✔
239
      void*   nextNextVal = NULL;
275✔
240
      int32_t nextNextVLen = 0;
275✔
241
      code = pAPI->stateStore.streamStateFillGetGroupKVByCur(pCur, &nextNextKey, (const void**)&nextNextVal, &nextNextVLen);
275✔
242
      if (code == TSDB_CODE_SUCCESS) {
275✔
243
        pFillSup->nextNext.key = nextNextKey.ts;
83✔
244
        pFillSup->nextNext.pRowVal = nextNextVal;
83✔
245
      }
246
    }
247
  }
248
  pAPI->stateStore.streamStateFreeCur(pCur);
1,258✔
249
}
1,258✔
250

251
bool hasCurWindow(SStreamFillSupporter* pFillSup) { return pFillSup->cur.key != INT64_MIN; }
×
252
bool hasPrevWindow(SStreamFillSupporter* pFillSup) { return pFillSup->prev.key != INT64_MIN; }
7,112✔
253
bool hasNextWindow(SStreamFillSupporter* pFillSup) { return pFillSup->next.key != INT64_MIN; }
5,266✔
254
static bool hasNextNextWindow(SStreamFillSupporter* pFillSup) { return pFillSup->nextNext.key != INT64_MIN; }
102✔
255

256
static void transBlockToResultRow(const SSDataBlock* pBlock, int32_t rowId, TSKEY ts, SResultRowData* pRowVal) {
1,312✔
257
  int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
1,312✔
258
  for (int32_t i = 0; i < numOfCols; ++i) {
19,385✔
259
    SColumnInfoData* pColData = taosArrayGet(pBlock->pDataBlock, i);
18,071✔
260
    SResultCellData* pCell = getResultCell(pRowVal, i);
18,071✔
261
    if (!colDataIsNull_s(pColData, rowId)) {
36,146!
262
      pCell->isNull = false;
18,073✔
263
      pCell->type = pColData->info.type;
18,073✔
264
      pCell->bytes = pColData->info.bytes;
18,073✔
265
      char* val = colDataGetData(pColData, rowId);
18,073!
266
      if (IS_VAR_DATA_TYPE(pCell->type)) {
18,073!
267
        memcpy(pCell->pData, val, varDataTLen(val));
41✔
268
      } else {
269
        memcpy(pCell->pData, val, pCell->bytes);
18,032✔
270
      }
271
    } else {
272
      pCell->isNull = true;
×
273
    }
274
  }
275
  pRowVal->key = ts;
1,314✔
276
}
1,314✔
277

278
static void calcRowDeltaData(SResultRowData* pEndRow, SArray* pEndPoins, SFillColInfo* pFillCol, int32_t numOfCol) {
204✔
279
  for (int32_t i = 0; i < numOfCol; i++) {
3,187✔
280
    if (!pFillCol[i].notFillCol) {
2,983✔
281
      int32_t          slotId = GET_DEST_SLOT_ID(pFillCol + i);
2,779✔
282
      SResultCellData* pECell = getResultCell(pEndRow, slotId);
2,779✔
283
      SPoint*          pPoint = taosArrayGet(pEndPoins, slotId);
2,779✔
284
      pPoint->key = pEndRow->key;
2,779✔
285
      memcpy(pPoint->val, pECell->pData, pECell->bytes);
2,779✔
286
    }
287
  }
288
}
204✔
289

290
static void setFillInfoStart(TSKEY ts, SInterval* pInterval, SStreamFillInfo* pFillInfo) {
970✔
291
  ts = taosTimeAdd(ts, pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
970✔
292
  pFillInfo->start = ts;
970✔
293
}
970✔
294

295
static void setFillInfoEnd(TSKEY ts, SInterval* pInterval, SStreamFillInfo* pFillInfo) {
970✔
296
  ts = taosTimeAdd(ts, pInterval->sliding * -1, pInterval->slidingUnit, pInterval->precision, NULL);
970✔
297
  pFillInfo->end = ts;
970✔
298
}
970✔
299

300
static void setFillKeyInfo(TSKEY start, TSKEY end, SInterval* pInterval, SStreamFillInfo* pFillInfo) {
970✔
301
  setFillInfoStart(start, pInterval, pFillInfo);
970✔
302
  pFillInfo->current = pFillInfo->start;
970✔
303
  setFillInfoEnd(end, pInterval, pFillInfo);
970✔
304
}
970✔
305

306
void setDeleteFillValueInfo(TSKEY start, TSKEY end, SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo) {
140✔
307
  if (!hasPrevWindow(pFillSup) || !hasNextWindow(pFillSup)) {
140✔
308
    pFillInfo->needFill = false;
60✔
309
    return;
60✔
310
  }
311

312
  TSKEY realStart = taosTimeAdd(pFillSup->prev.key, pFillSup->interval.sliding, pFillSup->interval.slidingUnit,
80✔
313
                                pFillSup->interval.precision, NULL);
80✔
314

315
  pFillInfo->needFill = true;
80✔
316
  pFillInfo->start = realStart;
80✔
317
  pFillInfo->current = pFillInfo->start;
80✔
318
  pFillInfo->end = end;
80✔
319
  pFillInfo->pos = FILL_POS_INVALID;
80✔
320
  switch (pFillInfo->type) {
80!
321
    case TSDB_FILL_NULL:
32✔
322
    case TSDB_FILL_NULL_F:
323
    case TSDB_FILL_SET_VALUE:
324
    case TSDB_FILL_SET_VALUE_F:
325
      break;
32✔
326
    case TSDB_FILL_PREV:
16✔
327
      pFillInfo->pResRow = &pFillSup->prev;
16✔
328
      break;
16✔
329
    case TSDB_FILL_NEXT:
16✔
330
      pFillInfo->pResRow = &pFillSup->next;
16✔
331
      break;
16✔
332
    case TSDB_FILL_LINEAR: {
16✔
333
      setFillKeyInfo(pFillSup->prev.key, pFillSup->next.key, &pFillSup->interval, pFillInfo);
16✔
334
      pFillInfo->pLinearInfo->hasNext = false;
16✔
335
      pFillInfo->pLinearInfo->nextEnd = INT64_MIN;
16✔
336
      calcRowDeltaData(&pFillSup->next, pFillInfo->pLinearInfo->pEndPoints, pFillSup->pAllColInfo,
16✔
337
                       pFillSup->numOfAllCols);
338
      pFillInfo->pResRow = &pFillSup->prev;
16✔
339
      pFillInfo->pLinearInfo->winIndex = 0;
16✔
340
    } break;
16✔
341
    default:
×
342
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
343
      break;
×
344
  }
345
}
346

347
void copyNotFillExpData(SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo) {
582✔
348
  for (int32_t i = pFillSup->numOfFillCols; i < pFillSup->numOfAllCols; ++i) {
1,300✔
349
    SFillColInfo*    pFillCol = pFillSup->pAllColInfo + i;
718✔
350
    int32_t          slotId = GET_DEST_SLOT_ID(pFillCol);
718✔
351
    SResultCellData* pCell = getResultCell(pFillInfo->pResRow, slotId);
718✔
352
    SResultCellData* pCurCell = getResultCell(&pFillSup->cur, slotId);
718✔
353
    pCell->isNull = pCurCell->isNull;
718✔
354
    if (!pCurCell->isNull) {
718!
355
      memcpy(pCell->pData, pCurCell->pData, pCell->bytes);
718✔
356
    }
357
  }
358
}
582✔
359

360
void setFillValueInfo(SSDataBlock* pBlock, TSKEY ts, int32_t rowId, SStreamFillSupporter* pFillSup,
1,118✔
361
                      SStreamFillInfo* pFillInfo) {
362
  pFillInfo->preRowKey = pFillSup->cur.key;
1,118✔
363
  if (!hasPrevWindow(pFillSup) && !hasNextWindow(pFillSup)) {
1,118✔
364
    pFillInfo->needFill = false;
183✔
365
    pFillInfo->pos = FILL_POS_START;
183✔
366
    return;
183✔
367
  }
368
  TSKEY prevWKey = INT64_MIN;
935✔
369
  TSKEY nextWKey = INT64_MIN;
935✔
370
  if (hasPrevWindow(pFillSup)) {
935✔
371
    prevWKey = pFillSup->prev.key;
458✔
372
  }
373
  if (hasNextWindow(pFillSup)) {
935✔
374
    nextWKey = pFillSup->next.key;
621✔
375
  }
376

377
  pFillInfo->needFill = true;
935✔
378
  pFillInfo->pos = FILL_POS_INVALID;
935✔
379
  switch (pFillInfo->type) {
935!
380
    case TSDB_FILL_NULL:
400✔
381
    case TSDB_FILL_NULL_F:
382
    case TSDB_FILL_SET_VALUE:
383
    case TSDB_FILL_SET_VALUE_F: {
384
      if (pFillSup->prev.key == pFillInfo->preRowKey) {
400!
385
        resetFillWindow(&pFillSup->prev);
×
386
      }
387
      if (hasPrevWindow(pFillSup) && hasNextWindow(pFillSup)) {
400✔
388
        if (pFillSup->next.key == pFillInfo->nextRowKey) {
70✔
389
          pFillInfo->preRowKey = INT64_MIN;
68✔
390
          setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
68✔
391
          pFillInfo->pos = FILL_POS_END;
68✔
392
        } else {
393
          pFillInfo->needFill = false;
2✔
394
          pFillInfo->pos = FILL_POS_START;
2✔
395
        }
396
      } else if (hasPrevWindow(pFillSup)) {
330✔
397
        setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
134✔
398
        pFillInfo->pos = FILL_POS_END;
134✔
399
      } else {
400
        setFillKeyInfo(ts, nextWKey, &pFillSup->interval, pFillInfo);
196✔
401
        pFillInfo->pos = FILL_POS_START;
196✔
402
      }
403
      copyNotFillExpData(pFillSup, pFillInfo);
400✔
404
    } break;
400✔
405
    case TSDB_FILL_PREV: {
179✔
406
      if (hasNextWindow(pFillSup) && ((pFillSup->next.key != pFillInfo->nextRowKey) ||
179✔
407
                                      (pFillSup->next.key == pFillInfo->nextRowKey && hasNextNextWindow(pFillSup)) ||
102!
408
                                      (pFillSup->next.key == pFillInfo->nextRowKey && !hasPrevWindow(pFillSup)))) {
84!
409
        setFillKeyInfo(ts, nextWKey, &pFillSup->interval, pFillInfo);
91✔
410
        pFillInfo->pos = FILL_POS_START;
91✔
411
        resetFillWindow(&pFillSup->prev);
91✔
412
        pFillSup->prev.key = pFillSup->cur.key;
91✔
413
        pFillSup->prev.pRowVal = pFillSup->cur.pRowVal;
91✔
414
      } else if (hasPrevWindow(pFillSup)) {
88!
415
        setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
88✔
416
        pFillInfo->pos = FILL_POS_END;
88✔
417
        pFillInfo->preRowKey = INT64_MIN;
88✔
418
      }
419
      pFillInfo->pResRow = &pFillSup->prev;
179✔
420
    } break;
179✔
421
    case TSDB_FILL_NEXT: {
189✔
422
      if (hasPrevWindow(pFillSup)) {
189✔
423
        setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
93✔
424
        pFillInfo->pos = FILL_POS_END;
93✔
425
        resetFillWindow(&pFillSup->next);
93✔
426
        pFillSup->next.key = pFillSup->cur.key;
93✔
427
        pFillSup->next.pRowVal = pFillSup->cur.pRowVal;
93✔
428
        pFillInfo->preRowKey = INT64_MIN;
93✔
429
      } else {
430
        setFillKeyInfo(ts, nextWKey, &pFillSup->interval, pFillInfo);
96✔
431
        pFillInfo->pos = FILL_POS_START;
96✔
432
      }
433
      pFillInfo->pResRow = &pFillSup->next;
189✔
434
    } break;
189✔
435
    case TSDB_FILL_LINEAR: {
167✔
436
      pFillInfo->pLinearInfo->winIndex = 0;
167✔
437
      if (hasPrevWindow(pFillSup) && hasNextWindow(pFillSup)) {
167✔
438
        setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
21✔
439
        pFillInfo->pos = FILL_POS_MID;
21✔
440
        pFillInfo->pLinearInfo->nextEnd = nextWKey;
21✔
441
        calcRowDeltaData(&pFillSup->cur, pFillInfo->pLinearInfo->pEndPoints, pFillSup->pAllColInfo,
21✔
442
                         pFillSup->numOfAllCols);
443
        pFillInfo->pResRow = &pFillSup->prev;
21✔
444

445
        calcRowDeltaData(&pFillSup->next, pFillInfo->pLinearInfo->pNextEndPoints, pFillSup->pAllColInfo,
21✔
446
                         pFillSup->numOfAllCols);
447
        pFillInfo->pLinearInfo->hasNext = true;
21✔
448
      } else if (hasPrevWindow(pFillSup)) {
146✔
449
        setFillKeyInfo(prevWKey, ts, &pFillSup->interval, pFillInfo);
49✔
450
        pFillInfo->pos = FILL_POS_END;
49✔
451
        pFillInfo->pLinearInfo->nextEnd = INT64_MIN;
49✔
452
        calcRowDeltaData(&pFillSup->cur, pFillInfo->pLinearInfo->pEndPoints, pFillSup->pAllColInfo,
49✔
453
                         pFillSup->numOfAllCols);
454
        pFillInfo->pResRow = &pFillSup->prev;
49✔
455
        pFillInfo->pLinearInfo->hasNext = false;
49✔
456
      } else {
457
        setFillKeyInfo(ts, nextWKey, &pFillSup->interval, pFillInfo);
97✔
458
        pFillInfo->pos = FILL_POS_START;
97✔
459
        pFillInfo->pLinearInfo->nextEnd = INT64_MIN;
97✔
460
        calcRowDeltaData(&pFillSup->next, pFillInfo->pLinearInfo->pEndPoints, pFillSup->pAllColInfo,
97✔
461
                         pFillSup->numOfAllCols);
462
        pFillInfo->pResRow = &pFillSup->cur;
97✔
463
        pFillInfo->pLinearInfo->hasNext = false;
97✔
464
      }
465
    } break;
167✔
466
    default:
×
467
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
468
      break;
×
469
  }
470
}
471

472
int32_t checkResult(SStreamFillSupporter* pFillSup, TSKEY ts, uint64_t groupId, bool* pRes) {
201,856✔
473
  int32_t code = TSDB_CODE_SUCCESS;
201,856✔
474
  int32_t lino = 0;
201,856✔
475
  SWinKey key = {.groupId = groupId, .ts = ts};
201,856✔
476
  if (tSimpleHashGet(pFillSup->pResMap, &key, sizeof(SWinKey)) != NULL) {
201,856✔
477
    (*pRes) = false;
50,200✔
478
    goto _end;
50,200✔
479
  }
480
  code = tSimpleHashPut(pFillSup->pResMap, &key, sizeof(SWinKey), NULL, 0);
151,818✔
481
  QUERY_CHECK_CODE(code, lino, _end);
151,752!
482
  (*pRes) = true;
151,752✔
483

484
_end:
201,952✔
485
  if (code != TSDB_CODE_SUCCESS) {
201,952!
486
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
487
  }
488
  return code;
201,951✔
489
}
490

491
static int32_t buildFillResult(SResultRowData* pResRow, SStreamFillSupporter* pFillSup, TSKEY ts, SSDataBlock* pBlock,
36,195✔
492
                               bool* pRes, bool isFilld) {
493
  int32_t code = TSDB_CODE_SUCCESS;
36,195✔
494
  int32_t lino = 0;
36,195✔
495
  if (pBlock->info.rows >= pBlock->info.capacity) {
36,195✔
496
    (*pRes) = false;
4✔
497
    goto _end;
4✔
498
  }
499
  uint64_t groupId = pBlock->info.id.groupId;
36,191✔
500
  bool     ckRes = true;
36,191✔
501
  code = checkResult(pFillSup, ts, groupId, &ckRes);
36,191✔
502
  QUERY_CHECK_CODE(code, lino, _end);
36,190!
503

504
  if (pFillSup->hasDelete && !ckRes) {
36,190✔
505
    (*pRes) = true;
28✔
506
    goto _end;
28✔
507
  }
508
  for (int32_t i = 0; i < pFillSup->numOfAllCols; ++i) {
139,598✔
509
    SFillColInfo*    pFillCol = pFillSup->pAllColInfo + i;
103,440✔
510
    int32_t          slotId = GET_DEST_SLOT_ID(pFillCol);
103,440✔
511
    SColumnInfoData* pColData = taosArrayGet(pBlock->pDataBlock, slotId);
103,440✔
512
    SFillInfo        tmpInfo = {
103,439✔
513
               .currentKey = ts,
514
               .order = TSDB_ORDER_ASC,
515
               .interval = pFillSup->interval,
516
               .isFilled = isFilld,
517
    };
518
    bool filled = fillIfWindowPseudoColumn(&tmpInfo, pFillCol, pColData, pBlock->info.rows);
103,439✔
519
    if (!filled) {
103,475✔
520
      SResultCellData* pCell = getResultCell(pResRow, slotId);
67,332✔
521
      code = setRowCell(pColData, pBlock->info.rows, pCell);
67,302✔
522
      QUERY_CHECK_CODE(code, lino, _end);
67,293!
523
    }
524
  }
525
  pBlock->info.rows++;
36,158✔
526
  (*pRes) = true;
36,158✔
527

528
_end:
36,190✔
529
  if (code != TSDB_CODE_SUCCESS) {
36,190!
530
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
531
  }
532
  return code;
36,194✔
533
}
534

535
bool hasRemainCalc(SStreamFillInfo* pFillInfo) {
204,725✔
536
  if (pFillInfo->current != INT64_MIN && pFillInfo->current <= pFillInfo->end) {
204,725✔
537
    return true;
199,365✔
538
  }
539
  return false;
5,360✔
540
}
541

542
static void doStreamFillNormal(SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo, SSDataBlock* pBlock) {
805✔
543
  int32_t code = TSDB_CODE_SUCCESS;
805✔
544
  int32_t lino = 0;
805✔
545
  while (hasRemainCalc(pFillInfo) && pBlock->info.rows < pBlock->info.capacity) {
35,202✔
546
    STimeWindow st = {.skey = pFillInfo->current, .ekey = pFillInfo->current};
34,397✔
547
    if (inWinRange(&pFillSup->winRange, &st)) {
34,397!
548
      bool res = true;
34,397✔
549
      code = buildFillResult(pFillInfo->pResRow, pFillSup, pFillInfo->current, pBlock, &res, true);
34,397✔
550
      QUERY_CHECK_CODE(code, lino, _end);
34,397!
551
    }
552
    pFillInfo->current = taosTimeAdd(pFillInfo->current, pFillSup->interval.sliding, pFillSup->interval.slidingUnit,
34,397✔
553
                                     pFillSup->interval.precision, NULL);
34,397✔
554
  }
555

556
_end:
805✔
557
  if (code != TSDB_CODE_SUCCESS) {
805!
558
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
559
  }
560
}
805✔
561

562
static void doStreamFillLinear(SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo, SSDataBlock* pBlock) {
199✔
563
  int32_t code = TSDB_CODE_SUCCESS;
199✔
564
  int32_t lino = 0;
199✔
565
  while (hasRemainCalc(pFillInfo) && pBlock->info.rows < pBlock->info.capacity) {
14,596✔
566
    uint64_t    groupId = pBlock->info.id.groupId;
14,397✔
567
    SWinKey     key = {.groupId = groupId, .ts = pFillInfo->current};
14,397✔
568
    STimeWindow st = {.skey = pFillInfo->current, .ekey = pFillInfo->current};
14,397✔
569
    bool        ckRes = true;
14,397✔
570
    code = checkResult(pFillSup, pFillInfo->current, groupId, &ckRes);
14,397✔
571
    QUERY_CHECK_CODE(code, lino, _end);
14,397!
572

573
    if ((pFillSup->hasDelete && !ckRes) || !inWinRange(&pFillSup->winRange, &st)) {
14,397!
574
      pFillInfo->current = taosTimeAdd(pFillInfo->current, pFillSup->interval.sliding, pFillSup->interval.slidingUnit,
16✔
575
                                       pFillSup->interval.precision, NULL);
8✔
576
      pFillInfo->pLinearInfo->winIndex++;
8✔
577
      continue;
8✔
578
    }
579
    pFillInfo->pLinearInfo->winIndex++;
14,389✔
580
    for (int32_t i = 0; i < pFillSup->numOfAllCols; ++i) {
46,735✔
581
      SFillColInfo* pFillCol = pFillSup->pAllColInfo + i;
32,346✔
582
      SFillInfo     tmp = {
32,346✔
583
              .currentKey = pFillInfo->current,
32,346✔
584
              .order = TSDB_ORDER_ASC,
585
              .interval = pFillSup->interval,
586
              .isFilled = true,
587
      };
588

589
      int32_t          slotId = GET_DEST_SLOT_ID(pFillCol);
32,346✔
590
      SColumnInfoData* pColData = taosArrayGet(pBlock->pDataBlock, slotId);
32,346✔
591
      int16_t          type = pColData->info.type;
32,346✔
592
      SResultCellData* pCell = getResultCell(pFillInfo->pResRow, slotId);
32,346✔
593
      int32_t          index = pBlock->info.rows;
32,346✔
594
      if (pFillCol->notFillCol) {
32,346✔
595
        bool filled = fillIfWindowPseudoColumn(&tmp, pFillCol, pColData, index);
14,389✔
596
        if (!filled) {
14,389!
597
          code = setRowCell(pColData, index, pCell);
×
598
          QUERY_CHECK_CODE(code, lino, _end);
×
599
        }
600
      } else {
601
        if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_BOOL || pCell->isNull) {
17,957!
602
          colDataSetNULL(pColData, index);
23!
603
          continue;
23✔
604
        }
605
        SPoint* pEnd = taosArrayGet(pFillInfo->pLinearInfo->pEndPoints, slotId);
17,934✔
606
        double  vCell = 0;
17,934✔
607
        SPoint  start = {0};
17,934✔
608
        start.key = pFillInfo->pResRow->key;
17,934✔
609
        start.val = pCell->pData;
17,934✔
610

611
        SPoint cur = {0};
17,934✔
612
        cur.key = pFillInfo->current;
17,934✔
613
        cur.val = taosMemoryCalloc(1, pCell->bytes);
17,934!
614
        QUERY_CHECK_NULL(cur.val, code, lino, _end, terrno);
17,934!
615
        taosGetLinearInterpolationVal(&cur, pCell->type, &start, pEnd, pCell->type, typeGetTypeModFromColInfo(&pColData->info));
17,934✔
616
        code = colDataSetVal(pColData, index, (const char*)cur.val, false);
17,934✔
617
        QUERY_CHECK_CODE(code, lino, _end);
17,934!
618
        destroySPoint(&cur);
17,934✔
619
      }
620
    }
621
    pFillInfo->current = taosTimeAdd(pFillInfo->current, pFillSup->interval.sliding, pFillSup->interval.slidingUnit,
28,778✔
622
                                     pFillSup->interval.precision, NULL);
14,389✔
623
    pBlock->info.rows++;
14,389✔
624
  }
625

626
_end:
199✔
627
  if (code != TSDB_CODE_SUCCESS) {
199!
628
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
629
  }
630
}
199✔
631

632
static void keepResultInDiscBuf(SOperatorInfo* pOperator, uint64_t groupId, SResultRowData* pRow, int32_t len) {
1,118✔
633
  SStorageAPI* pAPI = &pOperator->pTaskInfo->storageAPI;
1,118✔
634

635
  SWinKey key = {.groupId = groupId, .ts = pRow->key};
1,118✔
636
  int32_t code = pAPI->stateStore.streamStateFillPut(pOperator->pTaskInfo->streamInfo.pState, &key, pRow->pRowVal, len);
1,118✔
637
  qDebug("===stream===fill operator save key ts:%" PRId64 " group id:%" PRIu64 "  code:%d", key.ts, key.groupId, code);
1,118!
638
  if (code != TSDB_CODE_SUCCESS) {
1,118!
639
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
640
  }
641
}
1,118✔
642

643
void doStreamFillRange(SStreamFillInfo* pFillInfo, SStreamFillSupporter* pFillSup, SSDataBlock* pRes) {
1,168✔
644
  int32_t code = TSDB_CODE_SUCCESS;
1,168✔
645
  int32_t lino = 0;
1,168✔
646
  bool    res = false;
1,168✔
647
  if (pFillInfo->needFill == false) {
1,168✔
648
    code = buildFillResult(&pFillSup->cur, pFillSup, pFillSup->cur.key, pRes, &res, false);
185✔
649
    QUERY_CHECK_CODE(code, lino, _end);
185!
650
    return;
185✔
651
  }
652

653
  if (pFillInfo->pos == FILL_POS_START) {
983✔
654
    code = buildFillResult(&pFillSup->cur, pFillSup, pFillSup->cur.key, pRes, &res, false);
480✔
655
    QUERY_CHECK_CODE(code, lino, _end);
480!
656
    if (res) {
480!
657
      pFillInfo->pos = FILL_POS_INVALID;
480✔
658
    }
659
  }
660
  if (pFillInfo->type != TSDB_FILL_LINEAR) {
983✔
661
    doStreamFillNormal(pFillSup, pFillInfo, pRes);
805✔
662
  } else {
663
    doStreamFillLinear(pFillSup, pFillInfo, pRes);
178✔
664

665
    if (pFillInfo->pos == FILL_POS_MID) {
178✔
666
      code = buildFillResult(&pFillSup->cur, pFillSup, pFillSup->cur.key, pRes, &res, false);
21✔
667
      QUERY_CHECK_CODE(code, lino, _end);
21!
668
      if (res) {
21!
669
        pFillInfo->pos = FILL_POS_INVALID;
21✔
670
      }
671
    }
672

673
    if (pFillInfo->current > pFillInfo->end && pFillInfo->pLinearInfo->hasNext) {
178✔
674
      pFillInfo->pLinearInfo->hasNext = false;
21✔
675
      pFillInfo->pLinearInfo->winIndex = 0;
21✔
676
      taosArraySwap(pFillInfo->pLinearInfo->pEndPoints, pFillInfo->pLinearInfo->pNextEndPoints);
21✔
677
      pFillInfo->pResRow = &pFillSup->cur;
21✔
678
      setFillKeyInfo(pFillSup->cur.key, pFillInfo->pLinearInfo->nextEnd, &pFillSup->interval, pFillInfo);
21✔
679
      doStreamFillLinear(pFillSup, pFillInfo, pRes);
21✔
680
    }
681
  }
682
  if (pFillInfo->pos == FILL_POS_END) {
983✔
683
    code = buildFillResult(&pFillSup->cur, pFillSup, pFillSup->cur.key, pRes, &res, false);
436✔
684
    QUERY_CHECK_CODE(code, lino, _end);
436!
685
    if (res) {
436✔
686
      pFillInfo->pos = FILL_POS_INVALID;
432✔
687
    }
688
  }
689

690
_end:
551✔
691
  if (code != TSDB_CODE_SUCCESS) {
983!
692
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
693
  }
694
}
695

696
int32_t keepBlockRowInDiscBuf(SOperatorInfo* pOperator, SStreamFillInfo* pFillInfo, SSDataBlock* pBlock, TSKEY* tsCol,
1,118✔
697
                           int32_t rowId, uint64_t groupId, int32_t rowSize) {
698
  int32_t code = TSDB_CODE_SUCCESS;
1,118✔
699
  int32_t lino = 0;
1,118✔
700
  TSKEY ts = tsCol[rowId];
1,118✔
701
  pFillInfo->nextRowKey = ts;
1,118✔
702
  SResultRowData tmpNextRow = {.key = ts};
1,118✔
703
  tmpNextRow.pRowVal = taosMemoryCalloc(1, rowSize);
1,118!
704
  QUERY_CHECK_NULL(tmpNextRow.pRowVal, code, lino, _end, terrno);
1,118!
705
  transBlockToResultRow(pBlock, rowId, ts, &tmpNextRow);
1,118✔
706
  keepResultInDiscBuf(pOperator, groupId, &tmpNextRow, rowSize);
1,118✔
707
  taosMemoryFreeClear(tmpNextRow.pRowVal);
1,118!
708

709
_end:
×
710
  if (code != TSDB_CODE_SUCCESS) {
1,118!
711
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
712
  }
713
  return code;
1,118✔
714
}
715

716
static void doFillResults(SOperatorInfo* pOperator, SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo,
1,118✔
717
                          SSDataBlock* pBlock, TSKEY* tsCol, int32_t rowId, SSDataBlock* pRes) {
718
  uint64_t groupId = pBlock->info.id.groupId;
1,118✔
719
  getWindowFromDiscBuf(pOperator, tsCol[rowId], groupId, pFillSup);
1,118✔
720
  if (pFillSup->prev.key == pFillInfo->preRowKey) {
1,118✔
721
    resetFillWindow(&pFillSup->prev);
660✔
722
  }
723
  setFillValueInfo(pBlock, tsCol[rowId], rowId, pFillSup, pFillInfo);
1,118✔
724
  doStreamFillRange(pFillInfo, pFillSup, pRes);
1,118✔
725
}
1,118✔
726

727
static void doStreamFillImpl(SOperatorInfo* pOperator) {
519✔
728
  int32_t                  code = TSDB_CODE_SUCCESS;
519✔
729
  int32_t                  lino = 0;
519✔
730
  SStreamFillOperatorInfo* pInfo = pOperator->info;
519✔
731
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
519✔
732
  SStreamFillSupporter*    pFillSup = pInfo->pFillSup;
519✔
733
  SStreamFillInfo*         pFillInfo = pInfo->pFillInfo;
519✔
734
  SSDataBlock*             pBlock = pInfo->pSrcBlock;
519✔
735
  uint64_t                 groupId = pBlock->info.id.groupId;
519✔
736
  SSDataBlock*             pRes = pInfo->pRes;
519✔
737
  SColumnInfoData*         pTsCol = taosArrayGet(pInfo->pSrcBlock->pDataBlock, pInfo->primaryTsCol);
519✔
738
  TSKEY*                   tsCol = (TSKEY*)pTsCol->pData;
519✔
739
  pRes->info.id.groupId = groupId;
519✔
740
  pInfo->srcRowIndex++;
519✔
741

742
  if (pInfo->srcRowIndex == 0) {
519✔
743
    code = keepBlockRowInDiscBuf(pOperator, pFillInfo, pBlock, tsCol, pInfo->srcRowIndex, groupId, pFillSup->rowSize);
516✔
744
    QUERY_CHECK_CODE(code, lino, _end);
516!
745
    pInfo->srcRowIndex++;
516✔
746
  }
747

748
  while (pInfo->srcRowIndex < pBlock->info.rows) {
1,118✔
749
    code = keepBlockRowInDiscBuf(pOperator, pFillInfo, pBlock, tsCol, pInfo->srcRowIndex, groupId, pFillSup->rowSize);
602✔
750
    QUERY_CHECK_CODE(code, lino, _end);
602!
751
    doFillResults(pOperator, pFillSup, pFillInfo, pBlock, tsCol, pInfo->srcRowIndex - 1, pRes);
602✔
752
    if (pInfo->pRes->info.rows == pInfo->pRes->info.capacity) {
602✔
753
      code = blockDataUpdateTsWindow(pRes, pInfo->primaryTsCol);
3✔
754
      QUERY_CHECK_CODE(code, lino, _end);
3!
755
      return;
3✔
756
    }
757
    pInfo->srcRowIndex++;
599✔
758
  }
759
  doFillResults(pOperator, pFillSup, pFillInfo, pBlock, tsCol, pInfo->srcRowIndex - 1, pRes);
516✔
760
  code = blockDataUpdateTsWindow(pRes, pInfo->primaryTsCol);
516✔
761
  QUERY_CHECK_CODE(code, lino, _end);
516!
762
  blockDataCleanup(pInfo->pSrcBlock);
516✔
763

764
_end:
516✔
765
  if (code != TSDB_CODE_SUCCESS) {
516!
766
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
767
  }
768
}
769

770
static int32_t buildDeleteRange(SOperatorInfo* pOp, TSKEY start, TSKEY end, uint64_t groupId, SSDataBlock* delRes) {
60✔
771
  int32_t          code = TSDB_CODE_SUCCESS;
60✔
772
  int32_t          lino = 0;
60✔
773
  SStorageAPI*     pAPI = &pOp->pTaskInfo->storageAPI;
60✔
774
  void*            pState = pOp->pTaskInfo->streamInfo.pState;
60✔
775
  SExecTaskInfo*   pTaskInfo = pOp->pTaskInfo;
60✔
776
  SSDataBlock*     pBlock = delRes;
60✔
777
  SColumnInfoData* pStartCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
60✔
778
  SColumnInfoData* pEndCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
60✔
779
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
60✔
780
  SColumnInfoData* pGroupCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
60✔
781
  SColumnInfoData* pCalStartCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
60✔
782
  SColumnInfoData* pCalEndCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
60✔
783
  SColumnInfoData* pTbNameCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
60✔
784
  code = colDataSetVal(pStartCol, pBlock->info.rows, (const char*)&start, false);
60✔
785
  QUERY_CHECK_CODE(code, lino, _end);
60!
786

787
  code = colDataSetVal(pEndCol, pBlock->info.rows, (const char*)&end, false);
60✔
788
  QUERY_CHECK_CODE(code, lino, _end);
60!
789

790
  colDataSetNULL(pUidCol, pBlock->info.rows);
60!
791
  code = colDataSetVal(pGroupCol, pBlock->info.rows, (const char*)&groupId, false);
60✔
792
  QUERY_CHECK_CODE(code, lino, _end);
60!
793

794
  colDataSetNULL(pCalStartCol, pBlock->info.rows);
60!
795
  colDataSetNULL(pCalEndCol, pBlock->info.rows);
60!
796

797
  SColumnInfoData* pTableCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
60✔
798

799
  void*   tbname = NULL;
60✔
800
  int32_t winCode = TSDB_CODE_SUCCESS;
60✔
801
  code = pAPI->stateStore.streamStateGetParName(pOp->pTaskInfo->streamInfo.pState, groupId, &tbname, false, &winCode);
60✔
802
  QUERY_CHECK_CODE(code, lino, _end);
60!
803
  if (winCode != TSDB_CODE_SUCCESS) {
60✔
804
    colDataSetNULL(pTableCol, pBlock->info.rows);
15!
805
  } else {
806
    char parTbName[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN];
807
    STR_WITH_MAXSIZE_TO_VARSTR(parTbName, tbname, sizeof(parTbName));
45✔
808
    code = colDataSetVal(pTableCol, pBlock->info.rows, (const char*)parTbName, false);
45✔
809
    QUERY_CHECK_CODE(code, lino, _end);
45!
810
    pAPI->stateStore.streamStateFreeVal(tbname);
45✔
811
  }
812

813
  pBlock->info.rows++;
60✔
814

815
_end:
60✔
816
  if (code != TSDB_CODE_SUCCESS) {
60!
817
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
818
  }
819
  return code;
60✔
820
}
821

822
int32_t buildDeleteResult(SOperatorInfo* pOperator, TSKEY startTs, TSKEY endTs, uint64_t groupId, SSDataBlock* delRes) {
60✔
823
  int32_t                  code = TSDB_CODE_SUCCESS;
60✔
824
  int32_t                  lino = 0;
60✔
825
  SStreamFillOperatorInfo* pInfo = pOperator->info;
60✔
826
  SStreamFillSupporter*    pFillSup = pInfo->pFillSup;
60✔
827
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
60✔
828
  if (hasPrevWindow(pFillSup)) {
60✔
829
    TSKEY start = getNextWindowTs(pFillSup->prev.key, &pFillSup->interval);
40✔
830
    code = buildDeleteRange(pOperator, start, endTs, groupId, delRes);
40✔
831
    QUERY_CHECK_CODE(code, lino, _end);
40!
832
  } else if (hasNextWindow(pFillSup)) {
20✔
833
    TSKEY end = getPrevWindowTs(pFillSup->next.key, &pFillSup->interval);
15✔
834
    code = buildDeleteRange(pOperator, startTs, end, groupId, delRes);
15✔
835
    QUERY_CHECK_CODE(code, lino, _end);
15!
836
  } else {
837
    code = buildDeleteRange(pOperator, startTs, endTs, groupId, delRes);
5✔
838
    QUERY_CHECK_CODE(code, lino, _end);
5!
839
  }
840

841
_end:
5✔
842
  if (code != TSDB_CODE_SUCCESS) {
60!
843
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
844
  }
845
  return code;
60✔
846
}
847

848
static int32_t doDeleteFillResultImpl(SOperatorInfo* pOperator, TSKEY startTs, TSKEY endTs, uint64_t groupId) {
100✔
849
  int32_t                  code = TSDB_CODE_SUCCESS;
100✔
850
  int32_t                  lino = 0;
100✔
851
  SStorageAPI*             pAPI = &pOperator->pTaskInfo->storageAPI;
100✔
852
  SStreamFillOperatorInfo* pInfo = pOperator->info;
100✔
853
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
100✔
854
  getWindowFromDiscBuf(pOperator, startTs, groupId, pInfo->pFillSup);
100✔
855
  setDeleteFillValueInfo(startTs, endTs, pInfo->pFillSup, pInfo->pFillInfo);
100✔
856
  SWinKey key = {.ts = startTs, .groupId = groupId};
100✔
857
  pAPI->stateStore.streamStateFillDel(pOperator->pTaskInfo->streamInfo.pState, &key);
100✔
858
  if (!pInfo->pFillInfo->needFill) {
100✔
859
    code = buildDeleteResult(pOperator, startTs, endTs, groupId, pInfo->pDelRes);
60✔
860
    QUERY_CHECK_CODE(code, lino, _end);
60!
861
  } else {
862
    STimeFillRange tw = {
40✔
863
        .skey = startTs,
864
        .ekey = endTs,
865
        .groupId = groupId,
866
    };
867
    void* tmp = taosArrayPush(pInfo->pFillInfo->delRanges, &tw);
40✔
868
    if (!tmp) {
40!
869
      code = terrno;
×
870
      QUERY_CHECK_CODE(code, lino, _end);
×
871
    }
872
  }
873

874
_end:
100✔
875
  if (code != TSDB_CODE_SUCCESS) {
100!
876
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
877
  }
878
  return code;
100✔
879
}
880

881
static void getWindowInfoByKey(SStorageAPI* pAPI, void* pState, TSKEY ts, int64_t groupId, SResultRowData* pWinData) {
×
882
  SWinKey key = {.ts = ts, .groupId = groupId};
×
883
  void*   val = NULL;
×
884
  int32_t len = 0;
×
885
  int32_t code = pAPI->stateStore.streamStateFillGet(pState, &key, (void**)&val, &len, NULL);
×
886
  if (code != TSDB_CODE_SUCCESS) {
×
887
    qDebug("get window info by key failed, Data may be deleted, try next window. ts:%" PRId64 ", groupId:%" PRId64, ts,
×
888
           groupId);
889
    SStreamStateCur* pCur = pAPI->stateStore.streamStateFillSeekKeyNext(pState, &key);
×
890
    code = pAPI->stateStore.streamStateFillGetGroupKVByCur(pCur, &key, (const void**)&val, &len);
×
891
    pAPI->stateStore.streamStateFreeCur(pCur);
×
892
    qDebug("get window info by key ts:%" PRId64 ", groupId:%" PRId64 ", res%d", ts, groupId, code);
×
893
  }
894

895
  if (code == TSDB_CODE_SUCCESS) {
×
896
    resetFillWindow(pWinData);
×
897
    pWinData->key = key.ts;
×
898
    pWinData->pRowVal = val;
×
899
  }
900
}
×
901

902
static void doDeleteFillFinalize(SOperatorInfo* pOperator) {
1,470✔
903
  SStorageAPI* pAPI = &pOperator->pTaskInfo->storageAPI;
1,470✔
904

905
  SStreamFillOperatorInfo* pInfo = pOperator->info;
1,470✔
906
  SStreamFillInfo*         pFillInfo = pInfo->pFillInfo;
1,470✔
907
  int32_t                  size = taosArrayGetSize(pFillInfo->delRanges);
1,470✔
908
  while (pFillInfo->delIndex < size) {
1,510✔
909
    STimeFillRange* range = taosArrayGet(pFillInfo->delRanges, pFillInfo->delIndex);
40✔
910
    if (pInfo->pRes->info.id.groupId != 0 && pInfo->pRes->info.id.groupId != range->groupId) {
40!
911
      return;
×
912
    }
913
    getWindowFromDiscBuf(pOperator, range->skey, range->groupId, pInfo->pFillSup);
40✔
914
    TSKEY realEnd = range->ekey + 1;
40✔
915
    if (pInfo->pFillInfo->type == TSDB_FILL_NEXT && pInfo->pFillSup->next.key != realEnd) {
40!
916
      getWindowInfoByKey(pAPI, pOperator->pTaskInfo->streamInfo.pState, realEnd, range->groupId,
×
917
                         &pInfo->pFillSup->next);
×
918
    }
919
    setDeleteFillValueInfo(range->skey, range->ekey, pInfo->pFillSup, pInfo->pFillInfo);
40✔
920
    pFillInfo->delIndex++;
40✔
921
    if (pInfo->pFillInfo->needFill) {
40!
922
      doStreamFillRange(pInfo->pFillInfo, pInfo->pFillSup, pInfo->pRes);
40✔
923
      pInfo->pRes->info.id.groupId = range->groupId;
40✔
924
    }
925
  }
926
}
927

928
static int32_t doDeleteFillResult(SOperatorInfo* pOperator) {
90✔
929
  int32_t                  code = TSDB_CODE_SUCCESS;
90✔
930
  int32_t                  lino = 0;
90✔
931
  SStorageAPI*             pAPI = &pOperator->pTaskInfo->storageAPI;
90✔
932
  SStreamFillOperatorInfo* pInfo = pOperator->info;
90✔
933
  SStreamFillInfo*         pFillInfo = pInfo->pFillInfo;
90✔
934
  SSDataBlock*             pBlock = pInfo->pSrcDelBlock;
90✔
935
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
90✔
936

937
  SColumnInfoData* pStartCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
90✔
938
  TSKEY*           tsStarts = (TSKEY*)pStartCol->pData;
90✔
939
  SColumnInfoData* pGroupCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
90✔
940
  uint64_t*        groupIds = (uint64_t*)pGroupCol->pData;
90✔
941
  while (pInfo->srcDelRowIndex < pBlock->info.rows) {
235✔
942
    TSKEY            ts = tsStarts[pInfo->srcDelRowIndex];
145✔
943
    TSKEY            endTs = ts;
145✔
944
    uint64_t         groupId = groupIds[pInfo->srcDelRowIndex];
145✔
945
    SWinKey          key = {.ts = ts, .groupId = groupId};
145✔
946
    SStreamStateCur* pCur = pAPI->stateStore.streamStateGetAndCheckCur(pOperator->pTaskInfo->streamInfo.pState, &key);
145✔
947

948
    if (!pCur) {
145✔
949
      pInfo->srcDelRowIndex++;
45✔
950
      continue;
45✔
951
    }
952

953
    SWinKey nextKey = {.groupId = groupId, .ts = ts};
100✔
954
    while (pInfo->srcDelRowIndex < pBlock->info.rows) {
286✔
955
      TSKEY    delTs = tsStarts[pInfo->srcDelRowIndex];
241✔
956
      uint64_t delGroupId = groupIds[pInfo->srcDelRowIndex];
241✔
957
      int32_t  winCode = TSDB_CODE_SUCCESS;
241✔
958
      if (groupId != delGroupId) {
241!
959
        break;
55✔
960
      }
961
      if (delTs > nextKey.ts) {
241✔
962
        break;
10✔
963
      }
964

965
      SWinKey delKey = {.groupId = delGroupId, .ts = delTs};
231✔
966
      if (delTs == nextKey.ts) {
231✔
967
        pAPI->stateStore.streamStateCurNext(pOperator->pTaskInfo->streamInfo.pState, pCur);
171✔
968
        winCode = pAPI->stateStore.streamStateFillGetGroupKVByCur(pCur, &nextKey, NULL, NULL);
171✔
969
        // ts will be deleted later
970
        if (delTs != ts) {
171✔
971
          pAPI->stateStore.streamStateFillDel(pOperator->pTaskInfo->streamInfo.pState, &delKey);
71✔
972
          pAPI->stateStore.streamStateFreeCur(pCur);
71✔
973
          pCur = pAPI->stateStore.streamStateGetAndCheckCur(pOperator->pTaskInfo->streamInfo.pState, &nextKey);
71✔
974
        }
975
        endTs = TMAX(delTs, nextKey.ts - 1);
171✔
976
        if (winCode != TSDB_CODE_SUCCESS) {
171✔
977
          break;
45✔
978
        }
979
      }
980
      pInfo->srcDelRowIndex++;
186✔
981
    }
982

983
    pAPI->stateStore.streamStateFreeCur(pCur);
100✔
984
    code = doDeleteFillResultImpl(pOperator, ts, endTs, groupId);
100✔
985
    QUERY_CHECK_CODE(code, lino, _end);
100!
986
  }
987

988
  pFillInfo->current = pFillInfo->end + 1;
90✔
989

990
_end:
90✔
991
  if (code != TSDB_CODE_SUCCESS) {
90!
992
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
993
  }
994
  return code;
90✔
995
}
996

997
void resetStreamFillSup(SStreamFillSupporter* pFillSup) {
5,176✔
998
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);
5,176✔
999
  SSHashObj* pNewMap = tSimpleHashInit(16, hashFn);
5,176✔
1000
  if (pNewMap != NULL) {
5,176!
1001
    tSimpleHashCleanup(pFillSup->pResMap);
5,176✔
1002
    pFillSup->pResMap = pNewMap;
5,176✔
1003
  } else {
1004
    tSimpleHashClear(pFillSup->pResMap);
×
1005
  }
1006
  pFillSup->hasDelete = false;
5,176✔
1007
}
5,176✔
1008
void resetStreamFillInfo(SStreamFillOperatorInfo* pInfo) {
2,465✔
1009
  resetStreamFillSup(pInfo->pFillSup);
2,465✔
1010
  taosArrayClear(pInfo->pFillInfo->delRanges);
2,465✔
1011
  pInfo->pFillInfo->delIndex = 0;
2,465✔
1012
}
2,465✔
1013

1014
int32_t doApplyStreamScalarCalculation(SOperatorInfo* pOperator, SSDataBlock* pSrcBlock,
643✔
1015
                                              SSDataBlock* pDstBlock) {
1016
  int32_t                  code = TSDB_CODE_SUCCESS;
643✔
1017
  int32_t                  lino = 0;
643✔
1018
  SStreamFillOperatorInfo* pInfo = pOperator->info;
643✔
1019
  SExprSupp*               pSup = &pOperator->exprSupp;
643✔
1020
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
643✔
1021

1022
  blockDataCleanup(pDstBlock);
643✔
1023
  code = blockDataEnsureCapacity(pDstBlock, pSrcBlock->info.rows);
642✔
1024
  QUERY_CHECK_CODE(code, lino, _end);
642!
1025

1026
  code = setInputDataBlock(pSup, pSrcBlock, TSDB_ORDER_ASC, MAIN_SCAN, false);
642✔
1027
  QUERY_CHECK_CODE(code, lino, _end);
643!
1028
  code = projectApplyFunctions(pSup->pExprInfo, pDstBlock, pSrcBlock, pSup->pCtx, pSup->numOfExprs, NULL);
643✔
1029
  QUERY_CHECK_CODE(code, lino, _end);
643!
1030

1031
  pDstBlock->info.rows = 0;
643✔
1032
  pSup = &pInfo->pFillSup->notFillExprSup;
643✔
1033
  code = setInputDataBlock(pSup, pSrcBlock, TSDB_ORDER_ASC, MAIN_SCAN, false);
643✔
1034
  QUERY_CHECK_CODE(code, lino, _end);
643!
1035
  code = projectApplyFunctions(pSup->pExprInfo, pDstBlock, pSrcBlock, pSup->pCtx, pSup->numOfExprs, NULL);
643✔
1036
  QUERY_CHECK_CODE(code, lino, _end);
643!
1037

1038
  pDstBlock->info.id.groupId = pSrcBlock->info.id.groupId;
643✔
1039

1040
  code = blockDataUpdateTsWindow(pDstBlock, pInfo->primaryTsCol);
643✔
1041

1042
_end:
643✔
1043
  if (code != TSDB_CODE_SUCCESS) {
643!
1044
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
1045
  }
1046
  return code;
643✔
1047
}
1048

1049
static int32_t doStreamFillNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2,133✔
1050
  int32_t                  code = TSDB_CODE_SUCCESS;
2,133✔
1051
  int32_t                  lino = 0;
2,133✔
1052
  SStreamFillOperatorInfo* pInfo = pOperator->info;
2,133✔
1053
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
2,133✔
1054

1055
  if (pOperator->status == OP_EXEC_DONE) {
2,133!
1056
    (*ppRes) = NULL;
×
1057
    return code;
×
1058
  }
1059
  blockDataCleanup(pInfo->pRes);
2,133✔
1060
  if (hasRemainCalc(pInfo->pFillInfo) ||
2,132✔
1061
      (pInfo->pFillInfo->pos != FILL_POS_INVALID && pInfo->pFillInfo->needFill == true)) {
2,123!
1062
    doStreamFillRange(pInfo->pFillInfo, pInfo->pFillSup, pInfo->pRes);
10✔
1063
    if (pInfo->pRes->info.rows > 0) {
10!
1064
      printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
10✔
1065
      (*ppRes) = pInfo->pRes;
10✔
1066
      return code;
10✔
1067
    }
1068
  }
1069
  if (pOperator->status == OP_RES_TO_RETURN) {
2,123✔
1070
    doDeleteFillFinalize(pOperator);
31✔
1071
    if (pInfo->pRes->info.rows > 0) {
31!
1072
      printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
×
1073
      (*ppRes) = pInfo->pRes;
×
1074
      return code;
×
1075
    }
1076
    setOperatorCompleted(pOperator);
31✔
1077
    resetStreamFillInfo(pInfo);
31✔
1078
    (*ppRes) = NULL;
31✔
1079
    return code;
31✔
1080
  }
1081

1082
  SSDataBlock*   fillResult = NULL;
2,092✔
1083
  SOperatorInfo* downstream = pOperator->pDownstream[0];
2,092✔
1084
  while (1) {
1085
    if (pInfo->srcRowIndex >= pInfo->pSrcBlock->info.rows || pInfo->pSrcBlock->info.rows == 0) {
2,122✔
1086
      // If there are delete datablocks, we receive  them first.
1087
      SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
2,119✔
1088
      if (pBlock == NULL) {
2,119✔
1089
        pOperator->status = OP_RES_TO_RETURN;
1,439✔
1090
        pInfo->pFillInfo->preRowKey = INT64_MIN;
1,439✔
1091
        if (pInfo->pRes->info.rows > 0) {
1,439!
1092
          printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
×
1093
          (*ppRes) = pInfo->pRes;
×
1094
          return code;
×
1095
        }
1096
        break;
1,439✔
1097
      }
1098
      printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "recv", GET_TASKID(pTaskInfo));
680✔
1099

1100
      if (pInfo->pFillInfo->curGroupId != pBlock->info.id.groupId) {
680✔
1101
        pInfo->pFillInfo->curGroupId = pBlock->info.id.groupId;
173✔
1102
        pInfo->pFillInfo->preRowKey = INT64_MIN;
173✔
1103
      }
1104

1105
      pInfo->pFillSup->winRange = pTaskInfo->streamInfo.fillHistoryWindow;
680✔
1106
      if (pInfo->pFillSup->winRange.ekey <= 0) {
680!
1107
        pInfo->pFillSup->winRange.ekey = INT64_MAX;
×
1108
      }
1109

1110
      switch (pBlock->info.type) {
680!
1111
        case STREAM_RETRIEVE:
5✔
1112
          (*ppRes) = pBlock;
5✔
1113
          return code;
5✔
1114
        case STREAM_DELETE_RESULT: {
90✔
1115
          pInfo->pSrcDelBlock = pBlock;
90✔
1116
          pInfo->srcDelRowIndex = 0;
90✔
1117
          blockDataCleanup(pInfo->pDelRes);
90✔
1118
          pInfo->pFillSup->hasDelete = true;
90✔
1119
          code = doDeleteFillResult(pOperator);
90✔
1120
          QUERY_CHECK_CODE(code, lino, _end);
90!
1121

1122
          if (pInfo->pDelRes->info.rows > 0) {
90✔
1123
            printDataBlock(pInfo->pDelRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
60✔
1124
            (*ppRes) = pInfo->pDelRes;
60✔
1125
            return code;
60✔
1126
          }
1127
          continue;
30✔
1128
        } break;
1129
        case STREAM_NORMAL:
516✔
1130
        case STREAM_INVALID:
1131
        case STREAM_PULL_DATA: {
1132
          code = doApplyStreamScalarCalculation(pOperator, pBlock, pInfo->pSrcBlock);
516✔
1133
          QUERY_CHECK_CODE(code, lino, _end);
516!
1134

1135
          memcpy(pInfo->pSrcBlock->info.parTbName, pBlock->info.parTbName, TSDB_TABLE_NAME_LEN);
516✔
1136
          pInfo->srcRowIndex = -1;
516✔
1137
        } break;
516✔
1138
        case STREAM_CHECKPOINT:
69✔
1139
        case STREAM_CREATE_CHILD_TABLE: {
1140
          (*ppRes) = pBlock;
69✔
1141
          return code;
69✔
1142
        } break;
1143
        default:
×
1144
          return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1145
      }
1146
    }
1147

1148
    doStreamFillImpl(pOperator);
519✔
1149
    code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, &pInfo->matchInfo);
519✔
1150
    QUERY_CHECK_CODE(code, lino, _end);
519!
1151

1152
    memcpy(pInfo->pRes->info.parTbName, pInfo->pSrcBlock->info.parTbName, TSDB_TABLE_NAME_LEN);
519✔
1153
    pOperator->resultInfo.totalRows += pInfo->pRes->info.rows;
519✔
1154
    if (pInfo->pRes->info.rows > 0) {
519!
1155
      break;
519✔
1156
    }
1157
  }
1158
  if (pOperator->status == OP_RES_TO_RETURN) {
1,958✔
1159
    doDeleteFillFinalize(pOperator);
1,439✔
1160
  }
1161

1162
  if (pInfo->pRes->info.rows == 0) {
1,958✔
1163
    setOperatorCompleted(pOperator);
1,408✔
1164
    resetStreamFillInfo(pInfo);
1,408✔
1165
    (*ppRes) = NULL;
1,408✔
1166
    return code;
1,408✔
1167
  }
1168

1169
  pOperator->resultInfo.totalRows += pInfo->pRes->info.rows;
550✔
1170
  printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
550✔
1171
  (*ppRes) = pInfo->pRes;
550✔
1172
  return code;
550✔
1173

1174
_end:
×
1175
  if (code != TSDB_CODE_SUCCESS) {
×
1176
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
1177
    pTaskInfo->code = code;
×
1178
    T_LONG_JMP(pTaskInfo->env, code);
×
1179
  }
1180
  setOperatorCompleted(pOperator);
×
1181
  resetStreamFillInfo(pInfo);
×
1182
  (*ppRes) = NULL;
×
1183
  return code;
×
1184
}
1185

1186
static void resetForceFillWindow(SResultRowData* pRowData) {
676✔
1187
  pRowData->key = INT64_MIN;
676✔
1188
  pRowData->pRowVal = NULL;
676✔
1189
}
676✔
1190

1191
void doBuildForceFillResultImpl(SOperatorInfo* pOperator, SStreamFillSupporter* pFillSup,
527✔
1192
                                SStreamFillInfo* pFillInfo, SSDataBlock* pBlock, SGroupResInfo* pGroupResInfo) {
1193
  int32_t code = TSDB_CODE_SUCCESS;
527✔
1194
  int32_t lino = 0;
527✔
1195

1196
  SStreamFillOperatorInfo* pInfo = pOperator->info;
527✔
1197
  bool                     res = false;
527✔
1198
  int32_t                  numOfRows = getNumOfTotalRes(pGroupResInfo);
527✔
1199
  for (; pGroupResInfo->index < numOfRows; pGroupResInfo->index++) {
1,203✔
1200
    SWinKey* pKey = (SWinKey*)taosArrayGet(pGroupResInfo->pRows, pGroupResInfo->index);
834✔
1201
    if (pBlock->info.id.groupId == 0) {
834✔
1202
      pBlock->info.id.groupId = pKey->groupId;
527✔
1203
    } else if (pBlock->info.id.groupId != pKey->groupId) {
307✔
1204
      break;
158✔
1205
    }
1206

1207
    SRowBuffPos* pValPos = NULL;
676✔
1208
    int32_t      len = 0;
676✔
1209
    int32_t      winCode = TSDB_CODE_SUCCESS;
676✔
1210
    code = pInfo->stateStore.streamStateFillGet(pInfo->pState, pKey, (void**)&pValPos, &len, &winCode);
676✔
1211
    QUERY_CHECK_CODE(code, lino, _end);
676!
1212
    qDebug("===stream=== build force fill res. key:%" PRId64 ",groupId:%" PRId64".res:%d", pKey->ts, pKey->groupId, winCode);
676!
1213
    if (winCode == TSDB_CODE_SUCCESS) {
676✔
1214
      pFillSup->cur.key = pKey->ts;
194✔
1215
      pFillSup->cur.pRowVal = pValPos->pRowBuff;
194✔
1216
      code = buildFillResult(&pFillSup->cur, pFillSup, pKey->ts, pBlock, &res, false);
194✔
1217
      QUERY_CHECK_CODE(code, lino, _end);
194!
1218
      resetForceFillWindow(&pFillSup->cur);
194✔
1219
      releaseOutputBuf(pInfo->pState, pValPos, &pInfo->stateStore);
194✔
1220
    } else {
1221
      SWinKey      preKey = {.ts = INT64_MIN, .groupId = pKey->groupId};
482✔
1222
      SRowBuffPos* prePos = NULL;
482✔
1223
      int32_t      preVLen = 0;
482✔
1224
      code = pInfo->stateStore.streamStateFillGetPrev(pInfo->pState, pKey, &preKey,
482✔
1225
                                                      (void**)&prePos, &preVLen, &winCode);
1226
      QUERY_CHECK_CODE(code, lino, _end);
482!
1227
      if (winCode == TSDB_CODE_SUCCESS) {
482!
1228
        pFillSup->cur.key = pKey->ts;
482✔
1229
        pFillSup->cur.pRowVal = prePos->pRowBuff;
482✔
1230
        if (pFillInfo->type == TSDB_FILL_PREV) {
482✔
1231
          code = buildFillResult(&pFillSup->cur, pFillSup, pKey->ts, pBlock, &res, true);
300✔
1232
          QUERY_CHECK_CODE(code, lino, _end);
300!
1233
        } else {
1234
          copyNotFillExpData(pFillSup, pFillInfo);
182✔
1235
          pFillInfo->pResRow->key = pKey->ts;
182✔
1236
          code = buildFillResult(pFillInfo->pResRow, pFillSup, pKey->ts, pBlock, &res, true);
182✔
1237
          QUERY_CHECK_CODE(code, lino, _end);
182!
1238
        }
1239
        resetForceFillWindow(&pFillSup->cur);
482✔
1240
      }
1241
      releaseOutputBuf(pInfo->pState, prePos, &pInfo->stateStore);
482✔
1242
    }
1243
  }
1244

1245
  if (pBlock->info.parTbName[0] == 0 && pBlock->info.id.groupId != 0) {
527!
1246
    void*   tbname = NULL;
527✔
1247
    int32_t winCode = TSDB_CODE_SUCCESS;
527✔
1248

1249
    code = pInfo->stateStore.streamStateGetParName(pInfo->pState, pBlock->info.id.groupId, &tbname, false, &winCode);
527✔
1250
    QUERY_CHECK_CODE(code, lino, _end);
527!
1251
    if (winCode == TSDB_CODE_SUCCESS) {
527✔
1252
      memcpy(pBlock->info.parTbName, tbname, TSDB_TABLE_NAME_LEN);
31✔
1253
      pInfo->stateStore.streamStateFreeVal(tbname);
31✔
1254
    }
1255
  }
1256

1257
_end:
×
1258
  if (code != TSDB_CODE_SUCCESS) {
527!
1259
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1260
  }
1261
}
527✔
1262

1263
void doBuildForceFillResult(SOperatorInfo* pOperator, SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo,
1,553✔
1264
                            SSDataBlock* pBlock, SGroupResInfo* pGroupResInfo) {
1265
  blockDataCleanup(pBlock);
1,553✔
1266
  if (!hasRemainResults(pGroupResInfo)) {
1,552✔
1267
    return;
1,025✔
1268
  }
1269

1270
  // clear the existed group id
1271
  pBlock->info.id.groupId = 0;
527✔
1272
  memset(pBlock->info.parTbName, 0, tListLen(pBlock->info.parTbName));
527✔
1273

1274
  doBuildForceFillResultImpl(pOperator, pFillSup, pFillInfo, pBlock, pGroupResInfo);
527✔
1275
}
1276

1277
static int32_t buildForceFillResult(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,553✔
1278
  int32_t                  code = TSDB_CODE_SUCCESS;
1,553✔
1279
  int32_t                  lino = 0;
1,553✔
1280
  SStreamFillOperatorInfo* pInfo = pOperator->info;
1,553✔
1281
  uint16_t                 opType = pOperator->operatorType;
1,553✔
1282
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
1,553✔
1283

1284
  doBuildForceFillResult(pOperator, pInfo->pFillSup, pInfo->pFillInfo, pInfo->pRes, &pInfo->groupResInfo);
1,553✔
1285
  if (pInfo->pRes->info.rows != 0) {
1,551✔
1286
    printDataBlock(pInfo->pRes, getStreamOpName(opType), GET_TASKID(pTaskInfo));
526✔
1287
    (*ppRes) = pInfo->pRes;
527✔
1288
    goto _end;
527✔
1289
  }
1290

1291
  (*ppRes) = NULL;
1,025✔
1292

1293
_end:
1,552✔
1294
  if (code != TSDB_CODE_SUCCESS) {
1,552!
1295
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1296
  }
1297
  return code;
1,552✔
1298
}
1299

1300
static void keepResultInStateBuf(SStreamFillOperatorInfo* pInfo, uint64_t groupId, SResultRowData* pRow) {
194✔
1301
  int32_t code = TSDB_CODE_SUCCESS;
194✔
1302
  int32_t lino = 0;
194✔
1303

1304
  SWinKey      key = {.groupId = groupId, .ts = pRow->key};
194✔
1305
  int32_t      curVLen = 0;
194✔
1306
  SRowBuffPos* pStatePos = NULL;
194✔
1307
  int32_t      winCode = TSDB_CODE_SUCCESS;
194✔
1308
  code = pInfo->stateStore.streamStateFillAddIfNotExist(pInfo->pState, &key, (void**)&pStatePos,
194✔
1309
                                                        &curVLen, &winCode);
1310
  QUERY_CHECK_CODE(code, lino, _end);
194!
1311
  memcpy(pStatePos->pRowBuff, pRow->pRowVal, pInfo->pFillSup->rowSize);
194✔
1312
  qDebug("===stream===fill operator save key ts:%" PRId64 " group id:%" PRIu64 "  code:%d", key.ts, key.groupId, code);
194!
1313

1314
_end:
×
1315
  if (code != TSDB_CODE_SUCCESS) {
194!
1316
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1317
  }
1318
}
194✔
1319

1320
int32_t keepBlockRowInStateBuf(SStreamFillOperatorInfo* pInfo, SStreamFillInfo* pFillInfo, SSDataBlock* pBlock, TSKEY* tsCol,
194✔
1321
                               int32_t rowId, uint64_t groupId, int32_t rowSize) {
1322
  int32_t code = TSDB_CODE_SUCCESS;
194✔
1323
  int32_t lino = 0;
194✔
1324
  TSKEY ts = tsCol[rowId];
194✔
1325
  pFillInfo->nextRowKey = ts;
194✔
1326
  TAOS_MEMSET(pFillInfo->pTempBuff, 0, rowSize);
194✔
1327
  SResultRowData tmpNextRow = {.key = ts, .pRowVal = pFillInfo->pTempBuff};
194✔
1328

1329
  transBlockToResultRow(pBlock, rowId, ts, &tmpNextRow);
194✔
1330
  keepResultInStateBuf(pInfo, groupId, &tmpNextRow);
194✔
1331

1332
_end:
194✔
1333
  if (code != TSDB_CODE_SUCCESS) {
194!
1334
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1335
  }
1336
  return code;
194✔
1337
}
1338

1339
// force window close impl
1340
static int32_t doStreamForceFillImpl(SOperatorInfo* pOperator) {
127✔
1341
  int32_t                  code = TSDB_CODE_SUCCESS;
127✔
1342
  int32_t                  lino = 0;
127✔
1343
  SStreamFillOperatorInfo* pInfo = pOperator->info;
127✔
1344
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
127✔
1345
  SStreamFillSupporter*    pFillSup = pInfo->pFillSup;
127✔
1346
  SStreamFillInfo*         pFillInfo = pInfo->pFillInfo;
127✔
1347
  SSDataBlock*             pBlock = pInfo->pSrcBlock;
127✔
1348
  uint64_t                 groupId = pBlock->info.id.groupId;
127✔
1349
  SColumnInfoData*         pTsCol = taosArrayGet(pInfo->pSrcBlock->pDataBlock, pInfo->primaryTsCol);
127✔
1350
  TSKEY*                   tsCol = (TSKEY*)pTsCol->pData;
127✔
1351
  for (int32_t i = 0; i < pBlock->info.rows; i++){
321✔
1352
    code = keepBlockRowInStateBuf(pInfo, pFillInfo, pBlock, tsCol, i, groupId, pFillSup->rowSize);
194✔
1353
    QUERY_CHECK_CODE(code, lino, _end);
194!
1354

1355
    int32_t size =  taosArrayGetSize(pInfo->pCloseTs);
194✔
1356
    if (size > 0) {
194!
1357
      TSKEY* pTs = (TSKEY*) taosArrayGet(pInfo->pCloseTs, 0);
194✔
1358
      TSKEY  resTs = tsCol[i];
194✔
1359
      while (resTs < (*pTs)) {
311✔
1360
        SWinKey key = {.groupId = groupId, .ts = resTs};
149✔
1361
        void* pPushRes = taosArrayPush(pInfo->pUpdated, &key);
149✔
1362
        QUERY_CHECK_NULL(pPushRes, code, lino, _end, terrno);
149!
1363

1364
        if (IS_FILL_CONST_VALUE(pFillSup->type)) {
149!
1365
          break;
1366
        }
1367
        resTs = taosTimeAdd(resTs, pFillSup->interval.sliding, pFillSup->interval.slidingUnit,
117✔
1368
                            pFillSup->interval.precision, NULL);
117✔
1369
      }
1370
    }
1371
  }
1372
  code = pInfo->stateStore.streamStateGroupPut(pInfo->pState, groupId, NULL, 0);
127✔
1373
  QUERY_CHECK_CODE(code, lino, _end);
127!
1374

1375
_end:
127✔
1376
  if (code != TSDB_CODE_SUCCESS) {
127!
1377
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
1378
  }
1379
  return code;
127✔
1380
}
1381

1382
int32_t buildAllResultKey(SStateStore* pStateStore, SStreamState* pState, TSKEY ts, SArray* pUpdated) {
3,421✔
1383
  int32_t          code = TSDB_CODE_SUCCESS;
3,421✔
1384
  int32_t          lino = 0;
3,421✔
1385
  int64_t          groupId = 0;
3,421✔
1386
  SStreamStateCur* pCur = pStateStore->streamStateGroupGetCur(pState);
3,421✔
1387
  while (1) {  
1,469✔
1388
    int32_t winCode = pStateStore->streamStateGroupGetKVByCur(pCur, &groupId, NULL, NULL);
4,857✔
1389
    if (winCode != TSDB_CODE_SUCCESS) {
4,908✔
1390
      break;
3,421✔
1391
    }
1392
    SWinKey key = {.ts = ts, .groupId = groupId};
1,487✔
1393
    void* pPushRes = taosArrayPush(pUpdated, &key);
1,487✔
1394
    QUERY_CHECK_NULL(pPushRes, code, lino, _end, terrno);
1,487!
1395

1396
    pStateStore->streamStateGroupCurNext(pCur);
1,487✔
1397
  }
1398
  pStateStore->streamStateFreeCur(pCur);
3,421✔
1399
  pCur = NULL;
3,421✔
1400

1401
_end:
3,421✔
1402
  if (code != TSDB_CODE_SUCCESS) {
3,421!
1403
    pStateStore->streamStateFreeCur(pCur);
×
1404
    pCur = NULL;
×
1405
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1406
  }
1407
  return code;
3,420✔
1408
}
1409

1410
void removeDuplicateResult(SArray* pTsArrray, __compar_fn_t fn) {
1,026✔
1411
  taosArraySort(pTsArrray, fn);
1,026✔
1412
  taosArrayRemoveDuplicate(pTsArrray, fn, NULL);
1,026✔
1413
}
1,026✔
1414

1415
// force window close
1416
static int32_t doStreamForceFillNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,557✔
1417
  int32_t                  code = TSDB_CODE_SUCCESS;
1,557✔
1418
  int32_t                  lino = 0;
1,557✔
1419
  SStreamFillOperatorInfo* pInfo = pOperator->info;
1,557✔
1420
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
1,557✔
1421
  qDebug("%s ===stream===return data:%s.", __FUNCTION__, getStreamOpName(pOperator->operatorType));
1,557!
1422

1423
  if (pOperator->status == OP_EXEC_DONE) {
1,557!
1424
    (*ppRes) = NULL;
×
1425
    return code;
×
1426
  }
1427

1428
  if (pOperator->status == OP_RES_TO_RETURN) {
1,557✔
1429
    SSDataBlock* resBlock = NULL;
527✔
1430
    code = buildForceFillResult(pOperator, &resBlock);
527✔
1431
    QUERY_CHECK_CODE(code, lino, _end);
527!
1432

1433
    if (resBlock != NULL) {
527✔
1434
      (*ppRes) = resBlock;
158✔
1435
      goto _end;
158✔
1436
    }
1437

1438
    pInfo->stateStore.streamStateClearExpiredState(pInfo->pState, 1, INT64_MAX);
369✔
1439
    resetStreamFillInfo(pInfo);
369✔
1440
    setStreamOperatorCompleted(pOperator);
369✔
1441
    (*ppRes) = NULL;
369✔
1442
    goto _end;
369✔
1443
  }
1444

1445
  SSDataBlock*   fillResult = NULL;
1,030✔
1446
  SOperatorInfo* downstream = pOperator->pDownstream[0];
1,030✔
1447
  while (1) {
1,152✔
1448
    SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
2,182✔
1449
    if (pBlock == NULL) {
2,182✔
1450
      pOperator->status = OP_RES_TO_RETURN;
1,026✔
1451
      qDebug("===stream===return data:%s.", getStreamOpName(pOperator->operatorType));
1,026!
1452
      break;
1,026✔
1453
    }
1454
    printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "recv", GET_TASKID(pTaskInfo));
1,156✔
1455
    setStreamOperatorState(&pInfo->basic, pBlock->info.type);
1,156✔
1456

1457
    switch (pBlock->info.type) {
1,156!
1458
      case STREAM_NORMAL:
127✔
1459
      case STREAM_INVALID: {
1460
        code = doApplyStreamScalarCalculation(pOperator, pBlock, pInfo->pSrcBlock);
127✔
1461
        QUERY_CHECK_CODE(code, lino, _end);
127!
1462

1463
        memcpy(pInfo->pSrcBlock->info.parTbName, pBlock->info.parTbName, TSDB_TABLE_NAME_LEN);
127✔
1464
        pInfo->srcRowIndex = -1;
127✔
1465
      } break;
127✔
1466
      case STREAM_CHECKPOINT: {
1✔
1467
        pInfo->stateStore.streamStateCommit(pInfo->pState);
1✔
1468
        (*ppRes) = pBlock;
1✔
1469
        goto _end;
1✔
1470
      } break;
1471
      case STREAM_CREATE_CHILD_TABLE: {
3✔
1472
        (*ppRes) = pBlock;
3✔
1473
        goto _end;
3✔
1474
      } break;
1475
      case STREAM_GET_RESULT: {
2,050✔
1476
        void* pPushRes = taosArrayPush(pInfo->pCloseTs, &pBlock->info.window.skey);
1,025✔
1477
        QUERY_CHECK_NULL(pPushRes, code, lino, _end, terrno);
1,025!
1478
        continue;
1,025✔
1479
      }
1480
      default:
×
1481
        code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1482
        QUERY_CHECK_CODE(code, lino, _end);
×
1483
    }
1484

1485
    code = doStreamForceFillImpl(pOperator);
127✔
1486
    QUERY_CHECK_CODE(code, lino, _end);
127!
1487
  }
1488

1489
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->pCloseTs); i++) {
2,051✔
1490
    TSKEY ts = *(TSKEY*) taosArrayGet(pInfo->pCloseTs, i);
1,025✔
1491
    code = buildAllResultKey(&pInfo->stateStore, pInfo->pState, ts, pInfo->pUpdated);
1,025✔
1492
    QUERY_CHECK_CODE(code, lino, _end);
1,025!
1493
  }
1494
  taosArrayClear(pInfo->pCloseTs);
1,026✔
1495
  removeDuplicateResult(pInfo->pUpdated, winKeyCmprImpl);
1,026✔
1496

1497
  initMultiResInfoFromArrayList(&pInfo->groupResInfo, pInfo->pUpdated);
1,026✔
1498
  pInfo->groupResInfo.freeItem = false;
1,026✔
1499

1500
  pInfo->pUpdated = taosArrayInit(1024, sizeof(SWinKey));
1,026✔
1501
  QUERY_CHECK_NULL(pInfo->pUpdated, code, lino, _end, terrno);
1,026!
1502

1503
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
1,026✔
1504
  QUERY_CHECK_CODE(code, lino, _end);
1,026!
1505

1506
  code = buildForceFillResult(pOperator, ppRes);
1,026✔
1507
  QUERY_CHECK_CODE(code, lino, _end);
1,025!
1508

1509
  if ((*ppRes) == NULL) {
1,025✔
1510
    pInfo->stateStore.streamStateClearExpiredState(pInfo->pState, 1, INT64_MAX);
656✔
1511
    resetStreamFillInfo(pInfo);
657✔
1512
    setStreamOperatorCompleted(pOperator);
657✔
1513
  }
1514

1515
_end:
369✔
1516
  if (code != TSDB_CODE_SUCCESS) {
1,557!
1517
    qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
1518
    pTaskInfo->code = code;
×
1519
  }
1520
  return code;
1,557✔
1521
}
1522

1523
static int32_t initResultBuf(SSDataBlock* pInputRes, SStreamFillSupporter* pFillSup) {
253✔
1524
  int32_t numOfCols = taosArrayGetSize(pInputRes->pDataBlock);
253✔
1525
  pFillSup->rowSize = sizeof(SResultCellData) * numOfCols;
253✔
1526
  for (int i = 0; i < numOfCols; i++) {
3,698✔
1527
    SColumnInfoData* pCol = taosArrayGet(pInputRes->pDataBlock, i);
3,445✔
1528
    pFillSup->rowSize += pCol->info.bytes;
3,445✔
1529
  }
1530
  pFillSup->next.key = INT64_MIN;
253✔
1531
  pFillSup->nextNext.key = INT64_MIN;
253✔
1532
  pFillSup->prev.key = INT64_MIN;
253✔
1533
  pFillSup->cur.key = INT64_MIN;
253✔
1534
  pFillSup->next.pRowVal = NULL;
253✔
1535
  pFillSup->nextNext.pRowVal = NULL;
253✔
1536
  pFillSup->prev.pRowVal = NULL;
253✔
1537
  pFillSup->cur.pRowVal = NULL;
253✔
1538

1539
  return TSDB_CODE_SUCCESS;
253✔
1540
}
1541

1542
static SStreamFillSupporter* initStreamFillSup(SStreamFillPhysiNode* pPhyFillNode, SInterval* pInterval,
253✔
1543
                                               SExprInfo* pFillExprInfo, int32_t numOfFillCols, SStorageAPI* pAPI, SSDataBlock* pInputRes) {
1544
  int32_t               code = TSDB_CODE_SUCCESS;
253✔
1545
  int32_t               lino = 0;
253✔
1546
  SStreamFillSupporter* pFillSup = taosMemoryCalloc(1, sizeof(SStreamFillSupporter));
253!
1547
  if (!pFillSup) {
253!
1548
    code = terrno;
×
1549
    QUERY_CHECK_CODE(code, lino, _end);
×
1550
  }
1551
  pFillSup->numOfFillCols = numOfFillCols;
253✔
1552
  int32_t    numOfNotFillCols = 0;
253✔
1553
  SExprInfo* noFillExprInfo = NULL;
253✔
1554

1555
  code = createExprInfo(pPhyFillNode->pNotFillExprs, NULL, &noFillExprInfo, &numOfNotFillCols);
253✔
1556
  QUERY_CHECK_CODE(code, lino, _end);
253!
1557

1558
  pFillSup->pAllColInfo = createFillColInfo(pFillExprInfo, pFillSup->numOfFillCols, noFillExprInfo, numOfNotFillCols,
506✔
1559
                                            NULL, 0, (const SNodeListNode*)(pPhyFillNode->pValues));
253✔
1560
  if (pFillSup->pAllColInfo == NULL) {
253!
1561
    code = terrno;
×
1562
    lino = __LINE__;
×
1563
    destroyExprInfo(noFillExprInfo, numOfNotFillCols);
×
1564
    goto _end;
×
1565
  }
1566

1567
  pFillSup->type = convertFillType(pPhyFillNode->mode);
253✔
1568
  pFillSup->numOfAllCols = pFillSup->numOfFillCols + numOfNotFillCols;
253✔
1569
  pFillSup->interval = *pInterval;
253✔
1570
  pFillSup->pAPI = pAPI;
253✔
1571

1572
  code = initResultBuf(pInputRes, pFillSup);
253✔
1573
  QUERY_CHECK_CODE(code, lino, _end);
253!
1574

1575
  SExprInfo* noFillExpr = NULL;
253✔
1576
  code = createExprInfo(pPhyFillNode->pNotFillExprs, NULL, &noFillExpr, &numOfNotFillCols);
253✔
1577
  QUERY_CHECK_CODE(code, lino, _end);
253!
1578

1579
  code = initExprSupp(&pFillSup->notFillExprSup, noFillExpr, numOfNotFillCols, &pAPI->functionStore);
253✔
1580
  QUERY_CHECK_CODE(code, lino, _end);
253!
1581

1582
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);
253✔
1583
  pFillSup->pResMap = tSimpleHashInit(16, hashFn);
253✔
1584
  QUERY_CHECK_NULL(pFillSup->pResMap, code, lino, _end, terrno);
253!
1585
  pFillSup->hasDelete = false;
253✔
1586
  pFillSup->normalFill = true;
253✔
1587
  pFillSup->pResultRange = taosArrayInit(2, POINTER_BYTES);
253✔
1588

1589

1590
_end:
253✔
1591
  if (code != TSDB_CODE_SUCCESS) {
253!
1592
    destroyStreamFillSupporter(pFillSup);
×
1593
    pFillSup = NULL;
×
1594
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1595
  }
1596
  return pFillSup;
253✔
1597
}
1598

1599
SStreamFillInfo* initStreamFillInfo(SStreamFillSupporter* pFillSup, SSDataBlock* pRes) {
495✔
1600
  int32_t          code = TSDB_CODE_SUCCESS;
495✔
1601
  int32_t          lino = 0;
495✔
1602
  SStreamFillInfo* pFillInfo = taosMemoryCalloc(1, sizeof(SStreamFillInfo));
495!
1603
  if (!pFillInfo) {
495!
1604
    code = terrno;
×
1605
    QUERY_CHECK_CODE(code, lino, _end);
×
1606
  }
1607

1608
  pFillInfo->start = INT64_MIN;
495✔
1609
  pFillInfo->current = INT64_MIN;
495✔
1610
  pFillInfo->end = INT64_MIN;
495✔
1611
  pFillInfo->preRowKey = INT64_MIN;
495✔
1612
  pFillInfo->needFill = false;
495✔
1613
  pFillInfo->pLinearInfo = taosMemoryCalloc(1, sizeof(SStreamFillLinearInfo));
495!
1614
  if (!pFillInfo) {
495!
1615
    code = terrno;
×
1616
    QUERY_CHECK_CODE(code, lino, _end);
×
1617
  }
1618

1619
  pFillInfo->pLinearInfo->hasNext = false;
495✔
1620
  pFillInfo->pLinearInfo->nextEnd = INT64_MIN;
495✔
1621
  pFillInfo->pLinearInfo->pEndPoints = NULL;
495✔
1622
  pFillInfo->pLinearInfo->pNextEndPoints = NULL;
495✔
1623
  if (pFillSup->type == TSDB_FILL_LINEAR) {
495✔
1624
    pFillInfo->pLinearInfo->pEndPoints = taosArrayInit(pFillSup->numOfAllCols, sizeof(SPoint));
70✔
1625
    if (!pFillInfo->pLinearInfo->pEndPoints) {
70!
1626
      code = terrno;
×
1627
      QUERY_CHECK_CODE(code, lino, _end);
×
1628
    }
1629

1630
    pFillInfo->pLinearInfo->pNextEndPoints = taosArrayInit(pFillSup->numOfAllCols, sizeof(SPoint));
70✔
1631
    if (!pFillInfo->pLinearInfo->pNextEndPoints) {
70!
1632
      code = terrno;
×
1633
      QUERY_CHECK_CODE(code, lino, _end);
×
1634
    }
1635

1636
    for (int32_t i = 0; i < pFillSup->numOfAllCols; i++) {
806✔
1637
      SColumnInfoData* pColData = taosArrayGet(pRes->pDataBlock, i);
736✔
1638
      if (pColData == NULL) {
736✔
1639
        SPoint dummy = {0};
16✔
1640
        dummy.val = taosMemoryCalloc(1, 1);
16!
1641
        void* tmpRes = taosArrayPush(pFillInfo->pLinearInfo->pEndPoints, &dummy);
16✔
1642
        QUERY_CHECK_NULL(tmpRes, code, lino, _end, terrno);
16!
1643

1644
        dummy.val = taosMemoryCalloc(1, 1);
16!
1645
        tmpRes = taosArrayPush(pFillInfo->pLinearInfo->pNextEndPoints, &dummy);
16✔
1646
        QUERY_CHECK_NULL(tmpRes, code, lino, _end, terrno);
16!
1647

1648
        continue;
16✔
1649
      }
1650
      SPoint value = {0};
720✔
1651
      value.val = taosMemoryCalloc(1, pColData->info.bytes);
720!
1652
      QUERY_CHECK_NULL(value.val, code, lino, _end, terrno);
720!
1653

1654
      void* tmpRes = taosArrayPush(pFillInfo->pLinearInfo->pEndPoints, &value);
720✔
1655
      QUERY_CHECK_NULL(tmpRes, code, lino, _end, terrno);
720!
1656

1657
      value.val = taosMemoryCalloc(1, pColData->info.bytes);
720!
1658
      QUERY_CHECK_NULL(value.val, code, lino, _end, terrno);
720!
1659

1660
      tmpRes = taosArrayPush(pFillInfo->pLinearInfo->pNextEndPoints, &value);
720✔
1661
      QUERY_CHECK_NULL(tmpRes, code, lino, _end, terrno);
720!
1662
    }
1663
  }
1664
  pFillInfo->pLinearInfo->winIndex = 0;
495✔
1665

1666
  pFillInfo->pNonFillRow = NULL;
495✔
1667
  pFillInfo->pResRow = NULL;
495✔
1668
  if (pFillSup->type == TSDB_FILL_SET_VALUE || pFillSup->type == TSDB_FILL_SET_VALUE_F ||
495✔
1669
      pFillSup->type == TSDB_FILL_NULL || pFillSup->type == TSDB_FILL_NULL_F) {
392✔
1670
    pFillInfo->pResRow = taosMemoryCalloc(1, sizeof(SResultRowData));
215!
1671
    QUERY_CHECK_NULL(pFillInfo->pResRow, code, lino, _end, terrno);
215!
1672

1673
    pFillInfo->pResRow->key = INT64_MIN;
215✔
1674
    pFillInfo->pResRow->pRowVal = taosMemoryCalloc(1, pFillSup->rowSize);
215!
1675
    QUERY_CHECK_NULL(pFillInfo->pResRow->pRowVal, code, lino, _end, terrno);
215!
1676

1677
    for (int32_t i = 0; i < pFillSup->numOfAllCols; ++i) {
2,209✔
1678
      SColumnInfoData* pColData = taosArrayGet(pRes->pDataBlock, i);
1,994✔
1679
      SResultCellData* pCell = getResultCell(pFillInfo->pResRow, i);
1,994✔
1680
      if (pColData == NULL) {
1,994✔
1681
        pCell->bytes = 1;
73✔
1682
        pCell->type = 4;
73✔
1683
        continue;
73✔
1684
      }
1685
      pCell->bytes = pColData->info.bytes;
1,921✔
1686
      pCell->type = pColData->info.type;
1,921✔
1687
    }
1688

1689
    int32_t numOfResCol = taosArrayGetSize(pRes->pDataBlock);
215✔
1690
    if (numOfResCol < pFillSup->numOfAllCols) {
215✔
1691
      int32_t* pTmpBuf = (int32_t*)taosMemoryRealloc(pFillSup->pOffsetInfo, pFillSup->numOfAllCols * sizeof(int32_t));
69!
1692
      QUERY_CHECK_NULL(pTmpBuf, code, lino, _end, terrno);
69!
1693
      pFillSup->pOffsetInfo = pTmpBuf;
69✔
1694

1695
      SResultCellData* pCell = getResultCell(pFillInfo->pResRow, numOfResCol - 1);
69✔
1696
      int32_t preLength = pFillSup->pOffsetInfo[numOfResCol - 1] + pCell->bytes + sizeof(SResultCellData);
69✔
1697
      for (int32_t i = numOfResCol; i < pFillSup->numOfAllCols; i++) {
142✔
1698
        pFillSup->pOffsetInfo[i] = preLength;
73✔
1699
        pCell = getResultCell(pFillInfo->pResRow, i);
73✔
1700
        preLength += pCell->bytes + sizeof(SResultCellData);
73✔
1701
      }
1702
    }
1703

1704
    pFillInfo->pNonFillRow = taosMemoryCalloc(1, sizeof(SResultRowData));
215!
1705
    QUERY_CHECK_NULL(pFillInfo->pNonFillRow, code, lino, _end, terrno);
215!
1706
    pFillInfo->pNonFillRow->key = INT64_MIN;
215✔
1707
    pFillInfo->pNonFillRow->pRowVal = taosMemoryCalloc(1, pFillSup->rowSize);
215!
1708
    memcpy(pFillInfo->pNonFillRow->pRowVal, pFillInfo->pResRow->pRowVal, pFillSup->rowSize);
215✔
1709
  }
1710

1711
  pFillInfo->type = pFillSup->type;
495✔
1712
  pFillInfo->delRanges = taosArrayInit(16, sizeof(STimeFillRange));
495✔
1713
  if (!pFillInfo->delRanges) {
495!
1714
    code = terrno;
×
1715
    QUERY_CHECK_CODE(code, lino, _end);
×
1716
  }
1717

1718
  pFillInfo->delIndex = 0;
495✔
1719
  pFillInfo->curGroupId = 0;
495✔
1720
  pFillInfo->hasNext = false;
495✔
1721
  pFillInfo->pTempBuff = taosMemoryCalloc(1, pFillSup->rowSize);
495!
1722
  return pFillInfo;
495✔
1723

1724
_end:
×
1725
  if (code != TSDB_CODE_SUCCESS) {
×
1726
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1727
  }
1728
  destroyStreamFillInfo(pFillInfo);
×
1729
  return NULL;
×
1730
}
1731

1732
static void setValueForFillInfo(SStreamFillSupporter* pFillSup, SStreamFillInfo* pFillInfo) {
253✔
1733
  if (pFillInfo->type == TSDB_FILL_SET_VALUE || pFillInfo->type == TSDB_FILL_SET_VALUE_F) {
253✔
1734
    for (int32_t i = 0; i < pFillSup->numOfAllCols; ++i) {
713✔
1735
      SFillColInfo*    pFillCol = pFillSup->pAllColInfo + i;
662✔
1736
      int32_t          slotId = GET_DEST_SLOT_ID(pFillCol);
662✔
1737
      SResultCellData* pCell = getResultCell(pFillInfo->pResRow, slotId);
662✔
1738
      SVariant*        pVar = &(pFillCol->fillVal);
662✔
1739
      if (pCell->type == TSDB_DATA_TYPE_FLOAT) {
662!
1740
        float v = 0;
×
1741
        GET_TYPED_DATA(v, float, pVar->nType, &pVar->i, 0);
×
1742
        SET_TYPED_DATA(pCell->pData, pCell->type, v);
×
1743
      } else if (IS_FLOAT_TYPE(pCell->type)) {
870!
1744
        double v = 0;
208✔
1745
        GET_TYPED_DATA(v, double, pVar->nType, &pVar->i, 0);
208!
1746
        SET_TYPED_DATA(pCell->pData, pCell->type, v);
208!
1747
      } else if (IS_INTEGER_TYPE(pCell->type)) {
857!
1748
        int64_t v = 0;
403✔
1749
        GET_TYPED_DATA(v, int64_t, pVar->nType, &pVar->i, 0);
403!
1750
        SET_TYPED_DATA(pCell->pData, pCell->type, v);
403!
1751
      } else {
1752
        pCell->isNull = true;
51✔
1753
      }
1754
    }
1755
  } else if (pFillInfo->type == TSDB_FILL_NULL || pFillInfo->type == TSDB_FILL_NULL_F) {
202✔
1756
    for (int32_t i = 0; i < pFillSup->numOfAllCols; ++i) {
980✔
1757
      SFillColInfo*    pFillCol = pFillSup->pAllColInfo + i;
910✔
1758
      int32_t          slotId = GET_DEST_SLOT_ID(pFillCol);
910✔
1759
      SResultCellData* pCell = getResultCell(pFillInfo->pResRow, slotId);
910✔
1760
      pCell->isNull = true;
910✔
1761
    }
1762
  }
1763
}
253✔
1764

1765
int32_t getDownStreamInfo(SOperatorInfo* downstream, int8_t* triggerType, SInterval* pInterval,
253✔
1766
                          int16_t* pOperatorFlag) {
1767
  int32_t code = TSDB_CODE_SUCCESS;
253✔
1768
  int32_t lino = 0;
253✔
1769
  if (IS_NORMAL_INTERVAL_OP(downstream)) {
253✔
1770
    SStreamIntervalOperatorInfo* pInfo = downstream->info;
209✔
1771
    *triggerType = pInfo->twAggSup.calTrigger;
209✔
1772
    *pInterval = pInfo->interval;
209✔
1773
    *pOperatorFlag = pInfo->basic.operatorFlag;
209✔
1774
  } else {
1775
    SStreamIntervalSliceOperatorInfo* pInfo = downstream->info;
44✔
1776
    *triggerType = pInfo->twAggSup.calTrigger;
44✔
1777
    *pInterval = pInfo->interval;
44✔
1778
    pInfo->hasFill = true;
44✔
1779
    *pOperatorFlag = pInfo->basic.operatorFlag;
44✔
1780
  }
1781

1782
  QUERY_CHECK_CODE(code, lino, _end);
253!
1783

1784
_end:
253✔
1785
  if (code != TSDB_CODE_SUCCESS) {
253!
1786
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1787
  }
1788
  return code;
253✔
1789
}
1790

1791
int32_t initFillOperatorStateBuff(SStreamFillOperatorInfo* pInfo, SStreamState* pState, SStateStore* pStore,
44✔
1792
                                  SReadHandle* pHandle, const char* taskIdStr, SStorageAPI* pApi) {
1793
  int32_t code = TSDB_CODE_SUCCESS;
44✔
1794
  int32_t lino = 0;
44✔
1795

1796
  pInfo->stateStore = *pStore;
44✔
1797
  pInfo->pState = taosMemoryCalloc(1, sizeof(SStreamState));
44!
1798
  QUERY_CHECK_NULL(pInfo->pState, code, lino, _end, terrno);
44!
1799

1800
  *(pInfo->pState) = *pState;
44✔
1801
  pInfo->stateStore.streamStateSetNumber(pInfo->pState, -1, pInfo->primaryTsCol);
44✔
1802
  code = pInfo->stateStore.streamFileStateInit(tsStreamBufferSize, sizeof(SWinKey), pInfo->pFillSup->rowSize, 0, compareTs,
44✔
1803
                                               pInfo->pState, INT64_MAX, taskIdStr, pHandle->checkpointId,
44✔
1804
                                               STREAM_STATE_BUFF_HASH_SORT, &pInfo->pState->pFileState);
44✔
1805
  QUERY_CHECK_CODE(code, lino, _end);
44!
1806

1807
_end:
44✔
1808
  if (code != TSDB_CODE_SUCCESS) {
44!
1809
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1810
  }
1811
  return code;
44✔
1812
}
1813

1814
int32_t createStreamFillOperatorInfo(SOperatorInfo* downstream, SStreamFillPhysiNode* pPhyFillNode,
253✔
1815
                                     SExecTaskInfo* pTaskInfo, SReadHandle* pHandle, SOperatorInfo** pOptrInfo) {
1816
  QRY_PARAM_CHECK(pOptrInfo);
253!
1817

1818
  int32_t                  code = TSDB_CODE_SUCCESS;
253✔
1819
  int32_t                  lino = 0;
253✔
1820
  SStreamFillOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamFillOperatorInfo));
253!
1821
  SOperatorInfo*           pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
253!
1822
  if (pInfo == NULL || pOperator == NULL) {
253!
1823
    code = terrno;
×
1824
    QUERY_CHECK_CODE(code, lino, _error);
×
1825
  }
1826

1827
  int32_t    numOfFillCols = 0;
253✔
1828
  SExprInfo* pFillExprInfo = NULL;
253✔
1829

1830
  code = createExprInfo(pPhyFillNode->pFillExprs, NULL, &pFillExprInfo, &numOfFillCols);
253✔
1831
  QUERY_CHECK_CODE(code, lino, _error);
253!
1832

1833
  code = initExprSupp(&pOperator->exprSupp, pFillExprInfo, numOfFillCols, &pTaskInfo->storageAPI.functionStore);
253✔
1834
  QUERY_CHECK_CODE(code, lino, _error);
253!
1835

1836
  pInfo->pSrcBlock = createDataBlockFromDescNode(pPhyFillNode->node.pOutputDataBlockDesc);
253✔
1837
  QUERY_CHECK_NULL(pInfo->pSrcBlock, code, lino, _error, terrno);
253!
1838

1839
  int8_t triggerType = 0;
253✔
1840
  SInterval interval = {0};
253✔
1841
  int16_t opFlag = 0;
253✔
1842
  code = getDownStreamInfo(downstream, &triggerType, &interval, &opFlag);
253✔
1843
  QUERY_CHECK_CODE(code, lino, _error);
253!
1844

1845
  pInfo->pFillSup = initStreamFillSup(pPhyFillNode, &interval, pFillExprInfo, numOfFillCols, &pTaskInfo->storageAPI,
253✔
1846
                                      pInfo->pSrcBlock);
1847
  if (!pInfo->pFillSup) {
253!
1848
    code = TSDB_CODE_FAILED;
×
1849
    QUERY_CHECK_CODE(code, lino, _error);
×
1850
  }
1851

1852
  initResultSizeInfo(&pOperator->resultInfo, 4096);
253✔
1853
  pInfo->pRes = createDataBlockFromDescNode(pPhyFillNode->node.pOutputDataBlockDesc);
253✔
1854
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
253!
1855

1856
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
253✔
1857
  QUERY_CHECK_CODE(code, lino, _error);
253!
1858

1859
  code = blockDataEnsureCapacity(pInfo->pSrcBlock, pOperator->resultInfo.capacity);
253✔
1860
  QUERY_CHECK_CODE(code, lino, _error);
253!
1861

1862
  pInfo->pFillInfo = initStreamFillInfo(pInfo->pFillSup, pInfo->pRes);
253✔
1863
  if (!pInfo->pFillInfo) {
253!
1864
    goto _error;
×
1865
  }
1866

1867
  setValueForFillInfo(pInfo->pFillSup, pInfo->pFillInfo);
253✔
1868

1869
  code = createSpecialDataBlock(STREAM_DELETE_RESULT, &pInfo->pDelRes);
253✔
1870
  QUERY_CHECK_CODE(code, lino, _error);
253!
1871

1872
  code = blockDataEnsureCapacity(pInfo->pDelRes, pOperator->resultInfo.capacity);
253✔
1873
  QUERY_CHECK_CODE(code, lino, _error);
253!
1874

1875
  pInfo->pUpdated = taosArrayInit(1024, sizeof(SWinKey));
253✔
1876
  QUERY_CHECK_NULL(pInfo->pUpdated, code, lino, _error, terrno);
253!
1877

1878
  pInfo->pCloseTs = taosArrayInit(1024, sizeof(TSKEY));
253✔
1879
  QUERY_CHECK_NULL(pInfo->pCloseTs, code, lino, _error, terrno);
253!
1880

1881
  pInfo->primaryTsCol = ((STargetNode*)pPhyFillNode->pWStartTs)->slotId;
253✔
1882
  pInfo->primarySrcSlotId = ((SColumnNode*)((STargetNode*)pPhyFillNode->pWStartTs)->pExpr)->slotId;
253✔
1883

1884
  int32_t numOfOutputCols = 0;
253✔
1885
  code = extractColMatchInfo(pPhyFillNode->pFillExprs, pPhyFillNode->node.pOutputDataBlockDesc, &numOfOutputCols,
253✔
1886
                             COL_MATCH_FROM_SLOT_ID, &pInfo->matchInfo);
1887
  QUERY_CHECK_CODE(code, lino, _error);
253!
1888

1889
  code = filterInitFromNode((SNode*)pPhyFillNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
253✔
1890
  QUERY_CHECK_CODE(code, lino, _error);
253!
1891

1892
  pInfo->srcRowIndex = -1;
253✔
1893
  setOperatorInfo(pOperator, "StreamFillOperator", nodeType(pPhyFillNode), false, OP_NOT_OPENED, pInfo,
253✔
1894
                  pTaskInfo);
1895

1896
  if (triggerType == STREAM_TRIGGER_FORCE_WINDOW_CLOSE) {
253✔
1897
    code = initFillOperatorStateBuff(pInfo, pTaskInfo->streamInfo.pState, &pTaskInfo->storageAPI.stateStore, pHandle,
44✔
1898
                              GET_TASKID(pTaskInfo), &pTaskInfo->storageAPI);
44✔
1899
    QUERY_CHECK_CODE(code, lino, _error);
44!
1900
    pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doStreamForceFillNext, NULL, destroyStreamFillOperatorInfo,
44✔
1901
                                           optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
1902
  } else if (triggerType == STREAM_TRIGGER_CONTINUOUS_WINDOW_CLOSE) {
209!
1903
    code = initFillOperatorStateBuff(pInfo, pTaskInfo->streamInfo.pState, &pTaskInfo->storageAPI.stateStore, pHandle,
×
1904
                              GET_TASKID(pTaskInfo), &pTaskInfo->storageAPI);
×
1905
    QUERY_CHECK_CODE(code, lino, _error);
×
1906

1907
    initNonBlockAggSupptor(&pInfo->nbSup, &pInfo->pFillSup->interval, downstream);
×
1908
    code = initStreamBasicInfo(&pInfo->basic, pOperator);
×
1909
    QUERY_CHECK_CODE(code, lino, _error);
×
1910

1911
    code = streamClientCheckCfg(&pInfo->nbSup.recParam);
×
1912
    QUERY_CHECK_CODE(code, lino, _error);
×
1913

1914
    pInfo->basic.operatorFlag = opFlag;
×
1915
    if (isFinalOperator(&pInfo->basic)) {
×
1916
      pInfo->nbSup.numOfKeep++;
×
1917
    }
1918
    code = initFillSupRowInfo(pInfo->pFillSup, pInfo->pRes);
×
1919
    QUERY_CHECK_CODE(code, lino, _error);
×
1920
    pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doStreamNonblockFillNext, NULL, destroyStreamNonblockFillOperatorInfo,
×
1921
                                           optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
1922
  } else {
1923
    pInfo->pState = NULL;
209✔
1924
    pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doStreamFillNext, NULL, destroyStreamFillOperatorInfo,
209✔
1925
                                           optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
1926
  }
1927
  setOperatorStreamStateFn(pOperator, streamOpReleaseState, streamOpReloadState);
253✔
1928

1929
  code = appendDownstream(pOperator, &downstream, 1);
253✔
1930
  QUERY_CHECK_CODE(code, lino, _error);
253!
1931

1932
  *pOptrInfo = pOperator;
253✔
1933
  return TSDB_CODE_SUCCESS;
253✔
1934

1935
_error:
×
1936
  qError("%s failed at line %d since %s. task:%s", __func__, lino, tstrerror(code), GET_TASKID(pTaskInfo));
×
1937

1938
  if (pInfo != NULL) destroyStreamFillOperatorInfo(pInfo);
×
1939
  destroyOperatorAndDownstreams(pOperator, &downstream, 1);
×
1940
  pTaskInfo->code = code;
×
1941
  return code;
×
1942
}
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