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

taosdata / TDengine / #3653

14 Mar 2025 08:10AM UTC coverage: 22.565% (-41.0%) from 63.596%
#3653

push

travis-ci

web-flow
feat(keep): support keep on super table level. (#30097)

* Feat: support use keep while create super table.

* Test(keep): add test for create super table with keep option.

* Feat(keep): Add tmsg for create keep.

* Feat(keep): support alter table option keep.

* Fix(keep): Add baisc test for alter table option.

* Fix(keep): memory leek.

* Feat(keep): add keep to metaEntry&metaCache and fix earliestTs with stn keep.

* Test(keep): add some cases for select with stb keep.

* Fix: fix ci core while alter stb.

* Feat(keep): delete expired data in super table level.

* Feat: remove get stb keep while query.

* Fix : build error.

* Revert "Fix : build error."

This reverts commit 0ed66e4e8.

* Revert "Feat(keep): delete expired data in super table level."

This reverts commit 36330f6b4.

* Fix : build errors.

* Feat : support restart taosd.

* Fix : alter table comment problems.

* Test : add tests for super table keep.

* Fix: change sdb stb reserve size.

* Test: add more tests.

* Feat: Disable normal tables and sub tables from setting the keep parameter

* Fix: add more checks to avoid unknown address.

* Docs: Add docs for stable keep.

* Fix: some review changes.

* Fix: review errors.

49248 of 302527 branches covered (16.28%)

Branch coverage included in aggregate %.

53 of 99 new or added lines in 12 files covered. (53.54%)

155872 existing lines in 443 files now uncovered.

87359 of 302857 relevant lines covered (28.84%)

570004.22 hits per line

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

10.37
/source/dnode/mnode/impl/src/mndStreamUtil.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 "mndDb.h"
17
#include "mndStb.h"
18
#include "mndStream.h"
19
#include "mndTrans.h"
20
#include "mndVgroup.h"
21
#include "taoserror.h"
22
#include "tmisce.h"
23

24
struct SStreamTaskIter {
25
  SStreamObj  *pStream;
26
  int32_t      level;
27
  int32_t      ordinalIndex;
28
  int32_t      totalLevel;
29
  SStreamTask *pTask;
30
};
31

32
int32_t doRemoveTasks(SStreamExecInfo *pExecNode, STaskId *pRemovedId);
33

34
int32_t createStreamTaskIter(SStreamObj *pStream, SStreamTaskIter **pIter) {
1✔
35
  *pIter = taosMemoryCalloc(1, sizeof(SStreamTaskIter));
1!
36
  if (*pIter == NULL) {
1!
37
    return terrno;
×
38
  }
39

40
  (*pIter)->level = -1;
1✔
41
  (*pIter)->ordinalIndex = 0;
1✔
42
  (*pIter)->pStream = pStream;
1✔
43
  (*pIter)->totalLevel = taosArrayGetSize(pStream->tasks);
1✔
44
  (*pIter)->pTask = NULL;
1✔
45

46
  return 0;
1✔
47
}
48

49
bool streamTaskIterNextTask(SStreamTaskIter *pIter) {
2✔
50
  if (pIter->level >= pIter->totalLevel) {
2!
51
    pIter->pTask = NULL;
×
52
    return false;
×
53
  }
54

55
  if (pIter->level == -1) {
2✔
56
    pIter->level += 1;
1✔
57
  }
58

59
  while (pIter->level < pIter->totalLevel) {
3✔
60
    SArray *pList = taosArrayGetP(pIter->pStream->tasks, pIter->level);
2✔
61
    if (pIter->ordinalIndex >= taosArrayGetSize(pList)) {
2✔
62
      pIter->level += 1;
1✔
63
      pIter->ordinalIndex = 0;
1✔
64
      pIter->pTask = NULL;
1✔
65
      continue;
1✔
66
    }
67

68
    pIter->pTask = taosArrayGetP(pList, pIter->ordinalIndex);
1✔
69
    pIter->ordinalIndex += 1;
1✔
70
    return true;
1✔
71
  }
72

73
  pIter->pTask = NULL;
1✔
74
  return false;
1✔
75
}
76

77
int32_t streamTaskIterGetCurrent(SStreamTaskIter *pIter, SStreamTask **pTask) {
1✔
78
  if (pTask) {
1!
79
    *pTask = pIter->pTask;
1✔
80
    if (*pTask != NULL) {
1!
81
      return TSDB_CODE_SUCCESS;
1✔
82
    }
83
  }
84

85
  return TSDB_CODE_INVALID_PARA;
×
86
}
87

88
void destroyStreamTaskIter(SStreamTaskIter *pIter) { taosMemoryFree(pIter); }
1!
89

90
static bool checkStatusForEachReplica(SVgObj *pVgroup) {
1✔
91
  for (int32_t i = 0; i < pVgroup->replica; ++i) {
1!
92
    if (!pVgroup->vnodeGid[i].syncRestore) {
1!
93
      mInfo("vgId:%d not restored, not ready for checkpoint or other operations", pVgroup->vgId);
1!
94
      return false;
1✔
95
    }
96

UNCOV
97
    ESyncState state = pVgroup->vnodeGid[i].syncState;
×
UNCOV
98
    if (state == TAOS_SYNC_STATE_OFFLINE || state == TAOS_SYNC_STATE_ERROR || state == TAOS_SYNC_STATE_LEARNER ||
×
99
        state == TAOS_SYNC_STATE_CANDIDATE) {
UNCOV
100
      mInfo("vgId:%d state:%d , not ready for checkpoint or other operations, not check other vgroups", pVgroup->vgId,
×
101
            state);
UNCOV
102
      return false;
×
103
    }
104
  }
105

UNCOV
106
  return true;
×
107
}
108

109
static int32_t mndAddSnodeInfo(SMnode *pMnode, SArray *pVgroupList) {
2✔
110
  SSnodeObj *pObj = NULL;
2✔
111
  void      *pIter = NULL;
2✔
112
  int32_t    code = 0;
2✔
113

UNCOV
114
  while (1) {
×
115
    pIter = sdbFetch(pMnode->pSdb, SDB_SNODE, pIter, (void **)&pObj);
2✔
116
    if (pIter == NULL) {
2!
117
      break;
2✔
118
    }
119

UNCOV
120
    SNodeEntry entry = {.nodeId = SNODE_HANDLE};
×
UNCOV
121
    code = addEpIntoEpSet(&entry.epset, pObj->pDnode->fqdn, pObj->pDnode->port);
×
UNCOV
122
    if (code) {
×
123
      sdbRelease(pMnode->pSdb, pObj);
×
124
      sdbCancelFetch(pMnode->pSdb, pIter);
×
125
      mError("failed to extract epset for fqdn:%s during task vgroup snapshot", pObj->pDnode->fqdn);
×
126
      return code;
×
127
    }
128

UNCOV
129
    char buf[256] = {0};
×
UNCOV
130
    code = epsetToStr(&entry.epset, buf, tListLen(buf));
×
UNCOV
131
    if (code != 0) {  // print error and continue
×
132
      mError("failed to convert epset to str, code:%s", tstrerror(code));
×
133
    }
134

UNCOV
135
    void *p = taosArrayPush(pVgroupList, &entry);
×
UNCOV
136
    if (p == NULL) {
×
137
      code = terrno;
×
138
      sdbRelease(pMnode->pSdb, pObj);
×
139
      sdbCancelFetch(pMnode->pSdb, pIter);
×
140
      mError("failed to put entry in vgroup list, nodeId:%d code:%s", entry.nodeId, tstrerror(code));
×
141
      return code;
×
142
    } else {
UNCOV
143
      mDebug("take snode snapshot, nodeId:%d %s", entry.nodeId, buf);
×
144
    }
145

UNCOV
146
    sdbRelease(pMnode->pSdb, pObj);
×
147
  }
148

149
  return code;
2✔
150
}
151

152
static int32_t mndCheckMnodeStatus(SMnode* pMnode) {
2✔
153
  int32_t    code = 0;
2✔
154
  ESdbStatus objStatus;
155
  void      *pIter = NULL;
2✔
156
  SMnodeObj *pObj = NULL;
2✔
157

158
  while (1) {
159
    pIter = sdbFetchAll(pMnode->pSdb, SDB_MNODE, pIter, (void **)&pObj, &objStatus, true);
4✔
160
    if (pIter == NULL) {
4✔
161
      break;
2✔
162
    }
163

164
    if (pObj->syncState != TAOS_SYNC_STATE_LEADER && pObj->syncState != TAOS_SYNC_STATE_FOLLOWER) {
2!
UNCOV
165
      mDebug("mnode sync state:%d not leader/follower", pObj->syncState);
×
UNCOV
166
      sdbRelease(pMnode->pSdb, pObj);
×
UNCOV
167
      sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
168
      return TSDB_CODE_FAILED;
×
169
    }
170

171
    if (objStatus != SDB_STATUS_READY) {
2!
UNCOV
172
      mWarn("mnode status:%d not ready", objStatus);
×
UNCOV
173
      sdbRelease(pMnode->pSdb, pObj);
×
UNCOV
174
      sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
175
      return TSDB_CODE_FAILED;
×
176
    }
177

178
    sdbRelease(pMnode->pSdb, pObj);
2✔
179
  }
180

181
  return TSDB_CODE_SUCCESS;
2✔
182
}
183

184
static int32_t mndCheckAndAddVgroupsInfo(SMnode *pMnode, SArray *pVgroupList, bool* allReady) {
2✔
185
  SSdb     *pSdb = pMnode->pSdb;
2✔
186
  void     *pIter = NULL;
2✔
187
  SVgObj   *pVgroup = NULL;
2✔
188
  int32_t   code = 0;
2✔
189
  SHashObj *pHash = NULL;
2✔
190

191
  pHash = taosHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2✔
192
  if (pHash == NULL) {
2!
193
    mError("failed to prepare hashmap during take vgroup snapshot, code:%s", tstrerror(terrno));
×
194
    return terrno;
×
195
  }
196

197
  while (1) {
2✔
198
    pIter = sdbFetch(pSdb, SDB_VGROUP, pIter, (void **)&pVgroup);
4✔
199
    if (pIter == NULL) {
4✔
200
      break;
2✔
201
    }
202

203
    SNodeEntry entry = {.nodeId = pVgroup->vgId, .hbTimestamp = pVgroup->updateTime};
2✔
204
    entry.epset = mndGetVgroupEpset(pMnode, pVgroup);
2✔
205

206
    int8_t *pReplica = taosHashGet(pHash, &pVgroup->dbUid, sizeof(pVgroup->dbUid));
2✔
207
    if (pReplica == NULL) {  // not exist, add it into hash map
2✔
208
      code = taosHashPut(pHash, &pVgroup->dbUid, sizeof(pVgroup->dbUid), &pVgroup->replica, sizeof(pVgroup->replica));
1✔
209
      if (code) {
1!
210
        mError("failed to put info into hashmap during task vgroup snapshot, code:%s", tstrerror(code));
×
211
        sdbRelease(pSdb, pVgroup);
×
212
        sdbCancelFetch(pSdb, pIter);
×
213
        goto _end;  // take snapshot failed, and not all ready
×
214
      }
215
    } else {
216
      if (*pReplica != pVgroup->replica) {
1!
UNCOV
217
        mInfo("vgId:%d replica:%d inconsistent with other vgroups replica:%d, not ready for stream operations",
×
218
              pVgroup->vgId, pVgroup->replica, *pReplica);
UNCOV
219
        *allReady = false;  // task snap success, but not all ready
×
220
      }
221
    }
222

223
    // if not all ready till now, no need to check the remaining vgroups,
224
    // but still we need to put the info of the existed vgroups into the snapshot list
225
    if (*allReady) {
2✔
226
      *allReady = checkStatusForEachReplica(pVgroup);
1✔
227
    }
228

229
    char buf[256] = {0};
2✔
230
    code = epsetToStr(&entry.epset, buf, tListLen(buf));
2✔
231
    if (code != 0) {  // print error and continue
2!
232
      mError("failed to convert epset to str, code:%s", tstrerror(code));
×
233
    }
234

235
    void *p = taosArrayPush(pVgroupList, &entry);
2✔
236
    if (p == NULL) {
2!
237
      mError("failed to put entry in vgroup list, nodeId:%d code:out of memory", entry.nodeId);
×
238
      code = terrno;
×
239
      sdbRelease(pSdb, pVgroup);
×
240
      sdbCancelFetch(pSdb, pIter);
×
241
      goto _end;
×
242
    } else {
243
      mDebug("take node snapshot, nodeId:%d %s", entry.nodeId, buf);
2!
244
    }
245

246
    sdbRelease(pSdb, pVgroup);
2✔
247
  }
248

249
_end:
2✔
250
  taosHashCleanup(pHash);
2✔
251
  return code;
2✔
252
}
253

254
int32_t mndTakeVgroupSnapshot(SMnode *pMnode, bool *allReady, SArray **pList) {
2✔
255
  int32_t   code = 0;
2✔
256
  SArray   *pVgroupList = NULL;
2✔
257

258
  *pList = NULL;
2✔
259
  *allReady = true;
2✔
260

261
  pVgroupList = taosArrayInit(4, sizeof(SNodeEntry));
2✔
262
  if (pVgroupList == NULL) {
2!
263
    mError("failed to prepare arraylist during take vgroup snapshot, code:%s", tstrerror(terrno));
×
264
    code = terrno;
×
265
    goto _err;
×
266
  }
267

268
  // 1. check for all vnodes status
269
  code = mndCheckAndAddVgroupsInfo(pMnode, pVgroupList, allReady);
2✔
270
  if (code) {
2!
271
    goto _err;
×
272
  }
273

274
  // 2. add snode info
275
  code = mndAddSnodeInfo(pMnode, pVgroupList);
2✔
276
  if (code) {
2!
277
    goto _err;
×
278
  }
279

280
  // 3. check for mnode status
281
  code = mndCheckMnodeStatus(pMnode);
2✔
282
  if (code != TSDB_CODE_SUCCESS) {
2!
UNCOV
283
    *allReady = false;
×
284
  }
285

286
  *pList = pVgroupList;
2✔
287
  return code;
2✔
288

289
_err:
×
290
  *allReady = false;
×
291
  taosArrayDestroy(pVgroupList);
×
292
  return code;
×
293
}
294

295
int32_t mndGetStreamObj(SMnode *pMnode, int64_t streamId, SStreamObj **pStream) {
1✔
296
  void *pIter = NULL;
1✔
297
  SSdb *pSdb = pMnode->pSdb;
1✔
298
  *pStream = NULL;
1✔
299

300
  SStreamObj *p = NULL;
1✔
301
  while ((pIter = sdbFetch(pSdb, SDB_STREAM, pIter, (void **)&p)) != NULL) {
1!
UNCOV
302
    if (p->uid == streamId) {
×
UNCOV
303
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
304
      *pStream = p;
×
UNCOV
305
      return TSDB_CODE_SUCCESS;
×
306
    }
UNCOV
307
    sdbRelease(pSdb, p);
×
308
  }
309

310
  return TSDB_CODE_STREAM_TASK_NOT_EXIST;
1✔
311
}
312

313
void mndKillTransImpl(SMnode *pMnode, int32_t transId, const char *pDbName) {
×
314
  STrans *pTrans = mndAcquireTrans(pMnode, transId);
×
315
  if (pTrans != NULL) {
×
316
    mInfo("kill active transId:%d in Db:%s", transId, pDbName);
×
317
    int32_t code = mndKillTrans(pMnode, pTrans);
×
318
    mndReleaseTrans(pMnode, pTrans);
×
319
    if (code) {
×
320
      mError("failed to kill transId:%d, code:%s", pTrans->id, tstrerror(code));
×
321
    }
322
  } else {
323
    mError("failed to acquire trans in Db:%s, transId:%d", pDbName, transId);
×
324
  }
325
}
×
326

327
int32_t extractNodeEpset(SMnode *pMnode, SEpSet *pEpSet, bool *hasEpset, int32_t taskId, int32_t nodeId) {
1✔
328
  *hasEpset = false;
1✔
329

330
  pEpSet->numOfEps = 0;
1✔
331
  if (nodeId == SNODE_HANDLE) {
1!
UNCOV
332
    SSnodeObj *pObj = NULL;
×
UNCOV
333
    void      *pIter = NULL;
×
334

UNCOV
335
    pIter = sdbFetch(pMnode->pSdb, SDB_SNODE, pIter, (void **)&pObj);
×
UNCOV
336
    if (pIter != NULL) {
×
UNCOV
337
      int32_t code = addEpIntoEpSet(pEpSet, pObj->pDnode->fqdn, pObj->pDnode->port);
×
UNCOV
338
      sdbRelease(pMnode->pSdb, pObj);
×
UNCOV
339
      sdbCancelFetch(pMnode->pSdb, pIter);
×
UNCOV
340
      if (code) {
×
341
        *hasEpset = false;
×
342
        mError("failed to set epset");
×
343
      } else {
UNCOV
344
        *hasEpset = true;
×
345
      }
UNCOV
346
      return code;
×
347
    } else {
348
      mError("failed to acquire snode epset");
×
349
      return TSDB_CODE_INVALID_PARA;
×
350
    }
351
  } else {
352
    SVgObj *pVgObj = mndAcquireVgroup(pMnode, nodeId);
1✔
353
    if (pVgObj != NULL) {
1!
UNCOV
354
      SEpSet epset = mndGetVgroupEpset(pMnode, pVgObj);
×
UNCOV
355
      mndReleaseVgroup(pMnode, pVgObj);
×
356

UNCOV
357
      epsetAssign(pEpSet, &epset);
×
UNCOV
358
      *hasEpset = true;
×
UNCOV
359
      return TSDB_CODE_SUCCESS;
×
360
    } else {
361
      mDebug("orphaned task:0x%x need to be dropped, nodeId:%d, no redo action", taskId, nodeId);
1!
362
      return TSDB_CODE_SUCCESS;
1✔
363
    }
364
  }
365
}
366

UNCOV
367
int32_t mndGetStreamTask(STaskId *pId, SStreamObj *pStream, SStreamTask **pTask) {
×
UNCOV
368
  *pTask = NULL;
×
369

UNCOV
370
  SStreamTask     *p = NULL;
×
UNCOV
371
  SStreamTaskIter *pIter = NULL;
×
UNCOV
372
  int32_t          code = createStreamTaskIter(pStream, &pIter);
×
UNCOV
373
  if (code) {
×
374
    mError("failed to create stream task iter:%s", pStream->name);
×
375
    return code;
×
376
  }
377

UNCOV
378
  while (streamTaskIterNextTask(pIter)) {
×
UNCOV
379
    code = streamTaskIterGetCurrent(pIter, &p);
×
UNCOV
380
    if (code) {
×
381
      continue;
×
382
    }
383

UNCOV
384
    if (p->id.taskId == pId->taskId) {
×
UNCOV
385
      destroyStreamTaskIter(pIter);
×
UNCOV
386
      *pTask = p;
×
UNCOV
387
      return 0;
×
388
    }
389
  }
390

391
  destroyStreamTaskIter(pIter);
×
392
  return TSDB_CODE_FAILED;
×
393
}
394

UNCOV
395
int32_t mndGetNumOfStreamTasks(const SStreamObj *pStream) {
×
UNCOV
396
  int32_t num = 0;
×
UNCOV
397
  for (int32_t i = 0; i < taosArrayGetSize(pStream->tasks); ++i) {
×
UNCOV
398
    SArray *pLevel = taosArrayGetP(pStream->tasks, i);
×
UNCOV
399
    num += taosArrayGetSize(pLevel);
×
400
  }
401

UNCOV
402
  return num;
×
403
}
404

UNCOV
405
int32_t mndGetNumOfStreams(SMnode *pMnode, char *dbName, int32_t *pNumOfStreams) {
×
UNCOV
406
  SSdb   *pSdb = pMnode->pSdb;
×
UNCOV
407
  SDbObj *pDb = mndAcquireDb(pMnode, dbName);
×
UNCOV
408
  if (pDb == NULL) {
×
409
    TAOS_RETURN(TSDB_CODE_MND_DB_NOT_SELECTED);
×
410
  }
411

UNCOV
412
  int32_t numOfStreams = 0;
×
UNCOV
413
  void   *pIter = NULL;
×
414
  while (1) {
×
UNCOV
415
    SStreamObj *pStream = NULL;
×
UNCOV
416
    pIter = sdbFetch(pSdb, SDB_STREAM, pIter, (void **)&pStream);
×
UNCOV
417
    if (pIter == NULL) break;
×
418

419
    if (pStream->sourceDbUid == pDb->uid) {
×
420
      numOfStreams++;
×
421
    }
422

423
    sdbRelease(pSdb, pStream);
×
424
  }
425

UNCOV
426
  *pNumOfStreams = numOfStreams;
×
UNCOV
427
  mndReleaseDb(pMnode, pDb);
×
UNCOV
428
  return 0;
×
429
}
430

UNCOV
431
static void freeTaskList(void *param) {
×
UNCOV
432
  SArray **pList = (SArray **)param;
×
UNCOV
433
  taosArrayDestroy(*pList);
×
UNCOV
434
}
×
435

436
int32_t mndInitExecInfo() {
9✔
437
  int32_t code = taosThreadMutexInit(&execInfo.lock, NULL);
9✔
438
  if (code) {
9!
439
    return code;
×
440
  }
441

442
  _hash_fn_t fn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR);
9✔
443

444
  execInfo.pTaskList = taosArrayInit(4, sizeof(STaskId));
9✔
445
  execInfo.pTaskMap = taosHashInit(64, fn, true, HASH_NO_LOCK);
9✔
446
  execInfo.transMgmt.pDBTrans = taosHashInit(32, fn, true, HASH_NO_LOCK);
9✔
447
  execInfo.pTransferStateStreams = taosHashInit(32, fn, true, HASH_NO_LOCK);
9✔
448
  execInfo.pChkptStreams = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
9✔
449
  execInfo.pStreamConsensus = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
9✔
450
  execInfo.pNodeList = taosArrayInit(4, sizeof(SNodeEntry));
9✔
451
  execInfo.pKilledChkptTrans = taosArrayInit(4, sizeof(SStreamTaskResetMsg));
9✔
452

453
  if (execInfo.pTaskList == NULL || execInfo.pTaskMap == NULL || execInfo.transMgmt.pDBTrans == NULL ||
9!
454
      execInfo.pTransferStateStreams == NULL || execInfo.pChkptStreams == NULL || execInfo.pStreamConsensus == NULL ||
9!
455
      execInfo.pNodeList == NULL || execInfo.pKilledChkptTrans == NULL) {
9!
456
    mError("failed to initialize the stream runtime env, code:%s", tstrerror(terrno));
×
457
    return terrno;
×
458
  }
459

460
  execInfo.role = NODE_ROLE_UNINIT;
9✔
461
  execInfo.switchFromFollower = false;
9✔
462

463
  taosHashSetFreeFp(execInfo.pTransferStateStreams, freeTaskList);
9✔
464
  taosHashSetFreeFp(execInfo.pChkptStreams, freeTaskList);
9✔
465
  taosHashSetFreeFp(execInfo.pStreamConsensus, freeTaskList);
9✔
466
  return 0;
9✔
467
}
468

UNCOV
469
void removeExpiredNodeInfo(const SArray *pNodeSnapshot) {
×
UNCOV
470
  SArray *pValidList = taosArrayInit(4, sizeof(SNodeEntry));
×
UNCOV
471
  if (pValidList == NULL) {  // not continue
×
472
    return;
×
473
  }
474

UNCOV
475
  int32_t size = taosArrayGetSize(pNodeSnapshot);
×
UNCOV
476
  int32_t oldSize = taosArrayGetSize(execInfo.pNodeList);
×
477

UNCOV
478
  for (int32_t i = 0; i < oldSize; ++i) {
×
UNCOV
479
    SNodeEntry *p = taosArrayGet(execInfo.pNodeList, i);
×
UNCOV
480
    if (p == NULL) {
×
481
      continue;
×
482
    }
483

UNCOV
484
    for (int32_t j = 0; j < size; ++j) {
×
UNCOV
485
      SNodeEntry *pEntry = taosArrayGet(pNodeSnapshot, j);
×
UNCOV
486
      if (pEntry == NULL) {
×
487
        continue;
×
488
      }
489

UNCOV
490
      if (pEntry->nodeId == p->nodeId) {
×
UNCOV
491
        p->hbTimestamp = pEntry->hbTimestamp;
×
492

UNCOV
493
        void *px = taosArrayPush(pValidList, p);
×
UNCOV
494
        if (px == NULL) {
×
495
          mError("failed to put node into list, nodeId:%d", p->nodeId);
×
496
        } else {
UNCOV
497
          mDebug("vgId:%d ts:%" PRId64 " HbMsgId:%d is valid", p->nodeId, p->hbTimestamp, p->lastHbMsgId);
×
498
        }
UNCOV
499
        break;
×
500
      }
501
    }
502
  }
503

UNCOV
504
  taosArrayDestroy(execInfo.pNodeList);
×
UNCOV
505
  execInfo.pNodeList = pValidList;
×
506

UNCOV
507
  mDebug("remain %d valid node entries after clean expired nodes info, prev size:%d",
×
508
         (int32_t)taosArrayGetSize(pValidList), oldSize);
509
}
510

UNCOV
511
int32_t doRemoveTasks(SStreamExecInfo *pExecNode, STaskId *pRemovedId) {
×
UNCOV
512
  void *p = taosHashGet(pExecNode->pTaskMap, pRemovedId, sizeof(*pRemovedId));
×
UNCOV
513
  if (p == NULL) {
×
UNCOV
514
    return TSDB_CODE_SUCCESS;
×
515
  }
516

UNCOV
517
  int32_t code = taosHashRemove(pExecNode->pTaskMap, pRemovedId, sizeof(*pRemovedId));
×
UNCOV
518
  if (code) {
×
519
    return code;
×
520
  }
521

UNCOV
522
  for (int32_t k = 0; k < taosArrayGetSize(pExecNode->pTaskList); ++k) {
×
UNCOV
523
    STaskId *pId = taosArrayGet(pExecNode->pTaskList, k);
×
UNCOV
524
    if (pId == NULL) {
×
525
      continue;
×
526
    }
527

UNCOV
528
    if (pId->taskId == pRemovedId->taskId && pId->streamId == pRemovedId->streamId) {
×
UNCOV
529
      taosArrayRemove(pExecNode->pTaskList, k);
×
530

UNCOV
531
      int32_t num = taosArrayGetSize(pExecNode->pTaskList);
×
UNCOV
532
      mInfo("s-task:0x%x removed from buffer, remain:%d in buffer list", (int32_t)pRemovedId->taskId, num);
×
UNCOV
533
      break;
×
534
    }
535
  }
536

UNCOV
537
  return TSDB_CODE_SUCCESS;
×
538
}
539

UNCOV
540
void removeTasksInBuf(SArray *pTaskIds, SStreamExecInfo *pExecInfo) {
×
UNCOV
541
  for (int32_t i = 0; i < taosArrayGetSize(pTaskIds); ++i) {
×
542
    STaskId *pId = taosArrayGet(pTaskIds, i);
×
543
    if (pId == NULL) {
×
544
      continue;
×
545
    }
546

547
    int32_t code = doRemoveTasks(pExecInfo, pId);
×
548
    if (code) {
×
549
      mError("failed to remove task in buffer list, 0x%" PRIx64, pId->taskId);
×
550
    }
551
  }
UNCOV
552
}
×
553

UNCOV
554
void removeStreamTasksInBuf(SStreamObj *pStream, SStreamExecInfo *pExecNode) {
×
UNCOV
555
  SStreamTaskIter *pIter = NULL;
×
UNCOV
556
  streamMutexLock(&pExecNode->lock);
×
557

558
  // 1. remove task entries
UNCOV
559
  int32_t code = createStreamTaskIter(pStream, &pIter);
×
UNCOV
560
  if (code) {
×
561
    streamMutexUnlock(&pExecNode->lock);
×
562
    mError("failed to create stream task iter:%s", pStream->name);
×
563
    return;
×
564
  }
565

UNCOV
566
  while (streamTaskIterNextTask(pIter)) {
×
UNCOV
567
    SStreamTask *pTask = NULL;
×
UNCOV
568
    code = streamTaskIterGetCurrent(pIter, &pTask);
×
UNCOV
569
    if (code) {
×
570
      continue;
×
571
    }
572

UNCOV
573
    STaskId id = {.streamId = pTask->id.streamId, .taskId = pTask->id.taskId};
×
UNCOV
574
    code = doRemoveTasks(pExecNode, &id);
×
UNCOV
575
    if (code) {
×
576
      mError("failed to remove task in buffer list, 0x%" PRIx64, id.taskId);
×
577
    }
578
  }
579

UNCOV
580
  if (taosHashGetSize(pExecNode->pTaskMap) != taosArrayGetSize(pExecNode->pTaskList)) {
×
581
    streamMutexUnlock(&pExecNode->lock);
×
582
    destroyStreamTaskIter(pIter);
×
583
    mError("task map size, task list size, not equal");
×
584
    return;
×
585
  }
586

587
  // 2. remove stream entry in consensus hash table and checkpoint-report hash table
UNCOV
588
  code = mndClearConsensusCheckpointId(execInfo.pStreamConsensus, pStream->uid);
×
UNCOV
589
  if (code) {
×
590
    mError("failed to clear consensus checkpointId, code:%s", tstrerror(code));
×
591
  }
592

UNCOV
593
  code = mndClearChkptReportInfo(execInfo.pChkptStreams, pStream->uid);
×
UNCOV
594
  if (code) {
×
UNCOV
595
    mError("failed to clear the checkpoint report info, code:%s", tstrerror(code));
×
596
  }
597

UNCOV
598
  streamMutexUnlock(&pExecNode->lock);
×
UNCOV
599
  destroyStreamTaskIter(pIter);
×
600
}
601

UNCOV
602
static bool taskNodeExists(SArray *pList, int32_t nodeId) {
×
UNCOV
603
  size_t num = taosArrayGetSize(pList);
×
604

UNCOV
605
  for (int32_t i = 0; i < num; ++i) {
×
UNCOV
606
    SNodeEntry *pEntry = taosArrayGet(pList, i);
×
UNCOV
607
    if (pEntry == NULL) {
×
608
      continue;
×
609
    }
610

UNCOV
611
    if (pEntry->nodeId == nodeId) {
×
UNCOV
612
      return true;
×
613
    }
614
  }
615

616
  return false;
×
617
}
618

UNCOV
619
int32_t removeExpiredNodeEntryAndTaskInBuf(SArray *pNodeSnapshot) {
×
UNCOV
620
  SArray *pRemovedTasks = taosArrayInit(4, sizeof(STaskId));
×
UNCOV
621
  if (pRemovedTasks == NULL) {
×
622
    return terrno;
×
623
  }
624

UNCOV
625
  int32_t numOfTask = taosArrayGetSize(execInfo.pTaskList);
×
UNCOV
626
  for (int32_t i = 0; i < numOfTask; ++i) {
×
UNCOV
627
    STaskId *pId = taosArrayGet(execInfo.pTaskList, i);
×
UNCOV
628
    if (pId == NULL) {
×
629
      continue;
×
630
    }
631

UNCOV
632
    STaskStatusEntry *pEntry = taosHashGet(execInfo.pTaskMap, pId, sizeof(*pId));
×
UNCOV
633
    if (pEntry == NULL) {
×
634
      continue;
×
635
    }
636

UNCOV
637
    if (pEntry->nodeId == SNODE_HANDLE) {
×
UNCOV
638
      continue;
×
639
    }
640

UNCOV
641
    bool existed = taskNodeExists(pNodeSnapshot, pEntry->nodeId);
×
UNCOV
642
    if (!existed) {
×
643
      void *p = taosArrayPush(pRemovedTasks, pId);
×
644
      if (p == NULL) {
×
645
        mError("failed to put task entry into remove list, taskId:0x%" PRIx64, pId->taskId);
×
646
      }
647
    }
648
  }
649

UNCOV
650
  removeTasksInBuf(pRemovedTasks, &execInfo);
×
651

UNCOV
652
  mDebug("remove invalid stream tasks:%d, remain:%d", (int32_t)taosArrayGetSize(pRemovedTasks),
×
653
         (int32_t)taosArrayGetSize(execInfo.pTaskList));
654

UNCOV
655
  removeExpiredNodeInfo(pNodeSnapshot);
×
656

UNCOV
657
  taosArrayDestroy(pRemovedTasks);
×
UNCOV
658
  return 0;
×
659
}
660

UNCOV
661
static int32_t allTasksSendChkptReport(SChkptReportInfo* pReportInfo, int32_t numOfTasks, const char* pName) {
×
UNCOV
662
  int64_t checkpointId = -1;
×
UNCOV
663
  int32_t transId = -1;
×
UNCOV
664
  int32_t taskId = -1;
×
665

UNCOV
666
  int32_t existed = (int32_t)taosArrayGetSize(pReportInfo->pTaskList);
×
UNCOV
667
  if (existed != numOfTasks) {
×
UNCOV
668
    mDebug("stream:0x%" PRIx64 " %s %d/%d tasks send checkpoint-report, %d not send", pReportInfo->streamId, pName,
×
669
           existed, numOfTasks, numOfTasks - existed);
UNCOV
670
    return -1;
×
671
  }
672

673
  // acquire current active checkpointId, and do cross-check checkpointId info in exec.pTaskList
UNCOV
674
  for(int32_t i = 0; i < numOfTasks; ++i) {
×
UNCOV
675
    STaskChkptInfo *pInfo = taosArrayGet(pReportInfo->pTaskList, i);
×
UNCOV
676
    if (pInfo == NULL) {
×
677
      continue;
×
678
    }
679

UNCOV
680
    if (checkpointId == -1) {
×
UNCOV
681
      checkpointId = pInfo->checkpointId;
×
UNCOV
682
      transId = pInfo->transId;
×
UNCOV
683
      taskId = pInfo->taskId;
×
UNCOV
684
    } else if (checkpointId != pInfo->checkpointId) {
×
685
      mError("stream:0x%" PRIx64
×
686
             " checkpointId in checkpoint-report list are not identical, type 1 taskId:0x%x checkpointId:%" PRId64
687
             ", type 2 taskId:0x%x checkpointId:%" PRId64,
688
             pReportInfo->streamId, taskId, checkpointId, pInfo->taskId, pInfo->checkpointId);
689
      return -1;
×
690
    }
691
  }
692

693
  // check for the correct checkpointId for current task info in STaskChkptInfo
UNCOV
694
  STaskChkptInfo  *p = taosArrayGet(pReportInfo->pTaskList, 0);
×
UNCOV
695
  STaskId id = {.streamId = p->streamId, .taskId = p->taskId};
×
UNCOV
696
  STaskStatusEntry *pe = taosHashGet(execInfo.pTaskMap, &id, sizeof(id));
×
697

698
  // cross-check failed, there must be something unknown wrong
UNCOV
699
  SStreamTransInfo *pTransInfo = taosHashGet(execInfo.transMgmt.pDBTrans, &id.streamId, sizeof(id.streamId));
×
UNCOV
700
  if (pTransInfo == NULL) {
×
UNCOV
701
    mWarn("stream:0x%" PRIx64 " no active trans exists for checkpoint transId:%d, it may have been cleared already",
×
702
           id.streamId, transId);
703

UNCOV
704
    if (pe->checkpointInfo.activeId != 0 && pe->checkpointInfo.activeId != checkpointId) {
×
705
      mWarn("stream:0x%" PRIx64 " active checkpointId is not equalled to the required, current:%" PRId64
×
706
            ", req:%" PRId64 " recheck next time",
707
            id.streamId, pe->checkpointInfo.activeId, checkpointId);
708
      return -1;
×
709
    } else {
710
      //  do nothing
711
    }
712
  } else {
UNCOV
713
    if (pTransInfo->transId != transId) {
×
UNCOV
714
      mError("stream:0x%" PRIx64
×
715
             " checkpoint-report list info are expired, active transId:%d trans in list:%d, recheck next time",
716
             id.streamId, pTransInfo->transId, transId);
UNCOV
717
      return -1;
×
718
    }
719
  }
720

UNCOV
721
  mDebug("stream:0x%" PRIx64 " %s all %d tasks send checkpoint-report, start to update checkpoint-info", id.streamId,
×
722
         pName, numOfTasks);
723

UNCOV
724
  return TSDB_CODE_SUCCESS;
×
725
}
726

UNCOV
727
int32_t mndScanCheckpointReportInfo(SRpcMsg *pReq) {
×
UNCOV
728
  SMnode *pMnode = pReq->info.node;
×
UNCOV
729
  void   *pIter = NULL;
×
UNCOV
730
  int32_t code = 0;
×
UNCOV
731
  SArray *pDropped = taosArrayInit(4, sizeof(int64_t));
×
UNCOV
732
  if (pDropped == NULL) {
×
733
    return terrno;
×
734
  }
735

UNCOV
736
  mDebug("start to scan checkpoint report info");
×
737

UNCOV
738
  streamMutexLock(&execInfo.lock);
×
739

UNCOV
740
  while ((pIter = taosHashIterate(execInfo.pChkptStreams, pIter)) != NULL) {
×
UNCOV
741
    SChkptReportInfo *px = (SChkptReportInfo *)pIter;
×
UNCOV
742
    if (taosArrayGetSize(px->pTaskList) == 0) {
×
UNCOV
743
      continue;
×
744
    }
745

UNCOV
746
    STaskChkptInfo *pInfo = taosArrayGet(px->pTaskList, 0);
×
UNCOV
747
    if (pInfo == NULL) {
×
748
      continue;
×
749
    }
750

UNCOV
751
    SStreamObj *pStream = NULL;
×
UNCOV
752
    code = mndGetStreamObj(pMnode, pInfo->streamId, &pStream);
×
UNCOV
753
    if (pStream == NULL || code != 0) {
×
UNCOV
754
      mDebug("failed to acquire stream:0x%" PRIx64 " remove it from checkpoint-report list", pInfo->streamId);
×
UNCOV
755
      void *p = taosArrayPush(pDropped, &pInfo->streamId);
×
UNCOV
756
      if (p == NULL) {
×
757
        mError("failed to put stream into drop list:0x%" PRIx64, pInfo->streamId);
×
758
      }
UNCOV
759
      continue;
×
760
    }
761

UNCOV
762
    int32_t total = mndGetNumOfStreamTasks(pStream);
×
UNCOV
763
    int32_t ret = allTasksSendChkptReport(px, total, pStream->name);
×
UNCOV
764
    if (ret == 0) {
×
UNCOV
765
      code = mndStreamTransConflictCheck(pMnode, pStream->uid, MND_STREAM_CHKPT_UPDATE_NAME, false);
×
UNCOV
766
      if (code == 0) {
×
UNCOV
767
        code = mndCreateStreamChkptInfoUpdateTrans(pMnode, pStream, px->pTaskList);
×
UNCOV
768
        if (code == TSDB_CODE_SUCCESS || code == TSDB_CODE_ACTION_IN_PROGRESS) {  // remove this entry
×
UNCOV
769
          taosArrayClear(px->pTaskList);
×
UNCOV
770
          mInfo("stream:0x%" PRIx64 " clear checkpoint-report list and update the report checkpointId from:%" PRId64
×
771
                " to %" PRId64,
772
                pInfo->streamId, px->reportChkpt, pInfo->checkpointId);
UNCOV
773
          px->reportChkpt = pInfo->checkpointId;
×
774
        } else {
775
          mDebug("stream:0x%" PRIx64 " not launch chkpt-info update trans, due to checkpoint not finished yet",
×
776
                 pInfo->streamId);
777
        }
778

UNCOV
779
        sdbRelease(pMnode->pSdb, pStream);
×
UNCOV
780
        break;
×
781
      } else {
782
        mDebug("stream:0x%" PRIx64 " active checkpoint trans not finished yet, wait", pInfo->streamId);
×
783
      }
784
    }
785

UNCOV
786
    sdbRelease(pMnode->pSdb, pStream);
×
787
  }
788

UNCOV
789
  int32_t size = taosArrayGetSize(pDropped);
×
UNCOV
790
  if (size > 0) {
×
UNCOV
791
    for (int32_t i = 0; i < size; ++i) {
×
UNCOV
792
      int64_t *pStreamId = (int64_t *)taosArrayGet(pDropped, i);
×
UNCOV
793
      if (pStreamId == NULL) {
×
794
        continue;
×
795
      }
796

UNCOV
797
      code = taosHashRemove(execInfo.pChkptStreams, pStreamId, sizeof(*pStreamId));
×
UNCOV
798
      if (code) {
×
799
        mError("failed to remove stream in buf:0x%" PRIx64, *pStreamId);
×
800
      }
801
    }
802

UNCOV
803
    int32_t numOfStreams = taosHashGetSize(execInfo.pChkptStreams);
×
UNCOV
804
    mDebug("drop %d stream(s) in checkpoint-report list, remain:%d", size, numOfStreams);
×
805
  }
806

UNCOV
807
  streamMutexUnlock(&execInfo.lock);
×
808

UNCOV
809
  taosArrayDestroy(pDropped);
×
810

UNCOV
811
  mDebug("end to scan checkpoint report info")
×
UNCOV
812
  return TSDB_CODE_SUCCESS;
×
813
}
814

UNCOV
815
int32_t mndCreateSetConsensusChkptIdTrans(SMnode *pMnode, SStreamObj *pStream, int32_t taskId, int64_t checkpointId,
×
816
                                          int64_t ts) {
UNCOV
817
  char         msg[128] = {0};
×
UNCOV
818
  STrans      *pTrans = NULL;
×
UNCOV
819
  SStreamTask *pTask = NULL;
×
820

UNCOV
821
  snprintf(msg, tListLen(msg), "set consen-chkpt-id for task:0x%x", taskId);
×
822

UNCOV
823
  int32_t code = doCreateTrans(pMnode, pStream, NULL, TRN_CONFLICT_NOTHING, MND_STREAM_CHKPT_CONSEN_NAME, msg, &pTrans);
×
UNCOV
824
  if (pTrans == NULL || code != 0) {
×
825
    return terrno;
×
826
  }
827

UNCOV
828
  STaskId id = {.streamId = pStream->uid, .taskId = taskId};
×
UNCOV
829
  code = mndGetStreamTask(&id, pStream, &pTask);
×
UNCOV
830
  if (code) {
×
831
    mError("failed to get task:0x%x in stream:%s, failed to create consensus-checkpointId", taskId, pStream->name);
×
832
    sdbRelease(pMnode->pSdb, pStream);
×
833
    return code;
×
834
  }
835

UNCOV
836
  code = mndStreamRegisterTrans(pTrans, MND_STREAM_CHKPT_CONSEN_NAME, pStream->uid);
×
UNCOV
837
  if (code) {
×
838
    sdbRelease(pMnode->pSdb, pStream);
×
839
    return code;
×
840
  }
841

UNCOV
842
  code = mndStreamSetChkptIdAction(pMnode, pTrans, pTask, checkpointId, ts);
×
UNCOV
843
  if (code != 0) {
×
844
    sdbRelease(pMnode->pSdb, pStream);
×
845
    mndTransDrop(pTrans);
×
846
    return code;
×
847
  }
848

UNCOV
849
  code = mndPersistTransLog(pStream, pTrans, SDB_STATUS_READY);
×
UNCOV
850
  if (code) {
×
851
    sdbRelease(pMnode->pSdb, pStream);
×
852
    mndTransDrop(pTrans);
×
853
    return code;
×
854
  }
855

UNCOV
856
  code = mndTransPrepare(pMnode, pTrans);
×
UNCOV
857
  if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_ACTION_IN_PROGRESS) {
×
858
    mError("trans:%d, failed to prepare set consensus-chkptId trans since %s", pTrans->id, terrstr());
×
859
    sdbRelease(pMnode->pSdb, pStream);
×
860
    mndTransDrop(pTrans);
×
861
    return code;
×
862
  }
863

UNCOV
864
  sdbRelease(pMnode->pSdb, pStream);
×
UNCOV
865
  mndTransDrop(pTrans);
×
866

UNCOV
867
  return TSDB_CODE_ACTION_IN_PROGRESS;
×
868
}
869

UNCOV
870
int32_t mndGetConsensusInfo(SHashObj *pHash, int64_t streamId, int32_t numOfTasks, SCheckpointConsensusInfo **pInfo) {
×
UNCOV
871
  *pInfo = NULL;
×
872

UNCOV
873
  void *px = taosHashGet(pHash, &streamId, sizeof(streamId));
×
UNCOV
874
  if (px != NULL) {
×
UNCOV
875
    *pInfo = px;
×
UNCOV
876
    return 0;
×
877
  }
878

UNCOV
879
  SCheckpointConsensusInfo p = {
×
UNCOV
880
      .pTaskList = taosArrayInit(4, sizeof(SCheckpointConsensusEntry)),
×
881
      .numOfTasks = numOfTasks,
882
      .streamId = streamId,
883
  };
884

UNCOV
885
  if (p.pTaskList == NULL) {
×
886
    return terrno;
×
887
  }
888

UNCOV
889
  int32_t code = taosHashPut(pHash, &streamId, sizeof(streamId), &p, sizeof(p));
×
UNCOV
890
  if (code == 0) {
×
UNCOV
891
    void *pChkptInfo = (SCheckpointConsensusInfo *)taosHashGet(pHash, &streamId, sizeof(streamId));
×
UNCOV
892
    *pInfo = pChkptInfo;
×
893
  } else {
894
    *pInfo = NULL;
×
895
  }
896

UNCOV
897
  return code;
×
898
}
899

900
// no matter existed or not, add the request into info list anyway, since we need to send rsp mannually
901
// discard the msg may lead to the lost of connections.
UNCOV
902
void mndAddConsensusTasks(SCheckpointConsensusInfo *pInfo, const SRestoreCheckpointInfo *pRestoreInfo) {
×
UNCOV
903
  SCheckpointConsensusEntry info = {.ts = taosGetTimestampMs()};
×
UNCOV
904
  memcpy(&info.req, pRestoreInfo, sizeof(info.req));
×
905

UNCOV
906
  int32_t num = (int32_t) taosArrayGetSize(pInfo->pTaskList);
×
UNCOV
907
  for (int32_t i = 0; i < num; ++i) {
×
UNCOV
908
    SCheckpointConsensusEntry *p = taosArrayGet(pInfo->pTaskList, i);
×
UNCOV
909
    if (p == NULL) {
×
910
      continue;
×
911
    }
912

UNCOV
913
    if (p->req.taskId == info.req.taskId) {
×
UNCOV
914
      mDebug("s-task:0x%x already in consensus-checkpointId list for stream:0x%" PRIx64 ", update ts %" PRId64
×
915
             "->%" PRId64 " checkpointId:%" PRId64 " -> %" PRId64 " total existed:%d",
916
             pRestoreInfo->taskId, pRestoreInfo->streamId, p->req.startTs, info.req.startTs, p->req.checkpointId,
917
             info.req.checkpointId, num);
UNCOV
918
      p->req.startTs = info.req.startTs;
×
UNCOV
919
      p->req.checkpointId = info.req.checkpointId;
×
UNCOV
920
      p->req.transId = info.req.transId;
×
UNCOV
921
      return;
×
922
    }
923
  }
924

UNCOV
925
  void *p = taosArrayPush(pInfo->pTaskList, &info);
×
UNCOV
926
  if (p == NULL) {
×
927
    mError("s-task:0x%x failed to put task into consensus-checkpointId list, code: out of memory", info.req.taskId);
×
928
  } else {
UNCOV
929
    num = taosArrayGetSize(pInfo->pTaskList);
×
UNCOV
930
    mDebug("s-task:0x%x checkpointId:%" PRId64 " added into consensus-checkpointId list, stream:0x%" PRIx64
×
931
           " waiting tasks:%d",
932
           pRestoreInfo->taskId, pRestoreInfo->checkpointId, pRestoreInfo->streamId, num);
933
  }
934
}
935

UNCOV
936
void mndClearConsensusRspEntry(SCheckpointConsensusInfo *pInfo) {
×
UNCOV
937
  taosArrayDestroy(pInfo->pTaskList);
×
UNCOV
938
  pInfo->pTaskList = NULL;
×
UNCOV
939
}
×
940

UNCOV
941
int32_t mndClearConsensusCheckpointId(SHashObj *pHash, int64_t streamId) {
×
UNCOV
942
  int32_t code = 0;
×
UNCOV
943
  int32_t numOfStreams = taosHashGetSize(pHash);
×
UNCOV
944
  if (numOfStreams == 0) {
×
UNCOV
945
    return code;
×
946
  }
947

UNCOV
948
  code = taosHashRemove(pHash, &streamId, sizeof(streamId));
×
UNCOV
949
  if (code == 0) {
×
UNCOV
950
    mDebug("drop stream:0x%" PRIx64 " in consensus-checkpointId list, remain:%d", streamId, numOfStreams);
×
951
  } else {
952
    mError("failed to remove stream:0x%" PRIx64 " in consensus-checkpointId list, remain:%d", streamId, numOfStreams);
×
953
  }
954

UNCOV
955
  return code;
×
956
}
957

UNCOV
958
int32_t mndClearChkptReportInfo(SHashObj *pHash, int64_t streamId) {
×
UNCOV
959
  int32_t code = 0;
×
UNCOV
960
  int32_t numOfStreams = taosHashGetSize(pHash);
×
UNCOV
961
  if (numOfStreams == 0) {
×
UNCOV
962
    return code;
×
963
  }
964

UNCOV
965
  code = taosHashRemove(pHash, &streamId, sizeof(streamId));
×
UNCOV
966
  if (code == 0) {
×
UNCOV
967
    mDebug("drop stream:0x%" PRIx64 " in chkpt-report list, remain:%d", streamId, numOfStreams);
×
968
  } else {
UNCOV
969
    mError("failed to remove stream:0x%" PRIx64 " in chkpt-report list, remain:%d", streamId, numOfStreams);
×
970
  }
971

UNCOV
972
  return code;
×
973
}
974

975
int32_t mndResetChkptReportInfo(SHashObj *pHash, int64_t streamId) {
×
976
  SChkptReportInfo *pInfo = taosHashGet(pHash, &streamId, sizeof(streamId));
×
977
  if (pInfo != NULL) {
×
978
    taosArrayClear(pInfo->pTaskList);
×
979
    mDebug("stream:0x%" PRIx64 " checkpoint-report list cleared, prev report checkpointId:%" PRId64, streamId,
×
980
           pInfo->reportChkpt);
981
    return 0;
×
982
  }
983

984
  return TSDB_CODE_MND_STREAM_NOT_EXIST;
×
985
}
986

UNCOV
987
static void mndShowStreamStatus(char *dst, int8_t status) {
×
UNCOV
988
  if (status == STREAM_STATUS__NORMAL) {
×
UNCOV
989
    tstrncpy(dst, "ready", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
990
  } else if (status == STREAM_STATUS__STOP) {
×
991
    tstrncpy(dst, "stop", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
992
  } else if (status == STREAM_STATUS__FAILED) {
×
993
    tstrncpy(dst, "failed", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
994
  } else if (status == STREAM_STATUS__RECOVER) {
×
995
    tstrncpy(dst, "recover", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
996
  } else if (status == STREAM_STATUS__PAUSE) {
×
UNCOV
997
    tstrncpy(dst, "paused", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
998
  } else if (status == STREAM_STATUS__INIT) {
×
UNCOV
999
    tstrncpy(dst, "init", MND_STREAM_TRIGGER_NAME_SIZE);
×
1000
  }
UNCOV
1001
}
×
1002

UNCOV
1003
static void mndShowStreamTrigger(char *dst, SStreamObj *pStream) {
×
UNCOV
1004
  int8_t trigger = pStream->conf.trigger;
×
UNCOV
1005
  if (trigger == STREAM_TRIGGER_AT_ONCE) {
×
UNCOV
1006
    tstrncpy(dst, "at once", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
1007
  } else if (trigger == STREAM_TRIGGER_WINDOW_CLOSE) {
×
UNCOV
1008
    tstrncpy(dst, "window close", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
1009
  } else if (trigger == STREAM_TRIGGER_MAX_DELAY) {
×
UNCOV
1010
    tstrncpy(dst, "max delay", MND_STREAM_TRIGGER_NAME_SIZE);
×
UNCOV
1011
  } else if (trigger == STREAM_TRIGGER_FORCE_WINDOW_CLOSE) {
×
UNCOV
1012
    tstrncpy(dst, "force window close", MND_STREAM_TRIGGER_NAME_SIZE);
×
1013
  }
UNCOV
1014
}
×
1015

UNCOV
1016
static void int64ToHexStr(int64_t id, char *pBuf, int32_t bufLen) {
×
UNCOV
1017
  memset(pBuf, 0, bufLen);
×
UNCOV
1018
  pBuf[2] = '0';
×
UNCOV
1019
  pBuf[3] = 'x';
×
1020

UNCOV
1021
  int32_t len = tintToHex(id, &pBuf[4]);
×
UNCOV
1022
  varDataSetLen(pBuf, len + 2);
×
UNCOV
1023
}
×
1024

UNCOV
1025
static int32_t isAllTaskPaused(SStreamObj *pStream, bool *pRes) {
×
UNCOV
1026
  int32_t          code = TSDB_CODE_SUCCESS;
×
UNCOV
1027
  int32_t          lino = 0;
×
UNCOV
1028
  SStreamTaskIter *pIter = NULL;
×
UNCOV
1029
  bool             isPaused =  true;
×
1030

UNCOV
1031
  taosRLockLatch(&pStream->lock);
×
UNCOV
1032
  code = createStreamTaskIter(pStream, &pIter);
×
UNCOV
1033
  TSDB_CHECK_CODE(code, lino, _end);
×
1034

UNCOV
1035
  while (streamTaskIterNextTask(pIter)) {
×
UNCOV
1036
    SStreamTask *pTask = NULL;
×
UNCOV
1037
    code = streamTaskIterGetCurrent(pIter, &pTask);
×
UNCOV
1038
    TSDB_CHECK_CODE(code, lino, _end);
×
1039

UNCOV
1040
    STaskId           id = {.streamId = pTask->id.streamId, .taskId = pTask->id.taskId};
×
UNCOV
1041
    STaskStatusEntry *pe = taosHashGet(execInfo.pTaskMap, &id, sizeof(id));
×
UNCOV
1042
    if (pe == NULL) {
×
UNCOV
1043
      continue;
×
1044
    }
UNCOV
1045
    if (pe->status != TASK_STATUS__PAUSE) {
×
UNCOV
1046
      isPaused = false;
×
1047
    }
1048
  }
UNCOV
1049
  (*pRes) = isPaused;
×
1050

UNCOV
1051
_end:
×
UNCOV
1052
  destroyStreamTaskIter(pIter);
×
1053
  taosRUnLockLatch(&pStream->lock);
×
UNCOV
1054
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
1055
    mError("error happens when get stream status, lino:%d, code:%s", lino, tstrerror(code));
×
1056
  }
UNCOV
1057
  return code;
×
1058
}
1059

UNCOV
1060
int32_t setStreamAttrInResBlock(SStreamObj *pStream, SSDataBlock *pBlock, int32_t numOfRows) {
×
UNCOV
1061
  int32_t code = 0;
×
UNCOV
1062
  int32_t cols = 0;
×
UNCOV
1063
  int32_t lino = 0;
×
1064

UNCOV
1065
  char streamName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1066
  STR_WITH_MAXSIZE_TO_VARSTR(streamName, mndGetDbStr(pStream->name), sizeof(streamName));
×
UNCOV
1067
  SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1068
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1069

UNCOV
1070
  code = colDataSetVal(pColInfo, numOfRows, (const char *)streamName, false);
×
UNCOV
1071
  TSDB_CHECK_CODE(code, lino, _end);
×
1072

1073
  // create time
UNCOV
1074
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1075
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
UNCOV
1076
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pStream->createTime, false);
×
UNCOV
1077
  TSDB_CHECK_CODE(code, lino, _end);
×
1078

1079
  // stream id
UNCOV
1080
  char buf[128] = {0};
×
UNCOV
1081
  int64ToHexStr(pStream->uid, buf, tListLen(buf));
×
UNCOV
1082
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1083
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
UNCOV
1084
  code = colDataSetVal(pColInfo, numOfRows, buf, false);
×
UNCOV
1085
  TSDB_CHECK_CODE(code, lino, _end);
×
1086

1087
  // related fill-history stream id
UNCOV
1088
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1089
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1090
  if (pStream->hTaskUid != 0) {
×
UNCOV
1091
    int64ToHexStr(pStream->hTaskUid, buf, tListLen(buf));
×
UNCOV
1092
    code = colDataSetVal(pColInfo, numOfRows, buf, false);
×
1093
  } else {
UNCOV
1094
    code = colDataSetVal(pColInfo, numOfRows, buf, true);
×
1095
  }
UNCOV
1096
  TSDB_CHECK_CODE(code, lino, _end);
×
1097

1098
  // related fill-history stream id
UNCOV
1099
  char sql[TSDB_SHOW_SQL_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1100
  STR_WITH_MAXSIZE_TO_VARSTR(sql, pStream->sql, sizeof(sql));
×
UNCOV
1101
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1102
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
UNCOV
1103
  code = colDataSetVal(pColInfo, numOfRows, (const char *)sql, false);
×
UNCOV
1104
  TSDB_CHECK_CODE(code, lino, _end);
×
1105

UNCOV
1106
  char status[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1107
  char status2[MND_STREAM_TRIGGER_NAME_SIZE] = {0};
×
UNCOV
1108
  bool isPaused = false;
×
UNCOV
1109
  code = isAllTaskPaused(pStream, &isPaused);
×
UNCOV
1110
  TSDB_CHECK_CODE(code, lino, _end);
×
1111

UNCOV
1112
  int8_t streamStatus = atomic_load_8(&pStream->status);
×
UNCOV
1113
  if (isPaused && pStream->tasks != NULL) {
×
UNCOV
1114
    streamStatus = STREAM_STATUS__PAUSE;
×
1115
  }
UNCOV
1116
  mndShowStreamStatus(status2, streamStatus);
×
UNCOV
1117
  STR_WITH_MAXSIZE_TO_VARSTR(status, status2, sizeof(status));
×
UNCOV
1118
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1119
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1120

UNCOV
1121
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&status, false);
×
UNCOV
1122
  TSDB_CHECK_CODE(code, lino, _end);
×
1123

UNCOV
1124
  char sourceDB[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1125
  STR_WITH_MAXSIZE_TO_VARSTR(sourceDB, mndGetDbStr(pStream->sourceDb), sizeof(sourceDB));
×
UNCOV
1126
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1127
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1128

UNCOV
1129
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&sourceDB, false);
×
UNCOV
1130
  TSDB_CHECK_CODE(code, lino, _end);
×
1131

UNCOV
1132
  char targetDB[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1133
  STR_WITH_MAXSIZE_TO_VARSTR(targetDB, mndGetDbStr(pStream->targetDb), sizeof(targetDB));
×
UNCOV
1134
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1135
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1136

UNCOV
1137
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&targetDB, false);
×
UNCOV
1138
  TSDB_CHECK_CODE(code, lino, _end);
×
1139

UNCOV
1140
  if (pStream->targetSTbName[0] == 0) {
×
UNCOV
1141
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1142
    TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1143

UNCOV
1144
    code = colDataSetVal(pColInfo, numOfRows, NULL, true);
×
1145
  } else {
UNCOV
1146
    char targetSTB[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1147
    STR_WITH_MAXSIZE_TO_VARSTR(targetSTB, mndGetStbStr(pStream->targetSTbName), sizeof(targetSTB));
×
UNCOV
1148
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1149
    TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1150

UNCOV
1151
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&targetSTB, false);
×
1152
  }
UNCOV
1153
  TSDB_CHECK_CODE(code, lino, _end);
×
1154

UNCOV
1155
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1156
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1157

UNCOV
1158
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pStream->conf.watermark, false);
×
UNCOV
1159
  TSDB_CHECK_CODE(code, lino, _end);
×
1160

UNCOV
1161
  char trigger[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1162
  char trigger2[MND_STREAM_TRIGGER_NAME_SIZE] = {0};
×
UNCOV
1163
  mndShowStreamTrigger(trigger2, pStream);
×
UNCOV
1164
  STR_WITH_MAXSIZE_TO_VARSTR(trigger, trigger2, sizeof(trigger));
×
UNCOV
1165
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1166
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1167

UNCOV
1168
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&trigger, false);
×
UNCOV
1169
  TSDB_CHECK_CODE(code, lino, _end);
×
1170

1171
  // sink_quota
UNCOV
1172
  char sinkQuota[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1173
  sinkQuota[0] = '0';
×
UNCOV
1174
  char dstStr[20] = {0};
×
UNCOV
1175
  STR_TO_VARSTR(dstStr, sinkQuota)
×
UNCOV
1176
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1177
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1178

UNCOV
1179
  code = colDataSetVal(pColInfo, numOfRows, (const char *)dstStr, false);
×
UNCOV
1180
  TSDB_CHECK_CODE(code, lino, _end);
×
1181

1182
  // checkpoint interval
UNCOV
1183
  char tmp[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1184
  (void)tsnprintf(varDataVal(tmp), sizeof(tmp) - VARSTR_HEADER_SIZE, "%d sec", tsStreamCheckpointInterval);
×
UNCOV
1185
  varDataSetLen(tmp, strlen(varDataVal(tmp)));
×
1186

UNCOV
1187
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1188
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1189

UNCOV
1190
  code = colDataSetVal(pColInfo, numOfRows, (const char *)tmp, false);
×
UNCOV
1191
  TSDB_CHECK_CODE(code, lino, _end);
×
1192

1193
  // checkpoint backup type
UNCOV
1194
  char backup[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1195
  STR_TO_VARSTR(backup, "none")
×
UNCOV
1196
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1197
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1198

UNCOV
1199
  code = colDataSetVal(pColInfo, numOfRows, (const char *)backup, false);
×
UNCOV
1200
  TSDB_CHECK_CODE(code, lino, _end);
×
1201

1202
  // history scan idle
UNCOV
1203
  char scanHistoryIdle[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1204
  tstrncpy(scanHistoryIdle, "100a", sizeof(scanHistoryIdle));
×
1205

UNCOV
1206
  memset(dstStr, 0, tListLen(dstStr));
×
UNCOV
1207
  STR_TO_VARSTR(dstStr, scanHistoryIdle)
×
UNCOV
1208
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1209
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1210

UNCOV
1211
  code = colDataSetVal(pColInfo, numOfRows, (const char *)dstStr, false);
×
UNCOV
1212
  TSDB_CHECK_CODE(code, lino, _end);
×
1213

UNCOV
1214
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1215
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
UNCOV
1216
  char msg[TSDB_RESERVE_VALUE_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1217
  if (streamStatus == STREAM_STATUS__FAILED){
×
UNCOV
1218
    STR_TO_VARSTR(msg, pStream->reserve)
×
1219
  } else {
UNCOV
1220
    STR_TO_VARSTR(msg, " ")
×
1221
  }
UNCOV
1222
  code = colDataSetVal(pColInfo, numOfRows, (const char *)msg, false);
×
1223

UNCOV
1224
_end:
×
UNCOV
1225
  if (code) {
×
UNCOV
1226
    mError("error happens when build stream attr result block, lino:%d, code:%s", lino, tstrerror(code));
×
1227
  }
UNCOV
1228
  return code;
×
1229
}
1230

UNCOV
1231
int32_t setTaskAttrInResBlock(SStreamObj *pStream, SStreamTask *pTask, SSDataBlock *pBlock, int32_t numOfRows,
×
1232
                              int32_t precision) {
UNCOV
1233
  SColumnInfoData *pColInfo = NULL;
×
UNCOV
1234
  int32_t          cols = 0;
×
UNCOV
1235
  int32_t          code = 0;
×
UNCOV
1236
  int32_t          lino = 0;
×
1237

UNCOV
1238
  STaskId id = {.streamId = pTask->id.streamId, .taskId = pTask->id.taskId};
×
1239

UNCOV
1240
  STaskStatusEntry *pe = taosHashGet(execInfo.pTaskMap, &id, sizeof(id));
×
UNCOV
1241
  if (pe == NULL) {
×
UNCOV
1242
    mError("task:0x%" PRIx64 " not exists in any vnodes, streamName:%s, streamId:0x%" PRIx64 " createTs:%" PRId64
×
1243
           " no valid status/stage info",
1244
           id.taskId, pStream->name, pStream->uid, pStream->createTime);
UNCOV
1245
    return TSDB_CODE_STREAM_TASK_NOT_EXIST;
×
1246
  }
1247

1248
  // stream name
UNCOV
1249
  char streamName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1250
  STR_WITH_MAXSIZE_TO_VARSTR(streamName, mndGetDbStr(pStream->name), sizeof(streamName));
×
1251

UNCOV
1252
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1253
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1254

UNCOV
1255
  code = colDataSetVal(pColInfo, numOfRows, (const char *)streamName, false);
×
UNCOV
1256
  TSDB_CHECK_CODE(code, lino, _end);
×
1257

1258
  // task id
UNCOV
1259
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1260
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1261

UNCOV
1262
  char idstr[128] = {0};
×
UNCOV
1263
  int64ToHexStr(pTask->id.taskId, idstr, tListLen(idstr));
×
UNCOV
1264
  code = colDataSetVal(pColInfo, numOfRows, idstr, false);
×
UNCOV
1265
  TSDB_CHECK_CODE(code, lino, _end);
×
1266

1267
  // node type
UNCOV
1268
  char nodeType[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1269
  varDataSetLen(nodeType, 5);
×
UNCOV
1270
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1271
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1272

UNCOV
1273
  if (pTask->info.nodeId > 0) {
×
UNCOV
1274
    memcpy(varDataVal(nodeType), "vnode", 5);
×
1275
  } else {
UNCOV
1276
    memcpy(varDataVal(nodeType), "snode", 5);
×
1277
  }
UNCOV
1278
  code = colDataSetVal(pColInfo, numOfRows, nodeType, false);
×
UNCOV
1279
  TSDB_CHECK_CODE(code, lino, _end);
×
1280

1281
  // node id
UNCOV
1282
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1283
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1284

UNCOV
1285
  int64_t nodeId = TMAX(pTask->info.nodeId, 0);
×
UNCOV
1286
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&nodeId, false);
×
UNCOV
1287
  TSDB_CHECK_CODE(code, lino, _end);
×
1288

1289
  // level
UNCOV
1290
  char level[20 + VARSTR_HEADER_SIZE] = {0};
×
UNCOV
1291
  if (pTask->info.taskLevel == TASK_LEVEL__SOURCE) {
×
UNCOV
1292
    STR_WITH_SIZE_TO_VARSTR(level, "source", 6);
×
UNCOV
1293
  } else if (pTask->info.taskLevel == TASK_LEVEL__AGG) {
×
UNCOV
1294
    STR_WITH_SIZE_TO_VARSTR(level, "agg", 3);
×
UNCOV
1295
  } else if (pTask->info.taskLevel == TASK_LEVEL__SINK) {
×
UNCOV
1296
    STR_WITH_SIZE_TO_VARSTR(level, "sink", 4);
×
1297
  }
1298

UNCOV
1299
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1300
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1301

UNCOV
1302
  code = colDataSetVal(pColInfo, numOfRows, (const char *)level, false);
×
UNCOV
1303
  TSDB_CHECK_CODE(code, lino, _end);
×
1304

1305
  // status
UNCOV
1306
  char status[20 + VARSTR_HEADER_SIZE] = {0};
×
1307

UNCOV
1308
  const char *pStatus = streamTaskGetStatusStr(pe->status);
×
UNCOV
1309
  STR_TO_VARSTR(status, pStatus);
×
1310

1311
  // status
UNCOV
1312
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1313
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1314

UNCOV
1315
  code = colDataSetVal(pColInfo, numOfRows, (const char *)status, false);
×
UNCOV
1316
  TSDB_CHECK_CODE(code, lino, _end);
×
1317

1318
  // stage
UNCOV
1319
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1320
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1321

UNCOV
1322
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->stage, false);
×
UNCOV
1323
  TSDB_CHECK_CODE(code, lino, _end);
×
1324

1325
  // input queue
UNCOV
1326
  char        vbuf[TSDB_STREAM_NOTIFY_STAT_LEN + 2] = {0};
×
UNCOV
1327
  char        buf[TSDB_STREAM_NOTIFY_STAT_LEN] = {0};
×
UNCOV
1328
  const char *queueInfoStr = "%4.2f MiB (%6.2f%)";
×
UNCOV
1329
  snprintf(buf, tListLen(buf), queueInfoStr, pe->inputQUsed, pe->inputRate);
×
UNCOV
1330
  STR_TO_VARSTR(vbuf, buf);
×
1331

UNCOV
1332
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1333
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1334

UNCOV
1335
  code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1336
  TSDB_CHECK_CODE(code, lino, _end);
×
1337

1338
  // input total
UNCOV
1339
  const char *formatTotalMb = "%7.2f MiB";
×
UNCOV
1340
  const char *formatTotalGb = "%7.2f GiB";
×
UNCOV
1341
  if (pe->procsTotal < 1024) {
×
UNCOV
1342
    snprintf(buf, tListLen(buf), formatTotalMb, pe->procsTotal);
×
1343
  } else {
UNCOV
1344
    snprintf(buf, tListLen(buf), formatTotalGb, pe->procsTotal / 1024);
×
1345
  }
1346

UNCOV
1347
  memset(vbuf, 0, tListLen(vbuf));
×
UNCOV
1348
  STR_TO_VARSTR(vbuf, buf);
×
1349

UNCOV
1350
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1351
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1352

UNCOV
1353
  code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1354
  TSDB_CHECK_CODE(code, lino, _end);
×
1355

1356
  // process throughput
UNCOV
1357
  const char *formatKb = "%7.2f KiB/s";
×
UNCOV
1358
  const char *formatMb = "%7.2f MiB/s";
×
UNCOV
1359
  if (pe->procsThroughput < 1024) {
×
UNCOV
1360
    snprintf(buf, tListLen(buf), formatKb, pe->procsThroughput);
×
1361
  } else {
UNCOV
1362
    snprintf(buf, tListLen(buf), formatMb, pe->procsThroughput / 1024);
×
1363
  }
1364

UNCOV
1365
  memset(vbuf, 0, tListLen(vbuf));
×
UNCOV
1366
  STR_TO_VARSTR(vbuf, buf);
×
1367

UNCOV
1368
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1369
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1370

UNCOV
1371
  code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1372
  TSDB_CHECK_CODE(code, lino, _end);
×
1373

1374
  // output total
UNCOV
1375
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1376
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1377

UNCOV
1378
  if (pTask->info.taskLevel == TASK_LEVEL__SINK) {
×
UNCOV
1379
    colDataSetNULL(pColInfo, numOfRows);
×
1380
  } else {
UNCOV
1381
    (void)tsnprintf(buf, sizeof(buf), formatTotalMb, pe->outputTotal);
×
UNCOV
1382
    memset(vbuf, 0, tListLen(vbuf));
×
UNCOV
1383
    STR_TO_VARSTR(vbuf, buf);
×
1384

UNCOV
1385
    code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1386
    TSDB_CHECK_CODE(code, lino, _end);
×
1387
  }
1388

1389
  // output throughput
UNCOV
1390
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1391
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1392

UNCOV
1393
  if (pTask->info.taskLevel == TASK_LEVEL__SINK) {
×
UNCOV
1394
    colDataSetNULL(pColInfo, numOfRows);
×
1395
  } else {
UNCOV
1396
    if (pe->outputThroughput < 1024) {
×
UNCOV
1397
      snprintf(buf, tListLen(buf), formatKb, pe->outputThroughput);
×
1398
    } else {
UNCOV
1399
      snprintf(buf, tListLen(buf), formatMb, pe->outputThroughput / 1024);
×
1400
    }
1401

UNCOV
1402
    memset(vbuf, 0, tListLen(vbuf));
×
1403
    STR_TO_VARSTR(vbuf, buf);
×
1404

UNCOV
1405
    code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1406
    TSDB_CHECK_CODE(code, lino, _end);
×
1407
  }
1408
  // info
UNCOV
1409
  if (pTask->info.taskLevel == TASK_LEVEL__SINK) {
×
UNCOV
1410
    const char *sinkStr = "%.2f MiB";
×
UNCOV
1411
    snprintf(buf, tListLen(buf), sinkStr, pe->sinkDataSize);
×
UNCOV
1412
  } else if (pTask->info.taskLevel == TASK_LEVEL__SOURCE) {  // offset info
×
UNCOV
1413
    if (pTask->info.trigger == STREAM_TRIGGER_FORCE_WINDOW_CLOSE) {
×
UNCOV
1414
      int32_t ret = taosFormatUtcTime(buf, tListLen(buf), pe->processedVer, precision);
×
UNCOV
1415
      if (ret != 0) {
×
UNCOV
1416
        mError("failed to format processed timewindow, skey:%" PRId64, pe->processedVer);
×
UNCOV
1417
        memset(buf, 0, tListLen(buf));
×
1418
      }
1419
    } else {
UNCOV
1420
      const char *offsetStr = "%" PRId64 " [%" PRId64 ", %" PRId64 "]";
×
UNCOV
1421
      snprintf(buf, tListLen(buf), offsetStr, pe->processedVer, pe->verRange.minVer, pe->verRange.maxVer);
×
1422
    }
1423
  } else {
UNCOV
1424
    memset(buf, 0, tListLen(buf));
×
1425
  }
1426

UNCOV
1427
  STR_TO_VARSTR(vbuf, buf);
×
1428

UNCOV
1429
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1430
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1431

UNCOV
1432
  code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1433
  TSDB_CHECK_CODE(code, lino, _end);
×
1434

1435
  // start_time
UNCOV
1436
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1437
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1438

UNCOV
1439
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->startTime, false);
×
UNCOV
1440
  TSDB_CHECK_CODE(code, lino, _end);
×
1441

1442
  // start id
UNCOV
1443
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1444
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1445

UNCOV
1446
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->startCheckpointId, false);
×
UNCOV
1447
  TSDB_CHECK_CODE(code, lino, _end);
×
1448

1449
  // start ver
UNCOV
1450
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1451
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1452

UNCOV
1453
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->startCheckpointVer, false);
×
UNCOV
1454
  TSDB_CHECK_CODE(code, lino, _end);
×
1455

1456
  // checkpoint time
UNCOV
1457
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1458
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1459

UNCOV
1460
  if (pe->checkpointInfo.latestTime != 0) {
×
UNCOV
1461
    code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->checkpointInfo.latestTime, false);
×
1462
  } else {
UNCOV
1463
    code = colDataSetVal(pColInfo, numOfRows, 0, true);
×
1464
  }
UNCOV
1465
  TSDB_CHECK_CODE(code, lino, _end);
×
1466

1467
  // checkpoint_id
UNCOV
1468
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1469
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1470

UNCOV
1471
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->checkpointInfo.latestId, false);
×
UNCOV
1472
  TSDB_CHECK_CODE(code, lino, _end);
×
1473

1474
  // checkpoint version
UNCOV
1475
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1476
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1477

UNCOV
1478
  code = colDataSetVal(pColInfo, numOfRows, (const char *)&pe->checkpointInfo.latestVer, false);
×
UNCOV
1479
  TSDB_CHECK_CODE(code, lino, _end);
×
1480

1481
  // checkpoint size
UNCOV
1482
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1483
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1484

UNCOV
1485
  colDataSetNULL(pColInfo, numOfRows);
×
1486

1487
  // checkpoint backup status
UNCOV
1488
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1489
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1490

UNCOV
1491
  code = colDataSetVal(pColInfo, numOfRows, 0, true);
×
UNCOV
1492
  TSDB_CHECK_CODE(code, lino, _end);
×
1493

1494
  // ds_err_info
UNCOV
1495
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1496
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1497

UNCOV
1498
  code = colDataSetVal(pColInfo, numOfRows, 0, true);
×
UNCOV
1499
  TSDB_CHECK_CODE(code, lino, _end);
×
1500

1501
  // history_task_id
UNCOV
1502
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1503
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1504

UNCOV
1505
  if (pe->hTaskId != 0) {
×
UNCOV
1506
    int64ToHexStr(pe->hTaskId, idstr, tListLen(idstr));
×
UNCOV
1507
    code = colDataSetVal(pColInfo, numOfRows, idstr, false);
×
1508
  } else {
UNCOV
1509
    code = colDataSetVal(pColInfo, numOfRows, 0, true);
×
1510
  }
UNCOV
1511
  TSDB_CHECK_CODE(code, lino, _end);
×
1512

1513
  // history_task_status
UNCOV
1514
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1515
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1516

UNCOV
1517
  code = colDataSetVal(pColInfo, numOfRows, 0, true);
×
UNCOV
1518
  TSDB_CHECK_CODE(code, lino, _end);
×
1519

1520
  // notify_event_stat
UNCOV
1521
  int32_t offset =0;
×
UNCOV
1522
  if (pe->notifyEventStat.notifyEventAddTimes > 0) {
×
UNCOV
1523
    offset += tsnprintf(buf + offset, sizeof(buf) - offset, "Add %" PRId64 "x, %" PRId64 " elems in %lfs; ",
×
1524
                        pe->notifyEventStat.notifyEventAddTimes, pe->notifyEventStat.notifyEventAddElems,
1525
                        pe->notifyEventStat.notifyEventAddCostSec);
1526
  }
UNCOV
1527
  if (pe->notifyEventStat.notifyEventPushTimes > 0) {
×
UNCOV
1528
    offset += tsnprintf(buf + offset, sizeof(buf) - offset, "Push %" PRId64 "x, %" PRId64 " elems in %lfs; ",
×
1529
                        pe->notifyEventStat.notifyEventPushTimes, pe->notifyEventStat.notifyEventPushElems,
1530
                        pe->notifyEventStat.notifyEventPushCostSec);
1531
  }
UNCOV
1532
  if (pe->notifyEventStat.notifyEventPackTimes > 0) {
×
UNCOV
1533
    offset += tsnprintf(buf + offset, sizeof(buf) - offset, "Pack %" PRId64 "x, %" PRId64 " elems in %lfs; ",
×
1534
                        pe->notifyEventStat.notifyEventPackTimes, pe->notifyEventStat.notifyEventPackElems,
1535
                        pe->notifyEventStat.notifyEventPackCostSec);
1536
  }
UNCOV
1537
  if (pe->notifyEventStat.notifyEventSendTimes > 0) {
×
UNCOV
1538
    offset += tsnprintf(buf + offset, sizeof(buf) - offset, "Send %" PRId64 "x, %" PRId64 " elems in %lfs; ",
×
1539
                        pe->notifyEventStat.notifyEventSendTimes, pe->notifyEventStat.notifyEventSendElems,
1540
                        pe->notifyEventStat.notifyEventSendCostSec);
1541
  }
UNCOV
1542
  if (pe->notifyEventStat.notifyEventHoldElems > 0) {
×
UNCOV
1543
    offset += tsnprintf(buf + offset, sizeof(buf) - offset, "[Hold %" PRId64 " elems] ",
×
1544
                        pe->notifyEventStat.notifyEventHoldElems);
1545
  }
UNCOV
1546
  TSDB_CHECK_CONDITION(offset < sizeof(buf), code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
×
UNCOV
1547
  buf[offset] = '\0';
×
1548

UNCOV
1549
  STR_TO_VARSTR(vbuf, buf);
×
1550

UNCOV
1551
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
UNCOV
1552
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1553

UNCOV
1554
  if (offset == 0) {
×
UNCOV
1555
    colDataSetNULL(pColInfo, numOfRows);
×
1556
  } else {
UNCOV
1557
    code = colDataSetVal(pColInfo, numOfRows, (const char *)vbuf, false);
×
UNCOV
1558
    TSDB_CHECK_CODE(code, lino, _end);
×
1559
  }
1560

UNCOV
1561
_end:
×
1562
  if (code) {
×
UNCOV
1563
    mError("error happens during build task attr result blocks, lino:%d, code:%s", lino, tstrerror(code));
×
1564
  }
UNCOV
1565
  return code;
×
1566
}
1567

UNCOV
1568
static bool isNodeEpsetChanged(const SEpSet *pPrevEpset, const SEpSet *pCurrent) {
×
UNCOV
1569
  const SEp *pEp = GET_ACTIVE_EP(pPrevEpset);
×
UNCOV
1570
  const SEp *p = GET_ACTIVE_EP(pCurrent);
×
1571

UNCOV
1572
  if (pEp->port == p->port && strncmp(pEp->fqdn, p->fqdn, TSDB_FQDN_LEN) == 0) {
×
UNCOV
1573
    return false;
×
1574
  }
UNCOV
1575
  return true;
×
1576
}
1577

UNCOV
1578
void mndDestroyVgroupChangeInfo(SVgroupChangeInfo *pInfo) {
×
UNCOV
1579
  if (pInfo != NULL) {
×
UNCOV
1580
    taosArrayDestroy(pInfo->pUpdateNodeList);
×
UNCOV
1581
    taosHashCleanup(pInfo->pDBMap);
×
1582
  }
1583
}
×
1584

1585
// 1. increase the replica does not affect the stream process.
1586
// 2. decreasing the replica may affect the stream task execution in the way that there is one or more running stream
1587
// tasks on the will be removed replica.
1588
// 3. vgroup redistribution is an combination operation of first increase replica and then decrease replica. So we
1589
// will handle it as mentioned in 1 & 2 items.
1590
int32_t mndFindChangedNodeInfo(SMnode *pMnode, const SArray *pPrevNodeList, const SArray *pNodeList,
×
1591
                               SVgroupChangeInfo *pInfo) {
UNCOV
1592
  int32_t code = 0;
×
UNCOV
1593
  int32_t lino = 0;
×
1594

UNCOV
1595
  if (pInfo == NULL) {
×
UNCOV
1596
    return TSDB_CODE_INVALID_PARA;
×
1597
  }
1598

UNCOV
1599
  pInfo->pUpdateNodeList = taosArrayInit(4, sizeof(SNodeUpdateInfo));
×
UNCOV
1600
  pInfo->pDBMap = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, HASH_NO_LOCK);
×
1601

UNCOV
1602
  if (pInfo->pUpdateNodeList == NULL || pInfo->pDBMap == NULL) {
×
UNCOV
1603
    mndDestroyVgroupChangeInfo(pInfo);
×
UNCOV
1604
    TSDB_CHECK_NULL(NULL, code, lino, _err, terrno);
×
1605
  }
1606

UNCOV
1607
  int32_t numOfNodes = taosArrayGetSize(pPrevNodeList);
×
UNCOV
1608
  for (int32_t i = 0; i < numOfNodes; ++i) {
×
UNCOV
1609
    SNodeEntry *pPrevEntry = taosArrayGet(pPrevNodeList, i);
×
UNCOV
1610
    if (pPrevEntry == NULL) {
×
UNCOV
1611
      continue;
×
1612
    }
1613

UNCOV
1614
    int32_t num = taosArrayGetSize(pNodeList);
×
1615
    for (int32_t j = 0; j < num; ++j) {
×
1616
      SNodeEntry *pCurrent = taosArrayGet(pNodeList, j);
×
UNCOV
1617
      if (pCurrent == NULL) {
×
UNCOV
1618
        continue;
×
1619
      }
1620

UNCOV
1621
      if (pCurrent->nodeId == pPrevEntry->nodeId) {
×
UNCOV
1622
        if (pPrevEntry->stageUpdated || isNodeEpsetChanged(&pPrevEntry->epset, &pCurrent->epset)) {
×
UNCOV
1623
          const SEp *pPrevEp = GET_ACTIVE_EP(&pPrevEntry->epset);
×
1624

UNCOV
1625
          char buf[256] = {0};
×
UNCOV
1626
          code = epsetToStr(&pCurrent->epset, buf, tListLen(buf));  // ignore this error
×
UNCOV
1627
          if (code) {
×
UNCOV
1628
            mError("failed to convert epset string, code:%s", tstrerror(code));
×
UNCOV
1629
            TSDB_CHECK_CODE(code, lino, _err);
×
1630
          }
1631

UNCOV
1632
          mDebug("nodeId:%d restart/epset changed detected, old:%s:%d -> new:%s, stageUpdate:%d", pCurrent->nodeId,
×
1633
                 pPrevEp->fqdn, pPrevEp->port, buf, pPrevEntry->stageUpdated);
1634

UNCOV
1635
          SNodeUpdateInfo updateInfo = {.nodeId = pPrevEntry->nodeId};
×
UNCOV
1636
          epsetAssign(&updateInfo.prevEp, &pPrevEntry->epset);
×
UNCOV
1637
          epsetAssign(&updateInfo.newEp, &pCurrent->epset);
×
1638

UNCOV
1639
          void *p = taosArrayPush(pInfo->pUpdateNodeList, &updateInfo);
×
UNCOV
1640
          TSDB_CHECK_NULL(p, code, lino, _err, terrno);
×
1641
        }
1642

1643
        // todo handle the snode info
UNCOV
1644
        if (pCurrent->nodeId != SNODE_HANDLE) {
×
1645
          SVgObj *pVgroup = mndAcquireVgroup(pMnode, pCurrent->nodeId);
×
1646
          code = taosHashPut(pInfo->pDBMap, pVgroup->dbName, strlen(pVgroup->dbName), NULL, 0);
×
1647
          mndReleaseVgroup(pMnode, pVgroup);
×
1648
          TSDB_CHECK_CODE(code, lino, _err);
×
1649
        }
1650

UNCOV
1651
        break;
×
1652
      }
1653
    }
1654
  }
1655

UNCOV
1656
  return code;
×
1657

UNCOV
1658
_err:
×
UNCOV
1659
  mError("failed to find node change info, code:%s at %s line:%d", tstrerror(code), __func__, lino);
×
UNCOV
1660
  mndDestroyVgroupChangeInfo(pInfo);
×
UNCOV
1661
  return code;
×
1662
}
1663

UNCOV
1664
static int32_t doCheckForUpdated(SMnode *pMnode, SArray **ppNodeSnapshot) {
×
UNCOV
1665
  bool              allReady = false;
×
UNCOV
1666
  bool              nodeUpdated = false;
×
1667
  SVgroupChangeInfo changeInfo = {0};
×
1668

UNCOV
1669
  int32_t numOfNodes = extractStreamNodeList(pMnode);
×
1670

UNCOV
1671
  if (numOfNodes == 0) {
×
UNCOV
1672
    mDebug("stream task node change checking done, no vgroups exist, do nothing");
×
UNCOV
1673
    execInfo.ts = taosGetTimestampSec();
×
UNCOV
1674
    return false;
×
1675
  }
1676

UNCOV
1677
  for (int32_t i = 0; i < numOfNodes; ++i) {
×
1678
    SNodeEntry *pNodeEntry = taosArrayGet(execInfo.pNodeList, i);
×
UNCOV
1679
    if (pNodeEntry == NULL) {
×
UNCOV
1680
      continue;
×
1681
    }
1682

UNCOV
1683
    if (pNodeEntry->stageUpdated) {
×
UNCOV
1684
      mDebug("stream task not ready due to node update detected, checkpoint not issued");
×
UNCOV
1685
      return true;
×
1686
    }
1687
  }
1688

UNCOV
1689
  int32_t code = mndTakeVgroupSnapshot(pMnode, &allReady, ppNodeSnapshot);
×
UNCOV
1690
  if (code) {
×
UNCOV
1691
    mError("failed to get the vgroup snapshot, ignore it and continue");
×
1692
  }
1693

UNCOV
1694
  if (!allReady) {
×
UNCOV
1695
    mWarn("not all vnodes ready, quit from vnodes status check");
×
UNCOV
1696
    return true;
×
1697
  }
1698

UNCOV
1699
  code = mndFindChangedNodeInfo(pMnode, execInfo.pNodeList, *ppNodeSnapshot, &changeInfo);
×
UNCOV
1700
  if (code) {
×
UNCOV
1701
    nodeUpdated = false;
×
1702
  } else {
UNCOV
1703
    nodeUpdated = (taosArrayGetSize(changeInfo.pUpdateNodeList) > 0);
×
UNCOV
1704
    if (nodeUpdated) {
×
UNCOV
1705
      mDebug("stream tasks not ready due to node update");
×
1706
    }
1707
  }
1708

UNCOV
1709
  mndDestroyVgroupChangeInfo(&changeInfo);
×
UNCOV
1710
  return nodeUpdated;
×
1711
}
1712

1713
// check if the node update happens or not
UNCOV
1714
bool mndStreamNodeIsUpdated(SMnode *pMnode) {
×
UNCOV
1715
  SArray *pNodeSnapshot = NULL;
×
1716

UNCOV
1717
  streamMutexLock(&execInfo.lock);
×
UNCOV
1718
  bool updated = doCheckForUpdated(pMnode, &pNodeSnapshot);
×
UNCOV
1719
  streamMutexUnlock(&execInfo.lock);
×
1720

UNCOV
1721
  taosArrayDestroy(pNodeSnapshot);
×
UNCOV
1722
  return updated;
×
1723
}
1724

UNCOV
1725
int32_t mndCheckForSnode(SMnode *pMnode, SDbObj *pSrcDb) {
×
UNCOV
1726
  SSdb      *pSdb = pMnode->pSdb;
×
UNCOV
1727
  void      *pIter = NULL;
×
UNCOV
1728
  SSnodeObj *pObj = NULL;
×
1729

UNCOV
1730
  if (pSrcDb->cfg.replications == 1) {
×
UNCOV
1731
    return TSDB_CODE_SUCCESS;
×
1732
  } else {
1733
    while (1) {
UNCOV
1734
      pIter = sdbFetch(pSdb, SDB_SNODE, pIter, (void **)&pObj);
×
UNCOV
1735
      if (pIter == NULL) {
×
UNCOV
1736
        break;
×
1737
      }
1738

UNCOV
1739
      sdbRelease(pSdb, pObj);
×
UNCOV
1740
      sdbCancelFetch(pSdb, pIter);
×
UNCOV
1741
      return TSDB_CODE_SUCCESS;
×
1742
    }
1743

UNCOV
1744
    mError("snode not existed when trying to create stream in db with multiple replica");
×
UNCOV
1745
    return TSDB_CODE_SNODE_NOT_DEPLOYED;
×
1746
  }
1747
}
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