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

taosdata / TDengine / #5057

17 May 2026 01:15AM UTC coverage: 73.406% (+0.02%) from 73.384%
#5057

push

travis-ci

web-flow
feat (TDgpt): Dynamic Model Synchronization Enhancements (#35344)

* refactor: do some internal refactor.

* fix: fix multiprocess sync issue.

* feat: add dynamic anomaly detection and forecasting services

* fix: log error message for undeploying model in exception handling

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* fix: handle undeploy when model exists only on disk

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/286aafa0-c3ce-4c27-b803-2707571e9dc1

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: guard dynamic registry concurrent access

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/5e4db858-6458-40f4-ac28-d1b1b7f97c18

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: tighten service list locking scope

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/5e4db858-6458-40f4-ac28-d1b1b7f97c18

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: restore prophet support and update tests per review feedback

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/92298ae1-7da6-4d07-b20e-101c7cd0b26b

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: improve test name and move copy inside lock scope

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/92298ae1-7da6-4d07-b20e-101c7cd0b26b

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* Potential fix for pull request finding

Co-au... (continued)

281727 of 383795 relevant lines covered (73.41%)

136101761.85 hits per line

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

64.25
/source/dnode/vnode/src/meta/metaQuery.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 "meta.h"
17
#include "tencode.h"
18
#include "thash.h"
19
#include "types.h"
20

21
void _metaReaderInit(SMetaReader *pReader, void *pVnode, int32_t flags, SStoreMeta *pAPI) {
891,136,701✔
22
  SMeta *pMeta = ((SVnode *)pVnode)->pMeta;
891,136,701✔
23
  metaReaderDoInit(pReader, pMeta, flags);
891,495,731✔
24
  pReader->pAPI = pAPI;
891,482,597✔
25
}
891,489,786✔
26

27
void metaReaderDoInit(SMetaReader *pReader, SMeta *pMeta, int32_t flags) {
1,027,689,240✔
28
  memset(pReader, 0, sizeof(*pReader));
1,027,689,240✔
29
  pReader->pMeta = pMeta;
1,027,689,240✔
30
  pReader->flags = flags;
1,027,742,869✔
31
  if (pReader->pMeta && !(flags & META_READER_NOLOCK)) {
1,027,841,081✔
32
    metaRLock(pMeta);
875,598,785✔
33
  }
34
}
1,027,882,224✔
35

36
void metaReaderReleaseLock(SMetaReader *pReader) {
68,745,954✔
37
  if (pReader->pMeta && !(pReader->flags & META_READER_NOLOCK)) {
68,745,954✔
38
    metaULock(pReader->pMeta);
68,729,955✔
39
    pReader->flags |= META_READER_NOLOCK;
68,735,362✔
40
  }
41
}
68,777,381✔
42

43
void metaReaderClear(SMetaReader *pReader) {
1,260,772,492✔
44
  if (pReader->pMeta && !(pReader->flags & META_READER_NOLOCK)) {
1,260,772,492✔
45
    metaULock(pReader->pMeta);
807,037,215✔
46
  }
47
  tDecoderClear(&pReader->coder);
1,260,820,218✔
48
  tdbFree(pReader->pBuf);
1,261,204,953✔
49
  pReader->pBuf = NULL;
1,260,874,020✔
50
}
1,260,649,877✔
51

52
int metaGetTableEntryByVersion(SMetaReader *pReader, int64_t version, tb_uid_t uid) {
1,564,558,307✔
53
  int32_t  code = 0;
1,564,558,307✔
54
  SMeta   *pMeta = pReader->pMeta;
1,564,558,307✔
55
  STbDbKey tbDbKey = {.version = version, .uid = uid};
1,564,849,262✔
56

57
  // query table.db
58
  if ((code = tdbTbGet(pMeta->pTbDb, &tbDbKey, sizeof(tbDbKey), &pReader->pBuf, &pReader->szBuf)) < 0) {
1,564,878,318✔
59
    return terrno = (TSDB_CODE_NOT_FOUND == code ? TSDB_CODE_PAR_TABLE_NOT_EXIST : code);
×
60
  }
61

62
  // decode the entry
63
  tDecoderClear(&pReader->coder);
1,564,737,318✔
64
  tDecoderInit(&pReader->coder, pReader->pBuf, pReader->szBuf);
1,564,722,135✔
65

66
  code = metaDecodeEntry(&pReader->coder, &pReader->me);
1,564,860,511✔
67
  if (code) {
1,564,257,801✔
68
    tDecoderClear(&pReader->coder);
×
69
    return code;
×
70
  }
71
  // taosMemoryFreeClear(pReader->me.colCmpr.pColCmpr);
72

73
  return 0;
1,564,257,801✔
74
}
75

76
bool metaIsTableExist(void *pVnode, tb_uid_t uid) {
163,722,340✔
77
  SVnode *pVnodeObj = pVnode;
163,722,340✔
78
  metaRLock(pVnodeObj->pMeta);  // query uid.idx
163,722,340✔
79

80
  if (tdbTbGet(pVnodeObj->pMeta->pUidIdx, &uid, sizeof(uid), NULL, NULL) < 0) {
163,729,677✔
81
    metaULock(pVnodeObj->pMeta);
54,271✔
82
    return false;
54,271✔
83
  }
84

85
  metaULock(pVnodeObj->pMeta);
163,701,407✔
86
  return true;
163,705,536✔
87
}
88

89
int metaReaderGetTableEntryByUid(SMetaReader *pReader, tb_uid_t uid) {
267,209,109✔
90
  SMeta  *pMeta = pReader->pMeta;
267,209,109✔
91
  int64_t version1;
92

93
  // query uid.idx
94
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pReader->pBuf, &pReader->szBuf) < 0) {
267,214,682✔
95
    return terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
109,170✔
96
  }
97

98
  version1 = ((SUidIdxVal *)pReader->pBuf)[0].version;
267,115,458✔
99
  return metaGetTableEntryByVersion(pReader, version1, uid);
267,121,312✔
100
}
101

102
static int32_t getUidVersion(SMetaReader *pReader, int64_t *version, tb_uid_t uid) {
33,932✔
103
  int32_t code = 0;
33,932✔
104
  SMeta *pMeta = pReader->pMeta;
33,932✔
105
  void* pKey = NULL;
33,932✔
106
  int   kLen = 0;
33,932✔
107

108
  TBC* pCur = NULL;
33,932✔
109
  code = tdbTbcOpen(pMeta->pTbDb, (TBC**)&pCur, NULL);
33,932✔
110
  if (code != 0) {
33,932✔
111
    return TAOS_GET_TERRNO(code);
×
112
  }
113
  STbDbKey key = {.version = *version, .uid = INT64_MAX};
33,932✔
114
  int      c = 0;
33,932✔
115
  code = tdbTbcMoveTo(pCur, &key, sizeof(key), &c);
33,932✔
116
  if (code != 0) {
33,932✔
117
    goto END;
×
118
  }
119

120
  void   *tKey = NULL;
33,932✔
121
  code = tdbTbcGet(pCur, (const void**)&tKey, &kLen, NULL, NULL);
33,932✔
122
  if (code != 0) {
33,932✔
123
    goto END;
×
124
  }
125
  STbDbKey* tmp = (STbDbKey*)tKey;
33,932✔
126
  if (tmp->uid == uid && tmp->version <= *version) {
33,932✔
127
    *version = tmp->version;
13,835✔
128
    goto END;
13,835✔
129
  }
130
  
131
  while (1) {
35,127✔
132
    int32_t ret = tdbTbcPrev(pCur, &pKey, &kLen, NULL, NULL);
55,224✔
133
    if (ret < 0) break;
55,224✔
134

135
    STbDbKey* tmp = (STbDbKey*)pKey;
55,224✔
136
    if (tmp->uid == uid && tmp->version <= *version) {
55,224✔
137
      *version = tmp->version;
20,097✔
138
      goto END;
20,097✔
139
    }
140
  }
141
  code = TSDB_CODE_NOT_FOUND;
×
142
  metaError("%s uid:%" PRId64 " version:%" PRId64 " not found", __func__, uid, *version);
×
143
END:
33,932✔
144
  tdbFree(pKey);
33,932✔
145
  tdbTbcClose(pCur);
33,932✔
146
  return code;
33,932✔
147
} 
148

149
// get table entry according to the latest version number that is less than or equal to version and uid, if version < 0, get latest version
150
int metaReaderGetTableEntryByVersionUid(SMetaReader *pReader, int64_t version, tb_uid_t uid) {
73,245,846✔
151
  if (version < 0) {
73,245,846✔
152
    return metaReaderGetTableEntryByUid(pReader, uid);
73,248,778✔
153
  }
154
  SMeta *pMeta = pReader->pMeta;
×
155

156
  SMetaInfo info;
×
157
  int32_t   code = metaGetInfo(pMeta, uid, &info, pReader);
×
158
  if (TSDB_CODE_SUCCESS != code) {
×
159
    return terrno = (TSDB_CODE_NOT_FOUND == code ? TSDB_CODE_PAR_TABLE_NOT_EXIST : code);
×
160
  }
161
  if (info.version <= version) {
×
162
    version = info.version;
×
163
  } else {
164
    if (getUidVersion(pReader, &version, uid) != 0) {
×
165
      version = -1;
×
166
    }
167
  }
168
  return metaGetTableEntryByVersion(pReader, version, uid);
×
169
}
170

171
int metaReaderGetTableEntryByUidCache(SMetaReader *pReader, tb_uid_t uid) {
847,512,381✔
172
  SMeta *pMeta = pReader->pMeta;
847,512,381✔
173

174
  SMetaInfo info;
847,737,903✔
175
  int32_t   code = metaGetInfo(pMeta, uid, &info, pReader);
847,703,339✔
176
  if (TSDB_CODE_SUCCESS != code) {
847,467,515✔
177
    return terrno = (TSDB_CODE_NOT_FOUND == code ? TSDB_CODE_PAR_TABLE_NOT_EXIST : code);
249,491✔
178
  }
179

180
  return metaGetTableEntryByVersion(pReader, info.version, uid);
847,218,024✔
181
}
182

183
int metaGetTableEntryByVersionName(SMetaReader *pReader, int64_t version, const char *name) {
81,236,359✔
184
  SMeta   *pMeta = pReader->pMeta;
81,236,359✔
185
  tb_uid_t uid;
186

187
  // query name.idx
188
  if (tdbTbGet(pMeta->pNameIdx, name, strlen(name) + 1, &pReader->pBuf, &pReader->szBuf) < 0) {
81,239,407✔
189
    return terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
11,004,607✔
190
  }
191

192
  uid = *(tb_uid_t *)pReader->pBuf;
70,233,562✔
193
  return metaReaderGetTableEntryByVersionUid(pReader, version, uid);
70,235,047✔
194
}
195

196
int metaGetTableEntryByName(SMetaReader *pReader, const char *name) {
80,845,244✔
197
  return metaGetTableEntryByVersionName(pReader, -1, name);
80,845,244✔
198
}
199

200
tb_uid_t metaGetTableEntryUidByName(SMeta *pMeta, const char *name) {
83,893,862✔
201
  void    *pData = NULL;
83,893,862✔
202
  int      nData = 0;
83,897,256✔
203
  tb_uid_t uid = 0;
83,899,876✔
204

205
  metaRLock(pMeta);
83,899,876✔
206

207
  if (tdbTbGet(pMeta->pNameIdx, name, strlen(name) + 1, &pData, &nData) == 0) {
83,900,291✔
208
    uid = *(tb_uid_t *)pData;
10,885,322✔
209
    tdbFree(pData);
10,885,496✔
210
  }
211

212
  metaULock(pMeta);
83,887,266✔
213

214
  return uid;
83,890,559✔
215
}
216

217
int metaGetTableNameByUid(void *pVnode, uint64_t uid, char *tbName) {
8,291,669✔
218
  int         code = 0;
8,291,669✔
219
  SMetaReader mr = {0};
8,291,669✔
220
  metaReaderDoInit(&mr, ((SVnode *)pVnode)->pMeta, META_READER_LOCK);
8,291,669✔
221
  code = metaReaderGetTableEntryByUid(&mr, uid);
8,289,891✔
222
  if (code < 0) {
8,286,665✔
223
    metaReaderClear(&mr);
307✔
224
    return code;
307✔
225
  }
226

227
  STR_TO_VARSTR(tbName, mr.me.name);
8,286,358✔
228
  metaReaderClear(&mr);
8,287,804✔
229

230
  return 0;
8,286,192✔
231
}
232

233
int metaGetTableSzNameByUid(void *meta, uint64_t uid, char *tbName) {
×
234
  int         code = 0;
×
235
  SMetaReader mr = {0};
×
236
  metaReaderDoInit(&mr, (SMeta *)meta, META_READER_LOCK);
×
237
  code = metaReaderGetTableEntryByUid(&mr, uid);
×
238
  if (code < 0) {
×
239
    metaReaderClear(&mr);
×
240
    return code;
×
241
  }
242
  tstrncpy(tbName, mr.me.name, TSDB_TABLE_NAME_LEN);
×
243
  metaReaderClear(&mr);
×
244

245
  return 0;
×
246
}
247

248
int metaGetTableUidByName(void *pVnode, char *tbName, uint64_t *uid) {
790,191✔
249
  int         code = 0;
790,191✔
250
  SMetaReader mr = {0};
790,191✔
251
  metaReaderDoInit(&mr, ((SVnode *)pVnode)->pMeta, META_READER_LOCK);
790,191✔
252

253
  SMetaReader *pReader = &mr;
790,189✔
254

255
  // query name.idx
256
  if (tdbTbGet(((SMeta *)pReader->pMeta)->pNameIdx, tbName, strlen(tbName) + 1, &pReader->pBuf, &pReader->szBuf) < 0) {
790,189✔
257
    metaReaderClear(&mr);
322,397✔
258
    return terrno = TSDB_CODE_TDB_TABLE_NOT_EXIST;
322,397✔
259
  }
260

261
  *uid = *(tb_uid_t *)pReader->pBuf;
467,794✔
262

263
  metaReaderClear(&mr);
467,794✔
264

265
  return 0;
467,783✔
266
}
267

268
int metaGetTableTypeSuidByName(void *pVnode, char *tbName, ETableType *tbType, uint64_t *suid) {
465,067✔
269
  int         code = 0;
465,067✔
270
  SMetaReader mr = {0};
465,067✔
271
  metaReaderDoInit(&mr, ((SVnode *)pVnode)->pMeta, META_READER_LOCK);
465,067✔
272

273
  code = metaGetTableEntryByName(&mr, tbName);
465,067✔
274
  if (code == 0) *tbType = mr.me.type;
465,067✔
275
  if (TSDB_CHILD_TABLE == mr.me.type || TSDB_VIRTUAL_CHILD_TABLE == mr.me.type) {
465,067✔
276
    *suid = mr.me.ctbEntry.suid;
455,052✔
277
  } else if (TSDB_SUPER_TABLE == mr.me.type) {
10,015✔
278
    *suid = mr.me.uid;
7,875✔
279
  } else {
280
    *suid = 0;
2,140✔
281
  }
282

283
  metaReaderClear(&mr);
465,067✔
284
  return code;
465,067✔
285
}
286

287
int metaReadNext(SMetaReader *pReader) {
×
288
  SMeta *pMeta = pReader->pMeta;
×
289

290
  // TODO
291

292
  return 0;
×
293
}
294

295
int metaGetTableTtlByUid(void *meta, uint64_t uid, int64_t *ttlDays) {
×
296
  int         code = -1;
×
297
  SMetaReader mr = {0};
×
298
  metaReaderDoInit(&mr, (SMeta *)meta, META_READER_LOCK);
×
299
  code = metaReaderGetTableEntryByUid(&mr, uid);
×
300
  if (code < 0) {
×
301
    goto _exit;
×
302
  }
303
  if (mr.me.type == TSDB_CHILD_TABLE || mr.me.type == TSDB_VIRTUAL_CHILD_TABLE) {
×
304
    *ttlDays = mr.me.ctbEntry.ttlDays;
×
305
  } else if (mr.me.type == TSDB_NORMAL_TABLE || mr.me.type == TSDB_VIRTUAL_NORMAL_TABLE) {
×
306
    *ttlDays = mr.me.ntbEntry.ttlDays;
×
307
  } else {
308
    goto _exit;
×
309
  }
310

311
  code = 0;
×
312

313
_exit:
×
314
  metaReaderClear(&mr);
×
315
  return code;
×
316
}
317

318
#if 1  // ===================================================
319
SMTbCursor *metaOpenTbCursor(void *pVnode) {
20,079,451✔
320
  SMTbCursor *pTbCur = NULL;
20,079,451✔
321
  int32_t     code;
322

323
  pTbCur = (SMTbCursor *)taosMemoryCalloc(1, sizeof(*pTbCur));
20,079,451✔
324
  if (pTbCur == NULL) {
20,076,628✔
325
    return NULL;
×
326
  }
327

328
  SVnode *pVnodeObj = pVnode;
20,076,628✔
329
  // tdbTbcMoveToFirst((TBC *)pTbCur->pDbc);
330
  pTbCur->pMeta = pVnodeObj->pMeta;
20,076,628✔
331
  pTbCur->paused = 1;
20,078,340✔
332
  code = metaResumeTbCursor(pTbCur, 1, 0);
20,078,385✔
333
  if (code) {
20,076,102✔
334
    terrno = code;
×
335
    taosMemoryFree(pTbCur);
×
336
    return NULL;
×
337
  }
338
  return pTbCur;
20,076,102✔
339
}
340

341
void metaCloseTbCursor(SMTbCursor *pTbCur) {
40,152,290✔
342
  if (pTbCur) {
40,152,290✔
343
    tdbFree(pTbCur->pKey);
20,080,023✔
344
    tdbFree(pTbCur->pVal);
20,079,797✔
345
    if (!pTbCur->paused) {
20,080,061✔
346
      metaReaderClear(&pTbCur->mr);
9,377,862✔
347
      if (pTbCur->pDbc) {
9,376,714✔
348
        tdbTbcClose((TBC *)pTbCur->pDbc);
9,376,054✔
349
      }
350
    }
351
    taosMemoryFree(pTbCur);
20,079,496✔
352
  }
353
}
40,154,564✔
354

355
void metaPauseTbCursor(SMTbCursor *pTbCur) {
10,705,265✔
356
  if (!pTbCur->paused) {
10,705,265✔
357
    metaReaderClear(&pTbCur->mr);
10,705,842✔
358
    tdbTbcClose((TBC *)pTbCur->pDbc);
10,705,243✔
359
    pTbCur->paused = 1;
10,704,089✔
360
  }
361
}
10,704,792✔
362
int32_t metaResumeTbCursor(SMTbCursor *pTbCur, int8_t first, int8_t move) {
20,081,269✔
363
  int32_t code = 0;
20,081,269✔
364
  int32_t lino;
365
  int8_t  locked = 0;
20,081,269✔
366
  if (pTbCur->paused) {
20,081,269✔
367
    metaReaderDoInit(&pTbCur->mr, pTbCur->pMeta, META_READER_LOCK);
20,078,908✔
368
    locked = 1;
20,083,061✔
369
    code = tdbTbcOpen(((SMeta *)pTbCur->pMeta)->pUidIdx, (TBC **)&pTbCur->pDbc, NULL);
20,083,061✔
370
    if (code != 0) {
20,081,265✔
371
      TSDB_CHECK_CODE(code, lino, _exit);
×
372
    }
373

374
    if (first) {
20,081,265✔
375
      code = tdbTbcMoveToFirst((TBC *)pTbCur->pDbc);
20,078,359✔
376
      TSDB_CHECK_CODE(code, lino, _exit);
20,069,433✔
377
    } else {
378
      int c = 1;
2,906✔
379
      code = tdbTbcMoveTo(pTbCur->pDbc, pTbCur->pKey, pTbCur->kLen, &c);
2,906✔
380
      TSDB_CHECK_CODE(code, lino, _exit);
2,906✔
381
      if (c == 0) {
2,906✔
382
        if (move) tdbTbcMoveToNext(pTbCur->pDbc);
2,906✔
383
      } else if (c < 0) {
×
384
        code = tdbTbcMoveToPrev(pTbCur->pDbc);
×
385
        TSDB_CHECK_CODE(code, lino, _exit);
×
386
      } else {
387
        code = tdbTbcMoveToNext(pTbCur->pDbc);
×
388
        TSDB_CHECK_CODE(code, lino, _exit);
×
389
      }
390
    }
391

392
    pTbCur->paused = 0;
20,072,919✔
393
  }
394

395
_exit:
×
396
  if (code != 0 && locked) {
20,078,454✔
397
    metaReaderReleaseLock(&pTbCur->mr);
×
398
  }
399
  return code;
20,079,585✔
400
}
401

402
int32_t metaTbCursorNext(SMTbCursor *pTbCur, ETableType jumpTableType) {
470,383,290✔
403
  int    ret;
404
  void  *pBuf;
405
  STbCfg tbCfg;
406

407
  for (;;) {
408
    ret = tdbTbcNext((TBC *)pTbCur->pDbc, &pTbCur->pKey, &pTbCur->kLen, &pTbCur->pVal, &pTbCur->vLen);
470,383,290✔
409
    if (ret < 0) {
470,385,999✔
410
      return ret;
20,075,579✔
411
    }
412

413
    tDecoderClear(&pTbCur->mr.coder);
450,310,420✔
414

415
    ret = metaGetTableEntryByVersion(&pTbCur->mr, ((SUidIdxVal *)pTbCur->pVal)[0].version, *(tb_uid_t *)pTbCur->pKey);
450,327,082✔
416
    if (ret) return ret;
450,313,571✔
417

418
    if (pTbCur->mr.me.type == jumpTableType) {
450,313,571✔
419
      continue;
4,981,341✔
420
    }
421

422
    break;
445,345,659✔
423
  }
424

425
  return 0;
445,345,659✔
426
}
427

428
int32_t metaTbCursorPrev(SMTbCursor *pTbCur, ETableType jumpTableType) {
×
429
  int    ret;
430
  void  *pBuf;
431
  STbCfg tbCfg;
432

433
  for (;;) {
434
    ret = tdbTbcPrev((TBC *)pTbCur->pDbc, &pTbCur->pKey, &pTbCur->kLen, &pTbCur->pVal, &pTbCur->vLen);
×
435
    if (ret < 0) {
×
436
      return -1;
×
437
    }
438

439
    tDecoderClear(&pTbCur->mr.coder);
×
440

441
    ret = metaGetTableEntryByVersion(&pTbCur->mr, ((SUidIdxVal *)pTbCur->pVal)[0].version, *(tb_uid_t *)pTbCur->pKey);
×
442
    if (ret < 0) {
×
443
      return ret;
×
444
    }
445

446
    if (pTbCur->mr.me.type == jumpTableType) {
×
447
      continue;
×
448
    }
449

450
    break;
×
451
  }
452

453
  return 0;
×
454
}
455

456
static SSchemaWrapper * getSchemaWithVer(SMeta *pMeta, tb_uid_t uid, int32_t sver, SExtSchema **extSchema) {
303,200✔
457
  SDecoder        dc = {0};
303,200✔
458
  int32_t         code = 0;
303,200✔
459
  void           *pData = NULL;
303,200✔
460
  int             nData = 0;
303,200✔
461
  SSchemaWrapper  schema = {0};
303,200✔
462
  SSchemaWrapper *pSchema = NULL;
303,200✔
463
  
464
  // query from skm ext db
465
  if (extSchema != NULL) {
303,200✔
466
    if ((tdbTbGet(pMeta->pSkmExtDb, &(SSkmDbKey){.uid = uid, .sver = sver}, sizeof(SSkmDbKey), &pData, &nData)) == 0) {
299,056✔
467
      *extSchema = taosMemoryMalloc(nData);
5,533✔
468
      if (*extSchema) {
5,533✔
469
        (void)memcpy(*extSchema, pData, nData);
5,533✔
470
      }
471
    }
472
    tdbFree(pData);
298,673✔
473
    pData = NULL;
299,056✔
474
  }
475
  
476
  // query from skm db
477
  if ((code = tdbTbGet(pMeta->pSkmDb, &(SSkmDbKey){.uid = uid, .sver = sver}, sizeof(SSkmDbKey), &pData, &nData)) < 0) {
303,200✔
478
    goto _err;
7,446✔
479
  }
480

481
  tDecoderInit(&dc, pData, nData);
295,754✔
482
  if ((code = tDecodeSSchemaWrapperEx(&dc, &schema)) != 0) {
295,754✔
483
    tDecoderClear(&dc);
×
484
    goto _err;
×
485
  }
486
  pSchema = tCloneSSchemaWrapper(&schema);
295,371✔
487
  if (pSchema == NULL) {
295,371✔
488
    tDecoderClear(&dc);
×
489
    code = terrno;
×
490
    goto _err;
×
491
  }
492
  tDecoderClear(&dc);
295,371✔
493

494
  tdbFree(pData);
295,754✔
495
  return pSchema;
295,371✔
496

497
_err:
7,446✔
498
  tdbFree(pData);
7,446✔
499
  tDeleteSchemaWrapper(pSchema);
500
  if (extSchema != NULL) {
7,446✔
501
    taosMemoryFreeClear(*extSchema);
7,446✔
502
  }
503
  terrno = code;
7,446✔
504
  return NULL;
7,446✔
505
}
506

507
static int32_t metaGetSuidByUidIfTableNotExist(SMeta *pMeta, int64_t uid, int64_t* id){
250,873✔
508
  int32_t code = 0;
250,873✔
509
  void   *pKey = NULL;
250,873✔
510
  void   *pVal = NULL;
250,873✔
511
  int     kLen = 0;
250,873✔
512
  int     vLen = 0;
250,873✔
513

514
  int64_t* suid = (int64_t*)taosHashGet(pMeta->uidSuidHash, &uid, sizeof(uid));
250,873✔
515
  if (suid != NULL) {
250,490✔
516
    *id = *suid;
165,607✔
517
    return code;
165,607✔
518
  }
519

520
  TBC *pCur = NULL;
84,883✔
521
  code = tdbTbcOpen(pMeta->pTbDb, (TBC**)&pCur, NULL);
84,883✔
522
  if (code != 0) {
84,883✔
523
    goto END;
×
524
  }
525
  code = tdbTbcMoveToFirst(pCur);
84,883✔
526
  if (code != 0) {
84,883✔
527
    goto END;
×
528
  }
529

530
  void       *tKey = NULL;
84,883✔
531
  void       *tVal = NULL;
84,883✔
532
  int         tKLen = 0;
84,883✔
533
  int         tVLen = 0;
84,883✔
534
  code = tdbTbcGet(pCur, (const void**)&tKey, &tKLen, (const void**)&tVal, &tVLen);
84,883✔
535
  if (code != 0) {
84,883✔
536
    goto END;
×
537
  }
538
  STbDbKey *tmp = (STbDbKey*)tKey;
84,883✔
539
  if (tmp->uid == uid) {
84,883✔
540
    SMetaEntry  me = {0};
×
541
    SDecoder    dc = {0};
×
542
    tDecoderInit(&dc, tVal, tVLen);
×
543
    if (metaDecodeEntry(&dc, &me) == 0 && (me.type == TSDB_CHILD_TABLE || me.type == TSDB_VIRTUAL_CHILD_TABLE)) {
×
544
      *id = me.ctbEntry.suid;
×
545
    }
546
    tDecoderClear(&dc);
×
547
    goto END;
×
548
  }
549

550
  while (1) {
2,583,841✔
551
    int32_t ret = tdbTbcNext(pCur, &pKey, &kLen, &pVal, &vLen);
2,668,724✔
552
    if (ret < 0) break;
2,668,724✔
553

554
    tmp = (STbDbKey*)pKey;
2,668,724✔
555
    if (tmp->uid == uid) {
2,668,724✔
556
      SMetaEntry  me = {0};
84,883✔
557
      SDecoder    dc = {0};
84,883✔
558
      tDecoderInit(&dc, pVal, vLen);
84,883✔
559
      if (metaDecodeEntry(&dc, &me) == 0 && (me.type == TSDB_CHILD_TABLE || me.type == TSDB_VIRTUAL_CHILD_TABLE)) {
84,883✔
560
        *id = me.ctbEntry.suid;
83,845✔
561
      }
562
      tDecoderClear(&dc);
84,513✔
563
      goto END;
84,883✔
564
    }
565
  }
566

567
END:
84,883✔
568
  tdbFree(pKey);
84,883✔
569
  tdbFree(pVal);
84,883✔
570
  tdbTbcClose(pCur);
84,513✔
571
  if (code != 0) {
84,883✔
572
    metaError("%s failed to get suid reason:%s", __func__, tstrerror(terrno));
×
573
  } else {
574
    code = taosHashPut(pMeta->uidSuidHash, &uid, sizeof(uid), id, sizeof(*id));
84,883✔
575
    if (code != 0) {
84,883✔
576
      metaError("%s failed to put suid into uidSuidHash, reason:%s", __func__, tstrerror(terrno));
×
577
    }
578
  }
579
  return code;
84,883✔
580
}
581

582
int32_t metaGetTbnameByIdIfTableNotExist(SMeta *pMeta, int64_t uid, char *tbname){
241,740✔
583
  int32_t code = 0;
241,740✔
584
  void   *pKey = NULL;
241,740✔
585
  void   *pVal = NULL;
242,123✔
586
  int     kLen = 0;
242,123✔
587
  int     vLen = 0;
242,123✔
588

589
  char* name = (char*)taosHashGet(pMeta->uidNameHash, &uid, sizeof(uid));
242,123✔
590
  if (name != NULL) {
242,123✔
591
    tstrncpy(tbname, name, TSDB_TABLE_NAME_LEN);
159,256✔
592
    return code;
159,256✔
593
  }
594

595
  TBC *pCur = NULL;
82,867✔
596
  code = tdbTbcOpen(pMeta->pTbDb, (TBC**)&pCur, NULL);
82,867✔
597
  if (code != 0) {
82,867✔
598
    goto END;
×
599
  }
600
  code = tdbTbcMoveToFirst(pCur);
82,867✔
601
  if (code != 0) {
82,867✔
602
    goto END;
×
603
  }
604

605
  void       *tKey = NULL;
82,867✔
606
  void       *tVal = NULL;
82,867✔
607
  int         tKLen = 0;
82,867✔
608
  int         tVLen = 0;
82,867✔
609
  code = tdbTbcGet(pCur, (const void**)&tKey, &tKLen, (const void**)&tVal, &tVLen);
82,867✔
610
  if (code != 0) {
82,867✔
611
    goto END;
×
612
  }
613
  STbDbKey *tmp = (STbDbKey*)tKey;
82,867✔
614
  if (tmp->uid == uid) {
82,867✔
615
    SMetaEntry  me = {0};
×
616
    SDecoder    dc = {0};
×
617
    tDecoderInit(&dc, tVal, tVLen);
×
618
    code = metaDecodeEntry(&dc, &me);
×
619
    if (code == 0){
×
620
      tstrncpy(tbname, me.name, TSDB_TABLE_NAME_LEN);
×
621
    }
622
    tDecoderClear(&dc);
×
623
    goto END;
×
624
  }
625

626
  while (1) {
2,484,095✔
627
    int32_t ret = tdbTbcNext(pCur, &pKey, &kLen, &pVal, &vLen);
2,566,962✔
628
    if (ret < 0) break;
2,566,962✔
629

630
    tmp = (STbDbKey*)pKey;
2,566,962✔
631
    if (tmp->uid == uid) {
2,566,962✔
632
      SMetaEntry  me = {0};
82,867✔
633
      SDecoder    dc = {0};
82,867✔
634
      tDecoderInit(&dc, pVal, vLen);
82,867✔
635
      code = metaDecodeEntry(&dc, &me);
82,867✔
636
      if (code == 0){
82,867✔
637
        tstrncpy(tbname, me.name, TSDB_TABLE_NAME_LEN);
82,867✔
638
      }
639
      tDecoderClear(&dc);
82,867✔
640
      goto END;
82,867✔
641
    }
642
  }
643

644
END:
82,867✔
645
  tdbFree(pKey);
82,867✔
646
  tdbFree(pVal);
82,867✔
647
  tdbTbcClose(pCur);
82,497✔
648
  if (code != 0) {
82,497✔
649
    metaError("%s failed to get suid reason:%s", __func__, tstrerror(terrno));
×
650
  } else {
651
    code = taosHashPut(pMeta->uidNameHash, &uid, sizeof(uid), tbname, strlen(tbname) + 1);
82,497✔
652
    if (code != 0) {
82,867✔
653
      metaError("%s failed to put suid into uidSuidHash, reason:%s", __func__, tstrerror(terrno));
×
654
    }
655
  }
656
  return code;
82,867✔
657
}
658

659
/**
660
 * @param type 0x01 fetchRsmaSchema if table is rsma
661
 */
662
SSchemaWrapper *metaGetTableSchema(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock, SExtSchema **extSchema,
382,809,763✔
663
                                   int8_t type, bool ignoreExist) {
664
  int32_t         code = 0;
382,809,763✔
665
  void           *pData = NULL;
382,809,763✔
666
  int             nData = 0;
382,706,765✔
667
  int64_t         version;
668
  SSchemaWrapper  schema = {0};
382,654,835✔
669
  SSchemaWrapper *pSchema = NULL;
382,654,835✔
670
  SDecoder        dc = {0};
382,654,835✔
671
  if (lock) {
382,412,265✔
672
    metaRLock(pMeta);
376,756,469✔
673
  }
674
_query:
420,533,673✔
675
  code = tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData);
420,618,192✔
676
  if (code == TSDB_CODE_NOT_FOUND && ignoreExist){
420,673,036✔
677
    int64_t id = 0;
250,873✔
678
    code = metaGetSuidByUidIfTableNotExist(pMeta, uid, &id);
250,873✔
679
    if (code != 0) {
250,873✔
680
      goto _err;
×
681
    }
682
    pSchema = getSchemaWithVer(pMeta, id, sver, extSchema);
250,873✔
683
    goto _exit;
250,873✔
684
  }
685

686
  if (code < 0) {
420,422,163✔
687
    goto _err;
188,819✔
688
  }
689

690
  version = ((SUidIdxVal *)pData)[0].version;
420,233,344✔
691

692
  if ((code = tdbTbGet(pMeta->pTbDb, &(STbDbKey){.uid = uid, .version = version}, sizeof(STbDbKey), &pData, &nData)) !=
420,437,255✔
693
      0) {
694
    goto _err;
×
695
  }
696

697
  SMetaEntry me = {0};
420,344,576✔
698
  tDecoderInit(&dc, pData, nData);
420,396,025✔
699
  code = metaDecodeEntry(&dc, &me);
420,447,739✔
700
  if (code) {
420,283,217✔
701
    tDecoderClear(&dc);
×
702
    goto _err;
×
703
  }
704
  if (me.type == TSDB_SUPER_TABLE) {
420,283,217✔
705
    if (sver == -1 || sver == me.stbEntry.schemaRow.version) {
306,133,032✔
706
      pSchema = tCloneSSchemaWrapper(&me.stbEntry.schemaRow);
306,009,803✔
707
      if (extSchema != NULL) *extSchema = metaGetSExtSchema(&me);
306,009,803✔
708
      if (TABLE_IS_ROLLUP(me.flags)) {
306,003,293✔
709
        if ((type == 0x01) && (code = metaGetRsmaSchema(&me, &pSchema->pRsma)) != 0) {
118,133✔
710
          tDecoderClear(&dc);
×
711
          goto _err;
×
712
        }
713
      }
714
      tDecoderClear(&dc);
306,003,293✔
715
      goto _exit;
306,009,012✔
716
    }
717
  } else if (me.type == TSDB_CHILD_TABLE || me.type == TSDB_VIRTUAL_CHILD_TABLE) {
114,150,185✔
718
    uid = me.ctbEntry.suid;
37,830,410✔
719
    tDecoderClear(&dc);
37,830,410✔
720
    goto _query;
37,847,141✔
721
  } else {
722
    if (sver == -1 || sver == me.ntbEntry.schemaRow.version) {
76,319,775✔
723
      pSchema = tCloneSSchemaWrapper(&me.ntbEntry.schemaRow);
76,344,764✔
724
      if (extSchema != NULL) *extSchema = metaGetSExtSchema(&me);
76,344,764✔
725
      tDecoderClear(&dc);
76,368,616✔
726
      goto _exit;
76,351,214✔
727
    }
728
  }
729
  tDecoderClear(&dc);
94,058✔
730

731
  pSchema = getSchemaWithVer(pMeta, uid, sver, extSchema);
52,327✔
732

733
_exit:
382,663,414✔
734
  if (lock) {
382,714,369✔
735
    metaULock(pMeta);
377,101,025✔
736
  }
737
  tdbFree(pData);
382,578,031✔
738
  return pSchema;
382,598,424✔
739

740
_err:
145,823✔
741
  if (lock) {
188,819✔
742
    metaULock(pMeta);
188,819✔
743
  }
744
  tdbFree(pData);
188,819✔
745
  tDeleteSchemaWrapper(pSchema);
746
  if (extSchema != NULL) {
188,819✔
747
    taosMemoryFreeClear(*extSchema);
×
748
  }
749
  terrno = code;
188,819✔
750
  return NULL;
188,819✔
751
}
752

753
int64_t metaGetTableCreateTime(SMeta *pMeta, tb_uid_t uid, int lock) {
1,041,360✔
754
  void    *pData = NULL;
1,041,360✔
755
  int      nData = 0;
1,042,112✔
756
  int64_t  version = 0;
1,042,489✔
757
  SDecoder dc = {0};
1,042,489✔
758
  int64_t  createTime = INT64_MAX;
1,042,112✔
759
  if (lock) {
1,042,112✔
760
    metaRLock(pMeta);
1,042,864✔
761
  }
762

763
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData) < 0) {
1,041,358✔
764
    goto _exit;
826✔
765
  }
766

767
  version = ((SUidIdxVal *)pData)[0].version;
1,041,663✔
768

769
  if (tdbTbGet(pMeta->pTbDb, &(STbDbKey){.uid = uid, .version = version}, sizeof(STbDbKey), &pData, &nData) != 0) {
1,042,038✔
770
    goto _exit;
×
771
  }
772

773
  SMetaEntry me = {0};
1,041,663✔
774
  tDecoderInit(&dc, pData, nData);
1,041,663✔
775
  int32_t code = metaDecodeEntry(&dc, &me);
1,041,661✔
776
  if (code) {
1,041,286✔
777
    tDecoderClear(&dc);
×
778
    goto _exit;
×
779
  }
780
  if (me.type == TSDB_CHILD_TABLE || me.type == TSDB_VIRTUAL_CHILD_TABLE) {
1,041,286✔
781
    createTime = me.ctbEntry.btime;
1,040,903✔
782
  } else if (me.type == TSDB_NORMAL_TABLE || me.type == TSDB_VIRTUAL_NORMAL_TABLE) {
383✔
783
    createTime = me.ntbEntry.btime;
383✔
784
  }
785
  tDecoderClear(&dc);
1,041,286✔
786

787
_exit:
1,042,112✔
788
  if (lock) {
1,041,735✔
789
    metaULock(pMeta);
1,042,487✔
790
  }
791
  tdbFree(pData);
1,040,983✔
792
  return createTime;
1,042,487✔
793
}
794

795
SMCtbCursor *metaOpenCtbCursor(void *pVnode, tb_uid_t uid, int lock) {
153,770,942✔
796
  SMeta       *pMeta = ((SVnode *)pVnode)->pMeta;
153,770,942✔
797
  SMCtbCursor *pCtbCur = NULL;
153,790,868✔
798
  SCtbIdxKey   ctbIdxKey;
799
  int          ret = 0;
153,790,868✔
800
  int          c = 0;
153,790,868✔
801

802
  pCtbCur = (SMCtbCursor *)taosMemoryCalloc(1, sizeof(*pCtbCur));
153,790,868✔
803
  if (pCtbCur == NULL) {
153,686,422✔
804
    return NULL;
×
805
  }
806

807
  pCtbCur->pMeta = pMeta;
153,686,422✔
808
  pCtbCur->suid = uid;
153,700,093✔
809
  pCtbCur->lock = lock;
153,708,785✔
810
  pCtbCur->paused = 1;
153,721,992✔
811

812
  ret = metaResumeCtbCursor(pCtbCur, 1);
153,710,443✔
813
  if (ret < 0) {
153,751,286✔
814
    return NULL;
×
815
  }
816
  return pCtbCur;
153,751,286✔
817
}
818

819
void metaCloseCtbCursor(SMCtbCursor *pCtbCur) {
153,804,708✔
820
  if (pCtbCur) {
153,804,708✔
821
    if (!pCtbCur->paused) {
153,806,173✔
822
      if (pCtbCur->pMeta && pCtbCur->lock) metaULock(pCtbCur->pMeta);
148,616,706✔
823
      if (pCtbCur->pCur) {
148,624,077✔
824
        tdbTbcClose(pCtbCur->pCur);
148,619,640✔
825
      }
826
    }
827
    tdbFree(pCtbCur->pKey);
153,758,120✔
828
    tdbFree(pCtbCur->pVal);
153,717,202✔
829
  }
830
  taosMemoryFree(pCtbCur);
153,744,413✔
831
}
153,728,459✔
832

833
void metaPauseCtbCursor(SMCtbCursor *pCtbCur) {
5,189,049✔
834
  if (!pCtbCur->paused) {
5,189,049✔
835
    tdbTbcClose((TBC *)pCtbCur->pCur);
5,187,728✔
836
    if (pCtbCur->lock) {
5,186,287✔
837
      metaULock(pCtbCur->pMeta);
5,189,789✔
838
    }
839
    pCtbCur->paused = 1;
5,183,387✔
840
  }
841
}
5,192,624✔
842

843
int32_t metaResumeCtbCursor(SMCtbCursor *pCtbCur, int8_t first) {
153,771,704✔
844
  if (pCtbCur->paused) {
153,771,704✔
845
    pCtbCur->paused = 0;
153,703,674✔
846

847
    if (pCtbCur->lock) {
153,752,363✔
848
      metaRLock(pCtbCur->pMeta);
148,110,428✔
849
    }
850
    int ret = 0;
153,646,087✔
851
    ret = tdbTbcOpen(pCtbCur->pMeta->pCtbIdx, (TBC **)&pCtbCur->pCur, NULL);
153,646,087✔
852
    if (ret < 0) {
153,751,046✔
853
      metaCloseCtbCursor(pCtbCur);
×
854
      return -1;
×
855
    }
856

857
    if (first) {
153,751,058✔
858
      SCtbIdxKey ctbIdxKey;
153,746,570✔
859
      // move to the suid
860
      ctbIdxKey.suid = pCtbCur->suid;
153,756,564✔
861
      ctbIdxKey.uid = INT64_MIN;
153,783,109✔
862
      int c = 0;
153,783,109✔
863
      ret = tdbTbcMoveTo(pCtbCur->pCur, &ctbIdxKey, sizeof(ctbIdxKey), &c);
153,653,340✔
864
      if (c > 0) {
153,739,922✔
865
        ret = tdbTbcMoveToNext(pCtbCur->pCur);
28,404,720✔
866
      }
867
    } else {
868
      int c = 0;
×
869
      ret = tdbTbcMoveTo(pCtbCur->pCur, pCtbCur->pKey, pCtbCur->kLen, &c);
×
870
      if (c < 0) {
×
871
        ret = tdbTbcMoveToPrev(pCtbCur->pCur);
×
872
      } else {
873
        ret = tdbTbcMoveToNext(pCtbCur->pCur);
×
874
      }
875
    }
876
  }
877
  return 0;
153,761,086✔
878
}
879

880
tb_uid_t metaCtbCursorNext(SMCtbCursor *pCtbCur) {
848,882,143✔
881
  int         ret;
882
  SCtbIdxKey *pCtbIdxKey;
883

884
  ret = tdbTbcNext(pCtbCur->pCur, &pCtbCur->pKey, &pCtbCur->kLen, &pCtbCur->pVal, &pCtbCur->vLen);
848,882,143✔
885
  if (ret < 0) {
848,960,451✔
886
    return 0;
81,248,790✔
887
  }
888

889
  pCtbIdxKey = pCtbCur->pKey;
767,711,661✔
890
  if (pCtbIdxKey->suid > pCtbCur->suid) {
767,796,884✔
891
    return 0;
72,676,041✔
892
  }
893

894
  return pCtbIdxKey->uid;
695,085,944✔
895
}
896

897
struct SMStbCursor {
898
  SMeta   *pMeta;
899
  TBC     *pCur;
900
  tb_uid_t suid;
901
  void    *pKey;
902
  void    *pVal;
903
  int      kLen;
904
  int      vLen;
905
};
906

907
SMStbCursor *metaOpenStbCursor(SMeta *pMeta, tb_uid_t suid) {
80,063,501✔
908
  SMStbCursor *pStbCur = NULL;
80,063,501✔
909
  int          ret = 0;
80,063,501✔
910
  int          c = 0;
80,063,501✔
911

912
  pStbCur = (SMStbCursor *)taosMemoryCalloc(1, sizeof(*pStbCur));
80,063,501✔
913
  if (pStbCur == NULL) {
80,063,080✔
914
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
915
    return NULL;
×
916
  }
917

918
  pStbCur->pMeta = pMeta;
80,063,080✔
919
  pStbCur->suid = suid;
80,063,406✔
920
  metaRLock(pMeta);
80,062,778✔
921

922
  ret = tdbTbcOpen(pMeta->pSuidIdx, &pStbCur->pCur, NULL);
80,063,406✔
923
  if (ret < 0) {
80,052,916✔
924
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
925
    metaULock(pMeta);
×
926
    taosMemoryFree(pStbCur);
×
927
    return NULL;
×
928
  }
929

930
  // move to the suid
931
  ret = tdbTbcMoveTo(pStbCur->pCur, &suid, sizeof(suid), &c);
80,052,916✔
932
  if (c > 0) {
80,068,467✔
933
    ret = tdbTbcMoveToNext(pStbCur->pCur);
×
934
  }
935

936
  return pStbCur;
80,064,947✔
937
}
938

939
void metaCloseStbCursor(SMStbCursor *pStbCur) {
80,062,641✔
940
  if (pStbCur) {
80,062,641✔
941
    if (pStbCur->pMeta) metaULock(pStbCur->pMeta);
80,064,087✔
942
    if (pStbCur->pCur) {
80,068,084✔
943
      tdbTbcClose(pStbCur->pCur);
80,063,023✔
944

945
      tdbFree(pStbCur->pKey);
80,059,082✔
946
      tdbFree(pStbCur->pVal);
80,063,080✔
947
    }
948

949
    taosMemoryFree(pStbCur);
80,063,080✔
950
  }
951
}
80,056,516✔
952

953
tb_uid_t metaStbCursorNext(SMStbCursor *pStbCur) {
142,701,370✔
954
  int ret;
955

956
  ret = tdbTbcNext(pStbCur->pCur, &pStbCur->pKey, &pStbCur->kLen, &pStbCur->pVal, &pStbCur->vLen);
142,701,370✔
957
  if (ret < 0) {
142,695,448✔
958
    return 0;
80,068,467✔
959
  }
960
  return *(tb_uid_t *)pStbCur->pKey;
62,626,981✔
961
}
962

963
STSchema *metaGetTbTSchema(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock) {
323,485,902✔
964
  STSchema       *pTSchema = NULL;
323,485,902✔
965
  SSchemaWrapper *pSW = NULL;
323,485,902✔
966

967
  pSW = metaGetTableSchema(pMeta, uid, sver, lock, NULL, 0, false);
323,485,902✔
968
  if (!pSW) return NULL;
323,569,612✔
969

970
  pTSchema = tBuildTSchema(pSW->pSchema, pSW->nCols, pSW->version);
323,528,622✔
971

972
  tDeleteSchemaWrapper(pSW);
973
  return pTSchema;
323,332,128✔
974
}
975

976
/**
977
 * Fetch rsma schema if table is rsma
978
 */
979
SRSchema *metaGetTbTSchemaR(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock) {
30,525✔
980
  SRSchema       *pRSchema = NULL;
30,525✔
981
  SSchemaWrapper *pSW = NULL;
30,525✔
982

983
  if (!(pRSchema = (SRSchema *)taosMemoryCalloc(1, sizeof(SRSchema)))) goto _err;
30,525✔
984
  if (!(pSW = metaGetTableSchema(pMeta, uid, sver, lock, (SExtSchema **)&pRSchema->extSchema, 0x01, false))) goto _err;
30,525✔
985
  if (!(pRSchema->tSchema = tBuildTSchema(pSW->pSchema, pSW->nCols, pSW->version))) goto _err;
30,525✔
986

987
  if (pSW->pRsma) {
30,525✔
988
    if (!(pRSchema->funcIds = taosMemoryCalloc(pSW->nCols, sizeof(func_id_t)))) goto _err;
30,525✔
989
    memcpy(pRSchema->funcIds, pSW->pRsma->funcIds, pSW->nCols * sizeof(func_id_t));
30,525✔
990

991
    pRSchema->tbType = pSW->pRsma->tbType;
30,525✔
992
    pRSchema->tbUid = uid;
30,525✔
993
    tstrncpy(pRSchema->tbName, pSW->pRsma->tbName, TSDB_TABLE_NAME_LEN);
30,525✔
994
    pRSchema->interval[0] = pSW->pRsma->interval[0];
30,525✔
995
    pRSchema->interval[1] = pSW->pRsma->interval[1];
30,525✔
996
  }
997

998
_exit:
30,525✔
999
  tDeleteSchemaWrapper(pSW);
1000
  return pRSchema;
30,525✔
1001
_err:
×
1002
  tDeleteSchemaWrapper(pSW);
1003
  tFreeSRSchema(&pRSchema);
1004
  return NULL;
×
1005
}
1006

1007
int32_t metaGetTbTSchemaNotNull(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock, STSchema **ppTSchema) {
1,368,734✔
1008
  *ppTSchema = metaGetTbTSchema(pMeta, uid, sver, lock);
1,368,734✔
1009
  if (*ppTSchema == NULL) {
1,368,734✔
1010
    return terrno;
×
1011
  }
1012
  return TSDB_CODE_SUCCESS;
1,368,734✔
1013
}
1014

1015
int32_t metaGetTbTSchemaMaybeNull(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock, STSchema **ppTSchema) {
321,954,767✔
1016
  *ppTSchema = metaGetTbTSchema(pMeta, uid, sver, lock);
321,954,767✔
1017
  if (*ppTSchema == NULL && terrno == TSDB_CODE_OUT_OF_MEMORY) {
321,744,604✔
1018
    return terrno;
×
1019
  }
1020
  return TSDB_CODE_SUCCESS;
321,921,426✔
1021
}
1022

1023
int32_t metaGetTbTSchemaEx(SMeta *pMeta, tb_uid_t suid, tb_uid_t uid, int32_t sver, STSchema **ppTSchema) {
81,394,337✔
1024
  int32_t code = 0;
81,394,337✔
1025
  int32_t lino;
1026

1027
  void     *pData = NULL;
81,394,337✔
1028
  int       nData = 0;
81,448,560✔
1029
  SSkmDbKey skmDbKey;
81,464,744✔
1030
  if (sver <= 0) {
81,447,849✔
1031
    SMetaInfo info;
41,096,728✔
1032
    if (metaGetInfo(pMeta, suid ? suid : uid, &info, NULL) == 0) {
41,103,342✔
1033
      sver = info.skmVer;
41,075,306✔
1034
    } else {
1035
      TBC *pSkmDbC = NULL;
29,525✔
1036
      int  c;
40,620✔
1037

1038
      skmDbKey.uid = suid ? suid : uid;
40,620✔
1039
      skmDbKey.sver = INT32_MAX;
40,620✔
1040

1041
      code = tdbTbcOpen(pMeta->pSkmDb, &pSkmDbC, NULL);
40,620✔
1042
      TSDB_CHECK_CODE(code, lino, _exit);
40,620✔
1043
      metaRLock(pMeta);
40,620✔
1044

1045
      if (tdbTbcMoveTo(pSkmDbC, &skmDbKey, sizeof(skmDbKey), &c) < 0) {
40,620✔
1046
        metaULock(pMeta);
×
1047
        tdbTbcClose(pSkmDbC);
×
1048
        code = TSDB_CODE_NOT_FOUND;
×
1049
        goto _exit;
×
1050
      }
1051

1052
      if (c == 0) {
40,250✔
1053
        metaULock(pMeta);
×
1054
        tdbTbcClose(pSkmDbC);
×
1055
        code = TSDB_CODE_FAILED;
×
1056
        metaError("meta/query: incorrect c: %" PRId32 ".", c);
×
1057
        goto _exit;
×
1058
      }
1059

1060
      if (c < 0) {
40,250✔
1061
        int32_t ret = tdbTbcMoveToPrev(pSkmDbC);
×
1062
      }
1063

1064
      const void *pKey = NULL;
40,250✔
1065
      int32_t     nKey = 0;
40,250✔
1066
      int32_t     ret = tdbTbcGet(pSkmDbC, &pKey, &nKey, NULL, NULL);
40,620✔
1067

1068
      if (ret != 0 || ((SSkmDbKey *)pKey)->uid != skmDbKey.uid) {
40,990✔
1069
        metaULock(pMeta);
370✔
1070
        tdbTbcClose(pSkmDbC);
×
1071
        code = TSDB_CODE_NOT_FOUND;
×
1072
        goto _exit;
×
1073
      }
1074

1075
      sver = ((SSkmDbKey *)pKey)->sver;
40,620✔
1076

1077
      metaULock(pMeta);
40,250✔
1078
      tdbTbcClose(pSkmDbC);
40,990✔
1079
    }
1080
  }
1081

1082
  if (!(sver > 0)) {
81,401,436✔
1083
    code = TSDB_CODE_NOT_FOUND;
×
1084
    goto _exit;
×
1085
  }
1086

1087
  skmDbKey.uid = suid ? suid : uid;
81,401,436✔
1088
  skmDbKey.sver = sver;
81,401,436✔
1089
  metaRLock(pMeta);
81,401,436✔
1090
  if (tdbTbGet(pMeta->pSkmDb, &skmDbKey, sizeof(SSkmDbKey), &pData, &nData) < 0) {
81,512,971✔
1091
    metaULock(pMeta);
3,030✔
1092
    code = TSDB_CODE_NOT_FOUND;
3,030✔
1093
    goto _exit;
3,030✔
1094
  }
1095
  metaULock(pMeta);
81,464,170✔
1096

1097
  // decode
1098
  SDecoder        dc = {0};
81,523,298✔
1099
  SSchemaWrapper  schema;
81,533,413✔
1100
  SSchemaWrapper *pSchemaWrapper = &schema;
81,522,999✔
1101

1102
  tDecoderInit(&dc, pData, nData);
81,522,999✔
1103
  code = tDecodeSSchemaWrapper(&dc, pSchemaWrapper);
81,545,297✔
1104
  tDecoderClear(&dc);
81,545,297✔
1105
  tdbFree(pData);
81,496,484✔
1106
  if (TSDB_CODE_SUCCESS != code) {
81,441,164✔
1107
    taosMemoryFree(pSchemaWrapper->pSchema);
×
1108
    goto _exit;
×
1109
  }
1110

1111
  // convert
1112
  STSchema *pTSchema = tBuildTSchema(pSchemaWrapper->pSchema, pSchemaWrapper->nCols, pSchemaWrapper->version);
81,441,164✔
1113
  if (pTSchema == NULL) {
81,513,200✔
1114
    code = TSDB_CODE_OUT_OF_MEMORY;
×
1115
  }
1116

1117
  *ppTSchema = pTSchema;
81,513,200✔
1118
  taosMemoryFree(pSchemaWrapper->pSchema);
81,516,540✔
1119

1120
_exit:
81,462,298✔
1121
  return code;
81,464,315✔
1122
}
1123

1124
// N.B. Called by statusReq per second
1125
int64_t metaGetTbNum(SMeta *pMeta) {
178,005,109✔
1126
  // num of child tables (excluding normal tables , stables and others)
1127

1128
  /* int64_t num = 0; */
1129
  /* vnodeGetAllCtbNum(pMeta->pVnode, &num); */
1130

1131
  return pMeta->pVnode->config.vndStats.numOfCTables + pMeta->pVnode->config.vndStats.numOfNTables;
178,005,109✔
1132
}
1133

1134
void metaUpdTimeSeriesNum(SMeta *pMeta) {
80,055,923✔
1135
  int64_t nCtbTimeSeries = 0;
80,055,923✔
1136
  if (vnodeGetTimeSeriesNum(pMeta->pVnode, &nCtbTimeSeries) == 0) {
80,059,443✔
1137
    atomic_store_64(&pMeta->pVnode->config.vndStats.numOfTimeSeries, nCtbTimeSeries);
80,055,828✔
1138
  }
1139
}
80,059,060✔
1140

1141
static FORCE_INLINE int64_t metaGetTimeSeriesNumImpl(SMeta *pMeta, bool forceUpd) {
1142
  // sum of (number of columns of stable -  1) * number of ctables (excluding timestamp column)
1143
  SVnodeStats *pStats = &pMeta->pVnode->config.vndStats;
273,809,725✔
1144
  if (forceUpd || pStats->numOfTimeSeries <= 0) {
273,811,956✔
1145
    metaUpdTimeSeriesNum(pMeta);
80,036,007✔
1146
  }
1147

1148
  return pStats->numOfTimeSeries + pStats->numOfNTimeSeries;
273,844,560✔
1149
}
1150

1151
// type: 1 reported timeseries
1152
int64_t metaGetTimeSeriesNum(SMeta *pMeta, int type) {
273,794,141✔
1153
  int64_t nTimeSeries = metaGetTimeSeriesNumImpl(pMeta, false);
273,818,653✔
1154
  if (type == 1) {
273,818,653✔
1155
    atomic_store_64(&pMeta->pVnode->config.vndStats.numOfReportedTimeSeries, nTimeSeries);
178,005,109✔
1156
  }
1157
  return nTimeSeries;
273,777,264✔
1158
}
1159

1160
typedef struct {
1161
  SMeta   *pMeta;
1162
  TBC     *pCur;
1163
  tb_uid_t uid;
1164
  void    *pKey;
1165
  void    *pVal;
1166
  int      kLen;
1167
  int      vLen;
1168
} SMSmaCursor;
1169

1170
SMSmaCursor *metaOpenSmaCursor(SMeta *pMeta, tb_uid_t uid) {
×
1171
  SMSmaCursor *pSmaCur = NULL;
×
1172
  SSmaIdxKey   smaIdxKey;
×
1173
  int          ret;
1174
  int          c;
×
1175

1176
  pSmaCur = (SMSmaCursor *)taosMemoryCalloc(1, sizeof(*pSmaCur));
×
1177
  if (pSmaCur == NULL) {
×
1178
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1179
    return NULL;
×
1180
  }
1181

1182
  pSmaCur->pMeta = pMeta;
×
1183
  pSmaCur->uid = uid;
×
1184
  metaRLock(pMeta);
×
1185

1186
  ret = tdbTbcOpen(pMeta->pSmaIdx, &pSmaCur->pCur, NULL);
×
1187
  if (ret < 0) {
×
1188
    metaULock(pMeta);
×
1189
    taosMemoryFree(pSmaCur);
×
1190
    return NULL;
×
1191
  }
1192

1193
  // move to the suid
1194
  smaIdxKey.uid = uid;
×
1195
  smaIdxKey.smaUid = INT64_MIN;
×
1196
  ret = tdbTbcMoveTo(pSmaCur->pCur, &smaIdxKey, sizeof(smaIdxKey), &c);
×
1197
  if (c > 0) {
×
1198
    ret = tdbTbcMoveToNext(pSmaCur->pCur);
×
1199
  }
1200

1201
  return pSmaCur;
×
1202
}
1203

1204
void metaCloseSmaCursor(SMSmaCursor *pSmaCur) {
×
1205
  if (pSmaCur) {
×
1206
    if (pSmaCur->pMeta) metaULock(pSmaCur->pMeta);
×
1207
    if (pSmaCur->pCur) {
×
1208
      tdbTbcClose(pSmaCur->pCur);
×
1209
      pSmaCur->pCur = NULL;
×
1210

1211
      tdbFree(pSmaCur->pKey);
×
1212
      tdbFree(pSmaCur->pVal);
×
1213
    }
1214

1215
    taosMemoryFree(pSmaCur);
×
1216
  }
1217
}
×
1218

1219
tb_uid_t metaSmaCursorNext(SMSmaCursor *pSmaCur) {
×
1220
  int         ret;
1221
  SSmaIdxKey *pSmaIdxKey;
1222

1223
  ret = tdbTbcNext(pSmaCur->pCur, &pSmaCur->pKey, &pSmaCur->kLen, &pSmaCur->pVal, &pSmaCur->vLen);
×
1224
  if (ret < 0) {
×
1225
    return 0;
×
1226
  }
1227

1228
  pSmaIdxKey = pSmaCur->pKey;
×
1229
  if (pSmaIdxKey->uid > pSmaCur->uid) {
×
1230
    return 0;
×
1231
  }
1232

1233
  return pSmaIdxKey->uid;
×
1234
}
1235

1236
STSmaWrapper *metaGetSmaInfoByTable(SMeta *pMeta, tb_uid_t uid, bool deepCopy) {
×
1237
  STSmaWrapper *pSW = NULL;
×
1238
  SArray       *pSmaIds = NULL;
×
1239

1240
  if (!(pSmaIds = metaGetSmaIdsByTable(pMeta, uid))) {
×
1241
    return NULL;
×
1242
  }
1243

1244
  pSW = taosMemoryCalloc(1, sizeof(*pSW));
×
1245
  if (!pSW) {
×
1246
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1247
    goto _err;
×
1248
  }
1249

1250
  pSW->number = taosArrayGetSize(pSmaIds);
×
1251
  pSW->tSma = taosMemoryCalloc(pSW->number, sizeof(STSma));
×
1252

1253
  if (!pSW->tSma) {
×
1254
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1255
    goto _err;
×
1256
  }
1257

1258
  SMetaReader mr = {0};
×
1259
  metaReaderDoInit(&mr, pMeta, META_READER_LOCK);
×
1260
  int64_t smaId;
1261
  int     smaIdx = 0;
×
1262
  STSma  *pTSma = NULL;
×
1263
  for (int i = 0; i < pSW->number; ++i) {
×
1264
    smaId = *(tb_uid_t *)taosArrayGet(pSmaIds, i);
×
1265
    if (metaReaderGetTableEntryByUid(&mr, smaId) < 0) {
×
1266
      tDecoderClear(&mr.coder);
×
1267
      metaWarn("vgId:%d, no entry for tbId:%" PRIi64 ", smaId:%" PRIi64, TD_VID(pMeta->pVnode), uid, smaId);
×
1268
      continue;
×
1269
    }
1270
    tDecoderClear(&mr.coder);
×
1271
    pTSma = pSW->tSma + smaIdx;
×
1272
    memcpy(pTSma, mr.me.smaEntry.tsma, sizeof(STSma));
×
1273
    if (deepCopy) {
×
1274
      if (pTSma->exprLen > 0) {
×
1275
        if (!(pTSma->expr = taosMemoryCalloc(1, pTSma->exprLen))) {
×
1276
          terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1277
          goto _err;
×
1278
        }
1279
        memcpy((void *)pTSma->expr, mr.me.smaEntry.tsma->expr, pTSma->exprLen);
×
1280
      }
1281
      if (pTSma->tagsFilterLen > 0) {
×
1282
        if (!(pTSma->tagsFilter = taosMemoryCalloc(1, pTSma->tagsFilterLen))) {
×
1283
          terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1284
          goto _err;
×
1285
        }
1286
      }
1287
      memcpy((void *)pTSma->tagsFilter, mr.me.smaEntry.tsma->tagsFilter, pTSma->tagsFilterLen);
×
1288
    } else {
1289
      pTSma->exprLen = 0;
×
1290
      pTSma->expr = NULL;
×
1291
      pTSma->tagsFilterLen = 0;
×
1292
      pTSma->tagsFilter = NULL;
×
1293
    }
1294

1295
    ++smaIdx;
×
1296
  }
1297

1298
  if (smaIdx <= 0) goto _err;
×
1299
  pSW->number = smaIdx;
×
1300

1301
  metaReaderClear(&mr);
×
1302
  taosArrayDestroy(pSmaIds);
×
1303
  return pSW;
×
1304
_err:
×
1305
  metaReaderClear(&mr);
×
1306
  taosArrayDestroy(pSmaIds);
×
1307
  pSW = tFreeTSmaWrapper(pSW, deepCopy);
×
1308
  return NULL;
×
1309
}
1310

1311
STSma *metaGetSmaInfoByIndex(SMeta *pMeta, int64_t indexUid) {
×
1312
  STSma      *pTSma = NULL;
×
1313
  SMetaReader mr = {0};
×
1314
  metaReaderDoInit(&mr, pMeta, META_READER_LOCK);
×
1315
  if (metaReaderGetTableEntryByUid(&mr, indexUid) < 0) {
×
1316
    metaWarn("vgId:%d, failed to get table entry for smaId:%" PRIi64, TD_VID(pMeta->pVnode), indexUid);
×
1317
    metaReaderClear(&mr);
×
1318
    return NULL;
×
1319
  }
1320
  pTSma = (STSma *)taosMemoryMalloc(sizeof(STSma));
×
1321
  if (!pTSma) {
×
1322
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1323
    metaReaderClear(&mr);
×
1324
    return NULL;
×
1325
  }
1326

1327
  memcpy(pTSma, mr.me.smaEntry.tsma, sizeof(STSma));
×
1328

1329
  metaReaderClear(&mr);
×
1330
  return pTSma;
×
1331
}
1332

1333
SArray *metaGetSmaIdsByTable(SMeta *pMeta, tb_uid_t uid) {
×
1334
  SArray     *pUids = NULL;
×
1335
  SSmaIdxKey *pSmaIdxKey = NULL;
×
1336

1337
  SMSmaCursor *pCur = metaOpenSmaCursor(pMeta, uid);
×
1338
  if (!pCur) {
×
1339
    return NULL;
×
1340
  }
1341

1342
  while (1) {
×
1343
    tb_uid_t id = metaSmaCursorNext(pCur);
×
1344
    if (id == 0) {
×
1345
      break;
×
1346
    }
1347

1348
    if (!pUids) {
×
1349
      pUids = taosArrayInit(16, sizeof(tb_uid_t));
×
1350
      if (!pUids) {
×
1351
        terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1352
        metaCloseSmaCursor(pCur);
×
1353
        return NULL;
×
1354
      }
1355
    }
1356

1357
    pSmaIdxKey = (SSmaIdxKey *)pCur->pKey;
×
1358

1359
    if (!taosArrayPush(pUids, &pSmaIdxKey->smaUid)) {
×
1360
      terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1361
      metaCloseSmaCursor(pCur);
×
1362
      taosArrayDestroy(pUids);
×
1363
      return NULL;
×
1364
    }
1365
  }
1366

1367
  metaCloseSmaCursor(pCur);
×
1368
  return pUids;
×
1369
}
1370

1371
SArray *metaGetSmaTbUids(SMeta *pMeta) {
×
1372
  SArray     *pUids = NULL;
×
1373
  SSmaIdxKey *pSmaIdxKey = NULL;
×
1374
  tb_uid_t    lastUid = 0;
×
1375

1376
  SMSmaCursor *pCur = metaOpenSmaCursor(pMeta, 0);
×
1377
  if (!pCur) {
×
1378
    return NULL;
×
1379
  }
1380

1381
  while (1) {
×
1382
    tb_uid_t uid = metaSmaCursorNext(pCur);
×
1383
    if (uid == 0) {
×
1384
      break;
×
1385
    }
1386

1387
    if (lastUid == uid) {
×
1388
      continue;
×
1389
    }
1390

1391
    lastUid = uid;
×
1392

1393
    if (!pUids) {
×
1394
      pUids = taosArrayInit(16, sizeof(tb_uid_t));
×
1395
      if (!pUids) {
×
1396
        terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1397
        metaCloseSmaCursor(pCur);
×
1398
        return NULL;
×
1399
      }
1400
    }
1401

1402
    if (!taosArrayPush(pUids, &uid)) {
×
1403
      terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1404
      metaCloseSmaCursor(pCur);
×
1405
      taosArrayDestroy(pUids);
×
1406
      return NULL;
×
1407
    }
1408
  }
1409

1410
  metaCloseSmaCursor(pCur);
×
1411
  return pUids;
×
1412
}
1413

1414
#endif
1415

1416
const void *metaGetTableTagVal(const void *pTag, int16_t type, STagVal *val) {
1,174,929,713✔
1417
  STag *tag = (STag *)pTag;
1,174,929,713✔
1418
  if (type == TSDB_DATA_TYPE_JSON) {
1,174,929,713✔
1419
    return tag;
1,687,423✔
1420
  }
1421
  bool find = tTagGet(tag, val);
1,173,242,290✔
1422

1423
  if (!find) {
1,173,558,913✔
1424
    return NULL;
8,389,920✔
1425
  }
1426

1427
  return val;
1,165,168,993✔
1428
}
1429

1430
typedef struct {
1431
  SMeta   *pMeta;
1432
  TBC     *pCur;
1433
  tb_uid_t suid;
1434
  int16_t  cid;
1435
  int16_t  type;
1436
  void    *pKey;
1437
  void    *pVal;
1438
  int32_t  kLen;
1439
  int32_t  vLen;
1440
} SIdxCursor;
1441

1442
int32_t metaFilterCreateTime(void *pVnode, SMetaFltParam *arg, SArray *pUids) {
2,046✔
1443
  SMeta         *pMeta = ((SVnode *)pVnode)->pMeta;
2,046✔
1444
  SMetaFltParam *param = arg;
2,046✔
1445
  int32_t        ret = 0;
2,046✔
1446

1447
  SIdxCursor *pCursor = NULL;
2,046✔
1448
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
2,046✔
1449
  if (pCursor == NULL) {
2,046✔
1450
    return terrno;
×
1451
  }
1452
  pCursor->pMeta = pMeta;
2,046✔
1453
  pCursor->suid = param->suid;
2,046✔
1454
  pCursor->cid = param->cid;
2,046✔
1455
  pCursor->type = param->type;
2,046✔
1456

1457
  metaRLock(pMeta);
2,046✔
1458
  ret = tdbTbcOpen(pMeta->pBtimeIdx, &pCursor->pCur, NULL);
2,046✔
1459
  if (ret != 0) {
2,046✔
1460
    goto END;
×
1461
  }
1462
  int64_t uidLimit = param->reverse ? INT64_MAX : 0;
2,046✔
1463

1464
  SBtimeIdxKey  btimeKey = {.btime = *(int64_t *)(param->val), .uid = uidLimit};
2,046✔
1465
  SBtimeIdxKey *pBtimeKey = &btimeKey;
2,046✔
1466

1467
  int cmp = 0;
2,046✔
1468
  if (tdbTbcMoveTo(pCursor->pCur, &btimeKey, sizeof(btimeKey), &cmp) < 0) {
2,046✔
1469
    goto END;
×
1470
  }
1471

1472
  int32_t valid = 0;
2,046✔
1473
  int32_t count = 0;
2,046✔
1474

1475
  static const int8_t TRY_ERROR_LIMIT = 1;
1476
  do {
7,701,694✔
1477
    void   *entryKey = NULL;
7,703,740✔
1478
    int32_t nEntryKey = -1;
7,703,547✔
1479
    valid = tdbTbcGet(pCursor->pCur, (const void **)&entryKey, &nEntryKey, NULL, NULL);
7,703,740✔
1480
    if (valid < 0) break;
7,720,531✔
1481

1482
    SBtimeIdxKey *p = entryKey;
7,718,485✔
1483
    if (count > TRY_ERROR_LIMIT) break;
7,718,485✔
1484

1485
    terrno = TSDB_CODE_SUCCESS;
7,718,485✔
1486
    int32_t cmp = (*param->filterFunc)((void *)&p->btime, (void *)&pBtimeKey->btime, param->type);
7,716,169✔
1487
    if (terrno != TSDB_CODE_SUCCESS) {
7,710,572✔
1488
      ret = terrno;
×
1489
      break;
×
1490
    }
1491
    if (cmp == 0) {
7,715,011✔
1492
      if (taosArrayPush(pUids, &p->uid) == NULL) {
15,431,759✔
1493
        ret = terrno;
×
1494
        break;
×
1495
      }
1496
    } else {
1497
      if (param->equal == true) {
×
1498
        if (count > TRY_ERROR_LIMIT) break;
×
1499
        count++;
×
1500
      }
1501
    }
1502
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
7,716,748✔
1503
    if (valid < 0) break;
7,701,308✔
1504
  } while (1);
1505

1506
END:
2,046✔
1507
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
2,046✔
1508
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
2,046✔
1509
  taosMemoryFree(pCursor);
2,046✔
1510
  return ret;
2,046✔
1511
}
1512

1513
int32_t metaFilterTableName(void *pVnode, SMetaFltParam *arg, SArray *pUids) {
×
1514
  SMeta         *pMeta = ((SVnode *)pVnode)->pMeta;
×
1515
  SMetaFltParam *param = arg;
×
1516
  int32_t        ret = 0;
×
1517
  char          *buf = NULL;
×
1518

1519
  STagIdxKey *pKey = NULL;
×
1520
  int32_t     nKey = 0;
×
1521

1522
  SIdxCursor *pCursor = NULL;
×
1523
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
×
1524
  if (pCursor == NULL) {
×
1525
    return terrno;
×
1526
  }
1527
  pCursor->pMeta = pMeta;
×
1528
  pCursor->suid = param->suid;
×
1529
  pCursor->cid = param->cid;
×
1530
  pCursor->type = param->type;
×
1531

1532
  char *pName = param->val;
×
1533

1534
  metaRLock(pMeta);
×
1535
  ret = tdbTbcOpen(pMeta->pNameIdx, &pCursor->pCur, NULL);
×
1536
  if (ret != 0) {
×
1537
    goto END;
×
1538
  }
1539

1540
  int cmp = 0;
×
1541
  if (tdbTbcMoveTo(pCursor->pCur, pName, strlen(pName) + 1, &cmp) < 0) {
×
1542
    goto END;
×
1543
  }
1544
  int32_t valid = 0;
×
1545
  int32_t count = 0;
×
1546

1547
  int32_t TRY_ERROR_LIMIT = 1;
×
1548
  do {
×
1549
    void   *pEntryKey = NULL, *pEntryVal = NULL;
×
1550
    int32_t nEntryKey = -1, nEntryVal = 0;
×
1551
    valid = tdbTbcGet(pCursor->pCur, (const void **)pEntryKey, &nEntryKey, (const void **)&pEntryVal, &nEntryVal);
×
1552
    if (valid < 0) break;
×
1553

1554
    if (count > TRY_ERROR_LIMIT) break;
×
1555

1556
    char *pTableKey = (char *)pEntryKey;
×
1557
    terrno = TSDB_CODE_SUCCESS;
×
1558
    cmp = (*param->filterFunc)(pTableKey, pName, pCursor->type);
×
1559
    if (terrno != TSDB_CODE_SUCCESS) {
×
1560
      ret = terrno;
×
1561
      goto END;
×
1562
    }
1563
    if (cmp == 0) {
×
1564
      tb_uid_t tuid = *(tb_uid_t *)pEntryVal;
×
1565
      if (taosArrayPush(pUids, &tuid) == NULL) {
×
1566
        ret = terrno;
×
1567
        goto END;
×
1568
      }
1569
    } else {
1570
      if (param->equal == true) {
×
1571
        if (count > TRY_ERROR_LIMIT) break;
×
1572
        count++;
×
1573
      }
1574
    }
1575
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
×
1576
    if (valid < 0) {
×
1577
      break;
×
1578
    }
1579
  } while (1);
1580

1581
END:
×
1582
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
×
1583
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
×
1584
  taosMemoryFree(buf);
×
1585
  taosMemoryFree(pKey);
×
1586

1587
  taosMemoryFree(pCursor);
×
1588

1589
  return ret;
×
1590
}
1591
int32_t metaFilterTtl(void *pVnode, SMetaFltParam *arg, SArray *pUids) {
×
1592
  SMeta         *pMeta = ((SVnode *)pVnode)->pMeta;
×
1593
  SMetaFltParam *param = arg;
×
1594
  int32_t        ret = 0;
×
1595
  char          *buf = NULL;
×
1596

1597
  STtlIdxKey *pKey = NULL;
×
1598
  int32_t     nKey = 0;
×
1599

1600
  SIdxCursor *pCursor = NULL;
×
1601
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
×
1602
  if (pCursor == NULL) {
×
1603
    return terrno;
×
1604
  }
1605
  pCursor->pMeta = pMeta;
×
1606
  pCursor->suid = param->suid;
×
1607
  pCursor->cid = param->cid;
×
1608
  pCursor->type = param->type;
×
1609

1610
  metaRLock(pMeta);
×
1611
  // ret = tdbTbcOpen(pMeta->pTtlIdx, &pCursor->pCur, NULL);
1612

1613
END:
×
1614
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
×
1615
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
×
1616
  taosMemoryFree(buf);
×
1617
  taosMemoryFree(pKey);
×
1618

1619
  taosMemoryFree(pCursor);
×
1620

1621
  return ret;
×
1622
  // impl later
1623
  return 0;
1624
}
1625
int32_t metaFilterTableIds(void *pVnode, SMetaFltParam *arg, SArray *pUids) {
3,668,459✔
1626
  SMeta         *pMeta = ((SVnode *)pVnode)->pMeta;
3,668,459✔
1627
  SMetaFltParam *param = arg;
3,672,035✔
1628

1629
  SMetaEntry oStbEntry = {0};
3,672,035✔
1630
  int32_t    code = 0;
3,661,201✔
1631
  char      *buf = NULL;
3,661,201✔
1632
  void      *pData = NULL;
3,661,201✔
1633
  int        nData = 0;
3,656,762✔
1634

1635
  SDecoder    dc = {0};
3,656,897✔
1636
  STbDbKey    tbDbKey = {0};
3,655,269✔
1637
  STagIdxKey *pKey = NULL;
3,647,590✔
1638
  int32_t     nKey = 0;
3,667,333✔
1639

1640
  SIdxCursor *pCursor = NULL;
3,670,545✔
1641
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
3,670,545✔
1642
  if (!pCursor) {
3,656,153✔
1643
    return terrno;
×
1644
  }
1645
  pCursor->pMeta = pMeta;
3,656,153✔
1646
  pCursor->suid = param->suid;
3,666,759✔
1647
  pCursor->cid = param->cid;
3,657,153✔
1648
  pCursor->type = param->type;
3,671,857✔
1649

1650
  metaRLock(pMeta);
3,658,062✔
1651

1652
  TAOS_CHECK_GOTO(tdbTbGet(pMeta->pUidIdx, &param->suid, sizeof(tb_uid_t), &pData, &nData), NULL, END);
3,663,193✔
1653

1654
  tbDbKey.uid = param->suid;
3,675,120✔
1655
  tbDbKey.version = ((SUidIdxVal *)pData)[0].version;
3,675,120✔
1656

1657
  TAOS_CHECK_GOTO(tdbTbGet(pMeta->pTbDb, &tbDbKey, sizeof(tbDbKey), &pData, &nData), NULL, END);
3,668,643✔
1658

1659
  tDecoderInit(&dc, pData, nData);
3,675,112✔
1660

1661
  code = metaDecodeEntry(&dc, &oStbEntry);
3,674,270✔
1662
  if (code) {
3,669,898✔
1663
    tDecoderClear(&dc);
×
1664
    goto END;
×
1665
  }
1666

1667
  if (oStbEntry.stbEntry.schemaTag.pSchema == NULL || oStbEntry.stbEntry.schemaTag.pSchema == NULL) {
3,669,898✔
1668
    TAOS_CHECK_GOTO(TSDB_CODE_INVALID_PARA, NULL, END);
×
1669
  }
1670

1671
  code = TSDB_CODE_INVALID_PARA;
3,669,898✔
1672

1673
  for (int i = 0; i < oStbEntry.stbEntry.schemaTag.nCols; i++) {
4,891,448✔
1674
    SSchema *schema = oStbEntry.stbEntry.schemaTag.pSchema + i;
4,886,011✔
1675
    if (IS_IDX_ON(schema)) {
4,889,228✔
1676
      if (schema->colId == param->cid && param->type == schema->type) {
4,852,877✔
1677
        code = 0;
3,664,753✔
1678
        break;
3,664,753✔
1679
      }
1680
    }
1681
  }
1682

1683
  TAOS_CHECK_GOTO(code, NULL, END);
3,670,190✔
1684

1685
  code = tdbTbcOpen(pMeta->pTagIdx, &pCursor->pCur, NULL);
3,665,158✔
1686
  if (code != 0) {
3,665,440✔
1687
    TAOS_CHECK_GOTO(terrno, NULL, END);
×
1688
  }
1689

1690
  int32_t maxSize = 0;
3,665,440✔
1691
  int32_t nTagData = 0;
3,665,003✔
1692
  void   *tagData = NULL;
3,665,003✔
1693

1694
  if (param->val == NULL) {
3,665,003✔
1695
    metaError("vgId:%d, failed to filter NULL data", TD_VID(pMeta->pVnode));
×
1696
    goto END;
×
1697
  } else {
1698
    if (IS_VAR_DATA_TYPE(param->type)) {
3,658,388✔
1699
      tagData = varDataVal(param->val);
545,679✔
1700
      nTagData = varDataLen(param->val);
558,116✔
1701

1702
      if (param->type == TSDB_DATA_TYPE_NCHAR) {
558,116✔
1703
        maxSize = 4 * nTagData + 1;
74,619✔
1704
        buf = taosMemoryCalloc(1, maxSize);
74,619✔
1705
        if (buf == NULL) {
74,619✔
1706
          TAOS_CHECK_GOTO(terrno, NULL, END);
×
1707
        }
1708

1709
        if (false == taosMbsToUcs4(tagData, nTagData, (TdUcs4 *)buf, maxSize, &maxSize, NULL)) {
74,619✔
1710
          TAOS_CHECK_GOTO(terrno, NULL, END);
×
1711
        }
1712

1713
        tagData = buf;
74,619✔
1714
        nTagData = maxSize;
74,619✔
1715
      }
1716
    } else {
1717
      tagData = param->val;
3,106,887✔
1718
      nTagData = tDataTypes[param->type].bytes;
3,106,236✔
1719
    }
1720
  }
1721

1722
  TAOS_CHECK_GOTO(metaCreateTagIdxKey(pCursor->suid, pCursor->cid, tagData, nTagData, pCursor->type,
3,664,161✔
1723
                                      param->reverse ? INT64_MAX : INT64_MIN, &pKey, &nKey),
1724
                  NULL, END);
1725

1726
  int cmp = 0;
3,664,161✔
1727
  TAOS_CHECK_GOTO(tdbTbcMoveTo(pCursor->pCur, pKey, nKey, &cmp), 0, END);
3,665,295✔
1728

1729
  int     count = 0;
3,670,630✔
1730
  int32_t valid = 0;
3,670,630✔
1731
  bool    found = false;
3,670,630✔
1732

1733
  static const int8_t TRY_ERROR_LIMIT = 1;
1734

1735
  /// src:   [[suid, cid1, type1]....[suid, cid2, type2]....[suid, cid3, type3]...]
1736
  /// target:                        [suid, cid2, type2]
1737
  int diffCidCount = 0;
3,670,630✔
1738
  do {
35,332,928✔
1739
    void   *entryKey = NULL, *entryVal = NULL;
39,003,558✔
1740
    int32_t nEntryKey, nEntryVal;
38,996,677✔
1741

1742
    valid = tdbTbcGet(pCursor->pCur, (const void **)&entryKey, &nEntryKey, (const void **)&entryVal, &nEntryVal);
38,998,764✔
1743
    if (valid < 0) {
39,033,237✔
1744
      break;
1,832,088✔
1745
    }
1746
    if (count > TRY_ERROR_LIMIT) {
37,201,149✔
1747
      break;
1,188,568✔
1748
    }
1749

1750
    STagIdxKey *p = entryKey;
36,012,581✔
1751
    if (p == NULL) break;
36,012,581✔
1752

1753
    if (p->type != pCursor->type || p->suid != pCursor->suid || p->cid != pCursor->cid) {
36,012,581✔
1754
      if (found == true) break;  //
1,011,521✔
1755
      if (diffCidCount > TRY_ERROR_LIMIT) break;
360,423✔
1756
      diffCidCount++;
360,423✔
1757
      count++;
360,423✔
1758
      valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
360,423✔
1759
      if (valid < 0) {
360,423✔
1760
        code = valid;
×
1761
        break;
×
1762
      } else {
1763
        continue;
360,423✔
1764
      }
1765
    }
1766

1767
    terrno = TSDB_CODE_SUCCESS;
34,993,539✔
1768
    int32_t cmp = (*param->filterFunc)(p->data, pKey->data, pKey->type);
34,996,020✔
1769
    if (terrno != TSDB_CODE_SUCCESS) {
34,983,469✔
1770
      TAOS_CHECK_GOTO(terrno, NULL, END);
×
1771
      break;
×
1772
    }
1773
    if (cmp == 0) {
34,990,138✔
1774
      // match
1775
      tb_uid_t tuid = 0;
31,093,348✔
1776
      if (IS_VAR_DATA_TYPE(pKey->type)) {
31,092,055✔
1777
        tuid = *(tb_uid_t *)(p->data + varDataTLen(p->data));
420,674✔
1778
      } else {
1779
        tuid = *(tb_uid_t *)(p->data + tDataTypes[pCursor->type].bytes);
30,670,919✔
1780
      }
1781
      if (taosArrayPush(pUids, &tuid) == NULL) {
31,098,236✔
1782
        TAOS_CHECK_GOTO(terrno, NULL, END);
×
1783
      }
1784
      found = true;
31,093,571✔
1785
    } else {
1786
      if (param->equal == true) {
3,896,790✔
1787
        if (count > TRY_ERROR_LIMIT) break;
2,767,189✔
1788
        count++;
2,767,189✔
1789
      }
1790
    }
1791
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
34,991,342✔
1792
    if (valid < 0) {
34,974,456✔
1793
      code = valid;
×
1794
      break;
×
1795
    }
1796
  } while (1);
1797

1798
END:
3,668,756✔
1799
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
3,676,796✔
1800
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
3,676,375✔
1801
  if (oStbEntry.pBuf) taosMemoryFree(oStbEntry.pBuf);
3,671,506✔
1802
  tDecoderClear(&dc);
3,671,506✔
1803
  tdbFree(pData);
3,676,083✔
1804

1805
  taosMemoryFree(buf);
3,673,483✔
1806
  taosMemoryFree(pKey);
3,676,796✔
1807

1808
  taosMemoryFree(pCursor);
3,669,675✔
1809

1810
  return code;
3,673,884✔
1811
}
1812

1813
static int32_t metaGetTableTagByUidVersion(SMeta *pMeta, int64_t suid, int64_t uid, int64_t version, void** tag) {
56,222,760✔
1814
  void   *val = NULL;
56,222,760✔
1815
  int32_t len = 0;
56,223,356✔
1816
  int ret = 0;
56,226,231✔
1817

1818
  if (version != -1) {
56,226,231✔
1819
    SMetaReader mr = {0};
33,932✔
1820
    metaReaderDoInit(&mr, pMeta, META_READER_NOLOCK);
33,932✔
1821
    if (getUidVersion(&mr, &version, uid) != 0) {
33,932✔
1822
      version = -1;
×
1823
    }
1824

1825
    ret = metaGetTableEntryByVersion(&mr, version, uid);
33,932✔
1826
    if (ret == 0) {
33,932✔
1827
      val = mr.me.ctbEntry.pTags;
33,932✔
1828
      len = ((STag *)(mr.me.ctbEntry.pTags))->len;
33,932✔
1829
      *tag = taosMemoryMalloc(len);
33,932✔
1830
      if (*tag) {
33,932✔
1831
        memcpy(*tag, val, len);
33,932✔
1832
      } else {
1833
        ret = terrno;
×
1834
      }
1835
    }
1836
    metaReaderClear(&mr);
33,932✔
1837
  } else {
1838
    SCtbIdxKey ctbIdxKey = {.suid = suid, .uid = uid};
56,192,299✔
1839
    ret = tdbTbGet(pMeta->pCtbIdx, &ctbIdxKey, sizeof(SCtbIdxKey), &val, &len);
56,192,720✔
1840
    if (ret == 0) {
56,188,133✔
1841
      *tag = taosMemoryMalloc(len);
56,181,841✔
1842
      if (*tag) {
56,180,521✔
1843
        memcpy(*tag, val, len);
56,179,258✔
1844
      } else {
1845
        ret = terrno;
×
1846
      }
1847
      tdbFree(val);
56,181,008✔
1848
    }
1849
  }
1850

1851
  return ret;
56,225,133✔
1852
}
1853

1854
int32_t metaGetTableTagsByUidsVersion(void *pVnode, int64_t suid, SArray *uidList, int64_t version) {
8,122,944✔
1855
  SMeta        *pMeta = ((SVnode *)pVnode)->pMeta;
8,122,944✔
1856
  const int32_t LIMIT = 128;
8,123,696✔
1857

1858
  int32_t isLock = false;
8,123,696✔
1859
  int32_t sz = uidList ? taosArrayGetSize(uidList) : 0;
8,123,696✔
1860
  for (int i = 0; i < sz; i++) {
64,345,765✔
1861
    STUidTagInfo *p = taosArrayGet(uidList, i);
56,221,138✔
1862
    if (p->pTagVal != NULL) continue;
56,224,011✔
1863

1864
    if (i % LIMIT == 0) {
56,225,652✔
1865
      if (isLock) metaULock(pMeta);
8,123,337✔
1866

1867
      metaRLock(pMeta);
8,123,337✔
1868
      isLock = true;
8,122,741✔
1869
    }
1870
    int32_t code = metaGetTableTagByUidVersion(pMeta, suid, p->uid, version, &p->pTagVal);
56,225,056✔
1871
    if (code != 0) {
56,218,522✔
1872
      metaError("vgId:%d, failed to table tags, code:%d, suid: %" PRId64 ", uid: %" PRId64 " version: %" PRId64, TD_VID(pMeta->pVnode), code, suid, p->uid, version);
7,376✔
1873
    }
1874
  }
1875
  if (isLock) metaULock(pMeta);
8,124,627✔
1876
  return 0;
8,123,454✔
1877
}
1878

1879
int32_t metaGetTableTags(void *pVnode, uint64_t suid, SArray *pUidTagInfo) {
13,877,741✔
1880
  SMCtbCursor *pCur = metaOpenCtbCursor(pVnode, suid, 1);
13,877,741✔
1881
  if (!pCur) {
13,865,469✔
1882
    TAOS_RETURN(TSDB_CODE_OUT_OF_MEMORY);
×
1883
  }
1884

1885
  while (1) {
174,425,569✔
1886
    tb_uid_t uid = metaCtbCursorNext(pCur);
188,291,038✔
1887
    if (uid == 0) {
188,308,761✔
1888
      metaDebug("got uid 0 and uidTagSize:%d", (int32_t)taosArrayGetSize(pUidTagInfo));
13,882,786✔
1889
      break;
13,883,350✔
1890
    }
1891

1892
    STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
174,425,975✔
1893
    info.pTagVal = taosMemoryMalloc(pCur->vLen);
174,417,297✔
1894
    if (!info.pTagVal) {
174,355,621✔
1895
      metaCloseCtbCursor(pCur);
×
1896
      return terrno;
×
1897
    }
1898
    memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
174,355,621✔
1899
    if (taosArrayPush(pUidTagInfo, &info) == NULL) {
174,422,421✔
1900
      taosMemoryFreeClear(info.pTagVal);
×
1901
      metaCloseCtbCursor(pCur);
×
1902
      return terrno;
×
1903
    }
1904
  }
1905
  metaCloseCtbCursor(pCur);
13,883,350✔
1906
  return TSDB_CODE_SUCCESS;
13,876,839✔
1907
}
1908

1909
int32_t metaFlagCache(SVnode *pVnode) {
×
1910
  SMStbCursor *pCur = metaOpenStbCursor(pVnode->pMeta, 0);
×
1911
  if (!pCur) {
×
1912
    return terrno;
×
1913
  }
1914

1915
  SArray *suids = NULL;
×
1916
  while (1) {
×
1917
    tb_uid_t id = metaStbCursorNext(pCur);
×
1918
    if (id == 0) {
×
1919
      break;
×
1920
    }
1921

1922
    if (!suids) {
×
1923
      suids = taosArrayInit(8, sizeof(tb_uid_t));
×
1924
      if (!suids) {
×
1925
        return terrno;
×
1926
      }
1927
    }
1928

1929
    if (taosArrayPush(suids, &id) == NULL) {
×
1930
      taosArrayDestroy(suids);
×
1931
      return terrno;
×
1932
    }
1933
  }
1934

1935
  metaCloseStbCursor(pCur);
×
1936

1937
  for (int idx = 0; suids && idx < TARRAY_SIZE(suids); ++idx) {
×
1938
    tb_uid_t id = ((tb_uid_t *)TARRAY_DATA(suids))[idx];
×
1939
    STsdb   *pTsdb = pVnode->pTsdb;
×
1940
    SMeta   *pMeta = pVnode->pMeta;
×
1941
    SArray  *uids = NULL;
×
1942

1943
    int32_t code = metaGetChildUidsOfSuperTable(pMeta, id, &uids);
×
1944
    if (code) {
×
1945
      metaError("vgId:%d, failed to get subtables, suid:%" PRId64 " since %s.", TD_VID(pVnode), id, tstrerror(code));
×
1946

1947
      taosArrayDestroy(uids);
×
1948
      taosArrayDestroy(suids);
×
1949

1950
      return code;
×
1951
    }
1952

1953
    if (uids && TARRAY_SIZE(uids) > 0) {
×
1954
      STSchema *pTSchema = NULL;
×
1955

1956
      code = metaGetTbTSchemaEx(pMeta, id, id, -1, &pTSchema);
×
1957
      if (code) {
×
1958
        metaError("vgId:%d, failed to get schema, suid:%" PRId64 " since %s.", TD_VID(pVnode), id, tstrerror(code));
×
1959

1960
        taosArrayDestroy(uids);
×
1961
        taosArrayDestroy(suids);
×
1962

1963
        return code;
×
1964
      }
1965

1966
      int32_t nCol = pTSchema->numOfCols;
×
1967
      for (int32_t i = 0; i < nCol; ++i) {
×
1968
        int16_t cid = pTSchema->columns[i].colId;
×
1969
        int8_t  col_type = pTSchema->columns[i].type;
×
1970

1971
        code = tsdbCacheNewSTableColumn(pTsdb, uids, cid, col_type);
×
1972
        if (code) {
×
1973
          metaError("vgId:%d, failed to flag cache, suid:%" PRId64 " since %s.", TD_VID(pVnode), id, tstrerror(code));
×
1974

1975
          tDestroyTSchema(pTSchema);
×
1976
          taosArrayDestroy(uids);
×
1977
          taosArrayDestroy(suids);
×
1978

1979
          return code;
×
1980
        }
1981
      }
1982

1983
      tDestroyTSchema(pTSchema);
×
1984
    }
1985

1986
    taosArrayDestroy(uids);
×
1987
  }
1988

1989
  taosArrayDestroy(suids);
×
1990

1991
  return TSDB_CODE_SUCCESS;
×
1992
}
1993

1994
int32_t metaCacheGet(SMeta *pMeta, int64_t uid, SMetaInfo *pInfo);
1995

1996
int32_t metaGetInfo(SMeta *pMeta, int64_t uid, SMetaInfo *pInfo, SMetaReader *pReader) {
2,147,483,647✔
1997
  int32_t code = 0;
2,147,483,647✔
1998
  void   *pData = NULL;
2,147,483,647✔
1999
  int     nData = 0;
2,147,483,647✔
2000
  int     lock = 0;
2,147,483,647✔
2001

2002
  if (pReader && !(pReader->flags & META_READER_NOLOCK)) {
2,147,483,647✔
2003
    lock = 1;
847,760,931✔
2004
  }
2005

2006
  if (!lock) metaRLock(pMeta);
2,147,483,647✔
2007

2008
  // search cache
2009
  if (metaCacheGet(pMeta, uid, pInfo) == 0) {
2,147,483,647✔
2010
    if (!lock) metaULock(pMeta);
2,147,483,647✔
2011
    goto _exit;
2,147,483,647✔
2012
  }
2013

2014
  // search TDB
2015
  if ((code = tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData)) < 0) {
114,240,117✔
2016
    // not found
2017
    if (!lock) metaULock(pMeta);
103,395,404✔
2018
    goto _exit;
103,373,778✔
2019
  }
2020

2021
  if (!lock) metaULock(pMeta);
10,836,392✔
2022

2023
  pInfo->uid = uid;
10,836,392✔
2024
  pInfo->suid = ((SUidIdxVal *)pData)->suid;
10,836,217✔
2025
  pInfo->version = ((SUidIdxVal *)pData)->version;
10,836,940✔
2026
  pInfo->skmVer = ((SUidIdxVal *)pData)->skmVer;
10,836,217✔
2027

2028
  if (lock) {
10,837,115✔
2029
    metaULock(pReader->pMeta);
1,369,558✔
2030
    // metaReaderReleaseLock(pReader);
2031
  }
2032
  // upsert the cache
2033
  metaWLock(pMeta);
10,836,940✔
2034
  int32_t ret = metaCacheUpsert(pMeta, pInfo);
10,836,217✔
2035
  if (ret != 0) {
10,836,392✔
2036
    metaError("vgId:%d, failed to upsert cache, uid:%" PRId64, TD_VID(pMeta->pVnode), uid);
×
2037
  }
2038
  metaULock(pMeta);
10,836,392✔
2039

2040
  if (lock) {
10,836,217✔
2041
    metaRLock(pReader->pMeta);
1,368,660✔
2042
  }
2043

2044
_exit:
2,147,483,647✔
2045
  tdbFree(pData);
2,147,483,647✔
2046
  return code;
2,147,483,647✔
2047
}
2048

2049
int32_t metaGetStbStats(void *pVnode, int64_t uid, int64_t *numOfTables, int32_t *numOfCols, int8_t *flags) {
149,618,669✔
2050
  int32_t code = 0;
149,618,669✔
2051

2052
  if (!numOfTables && !numOfCols) goto _exit;
149,618,669✔
2053

2054
  SVnode *pVnodeObj = pVnode;
149,618,669✔
2055
  metaRLock(pVnodeObj->pMeta);
149,618,669✔
2056

2057
  // fast path: search cache
2058
  SMetaStbStats state = {0};
149,618,797✔
2059
  if (metaStatsCacheGet(pVnodeObj->pMeta, uid, &state) == TSDB_CODE_SUCCESS) {
149,626,013✔
2060
    metaULock(pVnodeObj->pMeta);
143,991,762✔
2061
    if (numOfTables) *numOfTables = state.ctbNum;
144,000,744✔
2062
    if (numOfCols) *numOfCols = state.colNum;
143,997,548✔
2063
    if (flags) *flags = state.flags;
143,981,284✔
2064
    goto _exit;
143,981,284✔
2065
  }
2066

2067
  // slow path: search TDB
2068
  int64_t ctbNum = 0;
5,652,963✔
2069
  int32_t colNum = 0;
5,653,090✔
2070
  int64_t keep = 0;
5,654,086✔
2071
  int8_t  flag = 0;
5,655,245✔
2072

2073
  code = vnodeGetCtbNum(pVnode, uid, &ctbNum);
5,655,262✔
2074
  if (TSDB_CODE_SUCCESS == code) {
5,655,478✔
2075
    code = vnodeGetStbColumnNum(pVnode, uid, &colNum);
5,655,478✔
2076
  }
2077
  if (TSDB_CODE_SUCCESS == code) {
5,654,388✔
2078
    code = vnodeGetStbInfo(pVnode, uid, &keep, &flag);
5,654,388✔
2079
  }
2080
  metaULock(pVnodeObj->pMeta);
5,655,515✔
2081
  if (TSDB_CODE_SUCCESS != code) {
5,656,488✔
2082
    goto _exit;
×
2083
  }
2084

2085
  if (numOfTables) *numOfTables = ctbNum;
5,656,488✔
2086
  if (numOfCols) *numOfCols = colNum;
5,656,488✔
2087
  if (flags) *flags = flag;
5,656,833✔
2088

2089
  state.uid = uid;
5,656,833✔
2090
  state.ctbNum = ctbNum;
5,656,833✔
2091
  state.colNum = colNum;
5,656,833✔
2092
  state.flags = flag;
5,656,833✔
2093
  state.keep = keep;
5,656,833✔
2094
  // upsert the cache
2095
  metaWLock(pVnodeObj->pMeta);
5,656,833✔
2096

2097
  int32_t ret = metaStatsCacheUpsert(pVnodeObj->pMeta, &state);
5,656,143✔
2098
  if (ret) {
5,655,086✔
2099
    metaError("failed to upsert stats, uid:%" PRId64 ", ctbNum:%" PRId64 ", colNum:%d, keep:%" PRId64 ", flags:%" PRIi8,
×
2100
              uid, ctbNum, colNum, keep, flag);
2101
  }
2102

2103
  metaULock(pVnodeObj->pMeta);
5,655,086✔
2104

2105
_exit:
149,637,772✔
2106
  return code;
149,650,218✔
2107
}
2108

2109
void metaUpdateStbStats(SMeta *pMeta, int64_t uid, int64_t deltaCtb, int32_t deltaCol, int64_t deltaKeep) {
89,815,992✔
2110
  SMetaStbStats stats = {0};
89,815,992✔
2111

2112
  if (metaStatsCacheGet(pMeta, uid, &stats) == TSDB_CODE_SUCCESS) {
89,836,498✔
2113
    stats.ctbNum += deltaCtb;
77,419,080✔
2114
    stats.colNum += deltaCol;
77,419,080✔
2115
    if (deltaKeep > 0) {
77,419,080✔
2116
      stats.keep = deltaKeep;
6,383✔
2117
    }
2118

2119
    int32_t code = metaStatsCacheUpsert(pMeta, &stats);
77,419,080✔
2120
    if (code) {
77,390,120✔
2121
      metaError("vgId:%d, failed to update stats, uid:%" PRId64 ", ctbNum:%" PRId64 ", colNum:%d, keep:%" PRId64,
×
2122
                TD_VID(pMeta->pVnode), uid, deltaCtb, deltaCol, deltaKeep > 0 ? deltaKeep : stats.keep);
2123
    }
2124
  }
2125
}
89,812,350✔
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