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

taosdata / TDengine / #3546

03 Dec 2024 10:02AM UTC coverage: 60.691% (-0.1%) from 60.839%
#3546

push

travis-ci

web-flow
Merge pull request #29015 from taosdata/fix/TS-5668

[TS-5668] fix(keeper): fix endpoint value too long for column/tag and eliminate warnings

120577 of 253823 branches covered (47.5%)

Branch coverage included in aggregate %.

201666 of 277134 relevant lines covered (72.77%)

18719900.08 hits per line

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

78.11
/source/libs/stream/src/streamQueue.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 "streamInt.h"
17

18
#define MAX_STREAM_EXEC_BATCH_NUM 32
19
#define MAX_SMOOTH_BURST_RATIO    5  // 5 sec
20

21
// todo refactor:
22
// read data from input queue
23
typedef struct SQueueReader {
24
  SStreamQueue* pQueue;
25
  int32_t       taskLevel;
26
  int32_t       maxBlocks;     // maximum block in one batch
27
  int32_t       waitDuration;  // maximum wait time to format several block into a batch to process, unit: ms
28
} SQueueReader;
29

30
#define streamQueueCurItem(_q) ((_q)->qItem)
31

32
static bool streamTaskExtractAvailableToken(STokenBucket* pBucket, const char* id);
33
static void streamTaskPutbackToken(STokenBucket* pBucket);
34
static void streamTaskConsumeQuota(STokenBucket* pBucket, int32_t bytes);
35

36
static void streamQueueCleanup(SStreamQueue* pQueue) {
28,063✔
37
  SStreamQueueItem* qItem = NULL;
28,063✔
38
  while (1) {
39
    streamQueueNextItem(pQueue, &qItem);
30,164✔
40
    if (qItem == NULL) {
30,177✔
41
      break;
28,078✔
42
    }
43
    streamFreeQitem(qItem);
2,099✔
44
  }
45
  pQueue->status = STREAM_QUEUE__SUCESS;
28,078✔
46
}
28,078✔
47

48
int32_t streamQueueOpen(int64_t cap, SStreamQueue** pQ) {
28,547✔
49
  *pQ = NULL;
28,547✔
50
  int32_t code = 0;
28,547✔
51

52
  SStreamQueue* pQueue = taosMemoryCalloc(1, sizeof(SStreamQueue));
28,547✔
53
  if (pQueue == NULL) {
28,589!
54
    return terrno;
×
55
  }
56

57
  code = taosOpenQueue(&pQueue->pQueue);
28,589✔
58
  if (code) {
28,560!
59
    taosMemoryFreeClear(pQueue);
×
60
    return code;
×
61
  }
62

63
  code = taosAllocateQall(&pQueue->qall);
28,560✔
64
  if (code) {
28,596!
65
    taosCloseQueue(pQueue->pQueue);
×
66
    taosMemoryFree(pQueue);
×
67
    return code;
×
68
  }
69

70
  pQueue->status = STREAM_QUEUE__SUCESS;
28,596✔
71
  taosSetQueueCapacity(pQueue->pQueue, cap);
28,596✔
72
  taosSetQueueMemoryCapacity(pQueue->pQueue, cap * 1024);
28,578✔
73

74
  *pQ = pQueue;
28,578✔
75
  return code;
28,578✔
76
}
77

78
void streamQueueClose(SStreamQueue* pQueue, int32_t taskId) {
28,063✔
79
  stDebug("s-task:0x%x free the queue:%p, items in queue:%d", taskId, pQueue->pQueue,
28,063✔
80
          taosQueueItemSize(pQueue->pQueue));
81
  streamQueueCleanup(pQueue);
28,063✔
82

83
  taosFreeQall(pQueue->qall);
28,067✔
84
  taosCloseQueue(pQueue->pQueue);
28,087✔
85
  taosMemoryFree(pQueue);
28,079✔
86
}
28,083✔
87

88
void streamQueueNextItem(SStreamQueue* pQueue, SStreamQueueItem** pItem) {
544,465✔
89
  *pItem = NULL;
544,465✔
90
  int8_t flag = atomic_exchange_8(&pQueue->status, STREAM_QUEUE__PROCESSING);
544,465✔
91

92
  if (flag == STREAM_QUEUE__FAILED) {
544,771✔
93
    *pItem = streamQueueCurItem(pQueue);
3,434✔
94
  } else {
95
    pQueue->qItem = NULL;
541,337✔
96
    (void) taosGetQitem(pQueue->qall, &pQueue->qItem);
541,337✔
97
    if (pQueue->qItem == NULL) {
541,242✔
98
      (void) taosReadAllQitems(pQueue->pQueue, pQueue->qall);
246,084✔
99
      (void) taosGetQitem(pQueue->qall, &pQueue->qItem);
246,142✔
100
    }
101

102
    *pItem = streamQueueCurItem(pQueue);
541,255✔
103
  }
104
}
544,689✔
105

106
void streamQueueProcessSuccess(SStreamQueue* queue) {
331,812✔
107
  if (atomic_load_8(&queue->status) != STREAM_QUEUE__PROCESSING) {
331,812!
108
    stError("invalid queue status:%d, expect:%d", atomic_load_8(&queue->status), STREAM_QUEUE__PROCESSING);
×
109
    return;
×
110
  }
111

112
  queue->qItem = NULL;
331,800✔
113
  atomic_store_8(&queue->status, STREAM_QUEUE__SUCESS);
331,800✔
114
}
115

116
void streamQueueProcessFail(SStreamQueue* queue) {
3,432✔
117
  if (atomic_load_8(&queue->status) != STREAM_QUEUE__PROCESSING) {
3,432!
118
    stError("invalid queue status:%d, expect:%d", atomic_load_8(&queue->status), STREAM_QUEUE__PROCESSING);
×
119
    return;
×
120
  }
121
  atomic_store_8(&queue->status, STREAM_QUEUE__FAILED);
3,433✔
122
}
123

124
bool streamQueueIsFull(const SStreamQueue* pQueue) {
635,079✔
125
  int32_t numOfItems = streamQueueGetNumOfItems(pQueue);
635,079✔
126
  if (numOfItems >= STREAM_TASK_QUEUE_CAPACITY) {
635,315!
127
    return true;
×
128
  }
129

130
  return (SIZE_IN_MiB(taosQueueMemorySize(pQueue->pQueue)) >= STREAM_TASK_QUEUE_CAPACITY_IN_SIZE);
635,315✔
131
}
132

133
int32_t streamQueueGetNumOfItems(const SStreamQueue* pQueue) {
1,255,455✔
134
  int32_t numOfItems1 = taosQueueItemSize(pQueue->pQueue);
1,255,455✔
135
  int32_t numOfItems2 = taosQallItemSize(pQueue->qall);
1,256,287✔
136

137
  return numOfItems1 + numOfItems2;
1,256,169✔
138
}
139

140
int32_t streamQueueGetNumOfUnAccessedItems(const SStreamQueue* pQueue) {
43,460✔
141
  int32_t numOfItems1 = taosQueueItemSize(pQueue->pQueue);
43,460✔
142
  int32_t numOfItems2 = taosQallUnAccessedItemSize(pQueue->qall);
43,461✔
143

144
  return numOfItems1 + numOfItems2;
43,463✔
145
}
146

147
int32_t streamQueueGetItemSize(const SStreamQueue* pQueue) {
50,331✔
148
  return taosQueueMemorySize(pQueue->pQueue) + taosQallUnAccessedMemSize(pQueue->qall);
50,331✔
149
}
150

151
int32_t streamQueueItemGetSize(const SStreamQueueItem* pItem) {
331,791✔
152
  STaosQnode* p = (STaosQnode*)((char*)pItem - sizeof(STaosQnode));
331,791✔
153
  return p->dataSize;
331,791✔
154
}
155

156
void streamQueueItemIncSize(const SStreamQueueItem* pItem, int32_t size) {
275,126✔
157
  STaosQnode* p = (STaosQnode*)((char*)pItem - sizeof(STaosQnode));
275,126✔
158
  p->dataSize += size;
275,126✔
159
}
275,126✔
160

161
const char* streamQueueItemGetTypeStr(int32_t type) {
71,332✔
162
  switch (type) {
71,332✔
163
    case STREAM_INPUT__CHECKPOINT:
4,753✔
164
      return "checkpoint";
4,753✔
165
    case STREAM_INPUT__CHECKPOINT_TRIGGER:
19,083✔
166
      return "checkpoint-trigger";
19,083✔
167
    case STREAM_INPUT__TRANS_STATE:
17,192✔
168
      return "trans-state";
17,192✔
169
    case STREAM_INPUT__REF_DATA_BLOCK:
5,723✔
170
      return "ref-block";
5,723✔
171
    default:
24,581✔
172
      return "datablock";
24,581✔
173
  }
174
}
175

176
EExtractDataCode streamTaskGetDataFromInputQ(SStreamTask* pTask, SStreamQueueItem** pInput, int32_t* numOfBlocks,
154,298✔
177
                                             int32_t* blockSize) {
178
  const char* id = pTask->id.idStr;
154,298✔
179
  int32_t     taskLevel = pTask->info.taskLevel;
154,298✔
180

181
  *pInput = NULL;
154,298✔
182
  *numOfBlocks = 0;
154,298✔
183
  *blockSize = 0;
154,298✔
184

185
  // no available token in bucket for sink task, let's wait for a little bit
186
  if (taskLevel == TASK_LEVEL__SINK && (!streamTaskExtractAvailableToken(pTask->outputInfo.pTokenBucket, id))) {
154,298!
187
    stDebug("s-task:%s no available token in bucket for sink data, wait for 10ms", id);
×
188
    return EXEC_AFTER_IDLE;
×
189
  }
190

191
  while (1) {
325,213✔
192
    if (streamTaskShouldPause(pTask) || streamTaskShouldStop(pTask)) {
479,506!
193
      stDebug("s-task:%s task should pause, extract input blocks:%d", id, *numOfBlocks);
×
194
      return EXEC_CONTINUE;
154,344✔
195
    }
196

197
    SStreamQueueItem* qItem = NULL;
479,386✔
198
    streamQueueNextItem(pTask->inputq.queue, (SStreamQueueItem**)&qItem);
479,386✔
199
    if (qItem == NULL) {
479,491✔
200
      // restore the token to bucket
201
      if (*numOfBlocks > 0) {
117,212✔
202
        *blockSize = streamQueueItemGetSize(*pInput);
46,562✔
203
        if (taskLevel == TASK_LEVEL__SINK) {
46,553✔
204
          streamTaskConsumeQuota(pTask->outputInfo.pTokenBucket, *blockSize);
12,508✔
205
        }
206
      } else {
207
        streamTaskPutbackToken(pTask->outputInfo.pTokenBucket);
70,650✔
208
      }
209

210
      return EXEC_CONTINUE;
117,180✔
211
    }
212

213
    // do not merge blocks for sink node and check point data block
214
    int8_t type = qItem->type;
362,279✔
215
    if (type == STREAM_INPUT__CHECKPOINT || type == STREAM_INPUT__CHECKPOINT_TRIGGER ||
362,279✔
216
        type == STREAM_INPUT__TRANS_STATE || type == STREAM_INPUT__REF_DATA_BLOCK) {
336,291✔
217
      const char* p = streamQueueItemGetTypeStr(type);
30,333✔
218

219
      if (*pInput == NULL) {
30,353✔
220
        stDebug("s-task:%s %s msg extracted, start to process immediately", id, p);
27,017✔
221

222
        // restore the token to bucket in case of checkpoint/trans-state msg
223
        streamTaskPutbackToken(pTask->outputInfo.pTokenBucket);
27,017✔
224
        *blockSize = 0;
27,017✔
225
        *numOfBlocks = 1;
27,017✔
226
        *pInput = qItem;
27,017✔
227
        return EXEC_CONTINUE;
27,017✔
228
      } else {  // previous existed blocks needs to be handle, before handle the checkpoint msg block
229
        stDebug("s-task:%s %s msg extracted, handle previous blocks, numOfBlocks:%d", id, p, *numOfBlocks);
3,336✔
230
        *blockSize = streamQueueItemGetSize(*pInput);
3,336✔
231
        if (taskLevel == TASK_LEVEL__SINK) {
3,330✔
232
          streamTaskConsumeQuota(pTask->outputInfo.pTokenBucket, *blockSize);
866✔
233
        }
234

235
        streamQueueProcessFail(pTask->inputq.queue);
3,330✔
236
        return EXEC_CONTINUE;
3,332✔
237
      }
238
    } else {
239
      if (*pInput == NULL) {
331,946✔
240
        *pInput = qItem;
56,658✔
241
      } else { // merge current block failed, let's handle the already merged blocks.
242
        void*   newRet = NULL;
275,288✔
243
        int32_t code = streamQueueMergeQueueItem(*pInput, qItem, (SStreamQueueItem**)&newRet);
275,288✔
244
        if (newRet == NULL) {
275,301✔
245
          if (code != -1) {
105!
246
            stError("s-task:%s failed to merge blocks from inputQ, numOfBlocks:%d, code:%s", id, *numOfBlocks,
×
247
                    tstrerror(code));
248
          }
249

250
          *blockSize = streamQueueItemGetSize(*pInput);
105✔
251
          if (taskLevel == TASK_LEVEL__SINK) {
102!
252
            streamTaskConsumeQuota(pTask->outputInfo.pTokenBucket, *blockSize);
×
253
          }
254

255
          streamQueueProcessFail(pTask->inputq.queue);
102✔
256
          return EXEC_CONTINUE;
102✔
257
        }
258

259
        *pInput = newRet;
275,196✔
260
      }
261

262
      *numOfBlocks += 1;
331,854✔
263
      streamQueueProcessSuccess(pTask->inputq.queue);
331,854✔
264

265
      if (*numOfBlocks >= MAX_STREAM_EXEC_BATCH_NUM) {
331,915✔
266
        stDebug("s-task:%s batch size limit:%d reached, start to process blocks", id, MAX_STREAM_EXEC_BATCH_NUM);
6,702✔
267

268
        *blockSize = streamQueueItemGetSize(*pInput);
6,702✔
269
        if (taskLevel == TASK_LEVEL__SINK) {
6,700✔
270
          streamTaskConsumeQuota(pTask->outputInfo.pTokenBucket, *blockSize);
3✔
271
        }
272

273
        return EXEC_CONTINUE;
6,700✔
274
      }
275
    }
276
  }
277
}
278

279
int32_t streamTaskPutDataIntoInputQ(SStreamTask* pTask, SStreamQueueItem* pItem) {
358,985✔
280
  int8_t      type = pItem->type;
358,985✔
281
  STaosQueue* pQueue = pTask->inputq.queue->pQueue;
358,985✔
282
  int32_t     total = streamQueueGetNumOfItems(pTask->inputq.queue) + 1;
358,985✔
283

284
  if (type == STREAM_INPUT__DATA_SUBMIT) {
359,045✔
285
    SStreamDataSubmit* px = (SStreamDataSubmit*)pItem;
308,194✔
286
    if ((pTask->info.taskLevel == TASK_LEVEL__SOURCE) && streamQueueIsFull(pTask->inputq.queue)) {
308,194!
287
      double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
13✔
288
      stTrace(
13!
289
          "s-task:%s inputQ is full, capacity(size:%d num:%dMiB), current(blocks:%d, size:%.2fMiB) stop to push data",
290
          pTask->id.idStr, STREAM_TASK_QUEUE_CAPACITY, STREAM_TASK_QUEUE_CAPACITY_IN_SIZE, total, size);
291
      streamDataSubmitDestroy(px);
13✔
292
      return TSDB_CODE_STREAM_INPUTQ_FULL;
13✔
293
    }
294

295
    int32_t msgLen = px->submit.msgLen;
308,200✔
296
    int64_t ver = px->submit.ver;
308,200✔
297

298
    int32_t code = taosWriteQitem(pQueue, pItem);
308,200✔
299
    if (code != TSDB_CODE_SUCCESS) {
308,184!
300
      streamDataSubmitDestroy(px);
×
301
      return code;
×
302
    }
303

304
    double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
308,184✔
305

306
    // use the local variable to avoid the pItem be freed by other threads, since it has been put into queue already.
307
    stDebug("s-task:%s submit enqueue msgLen:%d ver:%" PRId64 ", total in queue:%d, size:%.2fMiB", pTask->id.idStr,
308,188✔
308
            msgLen, ver, total, size + SIZE_IN_MiB(msgLen));
309
  } else if (type == STREAM_INPUT__DATA_BLOCK || type == STREAM_INPUT__REF_DATA_BLOCK) {
73,652✔
310
    if (streamQueueIsFull(pTask->inputq.queue)) {
22,827!
311
      double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
×
312

313
      stTrace("s-task:%s input queue is full, capacity:%d size:%d MiB, current(blocks:%d, size:%.2fMiB) abort",
×
314
              pTask->id.idStr, STREAM_TASK_QUEUE_CAPACITY, STREAM_TASK_QUEUE_CAPACITY_IN_SIZE, total, size);
315
      streamFreeQitem(pItem);
×
316
      return TSDB_CODE_STREAM_INPUTQ_FULL;
×
317
    }
318

319
    int32_t code = taosWriteQitem(pQueue, pItem);
22,802✔
320
    if (code != TSDB_CODE_SUCCESS) {
22,795!
321
      streamFreeQitem(pItem);
×
322
      return code;
×
323
    }
324

325
    double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
22,795✔
326
    stDebug("s-task:%s blockdata enqueue, total in queue:%d, size:%.2fMiB", pTask->id.idStr, total, size);
22,802✔
327
  } else if (type == STREAM_INPUT__CHECKPOINT || type == STREAM_INPUT__CHECKPOINT_TRIGGER ||
28,024✔
328
             type == STREAM_INPUT__TRANS_STATE || type == STREAM_INPUT__DATA_RETRIEVE) {
28,593✔
329
    int32_t code = taosWriteQitem(pQueue, pItem);
24,988✔
330
    if (code != TSDB_CODE_SUCCESS) {
25,017!
331
      streamFreeQitem(pItem);
×
332
      return code;
×
333
    }
334

335
    double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
25,017✔
336
    stDebug("s-task:%s level:%d %s blockdata enqueue, total in queue:%d, size:%.2fMiB", pTask->id.idStr,
24,995✔
337
            pTask->info.taskLevel, streamQueueItemGetTypeStr(type), total, size);
338
  } else if (type == STREAM_INPUT__GET_RES) {
3,036!
339
    // use the default memory limit, refactor later.
340
    int32_t code = taosWriteQitem(pQueue, pItem);
3,036✔
341
    if (code != TSDB_CODE_SUCCESS) {
3,036!
342
      streamFreeQitem(pItem);
×
343
      return code;
×
344
    }
345

346
    double size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
3,036✔
347
    stDebug("s-task:%s data res enqueue, current(blocks:%d, size:%.2fMiB)", pTask->id.idStr, total, size);
3,036✔
348
  } else {
349
    stError("s-task:%s invalid type:%d to put in inputQ", pTask->id.idStr, type);
×
350
    return TSDB_CODE_INVALID_PARA;
×
351
  }
352

353
  if (type != STREAM_INPUT__GET_RES && type != STREAM_INPUT__CHECKPOINT && type != STREAM_INPUT__CHECKPOINT_TRIGGER &&
359,036✔
354
      (pTask->info.delaySchedParam != 0)) {
341,079✔
355
    (void)atomic_val_compare_exchange_8(&pTask->schedInfo.status, TASK_TRIGGER_STATUS__INACTIVE,
11,885✔
356
                                        TASK_TRIGGER_STATUS__MAY_ACTIVE);
357
    stDebug("s-task:%s new data arrived, active the sched-trigger, triggerStatus:%d", pTask->id.idStr,
11,885✔
358
            pTask->schedInfo.status);
359
  }
360

361
  return 0;
359,017✔
362
}
363

364
int32_t streamTaskPutTranstateIntoInputQ(SStreamTask* pTask) {
2,356✔
365
  int32_t           code = 0;
2,356✔
366
  SStreamDataBlock* pTranstate = NULL;
2,356✔
367
  SSDataBlock*      pBlock = NULL;
2,356✔
368

369
  code = taosAllocateQitem(sizeof(SStreamDataBlock), DEF_QITEM, sizeof(SSDataBlock), (void**)&pTranstate);
2,356✔
370
  if (code) {
2,356!
371
    return code;
×
372
  }
373

374
  pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
2,356✔
375
  if (pBlock == NULL) {
2,356!
376
    code = terrno;
×
377
    goto _err;
×
378
  }
379

380
  pTranstate->type = STREAM_INPUT__TRANS_STATE;
2,356✔
381

382
  pBlock->info.type = STREAM_TRANS_STATE;
2,356✔
383
  pBlock->info.rows = 1;
2,356✔
384
  pBlock->info.childId = pTask->info.selfChildId;
2,356✔
385

386
  pTranstate->blocks = taosArrayInit(4, sizeof(SSDataBlock));  // pBlock;
2,356✔
387
  if (pTranstate->blocks == NULL) {
2,356!
388
    code = terrno;
×
389
    goto _err;
×
390
  }
391

392
  void* p = taosArrayPush(pTranstate->blocks, pBlock);
2,356✔
393
  if (p == NULL) {
2,356!
394
    code = terrno;
×
395
    goto _err;
×
396
  }
397

398
  taosMemoryFree(pBlock);
2,356✔
399
  if (streamTaskPutDataIntoInputQ(pTask, (SStreamQueueItem*)pTranstate) < 0) {
2,356!
400
    code = TSDB_CODE_OUT_OF_MEMORY;
×
401
    goto _err;
×
402
  }
403

404
  pTask->status.appendTranstateBlock = true;
2,356✔
405
  return TSDB_CODE_SUCCESS;
2,356✔
406

407
_err:
×
408
  taosMemoryFree(pBlock);
×
409
  taosFreeQitem(pTranstate);
×
410
  return code;
×
411
}
412

413
// the result should be put into the outputQ in any cases, the result may be lost otherwise.
414
int32_t streamTaskPutDataIntoOutputQ(SStreamTask* pTask, SStreamDataBlock* pBlock) {
18,345✔
415
  STaosQueue* pQueue = pTask->outputq.queue->pQueue;
18,345✔
416
  int32_t     code = taosWriteQitem(pQueue, pBlock);
18,345✔
417

418
  int32_t total = streamQueueGetNumOfItems(pTask->outputq.queue);
18,345✔
419
  double  size = SIZE_IN_MiB(taosQueueMemorySize(pQueue));
18,345✔
420
  if (code != 0) {
18,346!
421
    stError("s-task:%s failed to put res into outputQ, outputQ items:%d, size:%.2fMiB code:%s, result lost",
×
422
            pTask->id.idStr, total + 1, size, tstrerror(code));
423
  } else {
424
    if (streamQueueIsFull(pTask->outputq.queue)) {
18,346!
425
      stWarn(
×
426
          "s-task:%s outputQ is full(outputQ items:%d, size:%.2fMiB), set the output status BLOCKING, wait for 500ms "
427
          "after handle this batch of blocks",
428
          pTask->id.idStr, total, size);
429
    } else {
430
      stDebug("s-task:%s data put into outputQ, outputQ items:%d, size:%.2fMiB", pTask->id.idStr, total, size);
18,345✔
431
    }
432
  }
433

434
  return code;
18,345✔
435
}
436

437
int32_t streamTaskInitTokenBucket(STokenBucket* pBucket, int32_t numCap, int32_t numRate, float quotaRate,
14,279✔
438
                                  const char* id) {
439
  if (numCap < 10 || numRate < 10 || pBucket == NULL) {
14,279!
440
    stError("failed to init sink task bucket, cap:%d, rate:%d", numCap, numRate);
×
441
    return TSDB_CODE_INVALID_PARA;
×
442
  }
443

444
  pBucket->numCapacity = numCap;
14,291✔
445
  pBucket->numOfToken = numCap;
14,291✔
446
  pBucket->numRate = numRate;
14,291✔
447

448
  pBucket->quotaRate = quotaRate;
14,291✔
449
  pBucket->quotaCapacity = quotaRate * MAX_SMOOTH_BURST_RATIO;
14,291✔
450
  pBucket->quotaRemain = pBucket->quotaCapacity;
14,291✔
451

452
  pBucket->tokenFillTimestamp = taosGetTimestampMs();
14,299✔
453
  pBucket->quotaFillTimestamp = taosGetTimestampMs();
14,298✔
454
  stDebug("s-task:%s sink quotaRate:%.2fMiB, numRate:%d", id, quotaRate, numRate);
14,298✔
455
  return TSDB_CODE_SUCCESS;
14,299✔
456
}
457

458
static void fillTokenBucket(STokenBucket* pBucket, const char* id) {
50,398✔
459
  int64_t now = taosGetTimestampMs();
50,413✔
460

461
  int64_t deltaToken = now - pBucket->tokenFillTimestamp;
50,413✔
462
  if (pBucket->numOfToken < 0) {
50,413!
463
    return;
×
464
  }
465

466
  int32_t incNum = (deltaToken / 1000.0) * pBucket->numRate;
50,413✔
467
  if (incNum > 0) {
50,413✔
468
    pBucket->numOfToken = TMIN(pBucket->numOfToken + incNum, pBucket->numCapacity);
26,009✔
469
    pBucket->tokenFillTimestamp = now;
26,009✔
470
  }
471

472
  // increase the new available quota as time goes on
473
  int64_t deltaQuota = now - pBucket->quotaFillTimestamp;
50,413✔
474
  double  incSize = (deltaQuota / 1000.0) * pBucket->quotaRate;
50,413✔
475
  if (incSize > 0) {
50,413✔
476
    pBucket->quotaRemain = TMIN(pBucket->quotaRemain + incSize, pBucket->quotaCapacity);
31,357✔
477
    pBucket->quotaFillTimestamp = now;
31,357✔
478
  }
479

480
  if (incNum > 0 || incSize > 0) {
50,413✔
481
    stTrace("token/quota available, token:%d inc:%d, token_TsDelta:%" PRId64
31,327✔
482
            ", quota:%.2fMiB inc:%.3fMiB quotaTs:%" PRId64 " now:%" PRId64 "ms, %s",
483
            pBucket->numOfToken, incNum, deltaToken, pBucket->quotaRemain, incSize, deltaQuota, now, id);
484
  }
485
}
486

487
bool streamTaskExtractAvailableToken(STokenBucket* pBucket, const char* id) {
50,399✔
488
  fillTokenBucket(pBucket, id);
50,399✔
489

490
  if (pBucket->numOfToken > 0) {
50,417!
491
    if (pBucket->quotaRemain > 0) {
50,427!
492
      pBucket->numOfToken -= 1;
50,427✔
493
      return true;
50,427✔
494
    } else {  // no available size quota now
495
      return false;
×
496
    }
497
  } else {
498
    return false;
×
499
  }
500
}
501

502
void streamTaskPutbackToken(STokenBucket* pBucket) {
97,630✔
503
  pBucket->numOfToken = TMIN(pBucket->numOfToken + 1, pBucket->numCapacity);
97,630✔
504
}
97,630✔
505

506
// size in KB
507
void streamTaskConsumeQuota(STokenBucket* pBucket, int32_t bytes) { pBucket->quotaRemain -= SIZE_IN_MiB(bytes); }
13,378✔
508

509
void streamTaskInputFail(SStreamTask* pTask) { atomic_store_8(&pTask->inputq.status, TASK_INPUT_STATUS__FAILED); }
×
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