ctgCache.c 68.2 KB
Newer Older
D
dapan1121 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "catalogInt.h"
dengyihao's avatar
dengyihao 已提交
17
#include "query.h"
D
dapan1121 已提交
18
#include "systable.h"
dengyihao's avatar
dengyihao 已提交
19 20
#include "tname.h"
#include "trpc.h"
D
dapan1121 已提交
21

dengyihao's avatar
dengyihao 已提交
22 23 24 25 26 27 28 29 30 31 32
SCtgOperation gCtgCacheOperation[CTG_OP_MAX] = {{CTG_OP_UPDATE_VGROUP, "update vgInfo", ctgOpUpdateVgroup},
                                                {CTG_OP_UPDATE_TB_META, "update tbMeta", ctgOpUpdateTbMeta},
                                                {CTG_OP_DROP_DB_CACHE, "drop DB", ctgOpDropDbCache},
                                                {CTG_OP_DROP_DB_VGROUP, "drop DBVgroup", ctgOpDropDbVgroup},
                                                {CTG_OP_DROP_STB_META, "drop stbMeta", ctgOpDropStbMeta},
                                                {CTG_OP_DROP_TB_META, "drop tbMeta", ctgOpDropTbMeta},
                                                {CTG_OP_UPDATE_USER, "update user", ctgOpUpdateUser},
                                                {CTG_OP_UPDATE_VG_EPSET, "update epset", ctgOpUpdateEpset},
                                                {CTG_OP_UPDATE_TB_INDEX, "update tbIndex", ctgOpUpdateTbIndex},
                                                {CTG_OP_DROP_TB_INDEX, "drop tbIndex", ctgOpDropTbIndex},
                                                {CTG_OP_CLEAR_CACHE, "clear cache", ctgOpClearCache}};
D
dapan1121 已提交
33

D
dapan1121 已提交
34 35
int32_t ctgRLockVgInfo(SCatalog *pCtg, SCtgDBCache *dbCache, bool *inCache) {
  CTG_LOCK(CTG_READ, &dbCache->vgCache.vgLock);
dengyihao's avatar
dengyihao 已提交
36

D
dapan1121 已提交
37
  if (dbCache->deleted) {
D
dapan1121 已提交
38
    CTG_UNLOCK(CTG_READ, &dbCache->vgCache.vgLock);
D
dapan1121 已提交
39

dengyihao's avatar
dengyihao 已提交
40 41
    ctgDebug("db is dropping, dbId:0x%" PRIx64, dbCache->dbId);

D
dapan1121 已提交
42 43 44 45
    *inCache = false;
    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
46 47
  if (NULL == dbCache->vgCache.vgInfo) {
    CTG_UNLOCK(CTG_READ, &dbCache->vgCache.vgLock);
D
dapan1121 已提交
48 49

    *inCache = false;
dengyihao's avatar
dengyihao 已提交
50
    ctgDebug("db vgInfo is empty, dbId:0x%" PRIx64, dbCache->dbId);
D
dapan1121 已提交
51 52 53 54
    return TSDB_CODE_SUCCESS;
  }

  *inCache = true;
dengyihao's avatar
dengyihao 已提交
55

D
dapan1121 已提交
56 57 58
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
59 60
int32_t ctgWLockVgInfo(SCatalog *pCtg, SCtgDBCache *dbCache) {
  CTG_LOCK(CTG_WRITE, &dbCache->vgCache.vgLock);
D
dapan1121 已提交
61 62

  if (dbCache->deleted) {
dengyihao's avatar
dengyihao 已提交
63
    ctgDebug("db is dropping, dbId:0x%" PRIx64, dbCache->dbId);
D
dapan1121 已提交
64
    CTG_UNLOCK(CTG_WRITE, &dbCache->vgCache.vgLock);
D
dapan1121 已提交
65 66 67 68 69 70
    CTG_ERR_RET(TSDB_CODE_CTG_DB_DROPPED);
  }

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
71
void ctgRUnlockVgInfo(SCtgDBCache *dbCache) { CTG_UNLOCK(CTG_READ, &dbCache->vgCache.vgLock); }
D
dapan1121 已提交
72

dengyihao's avatar
dengyihao 已提交
73
void ctgWUnlockVgInfo(SCtgDBCache *dbCache) { CTG_UNLOCK(CTG_WRITE, &dbCache->vgCache.vgLock); }
D
dapan1121 已提交
74

dengyihao's avatar
dengyihao 已提交
75
void ctgReleaseDBCache(SCatalog *pCtg, SCtgDBCache *dbCache) { CTG_UNLOCK(CTG_READ, &dbCache->dbLock); }
D
dapan1121 已提交
76

dengyihao's avatar
dengyihao 已提交
77
int32_t ctgAcquireDBCacheImpl(SCatalog *pCtg, const char *dbFName, SCtgDBCache **pCache, bool acquire) {
D
dapan1121 已提交
78
  char *p = strchr(dbFName, '.');
D
dapan1121 已提交
79
  if (p && IS_SYS_DBNAME(p + 1)) {
D
dapan1121 已提交
80 81 82
    dbFName = p + 1;
  }

D
dapan1121 已提交
83
  SCtgDBCache *dbCache = (SCtgDBCache *)taosHashGet(pCtg->dbCache, dbFName, strlen(dbFName));
D
dapan1121 已提交
84 85 86 87 88 89
  if (NULL == dbCache) {
    *pCache = NULL;
    ctgDebug("db not in cache, dbFName:%s", dbFName);
    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
90 91 92 93
  if (acquire) {
    CTG_LOCK(CTG_READ, &dbCache->dbLock);
  }

D
dapan1121 已提交
94 95 96
  if (dbCache->deleted) {
    if (acquire) {
      ctgReleaseDBCache(pCtg, dbCache);
dengyihao's avatar
dengyihao 已提交
97 98
    }

D
dapan1121 已提交
99 100 101 102 103 104
    *pCache = NULL;
    ctgDebug("db is removing from cache, dbFName:%s", dbFName);
    return TSDB_CODE_SUCCESS;
  }

  *pCache = dbCache;
dengyihao's avatar
dengyihao 已提交
105

D
dapan1121 已提交
106 107 108
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
109
int32_t ctgAcquireDBCache(SCatalog *pCtg, const char *dbFName, SCtgDBCache **pCache) {
D
dapan1121 已提交
110 111 112
  CTG_RET(ctgAcquireDBCacheImpl(pCtg, dbFName, pCache, true));
}

dengyihao's avatar
dengyihao 已提交
113
int32_t ctgGetDBCache(SCatalog *pCtg, const char *dbFName, SCtgDBCache **pCache) {
D
dapan1121 已提交
114 115 116
  CTG_RET(ctgAcquireDBCacheImpl(pCtg, dbFName, pCache, false));
}

dengyihao's avatar
dengyihao 已提交
117
void ctgReleaseVgInfoToCache(SCatalog *pCtg, SCtgDBCache *dbCache) {
D
dapan1121 已提交
118 119 120
  ctgRUnlockVgInfo(dbCache);
  ctgReleaseDBCache(pCtg, dbCache);
}
D
dapan1121 已提交
121

dengyihao's avatar
dengyihao 已提交
122
void ctgReleaseTbMetaToCache(SCatalog *pCtg, SCtgDBCache *dbCache, SCtgTbCache *pCache) {
D
dapan1121 已提交
123 124
  if (pCache) {
    CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
125
    taosHashRelease(dbCache->tbCache, pCache);
D
dapan1121 已提交
126
  }
D
dapan1121 已提交
127

D
dapan1121 已提交
128 129
  if (dbCache) {
    ctgReleaseDBCache(pCtg, dbCache);
D
dapan1121 已提交
130
  }
D
dapan1121 已提交
131
}
D
dapan1121 已提交
132

dengyihao's avatar
dengyihao 已提交
133
void ctgReleaseTbIndexToCache(SCatalog *pCtg, SCtgDBCache *dbCache, SCtgTbCache *pCache) {
D
dapan1121 已提交
134 135
  if (pCache) {
    CTG_UNLOCK(CTG_READ, &pCache->indexLock);
dengyihao's avatar
dengyihao 已提交
136
    taosHashRelease(dbCache->tbCache, pCache);
D
dapan1121 已提交
137 138 139 140 141 142 143
  }

  if (dbCache) {
    ctgReleaseDBCache(pCtg, dbCache);
  }
}

dengyihao's avatar
dengyihao 已提交
144
int32_t ctgAcquireVgInfoFromCache(SCatalog *pCtg, const char *dbFName, SCtgDBCache **pCache) {
D
dapan1121 已提交
145
  SCtgDBCache *dbCache = NULL;
D
dapan1121 已提交
146
  ctgAcquireDBCache(pCtg, dbFName, &dbCache);
dengyihao's avatar
dengyihao 已提交
147
  if (NULL == dbCache) {
D
dapan1121 已提交
148 149 150 151 152
    ctgDebug("db %s not in cache", dbFName);
    goto _return;
  }

  bool inCache = false;
D
dapan1121 已提交
153
  ctgRLockVgInfo(pCtg, dbCache, &inCache);
D
dapan1121 已提交
154 155 156 157 158 159 160
  if (!inCache) {
    ctgDebug("vgInfo of db %s not in cache", dbFName);
    goto _return;
  }

  *pCache = dbCache;

D
dapan1121 已提交
161
  CTG_CACHE_STAT_INC(numOfVgHit, 1);
D
dapan1121 已提交
162 163

  ctgDebug("Got db vgInfo from cache, dbFName:%s", dbFName);
dengyihao's avatar
dengyihao 已提交
164

D
dapan1121 已提交
165 166 167 168 169 170 171 172 173 174
  return TSDB_CODE_SUCCESS;

_return:

  if (dbCache) {
    ctgReleaseDBCache(pCtg, dbCache);
  }

  *pCache = NULL;

D
dapan1121 已提交
175
  CTG_CACHE_STAT_INC(numOfVgMiss, 1);
dengyihao's avatar
dengyihao 已提交
176

D
dapan1121 已提交
177 178 179
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
180
int32_t ctgAcquireTbMetaFromCache(SCatalog *pCtg, char *dbFName, char *tbName, SCtgDBCache **pDb, SCtgTbCache **pTb) {
D
dapan1121 已提交
181
  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
182
  SCtgTbCache *pCache = NULL;
D
dapan1121 已提交
183 184 185 186 187
  ctgAcquireDBCache(pCtg, dbFName, &dbCache);
  if (NULL == dbCache) {
    ctgDebug("db %s not in cache", dbFName);
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
188

D
dapan1121 已提交
189
  pCache = taosHashAcquire(dbCache->tbCache, tbName, strlen(tbName));
D
dapan1121 已提交
190 191 192
  if (NULL == pCache) {
    ctgDebug("tb %s not in cache, dbFName:%s", tbName, dbFName);
    goto _return;
D
dapan1121 已提交
193 194
  }

D
dapan1121 已提交
195 196 197 198 199 200 201 202 203 204
  CTG_LOCK(CTG_READ, &pCache->metaLock);
  if (NULL == pCache->pMeta) {
    ctgDebug("tb %s meta not in cache, dbFName:%s", tbName, dbFName);
    goto _return;
  }

  *pDb = dbCache;
  *pTb = pCache;

  ctgDebug("tb %s meta got in cache, dbFName:%s", tbName, dbFName);
dengyihao's avatar
dengyihao 已提交
205

D
dapan1121 已提交
206
  CTG_CACHE_STAT_INC(numOfMetaHit, 1);
D
dapan1121 已提交
207 208 209 210 211 212 213

  return TSDB_CODE_SUCCESS;

_return:

  ctgReleaseTbMetaToCache(pCtg, dbCache, pCache);

D
dapan1121 已提交
214
  CTG_CACHE_STAT_INC(numOfMetaMiss, 1);
dengyihao's avatar
dengyihao 已提交
215

D
dapan1121 已提交
216 217 218
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
219 220 221
int32_t ctgAcquireStbMetaFromCache(SCatalog *pCtg, char *dbFName, uint64_t suid, SCtgDBCache **pDb, SCtgTbCache **pTb) {
  SCtgDBCache *dbCache = NULL;
  SCtgTbCache *pCache = NULL;
D
dapan1121 已提交
222 223 224 225 226
  ctgAcquireDBCache(pCtg, dbFName, &dbCache);
  if (NULL == dbCache) {
    ctgDebug("db %s not in cache", dbFName);
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
227 228

  char *stName = taosHashAcquire(dbCache->stbCache, &suid, sizeof(suid));
D
dapan1121 已提交
229
  if (NULL == stName) {
D
dapan1121 已提交
230
    ctgDebug("stb 0x%" PRIx64 " not in cache, dbFName:%s", suid, dbFName);
D
dapan1121 已提交
231 232 233 234 235
    goto _return;
  }

  pCache = taosHashAcquire(dbCache->tbCache, stName, strlen(stName));
  if (NULL == pCache) {
D
dapan1121 已提交
236
    ctgDebug("stb 0x%" PRIx64 " name %s not in cache, dbFName:%s", suid, stName, dbFName);
D
dapan1121 已提交
237 238 239 240 241 242
    taosHashRelease(dbCache->stbCache, stName);
    goto _return;
  }

  CTG_LOCK(CTG_READ, &pCache->metaLock);
  if (NULL == pCache->pMeta) {
D
dapan1121 已提交
243
    ctgDebug("stb 0x%" PRIx64 " meta not in cache, dbFName:%s", suid, dbFName);
D
dapan1121 已提交
244 245 246 247 248 249
    goto _return;
  }

  *pDb = dbCache;
  *pTb = pCache;

D
dapan1121 已提交
250
  ctgDebug("stb 0x%" PRIx64 " meta got in cache, dbFName:%s", suid, dbFName);
dengyihao's avatar
dengyihao 已提交
251

D
dapan1121 已提交
252
  CTG_CACHE_STAT_INC(numOfMetaHit, 1);
D
dapan1121 已提交
253 254 255 256 257 258 259

  return TSDB_CODE_SUCCESS;

_return:

  ctgReleaseTbMetaToCache(pCtg, dbCache, pCache);

D
dapan1121 已提交
260
  CTG_CACHE_STAT_INC(numOfMetaMiss, 1);
D
dapan1121 已提交
261 262 263

  *pDb = NULL;
  *pTb = NULL;
dengyihao's avatar
dengyihao 已提交
264

D
dapan1121 已提交
265 266 267
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
268
int32_t ctgAcquireTbIndexFromCache(SCatalog *pCtg, char *dbFName, char *tbName, SCtgDBCache **pDb, SCtgTbCache **pTb) {
D
dapan1121 已提交
269
  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
270
  SCtgTbCache *pCache = NULL;
D
dapan1121 已提交
271 272
  ctgAcquireDBCache(pCtg, dbFName, &dbCache);
  if (NULL == dbCache) {
D
dapan1121 已提交
273 274 275
    ctgDebug("db %s not in cache", dbFName);
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
276

D
dapan1121 已提交
277
  int32_t sz = 0;
D
dapan1121 已提交
278
  pCache = taosHashAcquire(dbCache->tbCache, tbName, strlen(tbName));
D
dapan1121 已提交
279 280 281 282 283 284 285 286 287
  if (NULL == pCache) {
    ctgDebug("tb %s not in cache, dbFName:%s", tbName, dbFName);
    goto _return;
  }

  CTG_LOCK(CTG_READ, &pCache->indexLock);
  if (NULL == pCache->pIndex) {
    ctgDebug("tb %s index not in cache, dbFName:%s", tbName, dbFName);
    goto _return;
D
dapan1121 已提交
288 289
  }

D
dapan1121 已提交
290 291 292 293
  *pDb = dbCache;
  *pTb = pCache;

  ctgDebug("tb %s index got in cache, dbFName:%s", tbName, dbFName);
dengyihao's avatar
dengyihao 已提交
294

D
dapan1121 已提交
295
  CTG_CACHE_STAT_INC(numOfIndexHit, 1);
D
dapan1121 已提交
296 297 298 299 300 301 302

  return TSDB_CODE_SUCCESS;

_return:

  ctgReleaseTbIndexToCache(pCtg, dbCache, pCache);

D
dapan1121 已提交
303
  CTG_CACHE_STAT_INC(numOfIndexMiss, 1);
dengyihao's avatar
dengyihao 已提交
304

D
dapan1121 已提交
305 306 307
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
308
int32_t ctgTbMetaExistInCache(SCatalog *pCtg, char *dbFName, char *tbName, int32_t *exist) {
D
dapan1121 已提交
309
  SCtgDBCache *dbCache = NULL;
D
dapan1121 已提交
310
  SCtgTbCache *tbCache = NULL;
D
dapan1121 已提交
311 312 313
  ctgAcquireTbMetaFromCache(pCtg, dbFName, tbName, &dbCache, &tbCache);
  if (NULL == tbCache) {
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
dengyihao's avatar
dengyihao 已提交
314

D
dapan1121 已提交
315 316 317 318 319
    *exist = 0;
    return TSDB_CODE_SUCCESS;
  }

  *exist = 1;
D
dapan1121 已提交
320
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
dengyihao's avatar
dengyihao 已提交
321

D
dapan1121 已提交
322 323 324
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
325 326
int32_t ctgReadTbMetaFromCache(SCatalog *pCtg, SCtgTbMetaCtx *ctx, STableMeta **pTableMeta) {
  int32_t      code = 0;
D
dapan1121 已提交
327
  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
328
  SCtgTbCache *tbCache = NULL;
D
dapan1121 已提交
329 330 331 332 333 334 335 336 337
  *pTableMeta = NULL;

  char dbFName[TSDB_DB_FNAME_LEN] = {0};
  if (CTG_FLAG_IS_SYS_DB(ctx->flag)) {
    strcpy(dbFName, ctx->pName->dbname);
  } else {
    tNameGetFullDbName(ctx->pName, dbFName);
  }

D
dapan1121 已提交
338 339 340
  ctgAcquireTbMetaFromCache(pCtg, dbFName, ctx->pName->tname, &dbCache, &tbCache);
  if (NULL == tbCache) {
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
341 342 343
    return TSDB_CODE_SUCCESS;
  }

dengyihao's avatar
dengyihao 已提交
344
  STableMeta *tbMeta = tbCache->pMeta;
D
dapan1121 已提交
345 346 347 348
  ctx->tbInfo.inCache = true;
  ctx->tbInfo.dbId = dbCache->dbId;
  ctx->tbInfo.suid = tbMeta->suid;
  ctx->tbInfo.tbType = tbMeta->tableType;
dengyihao's avatar
dengyihao 已提交
349

D
dapan1121 已提交
350
  if (tbMeta->tableType != TSDB_CHILD_TABLE) {
D
dapan1121 已提交
351 352 353 354 355 356 357 358
    int32_t metaSize = CTG_META_SIZE(tbMeta);
    *pTableMeta = taosMemoryCalloc(1, metaSize);
    if (NULL == *pTableMeta) {
      ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
      CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
    }

    memcpy(*pTableMeta, tbMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
359

D
dapan1121 已提交
360 361
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
    ctgDebug("Got tb %s meta from cache, type:%d, dbFName:%s", ctx->pName->tname, tbMeta->tableType, dbFName);
D
dapan1121 已提交
362 363
    return TSDB_CODE_SUCCESS;
  }
D
dapan1121 已提交
364 365

  // PROCESS FOR CHILD TABLE
dengyihao's avatar
dengyihao 已提交
366

D
dapan1121 已提交
367 368 369 370
  int32_t metaSize = sizeof(SCTableMeta);
  *pTableMeta = taosMemoryCalloc(1, metaSize);
  if (NULL == *pTableMeta) {
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
371 372
  }

D
dapan1121 已提交
373
  memcpy(*pTableMeta, tbMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
374

D
dapan1121 已提交
375
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
dengyihao's avatar
dengyihao 已提交
376 377
  ctgDebug("Got ctb %s meta from cache, will continue to get its stb meta, type:%d, dbFName:%s", ctx->pName->tname,
           ctx->tbInfo.tbType, dbFName);
D
dapan1121 已提交
378 379 380 381 382

  ctgAcquireStbMetaFromCache(pCtg, dbFName, ctx->tbInfo.suid, &dbCache, &tbCache);
  if (NULL == tbCache) {
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
    taosMemoryFreeClear(*pTableMeta);
D
dapan1121 已提交
383
    ctgDebug("stb 0x%" PRIx64 " meta not in cache", ctx->tbInfo.suid);
D
dapan1121 已提交
384 385
    return TSDB_CODE_SUCCESS;
  }
dengyihao's avatar
dengyihao 已提交
386 387 388

  STableMeta *stbMeta = tbCache->pMeta;
  if (stbMeta->suid != ctx->tbInfo.suid) {
D
dapan1121 已提交
389
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
dengyihao's avatar
dengyihao 已提交
390
    ctgError("stb suid 0x%" PRIx64 " in stbCache mis-match, expected suid 0x%" PRIx64, stbMeta->suid, ctx->tbInfo.suid);
D
dapan1121 已提交
391 392 393
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

D
dapan1121 已提交
394
  metaSize = CTG_META_SIZE(stbMeta);
D
dapan1121 已提交
395
  *pTableMeta = taosMemoryRealloc(*pTableMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
396
  if (NULL == *pTableMeta) {
D
dapan1121 已提交
397
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
398
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
399 400
  }

D
dapan1121 已提交
401
  memcpy(&(*pTableMeta)->sversion, &stbMeta->sversion, metaSize - sizeof(SCTableMeta));
D
dapan1121 已提交
402

D
dapan1121 已提交
403
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
404

D
dapan1121 已提交
405
  ctgDebug("Got tb %s meta from cache, dbFName:%s", ctx->pName->tname, dbFName);
dengyihao's avatar
dengyihao 已提交
406

D
dapan1121 已提交
407 408 409 410
  return TSDB_CODE_SUCCESS;

_return:

D
dapan1121 已提交
411
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
412
  taosMemoryFreeClear(*pTableMeta);
dengyihao's avatar
dengyihao 已提交
413

D
dapan1121 已提交
414 415 416
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
417 418
int32_t ctgReadTbVerFromCache(SCatalog *pCtg, SName *pTableName, int32_t *sver, int32_t *tver, int32_t *tbType,
                              uint64_t *suid, char *stbName) {
D
dapan1121 已提交
419
  *sver = -1;
D
dapan1121 已提交
420
  *tver = -1;
D
dapan1121 已提交
421 422

  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
423
  SCtgTbCache *tbCache = NULL;
D
dapan1121 已提交
424
  char         dbFName[TSDB_DB_FNAME_LEN] = {0};
D
dapan1121 已提交
425 426
  tNameGetFullDbName(pTableName, dbFName);

D
dapan1121 已提交
427 428 429
  ctgAcquireTbMetaFromCache(pCtg, dbFName, pTableName->tname, &dbCache, &tbCache);
  if (NULL == tbCache) {
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
430 431 432
    return TSDB_CODE_SUCCESS;
  }

dengyihao's avatar
dengyihao 已提交
433
  STableMeta *tbMeta = tbCache->pMeta;
D
dapan1121 已提交
434 435
  *tbType = tbMeta->tableType;
  *suid = tbMeta->suid;
D
dapan1121 已提交
436

D
dapan1121 已提交
437
  if (*tbType != TSDB_CHILD_TABLE) {
D
dapan1121 已提交
438 439 440
    *sver = tbMeta->sversion;
    *tver = tbMeta->tversion;

dengyihao's avatar
dengyihao 已提交
441 442
    ctgDebug("Got tb %s ver from cache, dbFName:%s, tbType:%d, sver:%d, tver:%d, suid:0x%" PRIx64, pTableName->tname,
             dbFName, *tbType, *sver, *tver, *suid);
D
dapan1121 已提交
443

D
dapan1121 已提交
444
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
445 446 447
    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
448
  // PROCESS FOR CHILD TABLE
dengyihao's avatar
dengyihao 已提交
449

D
dapan1121 已提交
450 451
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
  ctgDebug("Got ctb %s ver from cache, will continue to get its stb ver, dbFName:%s", pTableName->tname, dbFName);
dengyihao's avatar
dengyihao 已提交
452

D
dapan1121 已提交
453 454
  ctgAcquireStbMetaFromCache(pCtg, dbFName, *suid, &dbCache, &tbCache);
  if (NULL == tbCache) {
D
dapan1121 已提交
455
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
456
    ctgDebug("stb 0x%" PRIx64 " meta not in cache", *suid);
D
dapan1121 已提交
457 458
    return TSDB_CODE_SUCCESS;
  }
dengyihao's avatar
dengyihao 已提交
459 460

  STableMeta *stbMeta = tbCache->pMeta;
D
dapan1121 已提交
461
  if (stbMeta->suid != *suid) {
D
dapan1121 已提交
462
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
dengyihao's avatar
dengyihao 已提交
463
    ctgError("stb suid 0x%" PRIx64 " in stbCache mis-match, expected suid:0x%" PRIx64, stbMeta->suid, *suid);
D
dapan1121 已提交
464 465 466
    CTG_ERR_RET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

D
dapan1121 已提交
467
  size_t nameLen = 0;
D
dapan1121 已提交
468
  char  *name = taosHashGetKey(tbCache, &nameLen);
D
dapan1121 已提交
469 470 471 472

  strncpy(stbName, name, nameLen);
  stbName[nameLen] = 0;

D
dapan1121 已提交
473 474
  *sver = stbMeta->sversion;
  *tver = stbMeta->tversion;
D
dapan1121 已提交
475

D
dapan1121 已提交
476
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
477

dengyihao's avatar
dengyihao 已提交
478 479
  ctgDebug("Got tb %s sver %d tver %d from cache, type:%d, dbFName:%s", pTableName->tname, *sver, *tver, *tbType,
           dbFName);
D
dapan1121 已提交
480 481 482 483

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
484
int32_t ctgReadTbTypeFromCache(SCatalog *pCtg, char *dbFName, char *tbName, int32_t *tbType) {
D
dapan1121 已提交
485
  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
486
  SCtgTbCache *tbCache = NULL;
D
dapan1121 已提交
487
  CTG_ERR_RET(ctgAcquireTbMetaFromCache(pCtg, dbFName, tbName, &dbCache, &tbCache));
D
dapan1121 已提交
488 489
  if (NULL == tbCache) {
    ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
490 491
    return TSDB_CODE_SUCCESS;
  }
D
dapan1121 已提交
492 493 494 495

  *tbType = tbCache->pMeta->tableType;
  ctgReleaseTbMetaToCache(pCtg, dbCache, tbCache);

dengyihao's avatar
dengyihao 已提交
496 497
  ctgDebug("Got tb %s tbType %d from cache, dbFName:%s", tbName, *tbType, dbFName);

D
dapan1121 已提交
498 499 500
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
501 502
int32_t ctgReadTbIndexFromCache(SCatalog *pCtg, SName *pTableName, SArray **pRes) {
  int32_t      code = 0;
D
dapan1121 已提交
503
  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
504
  SCtgTbCache *tbCache = NULL;
D
dapan1121 已提交
505 506
  char         dbFName[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(pTableName, dbFName);
D
dapan1121 已提交
507

D
dapan1121 已提交
508
  *pRes = NULL;
D
dapan1121 已提交
509

D
dapan1121 已提交
510 511 512
  ctgAcquireTbIndexFromCache(pCtg, dbFName, pTableName->tname, &dbCache, &tbCache);
  if (NULL == tbCache) {
    ctgReleaseTbIndexToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
513 514 515
    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
516
  CTG_ERR_JRET(ctgCloneTableIndex(tbCache->pIndex->pIndex, pRes));
D
dapan1121 已提交
517

D
dapan1121 已提交
518
_return:
D
dapan1121 已提交
519

D
dapan1121 已提交
520
  ctgReleaseTbIndexToCache(pCtg, dbCache, tbCache);
D
dapan1121 已提交
521

D
dapan1121 已提交
522
  CTG_RET(code);
D
dapan1121 已提交
523 524
}

dengyihao's avatar
dengyihao 已提交
525
int32_t ctgChkAuthFromCache(SCatalog *pCtg, char *user, char *dbFName, AUTH_TYPE type, bool *inCache, bool *pass) {
526 527 528 529 530 531
  char *p = strchr(dbFName, '.');
  if (p) {
    ++p;
  } else {
    p = dbFName;
  }
dengyihao's avatar
dengyihao 已提交
532

533 534 535 536 537 538 539
  if (IS_SYS_DBNAME(p)) {
    *inCache = true;
    *pass = true;
    ctgDebug("sysdb %s, pass", dbFName);
    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
540 541 542 543 544 545 546 547 548
  SCtgUserAuth *pUser = (SCtgUserAuth *)taosHashGet(pCtg->userCache, user, strlen(user));
  if (NULL == pUser) {
    ctgDebug("user not in cache, user:%s", user);
    goto _return;
  }

  *inCache = true;

  ctgDebug("Got user from cache, user:%s", user);
D
dapan1121 已提交
549
  CTG_CACHE_STAT_INC(numOfUserHit, 1);
dengyihao's avatar
dengyihao 已提交
550

D
dapan1121 已提交
551 552 553 554 555 556 557 558 559 560 561
  if (pUser->superUser) {
    *pass = true;
    return TSDB_CODE_SUCCESS;
  }

  CTG_LOCK(CTG_READ, &pUser->lock);
  if (pUser->createdDbs && taosHashGet(pUser->createdDbs, dbFName, strlen(dbFName))) {
    *pass = true;
    CTG_UNLOCK(CTG_READ, &pUser->lock);
    return TSDB_CODE_SUCCESS;
  }
dengyihao's avatar
dengyihao 已提交
562

D
dapan1121 已提交
563 564 565
  if (pUser->readDbs && taosHashGet(pUser->readDbs, dbFName, strlen(dbFName)) && type == AUTH_TYPE_READ) {
    *pass = true;
  }
dengyihao's avatar
dengyihao 已提交
566

D
dapan1121 已提交
567 568 569 570 571
  if (pUser->writeDbs && taosHashGet(pUser->writeDbs, dbFName, strlen(dbFName)) && type == AUTH_TYPE_WRITE) {
    *pass = true;
  }

  CTG_UNLOCK(CTG_READ, &pUser->lock);
dengyihao's avatar
dengyihao 已提交
572

D
dapan1121 已提交
573 574 575 576 577
  return TSDB_CODE_SUCCESS;

_return:

  *inCache = false;
D
dapan1121 已提交
578
  CTG_CACHE_STAT_INC(numOfUserMiss, 1);
dengyihao's avatar
dengyihao 已提交
579

D
dapan1121 已提交
580 581 582
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
583
void ctgDequeue(SCtgCacheOperation **op) {
D
dapan1121 已提交
584
  SCtgQNode *orig = gCtgMgmt.queue.head;
dengyihao's avatar
dengyihao 已提交
585

D
dapan1121 已提交
586 587 588
  SCtgQNode *node = gCtgMgmt.queue.head->next;
  gCtgMgmt.queue.head = gCtgMgmt.queue.head->next;

D
dapan1121 已提交
589
  CTG_QUEUE_DEC();
dengyihao's avatar
dengyihao 已提交
590

D
dapan1121 已提交
591 592
  taosMemoryFreeClear(orig);

D
dapan1121 已提交
593
  *op = node->op;
D
dapan1121 已提交
594 595
}

dengyihao's avatar
dengyihao 已提交
596
int32_t ctgEnqueue(SCatalog *pCtg, SCtgCacheOperation *operation) {
D
dapan1121 已提交
597 598 599
  SCtgQNode *node = taosMemoryCalloc(1, sizeof(SCtgQNode));
  if (NULL == node) {
    qError("calloc %d failed", (int32_t)sizeof(SCtgQNode));
D
dapan1121 已提交
600 601
    taosMemoryFree(operation->data);
    taosMemoryFree(operation);
D
dapan1121 已提交
602
    CTG_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
603 604
  }

dengyihao's avatar
dengyihao 已提交
605 606
  bool  syncOp = operation->syncOp;
  char *opName = gCtgCacheOperation[operation->opId].name;
D
dapan1121 已提交
607 608 609
  if (operation->syncOp) {
    tsem_init(&operation->rspSem, 0, 0);
  }
dengyihao's avatar
dengyihao 已提交
610

D
dapan1121 已提交
611
  node->op = operation;
D
dapan1121 已提交
612 613

  CTG_LOCK(CTG_WRITE, &gCtgMgmt.queue.qlock);
D
dapan1121 已提交
614
  if (gCtgMgmt.queue.stopQueue) {
615 616 617 618
    ctgFreeQNode(node);
    CTG_UNLOCK(CTG_WRITE, &gCtgMgmt.queue.qlock);
    CTG_RET(TSDB_CODE_CTG_EXIT);
  }
D
dapan1121 已提交
619
  gCtgMgmt.queue.stopQueue = operation->stopQueue;
D
dapan1121 已提交
620 621 622 623
  gCtgMgmt.queue.tail->next = node;
  gCtgMgmt.queue.tail = node;
  CTG_UNLOCK(CTG_WRITE, &gCtgMgmt.queue.qlock);

D
dapan1121 已提交
624 625
  ctgDebug("action [%s] added into queue", opName);

D
dapan1121 已提交
626
  CTG_QUEUE_INC();
D
dapan1121 已提交
627
  CTG_RT_STAT_INC(numOfOpEnqueue, 1);
D
dapan1121 已提交
628 629 630

  tsem_post(&gCtgMgmt.queue.reqSem);

D
dapan1121 已提交
631
  if (syncOp) {
632 633 634
    if (!operation->unLocked) {
      CTG_UNLOCK(CTG_READ, &gCtgMgmt.lock);
    }
D
dapan1121 已提交
635
    tsem_wait(&operation->rspSem);
636 637 638
    if (!operation->unLocked) {
      CTG_LOCK(CTG_READ, &gCtgMgmt.lock);
    }
D
dapan1121 已提交
639
    taosMemoryFree(operation);
D
dapan1121 已提交
640 641 642 643 644
  }

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
645 646
int32_t ctgDropDbCacheEnqueue(SCatalog *pCtg, const char *dbFName, int64_t dbId) {
  int32_t             code = 0;
D
dapan1121 已提交
647 648
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_DROP_DB_CACHE;
dengyihao's avatar
dengyihao 已提交
649

D
dapan1121 已提交
650
  SCtgDropDBMsg *msg = taosMemoryMalloc(sizeof(SCtgDropDBMsg));
D
dapan1121 已提交
651
  if (NULL == msg) {
D
dapan1121 已提交
652
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgDropDBMsg));
D
dapan1121 已提交
653
    taosMemoryFree(op);
D
dapan1121 已提交
654
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
655 656 657
  }

  char *p = strchr(dbFName, '.');
D
dapan1121 已提交
658
  if (p && IS_SYS_DBNAME(p + 1)) {
D
dapan1121 已提交
659 660 661 662
    dbFName = p + 1;
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
663
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
D
dapan1121 已提交
664 665
  msg->dbId = dbId;

D
dapan1121 已提交
666
  op->data = msg;
D
dapan1121 已提交
667

D
dapan1121 已提交
668
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
669 670 671 672 673 674 675 676

  return TSDB_CODE_SUCCESS;

_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
677 678
int32_t ctgDropDbVgroupEnqueue(SCatalog *pCtg, const char *dbFName, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
679 680 681
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_DROP_DB_VGROUP;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
682

D
dapan1121 已提交
683 684 685
  SCtgDropDbVgroupMsg *msg = taosMemoryMalloc(sizeof(SCtgDropDbVgroupMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgDropDbVgroupMsg));
D
dapan1121 已提交
686
    taosMemoryFree(op);
D
dapan1121 已提交
687
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
688 689 690
  }

  char *p = strchr(dbFName, '.');
D
dapan1121 已提交
691
  if (p && IS_SYS_DBNAME(p + 1)) {
D
dapan1121 已提交
692 693 694 695
    dbFName = p + 1;
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
696
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
D
dapan1121 已提交
697

D
dapan1121 已提交
698
  op->data = msg;
D
dapan1121 已提交
699

D
dapan1121 已提交
700
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
701 702 703 704 705 706 707 708

  return TSDB_CODE_SUCCESS;

_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
709 710 711
int32_t ctgDropStbMetaEnqueue(SCatalog *pCtg, const char *dbFName, int64_t dbId, const char *stbName, uint64_t suid,
                              bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
712 713 714
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_DROP_STB_META;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
715

D
dapan1121 已提交
716
  SCtgDropStbMetaMsg *msg = taosMemoryMalloc(sizeof(SCtgDropStbMetaMsg));
D
dapan1121 已提交
717
  if (NULL == msg) {
D
dapan1121 已提交
718
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgDropStbMetaMsg));
D
dapan1121 已提交
719
    taosMemoryFree(op);
D
dapan1121 已提交
720
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
721 722 723
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
724 725
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
  tstrncpy(msg->stbName, stbName, sizeof(msg->stbName));
D
dapan1121 已提交
726 727 728
  msg->dbId = dbId;
  msg->suid = suid;

D
dapan1121 已提交
729
  op->data = msg;
D
dapan1121 已提交
730

D
dapan1121 已提交
731
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
732 733 734 735 736 737 738 739

  return TSDB_CODE_SUCCESS;

_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
740 741
int32_t ctgDropTbMetaEnqueue(SCatalog *pCtg, const char *dbFName, int64_t dbId, const char *tbName, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
742 743 744
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_DROP_TB_META;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
745

D
dapan1121 已提交
746
  SCtgDropTblMetaMsg *msg = taosMemoryMalloc(sizeof(SCtgDropTblMetaMsg));
D
dapan1121 已提交
747
  if (NULL == msg) {
D
dapan1121 已提交
748
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgDropTblMetaMsg));
D
dapan1121 已提交
749
    taosMemoryFree(op);
D
dapan1121 已提交
750
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
751 752 753
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
754 755
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
  tstrncpy(msg->tbName, tbName, sizeof(msg->tbName));
D
dapan1121 已提交
756 757
  msg->dbId = dbId;

D
dapan1121 已提交
758
  op->data = msg;
D
dapan1121 已提交
759

D
dapan1121 已提交
760
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
761 762 763 764 765 766 767 768

  return TSDB_CODE_SUCCESS;

_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
769 770
int32_t ctgUpdateVgroupEnqueue(SCatalog *pCtg, const char *dbFName, int64_t dbId, SDBVgInfo *dbInfo, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
771 772 773
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_UPDATE_VGROUP;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
774

D
dapan1121 已提交
775 776 777
  SCtgUpdateVgMsg *msg = taosMemoryMalloc(sizeof(SCtgUpdateVgMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgUpdateVgMsg));
D
dapan1121 已提交
778
    taosMemoryFree(op);
D
dapan1121 已提交
779
    ctgFreeVgInfo(dbInfo);
D
dapan1121 已提交
780
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
781 782 783
  }

  char *p = strchr(dbFName, '.');
D
dapan1121 已提交
784
  if (p && IS_SYS_DBNAME(p + 1)) {
D
dapan1121 已提交
785 786 787
    dbFName = p + 1;
  }

D
dapan1121 已提交
788
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
D
dapan1121 已提交
789 790 791 792
  msg->pCtg = pCtg;
  msg->dbId = dbId;
  msg->dbInfo = dbInfo;

D
dapan1121 已提交
793
  op->data = msg;
D
dapan1121 已提交
794

D
dapan1121 已提交
795
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
796 797 798 799 800 801 802 803 804

  return TSDB_CODE_SUCCESS;

_return:

  ctgFreeVgInfo(dbInfo);
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
805 806
int32_t ctgUpdateTbMetaEnqueue(SCatalog *pCtg, STableMetaOutput *output, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
807 808 809
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_UPDATE_TB_META;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
810

D
dapan1121 已提交
811
  SCtgUpdateTbMetaMsg *msg = taosMemoryMalloc(sizeof(SCtgUpdateTbMetaMsg));
D
dapan1121 已提交
812
  if (NULL == msg) {
D
dapan1121 已提交
813
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgUpdateTbMetaMsg));
D
dapan1121 已提交
814
    taosMemoryFree(op);
D
dapan1121 已提交
815
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
816 817 818
  }

  char *p = strchr(output->dbFName, '.');
D
dapan1121 已提交
819
  if (p && IS_SYS_DBNAME(p + 1)) {
D
dapan1121 已提交
820 821
    int32_t len = strlen(p + 1);
    memmove(output->dbFName, p + 1, len >= TSDB_DB_FNAME_LEN ? TSDB_DB_FNAME_LEN - 1 : len);
D
dapan1121 已提交
822 823 824
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
825
  msg->pMeta = output;
D
dapan1121 已提交
826

D
dapan1121 已提交
827
  op->data = msg;
D
dapan1121 已提交
828

D
dapan1121 已提交
829
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
830 831

  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
832

D
dapan1121 已提交
833 834
_return:

D
dapan1121 已提交
835 836 837 838 839
  if (output) {
    taosMemoryFree(output->tbMeta);
    taosMemoryFree(output);
  }

D
dapan1121 已提交
840 841 842
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
843 844
int32_t ctgUpdateVgEpsetEnqueue(SCatalog *pCtg, char *dbFName, int32_t vgId, SEpSet *pEpSet) {
  int32_t             code = 0;
D
dapan1121 已提交
845 846
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_UPDATE_VG_EPSET;
dengyihao's avatar
dengyihao 已提交
847

D
dapan1121 已提交
848 849 850
  SCtgUpdateEpsetMsg *msg = taosMemoryMalloc(sizeof(SCtgUpdateEpsetMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgUpdateEpsetMsg));
D
dapan1121 已提交
851
    taosMemoryFree(op);
D
dapan1121 已提交
852
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
853 854 855
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
856
  tstrncpy(msg->dbFName, dbFName, sizeof(msg->dbFName));
D
dapan1121 已提交
857 858 859
  msg->vgId = vgId;
  msg->epSet = *pEpSet;

D
dapan1121 已提交
860
  op->data = msg;
D
dapan1121 已提交
861

D
dapan1121 已提交
862
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
863 864

  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
865

D
dapan1121 已提交
866 867 868 869 870
_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
871 872
int32_t ctgUpdateUserEnqueue(SCatalog *pCtg, SGetUserAuthRsp *pAuth, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
873 874 875
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_UPDATE_USER;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
876

D
dapan1121 已提交
877 878 879
  SCtgUpdateUserMsg *msg = taosMemoryMalloc(sizeof(SCtgUpdateUserMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgUpdateUserMsg));
D
dapan1121 已提交
880
    taosMemoryFree(op);
D
dapan1121 已提交
881
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
882 883 884 885 886
  }

  msg->pCtg = pCtg;
  msg->userAuth = *pAuth;

D
dapan1121 已提交
887
  op->data = msg;
D
dapan1121 已提交
888

D
dapan1121 已提交
889
  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
dengyihao's avatar
dengyihao 已提交
890

D
dapan1121 已提交
891
  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
892

D
dapan1121 已提交
893 894 895
_return:

  tFreeSGetUserAuthRsp(pAuth);
dengyihao's avatar
dengyihao 已提交
896

D
dapan1121 已提交
897 898 899
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
900 901
int32_t ctgUpdateTbIndexEnqueue(SCatalog *pCtg, STableIndex **pIndex, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
902 903 904
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_UPDATE_TB_INDEX;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
905

D
dapan1121 已提交
906 907 908
  SCtgUpdateTbIndexMsg *msg = taosMemoryMalloc(sizeof(SCtgUpdateTbIndexMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgUpdateTbIndexMsg));
D
dapan1121 已提交
909
    taosMemoryFree(op);
D
dapan1121 已提交
910
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
911 912 913
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
914
  msg->pIndex = *pIndex;
D
dapan1121 已提交
915 916 917 918

  op->data = msg;

  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
D
dapan1121 已提交
919 920

  *pIndex = NULL;
D
dapan1121 已提交
921
  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
922

D
dapan1121 已提交
923 924
_return:

D
dapan1121 已提交
925
  taosArrayDestroyEx((*pIndex)->pIndex, tFreeSTableIndexInfo);
D
dapan1121 已提交
926
  taosMemoryFreeClear(*pIndex);
dengyihao's avatar
dengyihao 已提交
927

D
dapan1121 已提交
928 929 930
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
931 932
int32_t ctgDropTbIndexEnqueue(SCatalog *pCtg, SName *pName, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
933 934 935
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_DROP_TB_INDEX;
  op->syncOp = syncOp;
dengyihao's avatar
dengyihao 已提交
936

D
dapan1121 已提交
937 938 939
  SCtgDropTbIndexMsg *msg = taosMemoryMalloc(sizeof(SCtgDropTbIndexMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgDropTbIndexMsg));
D
dapan1121 已提交
940
    taosMemoryFree(op);
D
dapan1121 已提交
941
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
942 943 944 945 946 947 948 949 950
  }

  msg->pCtg = pCtg;
  tNameGetFullDbName(pName, msg->dbFName);
  strcpy(msg->tbName, pName->tname);

  op->data = msg;

  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
dengyihao's avatar
dengyihao 已提交
951

D
dapan1121 已提交
952
  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
953

D
dapan1121 已提交
954 955 956 957 958
_return:

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
959 960
int32_t ctgClearCacheEnqueue(SCatalog *pCtg, bool freeCtg, bool stopQueue, bool syncOp) {
  int32_t             code = 0;
D
dapan1121 已提交
961 962 963
  SCtgCacheOperation *op = taosMemoryCalloc(1, sizeof(SCtgCacheOperation));
  op->opId = CTG_OP_CLEAR_CACHE;
  op->syncOp = syncOp;
D
dapan1121 已提交
964
  op->stopQueue = stopQueue;
965
  op->unLocked = true;
dengyihao's avatar
dengyihao 已提交
966

D
dapan1121 已提交
967 968 969
  SCtgClearCacheMsg *msg = taosMemoryMalloc(sizeof(SCtgClearCacheMsg));
  if (NULL == msg) {
    ctgError("malloc %d failed", (int32_t)sizeof(SCtgClearCacheMsg));
D
dapan1121 已提交
970
    taosMemoryFree(op);
D
dapan1121 已提交
971
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
972 973 974
  }

  msg->pCtg = pCtg;
D
dapan1121 已提交
975
  msg->freeCtg = freeCtg;
D
dapan1121 已提交
976 977 978
  op->data = msg;

  CTG_ERR_JRET(ctgEnqueue(pCtg, op));
dengyihao's avatar
dengyihao 已提交
979

D
dapan1121 已提交
980
  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
981

D
dapan1121 已提交
982 983 984 985 986
_return:

  CTG_RET(code);
}

D
dapan1121 已提交
987 988 989 990 991 992
int32_t ctgMetaRentInit(SCtgRentMgmt *mgmt, uint32_t rentSec, int8_t type) {
  mgmt->slotRIdx = 0;
  mgmt->slotNum = rentSec / CTG_RENT_SLOT_SECOND;
  mgmt->type = type;

  size_t msgSize = sizeof(SCtgRentSlot) * mgmt->slotNum;
dengyihao's avatar
dengyihao 已提交
993

D
dapan1121 已提交
994 995 996
  mgmt->slots = taosMemoryCalloc(1, msgSize);
  if (NULL == mgmt->slots) {
    qError("calloc %d failed", (int32_t)msgSize);
D
dapan1121 已提交
997
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
998 999 1000
  }

  qDebug("meta rent initialized, type:%d, slotNum:%d", type, mgmt->slotNum);
dengyihao's avatar
dengyihao 已提交
1001

D
dapan1121 已提交
1002 1003 1004 1005 1006 1007 1008
  return TSDB_CODE_SUCCESS;
}

int32_t ctgMetaRentAdd(SCtgRentMgmt *mgmt, void *meta, int64_t id, int32_t size) {
  int16_t widx = abs((int)(id % mgmt->slotNum));

  SCtgRentSlot *slot = &mgmt->slots[widx];
dengyihao's avatar
dengyihao 已提交
1009 1010
  int32_t       code = 0;

D
dapan1121 已提交
1011 1012 1013 1014
  CTG_LOCK(CTG_WRITE, &slot->lock);
  if (NULL == slot->meta) {
    slot->meta = taosArrayInit(CTG_DEFAULT_RENT_SLOT_SIZE, size);
    if (NULL == slot->meta) {
dengyihao's avatar
dengyihao 已提交
1015 1016
      qError("taosArrayInit %d failed, id:0x%" PRIx64 ", slot idx:%d, type:%d", CTG_DEFAULT_RENT_SLOT_SIZE, id, widx,
             mgmt->type);
D
dapan1121 已提交
1017
      CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1018 1019 1020 1021
    }
  }

  if (NULL == taosArrayPush(slot->meta, meta)) {
dengyihao's avatar
dengyihao 已提交
1022
    qError("taosArrayPush meta to rent failed, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1023
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1024 1025 1026 1027
  }

  slot->needSort = true;

dengyihao's avatar
dengyihao 已提交
1028
  qDebug("add meta to rent, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1029 1030 1031 1032 1033 1034 1035

_return:

  CTG_UNLOCK(CTG_WRITE, &slot->lock);
  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
1036 1037
int32_t ctgMetaRentUpdate(SCtgRentMgmt *mgmt, void *meta, int64_t id, int32_t size, __compar_fn_t sortCompare,
                          __compar_fn_t searchCompare) {
D
dapan1121 已提交
1038 1039 1040
  int16_t widx = abs((int)(id % mgmt->slotNum));

  SCtgRentSlot *slot = &mgmt->slots[widx];
dengyihao's avatar
dengyihao 已提交
1041
  int32_t       code = 0;
D
dapan1121 已提交
1042 1043 1044

  CTG_LOCK(CTG_WRITE, &slot->lock);
  if (NULL == slot->meta) {
dengyihao's avatar
dengyihao 已提交
1045
    qDebug("empty meta slot, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1046 1047 1048 1049
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

  if (slot->needSort) {
dengyihao's avatar
dengyihao 已提交
1050 1051
    qDebug("meta slot before sorte, slot idx:%d, type:%d, size:%d", widx, mgmt->type,
           (int32_t)taosArrayGetSize(slot->meta));
D
dapan1121 已提交
1052 1053 1054 1055 1056 1057 1058
    taosArraySort(slot->meta, sortCompare);
    slot->needSort = false;
    qDebug("meta slot sorted, slot idx:%d, type:%d, size:%d", widx, mgmt->type, (int32_t)taosArrayGetSize(slot->meta));
  }

  void *orig = taosArraySearch(slot->meta, &id, searchCompare, TD_EQ);
  if (NULL == orig) {
dengyihao's avatar
dengyihao 已提交
1059 1060
    qDebug("meta not found in slot, id:0x%" PRIx64 ", slot idx:%d, type:%d, size:%d", id, widx, mgmt->type,
           (int32_t)taosArrayGetSize(slot->meta));
D
dapan1121 已提交
1061 1062 1063 1064 1065
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

  memcpy(orig, meta, size);

dengyihao's avatar
dengyihao 已提交
1066
  qDebug("meta in rent updated, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1067 1068 1069 1070 1071 1072

_return:

  CTG_UNLOCK(CTG_WRITE, &slot->lock);

  if (code) {
dengyihao's avatar
dengyihao 已提交
1073 1074
    qDebug("meta in rent update failed, will try to add it, code:%x, id:0x%" PRIx64 ", slot idx:%d, type:%d", code, id,
           widx, mgmt->type);
D
dapan1121 已提交
1075 1076 1077 1078 1079 1080 1081 1082 1083 1084
    CTG_RET(ctgMetaRentAdd(mgmt, meta, id, size));
  }

  CTG_RET(code);
}

int32_t ctgMetaRentRemove(SCtgRentMgmt *mgmt, int64_t id, __compar_fn_t sortCompare, __compar_fn_t searchCompare) {
  int16_t widx = abs((int)(id % mgmt->slotNum));

  SCtgRentSlot *slot = &mgmt->slots[widx];
dengyihao's avatar
dengyihao 已提交
1085 1086
  int32_t       code = 0;

D
dapan1121 已提交
1087 1088
  CTG_LOCK(CTG_WRITE, &slot->lock);
  if (NULL == slot->meta) {
dengyihao's avatar
dengyihao 已提交
1089
    qError("empty meta slot, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

  if (slot->needSort) {
    taosArraySort(slot->meta, sortCompare);
    slot->needSort = false;
    qDebug("meta slot sorted, slot idx:%d, type:%d", widx, mgmt->type);
  }

  int32_t idx = taosArraySearchIdx(slot->meta, &id, searchCompare, TD_EQ);
  if (idx < 0) {
dengyihao's avatar
dengyihao 已提交
1101
    qError("meta not found in slot, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1102 1103 1104 1105 1106
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

  taosArrayRemove(slot->meta, idx);

dengyihao's avatar
dengyihao 已提交
1107
  qDebug("meta in rent removed, id:0x%" PRIx64 ", slot idx:%d, type:%d", id, widx, mgmt->type);
D
dapan1121 已提交
1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123

_return:

  CTG_UNLOCK(CTG_WRITE, &slot->lock);

  CTG_RET(code);
}

int32_t ctgMetaRentGetImpl(SCtgRentMgmt *mgmt, void **res, uint32_t *num, int32_t size) {
  int16_t ridx = atomic_add_fetch_16(&mgmt->slotRIdx, 1);
  if (ridx >= mgmt->slotNum) {
    ridx %= mgmt->slotNum;
    atomic_store_16(&mgmt->slotRIdx, ridx);
  }

  SCtgRentSlot *slot = &mgmt->slots[ridx];
dengyihao's avatar
dengyihao 已提交
1124 1125
  int32_t       code = 0;

D
dapan1121 已提交
1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143
  CTG_LOCK(CTG_READ, &slot->lock);
  if (NULL == slot->meta) {
    qDebug("empty meta in slot:%d, type:%d", ridx, mgmt->type);
    *num = 0;
    goto _return;
  }

  size_t metaNum = taosArrayGetSize(slot->meta);
  if (metaNum <= 0) {
    qDebug("no meta in slot:%d, type:%d", ridx, mgmt->type);
    *num = 0;
    goto _return;
  }

  size_t msize = metaNum * size;
  *res = taosMemoryMalloc(msize);
  if (NULL == *res) {
    qError("malloc %d failed", (int32_t)msize);
D
dapan1121 已提交
1144
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190
  }

  void *meta = taosArrayGet(slot->meta, 0);

  memcpy(*res, meta, msize);

  *num = (uint32_t)metaNum;

  qDebug("Got %d meta from rent, type:%d", (int32_t)metaNum, mgmt->type);

_return:

  CTG_UNLOCK(CTG_READ, &slot->lock);

  CTG_RET(code);
}

int32_t ctgMetaRentGet(SCtgRentMgmt *mgmt, void **res, uint32_t *num, int32_t size) {
  while (true) {
    int64_t msec = taosGetTimestampMs();
    int64_t lsec = atomic_load_64(&mgmt->lastReadMsec);
    if ((msec - lsec) < CTG_RENT_SLOT_SECOND * 1000) {
      *res = NULL;
      *num = 0;
      qDebug("too short time period to get expired meta, type:%d", mgmt->type);
      return TSDB_CODE_SUCCESS;
    }

    if (lsec != atomic_val_compare_exchange_64(&mgmt->lastReadMsec, lsec, msec)) {
      continue;
    }

    break;
  }

  CTG_ERR_RET(ctgMetaRentGetImpl(mgmt, res, num, size));

  return TSDB_CODE_SUCCESS;
}

int32_t ctgAddNewDBCache(SCatalog *pCtg, const char *dbFName, uint64_t dbId) {
  int32_t code = 0;

  SCtgDBCache newDBCache = {0};
  newDBCache.dbId = dbId;

dengyihao's avatar
dengyihao 已提交
1191 1192
  newDBCache.tbCache = taosHashInit(gCtgMgmt.cfg.maxTblCacheNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY),
                                    true, HASH_ENTRY_LOCK);
D
dapan1121 已提交
1193
  if (NULL == newDBCache.tbCache) {
D
dapan1121 已提交
1194
    ctgError("taosHashInit %d metaCache failed", gCtgMgmt.cfg.maxTblCacheNum);
D
dapan1121 已提交
1195
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1196 1197
  }

dengyihao's avatar
dengyihao 已提交
1198 1199
  newDBCache.stbCache = taosHashInit(gCtgMgmt.cfg.maxTblCacheNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT),
                                     true, HASH_ENTRY_LOCK);
D
dapan1121 已提交
1200
  if (NULL == newDBCache.stbCache) {
D
dapan1121 已提交
1201
    ctgError("taosHashInit %d stbCache failed", gCtgMgmt.cfg.maxTblCacheNum);
D
dapan1121 已提交
1202
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1203 1204 1205 1206 1207 1208 1209 1210
  }

  code = taosHashPut(pCtg->dbCache, dbFName, strlen(dbFName), &newDBCache, sizeof(SCtgDBCache));
  if (code) {
    if (HASH_NODE_EXIST(code)) {
      ctgDebug("db already in cache, dbFName:%s", dbFName);
      goto _return;
    }
dengyihao's avatar
dengyihao 已提交
1211

D
dapan1121 已提交
1212
    ctgError("taosHashPut db to cache failed, dbFName:%s", dbFName);
D
dapan1121 已提交
1213
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1214 1215
  }

D
dapan1121 已提交
1216
  CTG_CACHE_STAT_INC(numOfDb, 1);
dengyihao's avatar
dengyihao 已提交
1217

D
dapan1121 已提交
1218
  SDbVgVersion vgVersion = {.dbId = newDBCache.dbId, .vgVersion = -1};
D
dapan1121 已提交
1219
  tstrncpy(vgVersion.dbFName, dbFName, sizeof(vgVersion.dbFName));
D
dapan1121 已提交
1220

dengyihao's avatar
dengyihao 已提交
1221
  ctgDebug("db added to cache, dbFName:%s, dbId:0x%" PRIx64, dbFName, dbId);
D
dapan1121 已提交
1222 1223 1224

  CTG_ERR_RET(ctgMetaRentAdd(&pCtg->dbRent, &vgVersion, dbId, sizeof(SDbVgVersion)));

dengyihao's avatar
dengyihao 已提交
1225
  ctgDebug("db added to rent, dbFName:%s, vgVersion:%d, dbId:0x%" PRIx64, dbFName, vgVersion.vgVersion, dbId);
D
dapan1121 已提交
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235

  return TSDB_CODE_SUCCESS;

_return:

  ctgFreeDbCache(&newDBCache);

  CTG_RET(code);
}

dengyihao's avatar
dengyihao 已提交
1236
void ctgRemoveStbRent(SCatalog *pCtg, SCtgDBCache *dbCache) {
D
dapan1121 已提交
1237 1238 1239
  if (NULL == dbCache->stbCache) {
    return;
  }
dengyihao's avatar
dengyihao 已提交
1240

D
dapan1121 已提交
1241 1242 1243 1244
  void *pIter = taosHashIterate(dbCache->stbCache, NULL);
  while (pIter) {
    uint64_t *suid = NULL;
    suid = taosHashGetKey(pIter, NULL);
D
dapan1121 已提交
1245

dengyihao's avatar
dengyihao 已提交
1246 1247 1248
    if (TSDB_CODE_SUCCESS ==
        ctgMetaRentRemove(&pCtg->stbRent, *suid, ctgStbVersionSortCompare, ctgStbVersionSearchCompare)) {
      ctgDebug("stb removed from rent, suid:0x%" PRIx64, *suid);
D
dapan1121 已提交
1249
    }
dengyihao's avatar
dengyihao 已提交
1250

D
dapan1121 已提交
1251
    pIter = taosHashIterate(dbCache->stbCache, pIter);
D
dapan1121 已提交
1252 1253 1254
  }
}

dengyihao's avatar
dengyihao 已提交
1255
int32_t ctgRemoveDBFromCache(SCatalog *pCtg, SCtgDBCache *dbCache, const char *dbFName) {
D
dapan1121 已提交
1256
  uint64_t dbId = dbCache->dbId;
dengyihao's avatar
dengyihao 已提交
1257 1258

  ctgInfo("start to remove db from cache, dbFName:%s, dbId:0x%" PRIx64, dbFName, dbCache->dbId);
D
dapan1121 已提交
1259

D
dapan1121 已提交
1260
  CTG_LOCK(CTG_WRITE, &dbCache->dbLock);
D
dapan1121 已提交
1261

D
dapan1121 已提交
1262
  atomic_store_8(&dbCache->deleted, 1);
D
dapan1121 已提交
1263
  ctgRemoveStbRent(pCtg, dbCache);
D
dapan1121 已提交
1264 1265
  ctgFreeDbCache(dbCache);

D
dapan1121 已提交
1266 1267 1268
  CTG_UNLOCK(CTG_WRITE, &dbCache->dbLock);

  CTG_ERR_RET(ctgMetaRentRemove(&pCtg->dbRent, dbId, ctgDbVgVersionSortCompare, ctgDbVgVersionSearchCompare));
dengyihao's avatar
dengyihao 已提交
1269
  ctgDebug("db removed from rent, dbFName:%s, dbId:0x%" PRIx64, dbFName, dbId);
D
dapan1121 已提交
1270 1271 1272 1273 1274 1275

  if (taosHashRemove(pCtg->dbCache, dbFName, strlen(dbFName))) {
    ctgInfo("taosHashRemove from dbCache failed, may be removed, dbFName:%s", dbFName);
    CTG_ERR_RET(TSDB_CODE_CTG_DB_DROPPED);
  }

D
dapan1121 已提交
1276
  CTG_CACHE_STAT_DEC(numOfDb, 1);
dengyihao's avatar
dengyihao 已提交
1277 1278
  ctgInfo("db removed from cache, dbFName:%s, dbId:0x%" PRIx64, dbFName, dbId);

D
dapan1121 已提交
1279 1280 1281
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
1282 1283
int32_t ctgGetAddDBCache(SCatalog *pCtg, const char *dbFName, uint64_t dbId, SCtgDBCache **pCache) {
  int32_t      code = 0;
D
dapan1121 已提交
1284 1285
  SCtgDBCache *dbCache = NULL;
  ctgGetDBCache(pCtg, dbFName, &dbCache);
dengyihao's avatar
dengyihao 已提交
1286

D
dapan1121 已提交
1287
  if (dbCache) {
dengyihao's avatar
dengyihao 已提交
1288
    // TODO OPEN IT
D
dapan1121 已提交
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
#if 0    
    if (dbCache->dbId == dbId) {
      *pCache = dbCache;
      return TSDB_CODE_SUCCESS;
    }
#else
    if (0 == dbId) {
      *pCache = dbCache;
      return TSDB_CODE_SUCCESS;
    }

    if (dbId && (dbCache->dbId == 0)) {
      dbCache->dbId = dbId;
      *pCache = dbCache;
      return TSDB_CODE_SUCCESS;
    }
dengyihao's avatar
dengyihao 已提交
1305

D
dapan1121 已提交
1306 1307 1308 1309 1310 1311 1312
    if (dbCache->dbId == dbId) {
      *pCache = dbCache;
      return TSDB_CODE_SUCCESS;
    }
#endif
    CTG_ERR_RET(ctgRemoveDBFromCache(pCtg, dbCache, dbFName));
  }
dengyihao's avatar
dengyihao 已提交
1313

D
dapan1121 已提交
1314 1315 1316 1317 1318 1319 1320 1321 1322
  CTG_ERR_RET(ctgAddNewDBCache(pCtg, dbFName, dbId));

  ctgGetDBCache(pCtg, dbFName, &dbCache);

  *pCache = dbCache;

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
1323 1324
int32_t ctgUpdateRentStbVersion(SCatalog *pCtg, char *dbFName, char *tbName, uint64_t dbId, uint64_t suid,
                                SCtgTbCache *pCache) {
D
dapan1121 已提交
1325 1326 1327 1328 1329 1330 1331 1332 1333
  SSTableVersion metaRent = {.dbId = dbId, .suid = suid};
  if (pCache->pMeta) {
    metaRent.sversion = pCache->pMeta->sversion;
    metaRent.tversion = pCache->pMeta->tversion;
  }

  if (pCache->pIndex) {
    metaRent.smaVer = pCache->pIndex->version;
  }
dengyihao's avatar
dengyihao 已提交
1334

D
dapan1121 已提交
1335 1336
  tstrncpy(metaRent.dbFName, dbFName, sizeof(metaRent.dbFName));
  tstrncpy(metaRent.stbName, tbName, sizeof(metaRent.stbName));
D
dapan1121 已提交
1337

dengyihao's avatar
dengyihao 已提交
1338 1339
  CTG_ERR_RET(ctgMetaRentUpdate(&pCtg->stbRent, &metaRent, metaRent.suid, sizeof(SSTableVersion),
                                ctgStbVersionSortCompare, ctgStbVersionSearchCompare));
D
dapan1121 已提交
1340

dengyihao's avatar
dengyihao 已提交
1341 1342
  ctgDebug("db %s,0x%" PRIx64 " stb %s,0x%" PRIx64 " sver %d tver %d smaVer %d updated to stbRent", dbFName, dbId,
           tbName, suid, metaRent.sversion, metaRent.tversion, metaRent.smaVer);
D
dapan1121 已提交
1343

dengyihao's avatar
dengyihao 已提交
1344 1345
  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
1346

dengyihao's avatar
dengyihao 已提交
1347 1348
int32_t ctgWriteTbMetaToCache(SCatalog *pCtg, SCtgDBCache *dbCache, char *dbFName, uint64_t dbId, char *tbName,
                              STableMeta *meta, int32_t metaSize) {
D
dapan1121 已提交
1349 1350
  if (NULL == dbCache->tbCache || NULL == dbCache->stbCache) {
    taosMemoryFree(meta);
dengyihao's avatar
dengyihao 已提交
1351
    ctgError("db is dropping, dbId:0x%" PRIx64, dbCache->dbId);
D
dapan1121 已提交
1352 1353 1354
    CTG_ERR_RET(TSDB_CODE_CTG_DB_DROPPED);
  }

dengyihao's avatar
dengyihao 已提交
1355 1356 1357 1358 1359 1360
  bool         isStb = meta->tableType == TSDB_SUPER_TABLE;
  SCtgTbCache *pCache = taosHashGet(dbCache->tbCache, tbName, strlen(tbName));
  STableMeta  *orig = (pCache ? pCache->pMeta : NULL);
  int8_t       origType = 0;
  uint64_t     origSuid = 0;

D
dapan1121 已提交
1361 1362 1363
  if (orig) {
    origType = orig->tableType;

dengyihao's avatar
dengyihao 已提交
1364 1365
    if (origType == meta->tableType && orig->uid == meta->uid &&
        (origType == TSDB_CHILD_TABLE || (orig->sversion >= meta->sversion && orig->tversion >= meta->tversion))) {
D
dapan1121 已提交
1366 1367
      taosMemoryFree(meta);
      ctgDebug("ignore table %s meta update", tbName);
D
dapan1121 已提交
1368 1369
      return TSDB_CODE_SUCCESS;
    }
dengyihao's avatar
dengyihao 已提交
1370

D
dapan1121 已提交
1371
    if (origType == TSDB_SUPER_TABLE) {
D
dapan1121 已提交
1372
      if (taosHashRemove(dbCache->stbCache, &orig->suid, sizeof(orig->suid))) {
dengyihao's avatar
dengyihao 已提交
1373
        ctgError("stb not exist in stbCache, dbFName:%s, stb:%s, suid:0x%" PRIx64, dbFName, tbName, orig->suid);
D
dapan1121 已提交
1374
      } else {
D
dapan1121 已提交
1375
        CTG_CACHE_STAT_DEC(numOfStb, 1);
dengyihao's avatar
dengyihao 已提交
1376
        ctgDebug("stb removed from stbCache, dbFName:%s, stb:%s, suid:0x%" PRIx64, dbFName, tbName, orig->suid);
D
dapan1121 已提交
1377
      }
dengyihao's avatar
dengyihao 已提交
1378

D
dapan1121 已提交
1379 1380 1381 1382
      origSuid = orig->suid;
    }
  }

D
dapan1121 已提交
1383 1384 1385 1386 1387 1388
  if (NULL == pCache) {
    SCtgTbCache cache = {0};
    cache.pMeta = meta;
    if (taosHashPut(dbCache->tbCache, tbName, strlen(tbName), &cache, sizeof(SCtgTbCache)) != 0) {
      taosMemoryFree(meta);
      ctgError("taosHashPut new tbCache failed, dbFName:%s, tbName:%s, tbType:%d", dbFName, tbName, meta->tableType);
D
dapan1121 已提交
1389
      CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1390
    }
dengyihao's avatar
dengyihao 已提交
1391

D
dapan1121 已提交
1392 1393
    pCache = taosHashGet(dbCache->tbCache, tbName, strlen(tbName));
  } else {
1394
    CTG_LOCK(CTG_WRITE, &pCache->metaLock);
D
dapan1121 已提交
1395 1396
    taosMemoryFree(pCache->pMeta);
    pCache->pMeta = meta;
1397
    CTG_UNLOCK(CTG_WRITE, &pCache->metaLock);
D
dapan1121 已提交
1398 1399 1400
  }

  if (NULL == orig) {
D
dapan1121 已提交
1401
    CTG_CACHE_STAT_INC(numOfTbl, 1);
D
dapan1121 已提交
1402 1403 1404 1405 1406 1407 1408 1409 1410
  }

  ctgDebug("tbmeta updated to cache, dbFName:%s, tbName:%s, tbType:%d", dbFName, tbName, meta->tableType);
  ctgdShowTableMeta(pCtg, tbName, meta);

  if (!isStb) {
    return TSDB_CODE_SUCCESS;
  }

dengyihao's avatar
dengyihao 已提交
1411 1412 1413
  if (origSuid != meta->suid &&
      taosHashPut(dbCache->stbCache, &meta->suid, sizeof(meta->suid), tbName, strlen(tbName) + 1) != 0) {
    ctgError("taosHashPut to stable cache failed, suid:0x%" PRIx64, meta->suid);
D
dapan1121 已提交
1414
    CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1415 1416
  }

D
dapan1121 已提交
1417
  CTG_CACHE_STAT_INC(numOfStb, 1);
D
dapan1121 已提交
1418

dengyihao's avatar
dengyihao 已提交
1419 1420
  ctgDebug("stb 0x%" PRIx64 " updated to cache, dbFName:%s, tbName:%s, tbType:%d", meta->suid, dbFName, tbName,
           meta->tableType);
D
dapan1121 已提交
1421

D
dapan1121 已提交
1422 1423 1424 1425
  if (pCache) {
    CTG_ERR_RET(ctgUpdateRentStbVersion(pCtg, dbFName, tbName, dbId, meta->suid, pCache));
  }
  
D
dapan1121 已提交
1426 1427 1428
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
1429
int32_t ctgWriteTbIndexToCache(SCatalog *pCtg, SCtgDBCache *dbCache, char *dbFName, char *tbName, STableIndex **index) {
D
dapan1121 已提交
1430
  if (NULL == dbCache->tbCache) {
D
dapan1121 已提交
1431
    ctgFreeSTableIndex(*index);
D
dapan1121 已提交
1432
    taosMemoryFreeClear(*index);
dengyihao's avatar
dengyihao 已提交
1433
    ctgError("db is dropping, dbId:0x%" PRIx64, dbCache->dbId);
D
dapan1121 已提交
1434 1435 1436
    CTG_ERR_RET(TSDB_CODE_CTG_DB_DROPPED);
  }

dengyihao's avatar
dengyihao 已提交
1437 1438 1439
  STableIndex *pIndex = *index;
  uint64_t     suid = pIndex->suid;
  SCtgTbCache *pCache = taosHashGet(dbCache->tbCache, tbName, strlen(tbName));
D
dapan1121 已提交
1440 1441 1442
  if (NULL == pCache) {
    SCtgTbCache cache = {0};
    cache.pIndex = pIndex;
dengyihao's avatar
dengyihao 已提交
1443

D
dapan1121 已提交
1444 1445
    if (taosHashPut(dbCache->tbCache, tbName, strlen(tbName), &cache, sizeof(cache)) != 0) {
      ctgFreeSTableIndex(*index);
D
dapan1121 已提交
1446 1447
      taosMemoryFreeClear(*index);
      ctgError("taosHashPut new tbCache failed, tbName:%s", tbName);
D
dapan1121 已提交
1448
      CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1449 1450 1451
    }

    *index = NULL;
dengyihao's avatar
dengyihao 已提交
1452 1453
    ctgDebug("table %s index updated to cache, ver:%d, num:%d", tbName, pIndex->version,
             (int32_t)taosArrayGetSize(pIndex->pIndex));
D
dapan1121 已提交
1454

D
dapan1121 已提交
1455
    if (suid) {
D
dapan1121 已提交
1456
      CTG_ERR_RET(ctgUpdateRentStbVersion(pCtg, dbFName, tbName, dbCache->dbId, pIndex->suid, &cache));
D
dapan1121 已提交
1457
    }
dengyihao's avatar
dengyihao 已提交
1458

D
dapan1121 已提交
1459 1460 1461
    return TSDB_CODE_SUCCESS;
  }

1462 1463
  CTG_LOCK(CTG_WRITE, &pCache->indexLock);

D
dapan1121 已提交
1464
  if (pCache->pIndex) {
D
dapan1121 已提交
1465 1466 1467
    if (0 == suid) {
      suid = pCache->pIndex->suid;
    }
D
dapan1121 已提交
1468 1469 1470 1471 1472
    taosArrayDestroyEx(pCache->pIndex->pIndex, tFreeSTableIndexInfo);
    taosMemoryFreeClear(pCache->pIndex);
  }

  pCache->pIndex = pIndex;
1473 1474
  CTG_UNLOCK(CTG_WRITE, &pCache->indexLock);

D
dapan1121 已提交
1475 1476
  *index = NULL;

dengyihao's avatar
dengyihao 已提交
1477 1478
  ctgDebug("table %s index updated to cache, ver:%d, num:%d", tbName, pIndex->version,
           (int32_t)taosArrayGetSize(pIndex->pIndex));
D
dapan1121 已提交
1479

D
dapan1121 已提交
1480 1481 1482
  if (suid) {
    CTG_ERR_RET(ctgUpdateRentStbVersion(pCtg, dbFName, tbName, dbCache->dbId, suid, pCache));
  }
dengyihao's avatar
dengyihao 已提交
1483

D
dapan1121 已提交
1484 1485 1486
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
1487 1488 1489 1490
int32_t ctgUpdateTbMetaToCache(SCatalog *pCtg, STableMetaOutput *pOut, bool syncReq) {
  STableMetaOutput *pOutput = NULL;
  int32_t           code = 0;

D
dapan1121 已提交
1491
  CTG_ERR_RET(ctgCloneMetaOutput(pOut, &pOutput));
D
dapan1121 已提交
1492 1493 1494
  code = ctgUpdateTbMetaEnqueue(pCtg, pOutput, syncReq);
  pOutput = NULL;
  CTG_ERR_JRET(code);
D
dapan1121 已提交
1495 1496

  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
1497

D
dapan1121 已提交
1498 1499 1500 1501 1502 1503
_return:

  ctgFreeSTableMetaOutput(pOutput);
  CTG_RET(code);
}

D
dapan1121 已提交
1504
void ctgClearAllInstance(void) {
dengyihao's avatar
dengyihao 已提交
1505
  SCatalog *pCtg = NULL;
1506

dengyihao's avatar
dengyihao 已提交
1507
  void *pIter = taosHashIterate(gCtgMgmt.pCluster, NULL);
1508
  while (pIter) {
dengyihao's avatar
dengyihao 已提交
1509
    pCtg = *(SCatalog **)pIter;
1510 1511

    if (pCtg) {
D
dapan1121 已提交
1512 1513 1514 1515 1516 1517 1518 1519
      ctgClearHandle(pCtg);
    }

    pIter = taosHashIterate(gCtgMgmt.pCluster, pIter);
  }
}

void ctgFreeAllInstance(void) {
dengyihao's avatar
dengyihao 已提交
1520
  SCatalog *pCtg = NULL;
D
dapan1121 已提交
1521

dengyihao's avatar
dengyihao 已提交
1522
  void *pIter = taosHashIterate(gCtgMgmt.pCluster, NULL);
D
dapan1121 已提交
1523
  while (pIter) {
dengyihao's avatar
dengyihao 已提交
1524
    pCtg = *(SCatalog **)pIter;
D
dapan1121 已提交
1525 1526 1527

    if (pCtg) {
      ctgFreeHandle(pCtg);
1528 1529 1530 1531 1532 1533 1534 1535
    }

    pIter = taosHashIterate(gCtgMgmt.pCluster, pIter);
  }

  taosHashClear(gCtgMgmt.pCluster);
}

D
dapan1121 已提交
1536
int32_t ctgOpUpdateVgroup(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1537
  int32_t          code = 0;
D
dapan1121 已提交
1538
  SCtgUpdateVgMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1539 1540 1541 1542
  SDBVgInfo       *dbInfo = msg->dbInfo;
  char            *dbFName = msg->dbFName;
  SCatalog        *pCtg = msg->pCtg;

D
dapan1121 已提交
1543
  if (NULL == dbInfo->vgHash) {
D
dapan1121 已提交
1544
    goto _return;
D
dapan1121 已提交
1545
  }
dengyihao's avatar
dengyihao 已提交
1546

D
dapan1121 已提交
1547
  if (dbInfo->vgVersion < 0 || taosHashGetSize(dbInfo->vgHash) <= 0) {
dengyihao's avatar
dengyihao 已提交
1548 1549
    ctgError("invalid db vgInfo, dbFName:%s, vgHash:%p, vgVersion:%d, vgHashSize:%d", dbFName, dbInfo->vgHash,
             dbInfo->vgVersion, taosHashGetSize(dbInfo->vgHash));
D
dapan1121 已提交
1550
    CTG_ERR_JRET(TSDB_CODE_APP_ERROR);
D
dapan1121 已提交
1551 1552
  }

dengyihao's avatar
dengyihao 已提交
1553
  bool         newAdded = false;
D
dapan1121 已提交
1554 1555 1556
  SDbVgVersion vgVersion = {.dbId = msg->dbId, .vgVersion = dbInfo->vgVersion, .numOfTable = dbInfo->numOfTable};

  SCtgDBCache *dbCache = NULL;
D
dapan1121 已提交
1557
  CTG_ERR_JRET(ctgGetAddDBCache(msg->pCtg, dbFName, msg->dbId, &dbCache));
D
dapan1121 已提交
1558
  if (NULL == dbCache) {
dengyihao's avatar
dengyihao 已提交
1559
    ctgInfo("conflict db update, ignore this update, dbFName:%s, dbId:0x%" PRIx64, dbFName, msg->dbId);
D
dapan1121 已提交
1560
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
D
dapan1121 已提交
1561 1562 1563
  }

  SCtgVgCache *vgCache = &dbCache->vgCache;
D
dapan1121 已提交
1564
  CTG_ERR_JRET(ctgWLockVgInfo(msg->pCtg, dbCache));
dengyihao's avatar
dengyihao 已提交
1565

D
dapan1121 已提交
1566 1567
  if (vgCache->vgInfo) {
    SDBVgInfo *vgInfo = vgCache->vgInfo;
dengyihao's avatar
dengyihao 已提交
1568

D
dapan1121 已提交
1569 1570 1571
    if (dbInfo->vgVersion < vgInfo->vgVersion) {
      ctgDebug("db vgVer is old, dbFName:%s, vgVer:%d, curVer:%d", dbFName, dbInfo->vgVersion, vgInfo->vgVersion);
      ctgWUnlockVgInfo(dbCache);
dengyihao's avatar
dengyihao 已提交
1572

D
dapan1121 已提交
1573
      goto _return;
D
dapan1121 已提交
1574 1575 1576
    }

    if (dbInfo->vgVersion == vgInfo->vgVersion && dbInfo->numOfTable == vgInfo->numOfTable) {
dengyihao's avatar
dengyihao 已提交
1577 1578
      ctgDebug("no new db vgVer or numOfTable, dbFName:%s, vgVer:%d, numOfTable:%d", dbFName, dbInfo->vgVersion,
               dbInfo->numOfTable);
D
dapan1121 已提交
1579
      ctgWUnlockVgInfo(dbCache);
dengyihao's avatar
dengyihao 已提交
1580

D
dapan1121 已提交
1581
      goto _return;
D
dapan1121 已提交
1582 1583 1584 1585 1586 1587 1588 1589
    }

    ctgFreeVgInfo(vgInfo);
  }

  vgCache->vgInfo = dbInfo;
  msg->dbInfo = NULL;

dengyihao's avatar
dengyihao 已提交
1590
  ctgDebug("db vgInfo updated, dbFName:%s, vgVer:%d, dbId:0x%" PRIx64, dbFName, vgVersion.vgVersion, vgVersion.dbId);
D
dapan1121 已提交
1591 1592 1593 1594 1595

  ctgWUnlockVgInfo(dbCache);

  dbCache = NULL;

D
dapan1121 已提交
1596
  tstrncpy(vgVersion.dbFName, dbFName, sizeof(vgVersion.dbFName));
dengyihao's avatar
dengyihao 已提交
1597 1598
  CTG_ERR_JRET(ctgMetaRentUpdate(&msg->pCtg->dbRent, &vgVersion, vgVersion.dbId, sizeof(SDbVgVersion),
                                 ctgDbVgVersionSortCompare, ctgDbVgVersionSearchCompare));
D
dapan1121 已提交
1599 1600 1601 1602 1603

_return:

  ctgFreeVgInfo(msg->dbInfo);
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1604

D
dapan1121 已提交
1605 1606 1607
  CTG_RET(code);
}

D
dapan1121 已提交
1608
int32_t ctgOpDropDbCache(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1609
  int32_t        code = 0;
D
dapan1121 已提交
1610
  SCtgDropDBMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1611
  SCatalog      *pCtg = msg->pCtg;
D
dapan1121 已提交
1612 1613 1614 1615 1616 1617

  SCtgDBCache *dbCache = NULL;
  ctgGetDBCache(msg->pCtg, msg->dbFName, &dbCache);
  if (NULL == dbCache) {
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
1618

D
dapan1121 已提交
1619
  if (dbCache->dbId != msg->dbId) {
dengyihao's avatar
dengyihao 已提交
1620 1621
    ctgInfo("dbId already updated, dbFName:%s, dbId:0x%" PRIx64 ", targetId:0x%" PRIx64, msg->dbFName, dbCache->dbId,
            msg->dbId);
D
dapan1121 已提交
1622 1623
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
1624

D
dapan1121 已提交
1625 1626 1627 1628 1629
  CTG_ERR_JRET(ctgRemoveDBFromCache(pCtg, dbCache, msg->dbFName));

_return:

  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1630

D
dapan1121 已提交
1631 1632 1633
  CTG_RET(code);
}

D
dapan1121 已提交
1634
int32_t ctgOpDropDbVgroup(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1635
  int32_t              code = 0;
D
dapan1121 已提交
1636
  SCtgDropDbVgroupMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1637
  SCatalog            *pCtg = msg->pCtg;
D
dapan1121 已提交
1638 1639 1640 1641 1642 1643

  SCtgDBCache *dbCache = NULL;
  ctgGetDBCache(msg->pCtg, msg->dbFName, &dbCache);
  if (NULL == dbCache) {
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
1644

dengyihao's avatar
dengyihao 已提交
1645
  CTG_ERR_JRET(ctgWLockVgInfo(pCtg, dbCache));
dengyihao's avatar
dengyihao 已提交
1646

D
dapan1121 已提交
1647 1648
  ctgFreeVgInfo(dbCache->vgCache.vgInfo);
  dbCache->vgCache.vgInfo = NULL;
D
dapan1121 已提交
1649 1650 1651

  ctgDebug("db vgInfo removed, dbFName:%s", msg->dbFName);

D
dapan1121 已提交
1652
  ctgWUnlockVgInfo(dbCache);
D
dapan1121 已提交
1653 1654 1655 1656

_return:

  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1657

D
dapan1121 已提交
1658 1659 1660
  CTG_RET(code);
}

D
dapan1121 已提交
1661
int32_t ctgOpUpdateTbMeta(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1662
  int32_t              code = 0;
D
dapan1121 已提交
1663
  SCtgUpdateTbMetaMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1664 1665 1666
  SCatalog            *pCtg = msg->pCtg;
  STableMetaOutput    *pMeta = msg->pMeta;
  SCtgDBCache         *dbCache = NULL;
D
dapan1121 已提交
1667

D
dapan1121 已提交
1668 1669
  if ((!CTG_IS_META_CTABLE(pMeta->metaType)) && NULL == pMeta->tbMeta) {
    ctgError("no valid tbmeta got from meta rsp, dbFName:%s, tbName:%s", pMeta->dbFName, pMeta->tbName);
D
dapan1121 已提交
1670 1671 1672
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

D
dapan1121 已提交
1673 1674
  if (CTG_IS_META_BOTH(pMeta->metaType) && TSDB_SUPER_TABLE != pMeta->tbMeta->tableType) {
    ctgError("table type error, expected:%d, actual:%d", TSDB_SUPER_TABLE, pMeta->tbMeta->tableType);
D
dapan1121 已提交
1675
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
dengyihao's avatar
dengyihao 已提交
1676 1677
  }

D
dapan1121 已提交
1678
  CTG_ERR_JRET(ctgGetAddDBCache(pCtg, pMeta->dbFName, pMeta->dbId, &dbCache));
D
dapan1121 已提交
1679
  if (NULL == dbCache) {
D
dapan1121 已提交
1680
    ctgInfo("conflict db update, ignore this update, dbFName:%s, dbId:0x%" PRIx64, pMeta->dbFName, pMeta->dbId);
D
dapan1121 已提交
1681 1682 1683
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
  }

D
dapan1121 已提交
1684 1685
  if (CTG_IS_META_TABLE(pMeta->metaType) || CTG_IS_META_BOTH(pMeta->metaType)) {
    int32_t metaSize = CTG_META_SIZE(pMeta->tbMeta);
D
dapan1121 已提交
1686
    code = ctgWriteTbMetaToCache(pCtg, dbCache, pMeta->dbFName, pMeta->dbId, pMeta->tbName, pMeta->tbMeta, metaSize);
D
dapan1121 已提交
1687
    pMeta->tbMeta = NULL;
D
dapan1121 已提交
1688
    CTG_ERR_JRET(code);
D
dapan1121 已提交
1689 1690
  }

D
dapan1121 已提交
1691
  if (CTG_IS_META_CTABLE(pMeta->metaType) || CTG_IS_META_BOTH(pMeta->metaType)) {
dengyihao's avatar
dengyihao 已提交
1692
    SCTableMeta *ctbMeta = taosMemoryMalloc(sizeof(SCTableMeta));
D
dapan1121 已提交
1693 1694 1695 1696
    if (NULL == ctbMeta) {
      CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
    }
    memcpy(ctbMeta, &pMeta->ctbMeta, sizeof(SCTableMeta));
dengyihao's avatar
dengyihao 已提交
1697 1698
    CTG_ERR_JRET(ctgWriteTbMetaToCache(pCtg, dbCache, pMeta->dbFName, pMeta->dbId, pMeta->ctbName,
                                       (STableMeta *)ctbMeta, sizeof(SCTableMeta)));
D
dapan1121 已提交
1699 1700 1701 1702
  }

_return:

D
dapan1121 已提交
1703 1704
  taosMemoryFreeClear(pMeta->tbMeta);
  taosMemoryFreeClear(pMeta);
dengyihao's avatar
dengyihao 已提交
1705

D
dapan1121 已提交
1706
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1707

D
dapan1121 已提交
1708 1709 1710
  CTG_RET(code);
}

D
dapan1121 已提交
1711
int32_t ctgOpDropStbMeta(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1712
  int32_t             code = 0;
D
dapan1121 已提交
1713
  SCtgDropStbMetaMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1714
  SCatalog           *pCtg = msg->pCtg;
D
dapan1121 已提交
1715 1716 1717 1718

  SCtgDBCache *dbCache = NULL;
  ctgGetDBCache(pCtg, msg->dbFName, &dbCache);
  if (NULL == dbCache) {
dengyihao's avatar
dengyihao 已提交
1719
    goto _return;
D
dapan1121 已提交
1720 1721 1722
  }

  if (msg->dbId && (dbCache->dbId != msg->dbId)) {
dengyihao's avatar
dengyihao 已提交
1723
    ctgDebug("dbId already modified, dbFName:%s, current:0x%" PRIx64 ", dbId:0x%" PRIx64 ", stb:%s, suid:0x%" PRIx64,
D
dapan1121 已提交
1724
             msg->dbFName, dbCache->dbId, msg->dbId, msg->stbName, msg->suid);
dengyihao's avatar
dengyihao 已提交
1725
    goto _return;
D
dapan1121 已提交
1726
  }
dengyihao's avatar
dengyihao 已提交
1727

D
dapan1121 已提交
1728
  if (taosHashRemove(dbCache->stbCache, &msg->suid, sizeof(msg->suid))) {
dengyihao's avatar
dengyihao 已提交
1729 1730
    ctgDebug("stb not exist in stbCache, may be removed, dbFName:%s, stb:%s, suid:0x%" PRIx64, msg->dbFName,
             msg->stbName, msg->suid);
D
dapan1121 已提交
1731
  } else {
D
dapan1121 已提交
1732
    CTG_CACHE_STAT_DEC(numOfStb, 1);
D
dapan1121 已提交
1733 1734
  }

dengyihao's avatar
dengyihao 已提交
1735
  SCtgTbCache *pTbCache = taosHashGet(dbCache->tbCache, msg->stbName, strlen(msg->stbName));
D
dapan1121 已提交
1736 1737 1738 1739 1740 1741 1742 1743 1744
  if (NULL == pTbCache) {
    ctgDebug("stb %s already not in cache", msg->stbName);
    goto _return;
  }

  CTG_LOCK(CTG_WRITE, &pTbCache->metaLock);
  ctgFreeTbCacheImpl(pTbCache);
  CTG_UNLOCK(CTG_WRITE, &pTbCache->metaLock);

dengyihao's avatar
dengyihao 已提交
1745 1746
  if (taosHashRemove(dbCache->tbCache, msg->stbName, strlen(msg->stbName))) {
    ctgError("stb not exist in cache, dbFName:%s, stb:%s, suid:0x%" PRIx64, msg->dbFName, msg->stbName, msg->suid);
D
dapan1121 已提交
1747
  } else {
D
dapan1121 已提交
1748
    CTG_CACHE_STAT_DEC(numOfTbl, 1);
D
dapan1121 已提交
1749
  }
dengyihao's avatar
dengyihao 已提交
1750 1751

  ctgInfo("stb removed from cache, dbFName:%s, stbName:%s, suid:0x%" PRIx64, msg->dbFName, msg->stbName, msg->suid);
D
dapan1121 已提交
1752 1753

  CTG_ERR_JRET(ctgMetaRentRemove(&msg->pCtg->stbRent, msg->suid, ctgStbVersionSortCompare, ctgStbVersionSearchCompare));
dengyihao's avatar
dengyihao 已提交
1754 1755 1756

  ctgDebug("stb removed from rent, dbFName:%s, stbName:%s, suid:0x%" PRIx64, msg->dbFName, msg->stbName, msg->suid);

D
dapan1121 已提交
1757 1758 1759
_return:

  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1760

D
dapan1121 已提交
1761 1762 1763
  CTG_RET(code);
}

D
dapan1121 已提交
1764
int32_t ctgOpDropTbMeta(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1765
  int32_t             code = 0;
D
dapan1121 已提交
1766
  SCtgDropTblMetaMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1767
  SCatalog           *pCtg = msg->pCtg;
D
dapan1121 已提交
1768 1769 1770 1771

  SCtgDBCache *dbCache = NULL;
  ctgGetDBCache(pCtg, msg->dbFName, &dbCache);
  if (NULL == dbCache) {
D
dapan1121 已提交
1772
    goto _return;
D
dapan1121 已提交
1773 1774 1775
  }

  if (dbCache->dbId != msg->dbId) {
dengyihao's avatar
dengyihao 已提交
1776 1777
    ctgDebug("dbId 0x%" PRIx64 " not match with curId 0x%" PRIx64 ", dbFName:%s, tbName:%s", msg->dbId, dbCache->dbId,
             msg->dbFName, msg->tbName);
D
dapan1121 已提交
1778
    goto _return;
D
dapan1121 已提交
1779 1780
  }

dengyihao's avatar
dengyihao 已提交
1781
  SCtgTbCache *pTbCache = taosHashGet(dbCache->tbCache, msg->tbName, strlen(msg->tbName));
D
dapan1121 已提交
1782 1783 1784 1785 1786 1787 1788 1789
  if (NULL == pTbCache) {
    ctgDebug("tb %s already not in cache", msg->tbName);
    goto _return;
  }

  CTG_LOCK(CTG_WRITE, &pTbCache->metaLock);
  ctgFreeTbCacheImpl(pTbCache);
  CTG_UNLOCK(CTG_WRITE, &pTbCache->metaLock);
dengyihao's avatar
dengyihao 已提交
1790

D
dapan1121 已提交
1791 1792 1793
  if (taosHashRemove(dbCache->tbCache, msg->tbName, strlen(msg->tbName))) {
    ctgError("tb %s not exist in cache, dbFName:%s", msg->tbName, msg->dbFName);
    CTG_ERR_JRET(TSDB_CODE_CTG_INTERNAL_ERROR);
D
dapan1121 已提交
1794
  } else {
D
dapan1121 已提交
1795
    CTG_CACHE_STAT_DEC(numOfTbl, 1);
D
dapan1121 已提交
1796 1797
  }

D
dapan1121 已提交
1798
  ctgDebug("table %s removed from cache, dbFName:%s", msg->tbName, msg->dbFName);
D
dapan1121 已提交
1799 1800 1801 1802 1803 1804 1805 1806

_return:

  taosMemoryFreeClear(msg);

  CTG_RET(code);
}

D
dapan1121 已提交
1807
int32_t ctgOpUpdateUser(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1808
  int32_t            code = 0;
D
dapan1121 已提交
1809
  SCtgUpdateUserMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1810 1811
  SCatalog          *pCtg = msg->pCtg;

D
dapan1121 已提交
1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
  SCtgUserAuth *pUser = (SCtgUserAuth *)taosHashGet(pCtg->userCache, msg->userAuth.user, strlen(msg->userAuth.user));
  if (NULL == pUser) {
    SCtgUserAuth userAuth = {0};

    userAuth.version = msg->userAuth.version;
    userAuth.superUser = msg->userAuth.superAuth;
    userAuth.createdDbs = msg->userAuth.createdDbs;
    userAuth.readDbs = msg->userAuth.readDbs;
    userAuth.writeDbs = msg->userAuth.writeDbs;

    if (taosHashPut(pCtg->userCache, msg->userAuth.user, strlen(msg->userAuth.user), &userAuth, sizeof(userAuth))) {
      ctgError("taosHashPut user %s to cache failed", msg->userAuth.user);
      CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
    }

    taosMemoryFreeClear(msg);

    return TSDB_CODE_SUCCESS;
  }

  pUser->version = msg->userAuth.version;

  CTG_LOCK(CTG_WRITE, &pUser->lock);

  taosHashCleanup(pUser->createdDbs);
  pUser->createdDbs = msg->userAuth.createdDbs;
  msg->userAuth.createdDbs = NULL;

  taosHashCleanup(pUser->readDbs);
  pUser->readDbs = msg->userAuth.readDbs;
  msg->userAuth.readDbs = NULL;

  taosHashCleanup(pUser->writeDbs);
  pUser->writeDbs = msg->userAuth.writeDbs;
  msg->userAuth.writeDbs = NULL;

  CTG_UNLOCK(CTG_WRITE, &pUser->lock);

_return:

  taosHashCleanup(msg->userAuth.createdDbs);
  taosHashCleanup(msg->userAuth.readDbs);
  taosHashCleanup(msg->userAuth.writeDbs);
dengyihao's avatar
dengyihao 已提交
1855

D
dapan1121 已提交
1856
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1857

D
dapan1121 已提交
1858 1859 1860
  CTG_RET(code);
}

D
dapan1121 已提交
1861
int32_t ctgOpUpdateEpset(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1862
  int32_t             code = 0;
D
dapan1121 已提交
1863
  SCtgUpdateEpsetMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1864 1865
  SCatalog           *pCtg = msg->pCtg;

D
dapan1121 已提交
1866
  SCtgDBCache *dbCache = NULL;
D
dapan1121 已提交
1867
  CTG_ERR_JRET(ctgGetDBCache(pCtg, msg->dbFName, &dbCache));
D
dapan1121 已提交
1868 1869 1870 1871 1872
  if (NULL == dbCache) {
    ctgDebug("db %s not exist, ignore epset update", msg->dbFName);
    goto _return;
  }

D
dapan1121 已提交
1873 1874
  CTG_ERR_JRET(ctgWLockVgInfo(pCtg, dbCache));

dengyihao's avatar
dengyihao 已提交
1875
  SDBVgInfo *vgInfo = dbCache->vgCache.vgInfo;
D
dapan1121 已提交
1876
  if (NULL == vgInfo) {
D
dapan1121 已提交
1877 1878 1879
    ctgDebug("vgroup in db %s not cached, ignore epset update", msg->dbFName);
    goto _return;
  }
dengyihao's avatar
dengyihao 已提交
1880 1881

  SVgroupInfo *pInfo = taosHashGet(vgInfo->vgHash, &msg->vgId, sizeof(msg->vgId));
D
dapan1121 已提交
1882 1883 1884 1885 1886
  if (NULL == pInfo) {
    ctgDebug("no vgroup %d in db %s, ignore epset update", msg->vgId, msg->dbFName);
    goto _return;
  }

dengyihao's avatar
dengyihao 已提交
1887 1888 1889 1890 1891
  SEp *pOrigEp = &pInfo->epSet.eps[pInfo->epSet.inUse];
  SEp *pNewEp = &msg->epSet.eps[msg->epSet.inUse];
  ctgDebug("vgroup %d epset updated from %d/%d=>%s:%d to %d/%d=>%s:%d, dbFName:%s in ctg", pInfo->vgId,
           pInfo->epSet.inUse, pInfo->epSet.numOfEps, pOrigEp->fqdn, pOrigEp->port, msg->epSet.inUse,
           msg->epSet.numOfEps, pNewEp->fqdn, pNewEp->port, msg->dbFName);
D
dapan1121 已提交
1892

D
dapan1121 已提交
1893
  pInfo->epSet = msg->epSet;
D
dapan1121 已提交
1894 1895 1896 1897

_return:

  if (dbCache) {
D
dapan1121 已提交
1898
    ctgWUnlockVgInfo(dbCache);
D
dapan1121 已提交
1899 1900 1901
  }

  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1902

D
dapan1121 已提交
1903 1904 1905
  CTG_RET(code);
}

D
dapan1121 已提交
1906
int32_t ctgOpUpdateTbIndex(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1907
  int32_t               code = 0;
D
dapan1121 已提交
1908
  SCtgUpdateTbIndexMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1909 1910 1911 1912
  SCatalog             *pCtg = msg->pCtg;
  STableIndex          *pIndex = msg->pIndex;
  SCtgDBCache          *dbCache = NULL;

D
dapan1121 已提交
1913
  CTG_ERR_JRET(ctgGetAddDBCache(pCtg, pIndex->dbFName, 0, &dbCache));
D
dapan1121 已提交
1914 1915 1916 1917 1918 1919 1920 1921 1922

  CTG_ERR_JRET(ctgWriteTbIndexToCache(pCtg, dbCache, pIndex->dbFName, pIndex->tbName, &pIndex));

_return:

  if (pIndex) {
    taosArrayDestroyEx(pIndex->pIndex, tFreeSTableIndexInfo);
    taosMemoryFreeClear(pIndex);
  }
dengyihao's avatar
dengyihao 已提交
1923

D
dapan1121 已提交
1924
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1925

D
dapan1121 已提交
1926 1927 1928 1929
  CTG_RET(code);
}

int32_t ctgOpDropTbIndex(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1930
  int32_t             code = 0;
D
dapan1121 已提交
1931
  SCtgDropTbIndexMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1932 1933 1934
  SCatalog           *pCtg = msg->pCtg;
  SCtgDBCache        *dbCache = NULL;

D
dapan1121 已提交
1935
  CTG_ERR_JRET(ctgGetDBCache(pCtg, msg->dbFName, &dbCache));
D
dapan1121 已提交
1936
  if (NULL == dbCache) {
D
dapan1121 已提交
1937 1938 1939
    return TSDB_CODE_SUCCESS;
  }

dengyihao's avatar
dengyihao 已提交
1940
  STableIndex *pIndex = taosMemoryCalloc(1, sizeof(STableIndex));
D
dapan1121 已提交
1941 1942
  if (NULL == pIndex) {
    CTG_ERR_JRET(TSDB_CODE_OUT_OF_MEMORY);
D
dapan1121 已提交
1943
  }
D
dapan1121 已提交
1944 1945 1946
  strcpy(pIndex->tbName, msg->tbName);
  strcpy(pIndex->dbFName, msg->dbFName);
  pIndex->version = -1;
D
dapan1121 已提交
1947

D
dapan1121 已提交
1948
  CTG_ERR_JRET(ctgWriteTbIndexToCache(pCtg, dbCache, pIndex->dbFName, pIndex->tbName, &pIndex));
D
dapan1121 已提交
1949 1950 1951 1952 1953 1954 1955

_return:

  if (pIndex) {
    taosArrayDestroyEx(pIndex->pIndex, tFreeSTableIndexInfo);
    taosMemoryFreeClear(pIndex);
  }
dengyihao's avatar
dengyihao 已提交
1956

D
dapan1121 已提交
1957
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1958

D
dapan1121 已提交
1959 1960 1961
  CTG_RET(code);
}

D
dapan1121 已提交
1962
int32_t ctgOpClearCache(SCtgCacheOperation *operation) {
dengyihao's avatar
dengyihao 已提交
1963
  int32_t            code = 0;
D
dapan1121 已提交
1964
  SCtgClearCacheMsg *msg = operation->data;
dengyihao's avatar
dengyihao 已提交
1965
  SCatalog          *pCtg = msg->pCtg;
D
dapan1121 已提交
1966

D
dapan1121 已提交
1967 1968
  CTG_LOCK(CTG_WRITE, &gCtgMgmt.lock);

D
dapan1121 已提交
1969
  if (pCtg) {
D
dapan1121 已提交
1970 1971 1972 1973 1974
    if (msg->freeCtg) {
      ctgFreeHandle(pCtg);
    } else {
      ctgClearHandle(pCtg);
    }
dengyihao's avatar
dengyihao 已提交
1975

D
dapan1121 已提交
1976 1977
    goto _return;
  }
D
dapan1121 已提交
1978 1979 1980 1981 1982 1983

  if (msg->freeCtg) {
    ctgFreeAllInstance();
  } else {
    ctgClearAllInstance();
  }
D
dapan1121 已提交
1984 1985

_return:
D
dapan1121 已提交
1986 1987

  CTG_UNLOCK(CTG_WRITE, &gCtgMgmt.lock);
dengyihao's avatar
dengyihao 已提交
1988

D
dapan1121 已提交
1989
  taosMemoryFreeClear(msg);
dengyihao's avatar
dengyihao 已提交
1990

D
dapan1121 已提交
1991 1992 1993
  CTG_RET(code);
}

D
dapan1121 已提交
1994
void ctgCleanupCacheQueue(void) {
dengyihao's avatar
dengyihao 已提交
1995 1996
  SCtgQNode          *node = NULL;
  SCtgQNode          *nodeNext = NULL;
D
dapan1121 已提交
1997
  SCtgCacheOperation *op = NULL;
dengyihao's avatar
dengyihao 已提交
1998
  bool                stopQueue = false;
D
dapan1121 已提交
1999 2000 2001 2002 2003

  while (true) {
    node = gCtgMgmt.queue.head->next;
    while (node) {
      if (node->op) {
D
dapan1121 已提交
2004 2005 2006 2007 2008 2009
        op = node->op;
        if (op->stopQueue) {
          SCatalog *pCtg = ((SCtgUpdateMsgHeader *)op->data)->pCtg;
          ctgDebug("process [%s] operation", gCtgCacheOperation[op->opId].name);
          (*gCtgCacheOperation[op->opId].func)(op);
          stopQueue = true;
dengyihao's avatar
dengyihao 已提交
2010
          CTG_RT_STAT_INC(numOfOpDequeue, 1);
D
dapan1121 已提交
2011 2012
        } else {
          taosMemoryFree(op->data);
dengyihao's avatar
dengyihao 已提交
2013
          CTG_RT_STAT_INC(numOfOpAbort, 1);
D
dapan1121 已提交
2014
        }
dengyihao's avatar
dengyihao 已提交
2015

D
dapan1121 已提交
2016 2017
        if (op->syncOp) {
          tsem_post(&op->rspSem);
D
dapan1121 已提交
2018
        } else {
D
dapan1121 已提交
2019
          taosMemoryFree(op);
D
dapan1121 已提交
2020
        }
D
dapan1121 已提交
2021
      }
D
dapan1121 已提交
2022 2023 2024

      nodeNext = node->next;
      taosMemoryFree(node);
dengyihao's avatar
dengyihao 已提交
2025

D
dapan1121 已提交
2026
      node = nodeNext;
D
dapan1121 已提交
2027 2028
    }

D
dapan1121 已提交
2029
    if (!stopQueue) {
D
dapan1121 已提交
2030 2031 2032 2033
      taosUsleep(1);
    } else {
      break;
    }
D
dapan1121 已提交
2034 2035 2036 2037 2038 2039
  }

  taosMemoryFreeClear(gCtgMgmt.queue.head);
  gCtgMgmt.queue.tail = NULL;
}

dengyihao's avatar
dengyihao 已提交
2040
void *ctgUpdateThreadFunc(void *param) {
D
dapan1121 已提交
2041
  setThreadName("catalog");
2042

D
dapan1121 已提交
2043 2044 2045 2046 2047 2048
  qInfo("catalog update thread started");

  while (true) {
    if (tsem_wait(&gCtgMgmt.queue.reqSem)) {
      qError("ctg tsem_wait failed, error:%s", tstrerror(TAOS_SYSTEM_ERROR(errno)));
    }
dengyihao's avatar
dengyihao 已提交
2049 2050

    if (atomic_load_8((int8_t *)&gCtgMgmt.exit)) {
D
dapan1121 已提交
2051
      ctgCleanupCacheQueue();
D
dapan1121 已提交
2052 2053 2054
      break;
    }

D
dapan1121 已提交
2055 2056 2057
    SCtgCacheOperation *operation = NULL;
    ctgDequeue(&operation);
    SCatalog *pCtg = ((SCtgUpdateMsgHeader *)operation->data)->pCtg;
D
dapan1121 已提交
2058

D
dapan1121 已提交
2059
    ctgDebug("process [%s] operation", gCtgCacheOperation[operation->opId].name);
dengyihao's avatar
dengyihao 已提交
2060

D
dapan1121 已提交
2061
    (*gCtgCacheOperation[operation->opId].func)(operation);
D
dapan1121 已提交
2062

D
dapan1121 已提交
2063
    if (operation->syncOp) {
D
dapan1121 已提交
2064
      tsem_post(&operation->rspSem);
D
dapan1121 已提交
2065 2066
    } else {
      taosMemoryFreeClear(operation);
D
dapan1121 已提交
2067 2068
    }

dengyihao's avatar
dengyihao 已提交
2069
    CTG_RT_STAT_INC(numOfOpDequeue, 1);
D
dapan1121 已提交
2070

D
dapan1121 已提交
2071
    ctgdShowCacheInfo();
D
dapan1121 已提交
2072 2073 2074 2075
    ctgdShowClusterCache(pCtg);
  }

  qInfo("catalog update thread stopped");
dengyihao's avatar
dengyihao 已提交
2076

D
dapan1121 已提交
2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088
  return NULL;
}

int32_t ctgStartUpdateThread() {
  TdThreadAttr thAttr;
  taosThreadAttrInit(&thAttr);
  taosThreadAttrSetDetachState(&thAttr, PTHREAD_CREATE_JOINABLE);

  if (taosThreadCreate(&gCtgMgmt.updateThread, &thAttr, ctgUpdateThreadFunc, NULL) != 0) {
    terrno = TAOS_SYSTEM_ERROR(errno);
    CTG_ERR_RET(terrno);
  }
dengyihao's avatar
dengyihao 已提交
2089

D
dapan1121 已提交
2090 2091 2092 2093
  taosThreadAttrDestroy(&thAttr);
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
2094
int32_t ctgGetTbMetaFromCache(SCatalog *pCtg, SRequestConnInfo *pConn, SCtgTbMetaCtx *ctx, STableMeta **pTableMeta) {
D
dapan1121 已提交
2095
  if (IS_SYS_DBNAME(ctx->pName->dbname)) {
D
dapan1121 已提交
2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116
    CTG_FLAG_SET_SYS_DB(ctx->flag);
  }

  CTG_ERR_RET(ctgReadTbMetaFromCache(pCtg, ctx, pTableMeta));

  if (*pTableMeta) {
    if (CTG_FLAG_MATCH_STB(ctx->flag, (*pTableMeta)->tableType) &&
        ((!CTG_FLAG_IS_FORCE_UPDATE(ctx->flag)) || (CTG_FLAG_IS_SYS_DB(ctx->flag)))) {
      return TSDB_CODE_SUCCESS;
    }

    taosMemoryFreeClear(*pTableMeta);
  }

  if (CTG_FLAG_IS_UNKNOWN_STB(ctx->flag)) {
    CTG_FLAG_SET_STB(ctx->flag, ctx->tbInfo.tbType);
  }

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
2117
#if 0
2118
int32_t ctgGetTbMetaBFromCache(SCatalog* pCtg, SRequestConnInfo *pConn, SCtgTbMetasCtx* ctx, SArray** pResList) {
D
dapan1121 已提交
2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155
  int32_t tbNum = taosArrayGetSize(ctx->pNames);
  SName* fName = taosArrayGet(ctx->pNames, 0);
  int32_t fIdx = 0;
  
  for (int32_t i = 0; i < tbNum; ++i) {
    SName* pName = taosArrayGet(ctx->pNames, i);
    SCtgTbMetaCtx nctx = {0};
    nctx.flag = CTG_FLAG_UNKNOWN_STB;
    nctx.pName = pName;
    
    if (IS_SYS_DBNAME(pName->dbname)) {
      CTG_FLAG_SET_SYS_DB(nctx.flag);
    }

    STableMeta *pTableMeta = NULL;
    CTG_ERR_RET(ctgReadTbMetaFromCache(pCtg, &nctx, &pTableMeta));
    SMetaRes res = {0};
    
    if (pTableMeta) {
      if (CTG_FLAG_MATCH_STB(nctx.flag, pTableMeta->tableType) &&
          ((!CTG_FLAG_IS_FORCE_UPDATE(nctx.flag)) || (CTG_FLAG_IS_SYS_DB(nctx.flag)))) {
        res.pRes = pTableMeta;
      } else {
        taosMemoryFreeClear(pTableMeta);
      }
    }

    if (NULL == res.pRes) {
      if (NULL == ctx->pFetchs) {
        ctx->pFetchs = taosArrayInit(tbNum, sizeof(SCtgFetch));
      }
      
      if (CTG_FLAG_IS_UNKNOWN_STB(nctx.flag)) {
        CTG_FLAG_SET_STB(nctx.flag, nctx.tbInfo.tbType);
      }

      SCtgFetch fetch = {0};
2156
      fetch.tbIdx = i;
D
dapan1121 已提交
2157 2158 2159 2160 2161 2162
      fetch.fetchIdx = fIdx++;
      fetch.flag = nctx.flag;

      taosArrayPush(ctx->pFetchs, &fetch);
    }
    
2163
    taosArrayPush(ctx->pResList, &res);
D
dapan1121 已提交
2164 2165 2166
  }

  if (NULL == ctx->pFetchs) {
2167
    TSWAP(*pResList, ctx->pResList);
D
dapan1121 已提交
2168 2169 2170 2171
  }

  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
2172 2173
#endif

dengyihao's avatar
dengyihao 已提交
2174 2175 2176 2177 2178 2179 2180 2181 2182
int32_t ctgGetTbMetasFromCache(SCatalog *pCtg, SRequestConnInfo *pConn, SCtgTbMetasCtx *ctx, int32_t dbIdx,
                               int32_t *fetchIdx, int32_t baseResIdx, SArray *pList) {
  int32_t     tbNum = taosArrayGetSize(pList);
  SName      *pName = taosArrayGet(pList, 0);
  char        dbFName[TSDB_DB_FNAME_LEN] = {0};
  int32_t     flag = CTG_FLAG_UNKNOWN_STB;
  uint64_t    lastSuid = 0;
  STableMeta *lastTableMeta = NULL;

D
dapan1121 已提交
2183 2184 2185 2186 2187 2188 2189 2190
  if (IS_SYS_DBNAME(pName->dbname)) {
    CTG_FLAG_SET_SYS_DB(flag);
    strcpy(dbFName, pName->dbname);
  } else {
    tNameGetFullDbName(pName, dbFName);
  }

  SCtgDBCache *dbCache = NULL;
dengyihao's avatar
dengyihao 已提交
2191
  SCtgTbCache *pCache = NULL;
D
dapan1121 已提交
2192
  ctgAcquireDBCache(pCtg, dbFName, &dbCache);
dengyihao's avatar
dengyihao 已提交
2193

D
dapan1121 已提交
2194 2195 2196
  if (NULL == dbCache) {
    ctgDebug("db %s not in cache", dbFName);
    for (int32_t i = 0; i < tbNum; ++i) {
2197 2198
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
D
dapan1121 已提交
2199 2200 2201 2202 2203 2204
    }

    return TSDB_CODE_SUCCESS;
  }

  for (int32_t i = 0; i < tbNum; ++i) {
dengyihao's avatar
dengyihao 已提交
2205
    SName *pName = taosArrayGet(pList, i);
D
dapan1121 已提交
2206 2207 2208 2209

    pCache = taosHashAcquire(dbCache->tbCache, pName->tname, strlen(pName->tname));
    if (NULL == pCache) {
      ctgDebug("tb %s not in cache, dbFName:%s", pName->tname, dbFName);
2210 2211
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
dengyihao's avatar
dengyihao 已提交
2212

D
dapan1121 已提交
2213 2214 2215 2216 2217 2218
      continue;
    }

    CTG_LOCK(CTG_READ, &pCache->metaLock);
    if (NULL == pCache->pMeta) {
      ctgDebug("tb %s meta not in cache, dbFName:%s", pName->tname, dbFName);
2219 2220
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
dengyihao's avatar
dengyihao 已提交
2221

D
dapan1121 已提交
2222 2223 2224
      continue;
    }

dengyihao's avatar
dengyihao 已提交
2225
    STableMeta *tbMeta = pCache->pMeta;
D
dapan1121 已提交
2226 2227 2228 2229 2230 2231 2232 2233

    SCtgTbMetaCtx nctx = {0};
    nctx.flag = flag;
    nctx.tbInfo.inCache = true;
    nctx.tbInfo.dbId = dbCache->dbId;
    nctx.tbInfo.suid = tbMeta->suid;
    nctx.tbInfo.tbType = tbMeta->tableType;

dengyihao's avatar
dengyihao 已提交
2234 2235
    SMetaRes    res = {0};
    STableMeta *pTableMeta = NULL;
D
dapan1121 已提交
2236 2237 2238 2239
    if (tbMeta->tableType != TSDB_CHILD_TABLE) {
      int32_t metaSize = CTG_META_SIZE(tbMeta);
      pTableMeta = taosMemoryCalloc(1, metaSize);
      if (NULL == pTableMeta) {
2240
        ctgReleaseTbMetaToCache(pCtg, dbCache, pCache);
D
dapan1121 已提交
2241 2242
        CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
      }
dengyihao's avatar
dengyihao 已提交
2243

D
dapan1121 已提交
2244
      memcpy(pTableMeta, tbMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
2245

2246
      CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2247 2248
      taosHashRelease(dbCache->tbCache, pCache);

D
dapan1121 已提交
2249
      ctgDebug("Got tb %s meta from cache, type:%d, dbFName:%s", pName->tname, tbMeta->tableType, dbFName);
dengyihao's avatar
dengyihao 已提交
2250

D
dapan1121 已提交
2251
      res.pRes = pTableMeta;
2252
      taosArrayPush(ctx->pResList, &res);
D
dapan1121 已提交
2253 2254 2255

      continue;
    }
dengyihao's avatar
dengyihao 已提交
2256

D
dapan1121 已提交
2257 2258 2259 2260
    // PROCESS FOR CHILD TABLE

    if (lastSuid && tbMeta->suid == lastSuid && lastTableMeta) {
      cloneTableMeta(lastTableMeta, &pTableMeta);
2261
      memcpy(pTableMeta, tbMeta, sizeof(SCTableMeta));
D
dapan1121 已提交
2262

2263
      CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2264 2265
      taosHashRelease(dbCache->tbCache, pCache);

D
dapan1121 已提交
2266
      ctgDebug("Got tb %s meta from cache, type:%d, dbFName:%s", pName->tname, tbMeta->tableType, dbFName);
dengyihao's avatar
dengyihao 已提交
2267

D
dapan1121 已提交
2268
      res.pRes = pTableMeta;
2269
      taosArrayPush(ctx->pResList, &res);
dengyihao's avatar
dengyihao 已提交
2270

D
dapan1121 已提交
2271 2272
      continue;
    }
dengyihao's avatar
dengyihao 已提交
2273

D
dapan1121 已提交
2274 2275 2276
    int32_t metaSize = sizeof(SCTableMeta);
    pTableMeta = taosMemoryCalloc(1, metaSize);
    if (NULL == pTableMeta) {
dengyihao's avatar
dengyihao 已提交
2277
      ctgReleaseTbMetaToCache(pCtg, dbCache, pCache);
D
dapan1121 已提交
2278 2279
      CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
    }
dengyihao's avatar
dengyihao 已提交
2280

D
dapan1121 已提交
2281
    memcpy(pTableMeta, tbMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
2282

2283
    CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2284 2285 2286 2287 2288 2289
    taosHashRelease(dbCache->tbCache, pCache);

    ctgDebug("Got ctb %s meta from cache, will continue to get its stb meta, type:%d, dbFName:%s", pName->tname,
             nctx.tbInfo.tbType, dbFName);

    char *stName = taosHashAcquire(dbCache->stbCache, &pTableMeta->suid, sizeof(pTableMeta->suid));
D
dapan1121 已提交
2290 2291
    if (NULL == stName) {
      ctgDebug("stb 0x%" PRIx64 " not in cache, dbFName:%s", pTableMeta->suid, dbFName);
2292 2293
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
D
dapan1121 已提交
2294 2295 2296 2297 2298 2299 2300 2301 2302

      taosMemoryFreeClear(pTableMeta);
      continue;
    }

    pCache = taosHashAcquire(dbCache->tbCache, stName, strlen(stName));
    if (NULL == pCache) {
      ctgDebug("stb 0x%" PRIx64 " name %s not in cache, dbFName:%s", pTableMeta->suid, stName, dbFName);
      taosHashRelease(dbCache->stbCache, stName);
dengyihao's avatar
dengyihao 已提交
2303

2304 2305
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
D
dapan1121 已提交
2306

dengyihao's avatar
dengyihao 已提交
2307
      taosMemoryFreeClear(pTableMeta);
D
dapan1121 已提交
2308 2309 2310 2311 2312 2313 2314 2315
      continue;
    }

    taosHashRelease(dbCache->stbCache, stName);

    CTG_LOCK(CTG_READ, &pCache->metaLock);
    if (NULL == pCache->pMeta) {
      ctgDebug("stb 0x%" PRIx64 " meta not in cache, dbFName:%s", pTableMeta->suid, dbFName);
2316
      CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2317 2318
      taosHashRelease(dbCache->tbCache, pCache);

2319 2320
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
D
dapan1121 已提交
2321 2322 2323 2324 2325

      taosMemoryFreeClear(pTableMeta);

      continue;
    }
dengyihao's avatar
dengyihao 已提交
2326 2327 2328

    STableMeta *stbMeta = pCache->pMeta;
    if (stbMeta->suid != nctx.tbInfo.suid) {
2329
      CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2330 2331 2332 2333 2334
      taosHashRelease(dbCache->tbCache, pCache);

      ctgError("stb suid 0x%" PRIx64 " in stbCache mis-match, expected suid 0x%" PRIx64, stbMeta->suid,
               nctx.tbInfo.suid);

2335 2336
      ctgAddFetch(&ctx->pFetchs, dbIdx, i, fetchIdx, baseResIdx + i, flag);
      taosArraySetSize(ctx->pResList, taosArrayGetSize(ctx->pResList) + 1);
D
dapan1121 已提交
2337 2338 2339 2340 2341

      taosMemoryFreeClear(pTableMeta);

      continue;
    }
dengyihao's avatar
dengyihao 已提交
2342

D
dapan1121 已提交
2343 2344
    metaSize = CTG_META_SIZE(stbMeta);
    pTableMeta = taosMemoryRealloc(pTableMeta, metaSize);
dengyihao's avatar
dengyihao 已提交
2345
    if (NULL == pTableMeta) {
2346
      ctgReleaseTbMetaToCache(pCtg, dbCache, pCache);
D
dapan1121 已提交
2347 2348
      CTG_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
    }
dengyihao's avatar
dengyihao 已提交
2349

D
dapan1121 已提交
2350
    memcpy(&pTableMeta->sversion, &stbMeta->sversion, metaSize - sizeof(SCTableMeta));
dengyihao's avatar
dengyihao 已提交
2351

2352
    CTG_UNLOCK(CTG_READ, &pCache->metaLock);
dengyihao's avatar
dengyihao 已提交
2353 2354
    taosHashRelease(dbCache->tbCache, pCache);

D
dapan1121 已提交
2355
    res.pRes = pTableMeta;
2356
    taosArrayPush(ctx->pResList, &res);
D
dapan1121 已提交
2357 2358 2359 2360 2361

    lastSuid = pTableMeta->suid;
    lastTableMeta = pTableMeta;
  }

2362
  ctgReleaseDBCache(pCtg, dbCache);
dengyihao's avatar
dengyihao 已提交
2363

D
dapan1121 已提交
2364 2365 2366
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
2367
int32_t ctgRemoveTbMetaFromCache(SCatalog *pCtg, SName *pTableName, bool syncReq) {
D
dapan1121 已提交
2368
  int32_t       code = 0;
dengyihao's avatar
dengyihao 已提交
2369
  STableMeta   *tblMeta = NULL;
D
dapan1121 已提交
2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397
  SCtgTbMetaCtx tbCtx = {0};
  tbCtx.flag = CTG_FLAG_UNKNOWN_STB;
  tbCtx.pName = pTableName;

  CTG_ERR_JRET(ctgReadTbMetaFromCache(pCtg, &tbCtx, &tblMeta));

  if (NULL == tblMeta) {
    ctgDebug("table already not in cache, db:%s, tblName:%s", pTableName->dbname, pTableName->tname);
    return TSDB_CODE_SUCCESS;
  }

  char dbFName[TSDB_DB_FNAME_LEN];
  tNameGetFullDbName(pTableName, dbFName);

  if (TSDB_SUPER_TABLE == tblMeta->tableType) {
    CTG_ERR_JRET(ctgDropStbMetaEnqueue(pCtg, dbFName, tbCtx.tbInfo.dbId, pTableName->tname, tblMeta->suid, syncReq));
  } else {
    CTG_ERR_JRET(ctgDropTbMetaEnqueue(pCtg, dbFName, tbCtx.tbInfo.dbId, pTableName->tname, syncReq));
  }

_return:

  taosMemoryFreeClear(tblMeta);

  CTG_RET(code);
}

int32_t ctgGetTbHashVgroupFromCache(SCatalog *pCtg, const SName *pTableName, SVgroupInfo **pVgroup) {
D
dapan1121 已提交
2398
  if (IS_SYS_DBNAME(pTableName->dbname)) {
D
dapan1121 已提交
2399 2400 2401 2402
    ctgError("no valid vgInfo for db, dbname:%s", pTableName->dbname);
    CTG_ERR_RET(TSDB_CODE_CTG_INVALID_INPUT);
  }

dengyihao's avatar
dengyihao 已提交
2403
  SCtgDBCache *dbCache = NULL;
D
dapan1121 已提交
2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429
  int32_t      code = 0;
  char         dbFName[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(pTableName, dbFName);

  CTG_ERR_RET(ctgAcquireVgInfoFromCache(pCtg, dbFName, &dbCache));

  if (NULL == dbCache) {
    *pVgroup = NULL;
    return TSDB_CODE_SUCCESS;
  }

  *pVgroup = taosMemoryCalloc(1, sizeof(SVgroupInfo));
  CTG_ERR_JRET(ctgGetVgInfoFromHashValue(pCtg, dbCache->vgCache.vgInfo, pTableName, *pVgroup));

_return:

  if (dbCache) {
    ctgReleaseVgInfoToCache(pCtg, dbCache);
  }

  if (code) {
    taosMemoryFreeClear(*pVgroup);
  }

  CTG_RET(code);
}