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

taosdata / TDengine / #4820

20 Oct 2025 09:08AM UTC coverage: 61.353% (-0.04%) from 61.392%
#4820

push

travis-ci

happyguoxy
add alarm result print

156500 of 324369 branches covered (48.25%)

Branch coverage included in aggregate %.

207878 of 269535 relevant lines covered (77.12%)

242602011.17 hits per line

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

66.07
/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
#include "mndSnode.h"
24

25
bool mstWaitLock(SRWLatch* pLock, bool readLock) {
72,569,684✔
26
  if (readLock) {
72,569,684✔
27
    while (taosRTryLockLatch(pLock)) {
67,173,720✔
28
      taosMsleep(1);
65,189✔
29
    }
30

31
    return true;
67,108,531✔
32
  }
33

34
  taosWWaitLockLatch(pLock);
5,461,153✔
35

36
  return true;
5,461,153✔
37
}
38

39
void mstDestroySStmVgStreamStatus(void* p) { 
766,605✔
40
  SStmVgStreamStatus* pStatus = (SStmVgStreamStatus*)p;
766,605✔
41
  taosArrayDestroy(pStatus->trigReaders); 
766,605✔
42
  taosArrayDestroy(pStatus->calcReaders); 
766,605✔
43
}
766,605✔
44

45
void mstDestroySStmSnodeStreamStatus(void* p) { 
479,938✔
46
  SStmSnodeStreamStatus* pStatus = (SStmSnodeStreamStatus*)p;
479,938✔
47
  for (int32_t i = 0; i < MND_STREAM_RUNNER_DEPLOY_NUM; ++i) {
1,919,752✔
48
    taosArrayDestroy(pStatus->runners[i]);
1,439,814✔
49
    pStatus->runners[i] = NULL;
1,439,814✔
50
  }
51
}
479,938✔
52

53

54
void mstDestroyVgroupStatus(SStmVgroupStatus* pVgStatus) {
×
55
  taosHashCleanup(pVgStatus->streamTasks);
×
56
  pVgStatus->streamTasks = NULL;
×
57
}
×
58

59
void mstDestroySStmTaskToDeployExt(void* param) {
2,827,622✔
60
  SStmTaskToDeployExt* pExt = (SStmTaskToDeployExt*)param;
2,827,622✔
61
  if (pExt->deployed) {
2,827,622✔
62
    return;
2,791,194✔
63
  }
64
  
65
  switch (pExt->deploy.task.type) {
36,428✔
66
    case STREAM_TRIGGER_TASK:
5,204✔
67
      taosArrayDestroy(pExt->deploy.msg.trigger.readerList);
5,204✔
68
      pExt->deploy.msg.trigger.readerList = NULL;
5,204✔
69
      taosArrayDestroy(pExt->deploy.msg.trigger.runnerList);
5,204✔
70
      pExt->deploy.msg.trigger.runnerList = NULL;
5,204✔
71
      break;
5,204✔
72
    case STREAM_RUNNER_TASK:
15,612✔
73
      taosMemoryFreeClear(pExt->deploy.msg.runner.pPlan);
15,612!
74
      break;
15,612✔
75
    default:  
15,612✔
76
      break;;
15,612✔
77
  }
78
}
79

80
void mstDestroyScanAddrList(void* param) {
567,921✔
81
  if (NULL == param) {
567,921!
82
    return;
×
83
  }
84
  SArray* pList = *(SArray**)param;
567,921✔
85
  taosArrayDestroy(pList);
567,921✔
86
}
87

88
void mstDestroySStmSnodeTasksDeploy(void* param) {
146,667✔
89
  SStmSnodeTasksDeploy* pSnode = (SStmSnodeTasksDeploy*)param;
146,667✔
90
  taosArrayDestroyEx(pSnode->triggerList, mstDestroySStmTaskToDeployExt);
146,667✔
91
  taosArrayDestroyEx(pSnode->runnerList, mstDestroySStmTaskToDeployExt);
146,667✔
92
}
146,667✔
93

94
void mstDestroySStmVgTasksToDeploy(void* param) {
360,871✔
95
  SStmVgTasksToDeploy* pVg = (SStmVgTasksToDeploy*)param;
360,871✔
96
  taosArrayDestroyEx(pVg->taskList, mstDestroySStmTaskToDeployExt);
360,871✔
97
}
360,871✔
98

99
void mstDestroySStmSnodeStatus(void* param) {
127,551✔
100
  SStmSnodeStatus* pSnode = (SStmSnodeStatus*)param;
127,551✔
101
  taosHashCleanup(pSnode->streamTasks);
127,551✔
102
}
127,551✔
103

104
void mstDestroySStmVgroupStatus(void* param) {
280,121✔
105
  SStmVgroupStatus* pVg = (SStmVgroupStatus*)param;
280,121✔
106
  taosHashCleanup(pVg->streamTasks);
280,121✔
107
}
280,121✔
108

109
void mstResetSStmStatus(SStmStatus* pStatus) {
437,624✔
110
  (void)mstWaitLock(&pStatus->resetLock, false);
437,624✔
111

112
  taosArrayDestroy(pStatus->trigReaders);
437,624✔
113
  pStatus->trigReaders = NULL;
437,624✔
114
  taosArrayDestroy(pStatus->trigOReaders);
437,624✔
115
  pStatus->trigOReaders = NULL;
437,624✔
116
  pStatus->calcReaders = tdListFree(pStatus->calcReaders);
437,624✔
117
  if (pStatus->triggerTask) {
437,624!
118
    (void)mstWaitLock(&pStatus->triggerTask->detailStatusLock, false);
437,624✔
119
    taosMemoryFreeClear(pStatus->triggerTask->detailStatus);
437,624!
120
    taosWUnLockLatch(&pStatus->triggerTask->detailStatusLock);
437,624✔
121
  }
122
  taosMemoryFreeClear(pStatus->triggerTask);
437,624!
123
  for (int32_t i = 0; i < MND_STREAM_RUNNER_DEPLOY_NUM; ++i) {
1,750,496✔
124
    taosArrayDestroy(pStatus->runners[i]);
1,312,872✔
125
    pStatus->runners[i] = NULL;
1,312,872✔
126
  }
127

128
  taosWUnLockLatch(&pStatus->resetLock);
437,624✔
129
}
437,624✔
130

131
void mstDestroySStmStatus(void* param) {
417,953✔
132
  SStmStatus* pStatus = (SStmStatus*)param;
417,953✔
133
  taosMemoryFreeClear(pStatus->streamName);
417,953!
134

135
  mstResetSStmStatus(pStatus);
417,953✔
136

137
  taosWLockLatch(&pStatus->userRecalcLock);
417,953✔
138
  taosArrayDestroy(pStatus->userRecalcList);
417,953✔
139
  taosWUnLockLatch(&pStatus->userRecalcLock);
417,953✔
140

141
  tFreeSCMCreateStreamReq(pStatus->pCreate);
417,953✔
142
  taosMemoryFreeClear(pStatus->pCreate);  
417,953!
143
}
417,953✔
144

145
void mstDestroySStmAction(void* param) {
590,048✔
146
  SStmAction* pAction = (SStmAction*)param;
590,048✔
147

148
  taosArrayDestroy(pAction->undeploy.taskList);
590,048✔
149
  taosArrayDestroy(pAction->recalc.recalcList);
590,048✔
150
}
590,048✔
151

152
void mstClearSStmStreamDeploy(SStmStreamDeploy* pDeploy) {
534,705✔
153
  pDeploy->readerTasks = NULL;
534,705✔
154
  pDeploy->triggerTask = NULL;
534,705✔
155
  pDeploy->runnerTasks = NULL;
534,705✔
156
}
534,705✔
157

158
int32_t mstIsStreamDropped(SMnode *pMnode, int64_t streamId, bool* dropped) {
187,013✔
159
  SSdb   *pSdb = pMnode->pSdb;
187,013✔
160
  void   *pIter = NULL;
187,013✔
161
  
162
  while (1) {
706,394✔
163
    SStreamObj *pStream = NULL;
893,407✔
164
    pIter = sdbFetch(pSdb, SDB_STREAM, pIter, (void **)&pStream);
893,407✔
165
    if (pIter == NULL) break;
893,407✔
166

167
    if (pStream->pCreate->streamId == streamId) {
820,580✔
168
      *dropped = pStream->userDropped ? true : false;
114,186✔
169
      sdbRelease(pSdb, pStream);
114,186✔
170
      sdbCancelFetch(pSdb, pIter);
114,186✔
171
      mstsDebug("stream found, dropped:%d", *dropped);
114,186!
172
      return TSDB_CODE_SUCCESS;
114,186✔
173
    }
174
    
175
    sdbRelease(pSdb, pStream);
706,394✔
176
  }
177

178
  *dropped = true;
72,827✔
179

180
  return TSDB_CODE_SUCCESS;
72,827✔
181
}
182

183
typedef struct SStmCheckDbInUseCtx {
184
  bool* dbStream;
185
  bool* vtableStream;
186
  bool  ignoreCurrDb;
187
} SStmCheckDbInUseCtx;
188

189
static bool mstChkSetDbInUse(SMnode *pMnode, void *pObj, void *p1, void *p2, void *p3) {
191,952✔
190
  SStreamObj *pStream = pObj;
191,952✔
191
  if (atomic_load_8(&pStream->userDropped)) {
191,952!
192
    return true;
×
193
  }
194

195
  SStmCheckDbInUseCtx* pCtx = (SStmCheckDbInUseCtx*)p2;
191,952✔
196
  if (pCtx->ignoreCurrDb && 0 == strcmp(pStream->pCreate->streamDB, p1)) {
191,952!
197
    return true;
13,858✔
198
  }
199
  
200
  if (pStream->pCreate->triggerDB && 0 == strcmp(pStream->pCreate->triggerDB, p1)) {
178,094!
201
    *pCtx->dbStream = true;
4,016✔
202
    return false;
4,016✔
203
  }
204

205
  int32_t calcDBNum = taosArrayGetSize(pStream->pCreate->calcDB);
174,078✔
206
  for (int32_t i = 0; i < calcDBNum; ++i) {
344,140✔
207
    char* calcDB = taosArrayGetP(pStream->pCreate->calcDB, i);
174,078✔
208
    if (0 == strcmp(calcDB, p1)) {
174,078!
209
      *pCtx->dbStream = true;
4,016✔
210
      return false;
4,016✔
211
    }
212
  }
213

214
  if (pStream->pCreate->vtableCalc || STREAM_IS_VIRTUAL_TABLE(pStream->pCreate->triggerTblType, pStream->pCreate->flags)) {
170,062!
215
    *pCtx->vtableStream = true;
45,390✔
216
    return true;
45,390✔
217
  }
218
  
219
  return true;
124,672✔
220
}
221

222
void mstCheckDbInUse(SMnode *pMnode, char *dbFName, bool *dbStream, bool *vtableStream, bool ignoreCurrDb) {
2,020,057✔
223
  int32_t streamNum = sdbGetSize(pMnode->pSdb, SDB_STREAM);
2,020,057✔
224
  if (streamNum <= 0) {
2,020,057✔
225
    return;
1,988,546✔
226
  }
227

228
  SStmCheckDbInUseCtx ctx = {dbStream, vtableStream, ignoreCurrDb};
31,511✔
229
  sdbTraverse(pMnode->pSdb, SDB_STREAM, mstChkSetDbInUse, dbFName, &ctx, NULL);
31,511✔
230
}
231

232
static void mstShowStreamStatus(char *dst, int8_t status, int32_t bufLen) {
×
233
  if (status == STREAM_STATUS_INIT) {
×
234
    tstrncpy(dst, "init", bufLen);
×
235
  } else if (status == STREAM_STATUS_RUNNING) {
×
236
    tstrncpy(dst, "running", bufLen);
×
237
  } else if (status == STREAM_STATUS_STOPPED) {
×
238
    tstrncpy(dst, "stopped", bufLen);
×
239
  } else if (status == STREAM_STATUS_FAILED) {
×
240
    tstrncpy(dst, "failed", bufLen);
×
241
  }
242
}
×
243

244
int32_t mstCheckSnodeExists(SMnode *pMnode) {
×
245
  SSdb      *pSdb = pMnode->pSdb;
×
246
  void      *pIter = NULL;
×
247
  SSnodeObj *pObj = NULL;
×
248

249
  while (1) {
250
    pIter = sdbFetch(pSdb, SDB_SNODE, pIter, (void **)&pObj);
×
251
    if (pIter == NULL) {
×
252
      break;
×
253
    }
254

255
    sdbRelease(pSdb, pObj);
×
256
    sdbCancelFetch(pSdb, pIter);
×
257
    return TSDB_CODE_SUCCESS;
×
258
  }
259

260
  return TSDB_CODE_SNODE_NOT_DEPLOYED;
×
261
}
262

263
void mstSetTaskStatusFromMsg(SStmGrpCtx* pCtx, SStmTaskStatus* pTask, SStmTaskStatusMsg* pMsg) {
×
264
  pTask->id.taskId = pMsg->taskId;
×
265
  pTask->id.deployId = pMsg->deployId;
×
266
  pTask->id.seriousId = pMsg->seriousId;
×
267
  pTask->id.nodeId = pMsg->nodeId;
×
268
  pTask->id.taskIdx = pMsg->taskIdx;
×
269

270
  pTask->type = pMsg->type;
×
271
  pTask->flags = pMsg->flags;
×
272
  pTask->status = pMsg->status;
×
273
  pTask->lastUpTs = pCtx->currTs;
×
274
}
×
275

276
bool mndStreamActionDequeue(SStmActionQ* pQueue, SStmQNode **param) {
2,180,063✔
277
  while (0 == atomic_load_64(&pQueue->qRemainNum)) {
2,180,063✔
278
    return false;
1,721,352✔
279
  }
280

281
  SStmQNode *orig = pQueue->head;
458,711✔
282

283
  SStmQNode *node = pQueue->head->next;
458,711✔
284
  pQueue->head = pQueue->head->next;
458,711✔
285

286
  *param = node;
458,711✔
287

288
  taosMemoryFreeClear(orig);
458,711!
289

290
  (void)atomic_sub_fetch_64(&pQueue->qRemainNum, 1);
458,711✔
291

292
  return true;
458,711✔
293
}
294

295
void mndStreamActionEnqueue(SStmActionQ* pQueue, SStmQNode* param) {
458,711✔
296
  taosWLockLatch(&pQueue->lock);
458,711✔
297
  pQueue->tail->next = param;
458,711✔
298
  pQueue->tail = param;
458,711✔
299
  taosWUnLockLatch(&pQueue->lock);
458,711✔
300

301
  (void)atomic_add_fetch_64(&pQueue->qRemainNum, 1);
458,711✔
302
}
458,711✔
303

304
char* mstGetStreamActionString(int32_t action) {
284,733✔
305
  switch (action) {
284,733!
306
    case STREAM_ACT_DEPLOY:
284,733✔
307
      return "DEPLOY";
284,733✔
308
    case STREAM_ACT_UNDEPLOY:
×
309
      return "UNDEPLOY";
×
310
    case STREAM_ACT_START:
×
311
      return "START";
×
312
    case STREAM_ACT_UPDATE_TRIGGER:
×
313
      return "UPDATE TRIGGER";
×
314
    case STREAM_ACT_RECALC:
×
315
      return "USER RECALC";
×
316
    default:
×
317
      break;
×
318
  }
319

320
  return "UNKNOWN";
×
321
}
322

323
void mstPostStreamAction(SStmActionQ*       actionQ, int64_t streamId, char* streamName, void* param, bool userAction, int32_t action) {
445,701✔
324
  SStmQNode *pNode = taosMemoryMalloc(sizeof(SStmQNode));
445,701!
325
  if (NULL == pNode) {
445,701!
326
    taosMemoryFreeClear(param);
×
327
    mstsError("%s failed at line %d, error:%s", __FUNCTION__, __LINE__, tstrerror(terrno));
×
328
    return;
×
329
  }
330

331
  pNode->type = action;
445,701✔
332
  pNode->streamAct = true;
445,701✔
333
  pNode->action.stream.streamId = streamId;
445,701✔
334
  TAOS_STRCPY(pNode->action.stream.streamName, streamName);
445,701!
335
  pNode->action.stream.userAction = userAction;
445,701✔
336
  pNode->action.stream.actionParam = param;
445,701✔
337
  
338
  pNode->next = NULL;
445,701✔
339

340
  mndStreamActionEnqueue(actionQ, pNode);
445,701✔
341

342
  mstsDebug("stream action %s posted enqueue", mstGetStreamActionString(action));
445,701✔
343
}
344

345
void mstPostTaskAction(SStmActionQ*        actionQ, SStmTaskAction* pAction, int32_t action) {
13,010✔
346
  SStmQNode *pNode = taosMemoryMalloc(sizeof(SStmQNode));
13,010!
347
  if (NULL == pNode) {
13,010!
348
    int64_t streamId = pAction->streamId;
×
349
    mstsError("%s failed at line %d, error:%s", __FUNCTION__, __LINE__, tstrerror(terrno));
×
350
    return;
×
351
  }
352

353
  pNode->type = action;
13,010✔
354
  pNode->streamAct = false;
13,010✔
355
  pNode->action.task = *pAction;
13,010✔
356
  
357
  pNode->next = NULL;
13,010✔
358

359
  mndStreamActionEnqueue(actionQ, pNode);
13,010✔
360
}
361

362
void mstDestroyDbVgroupsHash(SSHashObj *pDbVgs) {
76,205✔
363
  int32_t iter = 0;
76,205✔
364
  SDBVgHashInfo* pVg = NULL;
76,205✔
365
  void* p = NULL;
76,205✔
366
  while (NULL != (p = tSimpleHashIterate(pDbVgs, p, &iter))) {
562,290✔
367
    pVg = (SDBVgHashInfo*)p;
486,085✔
368
    taosArrayDestroy(pVg->vgArray);
486,085✔
369
  }
370
  
371
  tSimpleHashCleanup(pDbVgs);
76,205✔
372
}
76,205✔
373

374

375
int32_t mstBuildDBVgroupsMap(SMnode* pMnode, SSHashObj** ppRes) {
76,205✔
376
  void*   pIter = NULL;
76,205✔
377
  int32_t code = TSDB_CODE_SUCCESS;
76,205✔
378
  int32_t lino = 0;
76,205✔
379
  SArray* pTarget = NULL;
76,205✔
380
  SArray* pNew = NULL;
76,205✔
381
  SDbObj* pDb = NULL;
76,205✔
382
  SDBVgHashInfo dbInfo = {0}, *pDbInfo = NULL;
76,205✔
383
  SVgObj* pVgroup = NULL;
76,205✔
384

385
  SSHashObj* pDbVgroup = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
76,205✔
386
  TSDB_CHECK_NULL(pDbVgroup, code, lino, _exit, terrno);
76,205!
387

388
  while (1) {
543,318✔
389
    pIter = sdbFetch(pMnode->pSdb, SDB_VGROUP, pIter, (void**)&pVgroup);
619,523✔
390
    if (pIter == NULL) {
619,523✔
391
      break;
76,205✔
392
    }
393

394
    pDbInfo = (SDBVgHashInfo*)tSimpleHashGet(pDbVgroup, pVgroup->dbName, strlen(pVgroup->dbName) + 1);
543,318!
395
    if (NULL == pDbInfo) {
543,318✔
396
      pNew = taosArrayInit(20, sizeof(SVGroupHashInfo));
486,085✔
397
      if (NULL == pNew) {
486,085!
398
        sdbRelease(pMnode->pSdb, pVgroup);
×
399
        sdbCancelFetch(pMnode->pSdb, pIter);
×
400
        pVgroup = NULL;
×
401
        TSDB_CHECK_NULL(pNew, code, lino, _exit, terrno);
×
402
      }
403
      
404
      pDb = mndAcquireDb(pMnode, pVgroup->dbName);
486,085✔
405
      if (NULL == pDb) {
486,085!
406
        sdbRelease(pMnode->pSdb, pVgroup);
×
407
        sdbCancelFetch(pMnode->pSdb, pIter);      
×
408
        pVgroup = NULL;
×
409
        TSDB_CHECK_NULL(pDb, code, lino, _exit, terrno);
×
410
      }
411
      dbInfo.vgSorted = false;
486,085✔
412
      dbInfo.hashMethod = pDb->cfg.hashMethod;
486,085✔
413
      dbInfo.hashPrefix = pDb->cfg.hashPrefix;
486,085✔
414
      dbInfo.hashSuffix = pDb->cfg.hashSuffix;
486,085✔
415
      dbInfo.vgArray = pNew;
486,085✔
416
      
417
      mndReleaseDb(pMnode, pDb);
486,085✔
418

419
      pTarget = pNew;
486,085✔
420
    } else {
421
      pTarget = pDbInfo->vgArray;
57,233✔
422
    }
423

424
    SVGroupHashInfo vgInfo = {.vgId = pVgroup->vgId, .hashBegin = pVgroup->hashBegin, .hashEnd = pVgroup->hashEnd};
543,318✔
425
    if (NULL == taosArrayPush(pTarget, &vgInfo)) {
543,318!
426
      sdbRelease(pMnode->pSdb, pVgroup);
×
427
      sdbCancelFetch(pMnode->pSdb, pIter);      
×
428
      pVgroup = NULL;
×
429
      TSDB_CHECK_NULL(NULL, code, lino, _exit, terrno);
×
430
    }
431

432
    if (NULL == pDbInfo) {
543,318✔
433
      code = tSimpleHashPut(pDbVgroup, pVgroup->dbName, strlen(pVgroup->dbName) + 1, &dbInfo, sizeof(dbInfo));
486,085!
434
      if (code) {
486,085!
435
        sdbRelease(pMnode->pSdb, pVgroup);
×
436
        sdbCancelFetch(pMnode->pSdb, pIter);      
×
437
        pVgroup = NULL;
×
438
        TAOS_CHECK_EXIT(code);
×
439
      }
440
      pNew = NULL;
486,085✔
441
    }
442

443
    sdbRelease(pMnode->pSdb, pVgroup);
543,318✔
444
    pVgroup = NULL;
543,318✔
445
  }
446

447
  *ppRes = pDbVgroup;
76,205✔
448
  
449
_exit:
76,205✔
450

451
  taosArrayDestroy(pNew);
76,205✔
452

453
  if (code) {
76,205!
454
    mstError("%s failed at line %d, error:%s", __FUNCTION__, lino, tstrerror(code));
×
455
  }
456

457
  return code;
76,205✔
458
}
459

460
int mstDbVgInfoComp(const void* lp, const void* rp) {
58,564✔
461
  SVGroupHashInfo* pLeft = (SVGroupHashInfo*)lp;
58,564✔
462
  SVGroupHashInfo* pRight = (SVGroupHashInfo*)rp;
58,564✔
463
  if (pLeft->hashBegin < pRight->hashBegin) {
58,564!
464
    return -1;
58,564✔
465
  } else if (pLeft->hashBegin > pRight->hashBegin) {
×
466
    return 1;
×
467
  }
468

469
  return 0;
×
470
}
471

472
int32_t mstTableHashValueComp(void const* lp, void const* rp) {
168,430✔
473
  uint32_t*    key = (uint32_t*)lp;
168,430✔
474
  SVgroupInfo* pVg = (SVgroupInfo*)rp;
168,430✔
475

476
  if (*key < pVg->hashBegin) {
168,430!
477
    return -1;
×
478
  } else if (*key > pVg->hashEnd) {
168,430✔
479
    return 1;
23,958✔
480
  }
481

482
  return 0;
144,472✔
483
}
484

485

486
int32_t mstGetTableVgId(SSHashObj* pDbVgroups, char* dbFName, char *tbName, int32_t* vgId) {
144,472✔
487
  int32_t code = 0;
144,472✔
488
  int32_t lino = 0;
144,472✔
489
  SVgroupInfo* vgInfo = NULL;
144,472✔
490
  char         tbFullName[TSDB_TABLE_FNAME_LEN];
144,472✔
491

492
  SDBVgHashInfo* dbInfo = (SDBVgHashInfo*)tSimpleHashGet(pDbVgroups, dbFName, strlen(dbFName) + 1);
144,472!
493
  if (NULL == dbInfo) {
144,472!
494
    mstError("db %s does not exist", dbFName);
×
495
    TAOS_CHECK_EXIT(TSDB_CODE_MND_DB_NOT_EXIST);
×
496
  }
497
  
498
  (void)snprintf(tbFullName, sizeof(tbFullName), "%s.%s", dbFName, tbName);
144,472✔
499
  uint32_t hashValue = taosGetTbHashVal(tbFullName, (uint32_t)strlen(tbFullName), dbInfo->hashMethod,
288,944!
500
                                        dbInfo->hashPrefix, dbInfo->hashSuffix);
144,472!
501

502
  if (!dbInfo->vgSorted) {
144,472!
503
    taosArraySort(dbInfo->vgArray, mstDbVgInfoComp);
76,205✔
504
    dbInfo->vgSorted = true;
76,205✔
505
  }
506

507
  vgInfo = taosArraySearch(dbInfo->vgArray, &hashValue, mstTableHashValueComp, TD_EQ);
144,472✔
508
  if (NULL == vgInfo) {
144,472!
509
    mstError("no hash range found for hash value [%u], dbFName:%s, numOfVgId:%d", hashValue, dbFName,
×
510
             (int32_t)taosArrayGetSize(dbInfo->vgArray));
511
    TAOS_CHECK_EXIT(TSDB_CODE_MND_STREAM_INTERNAL_ERROR);
×
512
  }
513

514
  *vgId = vgInfo->vgId;
144,472✔
515

516
_exit:
144,472✔
517

518
  if (code) {
144,472!
519
    mstError("%s failed at line %d, error:%s", __FUNCTION__, lino, tstrerror(code));
×
520
  }
521

522
  return code;
144,472✔
523
}
524

525

526
void mstLogSStreamObj(char* tips, SStreamObj* p) {
425,244✔
527
  if (!(stDebugFlag & DEBUG_DEBUG)) {
425,244✔
528
    return;
158,298✔
529
  }
530
  
531
  if (NULL == p) {
266,946!
532
    mstDebug("%s: stream is NULL", tips);
×
533
    return;
×
534
  }
535

536
  mstDebug("%s: stream obj", tips);
266,946!
537
  mstDebug("name:%s mainSnodeId:%d userDropped:%d userStopped:%d createTime:%" PRId64 " updateTime:%" PRId64,
266,946!
538
      p->name, p->mainSnodeId, p->userDropped, p->userStopped, p->createTime, p->updateTime);
539

540
  SCMCreateStreamReq* q = p->pCreate;
266,946✔
541
  if (NULL == q) {
266,946!
542
    mstDebug("stream pCreate is NULL");
×
543
    return;
×
544
  }
545

546
  int64_t streamId = q->streamId;
266,946✔
547
  int32_t calcDBNum = taosArrayGetSize(q->calcDB);
266,946✔
548
  int32_t calcScanNum = taosArrayGetSize(q->calcScanPlanList);
266,946✔
549
  int32_t notifyUrlNum = taosArrayGetSize(q->pNotifyAddrUrls);
266,946✔
550
  int32_t outColNum = taosArrayGetSize(q->outCols);
266,946✔
551
  int32_t outTagNum = taosArrayGetSize(q->outTags);
266,946✔
552
  int32_t forceOutColNum = taosArrayGetSize(q->forceOutCols);
266,946✔
553

554
  mstsDebugL("create_info: name:%s sql:%s streamDB:%s triggerDB:%s outDB:%s calcDBNum:%d triggerTblName:%s outTblName:%s "
266,946!
555
      "igExists:%d triggerType:%d igDisorder:%d deleteReCalc:%d deleteOutTbl:%d fillHistory:%d fillHistroyFirst:%d "
556
      "calcNotifyOnly:%d lowLatencyCalc:%d igNoDataTrigger:%d notifyUrlNum:%d notifyEventTypes:%d addOptions:%d notifyHistory:%d "
557
      "outColsNum:%d outTagsNum:%d maxDelay:%" PRId64 " fillHistoryStartTs:%" PRId64 " watermark:%" PRId64 " expiredTime:%" PRId64 " "
558
      "triggerTblType:%d triggerTblUid:%" PRIx64 " triggerTblSuid:%" PRIx64 " vtableCalc:%d outTblType:%d outStbExists:%d outStbUid:%" PRIu64 " outStbSversion:%d "
559
      "eventTypes:0x%" PRIx64 " flags:0x%" PRIx64 " tsmaId:0x%" PRIx64 " placeHolderBitmap:0x%" PRIx64 " calcTsSlotId:%d triTsSlotId:%d "
560
      "triggerTblVgId:%d outTblVgId:%d calcScanPlanNum:%d forceOutCols:%d",
561
      q->name, q->sql, q->streamDB, q->triggerDB, q->outDB, calcDBNum, q->triggerTblName, q->outTblName,
562
      q->igExists, q->triggerType, q->igDisorder, q->deleteReCalc, q->deleteOutTbl, q->fillHistory, q->fillHistoryFirst,
563
      q->calcNotifyOnly, q->lowLatencyCalc, q->igNoDataTrigger, notifyUrlNum, q->notifyEventTypes, q->addOptions, q->notifyHistory,
564
      outColNum, outTagNum, q->maxDelay, q->fillHistoryStartTime, q->watermark, q->expiredTime,
565
      q->triggerTblType, q->triggerTblUid, q->triggerTblSuid, q->vtableCalc, q->outTblType, q->outStbExists, q->outStbUid, q->outStbSversion,
566
      q->eventTypes, q->flags, q->tsmaId, q->placeHolderBitmap, q->calcTsSlotId, q->triTsSlotId,
567
      q->triggerTblVgId, q->outTblVgId, calcScanNum, forceOutColNum);
568

569
  switch (q->triggerType) {
266,946!
570
    case WINDOW_TYPE_INTERVAL: {
151,665✔
571
      SSlidingTrigger* t = &q->trigger.sliding;
151,665✔
572
      mstsDebug("sliding trigger options, intervalUnit:%d, slidingUnit:%d, offsetUnit:%d, soffsetUnit:%d, precision:%d, interval:%" PRId64 ", offset:%" PRId64 ", sliding:%" PRId64 ", soffset:%" PRId64, 
151,665!
573
          t->intervalUnit, t->slidingUnit, t->offsetUnit, t->soffsetUnit, t->precision, t->interval, t->offset, t->sliding, t->soffset);
574
      break;
151,665✔
575
    }  
576
    case WINDOW_TYPE_SESSION: {
10,366✔
577
      SSessionTrigger* t = &q->trigger.session;
10,366✔
578
      mstsDebug("session trigger options, slotId:%d, sessionVal:%" PRId64, t->slotId, t->sessionVal);
10,366!
579
      break;
10,366✔
580
    }
581
    case WINDOW_TYPE_STATE: {
52,674✔
582
      SStateWinTrigger* t = &q->trigger.stateWin;
52,674✔
583
      mstsDebug("state trigger options, slotId:%d, expr:%s, extend:%d, trueForDuration:%" PRId64, t->slotId, (char *)t->expr, t->extend, t->trueForDuration);
52,674!
584
      break;
52,674✔
585
    }
586
    case WINDOW_TYPE_EVENT:{
25,398✔
587
      SEventTrigger* t = &q->trigger.event;
25,398✔
588
      mstsDebug("event trigger options, startCond:%s, endCond:%s, trueForDuration:%" PRId64, (char*)t->startCond, (char*)t->endCond, t->trueForDuration);
25,398!
589
      break;
25,398✔
590
    }
591
    case WINDOW_TYPE_COUNT: {
16,148✔
592
      SCountTrigger* t = &q->trigger.count;
16,148✔
593
      mstsDebug("count trigger options, countVal:%" PRId64 ", sliding:%" PRId64 ", condCols:%s", t->countVal, t->sliding, (char*)t->condCols);
16,148!
594
      break;
16,148✔
595
    }
596
    case WINDOW_TYPE_PERIOD: {
10,695✔
597
      SPeriodTrigger* t = &q->trigger.period;
10,695✔
598
      mstsDebug("period trigger options, periodUnit:%d, offsetUnit:%d, precision:%d, period:%" PRId64 ", offset:%" PRId64, 
10,695!
599
          t->periodUnit, t->offsetUnit, t->precision, t->period, t->offset);
600
      break;
10,695✔
601
    }
602
    default:
×
603
      mstsDebug("unknown triggerType:%d", q->triggerType);
×
604
      break;
×
605
  }
606

607
  mstsDebugL("create_info: triggerCols:[%s]", (char*)q->triggerCols);
266,946!
608

609
  mstsDebugL("create_info: partitionCols:[%s]", (char*)q->partitionCols);
266,946!
610

611
  mstsDebugL("create_info: triggerScanPlan:[%s]", (char*)q->triggerScanPlan);
266,946!
612

613
  mstsDebugL("create_info: calcPlan:[%s]", (char*)q->calcPlan);
266,946!
614

615
  mstsDebugL("create_info: subTblNameExpr:[%s]", (char*)q->subTblNameExpr);
266,946!
616

617
  mstsDebugL("create_info: tagValueExpr:[%s]", (char*)q->tagValueExpr);
266,946!
618

619

620
  for (int32_t i = 0; i < calcDBNum; ++i) {
532,561✔
621
    char* dbName = taosArrayGetP(q->calcDB, i);
265,615✔
622
    mstsDebug("create_info: calcDB[%d] - %s", i, dbName);
265,615!
623
  }
624

625
  for (int32_t i = 0; i < calcScanNum; ++i) {
617,415✔
626
    SStreamCalcScan* pScan = taosArrayGet(q->calcScanPlanList, i);
350,469✔
627
    int32_t vgNum = taosArrayGetSize(pScan->vgList);
350,469✔
628
    mstsDebugL("create_info: calcScanPlan[%d] - readFromCache:%d vgNum:%d scanPlan:[%s]", i, pScan->readFromCache, vgNum, (char*)pScan->scanPlan);
350,469!
629
    for (int32_t v = 0; v < vgNum; ++v) {
700,938✔
630
      mstsDebug("create_info: calcScanPlan[%d] vg[%d] - vgId:%d", i, v, *(int32_t*)taosArrayGet(pScan->vgList, v));
350,469!
631
    }
632
  }
633

634
  for (int32_t i = 0; i < notifyUrlNum; ++i) {
297,559✔
635
    char* url = taosArrayGetP(q->pNotifyAddrUrls, i);
30,613✔
636
    mstsDebug("create_info: notifyUrl[%d] - %s", i, url);
30,613!
637
  }
638

639
  for (int32_t i = 0; i < outColNum; ++i) {
1,283,836✔
640
    SFieldWithOptions* o = taosArrayGet(q->outCols, i);
1,016,890✔
641
    mstsDebug("create_info: outCol[%d] - name:%s type:%d flags:%d bytes:%d compress:%u typeMod:%d", 
1,016,890!
642
        i, o->name, o->type, o->flags, o->bytes, o->compress, o->typeMod);
643
  }
644
      
645
}
646

647
void mstLogSStmTaskStatus(char* name, int64_t streamId, SStmTaskStatus* pTask, int32_t idx) {
1,863,425✔
648
  mstsDebug("%s[%d]: task %" PRIx64 " deployId:%d SID:%" PRId64 " nodeId:%d tidx:%d type:%s flags:%" PRIx64 " status:%s lastUpTs:%" PRId64, 
1,863,425!
649
      name, idx, pTask->id.taskId, pTask->id.deployId, pTask->id.seriousId, pTask->id.nodeId, pTask->id.taskIdx,
650
      gStreamTaskTypeStr[pTask->type], pTask->flags, gStreamStatusStr[pTask->status], pTask->lastUpTs);
651
}
1,863,425✔
652

653
void mstLogSStmStatus(char* tips, int64_t streamId, SStmStatus* p) {
437,624✔
654
  if (!(stDebugFlag & DEBUG_DEBUG)) {
437,624✔
655
    return;
160,968✔
656
  }
657
  
658
  if (NULL == p) {
276,656!
659
    mstsDebug("%s: stream status is NULL", tips);
×
660
    return;
×
661
  }
662

663
  int32_t trigReaderNum = taosArrayGetSize(p->trigReaders);
276,656✔
664
  int32_t trigOReaderNum = taosArrayGetSize(p->trigOReaders);
276,656✔
665
  int32_t calcReaderNum = MST_LIST_SIZE(p->calcReaders);
276,656!
666
  int32_t triggerNum = p->triggerTask ? 1 : 0;
276,656✔
667
  int32_t runnerNum = 0;
276,656✔
668

669
  for (int32_t i = 0; i < p->runnerDeploys; ++i) {
1,102,631✔
670
    runnerNum += taosArrayGetSize(p->runners[i]);
825,975✔
671
  }
672

673
  mstsDebug("%s: stream status", tips);
276,656!
674
  mstsDebug("name:%s runnerNum:%d runnerDeploys:%d runnerReplica:%d lastActionTs:%" PRId64
276,656!
675
           " trigReaders:%d trigOReaders:%d calcReaders:%d trigger:%d runners:%d",
676
      p->streamName, p->runnerNum, p->runnerDeploys, p->runnerReplica, p->lastActionTs,
677
      trigReaderNum, trigOReaderNum, calcReaderNum, triggerNum, runnerNum);
678

679
  SStmTaskStatus* pTask = NULL;
276,656✔
680
  for (int32_t i = 0; i < trigReaderNum; ++i) {
681,327✔
681
    pTask = taosArrayGet(p->trigReaders, i);
404,671✔
682
    mstLogSStmTaskStatus("trigReader task", streamId, pTask, i);
404,671✔
683
  }
684

685
  for (int32_t i = 0; i < trigOReaderNum; ++i) {
276,656!
686
    pTask = taosArrayGet(p->trigOReaders, i);
×
687
    mstLogSStmTaskStatus("trigOReader task", streamId, pTask, i);
×
688
  }
689

690
  SListNode* pNode = listHead(p->calcReaders);
276,656✔
691
  for (int32_t i = 0; i < calcReaderNum; ++i) {
552,799✔
692
    pTask = (SStmTaskStatus*)pNode->data;
276,143✔
693
    mstLogSStmTaskStatus("calcReader task", streamId, pTask, i);
276,143✔
694
    pNode = TD_DLIST_NODE_NEXT(pNode);
276,143✔
695
  }
696

697
  if (triggerNum > 0) {
276,656!
698
    mstLogSStmTaskStatus("trigger task", streamId, p->triggerTask, 0);
276,656✔
699
  }
700

701
  for (int32_t i = 0; i < p->runnerDeploys; ++i) {
1,102,631✔
702
    int32_t num = taosArrayGetSize(p->runners[i]);
825,975✔
703
    if (num <= 0) {
825,975!
704
      continue;
×
705
    }
706
    
707
    mstsDebug("the %dth deploy runners status", i);
825,975!
708
    for (int32_t m = 0; m < num; ++m) {
1,731,930✔
709
      pTask = taosArrayGet(p->runners[i], m);
905,955✔
710
      mstLogSStmTaskStatus("runner task", streamId, pTask, m);
905,955✔
711
    }
712
  }
713
      
714
}
715

716
bool mstEventPassIsolation(int32_t num, int32_t event) {
11,453,787✔
717
  bool ret = ((mStreamMgmt.lastTs[event].ts + num * MST_SHORT_ISOLATION_DURATION) <= mStreamMgmt.hCtx.currentTs);
11,453,787✔
718
  if (ret) {
11,453,787✔
719
    mstDebug("event %s passed %d isolation, last:%" PRId64 ", curr:%" PRId64, 
8,485,053✔
720
        gMndStreamEvent[event], num, mStreamMgmt.lastTs[event].ts, mStreamMgmt.hCtx.currentTs);
721
  }
722

723
  return ret;
11,453,787✔
724
}
725

726
bool mstEventHandledChkSet(int32_t event) {
8,485,053✔
727
  if (0 == atomic_val_compare_exchange_8((int8_t*)&mStreamMgmt.lastTs[event].handled, 0, 1)) {
8,485,053✔
728
    mstDebug("event %s set handled", gMndStreamEvent[event]);
532,148✔
729
    return true;
532,148✔
730
  }
731
  return false;
7,952,905✔
732
}
733

734
int32_t mstGetStreamStatusStr(SStreamObj* pStream, char* status, int32_t statusSize, char* msg, int32_t msgSize) {
1,946,552✔
735
  int8_t active = atomic_load_8(&mStreamMgmt.active), state = atomic_load_8(&mStreamMgmt.state);
1,946,552✔
736
  if (0 == active || MND_STM_STATE_NORMAL != state) {
1,946,552!
737
    mstDebug("mnode streamMgmt not in active mode, active:%d, state:%d", active, state);
786!
738
    STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_UNDEPLOYED], statusSize);
786!
739
    STR_WITH_MAXSIZE_TO_VARSTR(msg, "Mnode may be unstable, try again later", msgSize);
786!
740
    return TSDB_CODE_SUCCESS;
786✔
741
  }
742

743
  if (atomic_load_8(&pStream->userDropped)) {
1,945,766!
744
    STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_DROPPING], statusSize);
×
745
    STR_WITH_MAXSIZE_TO_VARSTR(msg, "", msgSize);
×
746
    return TSDB_CODE_SUCCESS;
×
747
  }
748

749
  if (atomic_load_8(&pStream->userStopped)) {
1,945,766✔
750
    STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_STOPPED], statusSize);
28,738!
751
    STR_WITH_MAXSIZE_TO_VARSTR(msg, "", msgSize);
28,738!
752
    return TSDB_CODE_SUCCESS;
28,738✔
753
  }
754

755
  (void)mstWaitLock(&mStreamMgmt.runtimeLock, true);
1,917,028✔
756
  
757
  SStmStatus* pStatus = (SStmStatus*)taosHashGet(mStreamMgmt.streamMap, &pStream->pCreate->streamId, sizeof(pStream->pCreate->streamId));
1,917,028✔
758
  if (NULL == pStatus) {
1,917,028✔
759
    STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_UNDEPLOYED], statusSize);
280,736!
760
    STR_WITH_MAXSIZE_TO_VARSTR(msg, "", msgSize);
280,736!
761
    goto _exit;
280,736✔
762
  }
763

764
  char tmpBuf[256];
1,636,292✔
765
  int8_t stopped = atomic_load_8(&pStatus->stopped);
1,636,292✔
766
  switch (stopped) {
1,636,292!
767
    case 1:
49,438✔
768
      STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_FAILED], statusSize);
49,438!
769
      snprintf(tmpBuf, sizeof(tmpBuf), "Last error: %s, Failed times: %" PRId64, tstrerror(pStatus->fatalError), pStatus->fatalRetryTimes);
49,438✔
770
      STR_WITH_MAXSIZE_TO_VARSTR(msg, tmpBuf, msgSize);
49,438!
771
      goto _exit;
49,438✔
772
      break;
773
    case 4:
×
774
      STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_FAILED], statusSize);
×
775
      snprintf(tmpBuf, sizeof(tmpBuf), "Error: %s", tstrerror(TSDB_CODE_GRANT_STREAM_EXPIRED));
×
776
      STR_WITH_MAXSIZE_TO_VARSTR(msg, tmpBuf, msgSize);
×
777
      goto _exit;
×
778
      break;
779
    default:
1,586,854✔
780
      break;
1,586,854✔
781
  }
782

783
  if (pStatus->triggerTask && STREAM_STATUS_RUNNING == pStatus->triggerTask->status) {
1,586,854✔
784
    STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_RUNNING], statusSize);
734,186!
785
    strcpy(tmpBuf, "Running start from: ");
734,186✔
786
    (void)formatTimestampLocal(&tmpBuf[strlen(tmpBuf)], pStatus->triggerTask->runningStartTs, TSDB_TIME_PRECISION_MILLI);
734,186✔
787
    STR_WITH_MAXSIZE_TO_VARSTR(msg, tmpBuf, msgSize);
734,186!
788
    goto _exit;
734,186✔
789
  }
790

791
  STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[STREAM_STATUS_INIT], statusSize);
852,668!
792
  snprintf(tmpBuf, sizeof(tmpBuf), "Current deploy times: %" PRId64, pStatus->deployTimes);
852,668✔
793
  STR_WITH_MAXSIZE_TO_VARSTR(msg, tmpBuf, msgSize);
852,668!
794
  goto _exit;
852,668✔
795

796
_exit:
1,917,028✔
797
  
798
  taosRUnLockLatch(&mStreamMgmt.runtimeLock);
1,917,028✔
799

800
  return TSDB_CODE_SUCCESS;
1,917,028✔
801
}
802

803
int32_t mstSetStreamAttrResBlock(SMnode *pMnode, SStreamObj* pStream, SSDataBlock* pBlock, int32_t numOfRows) {
1,946,552✔
804
  int32_t code = 0;
1,946,552✔
805
  int32_t cols = 0;
1,946,552✔
806
  int32_t lino = 0;
1,946,552✔
807

808
  char streamName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
809
  STR_WITH_MAXSIZE_TO_VARSTR(streamName, mndGetStableStr(pStream->name), sizeof(streamName));
1,946,552!
810
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
811
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
812

813
  code = colDataSetVal(pColInfo, numOfRows, (const char*)streamName, false);
1,946,552✔
814
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
815

816
  // db_name
817
  char streamDB[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
818
  STR_WITH_MAXSIZE_TO_VARSTR(streamDB, mndGetDbStr(pStream->pCreate->streamDB), sizeof(streamDB));
1,946,552!
819
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
820
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
821
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&streamDB, false);
1,946,552✔
822
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
823

824
  // create time
825
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
826
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
827
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pStream->createTime, false);
1,946,552✔
828
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
829

830
  // stream id
831
  char streamId2[19] = {0};
1,946,552✔
832
  char streamId[19 + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
833
  snprintf(streamId2, sizeof(streamId2), "%" PRIx64, pStream->pCreate->streamId);
1,946,552✔
834
  STR_WITH_MAXSIZE_TO_VARSTR(streamId, streamId2, sizeof(streamId));
1,946,552!
835
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
836
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
837
  code = colDataSetVal(pColInfo, numOfRows, (const char*)streamId, false);
1,946,552✔
838
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
839

840
  // sql
841
  char sql[TSDB_SHOW_SQL_LEN + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
842
  STR_WITH_MAXSIZE_TO_VARSTR(sql, pStream->pCreate->sql, sizeof(sql));
1,946,552!
843
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
844
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
845
  code = colDataSetVal(pColInfo, numOfRows, (const char*)sql, false);
1,946,552✔
846
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
847

848
  // status
849
  char   status[20 + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
850
  char msg[TSDB_RESERVE_VALUE_LEN + VARSTR_HEADER_SIZE] = {0};
1,946,552✔
851
  code = mstGetStreamStatusStr(pStream, status, sizeof(status), msg, sizeof(msg));
1,946,552✔
852
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
853

854
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
855
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
856
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&status, false);
1,946,552✔
857
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
858

859
  // snodeLeader
860
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
861
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
862
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pStream->mainSnodeId, false);
1,946,552✔
863
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
864

865
  // snodeReplica
866
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
867
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
868
  SSnodeObj* pSnode = mndAcquireSnode(pMnode, pStream->mainSnodeId);
1,946,552✔
869
  int32_t replicaSnodeId = pSnode ? pSnode->replicaId : -1;
1,946,552!
870
  mndReleaseSnode(pMnode, pSnode);
1,946,552✔
871
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&replicaSnodeId, false);
1,946,552✔
872
  TSDB_CHECK_CODE(code, lino, _end);
1,946,552!
873

874
  // msg
875
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1,946,552✔
876
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
1,946,552!
877
  code = colDataSetVal(pColInfo, numOfRows, (const char*)msg, false);
1,946,552✔
878

879
_end:
1,946,552✔
880
  if (code) {
1,946,552!
881
    mError("error happens when build stream attr result block, lino:%d, code:%s", lino, tstrerror(code));
×
882
  }
883
  return code;
1,946,552✔
884
}
885

886

887
int32_t mstGetTaskStatusStr(SStmTaskStatus* pTask, char* status, int32_t statusSize, char* msg, int32_t msgSize) {
26,779,728✔
888
  char tmpBuf[256];
26,685,048✔
889
  
890
  STR_WITH_MAXSIZE_TO_VARSTR(status, gStreamStatusStr[pTask->status], statusSize);
26,779,728!
891
  if (STREAM_STATUS_FAILED == pTask->status && pTask->errCode) {
26,779,728!
892
    snprintf(tmpBuf, sizeof(tmpBuf), "Last error: %s", tstrerror(pTask->errCode));
×
893
    STR_WITH_MAXSIZE_TO_VARSTR(msg, tmpBuf, msgSize);
×
894
    return TSDB_CODE_SUCCESS;
×
895
  }
896

897
  if (STREAM_TRIGGER_TASK == pTask->type && mstWaitLock(&pTask->detailStatusLock, true)) {
26,779,728!
898
    if (pTask->detailStatus) {
4,348,807✔
899
      SSTriggerRuntimeStatus* pTrigger = (SSTriggerRuntimeStatus*)pTask->detailStatus;
3,141,798✔
900
      snprintf(tmpBuf, sizeof(tmpBuf), "Current RT/HI/RE session num: %d/%d/%d, histroy progress:%d%%, total AUTO/USER recalc num: %d/%d", 
3,141,798!
901
          pTrigger->realtimeSessionNum, pTrigger->historySessionNum, pTrigger->recalcSessionNum, pTrigger->histroyProgress,
902
          pTrigger->autoRecalcNum, (int32_t)taosArrayGetSize(pTrigger->userRecalcs));
3,141,798✔
903
      taosRUnLockLatch(&pTask->detailStatusLock);
3,141,798✔
904
      return TSDB_CODE_SUCCESS;
3,141,798✔
905
    }
906

907
    taosRUnLockLatch(&pTask->detailStatusLock);    
1,207,009✔
908
  }
909
  
910
  STR_WITH_MAXSIZE_TO_VARSTR(msg, "", msgSize);
23,637,930!
911
  
912
  return TSDB_CODE_SUCCESS;
23,637,930✔
913
}
914

915
int32_t mstGetTaskExtraStr(SStmTaskStatus* pTask, char* extraStr, int32_t extraSize) {
26,779,728✔
916
  switch (pTask->type) {
26,779,728✔
917
    case STREAM_READER_TASK:
8,603,852✔
918
      if (STREAM_IS_TRIGGER_READER(pTask->flags)) {
8,603,852✔
919
        STR_WITH_MAXSIZE_TO_VARSTR(extraStr, "trigReader", extraSize);
5,050,788!
920
      } else {
921
        STR_WITH_MAXSIZE_TO_VARSTR(extraStr, "calcReader", extraSize);
3,553,064!
922
      }
923
      return TSDB_CODE_SUCCESS;
8,603,852✔
924
    case STREAM_RUNNER_TASK:
13,827,069✔
925
      if (STREAM_IS_TOP_RUNNER(pTask->flags)) {
13,827,069✔
926
        STR_WITH_MAXSIZE_TO_VARSTR(extraStr, "topRunner", extraSize);
13,027,269!
927
        return TSDB_CODE_SUCCESS;
13,027,269✔
928
      }
929
      break;
799,800✔
930
    default:
4,348,807✔
931
      break;
4,348,807✔
932
  }
933

934
  STR_WITH_MAXSIZE_TO_VARSTR(extraStr, "", extraSize);
5,148,607!
935
  return TSDB_CODE_SUCCESS;
5,148,607✔
936
}
937

938

939
int32_t mstSetStreamTaskResBlock(SStreamObj* pStream, SStmTaskStatus* pTask, SSDataBlock* pBlock, int32_t numOfRows) {
26,779,728✔
940
  int32_t code = 0;
26,779,728✔
941
  int32_t cols = 0;
26,779,728✔
942
  int32_t lino = 0;
26,779,728✔
943

944
  // stream_name
945
  char streamName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
946
  STR_WITH_MAXSIZE_TO_VARSTR(streamName, mndGetStableStr(pStream->name), sizeof(streamName));
26,779,728!
947
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
948
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
949

950
  code = colDataSetVal(pColInfo, numOfRows, (const char*)streamName, false);
26,779,728✔
951
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
952

953
  // stream id
954
  char idstr[19 + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
955
  snprintf(&idstr[VARSTR_HEADER_SIZE], sizeof(idstr) - VARSTR_HEADER_SIZE, "%" PRIx64, pStream->pCreate->streamId);
26,779,728✔
956
  varDataSetLen(idstr, strlen(&idstr[VARSTR_HEADER_SIZE])); 
26,779,728✔
957
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
958
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
959
  code = colDataSetVal(pColInfo, numOfRows, (const char*)idstr, false);
26,779,728✔
960
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
961

962
  // task id
963
  snprintf(&idstr[VARSTR_HEADER_SIZE], sizeof(idstr) - VARSTR_HEADER_SIZE, "%" PRIx64, pTask->id.taskId);
26,779,728✔
964
  varDataSetLen(idstr, strlen(&idstr[VARSTR_HEADER_SIZE]));
26,779,728✔
965
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
966
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
967
  code = colDataSetVal(pColInfo, numOfRows, (const char*)idstr, false);
26,779,728✔
968
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
969

970
  // type
971
  char type[20 + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
972
  STR_WITH_MAXSIZE_TO_VARSTR(type, (STREAM_READER_TASK == pTask->type) ? "Reader" : ((STREAM_TRIGGER_TASK == pTask->type) ? "Trigger" : "Runner"), sizeof(type));
26,779,728!
973
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
974
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
975
  code = colDataSetVal(pColInfo, numOfRows, (const char*)type, false);
26,779,728✔
976
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
977

978
  // serious id
979
  snprintf(&idstr[VARSTR_HEADER_SIZE], sizeof(idstr) - VARSTR_HEADER_SIZE, "%" PRIx64, pTask->id.seriousId);
26,779,728✔
980
  varDataSetLen(idstr, strlen(&idstr[VARSTR_HEADER_SIZE]));
26,779,728✔
981
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
982
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
983
  code = colDataSetVal(pColInfo, numOfRows, (const char*)idstr, false);
26,779,728✔
984
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
985

986
  // deploy id
987
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
988
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
989
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->id.deployId, false);
26,779,728✔
990
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
991

992
  // node_type
993
  char nodeType[10 + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
994
  STR_WITH_MAXSIZE_TO_VARSTR(nodeType, (STREAM_READER_TASK == pTask->type) ? "vnode" : "snode", sizeof(nodeType));
26,779,728!
995
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
996
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
997
  code = colDataSetVal(pColInfo, numOfRows, (const char*)nodeType, false);
26,779,728✔
998
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
999

1000
  // node id
1001
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1002
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1003
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->id.nodeId, false);
26,779,728✔
1004
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1005

1006
  // task idx
1007
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1008
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1009
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->id.taskIdx, false);
26,779,728✔
1010
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1011

1012
  // status
1013
  char   status[20 + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
1014
  char msg[TSDB_RESERVE_VALUE_LEN + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
1015
  code = mstGetTaskStatusStr(pTask, status, sizeof(status), msg, sizeof(msg));
26,779,728✔
1016
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1017

1018
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1019
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1020
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&status, false);
26,779,728✔
1021
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1022

1023
  // start time
1024
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1025
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1026
  if (pTask->runningStartTs) {
26,779,728✔
1027
    code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->runningStartTs, false);
4,216,121✔
1028
  } else {
1029
    code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->runningStartTs, true);
22,563,607✔
1030
  }
1031
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1032

1033
  // last update
1034
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1035
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1036
  if (pTask->lastUpTs) {
26,779,728!
1037
    code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->lastUpTs, false);
26,779,728✔
1038
  } else {
1039
    code = colDataSetVal(pColInfo, numOfRows, (const char*)&pTask->lastUpTs, true);
×
1040
  }
1041
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1042

1043
  // extra info
1044
  char extra[64 + VARSTR_HEADER_SIZE] = {0};
26,779,728✔
1045
  code = mstGetTaskExtraStr(pTask, extra, sizeof(extra));
26,779,728✔
1046
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1047
  
1048
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1049
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1050
  code = colDataSetVal(pColInfo, numOfRows, (const char*)extra, false);
26,779,728✔
1051
  TSDB_CHECK_CODE(code, lino, _end);
26,779,728!
1052

1053
  // msg
1054
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
26,779,728✔
1055
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
26,779,728!
1056
  code = colDataSetVal(pColInfo, numOfRows, (const char*)msg, false);
26,779,728✔
1057

1058
_end:
26,779,728✔
1059
  if (code) {
26,779,728!
1060
    mError("error happens when build stream attr result block, lino:%d, code:%s", lino, tstrerror(code));
×
1061
  }
1062
  return code;
26,779,728✔
1063
}
1064

1065
int32_t mstGetNumOfStreamTasks(SStmStatus* pStatus) {
4,359,978✔
1066
  int32_t num = taosArrayGetSize(pStatus->trigReaders) + taosArrayGetSize(pStatus->trigOReaders) + MST_LIST_SIZE(pStatus->calcReaders) + (pStatus->triggerTask ? 1 : 0);
4,359,978!
1067
  for (int32_t i = 0; i < MND_STREAM_RUNNER_DEPLOY_NUM; ++i) {
17,439,912✔
1068
    num += taosArrayGetSize(pStatus->runners[i]);
13,079,934✔
1069
  }
1070

1071
  return num;
4,359,978✔
1072
}
1073

1074
int32_t mstSetStreamTasksResBlock(SStreamObj* pStream, SSDataBlock* pBlock, int32_t* numOfRows, int32_t rowsCapacity) {
4,658,733✔
1075
  int32_t code = 0;
4,658,733✔
1076
  int32_t lino = 0;
4,658,733✔
1077
  int64_t streamId = pStream->pCreate->streamId;
4,658,733✔
1078
  bool    statusLocked = false;
4,658,733✔
1079

1080
  (void)mstWaitLock(&mStreamMgmt.runtimeLock, true);
4,658,733✔
1081

1082
  SStmStatus* pStatus = (SStmStatus*)taosHashGet(mStreamMgmt.streamMap, &streamId, sizeof(streamId));
4,658,733✔
1083
  if (NULL == pStatus) {
4,658,733✔
1084
    mstsDebug("stream not in streamMap, ignore it, dropped:%d, stopped:%d", atomic_load_8(&pStream->userDropped), atomic_load_8(&pStream->userStopped));
298,755✔
1085
    goto _exit;
298,755✔
1086
  }
1087

1088
  int8_t stopped = atomic_load_8(&pStatus->stopped);
4,359,978✔
1089
  if (stopped) {
4,359,978!
1090
    mstsDebug("stream stopped %d, ignore it", stopped);
×
1091
    goto _exit;
×
1092
  }
1093

1094
  (void)mstWaitLock(&pStatus->resetLock, true);
4,359,978✔
1095
  statusLocked = true;
4,359,978✔
1096
  
1097
  int32_t count = mstGetNumOfStreamTasks(pStatus);
4,359,978✔
1098

1099
  if (*numOfRows + count > rowsCapacity) {
4,359,978✔
1100
    code = blockDataEnsureCapacity(pBlock, *numOfRows + count);
84,305✔
1101
    if (code) {
84,305!
1102
      mstError("failed to prepare the result block buffer, rows:%d", *numOfRows + count);
×
1103
      TAOS_CHECK_EXIT(code);
×
1104
    }
1105
  }
1106

1107
  SStmTaskStatus* pTask = NULL;
4,359,978✔
1108
  int32_t trigReaderNum = taosArrayGetSize(pStatus->trigReaders);
4,359,978✔
1109
  for (int32_t i = 0; i < trigReaderNum; ++i) {
9,172,517✔
1110
    pTask = taosArrayGet(pStatus->trigReaders, i);
4,812,539✔
1111
  
1112
    code = mstSetStreamTaskResBlock(pStream, pTask, pBlock, *numOfRows);
4,812,539✔
1113
    if (code == TSDB_CODE_SUCCESS) {
4,812,539!
1114
      (*numOfRows)++;
4,812,539✔
1115
    }
1116
  }
1117

1118
  trigReaderNum = taosArrayGetSize(pStatus->trigOReaders);
4,359,978✔
1119
  for (int32_t i = 0; i < trigReaderNum; ++i) {
4,598,227✔
1120
    pTask = taosArrayGet(pStatus->trigOReaders, i);
238,249✔
1121
  
1122
    code = mstSetStreamTaskResBlock(pStream, pTask, pBlock, *numOfRows);
238,249✔
1123
    if (code == TSDB_CODE_SUCCESS) {
238,249!
1124
      (*numOfRows)++;
238,249✔
1125
    }
1126
  }
1127

1128

1129
  int32_t calcReaderNum = MST_LIST_SIZE(pStatus->calcReaders);
4,359,978!
1130
  SListNode* pNode = listHead(pStatus->calcReaders);
4,359,978✔
1131
  for (int32_t i = 0; i < calcReaderNum; ++i) {
7,913,042✔
1132
    pTask = (SStmTaskStatus*)pNode->data;
3,553,064✔
1133
  
1134
    code = mstSetStreamTaskResBlock(pStream, pTask, pBlock, *numOfRows);
3,553,064✔
1135
    if (code == TSDB_CODE_SUCCESS) {
3,553,064!
1136
      (*numOfRows)++;
3,553,064✔
1137
    }
1138
    pNode = TD_DLIST_NODE_NEXT(pNode);
3,553,064✔
1139
  }
1140

1141
  if (pStatus->triggerTask) {
4,359,978✔
1142
    code = mstSetStreamTaskResBlock(pStream, pStatus->triggerTask, pBlock, *numOfRows);
4,348,807✔
1143
    if (code == TSDB_CODE_SUCCESS) {
4,348,807!
1144
      (*numOfRows)++;
4,348,807✔
1145
    }
1146
  }
1147

1148
  int32_t runnerNum = 0;
4,359,978✔
1149
  for (int32_t i = 0; i < MND_STREAM_RUNNER_DEPLOY_NUM; ++i) {
17,439,912✔
1150
    runnerNum = taosArrayGetSize(pStatus->runners[i]);
13,079,934✔
1151
    for (int32_t m = 0; m < runnerNum; ++m) {
26,907,003✔
1152
      pTask = taosArrayGet(pStatus->runners[i], m);
13,827,069✔
1153
    
1154
      code = mstSetStreamTaskResBlock(pStream, pTask, pBlock, *numOfRows);
13,827,069✔
1155
      if (code == TSDB_CODE_SUCCESS) {
13,827,069!
1156
        (*numOfRows)++;
13,827,069✔
1157
      }
1158
    }
1159
  }
1160
  
1161
  pBlock->info.rows = *numOfRows;
4,359,978✔
1162

1163
_exit:
4,658,733✔
1164

1165
  if (statusLocked) {
4,658,733✔
1166
    taosRUnLockLatch(&pStatus->resetLock);
4,359,978✔
1167
  }
1168
  
1169
  taosRUnLockLatch(&mStreamMgmt.runtimeLock);
4,658,733✔
1170

1171
  if (code) {
4,658,733!
1172
    mError("error happens when build stream tasks result block, lino:%d, code:%s", lino, tstrerror(code));
×
1173
  }
1174
  
1175
  return code;
4,658,733✔
1176
}
1177

1178

1179
int32_t mstAppendNewRecalcRange(int64_t streamId, SStmStatus *pStream, STimeWindow* pRange) {
17,055✔
1180
  int32_t code = 0;
17,055✔
1181
  int32_t lino = 0;
17,055✔
1182
  bool    locked = false;
17,055✔
1183
  SArray* userRecalcList = NULL;
17,055✔
1184

1185
  SStreamRecalcReq req = {.recalcId = 0, .start = pRange->skey, .end = pRange->ekey};
17,055✔
1186
  TAOS_CHECK_EXIT(taosGetSystemUUIDU64(&req.recalcId));
17,055!
1187
  
1188
  taosWLockLatch(&pStream->userRecalcLock);
17,055✔
1189
  locked = true;
17,055✔
1190
  
1191
  if (NULL == pStream->userRecalcList) {
17,055!
1192
    userRecalcList = taosArrayInit(2, sizeof(SStreamRecalcReq));
17,055✔
1193
    if (NULL == userRecalcList) {
17,055!
1194
      TAOS_CHECK_EXIT(terrno);
×
1195
    }
1196

1197
    TSDB_CHECK_NULL(taosArrayPush(userRecalcList, &req), code, lino, _exit, terrno);
17,055!
1198

1199
    atomic_store_ptr(&pStream->userRecalcList, userRecalcList);
17,055✔
1200
    userRecalcList = NULL;    
17,055✔
1201
  } else {
1202
    TSDB_CHECK_NULL(taosArrayPush(pStream->userRecalcList, &req), code, lino, _exit, terrno);
×
1203
  }
1204
  
1205
  mstsInfo("stream recalc ID:%" PRIx64 " range:%" PRId64 " - %" PRId64 " added", req.recalcId, pRange->skey, pRange->ekey);
17,055!
1206

1207
_exit:
17,055✔
1208

1209
  taosArrayDestroy(userRecalcList);
17,055✔
1210

1211
  if (locked) {
17,055!
1212
    taosWUnLockLatch(&pStream->userRecalcLock);
17,055✔
1213
  }
1214
  
1215
  if (code) {
17,055!
1216
    mstsError("%s failed at line %d, error:%s", __FUNCTION__, lino, tstrerror(code));
×
1217
  }
1218
  
1219
  return code;
17,055✔
1220
}
1221

1222

1223

1224
int32_t mstSetStreamRecalculateResBlock(SStreamObj* pStream, SSTriggerRecalcProgress* pProgress, SSDataBlock* pBlock, int32_t numOfRows) {
×
1225
  int32_t code = 0;
×
1226
  int32_t cols = 0;
×
1227
  int32_t lino = 0;
×
1228

1229
  // stream_name
1230
  char streamName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
1231
  STR_WITH_MAXSIZE_TO_VARSTR(streamName, mndGetStableStr(pStream->name), sizeof(streamName));
×
1232
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1233
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1234

1235
  code = colDataSetVal(pColInfo, numOfRows, (const char*)streamName, false);
×
1236
  TSDB_CHECK_CODE(code, lino, _end);
×
1237

1238
  // stream id
1239
  char idstr[19 + VARSTR_HEADER_SIZE] = {0};
×
1240
  snprintf(&idstr[VARSTR_HEADER_SIZE], sizeof(idstr) - VARSTR_HEADER_SIZE, "%" PRIx64, pStream->pCreate->streamId);
×
1241
  varDataSetLen(idstr, strlen(&idstr[VARSTR_HEADER_SIZE]) + VARSTR_HEADER_SIZE); 
×
1242
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1243
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1244
  code = colDataSetVal(pColInfo, numOfRows, (const char*)idstr, false);
×
1245
  TSDB_CHECK_CODE(code, lino, _end);
×
1246

1247
  // recalc id
1248
  snprintf(&idstr[VARSTR_HEADER_SIZE], sizeof(idstr) - VARSTR_HEADER_SIZE, "%" PRIx64, pProgress->recalcId);
×
1249
  varDataSetLen(idstr, strlen(&idstr[VARSTR_HEADER_SIZE]) + VARSTR_HEADER_SIZE);
×
1250
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1251
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1252
  code = colDataSetVal(pColInfo, numOfRows, (const char*)idstr, false);
×
1253
  TSDB_CHECK_CODE(code, lino, _end);
×
1254

1255
  // start
1256
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1257
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1258
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pProgress->start, false);
×
1259
  TSDB_CHECK_CODE(code, lino, _end);
×
1260

1261
  // end
1262
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1263
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1264
  code = colDataSetVal(pColInfo, numOfRows, (const char*)&pProgress->end, false);
×
1265
  TSDB_CHECK_CODE(code, lino, _end);
×
1266

1267
  // progress
1268
  char progress[20 + VARSTR_HEADER_SIZE] = {0};
×
1269
  snprintf(&progress[VARSTR_HEADER_SIZE], sizeof(progress) - VARSTR_HEADER_SIZE, "%d%%", pProgress->progress);
×
1270
  varDataSetLen(progress, strlen(&progress[VARSTR_HEADER_SIZE]) + VARSTR_HEADER_SIZE);
×
1271
  pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
×
1272
  TSDB_CHECK_NULL(pColInfo, code, lino, _end, terrno);
×
1273
  code = colDataSetVal(pColInfo, numOfRows, (const char*)progress, false);
×
1274
  TSDB_CHECK_CODE(code, lino, _end);
×
1275

1276
_end:
×
1277
  if (code) {
×
1278
    mError("error happens when build stream attr result block, lino:%d, code:%s", lino, tstrerror(code));
×
1279
  }
1280
  return code;
×
1281
}
1282

1283

1284
int32_t mstSetStreamRecalculatesResBlock(SStreamObj* pStream, SSDataBlock* pBlock, int32_t* numOfRows, int32_t rowsCapacity) {
1,370✔
1285
  int32_t code = 0;
1,370✔
1286
  int32_t lino = 0;
1,370✔
1287
  int64_t streamId = pStream->pCreate->streamId;
1,370✔
1288

1289
  (void)mstWaitLock(&mStreamMgmt.runtimeLock, true);
1,370✔
1290

1291
  SStmStatus* pStatus = (SStmStatus*)taosHashGet(mStreamMgmt.streamMap, &streamId, sizeof(streamId));
1,370✔
1292
  if (NULL == pStatus) {
1,370!
1293
    mstsDebug("stream not in streamMap, ignore it, dropped:%d, stopped:%d", atomic_load_8(&pStream->userDropped), atomic_load_8(&pStream->userStopped));
×
1294
    goto _exit;
×
1295
  }
1296

1297
  int8_t stopped = atomic_load_8(&pStatus->stopped);
1,370✔
1298
  if (stopped) {
1,370!
1299
    mstsDebug("stream stopped %d, ignore it", stopped);
×
1300
    goto _exit;
×
1301
  }
1302

1303
  if (NULL == pStatus->triggerTask) {
1,370!
1304
    mstsDebug("no trigger task now, deployTimes:%" PRId64 ", ignore it", pStatus->deployTimes);
×
1305
    goto _exit;
×
1306
  }
1307

1308
  (void)mstWaitLock(&pStatus->triggerTask->detailStatusLock, true);
1,370✔
1309
  if (NULL == pStatus->triggerTask->detailStatus) {
1,370!
1310
    mstsDebug("no trigger task now, deployTimes:%" PRId64 ", ignore it", pStatus->deployTimes);
×
1311
    taosRUnLockLatch(&pStatus->triggerTask->detailStatusLock);
×
1312
    goto _exit;
×
1313
  }
1314

1315
  SSTriggerRuntimeStatus* pTrigger = (SSTriggerRuntimeStatus*)pStatus->triggerTask->detailStatus;
1,370✔
1316
  int32_t count = taosArrayGetSize(pTrigger->userRecalcs);
1,370✔
1317

1318
  if (*numOfRows + count > rowsCapacity) {
1,370!
1319
    code = blockDataEnsureCapacity(pBlock, *numOfRows + count);
×
1320
    if (code) {
×
1321
      mstError("failed to prepare the result block buffer, rows:%d", *numOfRows + count);
×
1322
      taosRUnLockLatch(&pStatus->triggerTask->detailStatusLock);
×
1323
      TAOS_CHECK_EXIT(code);
×
1324
    }
1325
  }
1326

1327
  for (int32_t i = 0; i < count; ++i) {
1,370!
1328
    SSTriggerRecalcProgress* pProgress = taosArrayGet(pTrigger->userRecalcs, i);
×
1329
  
1330
    code = mstSetStreamRecalculateResBlock(pStream, pProgress, pBlock, *numOfRows);
×
1331
    if (code == TSDB_CODE_SUCCESS) {
×
1332
      (*numOfRows)++;
×
1333
    }
1334
  }
1335

1336
  taosRUnLockLatch(&pStatus->triggerTask->detailStatusLock);
1,370✔
1337
  
1338
  pBlock->info.rows = *numOfRows;
1,370✔
1339

1340
_exit:
1,370✔
1341
  
1342
  taosRUnLockLatch(&mStreamMgmt.runtimeLock);
1,370✔
1343

1344
  if (code) {
1,370!
1345
    mError("error happens when build stream recalculates result block, lino:%d, code:%s", lino, tstrerror(code));
×
1346
  }
1347
  
1348
  return code;
1,370✔
1349
}
1350

1351
int32_t mstGetScanUidFromPlan(int64_t streamId, void* scanPlan, int64_t* uid) {
384,431✔
1352
  SSubplan* pSubplan = NULL;
384,431✔
1353
  int32_t code = TSDB_CODE_SUCCESS, lino = 0;
384,431✔
1354
  
1355
  TAOS_CHECK_EXIT(nodesStringToNode(scanPlan, (SNode**)&pSubplan));
384,431!
1356

1357
  if (pSubplan->pNode && nodeType(pSubplan->pNode) == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
384,431!
1358
    SScanPhysiNode* pScanNode = (SScanPhysiNode*)pSubplan->pNode;
244,334✔
1359
    *uid = pScanNode->uid;
244,334✔
1360
  }
1361
  
1362
_exit:
382,853✔
1363

1364
  if (code) {
384,431!
1365
    mstsError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1366
  }
1367

1368
  nodesDestroyNode((SNode *)pSubplan);
384,431✔
1369

1370
  return code;
384,431✔
1371
}
1372

1373

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