metaQuery.c 38.6 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
Hongze Cheng 已提交
16
#include "meta.h"
H
Hongze Cheng 已提交
17

H
Hongze Cheng 已提交
18
void metaReaderInit(SMetaReader *pReader, SMeta *pMeta, int32_t flags) {
H
Hongze Cheng 已提交
19 20
  memset(pReader, 0, sizeof(*pReader));
  pReader->flags = flags;
H
Hongze Cheng 已提交
21
  pReader->pMeta = pMeta;
22
  if (pReader->pMeta && !(flags & META_READER_NOLOCK)) {
23 24
    metaRLock(pMeta);
  }
H
Hongze Cheng 已提交
25
}
H
Hongze Cheng 已提交
26

27 28 29 30 31 32 33
void metaReaderReleaseLock(SMetaReader *pReader) {
  if (pReader->pMeta && !(pReader->flags & META_READER_NOLOCK)) {
    metaULock(pReader->pMeta);
    pReader->flags |= META_READER_NOLOCK;
  }
}

H
Hongze Cheng 已提交
34
void metaReaderClear(SMetaReader *pReader) {
35
  if (pReader->pMeta && !(pReader->flags & META_READER_NOLOCK)) {
H
Hongze Cheng 已提交
36 37
    metaULock(pReader->pMeta);
  }
H
Hongze Cheng 已提交
38
  tDecoderClear(&pReader->coder);
H
Hongze Cheng 已提交
39
  tdbFree(pReader->pBuf);
H
Hongze Cheng 已提交
40 41
}

H
Hongze Cheng 已提交
42
int metaGetTableEntryByVersion(SMetaReader *pReader, int64_t version, tb_uid_t uid) {
H
Hongze Cheng 已提交
43
  SMeta   *pMeta = pReader->pMeta;
H
Hongze Cheng 已提交
44 45
  STbDbKey tbDbKey = {.version = version, .uid = uid};

H
Hongze Cheng 已提交
46
  // query table.db
H
Hongze Cheng 已提交
47
  if (tdbTbGet(pMeta->pTbDb, &tbDbKey, sizeof(tbDbKey), &pReader->pBuf, &pReader->szBuf) < 0) {
H
Hongze Cheng 已提交
48
    terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
H
Hongze Cheng 已提交
49 50 51 52
    goto _err;
  }

  // decode the entry
H
Hongze Cheng 已提交
53
  tDecoderInit(&pReader->coder, pReader->pBuf, pReader->szBuf);
H
Hongze Cheng 已提交
54 55 56 57 58 59 60 61 62 63 64

  if (metaDecodeEntry(&pReader->coder, &pReader->me) < 0) {
    goto _err;
  }

  return 0;

_err:
  return -1;
}

dengyihao's avatar
dengyihao 已提交
65
// int metaGetTableEntryByUidTest(void* meta, SArray *uidList) {
wmmhello's avatar
wmmhello 已提交
66
//
dengyihao's avatar
dengyihao 已提交
67 68 69 70 71
//   SArray* readerList = taosArrayInit(taosArrayGetSize(uidList), sizeof(SMetaReader));
//   SArray* uidVersion = taosArrayInit(taosArrayGetSize(uidList), sizeof(STbDbKey));
//   SMeta  *pMeta = meta;
//   int64_t version;
//   SHashObj *uHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
wmmhello's avatar
wmmhello 已提交
72
//
dengyihao's avatar
dengyihao 已提交
73 74 75 76 77 78 79 80 81 82
//   int64_t stt1 = taosGetTimestampUs();
//   for(int i = 0; i < taosArrayGetSize(uidList); i++) {
//     void* ppVal = NULL;
//     int vlen = 0;
//     uint64_t *  uid = taosArrayGet(uidList, i);
//     // query uid.idx
//     if (tdbTbGet(pMeta->pUidIdx, uid, sizeof(*uid), &ppVal, &vlen) < 0) {
//       continue;
//     }
//     version = *(int64_t *)ppVal;
wmmhello's avatar
wmmhello 已提交
83
//
dengyihao's avatar
dengyihao 已提交
84 85 86 87 88 89
//     STbDbKey tbDbKey = {.version = version, .uid = *uid};
//     taosArrayPush(uidVersion, &tbDbKey);
//     taosHashPut(uHash, uid, sizeof(int64_t), ppVal, sizeof(int64_t));
//   }
//   int64_t stt2 = taosGetTimestampUs();
//   qDebug("metaGetTableEntryByUidTest1 rows:%d, cost:%ld us", taosArrayGetSize(uidList), stt2-stt1);
wmmhello's avatar
wmmhello 已提交
90
//
dengyihao's avatar
dengyihao 已提交
91 92 93 94 95
//   TBC        *pCur = NULL;
//   tdbTbcOpen(pMeta->pTbDb, &pCur, NULL);
//   tdbTbcMoveToFirst(pCur);
//   void *pKey = NULL;
//   int   kLen = 0;
wmmhello's avatar
wmmhello 已提交
96
//
dengyihao's avatar
dengyihao 已提交
97 98 99 100 101 102 103 104 105 106
//   while(1){
//     SMetaReader pReader = {0};
//     int32_t ret = tdbTbcNext(pCur, &pKey, &kLen, &pReader.pBuf, &pReader.szBuf);
//     if (ret < 0) break;
//     STbDbKey *tmp = (STbDbKey*)pKey;
//     int64_t *ver = (int64_t*)taosHashGet(uHash, &tmp->uid, sizeof(int64_t));
//     if(ver == NULL || *ver != tmp->version) continue;
//     taosArrayPush(readerList, &pReader);
//   }
//   tdbTbcClose(pCur);
wmmhello's avatar
wmmhello 已提交
107
//
dengyihao's avatar
dengyihao 已提交
108 109 110 111 112
//   taosArrayClear(readerList);
//   int64_t stt3 = taosGetTimestampUs();
//   qDebug("metaGetTableEntryByUidTest2 rows:%d, cost:%ld us", taosArrayGetSize(uidList), stt3-stt2);
//   for(int i = 0; i < taosArrayGetSize(uidVersion); i++) {
//     SMetaReader pReader = {0};
wmmhello's avatar
wmmhello 已提交
113
//
dengyihao's avatar
dengyihao 已提交
114 115 116 117 118 119 120 121 122
//     STbDbKey *tbDbKey = taosArrayGet(uidVersion, i);
//     // query table.db
//     if (tdbTbGet(pMeta->pTbDb, tbDbKey, sizeof(STbDbKey), &pReader.pBuf, &pReader.szBuf) < 0) {
//       continue;
//     }
//     taosArrayPush(readerList, &pReader);
//   }
//   int64_t stt4 = taosGetTimestampUs();
//   qDebug("metaGetTableEntryByUidTest3 rows:%d, cost:%ld us", taosArrayGetSize(uidList), stt4-stt3);
wmmhello's avatar
wmmhello 已提交
123
//
dengyihao's avatar
dengyihao 已提交
124 125 126 127 128
//   for(int i = 0; i < taosArrayGetSize(readerList); i++){
//     SMetaReader* pReader  = taosArrayGet(readerList, i);
//     metaReaderInit(pReader, meta, 0);
//     // decode the entry
//     tDecoderInit(&pReader->coder, pReader->pBuf, pReader->szBuf);
wmmhello's avatar
wmmhello 已提交
129
//
dengyihao's avatar
dengyihao 已提交
130 131 132 133 134 135 136 137
//     if (metaDecodeEntry(&pReader->coder, &pReader->me) < 0) {
//     }
//     metaReaderClear(pReader);
//   }
//   int64_t stt5 = taosGetTimestampUs();
//   qDebug("metaGetTableEntryByUidTest4 rows:%d, cost:%ld us", taosArrayGetSize(readerList), stt5-stt4);
//   return 0;
// }
wmmhello's avatar
wmmhello 已提交
138

H
Hongze Cheng 已提交
139
bool metaIsTableExist(SMeta *pMeta, tb_uid_t uid) {
140
  // query uid.idx
141 142
  metaRLock(pMeta);

143
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), NULL, NULL) < 0) {
144 145
    metaULock(pMeta);

146 147 148
    return false;
  }

149 150
  metaULock(pMeta);

151 152 153
  return true;
}

H
Hongze Cheng 已提交
154
int metaGetTableEntryByUid(SMetaReader *pReader, tb_uid_t uid) {
155
  SMeta  *pMeta = pReader->pMeta;
156
  int64_t version1;
H
Hongze Cheng 已提交
157 158

  // query uid.idx
H
Hongze Cheng 已提交
159
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pReader->pBuf, &pReader->szBuf) < 0) {
H
Hongze Cheng 已提交
160
    terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
H
Hongze Cheng 已提交
161 162 163
    return -1;
  }

164 165
  version1 = ((SUidIdxVal *)pReader->pBuf)[0].version;
  return metaGetTableEntryByVersion(pReader, version1, uid);
H
Hongze Cheng 已提交
166 167
}

H
Haojun Liao 已提交
168 169
int metaGetTableEntryByUidCache(SMetaReader *pReader, tb_uid_t uid) {
  SMeta *pMeta = pReader->pMeta;
170 171 172 173 174 175 176 177

  SMetaInfo info;
  if (metaGetInfo(pMeta, uid, &info, pReader) == TSDB_CODE_NOT_FOUND) {
    terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
    return -1;
  }

  return metaGetTableEntryByVersion(pReader, info.version, uid);
H
Hongze Cheng 已提交
178 179
}

H
Hongze Cheng 已提交
180
int metaGetTableEntryByName(SMetaReader *pReader, const char *name) {
H
Hongze Cheng 已提交
181
  SMeta   *pMeta = pReader->pMeta;
H
Hongze Cheng 已提交
182 183 184
  tb_uid_t uid;

  // query name.idx
H
Hongze Cheng 已提交
185
  if (tdbTbGet(pMeta->pNameIdx, name, strlen(name) + 1, &pReader->pBuf, &pReader->szBuf) < 0) {
H
Hongze Cheng 已提交
186
    terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
H
Hongze Cheng 已提交
187 188 189 190
    return -1;
  }

  uid = *(tb_uid_t *)pReader->pBuf;
H
Hongze Cheng 已提交
191
  return metaGetTableEntryByUid(pReader, uid);
H
Hongze Cheng 已提交
192 193
}

H
Hongze Cheng 已提交
194
tb_uid_t metaGetTableEntryUidByName(SMeta *pMeta, const char *name) {
H
Hongze Cheng 已提交
195
  void    *pData = NULL;
H
Hongze Cheng 已提交
196 197 198
  int      nData = 0;
  tb_uid_t uid = 0;

wmmhello's avatar
wmmhello 已提交
199 200
  metaRLock(pMeta);

H
Hongze Cheng 已提交
201 202 203 204 205
  if (tdbTbGet(pMeta->pNameIdx, name, strlen(name) + 1, &pData, &nData) == 0) {
    uid = *(tb_uid_t *)pData;
    tdbFree(pData);
  }

wmmhello's avatar
wmmhello 已提交
206 207
  metaULock(pMeta);

208
  return uid;
H
Hongze Cheng 已提交
209 210
}

dengyihao's avatar
dengyihao 已提交
211
int metaGetTableNameByUid(void *meta, uint64_t uid, char *tbName) {
212
  int         code = 0;
D
dapan1121 已提交
213
  SMetaReader mr = {0};
dengyihao's avatar
dengyihao 已提交
214
  metaReaderInit(&mr, (SMeta *)meta, 0);
215 216 217 218 219
  code = metaGetTableEntryByUid(&mr, uid);
  if (code < 0) {
    metaReaderClear(&mr);
    return -1;
  }
D
dapan1121 已提交
220 221 222 223 224 225

  STR_TO_VARSTR(tbName, mr.me.name);
  metaReaderClear(&mr);

  return 0;
}
S
slzhou 已提交
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241

int metaGetTableSzNameByUid(void *meta, uint64_t uid, char *tbName) {
  int         code = 0;
  SMetaReader mr = {0};
  metaReaderInit(&mr, (SMeta *)meta, 0);
  code = metaGetTableEntryByUid(&mr, uid);
  if (code < 0) {
    metaReaderClear(&mr);
    return -1;
  }
  strncpy(tbName, mr.me.name, TSDB_TABLE_NAME_LEN);
  metaReaderClear(&mr);

  return 0;
}

H
Haojun Liao 已提交
242
int metaGetTableUidByName(void *meta, char *tbName, uint64_t *uid) {
dengyihao's avatar
dengyihao 已提交
243 244 245
  int         code = 0;
  SMetaReader mr = {0};
  metaReaderInit(&mr, (SMeta *)meta, 0);
dengyihao's avatar
dengyihao 已提交
246 247 248 249

  SMetaReader *pReader = &mr;

  // query name.idx
dengyihao's avatar
dengyihao 已提交
250
  if (tdbTbGet(pReader->pMeta->pNameIdx, tbName, strlen(tbName) + 1, &pReader->pBuf, &pReader->szBuf) < 0) {
dengyihao's avatar
dengyihao 已提交
251
    terrno = TSDB_CODE_PAR_TABLE_NOT_EXIST;
dengyihao's avatar
dengyihao 已提交
252 253 254 255
    metaReaderClear(&mr);
    return -1;
  }

dengyihao's avatar
dengyihao 已提交
256
  *uid = *(tb_uid_t *)pReader->pBuf;
dengyihao's avatar
dengyihao 已提交
257

dengyihao's avatar
dengyihao 已提交
258 259 260 261
  metaReaderClear(&mr);

  return 0;
}
D
dapan1121 已提交
262

dengyihao's avatar
dengyihao 已提交
263 264 265 266 267
int metaGetTableTypeByName(void *meta, char *tbName, ETableType *tbType) {
  int         code = 0;
  SMetaReader mr = {0};
  metaReaderInit(&mr, (SMeta *)meta, 0);

dengyihao's avatar
dengyihao 已提交
268 269 270
  code = metaGetTableEntryByName(&mr, tbName);
  if (code == 0) *tbType = mr.me.type;

dengyihao's avatar
dengyihao 已提交
271
  metaReaderClear(&mr);
dengyihao's avatar
dengyihao 已提交
272
  return code;
dengyihao's avatar
dengyihao 已提交
273 274
}

H
Hongze Cheng 已提交
275
int metaReadNext(SMetaReader *pReader) {
H
Hongze Cheng 已提交
276 277
  SMeta *pMeta = pReader->pMeta;

H
Hongze Cheng 已提交
278
  // TODO
H
Hongze Cheng 已提交
279

H
Hongze Cheng 已提交
280 281 282 283
  return 0;
}

#if 1  // ===================================================
H
Hongze Cheng 已提交
284 285 286 287 288 289 290 291
SMTbCursor *metaOpenTbCursor(SMeta *pMeta) {
  SMTbCursor *pTbCur = NULL;

  pTbCur = (SMTbCursor *)taosMemoryCalloc(1, sizeof(*pTbCur));
  if (pTbCur == NULL) {
    return NULL;
  }

H
Hongze Cheng 已提交
292
  metaReaderInit(&pTbCur->mr, pMeta, 0);
H
Hongze Cheng 已提交
293

H
Hongze Cheng 已提交
294
  tdbTbcOpen(pMeta->pUidIdx, &pTbCur->pDbc, NULL);
H
Hongze Cheng 已提交
295

H
Hongze Cheng 已提交
296
  tdbTbcMoveToFirst(pTbCur->pDbc);
H
Hongze Cheng 已提交
297

H
Hongze Cheng 已提交
298 299 300 301 302
  return pTbCur;
}

void metaCloseTbCursor(SMTbCursor *pTbCur) {
  if (pTbCur) {
H
Hongze Cheng 已提交
303 304
    tdbFree(pTbCur->pKey);
    tdbFree(pTbCur->pVal);
H
Hongze Cheng 已提交
305
    metaReaderClear(&pTbCur->mr);
H
Hongze Cheng 已提交
306
    if (pTbCur->pDbc) {
H
Hongze Cheng 已提交
307
      tdbTbcClose(pTbCur->pDbc);
H
Hongze Cheng 已提交
308 309 310 311 312
    }
    taosMemoryFree(pTbCur);
  }
}

X
Xiaoyu Wang 已提交
313
int32_t metaTbCursorNext(SMTbCursor *pTbCur, ETableType jumpTableType) {
H
Hongze Cheng 已提交
314
  int    ret;
H
Hongze Cheng 已提交
315
  void  *pBuf;
H
Hongze Cheng 已提交
316 317 318
  STbCfg tbCfg;

  for (;;) {
H
Hongze Cheng 已提交
319
    ret = tdbTbcNext(pTbCur->pDbc, &pTbCur->pKey, &pTbCur->kLen, &pTbCur->pVal, &pTbCur->vLen);
H
Hongze Cheng 已提交
320 321
    if (ret < 0) {
      return -1;
H
Hongze Cheng 已提交
322 323
    }

M
Minglei Jin 已提交
324 325
    tDecoderClear(&pTbCur->mr.coder);

H
Hongze Cheng 已提交
326
    metaGetTableEntryByVersion(&pTbCur->mr, ((SUidIdxVal *)pTbCur->pVal)[0].version, *(tb_uid_t *)pTbCur->pKey);
wmmhello's avatar
wmmhello 已提交
327
    if (pTbCur->mr.me.type == jumpTableType) {
H
Hongze Cheng 已提交
328 329
      continue;
    }
H
Hongze Cheng 已提交
330 331

    break;
H
Hongze Cheng 已提交
332 333
  }

H
Hongze Cheng 已提交
334
  return 0;
H
Hongze Cheng 已提交
335 336
}

337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360
int32_t metaTbCursorPrev(SMTbCursor *pTbCur) {
  int    ret;
  void  *pBuf;
  STbCfg tbCfg;

  for (;;) {
    ret = tdbTbcPrev(pTbCur->pDbc, &pTbCur->pKey, &pTbCur->kLen, &pTbCur->pVal, &pTbCur->vLen);
    if (ret < 0) {
      return -1;
    }

    tDecoderClear(&pTbCur->mr.coder);

    metaGetTableEntryByVersion(&pTbCur->mr, ((SUidIdxVal *)pTbCur->pVal)[0].version, *(tb_uid_t *)pTbCur->pKey);
    if (pTbCur->mr.me.type == TSDB_SUPER_TABLE) {
      continue;
    }

    break;
  }

  return 0;
}

361
SSchemaWrapper *metaGetTableSchema(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock) {
H
Hongze Cheng 已提交
362
  void           *pData = NULL;
H
Hongze Cheng 已提交
363 364 365 366 367
  int             nData = 0;
  int64_t         version;
  SSchemaWrapper  schema = {0};
  SSchemaWrapper *pSchema = NULL;
  SDecoder        dc = {0};
368 369 370
  if (lock) {
    metaRLock(pMeta);
  }
H
Hongze Cheng 已提交
371 372 373 374
_query:
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData) < 0) {
    goto _err;
  }
H
Hongze Cheng 已提交
375

H
Hongze Cheng 已提交
376
  version = ((SUidIdxVal *)pData)[0].version;
H
Hongze Cheng 已提交
377

H
Hongze Cheng 已提交
378 379 380 381 382 383
  tdbTbGet(pMeta->pTbDb, &(STbDbKey){.uid = uid, .version = version}, sizeof(STbDbKey), &pData, &nData);
  SMetaEntry me = {0};
  tDecoderInit(&dc, pData, nData);
  metaDecodeEntry(&dc, &me);
  if (me.type == TSDB_SUPER_TABLE) {
    if (sver == -1 || sver == me.stbEntry.schemaRow.version) {
H
Hongze Cheng 已提交
384
      pSchema = tCloneSSchemaWrapper(&me.stbEntry.schemaRow);
H
Hongze Cheng 已提交
385 386
      tDecoderClear(&dc);
      goto _exit;
H
Hongze Cheng 已提交
387
    }
388 389
    {  // Traverse to find the previous qualified data
      TBC *pCur;
390 391
      tdbTbcOpen(pMeta->pTbDb, &pCur, NULL);
      STbDbKey key = {.version = sver, .uid = INT64_MAX};
392
      int      c = 0;
393
      tdbTbcMoveTo(pCur, &key, sizeof(key), &c);
394
      if (c < 0) {
395 396 397 398 399 400
        tdbTbcMoveToPrev(pCur);
      }

      void *pKey = NULL;
      void *pVal = NULL;
      int   vLen = 0, kLen = 0;
401
      while (1) {
402 403 404
        int32_t ret = tdbTbcPrev(pCur, &pKey, &kLen, &pVal, &vLen);
        if (ret < 0) break;

405 406
        STbDbKey *tmp = (STbDbKey *)pKey;
        if (tmp->uid != uid) {
407 408 409 410 411 412 413 414 415
          continue;
        }
        SDecoder   dcNew = {0};
        SMetaEntry meNew = {0};
        tDecoderInit(&dcNew, pVal, vLen);
        metaDecodeEntry(&dcNew, &meNew);
        pSchema = tCloneSSchemaWrapper(&meNew.stbEntry.schemaRow);
        tDecoderClear(&dcNew);
        tdbTbcClose(pCur);
416 417
        tdbFree(pKey);
        tdbFree(pVal);
418 419
        goto _exit;
      }
420 421
      tdbFree(pKey);
      tdbFree(pVal);
422 423
      tdbTbcClose(pCur);
    }
H
Hongze Cheng 已提交
424 425
  } else if (me.type == TSDB_CHILD_TABLE) {
    uid = me.ctbEntry.suid;
H
Hongze Cheng 已提交
426
    tDecoderClear(&dc);
H
Hongze Cheng 已提交
427
    goto _query;
H
Hongze Cheng 已提交
428
  } else {
H
Hongze Cheng 已提交
429 430 431 432
    if (sver == -1 || sver == me.ntbEntry.schemaRow.version) {
      pSchema = tCloneSSchemaWrapper(&me.ntbEntry.schemaRow);
      tDecoderClear(&dc);
      goto _exit;
H
Hongze Cheng 已提交
433
    }
H
Hongze Cheng 已提交
434 435
  }
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
436

H
Hongze Cheng 已提交
437 438 439
  // query from skm db
  if (tdbTbGet(pMeta->pSkmDb, &(SSkmDbKey){.uid = uid, .sver = sver}, sizeof(SSkmDbKey), &pData, &nData) < 0) {
    goto _err;
H
Hongze Cheng 已提交
440
  }
H
Hongze Cheng 已提交
441

H
Hongze Cheng 已提交
442
  tDecoderInit(&dc, pData, nData);
M
Minglei Jin 已提交
443
  tDecodeSSchemaWrapperEx(&dc, &schema);
H
Hongze Cheng 已提交
444 445 446 447
  pSchema = tCloneSSchemaWrapper(&schema);
  tDecoderClear(&dc);

_exit:
448
  tDecoderClear(&dc);
449 450 451
  if (lock) {
    metaULock(pMeta);
  }
H
Hongze Cheng 已提交
452 453
  tdbFree(pData);
  return pSchema;
H
Hongze Cheng 已提交
454

H
Hongze Cheng 已提交
455
_err:
456
  tDecoderClear(&dc);
457 458 459
  if (lock) {
    metaULock(pMeta);
  }
H
Hongze Cheng 已提交
460 461
  tdbFree(pData);
  return NULL;
H
Hongze Cheng 已提交
462 463
}

H
Hongze Cheng 已提交
464 465 466
int metaTtlSmaller(SMeta *pMeta, uint64_t ttl, SArray *uidList) {
  TBC *pCur;
  int  ret = tdbTbcOpen(pMeta->pTtlIdx, &pCur, NULL);
467 468 469 470 471 472 473 474 475 476 477 478 479 480
  if (ret < 0) {
    return ret;
  }

  STtlIdxKey ttlKey = {0};
  ttlKey.dtime = ttl;
  ttlKey.uid = INT64_MAX;
  int c = 0;
  tdbTbcMoveTo(pCur, &ttlKey, sizeof(ttlKey), &c);
  if (c < 0) {
    tdbTbcMoveToPrev(pCur);
  }

  void *pKey = NULL;
H
Hongze Cheng 已提交
481 482
  int   kLen = 0;
  while (1) {
483 484 485 486
    ret = tdbTbcPrev(pCur, &pKey, &kLen, NULL, NULL);
    if (ret < 0) {
      break;
    }
H
Hongze Cheng 已提交
487
    ttlKey = *(STtlIdxKey *)pKey;
488 489
    taosArrayPush(uidList, &ttlKey.uid);
  }
wmmhello's avatar
wmmhello 已提交
490
  tdbFree(pKey);
491 492 493 494
  tdbTbcClose(pCur);
  return 0;
}

H
Hongze Cheng 已提交
495
struct SMCtbCursor {
H
Hongze Cheng 已提交
496 497
  SMeta   *pMeta;
  TBC     *pCur;
H
Hongze Cheng 已提交
498
  tb_uid_t suid;
H
Hongze Cheng 已提交
499 500
  void    *pKey;
  void    *pVal;
H
Hongze Cheng 已提交
501 502 503 504
  int      kLen;
  int      vLen;
};

505
SMCtbCursor *metaOpenCtbCursor(SMeta *pMeta, tb_uid_t uid, int lock) {
H
Hongze Cheng 已提交
506
  SMCtbCursor *pCtbCur = NULL;
H
Hongze Cheng 已提交
507
  SCtbIdxKey   ctbIdxKey;
C
Cary Xu 已提交
508 509
  int          ret = 0;
  int          c = 0;
H
Hongze Cheng 已提交
510

H
Hongze Cheng 已提交
511 512 513 514
  pCtbCur = (SMCtbCursor *)taosMemoryCalloc(1, sizeof(*pCtbCur));
  if (pCtbCur == NULL) {
    return NULL;
  }
H
Hongze Cheng 已提交
515

H
fix  
Hongze Cheng 已提交
516
  pCtbCur->pMeta = pMeta;
H
Hongze Cheng 已提交
517
  pCtbCur->suid = uid;
518 519 520
  if (lock) {
    metaRLock(pMeta);
  }
H
fix  
Hongze Cheng 已提交
521

H
Hongze Cheng 已提交
522
  ret = tdbTbcOpen(pMeta->pCtbIdx, &pCtbCur->pCur, NULL);
H
Hongze Cheng 已提交
523
  if (ret < 0) {
H
fix  
Hongze Cheng 已提交
524
    metaULock(pMeta);
H
Hongze Cheng 已提交
525 526 527
    taosMemoryFree(pCtbCur);
    return NULL;
  }
H
Hongze Cheng 已提交
528

H
Hongze Cheng 已提交
529 530 531
  // move to the suid
  ctbIdxKey.suid = uid;
  ctbIdxKey.uid = INT64_MIN;
H
Hongze Cheng 已提交
532
  tdbTbcMoveTo(pCtbCur->pCur, &ctbIdxKey, sizeof(ctbIdxKey), &c);
H
Hongze Cheng 已提交
533
  if (c > 0) {
H
Hongze Cheng 已提交
534
    tdbTbcMoveToNext(pCtbCur->pCur);
H
Hongze Cheng 已提交
535 536
  }

H
Hongze Cheng 已提交
537 538 539
  return pCtbCur;
}

540
void metaCloseCtbCursor(SMCtbCursor *pCtbCur, int lock) {
H
Hongze Cheng 已提交
541
  if (pCtbCur) {
542
    if (pCtbCur->pMeta && lock) metaULock(pCtbCur->pMeta);
H
Hongze Cheng 已提交
543
    if (pCtbCur->pCur) {
H
Hongze Cheng 已提交
544
      tdbTbcClose(pCtbCur->pCur);
H
Hongze Cheng 已提交
545

H
Hongze Cheng 已提交
546 547
      tdbFree(pCtbCur->pKey);
      tdbFree(pCtbCur->pVal);
H
Hongze Cheng 已提交
548
    }
H
Hongze Cheng 已提交
549

H
Hongze Cheng 已提交
550 551
    taosMemoryFree(pCtbCur);
  }
H
Hongze Cheng 已提交
552 553 554
}

tb_uid_t metaCtbCursorNext(SMCtbCursor *pCtbCur) {
H
Hongze Cheng 已提交
555 556
  int         ret;
  SCtbIdxKey *pCtbIdxKey;
H
Hongze Cheng 已提交
557

H
Hongze Cheng 已提交
558
  ret = tdbTbcNext(pCtbCur->pCur, &pCtbCur->pKey, &pCtbCur->kLen, &pCtbCur->pVal, &pCtbCur->vLen);
H
Hongze Cheng 已提交
559 560 561
  if (ret < 0) {
    return 0;
  }
H
Hongze Cheng 已提交
562

H
Hongze Cheng 已提交
563
  pCtbIdxKey = pCtbCur->pKey;
H
Hongze Cheng 已提交
564 565 566
  if (pCtbIdxKey->suid > pCtbCur->suid) {
    return 0;
  }
H
Hongze Cheng 已提交
567

H
Hongze Cheng 已提交
568
  return pCtbIdxKey->uid;
H
Hongze Cheng 已提交
569 570
}

C
Cary Xu 已提交
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587
struct SMStbCursor {
  SMeta   *pMeta;
  TBC     *pCur;
  tb_uid_t suid;
  void    *pKey;
  void    *pVal;
  int      kLen;
  int      vLen;
};

SMStbCursor *metaOpenStbCursor(SMeta *pMeta, tb_uid_t suid) {
  SMStbCursor *pStbCur = NULL;
  int          ret = 0;
  int          c = 0;

  pStbCur = (SMStbCursor *)taosMemoryCalloc(1, sizeof(*pStbCur));
  if (pStbCur == NULL) {
C
Cary Xu 已提交
588
    terrno = TSDB_CODE_OUT_OF_MEMORY;
C
Cary Xu 已提交
589 590 591 592 593 594 595 596 597
    return NULL;
  }

  pStbCur->pMeta = pMeta;
  pStbCur->suid = suid;
  metaRLock(pMeta);

  ret = tdbTbcOpen(pMeta->pSuidIdx, &pStbCur->pCur, NULL);
  if (ret < 0) {
C
Cary Xu 已提交
598
    terrno = TSDB_CODE_OUT_OF_MEMORY;
C
Cary Xu 已提交
599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633
    metaULock(pMeta);
    taosMemoryFree(pStbCur);
    return NULL;
  }

  // move to the suid
  tdbTbcMoveTo(pStbCur->pCur, &suid, sizeof(suid), &c);
  if (c > 0) {
    tdbTbcMoveToNext(pStbCur->pCur);
  }

  return pStbCur;
}

void metaCloseStbCursor(SMStbCursor *pStbCur) {
  if (pStbCur) {
    if (pStbCur->pMeta) metaULock(pStbCur->pMeta);
    if (pStbCur->pCur) {
      tdbTbcClose(pStbCur->pCur);

      tdbFree(pStbCur->pKey);
      tdbFree(pStbCur->pVal);
    }

    taosMemoryFree(pStbCur);
  }
}

tb_uid_t metaStbCursorNext(SMStbCursor *pStbCur) {
  int ret;

  ret = tdbTbcNext(pStbCur->pCur, &pStbCur->pKey, &pStbCur->kLen, &pStbCur->pVal, &pStbCur->vLen);
  if (ret < 0) {
    return 0;
  }
H
Hongze Cheng 已提交
634
  return *(tb_uid_t *)pStbCur->pKey;
C
Cary Xu 已提交
635 636
}

637
STSchema *metaGetTbTSchema(SMeta *pMeta, tb_uid_t uid, int32_t sver, int lock) {
H
Hongze Cheng 已提交
638
  STSchema       *pTSchema = NULL;
H
Hongze Cheng 已提交
639
  SSchemaWrapper *pSW = NULL;
H
Hongze Cheng 已提交
640
  SSchema        *pSchema = NULL;
H
Hongze Cheng 已提交
641

642
  pSW = metaGetTableSchema(pMeta, uid, sver, lock);
C
Cary Xu 已提交
643
  if (!pSW) return NULL;
H
fix  
Hongze Cheng 已提交
644

H
Hongze Cheng 已提交
645
  pTSchema = tBuildTSchema(pSW->pSchema, pSW->nCols, pSW->version);
H
Hongze Cheng 已提交
646

H
Hongze Cheng 已提交
647 648
  taosMemoryFree(pSW->pSchema);
  taosMemoryFree(pSW);
H
Hongze Cheng 已提交
649 650 651
  return pTSchema;
}

H
Hongze Cheng 已提交
652
int32_t metaGetTbTSchemaEx(SMeta *pMeta, tb_uid_t suid, tb_uid_t uid, int32_t sver, STSchema **ppTSchema) {
H
Hongze Cheng 已提交
653 654
  int32_t code = 0;

H
Hongze Cheng 已提交
655 656
  void     *pData = NULL;
  int       nData = 0;
H
Hongze Cheng 已提交
657 658 659
  SSkmDbKey skmDbKey;
  if (sver <= 0) {
    SMetaInfo info;
660
    if (metaGetInfo(pMeta, suid ? suid : uid, &info, NULL) == 0) {
H
Hongze Cheng 已提交
661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678
      sver = info.skmVer;
    } else {
      TBC *pSkmDbC = NULL;
      int  c;

      skmDbKey.uid = suid ? suid : uid;
      skmDbKey.sver = INT32_MAX;

      tdbTbcOpen(pMeta->pSkmDb, &pSkmDbC, NULL);
      metaRLock(pMeta);

      if (tdbTbcMoveTo(pSkmDbC, &skmDbKey, sizeof(skmDbKey), &c) < 0) {
        metaULock(pMeta);
        tdbTbcClose(pSkmDbC);
        code = TSDB_CODE_NOT_FOUND;
        goto _exit;
      }

679
      if (c == 0) {
680 681
        metaULock(pMeta);
        tdbTbcClose(pSkmDbC);
682
        code = TSDB_CODE_FAILED;
683
        metaError("meta/query: incorrect c: %" PRId32 ".", c);
684 685
        goto _exit;
      }
H
Hongze Cheng 已提交
686 687 688 689 690 691 692 693 694

      if (c < 0) {
        tdbTbcMoveToPrev(pSkmDbC);
      }

      const void *pKey = NULL;
      int32_t     nKey = 0;
      tdbTbcGet(pSkmDbC, &pKey, &nKey, NULL, NULL);

H
Hongze Cheng 已提交
695
      if (((SSkmDbKey *)pKey)->uid != skmDbKey.uid) {
H
Hongze Cheng 已提交
696 697 698 699 700 701 702 703 704 705 706 707
        metaULock(pMeta);
        tdbTbcClose(pSkmDbC);
        code = TSDB_CODE_NOT_FOUND;
        goto _exit;
      }

      sver = ((SSkmDbKey *)pKey)->sver;

      metaULock(pMeta);
      tdbTbcClose(pSkmDbC);
    }
  }
H
Hongze Cheng 已提交
708

709
  if (ASSERTS(sver > 0, "failed to get table schema version: %d", sver)) {
H
Hongze Cheng 已提交
710
    code = TSDB_CODE_NOT_FOUND;
711 712
    goto _exit;
  }
H
Hongze Cheng 已提交
713 714 715

  skmDbKey.uid = suid ? suid : uid;
  skmDbKey.sver = sver;
H
Hongze Cheng 已提交
716
  metaRLock(pMeta);
H
Hongze Cheng 已提交
717
  if (tdbTbGet(pMeta->pSkmDb, &skmDbKey, sizeof(SSkmDbKey), &pData, &nData) < 0) {
H
Hongze Cheng 已提交
718
    metaULock(pMeta);
H
Hongze Cheng 已提交
719 720
    code = TSDB_CODE_NOT_FOUND;
    goto _exit;
H
Hongze Cheng 已提交
721 722 723 724 725 726 727 728 729
  }
  metaULock(pMeta);

  // decode
  SDecoder        dc = {0};
  SSchemaWrapper  schema;
  SSchemaWrapper *pSchemaWrapper = &schema;

  tDecoderInit(&dc, pData, nData);
M
Minglei Jin 已提交
730
  (void)tDecodeSSchemaWrapper(&dc, pSchemaWrapper);
H
Hongze Cheng 已提交
731
  tDecoderClear(&dc);
732
  tdbFree(pData);
H
Hongze Cheng 已提交
733 734

  // convert
H
Hongze Cheng 已提交
735
  STSchema *pTSchema = tBuildTSchema(pSchemaWrapper->pSchema, pSchemaWrapper->nCols, pSchemaWrapper->version);
H
Haojun Liao 已提交
736 737 738 739
  if (pTSchema == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

H
Hongze Cheng 已提交
740 741 742
  *ppTSchema = pTSchema;
  taosMemoryFree(pSchemaWrapper->pSchema);

H
Hongze Cheng 已提交
743
_exit:
H
Hongze Cheng 已提交
744 745 746
  return code;
}

747 748
// N.B. Called by statusReq per second
int64_t metaGetTbNum(SMeta *pMeta) {
749 750 751 752 753
  // num of child tables (excluding normal tables , stables and others)

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

754
  return pMeta->pVnode->config.vndStats.numOfCTables + pMeta->pVnode->config.vndStats.numOfNTables;
755 756 757
}

// N.B. Called by statusReq per second
M
Minglei Jin 已提交
758
int64_t metaGetTimeSeriesNum(SMeta *pMeta) {
759
  // sum of (number of columns of stable -  1) * number of ctables (excluding timestamp column)
760 761
  if (pMeta->pVnode->config.vndStats.numOfTimeSeries <= 0 ||
      ++pMeta->pVnode->config.vndStats.itvTimeSeries % (60 * 5) == 0) {
762 763 764 765
    int64_t num = 0;
    vnodeGetTimeSeriesNum(pMeta->pVnode, &num);
    pMeta->pVnode->config.vndStats.numOfTimeSeries = num;

766
    pMeta->pVnode->config.vndStats.itvTimeSeries = (TD_VID(pMeta->pVnode) % 100) * 2;
767
  }
768

769
  return pMeta->pVnode->config.vndStats.numOfTimeSeries + pMeta->pVnode->config.vndStats.numOfNTimeSeries;
C
Cary Xu 已提交
770
}
H
Hongze Cheng 已提交
771

772
int64_t metaGetNtbNum(SMeta *pMeta) { return pMeta->pVnode->config.vndStats.numOfNTables; }
773

C
Cary Xu 已提交
774
typedef struct {
H
Hongze Cheng 已提交
775 776
  SMeta   *pMeta;
  TBC     *pCur;
C
Cary Xu 已提交
777
  tb_uid_t uid;
H
Hongze Cheng 已提交
778 779
  void    *pKey;
  void    *pVal;
C
Cary Xu 已提交
780 781 782 783 784 785 786 787 788 789 790 791 792
  int      kLen;
  int      vLen;
} SMSmaCursor;

SMSmaCursor *metaOpenSmaCursor(SMeta *pMeta, tb_uid_t uid) {
  SMSmaCursor *pSmaCur = NULL;
  SSmaIdxKey   smaIdxKey;
  int          ret;
  int          c;

  pSmaCur = (SMSmaCursor *)taosMemoryCalloc(1, sizeof(*pSmaCur));
  if (pSmaCur == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
793 794 795
    return NULL;
  }

C
Cary Xu 已提交
796 797 798 799
  pSmaCur->pMeta = pMeta;
  pSmaCur->uid = uid;
  metaRLock(pMeta);

H
Hongze Cheng 已提交
800
  ret = tdbTbcOpen(pMeta->pSmaIdx, &pSmaCur->pCur, NULL);
C
Cary Xu 已提交
801 802 803
  if (ret < 0) {
    metaULock(pMeta);
    taosMemoryFree(pSmaCur);
H
Hongze Cheng 已提交
804 805 806
    return NULL;
  }

C
Cary Xu 已提交
807 808 809
  // move to the suid
  smaIdxKey.uid = uid;
  smaIdxKey.smaUid = INT64_MIN;
H
Hongze Cheng 已提交
810
  tdbTbcMoveTo(pSmaCur->pCur, &smaIdxKey, sizeof(smaIdxKey), &c);
C
Cary Xu 已提交
811
  if (c > 0) {
H
Hongze Cheng 已提交
812
    tdbTbcMoveToNext(pSmaCur->pCur);
C
Cary Xu 已提交
813
  }
H
Hongze Cheng 已提交
814

C
Cary Xu 已提交
815 816
  return pSmaCur;
}
H
Hongze Cheng 已提交
817

C
Cary Xu 已提交
818 819 820 821
void metaCloseSmaCursor(SMSmaCursor *pSmaCur) {
  if (pSmaCur) {
    if (pSmaCur->pMeta) metaULock(pSmaCur->pMeta);
    if (pSmaCur->pCur) {
H
Hongze Cheng 已提交
822
      tdbTbcClose(pSmaCur->pCur);
H
Hongze Cheng 已提交
823

C
Cary Xu 已提交
824 825 826
      tdbFree(pSmaCur->pKey);
      tdbFree(pSmaCur->pVal);
    }
H
Hongze Cheng 已提交
827

C
Cary Xu 已提交
828 829 830 831 832 833 834 835
    taosMemoryFree(pSmaCur);
  }
}

tb_uid_t metaSmaCursorNext(SMSmaCursor *pSmaCur) {
  int         ret;
  SSmaIdxKey *pSmaIdxKey;

H
Hongze Cheng 已提交
836
  ret = tdbTbcNext(pSmaCur->pCur, &pSmaCur->pKey, &pSmaCur->kLen, &pSmaCur->pVal, &pSmaCur->vLen);
C
Cary Xu 已提交
837 838 839 840 841 842 843 844 845 846 847 848 849 850
  if (ret < 0) {
    return 0;
  }

  pSmaIdxKey = pSmaCur->pKey;
  if (pSmaIdxKey->uid > pSmaCur->uid) {
    return 0;
  }

  return pSmaIdxKey->uid;
}

STSmaWrapper *metaGetSmaInfoByTable(SMeta *pMeta, tb_uid_t uid, bool deepCopy) {
  STSmaWrapper *pSW = NULL;
H
Hongze Cheng 已提交
851
  SArray       *pSmaIds = NULL;
C
Cary Xu 已提交
852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874

  if (!(pSmaIds = metaGetSmaIdsByTable(pMeta, uid))) {
    return NULL;
  }

  pSW = taosMemoryCalloc(1, sizeof(*pSW));
  if (!pSW) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }

  pSW->number = taosArrayGetSize(pSmaIds);
  pSW->tSma = taosMemoryCalloc(pSW->number, sizeof(STSma));

  if (!pSW->tSma) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }

  SMetaReader mr = {0};
  metaReaderInit(&mr, pMeta, 0);
  int64_t smaId;
  int     smaIdx = 0;
H
Hongze Cheng 已提交
875
  STSma  *pTSma = NULL;
C
Cary Xu 已提交
876 877 878
  for (int i = 0; i < pSW->number; ++i) {
    smaId = *(tb_uid_t *)taosArrayGet(pSmaIds, i);
    if (metaGetTableEntryByUid(&mr, smaId) < 0) {
879
      tDecoderClear(&mr.coder);
S
Shengliang Guan 已提交
880
      metaWarn("vgId:%d, no entry for tbId:%" PRIi64 ", smaId:%" PRIi64, TD_VID(pMeta->pVnode), uid, smaId);
C
Cary Xu 已提交
881 882
      continue;
    }
883
    tDecoderClear(&mr.coder);
C
Cary Xu 已提交
884 885 886 887 888 889 890 891
    pTSma = pSW->tSma + smaIdx;
    memcpy(pTSma, mr.me.smaEntry.tsma, sizeof(STSma));
    if (deepCopy) {
      if (pTSma->exprLen > 0) {
        if (!(pTSma->expr = taosMemoryCalloc(1, pTSma->exprLen))) {
          terrno = TSDB_CODE_OUT_OF_MEMORY;
          goto _err;
        }
H
Hongze Cheng 已提交
892
        memcpy((void *)pTSma->expr, mr.me.smaEntry.tsma->expr, pTSma->exprLen);
H
Hongze Cheng 已提交
893
      }
C
Cary Xu 已提交
894 895 896 897 898
      if (pTSma->tagsFilterLen > 0) {
        if (!(pTSma->tagsFilter = taosMemoryCalloc(1, pTSma->tagsFilterLen))) {
          terrno = TSDB_CODE_OUT_OF_MEMORY;
          goto _err;
        }
H
Hongze Cheng 已提交
899
      }
H
Hongze Cheng 已提交
900
      memcpy((void *)pTSma->tagsFilter, mr.me.smaEntry.tsma->tagsFilter, pTSma->tagsFilterLen);
C
Cary Xu 已提交
901 902 903 904 905
    } else {
      pTSma->exprLen = 0;
      pTSma->expr = NULL;
      pTSma->tagsFilterLen = 0;
      pTSma->tagsFilter = NULL;
H
Hongze Cheng 已提交
906
    }
H
Hongze Cheng 已提交
907

C
Cary Xu 已提交
908
    ++smaIdx;
H
Hongze Cheng 已提交
909 910
  }

C
Cary Xu 已提交
911 912
  if (smaIdx <= 0) goto _err;
  pSW->number = smaIdx;
H
Hongze Cheng 已提交
913

C
Cary Xu 已提交
914 915
  metaReaderClear(&mr);
  taosArrayDestroy(pSmaIds);
H
Hongze Cheng 已提交
916
  return pSW;
C
Cary Xu 已提交
917 918 919
_err:
  metaReaderClear(&mr);
  taosArrayDestroy(pSmaIds);
C
Cary Xu 已提交
920
  tFreeTSmaWrapper(pSW, deepCopy);
H
Hongze Cheng 已提交
921 922 923
  return NULL;
}

C
Cary Xu 已提交
924
STSma *metaGetSmaInfoByIndex(SMeta *pMeta, int64_t indexUid) {
H
Hongze Cheng 已提交
925
  STSma      *pTSma = NULL;
C
Cary Xu 已提交
926 927 928
  SMetaReader mr = {0};
  metaReaderInit(&mr, pMeta, 0);
  if (metaGetTableEntryByUid(&mr, indexUid) < 0) {
S
Shengliang Guan 已提交
929
    metaWarn("vgId:%d, failed to get table entry for smaId:%" PRIi64, TD_VID(pMeta->pVnode), indexUid);
C
Cary Xu 已提交
930 931 932 933 934 935 936 937 938 939 940 941 942 943
    metaReaderClear(&mr);
    return NULL;
  }
  pTSma = (STSma *)taosMemoryMalloc(sizeof(STSma));
  if (!pTSma) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    metaReaderClear(&mr);
    return NULL;
  }

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

  metaReaderClear(&mr);
  return pTSma;
H
Hongze Cheng 已提交
944 945
}

C
Cary Xu 已提交
946
SArray *metaGetSmaIdsByTable(SMeta *pMeta, tb_uid_t uid) {
H
Hongze Cheng 已提交
947
  SArray     *pUids = NULL;
C
Cary Xu 已提交
948
  SSmaIdxKey *pSmaIdxKey = NULL;
H
Hongze Cheng 已提交
949

C
Cary Xu 已提交
950 951
  SMSmaCursor *pCur = metaOpenSmaCursor(pMeta, uid);
  if (!pCur) {
H
Hongze Cheng 已提交
952 953 954
    return NULL;
  }

C
Cary Xu 已提交
955 956 957 958 959
  while (1) {
    tb_uid_t id = metaSmaCursorNext(pCur);
    if (id == 0) {
      break;
    }
H
Hongze Cheng 已提交
960

C
Cary Xu 已提交
961 962
    if (!pUids) {
      pUids = taosArrayInit(16, sizeof(tb_uid_t));
H
Hongze Cheng 已提交
963
      if (!pUids) {
C
Cary Xu 已提交
964 965 966
        terrno = TSDB_CODE_OUT_OF_MEMORY;
        metaCloseSmaCursor(pCur);
        return NULL;
H
Hongze Cheng 已提交
967
      }
C
Cary Xu 已提交
968
    }
H
Hongze Cheng 已提交
969

C
Cary Xu 已提交
970
    pSmaIdxKey = (SSmaIdxKey *)pCur->pKey;
H
Hongze Cheng 已提交
971

C
Cary Xu 已提交
972
    if (!taosArrayPush(pUids, &pSmaIdxKey->smaUid)) {
C
Cary Xu 已提交
973 974 975 976
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      metaCloseSmaCursor(pCur);
      taosArrayDestroy(pUids);
      return NULL;
H
Hongze Cheng 已提交
977 978 979 980 981 982 983
    }
  }

  metaCloseSmaCursor(pCur);
  return pUids;
}

C
Cary Xu 已提交
984
SArray *metaGetSmaTbUids(SMeta *pMeta) {
H
Hongze Cheng 已提交
985
  SArray     *pUids = NULL;
C
Cary Xu 已提交
986 987 988 989 990
  SSmaIdxKey *pSmaIdxKey = NULL;
  tb_uid_t    lastUid = 0;

  SMSmaCursor *pCur = metaOpenSmaCursor(pMeta, 0);
  if (!pCur) {
H
Hongze Cheng 已提交
991 992 993
    return NULL;
  }

C
Cary Xu 已提交
994 995 996 997 998
  while (1) {
    tb_uid_t uid = metaSmaCursorNext(pCur);
    if (uid == 0) {
      break;
    }
H
Hongze Cheng 已提交
999

C
Cary Xu 已提交
1000 1001 1002
    if (lastUid == uid) {
      continue;
    }
H
Hongze Cheng 已提交
1003

C
Cary Xu 已提交
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
    lastUid = uid;

    if (!pUids) {
      pUids = taosArrayInit(16, sizeof(tb_uid_t));
      if (!pUids) {
        terrno = TSDB_CODE_OUT_OF_MEMORY;
        metaCloseSmaCursor(pCur);
        return NULL;
      }
    }

C
Cary Xu 已提交
1015
    if (!taosArrayPush(pUids, &uid)) {
C
Cary Xu 已提交
1016 1017 1018 1019 1020
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      metaCloseSmaCursor(pCur);
      taosArrayDestroy(pUids);
      return NULL;
    }
H
Hongze Cheng 已提交
1021 1022
  }

C
Cary Xu 已提交
1023 1024
  metaCloseSmaCursor(pCur);
  return pUids;
H
Hongze Cheng 已提交
1025 1026
}

L
Liu Jicong 已提交
1027
#endif
H
Hongze Cheng 已提交
1028

1029
const void *metaGetTableTagVal(void *pTag, int16_t type, STagVal *val) {
dengyihao's avatar
dengyihao 已提交
1030
  STag *tag = (STag *)pTag;
dengyihao's avatar
dengyihao 已提交
1031
  if (type == TSDB_DATA_TYPE_JSON) {
wmmhello's avatar
wmmhello 已提交
1032 1033
    return tag;
  }
wmmhello's avatar
wmmhello 已提交
1034 1035
  bool find = tTagGet(tag, val);

dengyihao's avatar
dengyihao 已提交
1036
  if (!find) {
wmmhello's avatar
wmmhello 已提交
1037 1038
    return NULL;
  }
wmmhello's avatar
wmmhello 已提交
1039

wmmhello's avatar
wmmhello 已提交
1040
#ifdef TAG_FILTER_DEBUG
wmmhello's avatar
wmmhello 已提交
1041
  if (IS_VAR_DATA_TYPE(val->type)) {
1042
    char *buf = taosMemoryCalloc(val->nData + 1, 1);
wmmhello's avatar
wmmhello 已提交
1043 1044 1045 1046 1047 1048 1049 1050 1051
    memcpy(buf, val->pData, val->nData);
    metaDebug("metaTag table val varchar index:%d cid:%d type:%d value:%s", 1, val->cid, val->type, buf);
    taosMemoryFree(buf);
  } else {
    double dval = 0;
    GET_TYPED_DATA(dval, double, val->type, &val->i64);
    metaDebug("metaTag table val number index:%d cid:%d type:%d value:%f", 1, val->cid, val->type, dval);
  }

1052 1053
  SArray *pTagVals = NULL;
  tTagToValArray((STag *)pTag, &pTagVals);
wmmhello's avatar
wmmhello 已提交
1054
  for (int i = 0; i < taosArrayGetSize(pTagVals); i++) {
1055
    STagVal *pTagVal = (STagVal *)taosArrayGet(pTagVals, i);
wmmhello's avatar
wmmhello 已提交
1056 1057

    if (IS_VAR_DATA_TYPE(pTagVal->type)) {
1058
      char *buf = taosMemoryCalloc(pTagVal->nData + 1, 1);
wmmhello's avatar
wmmhello 已提交
1059 1060 1061 1062 1063 1064 1065 1066 1067
      memcpy(buf, pTagVal->pData, pTagVal->nData);
      metaDebug("metaTag table varchar index:%d cid:%d type:%d value:%s", i, pTagVal->cid, pTagVal->type, buf);
      taosMemoryFree(buf);
    } else {
      double dval = 0;
      GET_TYPED_DATA(dval, double, pTagVal->type, &pTagVal->i64);
      metaDebug("metaTag table number index:%d cid:%d type:%d value:%f", i, pTagVal->cid, pTagVal->type, dval);
    }
  }
wmmhello's avatar
wmmhello 已提交
1068
#endif
wmmhello's avatar
wmmhello 已提交
1069

wmmhello's avatar
wmmhello 已提交
1070
  return val;
dengyihao's avatar
dengyihao 已提交
1071
}
wmmhello's avatar
wmmhello 已提交
1072 1073

typedef struct {
H
Hongze Cheng 已提交
1074 1075
  SMeta   *pMeta;
  TBC     *pCur;
wmmhello's avatar
wmmhello 已提交
1076 1077 1078
  tb_uid_t suid;
  int16_t  cid;
  int16_t  type;
H
Hongze Cheng 已提交
1079 1080
  void    *pKey;
  void    *pVal;
wmmhello's avatar
wmmhello 已提交
1081 1082 1083 1084
  int32_t  kLen;
  int32_t  vLen;
} SIdxCursor;

dengyihao's avatar
dengyihao 已提交
1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
int32_t metaFilterCreateTime(SMeta *pMeta, SMetaFltParam *param, SArray *pUids) {
  int32_t ret = 0;

  SIdxCursor *pCursor = NULL;
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
  pCursor->pMeta = pMeta;
  pCursor->suid = param->suid;
  pCursor->cid = param->cid;
  pCursor->type = param->type;

  metaRLock(pMeta);
  ret = tdbTbcOpen(pMeta->pCtimeIdx, &pCursor->pCur, NULL);
  if (ret != 0) {
    goto END;
  }
  int64_t uidLimit = param->reverse ? INT64_MAX : 0;

  SCtimeIdxKey  ctimeKey = {.ctime = *(int64_t *)(param->val), .uid = uidLimit};
  SCtimeIdxKey *pCtimeKey = &ctimeKey;

  int cmp = 0;
  if (tdbTbcMoveTo(pCursor->pCur, &ctimeKey, sizeof(ctimeKey), &cmp) < 0) {
    goto END;
  }
dengyihao's avatar
dengyihao 已提交
1109

dengyihao's avatar
dengyihao 已提交
1110
  int32_t valid = 0;
dengyihao's avatar
dengyihao 已提交
1111 1112
  int32_t count = 0;

dengyihao's avatar
dengyihao 已提交
1113
  static const int8_t TRY_ERROR_LIMIT = 1;
dengyihao's avatar
dengyihao 已提交
1114
  do {
dengyihao's avatar
dengyihao 已提交
1115 1116 1117 1118 1119 1120
    void   *entryKey = NULL;
    int32_t nEntryKey = -1;
    valid = tdbTbcGet(pCursor->pCur, (const void **)&entryKey, &nEntryKey, NULL, NULL);
    if (valid < 0) break;

    SCtimeIdxKey *p = entryKey;
dengyihao's avatar
dengyihao 已提交
1121
    if (count > TRY_ERROR_LIMIT) break;
dengyihao's avatar
dengyihao 已提交
1122

dengyihao's avatar
dengyihao 已提交
1123
    int32_t cmp = (*param->filterFunc)((void *)&p->ctime, (void *)&pCtimeKey->ctime, param->type);
dengyihao's avatar
dengyihao 已提交
1124 1125 1126
    if (cmp == 0)
      taosArrayPush(pUids, &p->uid);
    else {
dengyihao's avatar
dengyihao 已提交
1127 1128 1129 1130
      if (param->equal == true) {
        if (count > TRY_ERROR_LIMIT) break;
        count++;
      }
dengyihao's avatar
dengyihao 已提交
1131
    }
dengyihao's avatar
dengyihao 已提交
1132 1133
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
    if (valid < 0) break;
dengyihao's avatar
dengyihao 已提交
1134
  } while (1);
dengyihao's avatar
dengyihao 已提交
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156

END:
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
  taosMemoryFree(pCursor);
  return ret;
}

int32_t metaFilterTableName(SMeta *pMeta, SMetaFltParam *param, SArray *pUids) {
  int32_t ret = 0;
  char   *buf = NULL;

  STagIdxKey *pKey = NULL;
  int32_t     nKey = 0;

  SIdxCursor *pCursor = NULL;
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
  pCursor->pMeta = pMeta;
  pCursor->suid = param->suid;
  pCursor->cid = param->cid;
  pCursor->type = param->type;

dengyihao's avatar
dengyihao 已提交
1157 1158
  char *pName = param->val;

dengyihao's avatar
dengyihao 已提交
1159 1160
  metaRLock(pMeta);
  ret = tdbTbcOpen(pMeta->pNameIdx, &pCursor->pCur, NULL);
dengyihao's avatar
dengyihao 已提交
1161 1162 1163 1164 1165 1166 1167 1168 1169
  if (ret != 0) {
    goto END;
  }

  int cmp = 0;
  if (tdbTbcMoveTo(pCursor->pCur, pName, strlen(pName) + 1, &cmp) < 0) {
    goto END;
  }
  int32_t valid = 0;
dengyihao's avatar
dengyihao 已提交
1170 1171
  int32_t count = 0;

dengyihao's avatar
dengyihao 已提交
1172
  int32_t TRY_ERROR_LIMIT = 1;
dengyihao's avatar
dengyihao 已提交
1173
  do {
dengyihao's avatar
dengyihao 已提交
1174 1175 1176 1177 1178
    void   *pEntryKey = NULL, *pEntryVal = NULL;
    int32_t nEntryKey = -1, nEntryVal = 0;
    valid = tdbTbcGet(pCursor->pCur, (const void **)pEntryKey, &nEntryKey, (const void **)&pEntryVal, &nEntryVal);
    if (valid < 0) break;

dengyihao's avatar
dengyihao 已提交
1179 1180
    if (count > TRY_ERROR_LIMIT) break;

1181
    char *pTableKey = (char *)pEntryKey;
H
Haojun Liao 已提交
1182
    cmp = (*param->filterFunc)(pTableKey, pName, pCursor->type);
dengyihao's avatar
dengyihao 已提交
1183 1184 1185 1186
    if (cmp == 0) {
      tb_uid_t tuid = *(tb_uid_t *)pEntryVal;
      taosArrayPush(pUids, &tuid);
    } else {
dengyihao's avatar
dengyihao 已提交
1187 1188 1189 1190
      if (param->equal == true) {
        if (count > TRY_ERROR_LIMIT) break;
        count++;
      }
dengyihao's avatar
dengyihao 已提交
1191 1192 1193 1194 1195
    }
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
    if (valid < 0) {
      break;
    }
dengyihao's avatar
dengyihao 已提交
1196
  } while (1);
dengyihao's avatar
dengyihao 已提交
1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236

END:
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
  taosMemoryFree(buf);
  taosMemoryFree(pKey);

  taosMemoryFree(pCursor);

  return ret;
}
int32_t metaFilterTtl(SMeta *pMeta, SMetaFltParam *param, SArray *pUids) {
  int32_t ret = 0;
  char   *buf = NULL;

  STtlIdxKey *pKey = NULL;
  int32_t     nKey = 0;

  SIdxCursor *pCursor = NULL;
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
  pCursor->pMeta = pMeta;
  pCursor->suid = param->suid;
  pCursor->cid = param->cid;
  pCursor->type = param->type;

  metaRLock(pMeta);
  ret = tdbTbcOpen(pMeta->pTtlIdx, &pCursor->pCur, NULL);

END:
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
  taosMemoryFree(buf);
  taosMemoryFree(pKey);

  taosMemoryFree(pCursor);

  return ret;
  // impl later
  return 0;
}
dengyihao's avatar
dengyihao 已提交
1237 1238 1239
int32_t metaFilterTableIds(SMeta *pMeta, SMetaFltParam *param, SArray *pUids) {
  int32_t ret = 0;
  char   *buf = NULL;
wmmhello's avatar
wmmhello 已提交
1240

dengyihao's avatar
dengyihao 已提交
1241 1242 1243 1244
  STagIdxKey *pKey = NULL;
  int32_t     nKey = 0;

  SIdxCursor *pCursor = NULL;
wmmhello's avatar
wmmhello 已提交
1245 1246 1247 1248 1249 1250 1251
  pCursor = (SIdxCursor *)taosMemoryCalloc(1, sizeof(SIdxCursor));
  pCursor->pMeta = pMeta;
  pCursor->suid = param->suid;
  pCursor->cid = param->cid;
  pCursor->type = param->type;

  metaRLock(pMeta);
dengyihao's avatar
dengyihao 已提交
1252
  ret = tdbTbcOpen(pMeta->pTagIdx, &pCursor->pCur, NULL);
wmmhello's avatar
wmmhello 已提交
1253 1254 1255 1256
  if (ret < 0) {
    goto END;
  }

dengyihao's avatar
dengyihao 已提交
1257
  int32_t maxSize = 0;
wmmhello's avatar
wmmhello 已提交
1258
  int32_t nTagData = 0;
H
Hongze Cheng 已提交
1259
  void   *tagData = NULL;
dengyihao's avatar
dengyihao 已提交
1260

dengyihao's avatar
dengyihao 已提交
1261
  if (param->val == NULL) {
S
Shengliang Guan 已提交
1262
    metaError("vgId:%d, failed to filter NULL data", TD_VID(pMeta->pVnode));
dengyihao's avatar
dengyihao 已提交
1263
    ret = -1;
dengyihao's avatar
dengyihao 已提交
1264
    goto END;
dengyihao's avatar
dengyihao 已提交
1265
  } else {
dengyihao's avatar
dengyihao 已提交
1266 1267 1268 1269 1270 1271 1272 1273 1274 1275
    if (IS_VAR_DATA_TYPE(param->type)) {
      tagData = varDataVal(param->val);
      nTagData = varDataLen(param->val);

      if (param->type == TSDB_DATA_TYPE_NCHAR) {
        maxSize = 4 * nTagData + 1;
        buf = taosMemoryCalloc(1, maxSize);
        if (false == taosMbsToUcs4(tagData, nTagData, (TdUcs4 *)buf, maxSize, &maxSize)) {
          goto END;
        }
dengyihao's avatar
dengyihao 已提交
1276

dengyihao's avatar
dengyihao 已提交
1277 1278 1279 1280 1281 1282
        tagData = buf;
        nTagData = maxSize;
      }
    } else {
      tagData = param->val;
      nTagData = tDataTypes[param->type].bytes;
dengyihao's avatar
dengyihao 已提交
1283
    }
wmmhello's avatar
wmmhello 已提交
1284
  }
dengyihao's avatar
dengyihao 已提交
1285
  ret = metaCreateTagIdxKey(pCursor->suid, pCursor->cid, tagData, nTagData, pCursor->type,
wmmhello's avatar
wmmhello 已提交
1286
                            param->reverse ? INT64_MAX : INT64_MIN, &pKey, &nKey);
dengyihao's avatar
dengyihao 已提交
1287

wmmhello's avatar
wmmhello 已提交
1288 1289 1290 1291 1292 1293 1294
  if (ret != 0) {
    goto END;
  }
  int cmp = 0;
  if (tdbTbcMoveTo(pCursor->pCur, pKey, nKey, &cmp) < 0) {
    goto END;
  }
dengyihao's avatar
dengyihao 已提交
1295

dengyihao's avatar
dengyihao 已提交
1296
  int     count = 0;
dengyihao's avatar
dengyihao 已提交
1297
  int32_t valid = 0;
dengyihao's avatar
dengyihao 已提交
1298
  bool    found = false;
dengyihao's avatar
dengyihao 已提交
1299

dengyihao's avatar
dengyihao 已提交
1300
  static const int8_t TRY_ERROR_LIMIT = 1;
dengyihao's avatar
dengyihao 已提交
1301 1302 1303

  /// src:   [[suid, cid1, type1]....[suid, cid2, type2]....[suid, cid3, type3]...]
  /// target:                        [suid, cid2, type2]
1304
  int diffCidCount = 0;
dengyihao's avatar
dengyihao 已提交
1305
  do {
dengyihao's avatar
dengyihao 已提交
1306 1307 1308
    void   *entryKey = NULL, *entryVal = NULL;
    int32_t nEntryKey, nEntryVal;

wmmhello's avatar
wmmhello 已提交
1309 1310 1311 1312
    valid = tdbTbcGet(pCursor->pCur, (const void **)&entryKey, &nEntryKey, (const void **)&entryVal, &nEntryVal);
    if (valid < 0) {
      break;
    }
dengyihao's avatar
dengyihao 已提交
1313
    if (count > TRY_ERROR_LIMIT) {
dengyihao's avatar
dengyihao 已提交
1314 1315 1316
      break;
    }

wmmhello's avatar
wmmhello 已提交
1317
    STagIdxKey *p = entryKey;
dengyihao's avatar
dengyihao 已提交
1318
    if (p == NULL) break;
dengyihao's avatar
dengyihao 已提交
1319

dengyihao's avatar
dengyihao 已提交
1320
    if (p->type != pCursor->type || p->suid != pCursor->suid || p->cid != pCursor->cid) {
1321 1322 1323
      if (found == true) break;  //
      if (diffCidCount > TRY_ERROR_LIMIT) break;
      diffCidCount++;
dengyihao's avatar
dengyihao 已提交
1324 1325 1326
      count++;
      valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
      if (valid < 0) {
dengyihao's avatar
dengyihao 已提交
1327
        break;
dengyihao's avatar
dengyihao 已提交
1328 1329
      } else {
        continue;
dengyihao's avatar
dengyihao 已提交
1330 1331
      }
    }
dengyihao's avatar
dengyihao 已提交
1332

dengyihao's avatar
dengyihao 已提交
1333 1334 1335 1336 1337 1338
    int32_t cmp = (*param->filterFunc)(p->data, pKey->data, pKey->type);
    if (cmp == 0) {
      // match
      tb_uid_t tuid = 0;
      if (IS_VAR_DATA_TYPE(pKey->type)) {
        tuid = *(tb_uid_t *)(p->data + varDataTLen(p->data));
wmmhello's avatar
wmmhello 已提交
1339
      } else {
dengyihao's avatar
dengyihao 已提交
1340
        tuid = *(tb_uid_t *)(p->data + tDataTypes[pCursor->type].bytes);
wmmhello's avatar
wmmhello 已提交
1341
      }
dengyihao's avatar
dengyihao 已提交
1342
      taosArrayPush(pUids, &tuid);
dengyihao's avatar
dengyihao 已提交
1343
      found = true;
dengyihao's avatar
dengyihao 已提交
1344
    } else {
dengyihao's avatar
dengyihao 已提交
1345 1346 1347 1348
      if (param->equal == true) {
        if (count > TRY_ERROR_LIMIT) break;
        count++;
      }
wmmhello's avatar
wmmhello 已提交
1349 1350 1351 1352 1353
    }
    valid = param->reverse ? tdbTbcMoveToPrev(pCursor->pCur) : tdbTbcMoveToNext(pCursor->pCur);
    if (valid < 0) {
      break;
    }
dengyihao's avatar
dengyihao 已提交
1354
  } while (1);
dengyihao's avatar
dengyihao 已提交
1355

wmmhello's avatar
wmmhello 已提交
1356 1357 1358
END:
  if (pCursor->pMeta) metaULock(pCursor->pMeta);
  if (pCursor->pCur) tdbTbcClose(pCursor->pCur);
dengyihao's avatar
dengyihao 已提交
1359
  taosMemoryFree(buf);
dengyihao's avatar
dengyihao 已提交
1360
  taosMemoryFree(pKey);
wmmhello's avatar
wmmhello 已提交
1361 1362 1363 1364

  taosMemoryFree(pCursor);

  return ret;
dengyihao's avatar
dengyihao 已提交
1365
}
H
Hongze Cheng 已提交
1366

dengyihao's avatar
dengyihao 已提交
1367
static int32_t metaGetTableTagByUid(SMeta *pMeta, int64_t suid, int64_t uid, void **tag, int32_t *len, bool lock) {
dengyihao's avatar
dengyihao 已提交
1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
  int ret = 0;
  if (lock) {
    metaRLock(pMeta);
  }

  SCtbIdxKey ctbIdxKey = {.suid = suid, .uid = uid};
  ret = tdbTbGet(pMeta->pCtbIdx, &ctbIdxKey, sizeof(SCtbIdxKey), tag, len);
  if (lock) {
    metaULock(pMeta);
  }

  return ret;
}
H
Haojun Liao 已提交
1381

1382
int32_t metaGetTableTagsByUids(SMeta *pMeta, int64_t suid, SArray *uidList) {
dengyihao's avatar
dengyihao 已提交
1383 1384 1385
  const int32_t LIMIT = 128;

  int32_t isLock = false;
dengyihao's avatar
dengyihao 已提交
1386 1387
  int32_t sz = uidList ? taosArrayGetSize(uidList) : 0;
  for (int i = 0; i < sz; i++) {
1388
    STUidTagInfo *p = taosArrayGet(uidList, i);
dengyihao's avatar
dengyihao 已提交
1389 1390 1391 1392 1393 1394 1395 1396

    if (i % LIMIT == 0) {
      if (isLock) metaULock(pMeta);

      metaRLock(pMeta);
      isLock = true;
    }

X
Xiaoyu Wang 已提交
1397 1398 1399 1400 1401 1402 1403 1404 1405 1406
    //    if (taosHashGet(tags, &p->uid, sizeof(tb_uid_t)) == NULL) {
    void   *val = NULL;
    int32_t len = 0;
    if (metaGetTableTagByUid(pMeta, suid, p->uid, &val, &len, false) == 0) {
      p->pTagVal = taosMemoryMalloc(len);
      memcpy(p->pTagVal, val, len);
      tdbFree(val);
    } else {
      metaError("vgId:%d, failed to table tags, suid: %" PRId64 ", uid: %" PRId64 "", TD_VID(pMeta->pVnode), suid,
                p->uid);
dengyihao's avatar
dengyihao 已提交
1407 1408
    }
  }
X
Xiaoyu Wang 已提交
1409
  //  }
dengyihao's avatar
dengyihao 已提交
1410
  if (isLock) metaULock(pMeta);
dengyihao's avatar
dengyihao 已提交
1411 1412
  return 0;
}
dengyihao's avatar
dengyihao 已提交
1413

1414
int32_t metaGetTableTags(SMeta *pMeta, uint64_t suid, SArray *pUidTagInfo) {
1415
  SMCtbCursor *pCur = metaOpenCtbCursor(pMeta, suid, 1);
1416

1417 1418 1419 1420
  // If len > 0 means there already have uids, and we only want the
  // tags of the specified tables, of which uid in the uid list. Otherwise, all table tags are retrieved and kept
  // in the hash map, that may require a lot of memory
  SHashObj *pSepecifiedUidMap = NULL;
1421 1422
  size_t    numOfElems = taosArrayGetSize(pUidTagInfo);
  if (numOfElems > 0) {
X
Xiaoyu Wang 已提交
1423 1424
    pSepecifiedUidMap =
        taosHashInit(numOfElems / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
1425
    for (int i = 0; i < numOfElems; i++) {
H
Haojun Liao 已提交
1426 1427
      STUidTagInfo *pTagInfo = taosArrayGet(pUidTagInfo, i);
      taosHashPut(pSepecifiedUidMap, &pTagInfo->uid, sizeof(uint64_t), &i, sizeof(int32_t));
1428 1429 1430
    }
  }

H
Haojun Liao 已提交
1431 1432 1433 1434 1435 1436
  if (numOfElems == 0) {  // all data needs to be added into the pUidTagInfo list
    while (1) {
      tb_uid_t uid = metaCtbCursorNext(pCur);
      if (uid == 0) {
        break;
      }
1437

1438 1439 1440 1441
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
      taosArrayPush(pUidTagInfo, &info);
1442
    }
H
Haojun Liao 已提交
1443 1444 1445 1446 1447 1448
  } else {  // only the specified tables need to be added
    while (1) {
      tb_uid_t uid = metaCtbCursorNext(pCur);
      if (uid == 0) {
        break;
      }
1449

H
Haojun Liao 已提交
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460
      int32_t *index = taosHashGet(pSepecifiedUidMap, &uid, sizeof(uint64_t));
      if (index == NULL) {
        continue;
      }

      STUidTagInfo *pTagInfo = taosArrayGet(pUidTagInfo, *index);
      if (pTagInfo->pTagVal == NULL) {
        pTagInfo->pTagVal = taosMemoryMalloc(pCur->vLen);
        memcpy(pTagInfo->pTagVal, pCur->pVal, pCur->vLen);
      }
    }
1461 1462
  }

1463
  taosHashCleanup(pSepecifiedUidMap);
1464
  metaCloseCtbCursor(pCur, 1);
1465 1466
  return TSDB_CODE_SUCCESS;
}
wmmhello's avatar
wmmhello 已提交
1467

H
Hongze Cheng 已提交
1468 1469
int32_t metaCacheGet(SMeta *pMeta, int64_t uid, SMetaInfo *pInfo);

1470
int32_t metaGetInfo(SMeta *pMeta, int64_t uid, SMetaInfo *pInfo, SMetaReader *pReader) {
H
Hongze Cheng 已提交
1471 1472 1473
  int32_t code = 0;
  void   *pData = NULL;
  int     nData = 0;
1474
  int     lock = 0;
H
Hongze Cheng 已提交
1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498

  metaRLock(pMeta);

  // search cache
  if (metaCacheGet(pMeta, uid, pInfo) == 0) {
    metaULock(pMeta);
    goto _exit;
  }

  // search TDB
  if (tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData) < 0) {
    // not found
    metaULock(pMeta);
    code = TSDB_CODE_NOT_FOUND;
    goto _exit;
  }

  metaULock(pMeta);

  pInfo->uid = uid;
  pInfo->suid = ((SUidIdxVal *)pData)->suid;
  pInfo->version = ((SUidIdxVal *)pData)->version;
  pInfo->skmVer = ((SUidIdxVal *)pData)->skmVer;

1499 1500 1501 1502 1503 1504 1505
  if (pReader != NULL) {
    lock = !(pReader->flags & META_READER_NOLOCK);
    if (lock) {
      metaULock(pReader->pMeta);
      // metaReaderReleaseLock(pReader);
    }
  }
H
Hongze Cheng 已提交
1506 1507 1508 1509 1510
  // upsert the cache
  metaWLock(pMeta);
  metaCacheUpsert(pMeta, pInfo);
  metaULock(pMeta);

1511 1512 1513 1514
  if (lock) {
    metaRLock(pReader->pMeta);
  }

H
Hongze Cheng 已提交
1515 1516 1517 1518
_exit:
  tdbFree(pData);
  return code;
}
1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547

int32_t metaGetStbStats(SMeta *pMeta, int64_t uid, SMetaStbStats *pInfo) {
  int32_t code = 0;

  metaRLock(pMeta);

  // fast path: search cache
  if (metaStatsCacheGet(pMeta, uid, pInfo) == TSDB_CODE_SUCCESS) {
    metaULock(pMeta);
    goto _exit;
  }

  // slow path: search TDB
  int64_t ctbNum = 0;
  vnodeGetCtbNum(pMeta->pVnode, uid, &ctbNum);

  metaULock(pMeta);

  pInfo->uid = uid;
  pInfo->ctbNum = ctbNum;

  // upsert the cache
  metaWLock(pMeta);
  metaStatsCacheUpsert(pMeta, pInfo);
  metaULock(pMeta);

_exit:
  return code;
}
1548 1549 1550 1551 1552 1553 1554 1555 1556 1557

void metaUpdateStbStats(SMeta *pMeta, int64_t uid, int64_t delta) {
  SMetaStbStats stats = {0};

  if (metaStatsCacheGet(pMeta, uid, &stats) == TSDB_CODE_SUCCESS) {
    stats.ctbNum += delta;

    metaStatsCacheUpsert(pMeta, &stats);
  }
}