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

taosdata / TDengine / #4913

06 Jan 2026 01:30AM UTC coverage: 64.884% (-0.004%) from 64.888%
#4913

push

travis-ci

web-flow
merge: from main to 3.0 branch #34167

180 of 319 new or added lines in 14 files covered. (56.43%)

571 existing lines in 128 files now uncovered.

195016 of 300563 relevant lines covered (64.88%)

117540852.85 hits per line

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

78.07
/source/dnode/vnode/src/meta/metaCache.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
#include "meta.h"
16

17
#ifdef TD_ENTERPRISE
18
extern const char* tkLogStb[];
19
extern const char* tkAuditStb[];
20
extern const int   tkLogStbNum;
21
extern const int   tkAuditStbNum;
22
#endif
23

24
#define TAG_FILTER_RES_KEY_LEN  32
25
#define META_CACHE_BASE_BUCKET  1024
26
#define META_CACHE_STATS_BUCKET 16
27

28
// (uid , suid) : child table
29
// (uid,     0) : normal table
30
// (suid, suid) : super table
31
typedef struct SMetaCacheEntry SMetaCacheEntry;
32
struct SMetaCacheEntry {
33
  SMetaCacheEntry* next;
34
  SMetaInfo        info;
35
};
36

37
typedef struct SMetaStbStatsEntry {
38
  struct SMetaStbStatsEntry* next;
39
  SMetaStbStats              info;
40
} SMetaStbStatsEntry;
41

42
typedef struct STagFilterResEntry {
43
  SHashObj *set;    // the set of md5 digest, extracted from the serialized tag query condition
44
  uint32_t hitTimes;  // queried times for current super table
45
} STagFilterResEntry;
46

47
typedef struct STagCondFilterEntry {
48
  SArray*   pColIds;   // SArray<col_id_t>
49
  SHashObj* set;       // SHashObj<digest, SArray<uid>>
50
  uint32_t  hitTimes;  // queried times for current tag filter condition
51
} STagCondFilterEntry;
52

53
typedef struct STagConds {
54
  SHashObj* set;       // SHashObj<tagColIdStr, STagCondFilterEntry>
55
  uint32_t  hitTimes;  // queried times for current super table
56
  uint32_t  numTagDataEntries; // total num of tag data entries in this stable
57
} STagConds;
58

59
struct SMetaCache {
60
  // child, normal, super, table entry cache
61
  struct SEntryCache {
62
    int32_t           nEntry;
63
    int32_t           nBucket;
64
    SMetaCacheEntry** aBucket;
65
  } sEntryCache;
66

67
  // stable stats cache
68
  struct SStbStatsCache {
69
    int32_t              nEntry;
70
    int32_t              nBucket;
71
    SMetaStbStatsEntry** aBucket;
72
  } sStbStatsCache;
73

74
  // query cache
75
  struct STagFilterResCache {
76
    TdThreadMutex lock;
77
    uint32_t      accTimes;
78
    SHashObj*     pTableEntry;
79
    SLRUCache*    pUidResCache;
80
  } sTagFilterResCache;
81

82
  // cache table list for tag filter conditions
83
  // that match format "tag1 = v1 AND tag2 = v2 AND ..."
84
  struct SStableTagFilterResCache {
85
    TdThreadRwlock rwlock;
86
    SHashObj*      pTableEntry;  // HashObj<suid, STagConds>
87
    // access times
88
    uint64_t       accTimes;
89
    // hit times
90
    uint64_t       hitTimes;
91
    // total num of tag data entries in all stables
92
    uint32_t       numTagDataEntries;
93
  } sStableTagFilterResCache;
94

95
  struct STbGroupResCache {
96
    TdThreadMutex lock;
97
    uint32_t      accTimes;
98
    SHashObj*     pTableEntry;
99
    SLRUCache*    pResCache;
100
  } STbGroupResCache;
101

102
  struct STbFilterCache {
103
    SHashObj* pStb;
104
    SHashObj* pStbName;
105
  } STbFilterCache;
106

107
  struct STbRefDbCache {
108
    TdThreadMutex lock;
109
    SHashObj*     pStbRefs; // key: suid, value: SHashObj<dbName, refTimes>
110
  } STbRefDbCache;
111
};
112

113
static void entryCacheClose(SMeta* pMeta) {
7,504,319✔
114
  if (pMeta->pCache) {
7,504,319✔
115
    // close entry cache
116
    for (int32_t iBucket = 0; iBucket < pMeta->pCache->sEntryCache.nBucket; iBucket++) {
2,147,483,647✔
117
      SMetaCacheEntry* pEntry = pMeta->pCache->sEntryCache.aBucket[iBucket];
2,147,483,647✔
118
      while (pEntry) {
2,147,483,647✔
119
        SMetaCacheEntry* tEntry = pEntry->next;
89,618,887✔
120
        taosMemoryFree(pEntry);
89,618,887✔
121
        pEntry = tEntry;
89,616,794✔
122
      }
123
    }
124
    taosMemoryFree(pMeta->pCache->sEntryCache.aBucket);
7,505,176✔
125
  }
126
}
7,505,176✔
127

128
static void statsCacheClose(SMeta* pMeta) {
7,505,176✔
129
  if (pMeta->pCache) {
7,505,176✔
130
    // close entry cache
131
    for (int32_t iBucket = 0; iBucket < pMeta->pCache->sStbStatsCache.nBucket; iBucket++) {
129,376,364✔
132
      SMetaStbStatsEntry* pEntry = pMeta->pCache->sStbStatsCache.aBucket[iBucket];
121,871,313✔
133
      while (pEntry) {
125,687,675✔
134
        SMetaStbStatsEntry* tEntry = pEntry->next;
3,816,487✔
135
        taosMemoryFree(pEntry);
3,816,487✔
136
        pEntry = tEntry;
3,816,487✔
137
      }
138
    }
139
    taosMemoryFree(pMeta->pCache->sStbStatsCache.aBucket);
7,505,176✔
140
  }
141
}
7,504,346✔
142

143
static void freeCacheEntryFp(void* param) {
4,698✔
144
  STagFilterResEntry** p = param;
4,698✔
145
  taosHashCleanup((*p)->set);
4,698✔
146
  taosMemoryFreeClear(*p);
4,698✔
147
}
4,698✔
148

149
static void freeTagFilterEntryFp(void* param) {
3,021✔
150
  STagCondFilterEntry** p = param;
3,021✔
151
  taosArrayDestroy((*p)->pColIds);
3,021✔
152
  taosHashCleanup((*p)->set);
3,021✔
153
  taosMemoryFreeClear(*p);
3,021✔
154
}
3,021✔
155

156
static void freeTagCondsFp(void* param) {
636✔
157
  STagConds** p = param;
636✔
158
  taosHashCleanup((*p)->set);
636✔
159
  taosMemoryFreeClear(*p);
636✔
160
}
636✔
161

162
static void freeRefDbFp(void* param) {
17,097✔
163
  SHashObj** p = param;
17,097✔
164
  taosHashCleanup(*p);
17,097✔
165
  *p = NULL;
17,097✔
166
}
17,097✔
167

168
int32_t metaCacheOpen(SMeta* pMeta) {
7,496,776✔
169
  int32_t code = 0;
7,496,776✔
170
  int32_t lino;
171

172
  pMeta->pCache = (SMetaCache*)taosMemoryCalloc(1, sizeof(SMetaCache));
7,496,776✔
173
  if (pMeta->pCache == NULL) {
7,503,168✔
174
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
175
  }
176

177
  // open entry cache
178
  pMeta->pCache->sEntryCache.nEntry = 0;
7,504,057✔
179
  pMeta->pCache->sEntryCache.nBucket = META_CACHE_BASE_BUCKET;
7,503,247✔
180
  pMeta->pCache->sEntryCache.aBucket =
7,505,644✔
181
      (SMetaCacheEntry**)taosMemoryCalloc(pMeta->pCache->sEntryCache.nBucket, sizeof(SMetaCacheEntry*));
7,503,048✔
182
  if (pMeta->pCache->sEntryCache.aBucket == NULL) {
7,505,176✔
183
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
184
  }
185

186
  // open stats cache
187
  pMeta->pCache->sStbStatsCache.nEntry = 0;
7,504,467✔
188
  pMeta->pCache->sStbStatsCache.nBucket = META_CACHE_STATS_BUCKET;
7,503,839✔
189
  pMeta->pCache->sStbStatsCache.aBucket =
7,505,553✔
190
      (SMetaStbStatsEntry**)taosMemoryCalloc(pMeta->pCache->sStbStatsCache.nBucket, sizeof(SMetaStbStatsEntry*));
7,504,203✔
191
  if (pMeta->pCache->sStbStatsCache.aBucket == NULL) {
7,505,176✔
192
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
193
  }
194

195
  // open tag filter cache
196
  pMeta->pCache->sTagFilterResCache.pUidResCache = taosLRUCacheInit(5 * 1024 * 1024, -1, 0.5);
7,503,245✔
197
  if (pMeta->pCache->sTagFilterResCache.pUidResCache == NULL) {
7,505,176✔
198
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
199
  }
200

201
  pMeta->pCache->sTagFilterResCache.accTimes = 0;
7,505,176✔
202
  pMeta->pCache->sTagFilterResCache.pTableEntry =
15,010,352✔
203
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
15,008,714✔
204
  if (pMeta->pCache->sTagFilterResCache.pTableEntry == NULL) {
7,505,176✔
205
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
206
  }
207

208
  taosHashSetFreeFp(pMeta->pCache->sTagFilterResCache.pTableEntry, freeCacheEntryFp);
7,505,176✔
209
  (void)taosThreadMutexInit(&pMeta->pCache->sTagFilterResCache.lock, NULL);
7,504,379✔
210

211
  // open stable tag filter cache
212
  pMeta->pCache->sStableTagFilterResCache.accTimes = 0;
7,505,176✔
213
  pMeta->pCache->sStableTagFilterResCache.numTagDataEntries = 0;
7,504,773✔
214
  pMeta->pCache->sStableTagFilterResCache.pTableEntry =
15,009,473✔
215
    taosHashInit(1024,
15,009,117✔
216
      taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
217
  if (pMeta->pCache->sStableTagFilterResCache.pTableEntry == NULL) {
7,505,094✔
218
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
219
  }
220
  taosHashSetFreeFp(
7,505,094✔
221
    pMeta->pCache->sStableTagFilterResCache.pTableEntry, freeTagCondsFp);
7,505,176✔
222

223
  TAOS_UNUSED(taosThreadRwlockInit(
7,505,176✔
224
    &pMeta->pCache->sStableTagFilterResCache.rwlock, NULL));
225

226
  // open group res cache
227
  pMeta->pCache->STbGroupResCache.pResCache = taosLRUCacheInit(5 * 1024 * 1024, -1, 0.5);
7,505,176✔
228
  if (pMeta->pCache->STbGroupResCache.pResCache == NULL) {
7,505,176✔
229
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
230
  }
231

232
  pMeta->pCache->STbGroupResCache.accTimes = 0;
7,505,176✔
233
  pMeta->pCache->STbGroupResCache.pTableEntry =
15,009,860✔
234
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
15,008,237✔
235
  if (pMeta->pCache->STbGroupResCache.pTableEntry == NULL) {
7,505,176✔
236
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
237
  }
238

239
  taosHashSetFreeFp(pMeta->pCache->STbGroupResCache.pTableEntry, freeCacheEntryFp);
7,504,773✔
240
  (void)taosThreadMutexInit(&pMeta->pCache->STbGroupResCache.lock, NULL);
7,505,176✔
241

242
  // open filter cache
243
  pMeta->pCache->STbFilterCache.pStb =
15,010,352✔
244
      taosHashInit(0, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
15,009,520✔
245
  if (pMeta->pCache->STbFilterCache.pStb == NULL) {
7,505,176✔
246
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
247
  }
248

249
  pMeta->pCache->STbFilterCache.pStbName =
15,009,949✔
250
      taosHashInit(0, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
15,009,032✔
251
  if (pMeta->pCache->STbFilterCache.pStbName == NULL) {
7,505,176✔
252
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
253
  }
254

255
  // open ref db cache
256
  pMeta->pCache->STbRefDbCache.pStbRefs =
15,010,004✔
257
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
15,008,950✔
258
  if (pMeta->pCache->STbRefDbCache.pStbRefs == NULL) {
7,505,176✔
259
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
260
  }
261

262
  taosHashSetFreeFp(pMeta->pCache->STbRefDbCache.pStbRefs, freeRefDbFp);
7,505,176✔
263
  (void)taosThreadMutexInit(&pMeta->pCache->STbRefDbCache.lock, NULL);
7,505,176✔
264

265

266
_exit:
7,503,801✔
267
  if (code) {
7,503,801✔
268
    metaError("vgId:%d, %s failed at %s:%d since %s", TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
×
269
    metaCacheClose(pMeta);
×
270
  } else {
271
    metaDebug("vgId:%d, %s success", TD_VID(pMeta->pVnode), __func__);
7,503,801✔
272
  }
273
  return code;
7,503,801✔
274
}
275

276
void metaCacheClose(SMeta* pMeta) {
7,505,094✔
277
  if (pMeta->pCache) {
7,505,094✔
278
    entryCacheClose(pMeta);
7,505,094✔
279
    statsCacheClose(pMeta);
7,505,176✔
280

281
    taosHashClear(pMeta->pCache->sTagFilterResCache.pTableEntry);
7,505,051✔
282
    taosLRUCacheCleanup(pMeta->pCache->sTagFilterResCache.pUidResCache);
7,505,176✔
283
    (void)taosThreadMutexDestroy(&pMeta->pCache->sTagFilterResCache.lock);
7,505,176✔
284
    taosHashCleanup(pMeta->pCache->sTagFilterResCache.pTableEntry);
7,505,176✔
285

286
    (void)taosThreadRwlockDestroy(&pMeta->pCache->sStableTagFilterResCache.rwlock);
7,505,176✔
287
    taosHashCleanup(pMeta->pCache->sStableTagFilterResCache.pTableEntry);
7,505,176✔
288

289
    taosHashClear(pMeta->pCache->STbGroupResCache.pTableEntry);
7,505,176✔
290
    taosLRUCacheCleanup(pMeta->pCache->STbGroupResCache.pResCache);
7,505,087✔
291
    (void)taosThreadMutexDestroy(&pMeta->pCache->STbGroupResCache.lock);
7,505,176✔
292
    taosHashCleanup(pMeta->pCache->STbGroupResCache.pTableEntry);
7,505,176✔
293

294
    taosHashCleanup(pMeta->pCache->STbFilterCache.pStb);
7,505,176✔
295
    taosHashCleanup(pMeta->pCache->STbFilterCache.pStbName);
7,505,176✔
296

297
    taosHashClear(pMeta->pCache->STbRefDbCache.pStbRefs);
7,505,176✔
298
    (void)taosThreadMutexDestroy(&pMeta->pCache->STbRefDbCache.lock);
7,505,176✔
299
    taosHashCleanup(pMeta->pCache->STbRefDbCache.pStbRefs);
7,505,176✔
300

301
    taosMemoryFree(pMeta->pCache);
7,504,720✔
302
    pMeta->pCache = NULL;
7,505,176✔
303
  }
304
}
7,504,720✔
305

306
static void metaRehashCache(SMetaCache* pCache, int8_t expand) {
22,538✔
307
  int32_t code = 0;
22,538✔
308
  int32_t nBucket;
309

310
  if (expand) {
22,538✔
311
    nBucket = pCache->sEntryCache.nBucket * 2;
22,538✔
312
  } else {
313
    nBucket = pCache->sEntryCache.nBucket / 2;
×
314
  }
315

316
  SMetaCacheEntry** aBucket = (SMetaCacheEntry**)taosMemoryCalloc(nBucket, sizeof(SMetaCacheEntry*));
22,538✔
317
  if (aBucket == NULL) {
22,538✔
318
    return;
×
319
  }
320

321
  // rehash
322
  for (int32_t iBucket = 0; iBucket < pCache->sEntryCache.nBucket; iBucket++) {
77,400,074✔
323
    SMetaCacheEntry* pEntry = pCache->sEntryCache.aBucket[iBucket];
77,377,536✔
324

325
    while (pEntry) {
154,755,072✔
326
      SMetaCacheEntry* pTEntry = pEntry->next;
77,377,536✔
327

328
      pEntry->next = aBucket[TABS(pEntry->info.uid) % nBucket];
77,377,536✔
329
      aBucket[TABS(pEntry->info.uid) % nBucket] = pEntry;
77,377,536✔
330

331
      pEntry = pTEntry;
77,377,536✔
332
    }
333
  }
334

335
  // final set
336
  taosMemoryFree(pCache->sEntryCache.aBucket);
22,538✔
337
  pCache->sEntryCache.nBucket = nBucket;
22,538✔
338
  pCache->sEntryCache.aBucket = aBucket;
22,538✔
339
  return;
22,538✔
340
}
341

342
int32_t metaCacheUpsert(SMeta* pMeta, SMetaInfo* pInfo) {
110,790,124✔
343
  int32_t code = 0;
110,790,124✔
344

345
  // meta is wlocked for calling this func.
346

347
  // search
348
  SMetaCache*       pCache = pMeta->pCache;
110,790,124✔
349
  int32_t           iBucket = TABS(pInfo->uid) % pCache->sEntryCache.nBucket;
110,814,457✔
350
  SMetaCacheEntry** ppEntry = &pCache->sEntryCache.aBucket[iBucket];
110,776,541✔
351
  while (*ppEntry && (*ppEntry)->info.uid != pInfo->uid) {
129,957,528✔
352
    ppEntry = &(*ppEntry)->next;
19,201,276✔
353
  }
354

355
  if (*ppEntry) {  // update
110,733,579✔
356
    if (pInfo->suid != (*ppEntry)->info.suid) {
18,252,481✔
357
      metaError("meta/cache: suid should be same as the one in cache.");
×
358
      return TSDB_CODE_INVALID_PARA;
×
359
    }
360
    if (pInfo->version > (*ppEntry)->info.version) {
18,195,794✔
361
      (*ppEntry)->info.version = pInfo->version;
18,199,905✔
362
      (*ppEntry)->info.skmVer = pInfo->skmVer;
18,222,265✔
363
    }
364
  } else {  // insert
365
    if (pCache->sEntryCache.nEntry >= pCache->sEntryCache.nBucket) {
92,537,722✔
366
      metaRehashCache(pCache, 1);
22,538✔
367

368
      iBucket = TABS(pInfo->uid) % pCache->sEntryCache.nBucket;
22,538✔
369
    }
370

371
    SMetaCacheEntry* pEntryNew = (SMetaCacheEntry*)taosMemoryMalloc(sizeof(*pEntryNew));
92,545,141✔
372
    if (pEntryNew == NULL) {
92,537,613✔
373
      code = terrno;
×
374
      goto _exit;
×
375
    }
376

377
    pEntryNew->info = *pInfo;
92,537,613✔
378
    pEntryNew->next = pCache->sEntryCache.aBucket[iBucket];
92,535,179✔
379
    pCache->sEntryCache.aBucket[iBucket] = pEntryNew;
92,519,809✔
380
    pCache->sEntryCache.nEntry++;
92,511,320✔
381
  }
382

383
_exit:
110,774,576✔
384
  return code;
110,774,576✔
385
}
386

387
int32_t metaCacheDrop(SMeta* pMeta, int64_t uid) {
3,025,630✔
388
  int32_t code = 0;
3,025,630✔
389

390
  SMetaCache*       pCache = pMeta->pCache;
3,025,630✔
391
  int32_t           iBucket = TABS(uid) % pCache->sEntryCache.nBucket;
3,026,245✔
392
  SMetaCacheEntry** ppEntry = &pCache->sEntryCache.aBucket[iBucket];
3,026,158✔
393
  while (*ppEntry && (*ppEntry)->info.uid != uid) {
3,035,303✔
394
    ppEntry = &(*ppEntry)->next;
9,789✔
395
  }
396

397
  SMetaCacheEntry* pEntry = *ppEntry;
3,025,778✔
398
  if (pEntry) {
3,025,514✔
399
    *ppEntry = pEntry->next;
2,947,071✔
400
    taosMemoryFree(pEntry);
2,947,802✔
401
    pCache->sEntryCache.nEntry--;
2,947,503✔
402
    if (pCache->sEntryCache.nEntry < pCache->sEntryCache.nBucket / 4 &&
2,948,147✔
403
        pCache->sEntryCache.nBucket > META_CACHE_BASE_BUCKET) {
2,659,684✔
404
      metaRehashCache(pCache, 0);
×
405
    }
406
  } else {
407
    code = TSDB_CODE_NOT_FOUND;
78,443✔
408
  }
409

410
_exit:
3,026,034✔
411
  return code;
3,026,034✔
412
}
413

414
int32_t metaCacheGet(SMeta* pMeta, int64_t uid, SMetaInfo* pInfo) {
2,091,969,289✔
415
  int32_t code = 0;
2,091,969,289✔
416

417
  SMetaCache*      pCache = pMeta->pCache;
2,091,969,289✔
418
  int32_t          iBucket = TABS(uid) % pCache->sEntryCache.nBucket;
2,092,321,492✔
419
  SMetaCacheEntry* pEntry = pCache->sEntryCache.aBucket[iBucket];
2,092,326,707✔
420

421
  while (pEntry && pEntry->info.uid != uid) {
2,147,483,647✔
422
    pEntry = pEntry->next;
59,612,600✔
423
  }
424

425
  if (pEntry) {
2,092,242,937✔
426
    if (pInfo) {
1,993,149,991✔
427
      *pInfo = pEntry->info;
1,993,181,807✔
428
    }
429
  } else {
430
    code = TSDB_CODE_NOT_FOUND;
99,092,946✔
431
  }
432

433
  return code;
2,092,312,996✔
434
}
435

436
static int32_t metaRehashStatsCache(SMetaCache* pCache, int8_t expand) {
48,581✔
437
  int32_t code = 0;
48,581✔
438
  int32_t nBucket;
439

440
  if (expand) {
48,581✔
441
    nBucket = pCache->sStbStatsCache.nBucket * 2;
43,379✔
442
  } else {
443
    nBucket = pCache->sStbStatsCache.nBucket / 2;
5,202✔
444
  }
445

446
  SMetaStbStatsEntry** aBucket = (SMetaStbStatsEntry**)taosMemoryCalloc(nBucket, sizeof(SMetaStbStatsEntry*));
48,581✔
447
  if (aBucket == NULL) {
48,581✔
448
    code = terrno;
×
449
    goto _exit;
×
450
  }
451

452
  // rehash
453
  for (int32_t iBucket = 0; iBucket < pCache->sStbStatsCache.nBucket; iBucket++) {
2,185,141✔
454
    SMetaStbStatsEntry* pEntry = pCache->sStbStatsCache.aBucket[iBucket];
2,136,560✔
455

456
    while (pEntry) {
4,098,286✔
457
      SMetaStbStatsEntry* pTEntry = pEntry->next;
1,961,726✔
458

459
      pEntry->next = aBucket[TABS(pEntry->info.uid) % nBucket];
1,961,726✔
460
      aBucket[TABS(pEntry->info.uid) % nBucket] = pEntry;
1,961,726✔
461

462
      pEntry = pTEntry;
1,961,726✔
463
    }
464
  }
465

466
  // final set
467
  taosMemoryFree(pCache->sStbStatsCache.aBucket);
48,581✔
468
  pCache->sStbStatsCache.nBucket = nBucket;
48,581✔
469
  pCache->sStbStatsCache.aBucket = aBucket;
48,581✔
470

471
_exit:
48,581✔
472
  return code;
48,581✔
473
}
474

475
int32_t metaStatsCacheUpsert(SMeta* pMeta, SMetaStbStats* pInfo) {
62,160,579✔
476
  int32_t code = 0;
62,160,579✔
477

478
  // meta is wlocked for calling this func.
479

480
  // search
481
  SMetaCache*          pCache = pMeta->pCache;
62,160,579✔
482
  int32_t              iBucket = TABS(pInfo->uid) % pCache->sStbStatsCache.nBucket;
62,177,163✔
483
  SMetaStbStatsEntry** ppEntry = &pCache->sStbStatsCache.aBucket[iBucket];
62,157,450✔
484
  while (*ppEntry && (*ppEntry)->info.uid != pInfo->uid) {
63,781,225✔
485
    ppEntry = &(*ppEntry)->next;
1,634,451✔
486
  }
487

488
  if (*ppEntry) {  // update
62,164,266✔
489
    (*ppEntry)->info.ctbNum = pInfo->ctbNum;
57,787,056✔
490
    (*ppEntry)->info.colNum = pInfo->colNum;
57,796,717✔
491
    (*ppEntry)->info.flags = pInfo->flags;
57,768,904✔
492
    (*ppEntry)->info.keep = pInfo->keep;
57,782,584✔
493
  } else {  // insert
494
    if (pCache->sStbStatsCache.nEntry >= pCache->sStbStatsCache.nBucket) {
4,374,698✔
495
      TAOS_UNUSED(metaRehashStatsCache(pCache, 1));
43,379✔
496
      iBucket = TABS(pInfo->uid) % pCache->sStbStatsCache.nBucket;
43,379✔
497
    }
498

499
    SMetaStbStatsEntry* pEntryNew = (SMetaStbStatsEntry*)taosMemoryMalloc(sizeof(*pEntryNew));
4,375,063✔
500
    if (pEntryNew == NULL) {
4,375,062✔
501
      code = terrno;
×
502
      goto _exit;
×
503
    }
504

505
    pEntryNew->info = *pInfo;
4,375,062✔
506
    pEntryNew->next = pCache->sStbStatsCache.aBucket[iBucket];
4,375,481✔
507
    pCache->sStbStatsCache.aBucket[iBucket] = pEntryNew;
4,374,642✔
508
    pCache->sStbStatsCache.nEntry++;
4,374,285✔
509
  }
510

511
_exit:
62,113,564✔
512
  return code;
62,113,564✔
513
}
514

515
int32_t metaStatsCacheDrop(SMeta* pMeta, int64_t uid) {
864,700✔
516
  int32_t code = 0;
864,700✔
517

518
  SMetaCache*          pCache = pMeta->pCache;
864,700✔
519
  int32_t              iBucket = TABS(uid) % pCache->sStbStatsCache.nBucket;
864,398✔
520
  SMetaStbStatsEntry** ppEntry = &pCache->sStbStatsCache.aBucket[iBucket];
864,056✔
521
  while (*ppEntry && (*ppEntry)->info.uid != uid) {
973,382✔
522
    ppEntry = &(*ppEntry)->next;
109,326✔
523
  }
524

525
  SMetaStbStatsEntry* pEntry = *ppEntry;
864,700✔
526
  if (pEntry) {
865,042✔
527
    *ppEntry = pEntry->next;
559,521✔
528
    taosMemoryFree(pEntry);
559,521✔
529
    pCache->sStbStatsCache.nEntry--;
559,179✔
530
    if (pCache->sStbStatsCache.nEntry < pCache->sStbStatsCache.nBucket / 4 &&
559,179✔
531
        pCache->sStbStatsCache.nBucket > META_CACHE_STATS_BUCKET) {
409,451✔
532
      TAOS_UNUSED(metaRehashStatsCache(pCache, 0));
5,202✔
533
    }
534
  } else {
535
    code = TSDB_CODE_NOT_FOUND;
305,521✔
536
  }
537

538
_exit:
865,042✔
539
  return code;
865,042✔
540
}
541

542
int32_t metaStatsCacheGet(SMeta* pMeta, int64_t uid, SMetaStbStats* pInfo) {
188,717,485✔
543
  int32_t code = TSDB_CODE_SUCCESS;
188,717,485✔
544

545
  SMetaCache*         pCache = pMeta->pCache;
188,717,485✔
546
  int32_t             iBucket = TABS(uid) % pCache->sStbStatsCache.nBucket;
188,742,775✔
547
  SMetaStbStatsEntry* pEntry = pCache->sStbStatsCache.aBucket[iBucket];
188,731,181✔
548

549
  while (pEntry && pEntry->info.uid != uid) {
202,850,645✔
550
    pEntry = pEntry->next;
14,072,969✔
551
  }
552

553
  if (pEntry) {
188,778,455✔
554
    if (pInfo) {
174,007,508✔
555
      *pInfo = pEntry->info;
174,008,718✔
556
    }
557
  } else {
558
    code = TSDB_CODE_NOT_FOUND;
14,770,947✔
559
  }
560

561
  return code;
188,776,805✔
562
}
563

564
static FORCE_INLINE void setMD5DigestInKey(uint64_t* pBuf, const char* key, int32_t keyLen) {
565
  memcpy(&pBuf[2], key, keyLen);
136,549,653✔
566
}
136,547,356✔
567

568
// the format of key:
569
// hash table address(8bytes) + suid(8bytes) + MD5 digest(16bytes)
570
static void initCacheKey(uint64_t* buf, const SHashObj* pHashMap, uint64_t suid, const char* key, int32_t keyLen) {
136,526,881✔
571
  buf[0] = (uint64_t)pHashMap;
136,526,881✔
572
  buf[1] = suid;
136,534,488✔
573
  setMD5DigestInKey(buf, key, keyLen);
574
}
136,547,899✔
575

576
int32_t metaGetCachedTableUidList(void* pVnode, tb_uid_t suid, const uint8_t* pKey, int32_t keyLen, SArray* pList1,
29,383✔
577
                                  bool* acquireRes) {
578
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
29,383✔
579
  int32_t vgId = TD_VID(pMeta->pVnode);
29,383✔
580

581
  // generate the composed key for LRU cache
582
  SLRUCache*     pCache = pMeta->pCache->sTagFilterResCache.pUidResCache;
29,383✔
583
  SHashObj*      pTableMap = pMeta->pCache->sTagFilterResCache.pTableEntry;
29,383✔
584
  TdThreadMutex* pLock = &pMeta->pCache->sTagFilterResCache.lock;
29,383✔
585

586
  *acquireRes = 0;
29,383✔
587
  uint64_t key[4];
29,383✔
588
  initCacheKey(key, pTableMap, suid, (const char*)pKey, keyLen);
29,383✔
589
  
590
  // void* tmp = NULL;
591
  // uint32_t len = 0;
592
  // (void)taosAscii2Hex((const char*)key, 32, &tmp, &len);
593
  // qDebug("metaGetCachedTableUidList %p %"PRId64" key: %s", pTableMap, suid, tmp);
594
  // taosMemoryFree(tmp);
595

596
  (void)taosThreadMutexLock(pLock);
29,383✔
597
  pMeta->pCache->sTagFilterResCache.accTimes += 1;
29,383✔
598

599
  LRUHandle* pHandle = taosLRUCacheLookup(pCache, key, TAG_FILTER_RES_KEY_LEN);
29,383✔
600
  if (pHandle == NULL) {
29,383✔
601
    (void)taosThreadMutexUnlock(pLock);
7,257✔
602
    return TSDB_CODE_SUCCESS;
7,257✔
603
  }
604

605
  // do some book mark work after acquiring the filter result from cache
606
  STagFilterResEntry** pEntry = taosHashGet(pTableMap, &suid, sizeof(uint64_t));
22,126✔
607
  if (NULL == pEntry) {
22,126✔
608
    metaError("meta/cache: pEntry should not be NULL.");
×
609
    return TSDB_CODE_NOT_FOUND;
×
610
  }
611

612
  *acquireRes = 1;
22,126✔
613

614
  const char* p = taosLRUCacheValue(pCache, pHandle);
22,126✔
615
  int32_t     size = *(int32_t*)p;
22,126✔
616

617
  // set the result into the buffer
618
  if (taosArrayAddBatch(pList1, p + sizeof(int32_t), size) == NULL) {
22,126✔
619
    return terrno;
×
620
  }
621

622
  (*pEntry)->hitTimes += 1;
22,126✔
623

624
  uint32_t acc = pMeta->pCache->sTagFilterResCache.accTimes;
22,126✔
625
  if ((*pEntry)->hitTimes % 5000 == 0 && (*pEntry)->hitTimes > 0) {
22,126✔
626
    metaInfo("vgId:%d cache hit:%d, total acc:%d, rate:%.2f", vgId, (*pEntry)->hitTimes, acc,
×
627
             ((double)(*pEntry)->hitTimes) / acc);
628
  }
629

630
  bool ret = taosLRUCacheRelease(pCache, pHandle, false);
22,126✔
631

632
  // unlock meta
633
  (void)taosThreadMutexUnlock(pLock);
22,126✔
634
  return TSDB_CODE_SUCCESS;
22,126✔
635
}
636

637
int32_t metaStableTagFilterCacheGet(void* pVnode, tb_uid_t suid,
8,745✔
638
  const uint8_t* pTagCondKey, int32_t tagCondKeyLen,
639
  const uint8_t* pKey, int32_t keyLen, SArray* pList1, bool* acquireRes) {
640

641
  int32_t code = TSDB_CODE_SUCCESS;
8,745✔
642
  int32_t lino = 0;
8,745✔
643
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
8,745✔
644
  int32_t vgId = TD_VID(pMeta->pVnode);
8,745✔
645
  *acquireRes = 0;
8,745✔
646

647
  // generate the composed key for LRU cache
648
  SHashObj*       pTableMap = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
8,745✔
649
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
8,745✔
650

651
  code = taosThreadRwlockRdlock(pRwlock);
8,745✔
652
  TSDB_CHECK_CODE(code, lino, _end);
8,745✔
653
  pMeta->pCache->sStableTagFilterResCache.accTimes += 1;
8,745✔
654

655
  STagConds** pTagConds =
656
    (STagConds**)taosHashGet(pTableMap, &suid, sizeof(tb_uid_t));
8,745✔
657
  TSDB_CHECK_NULL(pTagConds, code, lino, _end, TSDB_CODE_SUCCESS);
8,745✔
658

659
  STagCondFilterEntry** pFilterEntry = (STagCondFilterEntry**)taosHashGet(
7,791✔
660
    (*pTagConds)->set, pTagCondKey, tagCondKeyLen);
7,791✔
661
  TSDB_CHECK_NULL(pFilterEntry, code, lino, _end, TSDB_CODE_SUCCESS);
7,791✔
662

663
  SArray** pArray = (SArray**)taosHashGet((*pFilterEntry)->set, pKey, keyLen);
5,565✔
664
  TSDB_CHECK_NULL(pArray, code, lino, _end, TSDB_CODE_SUCCESS);
5,565✔
665

666
  // set the result into the buffer
667
  *acquireRes = 1;
4,293✔
668
  TAOS_UNUSED(taosArrayAddBatch(
4,293✔
669
    pList1, TARRAY_GET_ELEM(*pArray, 0), taosArrayGetSize(*pArray)));
670

671
  // do some bookmark work after acquiring the filter result from cache
672
  (*pTagConds)->hitTimes += 1;
4,293✔
673
  (*pFilterEntry)->hitTimes += 1;
4,293✔
674
  uint64_t hit = ++pMeta->pCache->sStableTagFilterResCache.hitTimes;
4,293✔
675
  uint64_t acc = pMeta->pCache->sStableTagFilterResCache.accTimes;
4,293✔
676
  if ((*pTagConds)->hitTimes % 1000 == 0 && (*pTagConds)->hitTimes > 0) {
4,293✔
677
    metaDebug(
×
678
      "vgId:%d, suid:%" PRIu64 
679
      ", current stable cache hit:%" PRIu32 ", this tag condition hit:%" PRIu32
680
      ", total cache hit:%" PRIu64 ", acc:%" PRIu64 ", hit rate:%.2f%%",
681
      vgId, suid, (*pTagConds)->hitTimes, (*pFilterEntry)->hitTimes, hit, acc,
682
      ((double)hit / acc * 100));
683
  }
684

685
_end:
4,293✔
686
  if (TSDB_CODE_SUCCESS != code) {
8,745✔
687
    metaError("vgId:%d, %s failed at %s:%d since %s",
×
688
      vgId, __func__, __FILE__, lino, tstrerror(code));
689
  }
690
  // unlock meta
691
  code = taosThreadRwlockUnlock(pRwlock);
8,745✔
692
  if (TSDB_CODE_SUCCESS != code) {
8,745✔
693
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
694
      vgId, __func__, __FILE__, lino, tstrerror(code));
695
  }
696
  return code;
8,745✔
697
}
698

699
static void freeUidCachePayload(const void* key, size_t keyLen, void* value, void* ud) {
7,257✔
700
  (void)ud;
701
  if (value == NULL) {
7,257✔
702
    return;
×
703
  }
704

705
  const uint64_t* p = key;
7,257✔
706
  if (keyLen != sizeof(int64_t) * 4) {
7,257✔
707
    metaError("key length is invalid, length:%d, expect:%d", (int32_t)keyLen, (int32_t)sizeof(uint64_t) * 2);
×
708
    return;
×
709
  }
710

711
  SHashObj* pHashObj = (SHashObj*)p[0];
7,257✔
712

713
  STagFilterResEntry** pEntry = taosHashGet(pHashObj, &p[1], sizeof(uint64_t));
7,257✔
714

715
  if (pEntry != NULL && (*pEntry) != NULL) {
7,257✔
716
    int64_t st = taosGetTimestampUs();
1,754✔
717
    int32_t code = taosHashRemove((*pEntry)->set, &p[2], sizeof(uint64_t) * 2);
1,754✔
718
    if (code == TSDB_CODE_SUCCESS) {
1,754✔
719
      double el = (taosGetTimestampUs() - st) / 1000.0;
1,754✔
720
      metaInfo("clear items in meta-cache, remain cached item:%d, elapsed time:%.2fms", taosHashGetSize((*pEntry)->set),
1,754✔
721
               el);
722
    }
723
  }
724

725
  metaDebug("free uid cache payload in lru, suid: %" PRIu64
7,257✔
726
            " origin key:%" PRIu64 ",%" PRIu64,
727
            p[1], p[2], p[3]);
728
  taosMemoryFree(value);
7,257✔
729
}
730

731
static int32_t addNewEntry(SHashObj* pTableEntry, const void* pKey, int32_t keyLen, uint64_t suid) {
4,698✔
732
  int32_t             code = TSDB_CODE_SUCCESS;
4,698✔
733
  int32_t             lino = 0;
4,698✔
734
  STagFilterResEntry* p = taosMemoryMalloc(sizeof(STagFilterResEntry));
4,698✔
735
  TSDB_CHECK_NULL(p, code, lino, _end, terrno);
4,698✔
736

737
  p->hitTimes = 0;
4,698✔
738
  p->set = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
4,698✔
739
  TSDB_CHECK_NULL(p->set, code, lino, _end, terrno);
4,698✔
740
  code = taosHashPut(p->set, pKey, keyLen, NULL, 0);
4,698✔
741
  TSDB_CHECK_CODE(code, lino, _end);
4,698✔
742
  code = taosHashPut(pTableEntry, &suid, sizeof(uint64_t), &p, POINTER_BYTES);
4,698✔
743
  TSDB_CHECK_CODE(code, lino, _end);
4,698✔
744

745
_end:
4,698✔
746
  if (code != TSDB_CODE_SUCCESS) {
4,698✔
747
    metaError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
748
    if (p != NULL) {
×
749
      if (p->set != NULL) {
×
750
        taosHashCleanup(p->set);
×
751
      }
752
      taosMemoryFree(p);
×
753
    }
754
  }
755
  return code;
4,698✔
756
}
757

758
// check both the payload size and selectivity ratio
759
int32_t metaUidFilterCachePut(void* pVnode, uint64_t suid, const void* pKey, int32_t keyLen, void* pPayload,
7,257✔
760
                              int32_t payloadLen, double selectivityRatio) {
761
  int32_t code = 0;
7,257✔
762
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
7,257✔
763
  int32_t vgId = TD_VID(pMeta->pVnode);
7,257✔
764

765
  if (selectivityRatio > tsSelectivityRatio) {
7,257✔
766
    metaDebug("vgId:%d, suid:%" PRIu64
×
767
              " failed to add to uid list cache, due to selectivity ratio %.2f less than threshold %.2f",
768
              vgId, suid, selectivityRatio, tsSelectivityRatio);
769
    taosMemoryFree(pPayload);
×
770
    return TSDB_CODE_SUCCESS;
×
771
  }
772

773
  if (payloadLen > tsTagFilterResCacheSize) {
7,257✔
774
    metaDebug("vgId:%d, suid:%" PRIu64
×
775
              " failed to add to uid list cache, due to payload length %d greater than threshold %d",
776
              vgId, suid, payloadLen, tsTagFilterResCacheSize);
777
    taosMemoryFree(pPayload);
×
778
    return TSDB_CODE_SUCCESS;
×
779
  }
780

781
  SLRUCache*     pCache = pMeta->pCache->sTagFilterResCache.pUidResCache;
7,257✔
782
  SHashObj*      pTableEntry = pMeta->pCache->sTagFilterResCache.pTableEntry;
7,257✔
783
  TdThreadMutex* pLock = &pMeta->pCache->sTagFilterResCache.lock;
7,257✔
784

785
  uint64_t key[4] = {0};
7,257✔
786
  initCacheKey(key, pTableEntry, suid, pKey, keyLen);
7,257✔
787

788
  // void* tmp = NULL;
789
  // uint32_t len = 0;
790
  // (void)taosAscii2Hex((const char*)key, 32, &tmp, &len);
791
  // qDebug("metaUidFilterCachePut %p %"PRId64" key: %s", pTableEntry, suid, tmp);
792
  // taosMemoryFree(tmp);
793

794
  (void)taosThreadMutexLock(pLock);
7,257✔
795
  STagFilterResEntry** pEntry = taosHashGet(pTableEntry, &suid, sizeof(uint64_t));
7,257✔
796
  if (pEntry == NULL) {
7,257✔
797
    code = addNewEntry(pTableEntry, pKey, keyLen, suid);
4,698✔
798
    if (code != TSDB_CODE_SUCCESS) {
4,698✔
799
      goto _end;
×
800
    }
801
  } else {  // check if it exists or not
802
    code = taosHashPut((*pEntry)->set, pKey, keyLen, NULL, 0);
2,559✔
803
    if (code == TSDB_CODE_DUP_KEY) {
2,559✔
804
      // we have already found the existed items, no need to added to cache anymore.
805
      (void)taosThreadMutexUnlock(pLock);
×
806
      return TSDB_CODE_DUP_KEY;
×
807
    }
808
    if (code != TSDB_CODE_SUCCESS) {
2,559✔
809
      goto _end;
×
810
    }
811
  }
812

813
  // add to cache.
814
  (void)taosLRUCacheInsert(pCache, key, TAG_FILTER_RES_KEY_LEN, pPayload, payloadLen, freeUidCachePayload, NULL, NULL,
7,257✔
815
                           TAOS_LRU_PRIORITY_LOW, NULL);
816
_end:
7,257✔
817
  (void)taosThreadMutexUnlock(pLock);
7,257✔
818
  metaDebug("vgId:%d, suid:%" PRIu64 " list cache added into cache, total:%d, tables:%d", vgId, suid,
7,257✔
819
            (int32_t)taosLRUCacheGetUsage(pCache), taosHashGetSize(pTableEntry));
820

821
  return code;
7,257✔
822
}
823

824
static void freeSArrayPtr(void* pp) {
4,452✔
825
  SArray* pArray = *(SArray**)pp;
4,452✔
826
  taosArrayDestroy(pArray);
4,452✔
827
}
4,452✔
828

829
int32_t metaStableTagFilterCachePut(
4,452✔
830
  void* pVnode, uint64_t suid, const void* pTagCondKey, int32_t tagCondKeyLen,
831
  const void* pKey, int32_t keyLen, SArray* pUidList, SArray** pTagColIds) {
832

833
  int32_t code = TSDB_CODE_SUCCESS;
4,452✔
834
  int32_t lino = 0;
4,452✔
835
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
4,452✔
836
  int32_t vgId = TD_VID(pMeta->pVnode);
4,452✔
837

838
  SHashObj*       pTableEntry = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
4,452✔
839
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
4,452✔
840

841
  code = taosThreadRwlockWrlock(pRwlock);
4,452✔
842
  TSDB_CHECK_CODE(code, lino, _end);
4,452✔
843

844
  STagConds** pTagConds = 
845
    (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(uint64_t));
4,452✔
846
  if (pTagConds == NULL) {
4,452✔
847
    // add new (suid -> tag conds) entry
848
    STagConds* pEntry = (STagConds*)taosMemoryMalloc(sizeof(STagConds));
636✔
849
    TSDB_CHECK_NULL(pEntry, code, lino, _end, terrno);
636✔
850

851
    pEntry->hitTimes = 0;
636✔
852
    pEntry->set = taosHashInit(
636✔
853
      1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY),
854
      false, HASH_NO_LOCK);
855
    taosHashSetFreeFp(pEntry->set, freeTagFilterEntryFp);
636✔
856
    TSDB_CHECK_NULL(pEntry->set, code, lino, _end, terrno);
636✔
857

858
    code = taosHashPut(
636✔
859
      pTableEntry, &suid, sizeof(uint64_t), &pEntry, POINTER_BYTES);
860
    TSDB_CHECK_CODE(code, lino, _end);
636✔
861

862
    pTagConds = (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(uint64_t));
636✔
863
    TSDB_CHECK_NULL(pTagConds, code, lino, _end, TSDB_CODE_NOT_FOUND);
636✔
864
  }
865

866
  STagCondFilterEntry** pFilterEntry =
867
    (STagCondFilterEntry**)taosHashGet(
4,452✔
868
      (*pTagConds)->set, pTagCondKey, tagCondKeyLen);
4,452✔
869
  if (pFilterEntry == NULL) {
4,452✔
870
    // add new (tag cond -> filter entry) entry
871
    STagCondFilterEntry* pEntry = 
3,021✔
872
      (STagCondFilterEntry*)taosMemoryMalloc(sizeof(STagCondFilterEntry));
3,021✔
873
    TSDB_CHECK_NULL(pEntry, code, lino, _end, terrno);
3,021✔
874

875
    pEntry->hitTimes = 0;
3,021✔
876
    pEntry->set = taosHashInit(
3,021✔
877
      1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY),
878
      false, HASH_NO_LOCK);
879
    TSDB_CHECK_NULL(pEntry->set, code, lino, _end, terrno);
3,021✔
880
    taosHashSetFreeFp(pEntry->set, freeSArrayPtr);
3,021✔
881
    pEntry->pColIds = *pTagColIds;
3,021✔
882
    *pTagColIds = NULL;
3,021✔
883

884
    code = taosHashPut(
3,021✔
885
      (*pTagConds)->set, pTagCondKey, tagCondKeyLen, &pEntry, POINTER_BYTES);
3,021✔
886
    TSDB_CHECK_CODE(code, lino, _end);
3,021✔
887

888
    pFilterEntry = (STagCondFilterEntry**)taosHashGet(
3,021✔
889
      (*pTagConds)->set, pTagCondKey, tagCondKeyLen);
3,021✔
890
  } else {
891
    // pColIds is already set, so we can destroy the new one
892
    taosArrayDestroy(*pTagColIds);
1,431✔
893
    *pTagColIds = NULL;
1,431✔
894
  }
895

896
  // add to cache.
897
  SArray* pPayload = taosArrayDup(pUidList, NULL);
4,452✔
898
  code = taosHashPut(
4,452✔
899
    (*pFilterEntry)->set, pKey, keyLen, &pPayload, POINTER_BYTES);
4,452✔
900
  TSDB_CHECK_CODE(code, lino, _end);
4,452✔
901
  pMeta->pCache->sStableTagFilterResCache.numTagDataEntries += 1;
4,452✔
902
  (*pTagConds)->numTagDataEntries += 1;
4,452✔
903

904
_end:
4,452✔
905
  if (TSDB_CODE_SUCCESS != code) {
4,452✔
906
    metaError("vgId:%d, %s failed at %s:%d since %s",
×
907
      vgId, __func__, __FILE__, lino, tstrerror(code));
908
  } else {
909
    metaDebug("vgId:%d, suid:%" PRIu64 " new tag data filter entry added, "
4,452✔
910
      "uid num:%d, current stable tag conditions num:%d, "
911
      "this tag condition data entries num:%d, "
912
      "cache stable num:%d, total tag data entries num:%" PRIu32 ", "
913
      "total tag data entries num:%" PRIu32,
914
      vgId, suid, (int32_t)taosArrayGetSize(pUidList), 
915
      pTagConds ? (int32_t)taosHashGetSize((*pTagConds)->set) : 0,
916
      pFilterEntry ? (int32_t)taosHashGetSize((*pFilterEntry)->set) : 0,
917
      (int32_t)taosHashGetSize(pTableEntry),
918
      pMeta->pCache->sStableTagFilterResCache.numTagDataEntries,
919
      (*pTagConds)->numTagDataEntries);
920
  }
921
  // unlock meta
922
  code = taosThreadRwlockUnlock(pRwlock);
4,452✔
923
  if (TSDB_CODE_SUCCESS != code) {
4,452✔
924
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
925
      vgId, __func__, __FILE__, lino, tstrerror(code));
926
  }
927

928
  return code;
4,452✔
929
}
930

931
// drop all the cache entries for a super table 
932
int32_t metaStableTagFilterCacheDropSTable(
×
933
  SMeta* pMeta, tb_uid_t suid) {
934
  if (pMeta == NULL) {
×
935
    return TSDB_CODE_INVALID_PARA;
×
936
  }
937
  int32_t   lino = 0;
×
938
  int32_t   code = TSDB_CODE_SUCCESS;
×
939
  SHashObj* pTableEntry = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
×
940
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
×
941

942
  code = taosThreadRwlockWrlock(pRwlock);
×
943
  TSDB_CHECK_CODE(code, lino, _end);
×
944
  STagConds** pTagConds = taosHashGet(pTableEntry, &suid, sizeof(tb_uid_t));
×
945
  if (pTagConds != NULL) {
×
946
    pMeta->pCache->sStableTagFilterResCache.
×
947
      numTagDataEntries -= (*pTagConds)->numTagDataEntries;
×
948
  }
949
  code = taosHashRemove(pTableEntry, &suid, sizeof(tb_uid_t));
×
950
  TSDB_CHECK_CODE(code, lino, _end);
×
951

952
_end:
×
953
  if (TSDB_CODE_SUCCESS != code) {
×
954
    metaError("vgId:%d, %s failed at %s:%d since %s",
×
955
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
956
  } else {
957
    metaDebug(
×
958
      "vgId:%d, suid:%" PRIu64 " stable tag filter cache dropped from cache"
959
      "left stable num:%d, tag conditions num:%" PRIu32,
960
      TD_VID(pMeta->pVnode), suid, (int32_t)taosHashGetSize(pTableEntry),
961
      pMeta->pCache->sStableTagFilterResCache.numTagDataEntries);
962
  }
963
  code = taosThreadRwlockUnlock(pRwlock);
×
964
  if (TSDB_CODE_SUCCESS != code) {
×
965
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
966
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
967
  }
968
  return code;
×
969
}
970

971
static int32_t getTagColSize(
14,946✔
972
  const SSchema* pTagSchemas, int32_t nTagCols, col_id_t cid) {
973
  for (int32_t i = 0; i < nTagCols; i++) {
82,362✔
974
    if (pTagSchemas[i].colId == cid) {
82,362✔
975
      return pTagSchemas[i].bytes;
14,946✔
976
    }
977
  }
978
  return 0;
×
979
}
980

981
// when encode nchar tag into tag data entry key, need to convert it to var type
982
static FORCE_INLINE int32_t ncharToVar(char *pData, int32_t nData, char **ppOut) {
983
  int32_t code = TSDB_CODE_SUCCESS;
7,632✔
984

985
  char *t = taosMemoryCalloc(1, nData + VARSTR_HEADER_SIZE);
7,632✔
986
  if (NULL == t) {
7,632✔
987
    return terrno;
×
988
  }
989
  int32_t len = taosUcs4ToMbs(
7,632✔
990
    (TdUcs4 *)pData, nData, varDataVal(t), NULL);
991
  if (len < 0) {
7,632✔
992
    taosMemoryFree(t);
×
993
    return TSDB_CODE_SCALAR_CONVERT_ERROR;
×
994
  }
995
  varDataSetLen(t, len);
7,632✔
996

997
  *ppOut = taosMemoryCalloc(1, len + VARSTR_HEADER_SIZE);
7,632✔
998
  memcpy(*ppOut, t, len + VARSTR_HEADER_SIZE);
7,632✔
999

1000
_return:
7,632✔
1001
  taosMemoryFree(t);
7,632✔
1002
  return code;
7,632✔
1003
}
1004

1005
static int32_t buildTagDataEntryKey(const SArray* pColIds, const STag* pTag,
36,729✔
1006
  const SSchemaWrapper* pTagScheam, T_MD5_CTX* pContext) {
1007
  int32_t code = TSDB_CODE_SUCCESS;
36,729✔
1008
  int32_t lino = 0;
36,729✔
1009
  int32_t keyLen = 0;
36,729✔
1010
  char* pKey = NULL;
36,729✔
1011
  // get length first
1012
  for (int32_t i = 0; i < taosArrayGetSize(pColIds); i++) {
108,279✔
1013
    STagVal pTagValue = {.cid = *(col_id_t*)taosArrayGet(pColIds, i)};
73,299✔
1014
    if (tTagGet(pTag, &pTagValue)) {
73,299✔
1015
      keyLen += sizeof(col_id_t);
71,550✔
1016
      if (IS_VAR_DATA_TYPE(pTagValue.type)) {
71,550✔
1017
        int32_t varLen = getTagColSize(
29,892✔
1018
          pTagScheam->pSchema, pTagScheam->nCols, pTagValue.cid);
14,946✔
1019
        code = varLen > 0 ? TSDB_CODE_SUCCESS : TSDB_CODE_NOT_FOUND;
14,946✔
1020
        QUERY_CHECK_CODE(code, lino, _end);
14,946✔
1021
        keyLen += varLen;
14,946✔
1022
      } else {
1023
        keyLen += tDataTypes[pTagValue.type].bytes;
56,604✔
1024
      }
1025
    } else {
1026
      // tag value not found
1027
      code = TSDB_CODE_NOT_FOUND;
1,749✔
1028
      QUERY_CHECK_CODE(code, lino, _end);
1,749✔
1029
    }
1030
  }
1031

1032
  pKey = taosMemoryCalloc(1, keyLen);
34,980✔
1033
  if (NULL == pKey) {
34,980✔
1034
    code = terrno;
×
1035
    return code;
×
1036
  }
1037

1038
  // build the key
1039
  char* pStart = pKey;
34,980✔
1040
  for (int32_t i = 0; i < taosArrayGetSize(pColIds); i++) {
104,940✔
1041
    STagVal pTagValue = {.cid = *(col_id_t*)taosArrayGet(pColIds, i)};
69,960✔
1042
    if (tTagGet(pTag, &pTagValue)) {
69,960✔
1043
      // copy cid
1044
      memcpy(pStart, &pTagValue.cid, sizeof(col_id_t));
69,960✔
1045
      pStart += sizeof(col_id_t);
69,960✔
1046
      // copy value
1047
      if (IS_VAR_DATA_TYPE(pTagValue.type) && pTagValue.pData != NULL) {
69,960✔
1048
        if (TSDB_DATA_TYPE_NCHAR == pTagValue.type) {
14,946✔
1049
          // need to convert nchar to var
1050
          char *pVar = NULL;
7,632✔
1051
          code = ncharToVar((char *)pTagValue.pData, pTagValue.nData, &pVar);
7,632✔
1052
          QUERY_CHECK_CODE(code, lino, _end);
7,632✔
1053
          memcpy(pStart, varDataVal(pVar), varDataLen(pVar));
7,632✔
1054
          pStart += varDataLen(pVar);
7,632✔
1055
          taosMemoryFree(pVar);
7,632✔
1056
        } else {
1057
          memcpy(pStart, pTagValue.pData, pTagValue.nData);
7,314✔
1058
          pStart += pTagValue.nData;
7,314✔
1059
        }
1060
      } else {
1061
        memcpy(pStart, &pTagValue.i64, tDataTypes[pTagValue.type].bytes);
55,014✔
1062
        pStart += tDataTypes[pTagValue.type].bytes;
55,014✔
1063
      }
1064
    } else {
1065
      // tag value not found
1066
      code = TSDB_CODE_NOT_FOUND;
×
1067
      QUERY_CHECK_CODE(code, lino, _end);
×
1068
    }
1069
  }
1070

1071
  // update MD5
1072
  tMD5Init(pContext);
34,980✔
1073
  tMD5Update(pContext, (uint8_t*)pKey, (uint32_t)keyLen);
34,980✔
1074
  tMD5Final(pContext);
34,980✔
1075

1076
_end:
36,729✔
1077
  taosMemFreeClear(pKey);
36,729✔
1078
  return code;
36,729✔
1079
}
1080

1081
// remove the dropped table uid from all cache entries
1082
// pDroppedTable is the dropped child table meta entry
1083
int32_t metaStableTagFilterCacheUpdateUid(SMeta* pMeta,
75,637,649✔
1084
  const SMetaEntry* pChildTable, const SMetaEntry* pSuperTable,
1085
  ETagFilterCacheAction action) {
1086
  if (pMeta == NULL || pChildTable == NULL || pSuperTable == NULL) {
75,637,649✔
1087
    return TSDB_CODE_INVALID_PARA;
×
1088
  }
1089
  int32_t   lino = 0;
75,647,706✔
1090
  int32_t   code = TSDB_CODE_SUCCESS;
75,647,706✔
1091
  SHashObj* pTableEntry = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
75,647,706✔
1092
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
75,639,442✔
1093

1094
  code = taosThreadRwlockWrlock(pRwlock);
75,657,740✔
1095
  TSDB_CHECK_CODE(code, lino, _end);
75,644,609✔
1096

1097
  tb_uid_t suid = pChildTable->ctbEntry.suid;;
75,644,609✔
1098
  STagConds** pTagConds =
1099
    (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(tb_uid_t));
75,623,628✔
1100
  if (pTagConds != NULL) {
75,650,842✔
1101
    STagCondFilterEntry** ppFilterEntry = NULL;
8,109✔
1102
    while ((ppFilterEntry = taosHashIterate((*pTagConds)->set, ppFilterEntry))) {
43,089✔
1103
      STagCondFilterEntry* pFilterEntry = *ppFilterEntry;
36,729✔
1104
      // rebuild the tagCondKey and check existence
1105
      SArray* pColIds = pFilterEntry->pColIds;
36,729✔
1106
      // rebuild the tagCondFilterKey
1107
      int32_t keyLen = 0;
36,729✔
1108
      char*   pKey = NULL;
36,729✔
1109
      T_MD5_CTX context = {0};
36,729✔
1110
      code = buildTagDataEntryKey(pColIds, (STag*)pChildTable->ctbEntry.pTags, 
36,729✔
1111
        &pSuperTable->stbEntry.schemaTag, &context);
1112
      if (code != TSDB_CODE_SUCCESS) {
36,729✔
1113
        metaError("vgId:%d, suid:%" PRIu64 " failed to build tag condition"
1,749✔
1114
          " key for dropped table uid:%" PRIu64 " since %s",
1115
          TD_VID(pMeta->pVnode), suid, pChildTable->uid, tstrerror(code));
1116
        goto _end;
1,749✔
1117
      }
1118

1119
      SArray** pArray = (SArray**)taosHashGet(
34,980✔
1120
        pFilterEntry->set, context.digest, tListLen(context.digest));
1121
      if (pArray != NULL) {
34,980✔
1122
        // check and remove the dropped table uid from the array
1123
        // TODO(Tony Zhang): optimize this scan
1124
        if (action == STABLE_TAG_FILTER_CACHE_DROP_TABLE) {
4,770✔
1125
          for (int32_t i = 0; i < taosArrayGetSize(*pArray); i++) {
4,770✔
1126
            uint64_t uid = *(uint64_t*)taosArrayGet(*pArray, i);
4,770✔
1127
            if (uid == pChildTable->uid) {
4,770✔
1128
              taosArrayRemove(*pArray, i);
2,385✔
1129
              metaDebug("vgId:%d, suid:%" PRIu64
2,385✔
1130
                " removed dropped table uid:%" PRIu64
1131
                " from stable tag filter cache",
1132
                TD_VID(pMeta->pVnode), suid, pChildTable->uid);
1133
              break;
2,385✔
1134
            } 
1135
          }
1136
        } else {
1137
          // STABLE_TAG_FILTER_CACHE_ADD_TABLE
1138
          void* _tmp = taosArrayPush(*pArray, &pChildTable->uid);
4,770✔
1139
        }
1140
      }
1141
    }
1142
  }
1143

1144
_end:
75,650,842✔
1145
  if (TSDB_CODE_SUCCESS != code) {
75,640,821✔
1146
    metaError("vgId:%d, %s failed at %s:%d since %s",
1,749✔
1147
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
1148
  } else {
1149
    metaDebug(
75,639,072✔
1150
      "vgId:%d, suid:%" PRIu64 " update table uid:%" PRIu64
1151
      " in stable tag filter cache, action:%d",
1152
      TD_VID(pMeta->pVnode),
1153
      pChildTable->ctbEntry.suid, pChildTable->uid, action);
1154
  }
1155
  code = taosThreadRwlockUnlock(pRwlock);
75,641,854✔
1156
  if (TSDB_CODE_SUCCESS != code) {
75,645,497✔
1157
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
1158
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
1159
  }
1160
  return code;
75,644,356✔
1161
}
1162

1163
int32_t metaStableTagFilterCacheDropTag(
210,624✔
1164
  SMeta* pMeta, tb_uid_t suid, col_id_t cid) {
1165
  if (pMeta == NULL) {
210,624✔
1166
    return TSDB_CODE_INVALID_PARA;
×
1167
  }
1168
  int32_t   lino = 0;
210,624✔
1169
  int32_t   code = TSDB_CODE_SUCCESS;
210,624✔
1170
  SHashObj* pTableEntry = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
210,624✔
1171
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
208,160✔
1172

1173
  code = taosThreadRwlockWrlock(pRwlock);
209,392✔
1174
  TSDB_CHECK_CODE(code, lino, _end);
209,392✔
1175

1176
  STagConds** pTagConds =
1177
    (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(tb_uid_t));
209,392✔
1178
  if (pTagConds != NULL) {
209,675✔
1179
    void* pIter = taosHashIterate((*pTagConds)->set, NULL);
×
1180
    while (pIter) {
×
1181
      STagCondFilterEntry* pFilterEntry = *(STagCondFilterEntry**)pIter;
×
1182
      bool found = false;
×
1183
      for (int32_t i = 0; i < taosArrayGetSize(pFilterEntry->pColIds); i++) {
×
1184
        col_id_t existCid = *(col_id_t*)taosArrayGet(pFilterEntry->pColIds, i);
×
1185
        if (existCid == cid) {
×
1186
          found = true;
×
1187
          break;
×
1188
        }
1189
      }
1190
      if (found) {
×
1191
        uint32_t numEntries = taosHashGetSize(pFilterEntry->set);
×
1192
        size_t keyLen = 0;
×
1193
        char  *key = (char *)taosHashGetKey(pIter, &keyLen);
×
1194
        code = taosHashRemove((*pTagConds)->set, key, keyLen);
×
1195
        TSDB_CHECK_CODE(code, lino, _end);
×
1196
        (*pTagConds)->numTagDataEntries -= numEntries;
×
1197
        pMeta->pCache->sStableTagFilterResCache.numTagDataEntries -= numEntries;
×
1198
      }
UNCOV
1199
      pIter = taosHashIterate((*pTagConds)->set, pIter);
×
1200
    }
1201
  }
1202
_end:
209,675✔
1203
  if (TSDB_CODE_SUCCESS != code) {
210,624✔
1204
    metaError("vgId:%d, %s failed at %s:%d since %s",
×
1205
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
1206
  } else {
1207
    metaDebug(
210,624✔
1208
      "vgId:%d, suid:%" PRIu64 " dropped tag cid:%d "
1209
      "from stable tag filter cache",
1210
      TD_VID(pMeta->pVnode), suid, cid);
1211
  }
1212
  code = taosThreadRwlockUnlock(pRwlock);
210,624✔
1213
  if (TSDB_CODE_SUCCESS != code) {
210,008✔
1214
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
1215
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
1216
  }
1217
  return code;
209,392✔
1218
}
1219

1220
void metaCacheClear(SMeta* pMeta) {
3,750,039✔
1221
  metaWLock(pMeta);
3,750,039✔
1222
  metaCacheClose(pMeta);
3,750,039✔
1223
  (void)metaCacheOpen(pMeta);
3,750,039✔
1224
  metaULock(pMeta);
3,750,039✔
1225
}
3,750,039✔
1226

1227
// remove the lru cache that are expired due to the tags value update, or creating, or dropping, of child tables
1228
int32_t metaUidCacheClear(SMeta* pMeta, uint64_t suid) {
68,247,725✔
1229
  uint64_t  p[4] = {0};
68,247,725✔
1230
  int32_t   vgId = TD_VID(pMeta->pVnode);
68,251,541✔
1231
  SHashObj* pEntryHashMap = pMeta->pCache->sTagFilterResCache.pTableEntry;
68,241,696✔
1232

1233
  uint64_t dummy[2] = {0};
68,248,756✔
1234
  initCacheKey(p, pEntryHashMap, suid, (char*)&dummy[0], 16);
68,256,686✔
1235

1236
  TdThreadMutex* pLock = &pMeta->pCache->sTagFilterResCache.lock;
68,244,302✔
1237
  (void)taosThreadMutexLock(pLock);
68,240,682✔
1238

1239
  STagFilterResEntry** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
68,255,894✔
1240
  if (pEntry == NULL || taosHashGetSize((*pEntry)->set) == 0) {
68,262,213✔
1241
    (void)taosThreadMutexUnlock(pLock);
68,260,459✔
1242
    return TSDB_CODE_SUCCESS;
68,260,930✔
1243
  }
1244

1245
  (*pEntry)->hitTimes = 0;
1,754✔
1246

1247
  char *iter = taosHashIterate((*pEntry)->set, NULL);
1,754✔
1248
  while (iter != NULL) {
3,508✔
1249
    setMD5DigestInKey(p, iter, 2 * sizeof(uint64_t));
1250
    taosLRUCacheErase(pMeta->pCache->sTagFilterResCache.pUidResCache, p, TAG_FILTER_RES_KEY_LEN);
1,754✔
1251
    iter = taosHashIterate((*pEntry)->set, iter);
1,754✔
1252
  }
1253
  taosHashClear((*pEntry)->set);
1,754✔
1254
  (void)taosThreadMutexUnlock(pLock);
1,754✔
1255

1256
  metaDebug("vgId:%d suid:%" PRId64 " cached related tag filter uid list cleared", vgId, suid);
1,754✔
1257
  return TSDB_CODE_SUCCESS;
1,754✔
1258
}
1259

1260
int32_t metaGetCachedTbGroup(void* pVnode, tb_uid_t suid, const uint8_t* pKey, int32_t keyLen, SArray** pList) {
×
1261
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
×
1262
  int32_t vgId = TD_VID(pMeta->pVnode);
×
1263

1264
  // generate the composed key for LRU cache
1265
  SLRUCache*     pCache = pMeta->pCache->STbGroupResCache.pResCache;
×
1266
  SHashObj*      pTableMap = pMeta->pCache->STbGroupResCache.pTableEntry;
×
1267
  TdThreadMutex* pLock = &pMeta->pCache->STbGroupResCache.lock;
×
1268

1269
  *pList = NULL;
×
1270
  uint64_t key[4];
×
1271
  initCacheKey(key, pTableMap, suid, (const char*)pKey, keyLen);
×
1272

1273
  (void)taosThreadMutexLock(pLock);
×
1274
  pMeta->pCache->STbGroupResCache.accTimes += 1;
×
1275

1276
  LRUHandle* pHandle = taosLRUCacheLookup(pCache, key, TAG_FILTER_RES_KEY_LEN);
×
1277
  if (pHandle == NULL) {
×
1278
    (void)taosThreadMutexUnlock(pLock);
×
1279
    return TSDB_CODE_SUCCESS;
×
1280
  }
1281

1282
  STagFilterResEntry** pEntry = taosHashGet(pTableMap, &suid, sizeof(uint64_t));
×
1283
  if (NULL == pEntry) {
×
1284
    metaDebug("suid %" PRIu64 " not in tb group cache", suid);
×
1285
    return TSDB_CODE_NOT_FOUND;
×
1286
  }
1287

1288
  *pList = taosArrayDup(taosLRUCacheValue(pCache, pHandle), NULL);
×
1289

1290
  (*pEntry)->hitTimes += 1;
×
1291

1292
  uint32_t acc = pMeta->pCache->STbGroupResCache.accTimes;
×
1293
  if ((*pEntry)->hitTimes % 5000 == 0 && (*pEntry)->hitTimes > 0) {
×
1294
    metaInfo("vgId:%d tb group cache hit:%d, total acc:%d, rate:%.2f", vgId, (*pEntry)->hitTimes, acc,
×
1295
             ((double)(*pEntry)->hitTimes) / acc);
1296
  }
1297

1298
  bool ret = taosLRUCacheRelease(pCache, pHandle, false);
×
1299

1300
  // unlock meta
1301
  (void)taosThreadMutexUnlock(pLock);
×
1302
  return TSDB_CODE_SUCCESS;
×
1303
}
1304

1305
static void freeTbGroupCachePayload(const void* key, size_t keyLen, void* value, void* ud) {
×
1306
  (void)ud;
1307
  if (value == NULL) {
×
1308
    return;
×
1309
  }
1310

1311
  const uint64_t* p = key;
×
1312
  if (keyLen != sizeof(int64_t) * 4) {
×
1313
    metaError("tb group key length is invalid, length:%d, expect:%d", (int32_t)keyLen, (int32_t)sizeof(uint64_t) * 2);
×
1314
    return;
×
1315
  }
1316

1317
  SHashObj* pHashObj = (SHashObj*)p[0];
×
1318

1319
  STagFilterResEntry** pEntry = taosHashGet(pHashObj, &p[1], sizeof(uint64_t));
×
1320

1321
  if (pEntry != NULL && (*pEntry) != NULL) {
×
1322
    int64_t st = taosGetTimestampUs();
×
1323
    int32_t code = taosHashRemove((*pEntry)->set, &p[2], sizeof(uint64_t) * 2);
×
1324
    if (code == TSDB_CODE_SUCCESS) {
×
1325
      double el = (taosGetTimestampUs() - st) / 1000.0;
×
1326
      metaDebug("clear one item in tb group cache, remain cached item:%d, elapsed time:%.2fms",
×
1327
                taosHashGetSize((*pEntry)->set), el);
1328
    }
1329
  }
1330

1331
  taosArrayDestroy((SArray*)value);
×
1332
}
1333

1334
int32_t metaPutTbGroupToCache(void* pVnode, uint64_t suid, const void* pKey, int32_t keyLen, void* pPayload,
×
1335
                              int32_t payloadLen) {
1336
  int32_t code = 0;
×
1337
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
×
1338
  int32_t vgId = TD_VID(pMeta->pVnode);
×
1339

1340
  if (payloadLen > tsTagFilterResCacheSize) {
×
1341
    metaDebug("vgId:%d, suid:%" PRIu64
×
1342
              " ignore to add to tb group cache, due to payload length %d greater than threshold %d",
1343
              vgId, suid, payloadLen, tsTagFilterResCacheSize);
1344
    taosArrayDestroy((SArray*)pPayload);
×
1345
    return TSDB_CODE_SUCCESS;
×
1346
  }
1347

1348
  SLRUCache*     pCache = pMeta->pCache->STbGroupResCache.pResCache;
×
1349
  SHashObj*      pTableEntry = pMeta->pCache->STbGroupResCache.pTableEntry;
×
1350
  TdThreadMutex* pLock = &pMeta->pCache->STbGroupResCache.lock;
×
1351

1352
  uint64_t key[4] = {0};
×
1353
  initCacheKey(key, pTableEntry, suid, pKey, keyLen);
×
1354

1355
  (void)taosThreadMutexLock(pLock);
×
1356
  STagFilterResEntry** pEntry = taosHashGet(pTableEntry, &suid, sizeof(uint64_t));
×
1357
  if (pEntry == NULL) {
×
1358
    code = addNewEntry(pTableEntry, pKey, keyLen, suid);
×
1359
    if (code != TSDB_CODE_SUCCESS) {
×
1360
      goto _end;
×
1361
    }
1362
  } else {  // check if it exists or not
1363
    code = taosHashPut((*pEntry)->set, pKey, keyLen, NULL, 0);
×
1364
    if (code == TSDB_CODE_DUP_KEY) {
×
1365
      // we have already found the existed items, no need to added to cache anymore.
1366
      (void)taosThreadMutexUnlock(pLock);
×
1367
      return TSDB_CODE_SUCCESS;
×
1368
    }
1369
    if (code != TSDB_CODE_SUCCESS) {
×
1370
      goto _end;
×
1371
    }
1372
  }
1373

1374
  // add to cache.
1375
  (void)taosLRUCacheInsert(pCache, key, TAG_FILTER_RES_KEY_LEN, pPayload, payloadLen, freeTbGroupCachePayload, NULL, NULL,
×
1376
                           TAOS_LRU_PRIORITY_LOW, NULL);
1377
_end:
×
1378
  (void)taosThreadMutexUnlock(pLock);
×
1379
  metaDebug("vgId:%d, suid:%" PRIu64 " tb group added into cache, total:%d, tables:%d", vgId, suid,
×
1380
            (int32_t)taosLRUCacheGetUsage(pCache), taosHashGetSize(pTableEntry));
1381

1382
  return code;
×
1383
}
1384

1385
// remove the lru cache that are expired due to the tags value update, or creating, or dropping, of child tables
1386
int32_t metaTbGroupCacheClear(SMeta* pMeta, uint64_t suid) {
68,240,910✔
1387
  uint64_t  p[4] = {0};
68,240,910✔
1388
  int32_t   vgId = TD_VID(pMeta->pVnode);
68,242,853✔
1389
  SHashObj* pEntryHashMap = pMeta->pCache->STbGroupResCache.pTableEntry;
68,240,585✔
1390

1391
  uint64_t dummy[2] = {0};
68,240,576✔
1392
  initCacheKey(p, pEntryHashMap, suid, (char*)&dummy[0], 16);
68,255,297✔
1393

1394
  TdThreadMutex* pLock = &pMeta->pCache->STbGroupResCache.lock;
68,253,349✔
1395
  (void)taosThreadMutexLock(pLock);
68,251,342✔
1396

1397
  STagFilterResEntry** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
68,264,055✔
1398
  if (pEntry == NULL || taosHashGetSize((*pEntry)->set) == 0) {
68,264,850✔
1399
    (void)taosThreadMutexUnlock(pLock);
68,264,850✔
1400
    return TSDB_CODE_SUCCESS;
68,266,938✔
1401
  }
1402

1403
  (*pEntry)->hitTimes = 0;
×
1404

1405
  char *iter = taosHashIterate((*pEntry)->set, NULL);
×
1406
  while (iter != NULL) {
×
1407
    setMD5DigestInKey(p, iter, 2 * sizeof(uint64_t));
1408
    taosLRUCacheErase(pMeta->pCache->STbGroupResCache.pResCache, p, TAG_FILTER_RES_KEY_LEN);
×
1409
    iter = taosHashIterate((*pEntry)->set, iter);
×
1410
  }
1411
  taosHashClear((*pEntry)->set);
×
1412
  (void)taosThreadMutexUnlock(pLock);
×
1413

1414
  metaDebug("vgId:%d suid:%" PRId64 " cached related tb group cleared", vgId, suid);
×
1415
  return TSDB_CODE_SUCCESS;
×
1416
}
1417

1418
bool metaTbInFilterCache(SMeta* pMeta, const void* key, int8_t type) {
117,856,867✔
1419
  if (type == 0 && taosHashGet(pMeta->pCache->STbFilterCache.pStb, key, sizeof(tb_uid_t))) {
117,856,867✔
1420
    return true;
1,370✔
1421
  }
1422

1423
  if (type == 1 && taosHashGet(pMeta->pCache->STbFilterCache.pStbName, key, strlen(key))) {
117,855,497✔
1424
    return true;
×
1425
  }
1426

1427
  return false;
117,874,799✔
1428
}
1429

1430
int32_t metaPutTbToFilterCache(SMeta* pMeta, const void* key, int8_t type) {
8,220✔
1431
  if (type == 0) {
8,220✔
1432
    return taosHashPut(pMeta->pCache->STbFilterCache.pStb, key, sizeof(tb_uid_t), NULL, 0);
3,014✔
1433
  }
1434

1435
  if (type == 1) {
5,206✔
1436
    return taosHashPut(pMeta->pCache->STbFilterCache.pStbName, key, strlen(key), NULL, 0);
5,206✔
1437
  }
1438

1439
  return 0;
×
1440
}
1441

1442
int32_t metaSizeOfTbFilterCache(SMeta* pMeta, int8_t type) {
3,836✔
1443
  if (type == 0) {
3,836✔
1444
    return taosHashGetSize(pMeta->pCache->STbFilterCache.pStb);
3,836✔
1445
  }
1446
  return 0;
×
1447
}
1448

1449
int32_t metaInitTbFilterCache(SMeta* pMeta) {
3,753,943✔
1450
#ifdef TD_ENTERPRISE
1451
  int32_t      tbNum = 0;
3,753,943✔
1452
  const char** pTbArr = NULL;
3,753,943✔
1453
  const char*  dbName = NULL;
3,753,943✔
1454

1455
  if (!(dbName = strchr(pMeta->pVnode->config.dbname, '.'))) return 0;
3,753,943✔
1456
  if (0 == strncmp(++dbName, "log", TSDB_DB_NAME_LEN)) {
3,754,131✔
1457
    tbNum = tkLogStbNum;
274✔
1458
    pTbArr = (const char**)&tkLogStb;
274✔
1459
  } else if (0 == strncmp(dbName, "audit", TSDB_DB_NAME_LEN)) {
3,753,946✔
1460
    tbNum = tkAuditStbNum;
×
1461
    pTbArr = (const char**)&tkAuditStb;
×
1462
  }
1463
  if (tbNum && pTbArr) {
3,754,220✔
1464
    for (int32_t i = 0; i < tbNum; ++i) {
5,480✔
1465
      TAOS_CHECK_RETURN(metaPutTbToFilterCache(pMeta, pTbArr[i], 1));
5,206✔
1466
    }
1467
  }
1468
#else
1469
#endif
1470
  return 0;
3,754,220✔
1471
}
1472

1473
int64_t metaGetStbKeep(SMeta* pMeta, int64_t uid) {
124,621✔
1474
  SMetaStbStats stats = {0};
124,621✔
1475

1476
  if (metaStatsCacheGet(pMeta, uid, &stats) == TSDB_CODE_SUCCESS) {
124,621✔
1477
    return stats.keep;
113,824✔
1478
  }
1479

1480
  SMetaEntry* pEntry = NULL;
10,797✔
1481
  if (metaFetchEntryByUid(pMeta, uid, &pEntry) == TSDB_CODE_SUCCESS) {
10,797✔
1482
    int64_t keep = -1;
10,220✔
1483
    if (pEntry->type == TSDB_SUPER_TABLE) {
10,220✔
1484
      keep = pEntry->stbEntry.keep;
10,220✔
1485
    }
1486
    metaFetchEntryFree(&pEntry);
10,220✔
1487
    return keep;
10,220✔
1488
  }
1489
  
1490
  return -1;
577✔
1491
}
1492

1493
int32_t metaRefDbsCacheClear(SMeta* pMeta, uint64_t suid) {
255,990✔
1494
  int32_t        code = TSDB_CODE_SUCCESS;
255,990✔
1495
  int32_t        vgId = TD_VID(pMeta->pVnode);
255,990✔
1496
  SHashObj*      pEntryHashMap = pMeta->pCache->STbRefDbCache.pStbRefs;
255,990✔
1497
  TdThreadMutex* pLock = &pMeta->pCache->STbRefDbCache.lock;
255,990✔
1498

1499
  (void)taosThreadMutexLock(pLock);
255,990✔
1500

1501
  SHashObj** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
255,990✔
1502
  if (pEntry == NULL) {
255,990✔
1503
    goto _return;
254,000✔
1504
  }
1505

1506
  code = taosHashRemove(pEntryHashMap, &suid, sizeof(uint64_t));
1,990✔
1507

1508
  metaDebug("vgId:%d suid:%" PRId64 " cached virtual stable ref db cleared", vgId, suid);
1,990✔
1509

1510
_return:
1,578✔
1511
  (void)taosThreadMutexUnlock(pLock);
255,990✔
1512
  return code;
255,990✔
1513
}
1514

1515
int32_t metaGetCachedRefDbs(void* pVnode, tb_uid_t suid, SArray* pList) {
46,576✔
1516
  int32_t        code = TSDB_CODE_SUCCESS;
46,576✔
1517
  int32_t        line = 0;
46,576✔
1518
  SMeta*         pMeta = ((SVnode*)pVnode)->pMeta;
46,576✔
1519
  SHashObj*      pTableMap = pMeta->pCache->STbRefDbCache.pStbRefs;
46,576✔
1520
  TdThreadMutex* pLock = &pMeta->pCache->STbRefDbCache.lock;
46,576✔
1521

1522
  (void)taosThreadMutexLock(pLock);
46,576✔
1523

1524
  SHashObj** pEntry = taosHashGet(pTableMap, &suid, sizeof(uint64_t));
46,576✔
1525
  if (pEntry) {
46,576✔
1526
    void *iter = taosHashIterate(*pEntry, NULL);
29,479✔
1527
    while (iter != NULL) {
56,013✔
1528
      size_t   dbNameLen = 0;
26,534✔
1529
      char*    name = NULL;
26,534✔
1530
      char*    dbName = NULL;
26,534✔
1531
      name = taosHashGetKey(iter, &dbNameLen);
26,534✔
1532
      TSDB_CHECK_NULL(name, code, line, _return, terrno);
26,534✔
1533
      dbName = taosMemoryMalloc(dbNameLen + 1);
26,534✔
1534
      TSDB_CHECK_NULL(dbName, code, line, _return, terrno);
26,534✔
1535
      tstrncpy(dbName, name, dbNameLen + 1);
26,534✔
1536
      TSDB_CHECK_NULL(taosArrayPush(pList, &dbName), code, line, _return, terrno);
26,534✔
1537
      iter = taosHashIterate(*pEntry, iter);
26,534✔
1538
    }
1539
  }
1540

1541
_return:
46,576✔
1542
  if (code) {
46,576✔
1543
    metaError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
1544
  }
1545
  (void)taosThreadMutexUnlock(pLock);
46,576✔
1546
  return code;
46,576✔
1547
}
1548

1549
static int32_t addRefDbsCacheNewEntry(SHashObj* pRefDbs, uint64_t suid, SHashObj **pEntry) {
17,097✔
1550
  int32_t      code = TSDB_CODE_SUCCESS;
17,097✔
1551
  int32_t      lino = 0;
17,097✔
1552
  SHashObj*    p = NULL;
17,097✔
1553

1554
  p = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
17,097✔
1555
  TSDB_CHECK_NULL(p, code, lino, _end, terrno);
17,097✔
1556

1557
  code = taosHashPut(pRefDbs, &suid, sizeof(uint64_t), &p, POINTER_BYTES);
17,097✔
1558
  TSDB_CHECK_CODE(code, lino, _end);
17,097✔
1559

1560
  *pEntry = p;
17,097✔
1561

1562
_end:
17,097✔
1563
  if (code != TSDB_CODE_SUCCESS) {
17,097✔
1564
    metaError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1565
  }
1566
  return code;
17,097✔
1567
}
1568

1569
int32_t metaPutRefDbsToCache(void* pVnode, tb_uid_t suid, SArray* pList) {
20,042✔
1570
  int32_t        code = 0;
20,042✔
1571
  int32_t        line = 0;
20,042✔
1572
  SMeta*         pMeta = ((SVnode*)pVnode)->pMeta;
20,042✔
1573
  SHashObj*      pStbRefs = pMeta->pCache->STbRefDbCache.pStbRefs;
20,042✔
1574
  TdThreadMutex* pLock = &pMeta->pCache->STbRefDbCache.lock;
20,042✔
1575

1576
  (void)taosThreadMutexLock(pLock);
20,042✔
1577

1578
  SHashObj*  pEntry = NULL;
20,042✔
1579
  SHashObj** find = taosHashGet(pStbRefs, &suid, sizeof(uint64_t));
20,042✔
1580
  if (find == NULL) {
20,042✔
1581
    code = addRefDbsCacheNewEntry(pStbRefs, suid, &pEntry);
17,097✔
1582
    TSDB_CHECK_CODE(code, line, _return);
17,097✔
1583
  } else {  // check if it exists or not
1584
    pEntry = *find;
2,945✔
1585
  }
1586

1587
  for (int32_t i = 0; i < taosArrayGetSize(pList); i++) {
33,709✔
1588
    char* dbName = taosArrayGetP(pList, i);
13,667✔
1589
    void* pItem = taosHashGet(pEntry, dbName, strlen(dbName));
13,667✔
1590
    if (pItem == NULL) {
13,667✔
1591
      code = taosHashPut(pEntry, dbName, strlen(dbName), NULL, 0);
13,667✔
1592
      TSDB_CHECK_CODE(code, line, _return);
13,667✔
1593
    }
1594
  }
1595

1596
_return:
20,042✔
1597
  if (code) {
20,042✔
1598
    metaError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
1599
  }
1600
  (void)taosThreadMutexUnlock(pLock);
20,042✔
1601

1602
  return code;
20,042✔
1603
}
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