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

taosdata / TDengine / #4911

04 Jan 2026 09:05AM UTC coverage: 65.028% (-0.8%) from 65.864%
#4911

push

travis-ci

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

1206 of 4524 new or added lines in 22 files covered. (26.66%)

1517 existing lines in 134 files now uncovered.

195276 of 300296 relevant lines covered (65.03%)

116931714.52 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,195,904✔
114
  if (pMeta->pCache) {
7,195,904✔
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;
82,303,683✔
120
        taosMemoryFree(pEntry);
82,306,204✔
121
        pEntry = tEntry;
82,320,647✔
122
      }
123
    }
124
    taosMemoryFree(pMeta->pCache->sEntryCache.aBucket);
7,195,904✔
125
  }
126
}
7,195,904✔
127

128
static void statsCacheClose(SMeta* pMeta) {
7,195,904✔
129
  if (pMeta->pCache) {
7,195,904✔
130
    // close entry cache
131
    for (int32_t iBucket = 0; iBucket < pMeta->pCache->sStbStatsCache.nBucket; iBucket++) {
123,873,531✔
132
      SMetaStbStatsEntry* pEntry = pMeta->pCache->sStbStatsCache.aBucket[iBucket];
116,676,947✔
133
      while (pEntry) {
120,256,948✔
134
        SMetaStbStatsEntry* tEntry = pEntry->next;
3,579,321✔
135
        taosMemoryFree(pEntry);
3,579,321✔
136
        pEntry = tEntry;
3,579,321✔
137
      }
138
    }
139
    taosMemoryFree(pMeta->pCache->sStbStatsCache.aBucket);
7,195,836✔
140
  }
141
}
7,195,904✔
142

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

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

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

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

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

172
  pMeta->pCache = (SMetaCache*)taosMemoryCalloc(1, sizeof(SMetaCache));
7,183,851✔
173
  if (pMeta->pCache == NULL) {
7,191,720✔
174
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
175
  }
176

177
  // open entry cache
178
  pMeta->pCache->sEntryCache.nEntry = 0;
7,192,368✔
179
  pMeta->pCache->sEntryCache.nBucket = META_CACHE_BASE_BUCKET;
7,193,700✔
180
  pMeta->pCache->sEntryCache.aBucket =
7,196,678✔
181
      (SMetaCacheEntry**)taosMemoryCalloc(pMeta->pCache->sEntryCache.nBucket, sizeof(SMetaCacheEntry*));
7,194,347✔
182
  if (pMeta->pCache->sEntryCache.aBucket == NULL) {
7,195,904✔
183
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
184
  }
185

186
  // open stats cache
187
  pMeta->pCache->sStbStatsCache.nEntry = 0;
7,195,904✔
188
  pMeta->pCache->sStbStatsCache.nBucket = META_CACHE_STATS_BUCKET;
7,195,904✔
189
  pMeta->pCache->sStbStatsCache.aBucket =
7,196,521✔
190
      (SMetaStbStatsEntry**)taosMemoryCalloc(pMeta->pCache->sStbStatsCache.nBucket, sizeof(SMetaStbStatsEntry*));
7,195,904✔
191
  if (pMeta->pCache->sStbStatsCache.aBucket == NULL) {
7,195,747✔
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,195,904✔
197
  if (pMeta->pCache->sTagFilterResCache.pUidResCache == NULL) {
7,195,904✔
198
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
199
  }
200

201
  pMeta->pCache->sTagFilterResCache.accTimes = 0;
7,195,904✔
202
  pMeta->pCache->sTagFilterResCache.pTableEntry =
14,391,808✔
203
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
14,391,034✔
204
  if (pMeta->pCache->sTagFilterResCache.pTableEntry == NULL) {
7,195,904✔
205
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
206
  }
207

208
  taosHashSetFreeFp(pMeta->pCache->sTagFilterResCache.pTableEntry, freeCacheEntryFp);
7,195,904✔
209
  (void)taosThreadMutexInit(&pMeta->pCache->sTagFilterResCache.lock, NULL);
7,195,904✔
210

211
  // open stable tag filter cache
212
  pMeta->pCache->sStableTagFilterResCache.accTimes = 0;
7,195,904✔
213
  pMeta->pCache->sStableTagFilterResCache.numTagDataEntries = 0;
7,195,904✔
214
  pMeta->pCache->sStableTagFilterResCache.pTableEntry =
14,391,808✔
215
    taosHashInit(1024,
14,391,034✔
216
      taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
217
  if (pMeta->pCache->sStableTagFilterResCache.pTableEntry == NULL) {
7,195,904✔
218
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
219
  }
220
  taosHashSetFreeFp(
7,195,904✔
221
    pMeta->pCache->sStableTagFilterResCache.pTableEntry, freeTagCondsFp);
7,195,904✔
222

223
  TAOS_UNUSED(taosThreadRwlockInit(
7,195,904✔
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,195,904✔
228
  if (pMeta->pCache->STbGroupResCache.pResCache == NULL) {
7,195,904✔
229
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
230
  }
231

232
  pMeta->pCache->STbGroupResCache.accTimes = 0;
7,195,904✔
233
  pMeta->pCache->STbGroupResCache.pTableEntry =
14,391,808✔
234
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
14,391,034✔
235
  if (pMeta->pCache->STbGroupResCache.pTableEntry == NULL) {
7,195,904✔
236
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
237
  }
238

239
  taosHashSetFreeFp(pMeta->pCache->STbGroupResCache.pTableEntry, freeCacheEntryFp);
7,195,904✔
240
  (void)taosThreadMutexInit(&pMeta->pCache->STbGroupResCache.lock, NULL);
7,195,904✔
241

242
  // open filter cache
243
  pMeta->pCache->STbFilterCache.pStb =
14,391,808✔
244
      taosHashInit(0, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
14,391,034✔
245
  if (pMeta->pCache->STbFilterCache.pStb == NULL) {
7,195,904✔
246
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
247
  }
248

249
  pMeta->pCache->STbFilterCache.pStbName =
14,391,565✔
250
      taosHashInit(0, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), false, HASH_NO_LOCK);
14,390,062✔
251
  if (pMeta->pCache->STbFilterCache.pStbName == NULL) {
7,195,904✔
252
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
253
  }
254

255
  // open ref db cache
256
  pMeta->pCache->STbRefDbCache.pStbRefs =
14,391,808✔
257
      taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
14,391,034✔
258
  if (pMeta->pCache->STbRefDbCache.pStbRefs == NULL) {
7,195,904✔
259
    TSDB_CHECK_CODE(code = terrno, lino, _exit);
×
260
  }
261

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

265

266
_exit:
7,195,904✔
267
  if (code) {
7,195,904✔
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,195,904✔
272
  }
273
  return code;
7,195,904✔
274
}
275

276
void metaCacheClose(SMeta* pMeta) {
7,195,904✔
277
  if (pMeta->pCache) {
7,195,904✔
278
    entryCacheClose(pMeta);
7,195,650✔
279
    statsCacheClose(pMeta);
7,195,904✔
280

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

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

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

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

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

301
    taosMemoryFree(pMeta->pCache);
7,195,904✔
302
    pMeta->pCache = NULL;
7,195,904✔
303
  }
304
}
7,195,836✔
305

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

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

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

321
  // rehash
322
  for (int32_t iBucket = 0; iBucket < pCache->sEntryCache.nBucket; iBucket++) {
69,715,302✔
323
    SMetaCacheEntry* pEntry = pCache->sEntryCache.aBucket[iBucket];
69,694,464✔
324

325
    while (pEntry) {
139,388,928✔
326
      SMetaCacheEntry* pTEntry = pEntry->next;
69,694,464✔
327

328
      pEntry->next = aBucket[TABS(pEntry->info.uid) % nBucket];
69,694,464✔
329
      aBucket[TABS(pEntry->info.uid) % nBucket] = pEntry;
69,694,464✔
330

331
      pEntry = pTEntry;
69,694,464✔
332
    }
333
  }
334

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

342
int32_t metaCacheUpsert(SMeta* pMeta, SMetaInfo* pInfo) {
102,824,558✔
343
  int32_t code = 0;
102,824,558✔
344

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

347
  // search
348
  SMetaCache*       pCache = pMeta->pCache;
102,824,558✔
349
  int32_t           iBucket = TABS(pInfo->uid) % pCache->sEntryCache.nBucket;
102,845,494✔
350
  SMetaCacheEntry** ppEntry = &pCache->sEntryCache.aBucket[iBucket];
102,826,849✔
351
  while (*ppEntry && (*ppEntry)->info.uid != pInfo->uid) {
119,596,805✔
352
    ppEntry = &(*ppEntry)->next;
16,756,094✔
353
  }
354

355
  if (*ppEntry) {  // update
102,790,545✔
356
    if (pInfo->suid != (*ppEntry)->info.suid) {
17,641,213✔
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) {
17,625,570✔
361
      (*ppEntry)->info.version = pInfo->version;
17,652,182✔
362
      (*ppEntry)->info.skmVer = pInfo->skmVer;
17,627,968✔
363
    }
364
  } else {  // insert
365
    if (pCache->sEntryCache.nEntry >= pCache->sEntryCache.nBucket) {
85,196,862✔
366
      metaRehashCache(pCache, 1);
20,838✔
367

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

371
    SMetaCacheEntry* pEntryNew = (SMetaCacheEntry*)taosMemoryMalloc(sizeof(*pEntryNew));
85,186,393✔
372
    if (pEntryNew == NULL) {
85,189,953✔
373
      code = terrno;
×
374
      goto _exit;
×
375
    }
376

377
    pEntryNew->info = *pInfo;
85,189,953✔
378
    pEntryNew->next = pCache->sEntryCache.aBucket[iBucket];
85,190,602✔
379
    pCache->sEntryCache.aBucket[iBucket] = pEntryNew;
85,170,670✔
380
    pCache->sEntryCache.nEntry++;
85,192,478✔
381
  }
382

383
_exit:
102,825,596✔
384
  return code;
102,825,596✔
385
}
386

387
int32_t metaCacheDrop(SMeta* pMeta, int64_t uid) {
2,915,045✔
388
  int32_t code = 0;
2,915,045✔
389

390
  SMetaCache*       pCache = pMeta->pCache;
2,915,045✔
391
  int32_t           iBucket = TABS(uid) % pCache->sEntryCache.nBucket;
2,915,619✔
392
  SMetaCacheEntry** ppEntry = &pCache->sEntryCache.aBucket[iBucket];
2,915,696✔
393
  while (*ppEntry && (*ppEntry)->info.uid != uid) {
2,926,376✔
394
    ppEntry = &(*ppEntry)->next;
11,331✔
395
  }
396

397
  SMetaCacheEntry* pEntry = *ppEntry;
2,915,335✔
398
  if (pEntry) {
2,916,054✔
399
    *ppEntry = pEntry->next;
2,840,195✔
400
    taosMemoryFree(pEntry);
2,840,556✔
401
    pCache->sEntryCache.nEntry--;
2,839,041✔
402
    if (pCache->sEntryCache.nEntry < pCache->sEntryCache.nBucket / 4 &&
2,839,294✔
403
        pCache->sEntryCache.nBucket > META_CACHE_BASE_BUCKET) {
2,564,500✔
404
      metaRehashCache(pCache, 0);
×
405
    }
406
  } else {
407
    code = TSDB_CODE_NOT_FOUND;
75,859✔
408
  }
409

410
_exit:
2,915,696✔
411
  return code;
2,915,696✔
412
}
413

414
int32_t metaCacheGet(SMeta* pMeta, int64_t uid, SMetaInfo* pInfo) {
1,961,268,886✔
415
  int32_t code = 0;
1,961,268,886✔
416

417
  SMetaCache*      pCache = pMeta->pCache;
1,961,268,886✔
418
  int32_t          iBucket = TABS(uid) % pCache->sEntryCache.nBucket;
1,961,573,260✔
419
  SMetaCacheEntry* pEntry = pCache->sEntryCache.aBucket[iBucket];
1,961,607,477✔
420

421
  while (pEntry && pEntry->info.uid != uid) {
2,009,296,803✔
422
    pEntry = pEntry->next;
47,745,754✔
423
  }
424

425
  if (pEntry) {
1,961,739,100✔
426
    if (pInfo) {
1,870,192,658✔
427
      *pInfo = pEntry->info;
1,870,196,990✔
428
    }
429
  } else {
430
    code = TSDB_CODE_NOT_FOUND;
91,546,442✔
431
  }
432

433
  return code;
1,961,785,952✔
434
}
435

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

440
  if (expand) {
42,961✔
441
    nBucket = pCache->sStbStatsCache.nBucket * 2;
38,177✔
442
  } else {
443
    nBucket = pCache->sStbStatsCache.nBucket / 2;
4,784✔
444
  }
445

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

452
  // rehash
453
  for (int32_t iBucket = 0; iBucket < pCache->sStbStatsCache.nBucket; iBucket++) {
1,880,561✔
454
    SMetaStbStatsEntry* pEntry = pCache->sStbStatsCache.aBucket[iBucket];
1,837,600✔
455

456
    while (pEntry) {
3,524,112✔
457
      SMetaStbStatsEntry* pTEntry = pEntry->next;
1,686,512✔
458

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

462
      pEntry = pTEntry;
1,686,512✔
463
    }
464
  }
465

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

471
_exit:
42,961✔
472
  return code;
42,961✔
473
}
474

475
int32_t metaStatsCacheUpsert(SMeta* pMeta, SMetaStbStats* pInfo) {
59,272,827✔
476
  int32_t code = 0;
59,272,827✔
477

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

480
  // search
481
  SMetaCache*          pCache = pMeta->pCache;
59,272,827✔
482
  int32_t              iBucket = TABS(pInfo->uid) % pCache->sStbStatsCache.nBucket;
59,272,914✔
483
  SMetaStbStatsEntry** ppEntry = &pCache->sStbStatsCache.aBucket[iBucket];
59,281,780✔
484
  while (*ppEntry && (*ppEntry)->info.uid != pInfo->uid) {
60,985,446✔
485
    ppEntry = &(*ppEntry)->next;
1,723,452✔
486
  }
487

488
  if (*ppEntry) {  // update
59,246,627✔
489
    (*ppEntry)->info.ctbNum = pInfo->ctbNum;
55,145,398✔
490
    (*ppEntry)->info.colNum = pInfo->colNum;
55,149,609✔
491
    (*ppEntry)->info.flags = pInfo->flags;
55,157,129✔
492
    (*ppEntry)->info.keep = pInfo->keep;
55,141,190✔
493
  } else {  // insert
494
    if (pCache->sStbStatsCache.nEntry >= pCache->sStbStatsCache.nBucket) {
4,118,099✔
495
      TAOS_UNUSED(metaRehashStatsCache(pCache, 1));
38,177✔
496
      iBucket = TABS(pInfo->uid) % pCache->sStbStatsCache.nBucket;
38,177✔
497
    }
498

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

505
    pEntryNew->info = *pInfo;
4,119,234✔
506
    pEntryNew->next = pCache->sStbStatsCache.aBucket[iBucket];
4,119,524✔
507
    pCache->sStbStatsCache.aBucket[iBucket] = pEntryNew;
4,118,749✔
508
    pCache->sStbStatsCache.nEntry++;
4,119,514✔
509
  }
510

511
_exit:
59,264,015✔
512
  return code;
59,264,015✔
513
}
514

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

518
  SMetaCache*          pCache = pMeta->pCache;
836,224✔
519
  int32_t              iBucket = TABS(uid) % pCache->sStbStatsCache.nBucket;
836,224✔
520
  SMetaStbStatsEntry** ppEntry = &pCache->sStbStatsCache.aBucket[iBucket];
836,477✔
521
  while (*ppEntry && (*ppEntry)->info.uid != uid) {
947,464✔
522
    ppEntry = &(*ppEntry)->next;
111,240✔
523
  }
524

525
  SMetaStbStatsEntry* pEntry = *ppEntry;
836,224✔
526
  if (pEntry) {
835,934✔
527
    *ppEntry = pEntry->next;
540,192✔
528
    taosMemoryFree(pEntry);
540,482✔
529
    pCache->sStbStatsCache.nEntry--;
540,482✔
530
    if (pCache->sStbStatsCache.nEntry < pCache->sStbStatsCache.nBucket / 4 &&
540,482✔
531
        pCache->sStbStatsCache.nBucket > META_CACHE_STATS_BUCKET) {
401,133✔
532
      TAOS_UNUSED(metaRehashStatsCache(pCache, 0));
4,784✔
533
    }
534
  } else {
535
    code = TSDB_CODE_NOT_FOUND;
295,742✔
536
  }
537

538
_exit:
836,224✔
539
  return code;
836,224✔
540
}
541

542
int32_t metaStatsCacheGet(SMeta* pMeta, int64_t uid, SMetaStbStats* pInfo) {
175,657,666✔
543
  int32_t code = TSDB_CODE_SUCCESS;
175,657,666✔
544

545
  SMetaCache*         pCache = pMeta->pCache;
175,657,666✔
546
  int32_t             iBucket = TABS(uid) % pCache->sStbStatsCache.nBucket;
175,688,223✔
547
  SMetaStbStatsEntry* pEntry = pCache->sStbStatsCache.aBucket[iBucket];
175,675,735✔
548

549
  while (pEntry && pEntry->info.uid != uid) {
190,452,111✔
550
    pEntry = pEntry->next;
14,749,267✔
551
  }
552

553
  if (pEntry) {
175,704,457✔
554
    if (pInfo) {
161,469,928✔
555
      *pInfo = pEntry->info;
161,469,648✔
556
    }
557
  } else {
558
    code = TSDB_CODE_NOT_FOUND;
14,234,529✔
559
  }
560

561
  return code;
175,706,581✔
562
}
563

564
static FORCE_INLINE void setMD5DigestInKey(uint64_t* pBuf, const char* key, int32_t keyLen) {
565
  memcpy(&pBuf[2], key, keyLen);
131,023,447✔
566
}
131,021,213✔
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) {
131,013,275✔
571
  buf[0] = (uint64_t)pHashMap;
131,013,275✔
572
  buf[1] = suid;
131,018,032✔
573
  setMD5DigestInKey(buf, key, keyLen);
574
}
131,021,841✔
575

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

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

586
  *acquireRes = 0;
27,002✔
587
  uint64_t key[4];
27,002✔
588
  initCacheKey(key, pTableMap, suid, (const char*)pKey, keyLen);
27,002✔
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);
27,002✔
597
  pMeta->pCache->sTagFilterResCache.accTimes += 1;
27,002✔
598

599
  LRUHandle* pHandle = taosLRUCacheLookup(pCache, key, TAG_FILTER_RES_KEY_LEN);
27,002✔
600
  if (pHandle == NULL) {
27,002✔
601
    (void)taosThreadMutexUnlock(pLock);
6,647✔
602
    return TSDB_CODE_SUCCESS;
6,647✔
603
  }
604

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

612
  *acquireRes = 1;
20,355✔
613

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

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

622
  (*pEntry)->hitTimes += 1;
20,355✔
623

624
  uint32_t acc = pMeta->pCache->sTagFilterResCache.accTimes;
20,355✔
625
  if ((*pEntry)->hitTimes % 5000 == 0 && (*pEntry)->hitTimes > 0) {
20,355✔
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);
20,355✔
631

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

637
int32_t metaStableTagFilterCacheGet(void* pVnode, tb_uid_t suid,
7,975✔
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;
7,975✔
642
  int32_t lino = 0;
7,975✔
643
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
7,975✔
644
  int32_t vgId = TD_VID(pMeta->pVnode);
7,975✔
645
  *acquireRes = 0;
7,975✔
646

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

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

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

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

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

666
  // set the result into the buffer
667
  *acquireRes = 1;
3,915✔
668
  TAOS_UNUSED(taosArrayAddBatch(
3,915✔
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;
3,915✔
673
  (*pFilterEntry)->hitTimes += 1;
3,915✔
674
  uint64_t hit = ++pMeta->pCache->sStableTagFilterResCache.hitTimes;
3,915✔
675
  uint64_t acc = pMeta->pCache->sStableTagFilterResCache.accTimes;
3,915✔
676
  if ((*pTagConds)->hitTimes % 1000 == 0 && (*pTagConds)->hitTimes > 0) {
3,915✔
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:
3,915✔
686
  if (TSDB_CODE_SUCCESS != code) {
7,975✔
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);
7,975✔
692
  if (TSDB_CODE_SUCCESS != code) {
7,975✔
693
    metaError("vgId:%d, %s unlock failed at %s:%d since %s",
×
694
      vgId, __func__, __FILE__, lino, tstrerror(code));
695
  }
696
  return code;
7,975✔
697
}
698

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

705
  const uint64_t* p = key;
6,647✔
706
  if (keyLen != sizeof(int64_t) * 4) {
6,647✔
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];
6,647✔
712

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

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

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

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

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

745
_end:
4,294✔
746
  if (code != TSDB_CODE_SUCCESS) {
4,294✔
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,294✔
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,
6,647✔
760
                              int32_t payloadLen, double selectivityRatio) {
761
  int32_t code = 0;
6,647✔
762
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
6,647✔
763
  int32_t vgId = TD_VID(pMeta->pVnode);
6,647✔
764

765
  if (selectivityRatio > tsSelectivityRatio) {
6,647✔
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) {
6,647✔
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;
6,647✔
782
  SHashObj*      pTableEntry = pMeta->pCache->sTagFilterResCache.pTableEntry;
6,647✔
783
  TdThreadMutex* pLock = &pMeta->pCache->sTagFilterResCache.lock;
6,647✔
784

785
  uint64_t key[4] = {0};
6,647✔
786
  initCacheKey(key, pTableEntry, suid, pKey, keyLen);
6,647✔
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);
6,647✔
795
  STagFilterResEntry** pEntry = taosHashGet(pTableEntry, &suid, sizeof(uint64_t));
6,647✔
796
  if (pEntry == NULL) {
6,647✔
797
    code = addNewEntry(pTableEntry, pKey, keyLen, suid);
4,294✔
798
    if (code != TSDB_CODE_SUCCESS) {
4,294✔
799
      goto _end;
×
800
    }
801
  } else {  // check if it exists or not
802
    code = taosHashPut((*pEntry)->set, pKey, keyLen, NULL, 0);
2,353✔
803
    if (code == TSDB_CODE_DUP_KEY) {
2,353✔
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,353✔
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,
6,647✔
815
                           TAOS_LRU_PRIORITY_LOW, NULL);
816
_end:
6,647✔
817
  (void)taosThreadMutexUnlock(pLock);
6,647✔
818
  metaDebug("vgId:%d, suid:%" PRIu64 " list cache added into cache, total:%d, tables:%d", vgId, suid,
6,647✔
819
            (int32_t)taosLRUCacheGetUsage(pCache), taosHashGetSize(pTableEntry));
820

821
  return code;
6,647✔
822
}
823

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

829
int32_t metaStableTagFilterCachePut(
4,060✔
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,060✔
834
  int32_t lino = 0;
4,060✔
835
  SMeta*  pMeta = ((SVnode*)pVnode)->pMeta;
4,060✔
836
  int32_t vgId = TD_VID(pMeta->pVnode);
4,060✔
837

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

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

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

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

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

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

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

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

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

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

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

904
_end:
4,060✔
905
  if (TSDB_CODE_SUCCESS != code) {
4,060✔
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,060✔
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,060✔
923
  if (TSDB_CODE_SUCCESS != code) {
4,060✔
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,060✔
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(
13,630✔
972
  const SSchema* pTagSchemas, int32_t nTagCols, col_id_t cid) {
973
  for (int32_t i = 0; i < nTagCols; i++) {
75,110✔
974
    if (pTagSchemas[i].colId == cid) {
75,110✔
975
      return pTagSchemas[i].bytes;
13,630✔
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;
6,960✔
984

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

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

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

1005
static int32_t buildTagDataEntryKey(const SArray* pColIds, const STag* pTag,
33,495✔
1006
  const SSchemaWrapper* pTagScheam, T_MD5_CTX* pContext) {
1007
  int32_t code = TSDB_CODE_SUCCESS;
33,495✔
1008
  int32_t lino = 0;
33,495✔
1009
  int32_t keyLen = 0;
33,495✔
1010
  char* pKey = NULL;
33,495✔
1011
  // get length first
1012
  for (int32_t i = 0; i < taosArrayGetSize(pColIds); i++) {
98,745✔
1013
    STagVal pTagValue = {.cid = *(col_id_t*)taosArrayGet(pColIds, i)};
66,845✔
1014
    if (tTagGet(pTag, &pTagValue)) {
66,845✔
1015
      keyLen += sizeof(col_id_t);
65,250✔
1016
      if (IS_VAR_DATA_TYPE(pTagValue.type)) {
65,250✔
1017
        int32_t varLen = getTagColSize(
27,260✔
1018
          pTagScheam->pSchema, pTagScheam->nCols, pTagValue.cid);
13,630✔
1019
        code = varLen > 0 ? TSDB_CODE_SUCCESS : TSDB_CODE_NOT_FOUND;
13,630✔
1020
        QUERY_CHECK_CODE(code, lino, _end);
13,630✔
1021
        keyLen += varLen;
13,630✔
1022
      } else {
1023
        keyLen += tDataTypes[pTagValue.type].bytes;
51,620✔
1024
      }
1025
    } else {
1026
      // tag value not found
1027
      code = TSDB_CODE_NOT_FOUND;
1,595✔
1028
      QUERY_CHECK_CODE(code, lino, _end);
1,595✔
1029
    }
1030
  }
1031

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

1038
  // build the key
1039
  char* pStart = pKey;
31,900✔
1040
  for (int32_t i = 0; i < taosArrayGetSize(pColIds); i++) {
95,700✔
1041
    STagVal pTagValue = {.cid = *(col_id_t*)taosArrayGet(pColIds, i)};
63,800✔
1042
    if (tTagGet(pTag, &pTagValue)) {
63,800✔
1043
      // copy cid
1044
      memcpy(pStart, &pTagValue.cid, sizeof(col_id_t));
63,800✔
1045
      pStart += sizeof(col_id_t);
63,800✔
1046
      // copy value
1047
      if (IS_VAR_DATA_TYPE(pTagValue.type) && pTagValue.pData != NULL) {
63,800✔
1048
        if (TSDB_DATA_TYPE_NCHAR == pTagValue.type) {
13,630✔
1049
          // need to convert nchar to var
1050
          char *pVar = NULL;
6,960✔
1051
          code = ncharToVar((char *)pTagValue.pData, pTagValue.nData, &pVar);
6,960✔
1052
          QUERY_CHECK_CODE(code, lino, _end);
6,960✔
1053
          memcpy(pStart, varDataVal(pVar), varDataLen(pVar));
6,960✔
1054
          pStart += varDataLen(pVar);
6,960✔
1055
          taosMemoryFree(pVar);
6,960✔
1056
        } else {
1057
          memcpy(pStart, pTagValue.pData, pTagValue.nData);
6,670✔
1058
          pStart += pTagValue.nData;
6,670✔
1059
        }
1060
      } else {
1061
        memcpy(pStart, &pTagValue.i64, tDataTypes[pTagValue.type].bytes);
50,170✔
1062
        pStart += tDataTypes[pTagValue.type].bytes;
50,170✔
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);
31,900✔
1073
  tMD5Update(pContext, (uint8_t*)pKey, (uint32_t)keyLen);
31,900✔
1074
  tMD5Final(pContext);
31,900✔
1075

1076
_end:
33,495✔
1077
  taosMemFreeClear(pKey);
33,495✔
1078
  return code;
33,495✔
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,
72,783,813✔
1084
  const SMetaEntry* pChildTable, const SMetaEntry* pSuperTable,
1085
  ETagFilterCacheAction action) {
1086
  if (pMeta == NULL || pChildTable == NULL || pSuperTable == NULL) {
72,783,813✔
UNCOV
1087
    return TSDB_CODE_INVALID_PARA;
×
1088
  }
1089
  int32_t   lino = 0;
72,795,249✔
1090
  int32_t   code = TSDB_CODE_SUCCESS;
72,795,249✔
1091
  SHashObj* pTableEntry = pMeta->pCache->sStableTagFilterResCache.pTableEntry;
72,795,249✔
1092
  TdThreadRwlock* pRwlock = &pMeta->pCache->sStableTagFilterResCache.rwlock;
72,788,005✔
1093

1094
  code = taosThreadRwlockWrlock(pRwlock);
72,791,332✔
1095
  TSDB_CHECK_CODE(code, lino, _end);
72,778,103✔
1096

1097
  tb_uid_t suid = pChildTable->ctbEntry.suid;;
72,778,103✔
1098
  STagConds** pTagConds =
1099
    (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(tb_uid_t));
72,786,966✔
1100
  if (pTagConds != NULL) {
72,790,232✔
1101
    STagCondFilterEntry** ppFilterEntry = NULL;
7,395✔
1102
    while ((ppFilterEntry = taosHashIterate((*pTagConds)->set, ppFilterEntry))) {
39,295✔
1103
      STagCondFilterEntry* pFilterEntry = *ppFilterEntry;
33,495✔
1104
      // rebuild the tagCondKey and check existence
1105
      SArray* pColIds = pFilterEntry->pColIds;
33,495✔
1106
      // rebuild the tagCondFilterKey
1107
      int32_t keyLen = 0;
33,495✔
1108
      char*   pKey = NULL;
33,495✔
1109
      T_MD5_CTX context = {0};
33,495✔
1110
      code = buildTagDataEntryKey(pColIds, (STag*)pChildTable->ctbEntry.pTags, 
33,495✔
1111
        &pSuperTable->stbEntry.schemaTag, &context);
1112
      if (code != TSDB_CODE_SUCCESS) {
33,495✔
1113
        metaError("vgId:%d, suid:%" PRIu64 " failed to build tag condition"
1,595✔
1114
          " key for dropped table uid:%" PRIu64 " since %s",
1115
          TD_VID(pMeta->pVnode), suid, pChildTable->uid, tstrerror(code));
1116
        goto _end;
1,595✔
1117
      }
1118

1119
      SArray** pArray = (SArray**)taosHashGet(
31,900✔
1120
        pFilterEntry->set, context.digest, tListLen(context.digest));
1121
      if (pArray != NULL) {
31,900✔
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,350✔
1125
          for (int32_t i = 0; i < taosArrayGetSize(*pArray); i++) {
4,350✔
1126
            uint64_t uid = *(uint64_t*)taosArrayGet(*pArray, i);
4,350✔
1127
            if (uid == pChildTable->uid) {
4,350✔
1128
              taosArrayRemove(*pArray, i);
2,175✔
1129
              metaDebug("vgId:%d, suid:%" PRIu64
2,175✔
1130
                " removed dropped table uid:%" PRIu64
1131
                " from stable tag filter cache",
1132
                TD_VID(pMeta->pVnode), suid, pChildTable->uid);
1133
              break;
2,175✔
1134
            } 
1135
          }
1136
        } else {
1137
          // STABLE_TAG_FILTER_CACHE_ADD_TABLE
1138
          void* _tmp = taosArrayPush(*pArray, &pChildTable->uid);
4,350✔
1139
        }
1140
      }
1141
    }
1142
  }
1143

1144
_end:
72,790,232✔
1145
  if (TSDB_CODE_SUCCESS != code) {
72,785,352✔
1146
    metaError("vgId:%d, %s failed at %s:%d since %s",
1,595✔
1147
      TD_VID(pMeta->pVnode), __func__, __FILE__, lino, tstrerror(code));
1148
  } else {
1149
    metaDebug(
72,783,757✔
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);
72,785,948✔
1156
  if (TSDB_CODE_SUCCESS != code) {
72,775,034✔
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;
72,773,739✔
1161
}
1162

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

1173
  code = taosThreadRwlockWrlock(pRwlock);
192,021✔
1174
  TSDB_CHECK_CODE(code, lino, _end);
192,021✔
1175

1176
  STagConds** pTagConds =
1177
    (STagConds**)taosHashGet(pTableEntry, &suid, sizeof(tb_uid_t));
192,021✔
1178
  if (pTagConds != NULL) {
192,624✔
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:
192,624✔
1203
  if (TSDB_CODE_SUCCESS != code) {
192,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(
192,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);
192,624✔
1213
  if (TSDB_CODE_SUCCESS != code) {
191,418✔
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;
192,624✔
1218
}
1219

1220
void metaCacheClear(SMeta* pMeta) {
3,595,327✔
1221
  metaWLock(pMeta);
3,595,327✔
1222
  metaCacheClose(pMeta);
3,595,477✔
1223
  (void)metaCacheOpen(pMeta);
3,595,477✔
1224
  metaULock(pMeta);
3,595,477✔
1225
}
3,595,477✔
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) {
65,487,189✔
1229
  uint64_t  p[4] = {0};
65,487,189✔
1230
  int32_t   vgId = TD_VID(pMeta->pVnode);
65,492,951✔
1231
  SHashObj* pEntryHashMap = pMeta->pCache->sTagFilterResCache.pTableEntry;
65,495,541✔
1232

1233
  uint64_t dummy[2] = {0};
65,490,213✔
1234
  initCacheKey(p, pEntryHashMap, suid, (char*)&dummy[0], 16);
65,497,176✔
1235

1236
  TdThreadMutex* pLock = &pMeta->pCache->sTagFilterResCache.lock;
65,489,860✔
1237
  (void)taosThreadMutexLock(pLock);
65,493,602✔
1238

1239
  STagFilterResEntry** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
65,501,477✔
1240
  if (pEntry == NULL || taosHashGetSize((*pEntry)->set) == 0) {
65,497,447✔
1241
    (void)taosThreadMutexUnlock(pLock);
65,495,841✔
1242
    return TSDB_CODE_SUCCESS;
65,497,793✔
1243
  }
1244

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

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

1256
  metaDebug("vgId:%d suid:%" PRId64 " cached related tag filter uid list cleared", vgId, suid);
1,606✔
1257
  return TSDB_CODE_SUCCESS;
1,606✔
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) {
65,485,474✔
1387
  uint64_t  p[4] = {0};
65,485,474✔
1388
  int32_t   vgId = TD_VID(pMeta->pVnode);
65,490,441✔
1389
  SHashObj* pEntryHashMap = pMeta->pCache->STbGroupResCache.pTableEntry;
65,482,652✔
1390

1391
  uint64_t dummy[2] = {0};
65,493,692✔
1392
  initCacheKey(p, pEntryHashMap, suid, (char*)&dummy[0], 16);
65,491,347✔
1393

1394
  TdThreadMutex* pLock = &pMeta->pCache->STbGroupResCache.lock;
65,495,671✔
1395
  (void)taosThreadMutexLock(pLock);
65,493,539✔
1396

1397
  STagFilterResEntry** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
65,493,182✔
1398
  if (pEntry == NULL || taosHashGetSize((*pEntry)->set) == 0) {
65,498,820✔
1399
    (void)taosThreadMutexUnlock(pLock);
65,498,820✔
1400
    return TSDB_CODE_SUCCESS;
65,502,577✔
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) {
112,548,059✔
1419
  if (type == 0 && taosHashGet(pMeta->pCache->STbFilterCache.pStb, key, sizeof(tb_uid_t))) {
112,548,059✔
1420
    return true;
1,300✔
1421
  }
1422

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

1427
  return false;
112,565,770✔
1428
}
1429

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

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

1439
  return 0;
×
1440
}
1441

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

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

1455
  if (!(dbName = strchr(pMeta->pVnode->config.dbname, '.'))) return 0;
3,598,036✔
1456
  if (0 == strncmp(++dbName, "log", TSDB_DB_NAME_LEN)) {
3,599,977✔
1457
    tbNum = tkLogStbNum;
260✔
1458
    pTbArr = (const char**)&tkLogStb;
260✔
1459
  } else if (0 == strncmp(dbName, "audit", TSDB_DB_NAME_LEN)) {
3,599,717✔
1460
    tbNum = tkAuditStbNum;
×
1461
    pTbArr = (const char**)&tkAuditStb;
×
1462
  }
1463
  if (tbNum && pTbArr) {
3,599,977✔
1464
    for (int32_t i = 0; i < tbNum; ++i) {
5,200✔
1465
      TAOS_CHECK_RETURN(metaPutTbToFilterCache(pMeta, pTbArr[i], 1));
4,940✔
1466
    }
1467
  }
1468
#else
1469
#endif
1470
  return 0;
3,599,977✔
1471
}
1472

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

1476
  if (metaStatsCacheGet(pMeta, uid, &stats) == TSDB_CODE_SUCCESS) {
127,717✔
1477
    return stats.keep;
118,475✔
1478
  }
1479

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

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

1499
  (void)taosThreadMutexLock(pLock);
252,302✔
1500

1501
  SHashObj** pEntry = taosHashGet(pEntryHashMap, &suid, sizeof(uint64_t));
252,302✔
1502
  if (pEntry == NULL) {
252,302✔
1503
    goto _return;
250,408✔
1504
  }
1505

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

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

1510
_return:
1,518✔
1511
  (void)taosThreadMutexUnlock(pLock);
252,302✔
1512
  return code;
252,302✔
1513
}
1514

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

1522
  (void)taosThreadMutexLock(pLock);
775,893✔
1523

1524
  SHashObj** pEntry = taosHashGet(pTableMap, &suid, sizeof(uint64_t));
777,595✔
1525
  if (pEntry) {
777,595✔
1526
    void *iter = taosHashIterate(*pEntry, NULL);
759,236✔
1527
    while (iter != NULL) {
1,515,587✔
1528
      size_t   dbNameLen = 0;
756,351✔
1529
      char*    name = NULL;
755,773✔
1530
      char*    dbName = NULL;
755,773✔
1531
      name = taosHashGetKey(iter, &dbNameLen);
755,773✔
1532
      TSDB_CHECK_NULL(name, code, line, _return, terrno);
756,351✔
1533
      dbName = taosMemoryMalloc(dbNameLen + 1);
756,351✔
1534
      TSDB_CHECK_NULL(dbName, code, line, _return, terrno);
755,773✔
1535
      tstrncpy(dbName, name, dbNameLen + 1);
755,773✔
1536
      TSDB_CHECK_NULL(taosArrayPush(pList, &dbName), code, line, _return, terrno);
755,773✔
1537
      iter = taosHashIterate(*pEntry, iter);
755,773✔
1538
    }
1539
  }
1540

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

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

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

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

1560
  *pEntry = p;
18,359✔
1561

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

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

1576
  (void)taosThreadMutexLock(pLock);
21,244✔
1577

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

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

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

1602
  return code;
21,244✔
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