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

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

dengyihao's avatar
dengyihao 已提交
18
static int metaSaveJsonVarToIdx(SMeta *pMeta, const SMetaEntry *pCtbEntry, const SSchema *pSchema);
H
Hongze Cheng 已提交
19 20 21 22 23 24 25
static int metaHandleEntry(SMeta *pMeta, const SMetaEntry *pME);
static int metaSaveToTbDb(SMeta *pMeta, const SMetaEntry *pME);
static int metaUpdateUidIdx(SMeta *pMeta, const SMetaEntry *pME);
static int metaUpdateNameIdx(SMeta *pMeta, const SMetaEntry *pME);
static int metaUpdateTtlIdx(SMeta *pMeta, const SMetaEntry *pME);
static int metaSaveToSkmDb(SMeta *pMeta, const SMetaEntry *pME);
static int metaUpdateCtbIdx(SMeta *pMeta, const SMetaEntry *pME);
H
Hongze Cheng 已提交
26
static int metaUpdateTagIdx(SMeta *pMeta, const SMetaEntry *pCtbEntry);
H
Hongze Cheng 已提交
27
static int metaDropTableByUid(SMeta *pMeta, tb_uid_t uid, int *type);
H
Hongze Cheng 已提交
28

dengyihao's avatar
dengyihao 已提交
29
static int metaUpdateMetaRsp(tb_uid_t uid, char *tbName, SSchemaWrapper *pSchema, STableMetaRsp *pMetaRsp) {
D
dapan1121 已提交
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
  pMetaRsp->pSchemas = taosMemoryMalloc(pSchema->nCols * sizeof(SSchema));
  if (NULL == pMetaRsp->pSchemas) {
    terrno = TSDB_CODE_VND_OUT_OF_MEMORY;
    return -1;
  }

  strcpy(pMetaRsp->tbName, tbName);
  pMetaRsp->numOfColumns = pSchema->nCols;
  pMetaRsp->tableType = TSDB_NORMAL_TABLE;
  pMetaRsp->sversion = pSchema->version;
  pMetaRsp->tuid = uid;

  memcpy(pMetaRsp->pSchemas, pSchema->pSchema, pSchema->nCols * sizeof(SSchema));

  return 0;
}

dengyihao's avatar
dengyihao 已提交
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
static int metaSaveJsonVarToIdx(SMeta *pMeta, const SMetaEntry *pCtbEntry, const SSchema *pSchema) {
#ifdef USE_INVERTED_INDEX
  if (pMeta->pTagIvtIdx == NULL || pCtbEntry == NULL) {
    return -1;
  }
  void *      data = pCtbEntry->ctbEntry.pTags;
  const char *tagName = pSchema->name;

  tb_uid_t    suid = pCtbEntry->ctbEntry.suid;
  tb_uid_t    tuid = pCtbEntry->uid;
  const void *pTagData = pCtbEntry->ctbEntry.pTags;
  int32_t     nTagData = 0;

  SArray *pTagVals = NULL;
  if (tTagToValArray((const STag *)data, &pTagVals) != 0) {
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
64

dengyihao's avatar
dengyihao 已提交
65 66 67 68 69
  SIndexMultiTerm *terms = indexMultiTermCreate();
  int16_t          nCols = taosArrayGetSize(pTagVals);
  for (int i = 0; i < nCols; i++) {
    STagVal *pTagVal = (STagVal *)taosArrayGet(pTagVals, i);
    char     type = pTagVal->type;
dengyihao's avatar
dengyihao 已提交
70 71

    char *  key = pTagVal->pKey;
dengyihao's avatar
dengyihao 已提交
72
    int32_t nKey = strlen(key);
dengyihao's avatar
dengyihao 已提交
73 74 75

    SIndexTerm *term = NULL;
    if (type == TSDB_DATA_TYPE_NULL) {
dengyihao's avatar
dengyihao 已提交
76
      // handle null value
dengyihao's avatar
dengyihao 已提交
77 78
    } else if (type == TSDB_DATA_TYPE_NCHAR) {
      if (pTagVal->nData > 0) {
dengyihao's avatar
dengyihao 已提交
79 80 81
        char *  val = taosMemoryCalloc(1, pTagVal->nData + VARSTR_HEADER_SIZE);
        int32_t len = taosUcs4ToMbs((TdUcs4 *)pTagVal->pData, pTagVal->nData, val + VARSTR_HEADER_SIZE);
        memcpy(val, (uint16_t *)&len, VARSTR_HEADER_SIZE);
dengyihao's avatar
dengyihao 已提交
82
        type = TSDB_DATA_TYPE_VARCHAR;
dengyihao's avatar
dengyihao 已提交
83
        term = indexTermCreate(suid, ADD_VALUE, type, key, nKey, val, len);
dengyihao's avatar
dengyihao 已提交
84 85 86 87 88 89 90 91 92 93 94 95
      } else if (pTagVal->nData == 0) {
        char *  val = NULL;
        int32_t len = 0;
        // handle NULL key
      }
    } else if (type == TSDB_DATA_TYPE_DOUBLE) {
      double val = *(double *)(&pTagVal->i64);
      int    len = 0;
      term = indexTermCreate(suid, ADD_VALUE, type, key, nKey, (const char *)&val, len);
    } else if (type == TSDB_DATA_TYPE_BOOL) {
      int val = *(int *)(&pTagVal->i64);
      int len = 0;
dengyihao's avatar
dengyihao 已提交
96
      term = indexTermCreate(suid, ADD_VALUE, TSDB_DATA_TYPE_INT, key, nKey, (const char *)&val, len);
dengyihao's avatar
dengyihao 已提交
97
    }
dengyihao's avatar
dengyihao 已提交
98
    if (term != NULL) {
dengyihao's avatar
dengyihao 已提交
99 100 101
      indexMultiTermAdd(terms, term);
    }
  }
dengyihao's avatar
dengyihao 已提交
102
  indexJsonPut(pMeta->pTagIvtIdx, terms, tuid);
dengyihao's avatar
dengyihao 已提交
103 104
  indexMultiTermDestroy(terms);
#endif
dengyihao's avatar
dengyihao 已提交
105
  return 0;
dengyihao's avatar
dengyihao 已提交
106 107
}

H
Hongze Cheng 已提交
108
int metaCreateSTable(SMeta *pMeta, int64_t version, SVCreateStbReq *pReq) {
H
Hongze Cheng 已提交
109
  SMetaEntry  me = {0};
H
Hongze Cheng 已提交
110 111 112 113
  int         kLen = 0;
  int         vLen = 0;
  const void *pKey = NULL;
  const void *pVal = NULL;
dengyihao's avatar
dengyihao 已提交
114
  void *      pBuf = NULL;
H
Hongze Cheng 已提交
115
  int32_t     szBuf = 0;
dengyihao's avatar
dengyihao 已提交
116
  void *      p = NULL;
H
Hongze Cheng 已提交
117 118 119
  SMetaReader mr = {0};

  // validate req
H
Hongze Cheng 已提交
120
  metaReaderInit(&mr, pMeta, 0);
H
Hongze Cheng 已提交
121
  if (metaGetTableEntryByName(&mr, pReq->name) == 0) {
H
Hongze Cheng 已提交
122 123
// TODO: just for pass case
#if 0
H
Hongze Cheng 已提交
124 125 126
    terrno = TSDB_CODE_TDB_TABLE_ALREADY_EXIST;
    metaReaderClear(&mr);
    return -1;
H
Hongze Cheng 已提交
127 128 129 130
#else
    metaReaderClear(&mr);
    return 0;
#endif
H
Hongze Cheng 已提交
131
  }
H
Hongze Cheng 已提交
132
  metaReaderClear(&mr);
H
Hongze Cheng 已提交
133 134

  // set structs
H
Hongze Cheng 已提交
135
  me.version = version;
H
Hongze Cheng 已提交
136 137 138
  me.type = TSDB_SUPER_TABLE;
  me.uid = pReq->suid;
  me.name = pReq->name;
139
  me.stbEntry.schemaRow = pReq->schemaRow;
H
Hongze Cheng 已提交
140
  me.stbEntry.schemaTag = pReq->schemaTag;
H
Hongze Cheng 已提交
141

H
Hongze Cheng 已提交
142
  if (metaHandleEntry(pMeta, &me) < 0) goto _err;
H
Hongze Cheng 已提交
143

S
Shengliang Guan 已提交
144
  metaDebug("vgId:%d, super table is created, name:%s uid: %" PRId64, TD_VID(pMeta->pVnode), pReq->name, pReq->suid);
H
Hongze Cheng 已提交
145 146 147 148

  return 0;

_err:
S
Shengliang Guan 已提交
149
  metaError("vgId:%d, failed to create super table: %s uid: %" PRId64 " since %s", TD_VID(pMeta->pVnode), pReq->name,
H
Hongze Cheng 已提交
150 151 152 153 154
            pReq->suid, tstrerror(terrno));
  return -1;
}

int metaDropSTable(SMeta *pMeta, int64_t verison, SVDropStbReq *pReq) {
H
Hongze Cheng 已提交
155 156 157 158 159 160 161 162 163 164 165 166
  void *pKey = NULL;
  int   nKey = 0;
  void *pData = NULL;
  int   nData = 0;
  int   c = 0;
  int   rc = 0;

  // check if super table exists
  rc = tdbTbGet(pMeta->pNameIdx, pReq->name, strlen(pReq->name) + 1, &pData, &nData);
  if (rc < 0 || *(tb_uid_t *)pData != pReq->suid) {
    terrno = TSDB_CODE_VND_TABLE_NOT_EXIST;
    return -1;
H
Hongze Cheng 已提交
167 168
  }

H
Hongze Cheng 已提交
169
  // drop all child tables
dengyihao's avatar
dengyihao 已提交
170
  TBC *   pCtbIdxc = NULL;
H
Hongze Cheng 已提交
171
  SArray *pArray = taosArrayInit(8, sizeof(tb_uid_t));
H
Hongze Cheng 已提交
172

H
Hongze Cheng 已提交
173
  tdbTbcOpen(pMeta->pCtbIdx, &pCtbIdxc, &pMeta->txn);
H
Hongze Cheng 已提交
174 175
  rc = tdbTbcMoveTo(pCtbIdxc, &(SCtbIdxKey){.suid = pReq->suid, .uid = INT64_MIN}, sizeof(SCtbIdxKey), &c);
  if (rc < 0) {
H
Hongze Cheng 已提交
176
    tdbTbcClose(pCtbIdxc);
H
Hongze Cheng 已提交
177 178
    metaWLock(pMeta);
    goto _drop_super_table;
H
Hongze Cheng 已提交
179 180 181
  }

  for (;;) {
H
Hongze Cheng 已提交
182 183
    rc = tdbTbcNext(pCtbIdxc, &pKey, &nKey, NULL, NULL);
    if (rc < 0) break;
H
Hongze Cheng 已提交
184

H
Hongze Cheng 已提交
185 186 187 188 189
    if (((SCtbIdxKey *)pKey)->suid < pReq->suid) {
      continue;
    } else if (((SCtbIdxKey *)pKey)->suid > pReq->suid) {
      break;
    }
H
Hongze Cheng 已提交
190

H
Hongze Cheng 已提交
191 192 193 194 195 196
    taosArrayPush(pArray, &(((SCtbIdxKey *)pKey)->uid));
  }

  tdbTbcClose(pCtbIdxc);

  metaWLock(pMeta);
H
Hongze Cheng 已提交
197

H
Hongze Cheng 已提交
198 199
  for (int32_t iChild = 0; iChild < taosArrayGetSize(pArray); iChild++) {
    tb_uid_t uid = *(tb_uid_t *)taosArrayGet(pArray, iChild);
H
Hongze Cheng 已提交
200
    metaDropTableByUid(pMeta, uid, NULL);
H
Hongze Cheng 已提交
201 202
  }

H
Hongze Cheng 已提交
203 204 205 206 207 208 209 210 211 212 213 214
  taosArrayDestroy(pArray);

  // drop super table
_drop_super_table:
  tdbTbGet(pMeta->pUidIdx, &pReq->suid, sizeof(tb_uid_t), &pData, &nData);
  tdbTbDelete(pMeta->pTbDb, &(STbDbKey){.version = *(int64_t *)pData, .uid = pReq->suid}, sizeof(STbDbKey),
              &pMeta->txn);
  tdbTbDelete(pMeta->pNameIdx, pReq->name, strlen(pReq->name) + 1, &pMeta->txn);
  tdbTbDelete(pMeta->pUidIdx, &pReq->suid, sizeof(tb_uid_t), &pMeta->txn);

  metaULock(pMeta);

H
Hongze Cheng 已提交
215
_exit:
H
Hongze Cheng 已提交
216 217
  tdbFree(pKey);
  tdbFree(pData);
S
Shengliang Guan 已提交
218
  metaDebug("vgId:%d,  super table %s uid:%" PRId64 " is dropped", TD_VID(pMeta->pVnode), pReq->name, pReq->suid);
H
Hongze Cheng 已提交
219 220 221
  return 0;
}

H
Hongze Cheng 已提交
222 223 224
int metaAlterSTable(SMeta *pMeta, int64_t version, SVCreateStbReq *pReq) {
  SMetaEntry  oStbEntry = {0};
  SMetaEntry  nStbEntry = {0};
dengyihao's avatar
dengyihao 已提交
225 226
  TBC *       pUidIdxc = NULL;
  TBC *       pTbDbc = NULL;
H
Hongze Cheng 已提交
227 228 229 230 231 232 233
  const void *pData;
  int         nData;
  int64_t     oversion;
  SDecoder    dc = {0};
  int32_t     ret;
  int32_t     c;

H
Hongze Cheng 已提交
234 235
  tdbTbcOpen(pMeta->pUidIdx, &pUidIdxc, &pMeta->txn);
  ret = tdbTbcMoveTo(pUidIdxc, &pReq->suid, sizeof(tb_uid_t), &c);
H
Hongze Cheng 已提交
236 237 238 239 240
  if (ret < 0 || c) {
    ASSERT(0);
    return -1;
  }

H
Hongze Cheng 已提交
241
  ret = tdbTbcGet(pUidIdxc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
242 243 244 245 246 247 248
  if (ret < 0) {
    ASSERT(0);
    return -1;
  }

  oversion = *(int64_t *)pData;

H
Hongze Cheng 已提交
249 250
  tdbTbcOpen(pMeta->pTbDb, &pTbDbc, &pMeta->txn);
  ret = tdbTbcMoveTo(pTbDbc, &((STbDbKey){.uid = pReq->suid, .version = oversion}), sizeof(STbDbKey), &c);
H
Hongze Cheng 已提交
251 252
  ASSERT(ret == 0 && c == 0);

H
Hongze Cheng 已提交
253
  ret = tdbTbcGet(pTbDbc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
254 255
  ASSERT(ret == 0);

H
Hongze Cheng 已提交
256 257 258
  oStbEntry.pBuf = taosMemoryMalloc(nData);
  memcpy(oStbEntry.pBuf, pData, nData);
  tDecoderInit(&dc, oStbEntry.pBuf, nData);
H
Hongze Cheng 已提交
259 260 261 262 263 264
  metaDecodeEntry(&dc, &oStbEntry);

  nStbEntry.version = version;
  nStbEntry.type = TSDB_SUPER_TABLE;
  nStbEntry.uid = pReq->suid;
  nStbEntry.name = pReq->name;
265
  nStbEntry.stbEntry.schemaRow = pReq->schemaRow;
H
Hongze Cheng 已提交
266 267 268 269
  nStbEntry.stbEntry.schemaTag = pReq->schemaTag;

  metaWLock(pMeta);
  // compare two entry
270 271
  if (oStbEntry.stbEntry.schemaRow.version != pReq->schemaRow.version) {
    metaSaveToSkmDb(pMeta, &nStbEntry);
H
Hongze Cheng 已提交
272 273 274 275 276 277 278 279 280 281
  }

  // if (oStbEntry.stbEntry.schemaTag.sver != pReq->schemaTag.sver) {
  //   // change tag schema
  // }

  // update table.db
  metaSaveToTbDb(pMeta, &nStbEntry);

  // update uid index
H
Hongze Cheng 已提交
282
  tdbTbcUpsert(pUidIdxc, &pReq->suid, sizeof(tb_uid_t), &version, sizeof(version), 0);
H
Hongze Cheng 已提交
283

H
Hongze Cheng 已提交
284
  if (oStbEntry.pBuf) taosMemoryFree(oStbEntry.pBuf);
H
Hongze Cheng 已提交
285 286
  metaULock(pMeta);
  tDecoderClear(&dc);
H
Hongze Cheng 已提交
287 288
  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
H
Hongze Cheng 已提交
289 290 291
  return 0;
}

H
Hongze Cheng 已提交
292
int metaCreateTable(SMeta *pMeta, int64_t version, SVCreateTbReq *pReq) {
H
Hongze Cheng 已提交
293 294
  SMetaEntry  me = {0};
  SMetaReader mr = {0};
H
Hongze Cheng 已提交
295 296 297 298 299

  // validate message
  if (pReq->type != TSDB_CHILD_TABLE && pReq->type != TSDB_NORMAL_TABLE) {
    terrno = TSDB_CODE_INVALID_MSG;
    goto _err;
H
more  
Hongze Cheng 已提交
300 301
  }

H
Hongze Cheng 已提交
302
  // validate req
H
Hongze Cheng 已提交
303
  metaReaderInit(&mr, pMeta, 0);
H
Hongze Cheng 已提交
304
  if (metaGetTableEntryByName(&mr, pReq->name) == 0) {
H
Hongze Cheng 已提交
305 306 307 308
    pReq->uid = mr.me.uid;
    if (pReq->type == TSDB_CHILD_TABLE) {
      pReq->ctb.suid = mr.me.ctbEntry.suid;
    }
H
Hongze Cheng 已提交
309 310 311
    terrno = TSDB_CODE_TDB_TABLE_ALREADY_EXIST;
    metaReaderClear(&mr);
    return -1;
H
Hongze Cheng 已提交
312
  }
H
Hongze Cheng 已提交
313
  metaReaderClear(&mr);
H
Hongze Cheng 已提交
314 315

  // build SMetaEntry
H
Hongze Cheng 已提交
316
  me.version = version;
H
Hongze Cheng 已提交
317 318 319 320 321 322
  me.type = pReq->type;
  me.uid = pReq->uid;
  me.name = pReq->name;
  if (me.type == TSDB_CHILD_TABLE) {
    me.ctbEntry.ctime = pReq->ctime;
    me.ctbEntry.ttlDays = pReq->ttl;
wmmhello's avatar
wmmhello 已提交
323
    me.ctbEntry.commentLen = pReq->commentLen;
wmmhello's avatar
wmmhello 已提交
324
    me.ctbEntry.comment = pReq->comment;
H
Hongze Cheng 已提交
325 326 327 328 329
    me.ctbEntry.suid = pReq->ctb.suid;
    me.ctbEntry.pTags = pReq->ctb.pTag;
  } else {
    me.ntbEntry.ctime = pReq->ctime;
    me.ntbEntry.ttlDays = pReq->ttl;
wmmhello's avatar
wmmhello 已提交
330
    me.ntbEntry.commentLen = pReq->commentLen;
wmmhello's avatar
wmmhello 已提交
331
    me.ntbEntry.comment = pReq->comment;
332 333
    me.ntbEntry.schemaRow = pReq->ntb.schemaRow;
    me.ntbEntry.ncid = me.ntbEntry.schemaRow.pSchema[me.ntbEntry.schemaRow.nCols - 1].colId + 1;
H
more  
Hongze Cheng 已提交
334 335
  }

H
Hongze Cheng 已提交
336
  if (metaHandleEntry(pMeta, &me) < 0) goto _err;
H
Hongze Cheng 已提交
337

S
Shengliang Guan 已提交
338
  metaDebug("vgId:%d, table %s uid %" PRId64 " is created, type:%" PRId8, TD_VID(pMeta->pVnode), pReq->name, pReq->uid,
H
Hongze Cheng 已提交
339
            pReq->type);
H
refact  
Hongze Cheng 已提交
340
  return 0;
H
Hongze Cheng 已提交
341 342

_err:
S
Shengliang Guan 已提交
343
  metaError("vgId:%d, failed to create table:%s type:%s since %s", TD_VID(pMeta->pVnode), pReq->name,
H
Hongze Cheng 已提交
344 345
            pReq->type == TSDB_CHILD_TABLE ? "child table" : "normal table", tstrerror(terrno));
  return -1;
H
refact  
Hongze Cheng 已提交
346 347
}

348
int metaDropTable(SMeta *pMeta, int64_t version, SVDropTbReq *pReq, SArray *tbUids) {
dengyihao's avatar
dengyihao 已提交
349
  void *   pData = NULL;
H
Hongze Cheng 已提交
350 351 352 353
  int      nData = 0;
  int      rc = 0;
  tb_uid_t uid;
  int      type;
H
more  
Hongze Cheng 已提交
354

H
Hongze Cheng 已提交
355 356 357
  rc = tdbTbGet(pMeta->pNameIdx, pReq->name, strlen(pReq->name) + 1, &pData, &nData);
  if (rc < 0) {
    terrno = TSDB_CODE_VND_TABLE_NOT_EXIST;
H
more  
Hongze Cheng 已提交
358 359
    return -1;
  }
H
Hongze Cheng 已提交
360 361
  uid = *(tb_uid_t *)pData;

H
Hongze Cheng 已提交
362 363 364
  metaWLock(pMeta);
  metaDropTableByUid(pMeta, uid, &type);
  metaULock(pMeta);
H
Hongze Cheng 已提交
365

366
  if ((type == TSDB_CHILD_TABLE || type == TSDB_NORMAL_TABLE) && tbUids) {
H
Hongze Cheng 已提交
367
    taosArrayPush(tbUids, &uid);
H
Hongze Cheng 已提交
368
  }
H
Hongze Cheng 已提交
369

H
Hongze Cheng 已提交
370 371 372
  tdbFree(pData);
  return 0;
}
H
Hongze Cheng 已提交
373

374 375 376 377
int metaTtlDropTable(SMeta *pMeta, int64_t ttl, SArray *tbUids) {
  metaWLock(pMeta);
  int ret = metaTtlSmaller(pMeta, ttl, tbUids);
  if(ret != 0){
wmmhello's avatar
wmmhello 已提交
378
    metaULock(pMeta);
379 380 381 382 383
    return ret;
  }
  for (int i = 0; i < taosArrayGetSize(tbUids); ++i) {
    tb_uid_t *uid = (tb_uid_t *)taosArrayGet(tbUids, i);
    metaDropTableByUid(pMeta, *uid, NULL);
wmmhello's avatar
wmmhello 已提交
384
    metaDebug("ttl drop table:%"PRId64, *uid);
385 386 387 388 389 390
  }
  metaULock(pMeta);
  return 0;
}

static void metaBuildTtlIdxKey(STtlIdxKey *ttlKey, const SMetaEntry *pME){
wmmhello's avatar
wmmhello 已提交
391
  int64_t    ttlDays;
392 393 394 395 396 397 398 399 400 401 402 403 404
  int64_t    ctime;
  if (pME->type == TSDB_CHILD_TABLE) {
    ctime = pME->ctbEntry.ctime;
    ttlDays = pME->ctbEntry.ttlDays;
  } else if (pME->type == TSDB_NORMAL_TABLE) {
    ctime = pME->ntbEntry.ctime;
    ttlDays = pME->ntbEntry.ttlDays;
  } else {
    ASSERT(0);
  }

  if (ttlDays <= 0) return;

wmmhello's avatar
wmmhello 已提交
405
  ttlKey->dtime = ctime / 1000 + ttlDays * tsTtlUnit;
406 407 408 409 410 411 412 413 414 415 416
  ttlKey->uid = pME->uid;
}

static int metaDeleteTtlIdx(SMeta *pMeta, const SMetaEntry *pME) {
  STtlIdxKey ttlKey = {0};
  metaBuildTtlIdxKey(&ttlKey, pME);
  if(ttlKey.dtime == 0) return 0;
  return tdbTbDelete(pMeta->pTtlIdx, &ttlKey, sizeof(ttlKey), &pMeta->txn);
}


H
Hongze Cheng 已提交
417
static int metaDropTableByUid(SMeta *pMeta, tb_uid_t uid, int *type) {
dengyihao's avatar
dengyihao 已提交
418
  void *     pData = NULL;
H
Hongze Cheng 已提交
419 420 421 422 423 424
  int        nData = 0;
  int        rc = 0;
  SMetaEntry e = {0};
  SDecoder   dc = {0};

  rc = tdbTbGet(pMeta->pUidIdx, &uid, sizeof(uid), &pData, &nData);
wmmhello's avatar
wmmhello 已提交
425
  int64_t version = *(int64_t *)pData;
H
Hongze Cheng 已提交
426 427 428 429 430 431 432 433 434 435 436

  tdbTbGet(pMeta->pTbDb, &(STbDbKey){.version = version, .uid = uid}, sizeof(STbDbKey), &pData, &nData);

  tDecoderInit(&dc, pData, nData);
  metaDecodeEntry(&dc, &e);

  if (type) *type = e.type;

  tdbTbDelete(pMeta->pTbDb, &(STbDbKey){.version = version, .uid = uid}, sizeof(STbDbKey), &pMeta->txn);
  tdbTbDelete(pMeta->pNameIdx, e.name, strlen(e.name) + 1, &pMeta->txn);
  tdbTbDelete(pMeta->pUidIdx, &uid, sizeof(uid), &pMeta->txn);
437 438
  if(e.type != TSDB_SUPER_TABLE) metaDeleteTtlIdx(pMeta, &e);

H
Hongze Cheng 已提交
439 440 441 442 443 444
  if (e.type == TSDB_CHILD_TABLE) {
    tdbTbDelete(pMeta->pCtbIdx, &(SCtbIdxKey){.suid = e.ctbEntry.suid, .uid = uid}, sizeof(SCtbIdxKey), &pMeta->txn);
  } else if (e.type == TSDB_NORMAL_TABLE) {
    // drop schema.db (todo)
  } else if (e.type == TSDB_SUPER_TABLE) {
    // drop schema.db (todo)
H
Hongze Cheng 已提交
445 446
  }

H
Hongze Cheng 已提交
447 448
  tDecoderClear(&dc);
  tdbFree(pData);
H
Hongze Cheng 已提交
449

H
refact  
Hongze Cheng 已提交
450 451
  return 0;
}
H
Hongze Cheng 已提交
452

D
dapan1121 已提交
453
static int metaAlterTableColumn(SMeta *pMeta, int64_t version, SVAlterTbReq *pAlterTbReq, STableMetaRsp *pMetaRsp) {
dengyihao's avatar
dengyihao 已提交
454
  void *          pVal = NULL;
H
Hongze Cheng 已提交
455
  int             nVal = 0;
dengyihao's avatar
dengyihao 已提交
456
  const void *    pData = NULL;
H
Hongze Cheng 已提交
457 458 459 460
  int             nData = 0;
  int             ret = 0;
  tb_uid_t        uid;
  int64_t         oversion;
dengyihao's avatar
dengyihao 已提交
461
  SSchema *       pColumn = NULL;
H
Hongze Cheng 已提交
462 463 464 465 466
  SMetaEntry      entry = {0};
  SSchemaWrapper *pSchema;
  int             c;

  // search name index
H
Hongze Cheng 已提交
467
  ret = tdbTbGet(pMeta->pNameIdx, pAlterTbReq->tbName, strlen(pAlterTbReq->tbName) + 1, &pVal, &nVal);
H
Hongze Cheng 已提交
468 469 470 471 472 473 474 475 476 477
  if (ret < 0) {
    terrno = TSDB_CODE_VND_TABLE_NOT_EXIST;
    return -1;
  }

  uid = *(tb_uid_t *)pVal;
  tdbFree(pVal);
  pVal = NULL;

  // search uid index
H
Hongze Cheng 已提交
478
  TBC *pUidIdxc = NULL;
H
Hongze Cheng 已提交
479

H
Hongze Cheng 已提交
480 481
  tdbTbcOpen(pMeta->pUidIdx, &pUidIdxc, &pMeta->txn);
  tdbTbcMoveTo(pUidIdxc, &uid, sizeof(uid), &c);
H
Hongze Cheng 已提交
482 483
  ASSERT(c == 0);

H
Hongze Cheng 已提交
484
  tdbTbcGet(pUidIdxc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
485 486 487
  oversion = *(int64_t *)pData;

  // search table.db
H
Hongze Cheng 已提交
488
  TBC *pTbDbc = NULL;
H
Hongze Cheng 已提交
489

H
Hongze Cheng 已提交
490 491
  tdbTbcOpen(pMeta->pTbDb, &pTbDbc, &pMeta->txn);
  tdbTbcMoveTo(pTbDbc, &((STbDbKey){.uid = uid, .version = oversion}), sizeof(STbDbKey), &c);
H
Hongze Cheng 已提交
492
  ASSERT(c == 0);
H
Hongze Cheng 已提交
493
  tdbTbcGet(pTbDbc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
494 495 496

  // get table entry
  SDecoder dc = {0};
H
Hongze Cheng 已提交
497 498 499
  entry.pBuf = taosMemoryMalloc(nData);
  memcpy(entry.pBuf, pData, nData);
  tDecoderInit(&dc, entry.pBuf, nData);
H
Hongze Cheng 已提交
500 501
  ret = metaDecodeEntry(&dc, &entry);
  ASSERT(ret == 0);
H
Hongze Cheng 已提交
502 503 504 505 506 507 508

  if (entry.type != TSDB_NORMAL_TABLE) {
    terrno = TSDB_CODE_VND_INVALID_TABLE_ACTION;
    goto _err;
  }

  // search the column to add/drop/update
509
  pSchema = &entry.ntbEntry.schemaRow;
H
Hongze Cheng 已提交
510 511 512 513 514 515 516 517 518 519 520 521
  int32_t iCol = 0;
  for (;;) {
    pColumn = NULL;

    if (iCol >= pSchema->nCols) break;
    pColumn = &pSchema->pSchema[iCol];

    if (strcmp(pColumn->name, pAlterTbReq->colName) == 0) break;
    iCol++;
  }

  entry.version = version;
H
Hongze Cheng 已提交
522 523
  int      tlen;
  SSchema *pNewSchema = NULL;
H
Hongze Cheng 已提交
524 525 526 527 528 529
  switch (pAlterTbReq->action) {
    case TSDB_ALTER_TABLE_ADD_COLUMN:
      if (pColumn) {
        terrno = TSDB_CODE_VND_COL_ALREADY_EXISTS;
        goto _err;
      }
530
      pSchema->version++;
H
Hongze Cheng 已提交
531
      pSchema->nCols++;
H
Hongze Cheng 已提交
532 533 534
      pNewSchema = taosMemoryMalloc(sizeof(SSchema) * pSchema->nCols);
      memcpy(pNewSchema, pSchema->pSchema, sizeof(SSchema) * (pSchema->nCols - 1));
      pSchema->pSchema = pNewSchema;
535 536 537 538 539
      pSchema->pSchema[entry.ntbEntry.schemaRow.nCols - 1].bytes = pAlterTbReq->bytes;
      pSchema->pSchema[entry.ntbEntry.schemaRow.nCols - 1].type = pAlterTbReq->type;
      pSchema->pSchema[entry.ntbEntry.schemaRow.nCols - 1].flags = pAlterTbReq->flags;
      pSchema->pSchema[entry.ntbEntry.schemaRow.nCols - 1].colId = entry.ntbEntry.ncid++;
      strcpy(pSchema->pSchema[entry.ntbEntry.schemaRow.nCols - 1].name, pAlterTbReq->colName);
H
Hongze Cheng 已提交
540 541 542 543 544 545 546 547 548 549
      break;
    case TSDB_ALTER_TABLE_DROP_COLUMN:
      if (pColumn == NULL) {
        terrno = TSDB_CODE_VND_TABLE_COL_NOT_EXISTS;
        goto _err;
      }
      if (pColumn->colId == 0) {
        terrno = TSDB_CODE_VND_INVALID_TABLE_ACTION;
        goto _err;
      }
550
      pSchema->version++;
H
Hongze Cheng 已提交
551 552 553 554
      tlen = (pSchema->nCols - iCol - 1) * sizeof(SSchema);
      if (tlen) {
        memmove(pColumn, pColumn + 1, tlen);
      }
H
Hongze Cheng 已提交
555
      pSchema->nCols--;
H
Hongze Cheng 已提交
556 557 558 559 560 561
      break;
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES:
      if (pColumn == NULL) {
        terrno = TSDB_CODE_VND_TABLE_COL_NOT_EXISTS;
        goto _err;
      }
H
Hongze Cheng 已提交
562
      if (!IS_VAR_DATA_TYPE(pColumn->type) || pColumn->bytes > pAlterTbReq->colModBytes) {
H
Hongze Cheng 已提交
563 564 565
        terrno = TSDB_CODE_VND_INVALID_TABLE_ACTION;
        goto _err;
      }
566
      pSchema->version++;
H
Hongze Cheng 已提交
567
      pColumn->bytes = pAlterTbReq->colModBytes;
H
Hongze Cheng 已提交
568 569 570 571 572 573
      break;
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_NAME:
      if (pColumn == NULL) {
        terrno = TSDB_CODE_VND_TABLE_COL_NOT_EXISTS;
        goto _err;
      }
574
      pSchema->version++;
H
Hongze Cheng 已提交
575 576 577 578 579 580
      strcpy(pColumn->name, pAlterTbReq->colNewName);
      break;
  }

  entry.version = version;

H
Hongze Cheng 已提交
581 582 583 584 585 586
  // do actual write
  metaWLock(pMeta);

  // save to table db
  metaSaveToTbDb(pMeta, &entry);

H
Hongze Cheng 已提交
587
  tdbTbcUpsert(pUidIdxc, &entry.uid, sizeof(tb_uid_t), &version, sizeof(version), 0);
H
Hongze Cheng 已提交
588 589 590 591 592

  metaSaveToSkmDb(pMeta, &entry);

  metaULock(pMeta);

D
dapan1121 已提交
593 594
  metaUpdateMetaRsp(uid, pAlterTbReq->tbName, pSchema, pMetaRsp);

wmmhello's avatar
wmmhello 已提交
595
  if (entry.pBuf) taosMemoryFree(entry.pBuf);
H
Hongze Cheng 已提交
596
  if (pNewSchema) taosMemoryFree(pNewSchema);
H
Hongze Cheng 已提交
597 598
  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
599 600
  tDecoderClear(&dc);

H
Hongze Cheng 已提交
601
  return 0;
H
Hongze Cheng 已提交
602 603

_err:
wmmhello's avatar
wmmhello 已提交
604
  if (entry.pBuf) taosMemoryFree(entry.pBuf);
H
Hongze Cheng 已提交
605 606
  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
607 608
  tDecoderClear(&dc);

H
Hongze Cheng 已提交
609
  return -1;
H
Hongze Cheng 已提交
610 611 612
}

static int metaUpdateTableTagVal(SMeta *pMeta, int64_t version, SVAlterTbReq *pAlterTbReq) {
H
Hongze Cheng 已提交
613 614
  SMetaEntry  ctbEntry = {0};
  SMetaEntry  stbEntry = {0};
dengyihao's avatar
dengyihao 已提交
615
  void *      pVal = NULL;
H
Hongze Cheng 已提交
616 617 618 619 620 621 622 623 624
  int         nVal = 0;
  int         ret;
  int         c;
  tb_uid_t    uid;
  int64_t     oversion;
  const void *pData = NULL;
  int         nData = 0;

  // search name index
H
Hongze Cheng 已提交
625
  ret = tdbTbGet(pMeta->pNameIdx, pAlterTbReq->tbName, strlen(pAlterTbReq->tbName) + 1, &pVal, &nVal);
H
Hongze Cheng 已提交
626 627 628 629 630 631 632 633 634 635
  if (ret < 0) {
    terrno = TSDB_CODE_VND_TABLE_NOT_EXIST;
    return -1;
  }

  uid = *(tb_uid_t *)pVal;
  tdbFree(pVal);
  pVal = NULL;

  // search uid index
H
Hongze Cheng 已提交
636
  TBC *pUidIdxc = NULL;
H
Hongze Cheng 已提交
637

H
Hongze Cheng 已提交
638 639
  tdbTbcOpen(pMeta->pUidIdx, &pUidIdxc, &pMeta->txn);
  tdbTbcMoveTo(pUidIdxc, &uid, sizeof(uid), &c);
H
Hongze Cheng 已提交
640 641
  ASSERT(c == 0);

H
Hongze Cheng 已提交
642
  tdbTbcGet(pUidIdxc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
643 644 645
  oversion = *(int64_t *)pData;

  // search table.db
dengyihao's avatar
dengyihao 已提交
646
  TBC *    pTbDbc = NULL;
H
Hongze Cheng 已提交
647 648
  SDecoder dc1 = {0};
  SDecoder dc2 = {0};
H
Hongze Cheng 已提交
649

H
Hongze Cheng 已提交
650
  /* get ctbEntry */
H
Hongze Cheng 已提交
651 652
  tdbTbcOpen(pMeta->pTbDb, &pTbDbc, &pMeta->txn);
  tdbTbcMoveTo(pTbDbc, &((STbDbKey){.uid = uid, .version = oversion}), sizeof(STbDbKey), &c);
H
Hongze Cheng 已提交
653
  ASSERT(c == 0);
H
Hongze Cheng 已提交
654
  tdbTbcGet(pTbDbc, NULL, NULL, &pData, &nData);
H
Hongze Cheng 已提交
655

H
Hongze Cheng 已提交
656 657
  ctbEntry.pBuf = taosMemoryMalloc(nData);
  memcpy(ctbEntry.pBuf, pData, nData);
H
Hongze Cheng 已提交
658 659
  tDecoderInit(&dc1, ctbEntry.pBuf, nData);
  metaDecodeEntry(&dc1, &ctbEntry);
H
Hongze Cheng 已提交
660

H
Hongze Cheng 已提交
661
  /* get stbEntry*/
H
Hongze Cheng 已提交
662 663 664
  tdbTbGet(pMeta->pUidIdx, &ctbEntry.ctbEntry.suid, sizeof(tb_uid_t), &pVal, &nVal);
  tdbTbGet(pMeta->pTbDb, &((STbDbKey){.uid = ctbEntry.ctbEntry.suid, .version = *(int64_t *)pVal}), sizeof(STbDbKey),
           (void **)&stbEntry.pBuf, &nVal);
H
Hongze Cheng 已提交
665
  tdbFree(pVal);
H
Hongze Cheng 已提交
666 667
  tDecoderInit(&dc2, stbEntry.pBuf, nVal);
  metaDecodeEntry(&dc2, &stbEntry);
H
Hongze Cheng 已提交
668 669

  SSchemaWrapper *pTagSchema = &stbEntry.stbEntry.schemaTag;
dengyihao's avatar
dengyihao 已提交
670
  SSchema *       pColumn = NULL;
H
Hongze Cheng 已提交
671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
  int32_t         iCol = 0;
  for (;;) {
    pColumn = NULL;

    if (iCol >= pTagSchema->nCols) break;
    pColumn = &pTagSchema->pSchema[iCol];

    if (strcmp(pColumn->name, pAlterTbReq->tagName) == 0) break;
    iCol++;
  }

  if (pColumn == NULL) {
    terrno = TSDB_CODE_VND_TABLE_COL_NOT_EXISTS;
    goto _err;
  }
H
Hongze Cheng 已提交
686

H
Hongze Cheng 已提交
687 688
  if (iCol == 0) {
    // TODO : need to update tag index
H
Hongze Cheng 已提交
689
  }
H
Hongze Cheng 已提交
690
  ctbEntry.version = version;
H
Hongze Cheng 已提交
691
  if (pTagSchema->nCols == 1 && pTagSchema->pSchema[0].type == TSDB_DATA_TYPE_JSON) {
wmmhello's avatar
wmmhello 已提交
692
    ctbEntry.ctbEntry.pTags = taosMemoryMalloc(pAlterTbReq->nTagVal);
H
Hongze Cheng 已提交
693
    if (ctbEntry.ctbEntry.pTags == NULL) {
wmmhello's avatar
wmmhello 已提交
694 695 696
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      goto _err;
    }
H
Hongze Cheng 已提交
697 698
    memcpy((void *)ctbEntry.ctbEntry.pTags, pAlterTbReq->pTagVal, pAlterTbReq->nTagVal);
  } else {
C
Cary Xu 已提交
699
    const STag *pOldTag = (const STag *)ctbEntry.ctbEntry.pTags;
dengyihao's avatar
dengyihao 已提交
700 701
    STag *      pNewTag = NULL;
    SArray *    pTagArray = taosArrayInit(pTagSchema->nCols, sizeof(STagVal));
C
Cary Xu 已提交
702
    if (!pTagArray) {
C
Cary Xu 已提交
703 704 705
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      goto _err;
    }
wmmhello's avatar
wmmhello 已提交
706 707 708
    for (int32_t i = 0; i < pTagSchema->nCols; i++) {
      SSchema *pCol = &pTagSchema->pSchema[i];
      if (iCol == i) {
wmmhello's avatar
wmmhello 已提交
709 710 711 712 713 714
        STagVal val = {0};
        val.type = pCol->type;
        val.cid = pCol->colId;
        if (IS_VAR_DATA_TYPE(pCol->type)) {
          val.pData = pAlterTbReq->pTagVal;
          val.nData = pAlterTbReq->nTagVal;
715
        } else {
wmmhello's avatar
wmmhello 已提交
716 717 718
          memcpy(&val.i64, pAlterTbReq->pTagVal, pAlterTbReq->nTagVal);
        }
        taosArrayPush(pTagArray, &val);
wmmhello's avatar
wmmhello 已提交
719
      } else {
wmmhello's avatar
wmmhello 已提交
720
        STagVal val = {.cid = pCol->colId};
wmmhello's avatar
wmmhello 已提交
721 722
        if (tTagGet(pOldTag, &val)) {
          taosArrayPush(pTagArray, &val);
H
Hongze Cheng 已提交
723 724 725
        }
      }
    }
C
Cary Xu 已提交
726 727
    if ((terrno = tTagNew(pTagArray, pTagSchema->version, false, &pNewTag)) < 0) {
      taosArrayDestroy(pTagArray);
C
Cary Xu 已提交
728 729 730
      goto _err;
    }
    ctbEntry.ctbEntry.pTags = (uint8_t *)pNewTag;
C
Cary Xu 已提交
731
    taosArrayDestroy(pTagArray);
wmmhello's avatar
wmmhello 已提交
732
  }
H
Hongze Cheng 已提交
733 734 735 736 737

  // save to table.db
  metaSaveToTbDb(pMeta, &ctbEntry);

  // save to uid.idx
H
Hongze Cheng 已提交
738
  tdbTbUpsert(pMeta->pUidIdx, &ctbEntry.uid, sizeof(tb_uid_t), &version, sizeof(version), &pMeta->txn);
H
Hongze Cheng 已提交
739

H
Hongze Cheng 已提交
740 741
  tDecoderClear(&dc1);
  tDecoderClear(&dc2);
H
Hongze Cheng 已提交
742
  if (ctbEntry.ctbEntry.pTags) taosMemoryFree((void *)ctbEntry.ctbEntry.pTags);
H
Hongze Cheng 已提交
743 744
  if (ctbEntry.pBuf) taosMemoryFree(ctbEntry.pBuf);
  if (stbEntry.pBuf) tdbFree(stbEntry.pBuf);
H
Hongze Cheng 已提交
745 746
  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
H
Hongze Cheng 已提交
747
  return 0;
H
Hongze Cheng 已提交
748 749

_err:
H
Hongze Cheng 已提交
750 751
  tDecoderClear(&dc1);
  tDecoderClear(&dc2);
H
Hongze Cheng 已提交
752 753
  if (ctbEntry.pBuf) taosMemoryFree(ctbEntry.pBuf);
  if (stbEntry.pBuf) tdbFree(stbEntry.pBuf);
H
Hongze Cheng 已提交
754 755
  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
H
Hongze Cheng 已提交
756
  return -1;
H
Hongze Cheng 已提交
757 758 759
}

static int metaUpdateTableOptions(SMeta *pMeta, int64_t version, SVAlterTbReq *pAlterTbReq) {
wmmhello's avatar
wmmhello 已提交
760 761 762 763 764 765 766 767 768 769 770 771 772 773 774
  void *          pVal = NULL;
  int             nVal = 0;
  const void *    pData = NULL;
  int             nData = 0;
  int             ret = 0;
  tb_uid_t        uid;
  int64_t         oversion;
  SMetaEntry      entry = {0};
  int             c = 0;

  // search name index
  ret = tdbTbGet(pMeta->pNameIdx, pAlterTbReq->tbName, strlen(pAlterTbReq->tbName) + 1, &pVal, &nVal);
  if (ret < 0) {
    terrno = TSDB_CODE_VND_TABLE_NOT_EXIST;
    return -1;
775
  }
wmmhello's avatar
wmmhello 已提交
776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815

  uid = *(tb_uid_t *)pVal;
  tdbFree(pVal);
  pVal = NULL;

  // search uid index
  TBC *pUidIdxc = NULL;

  tdbTbcOpen(pMeta->pUidIdx, &pUidIdxc, &pMeta->txn);
  tdbTbcMoveTo(pUidIdxc, &uid, sizeof(uid), &c);
  ASSERT(c == 0);

  tdbTbcGet(pUidIdxc, NULL, NULL, &pData, &nData);
  oversion = *(int64_t *)pData;

  // search table.db
  TBC *pTbDbc = NULL;

  tdbTbcOpen(pMeta->pTbDb, &pTbDbc, &pMeta->txn);
  tdbTbcMoveTo(pTbDbc, &((STbDbKey){.uid = uid, .version = oversion}), sizeof(STbDbKey), &c);
  ASSERT(c == 0);
  tdbTbcGet(pTbDbc, NULL, NULL, &pData, &nData);

  // get table entry
  SDecoder dc = {0};
  entry.pBuf = taosMemoryMalloc(nData);
  memcpy(entry.pBuf, pData, nData);
  tDecoderInit(&dc, entry.pBuf, nData);
  ret = metaDecodeEntry(&dc, &entry);
  ASSERT(ret == 0);

  entry.version = version;
  metaWLock(pMeta);
  // build SMetaEntry
  if (entry.type == TSDB_CHILD_TABLE) {
    if(pAlterTbReq->updateTTL) {
      metaDeleteTtlIdx(pMeta, &entry);
      entry.ctbEntry.ttlDays = pAlterTbReq->newTTL;
      metaUpdateTtlIdx(pMeta, &entry);
    }
wmmhello's avatar
wmmhello 已提交
816 817 818 819
    if(pAlterTbReq->newCommentLen >= 0) {
      entry.ctbEntry.commentLen = pAlterTbReq->newCommentLen;
      entry.ctbEntry.comment = pAlterTbReq->newComment;
    }
wmmhello's avatar
wmmhello 已提交
820 821 822 823 824 825
  } else {
    if(pAlterTbReq->updateTTL) {
      metaDeleteTtlIdx(pMeta, &entry);
      entry.ntbEntry.ttlDays = pAlterTbReq->newTTL;
      metaUpdateTtlIdx(pMeta, &entry);
    }
wmmhello's avatar
wmmhello 已提交
826 827 828 829
    if(pAlterTbReq->newCommentLen >= 0) {
      entry.ntbEntry.commentLen = pAlterTbReq->newCommentLen;
      entry.ntbEntry.comment = pAlterTbReq->newComment;
    }
830
  }
wmmhello's avatar
wmmhello 已提交
831 832 833 834 835 836 837 838

  // save to table db
  metaSaveToTbDb(pMeta, &entry);
  tdbTbcUpsert(pUidIdxc, &entry.uid, sizeof(tb_uid_t), &version, sizeof(version), 0);
  metaULock(pMeta);

  tdbTbcClose(pTbDbc);
  tdbTbcClose(pUidIdxc);
839
  tDecoderClear(&dc);
wmmhello's avatar
wmmhello 已提交
840
  if (entry.pBuf) taosMemoryFree(entry.pBuf);
H
Hongze Cheng 已提交
841 842 843
  return 0;
}

D
dapan1121 已提交
844
int metaAlterTable(SMeta *pMeta, int64_t version, SVAlterTbReq *pReq, STableMetaRsp *pMetaRsp) {
H
Hongze Cheng 已提交
845 846 847 848 849
  switch (pReq->action) {
    case TSDB_ALTER_TABLE_ADD_COLUMN:
    case TSDB_ALTER_TABLE_DROP_COLUMN:
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES:
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_NAME:
D
dapan1121 已提交
850
      return metaAlterTableColumn(pMeta, version, pReq, pMetaRsp);
H
Hongze Cheng 已提交
851 852 853 854 855 856 857 858 859 860 861
    case TSDB_ALTER_TABLE_UPDATE_TAG_VAL:
      return metaUpdateTableTagVal(pMeta, version, pReq);
    case TSDB_ALTER_TABLE_UPDATE_OPTIONS:
      return metaUpdateTableOptions(pMeta, version, pReq);
    default:
      terrno = TSDB_CODE_VND_INVALID_TABLE_ACTION;
      return -1;
      break;
  }
}

H
Hongze Cheng 已提交
862
static int metaSaveToTbDb(SMeta *pMeta, const SMetaEntry *pME) {
H
Hongze Cheng 已提交
863
  STbDbKey tbDbKey;
dengyihao's avatar
dengyihao 已提交
864 865
  void *   pKey = NULL;
  void *   pVal = NULL;
H
Hongze Cheng 已提交
866 867
  int      kLen = 0;
  int      vLen = 0;
H
Hongze Cheng 已提交
868
  SEncoder coder = {0};
H
Hongze Cheng 已提交
869 870

  // set key and value
H
Hongze Cheng 已提交
871
  tbDbKey.version = pME->version;
H
Hongze Cheng 已提交
872 873 874 875
  tbDbKey.uid = pME->uid;

  pKey = &tbDbKey;
  kLen = sizeof(tbDbKey);
H
Hongze Cheng 已提交
876

wafwerar's avatar
wafwerar 已提交
877 878 879
  int32_t ret = 0;
  tEncodeSize(metaEncodeEntry, pME, vLen, ret);
  if (ret < 0) {
H
Hongze Cheng 已提交
880 881 882 883 884 885 886 887 888
    goto _err;
  }

  pVal = taosMemoryMalloc(vLen);
  if (pVal == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }

H
Hongze Cheng 已提交
889
  tEncoderInit(&coder, pVal, vLen);
H
Hongze Cheng 已提交
890 891 892 893 894

  if (metaEncodeEntry(&coder, pME) < 0) {
    goto _err;
  }

H
Hongze Cheng 已提交
895
  tEncoderClear(&coder);
H
Hongze Cheng 已提交
896 897

  // write to table.db
H
Hongze Cheng 已提交
898
  if (tdbTbInsert(pMeta->pTbDb, pKey, kLen, pVal, vLen, &pMeta->txn) < 0) {
H
Hongze Cheng 已提交
899 900 901 902 903 904 905 906 907 908 909
    goto _err;
  }

  taosMemoryFree(pVal);
  return 0;

_err:
  taosMemoryFree(pVal);
  return -1;
}

H
Hongze Cheng 已提交
910
static int metaUpdateUidIdx(SMeta *pMeta, const SMetaEntry *pME) {
H
Hongze Cheng 已提交
911
  return tdbTbInsert(pMeta->pUidIdx, &pME->uid, sizeof(tb_uid_t), &pME->version, sizeof(int64_t), &pMeta->txn);
H
Hongze Cheng 已提交
912 913
}

H
Hongze Cheng 已提交
914
static int metaUpdateNameIdx(SMeta *pMeta, const SMetaEntry *pME) {
H
Hongze Cheng 已提交
915
  return tdbTbInsert(pMeta->pNameIdx, pME->name, strlen(pME->name) + 1, &pME->uid, sizeof(tb_uid_t), &pMeta->txn);
H
Hongze Cheng 已提交
916 917
}

H
Hongze Cheng 已提交
918
static int metaUpdateTtlIdx(SMeta *pMeta, const SMetaEntry *pME) {
919 920 921
  STtlIdxKey ttlKey = {0};
  metaBuildTtlIdxKey(&ttlKey, pME);
  if(ttlKey.dtime == 0) return 0;
H
Hongze Cheng 已提交
922
  return tdbTbInsert(pMeta->pTtlIdx, &ttlKey, sizeof(ttlKey), NULL, 0, &pMeta->txn);
H
Hongze Cheng 已提交
923 924
}

H
Hongze Cheng 已提交
925 926
static int metaUpdateCtbIdx(SMeta *pMeta, const SMetaEntry *pME) {
  SCtbIdxKey ctbIdxKey = {.suid = pME->ctbEntry.suid, .uid = pME->uid};
H
Hongze Cheng 已提交
927
  return tdbTbInsert(pMeta->pCtbIdx, &ctbIdxKey, sizeof(ctbIdxKey), NULL, 0, &pMeta->txn);
H
Hongze Cheng 已提交
928 929
}

wmmhello's avatar
wmmhello 已提交
930
int metaCreateTagIdxKey(tb_uid_t suid, int32_t cid, const void *pTagData, int32_t nTagData, int8_t type, tb_uid_t uid,
931
                        STagIdxKey **ppTagIdxKey, int32_t *nTagIdxKey) {
dengyihao's avatar
dengyihao 已提交
932 933 934 935 936
  if (IS_VAR_DATA_TYPE(type)) {
    *nTagIdxKey = sizeof(STagIdxKey) + nTagData + VARSTR_HEADER_SIZE + sizeof(tb_uid_t);
  } else {
    *nTagIdxKey = sizeof(STagIdxKey) + nTagData + sizeof(tb_uid_t);
  }
H
Hongze Cheng 已提交
937 938 939 940 941 942 943 944 945 946 947

  *ppTagIdxKey = (STagIdxKey *)taosMemoryMalloc(*nTagIdxKey);
  if (*ppTagIdxKey == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

  (*ppTagIdxKey)->suid = suid;
  (*ppTagIdxKey)->cid = cid;
  (*ppTagIdxKey)->isNull = (pTagData == NULL) ? 1 : 0;
  (*ppTagIdxKey)->type = type;
dengyihao's avatar
dengyihao 已提交
948 949 950 951 952 953 954 955 956 957

  // refactor
  if (IS_VAR_DATA_TYPE(type)) {
    memcpy((*ppTagIdxKey)->data, (uint16_t *)&nTagData, VARSTR_HEADER_SIZE);
    memcpy((*ppTagIdxKey)->data + VARSTR_HEADER_SIZE, pTagData, nTagData);
    *(tb_uid_t *)((*ppTagIdxKey)->data + VARSTR_HEADER_SIZE + nTagData) = uid;
  } else {
    memcpy((*ppTagIdxKey)->data, pTagData, nTagData);
    *(tb_uid_t *)((*ppTagIdxKey)->data + nTagData) = uid;
  }
H
Hongze Cheng 已提交
958 959 960 961 962 963 964 965 966

  return 0;
}

static void metaDestroyTagIdxKey(STagIdxKey *pTagIdxKey) {
  if (pTagIdxKey) taosMemoryFree(pTagIdxKey);
}

static int metaUpdateTagIdx(SMeta *pMeta, const SMetaEntry *pCtbEntry) {
dengyihao's avatar
dengyihao 已提交
967
  void *         pData = NULL;
H
Hongze Cheng 已提交
968 969 970
  int            nData = 0;
  STbDbKey       tbDbKey = {0};
  SMetaEntry     stbEntry = {0};
dengyihao's avatar
dengyihao 已提交
971
  STagIdxKey *   pTagIdxKey = NULL;
H
Hongze Cheng 已提交
972 973
  int32_t        nTagIdxKey;
  const SSchema *pTagColumn;       // = &stbEntry.stbEntry.schema.pSchema[0];
dengyihao's avatar
dengyihao 已提交
974
  const void *   pTagData = NULL;  //
C
Cary Xu 已提交
975
  int32_t        nTagData = 0;
H
Hongze Cheng 已提交
976 977 978
  SDecoder       dc = {0};

  // get super table
wmmhello's avatar
wmmhello 已提交
979 980 981
  if(tdbTbGet(pMeta->pUidIdx, &pCtbEntry->ctbEntry.suid, sizeof(tb_uid_t), &pData, &nData) != 0){
    return -1;
  }
H
Hongze Cheng 已提交
982 983
  tbDbKey.uid = pCtbEntry->ctbEntry.suid;
  tbDbKey.version = *(int64_t *)pData;
H
Hongze Cheng 已提交
984
  tdbTbGet(pMeta->pTbDb, &tbDbKey, sizeof(tbDbKey), &pData, &nData);
H
Hongze Cheng 已提交
985 986 987 988 989

  tDecoderInit(&dc, pData, nData);
  metaDecodeEntry(&dc, &stbEntry);

  pTagColumn = &stbEntry.stbEntry.schemaTag.pSchema[0];
C
Cary Xu 已提交
990 991

  STagVal tagVal = {.cid = pTagColumn->colId};
992
  if (pTagColumn->type != TSDB_DATA_TYPE_JSON) {
wmmhello's avatar
wmmhello 已提交
993
    tTagGet((const STag *)pCtbEntry->ctbEntry.pTags, &tagVal);
994
    if (IS_VAR_DATA_TYPE(pTagColumn->type)) {
wmmhello's avatar
wmmhello 已提交
995 996
      pTagData = tagVal.pData;
      nTagData = (int32_t)tagVal.nData;
997
    } else {
wmmhello's avatar
wmmhello 已提交
998 999 1000
      pTagData = &(tagVal.i64);
      nTagData = tDataTypes[pTagColumn->type].bytes;
    }
1001 1002 1003
  } else {
    // pTagData = pCtbEntry->ctbEntry.pTags;
    // nTagData = ((const STag *)pCtbEntry->ctbEntry.pTags)->len;
dengyihao's avatar
dengyihao 已提交
1004 1005 1006
    pTagData = pCtbEntry->ctbEntry.pTags;
    nTagData = ((const STag *)pCtbEntry->ctbEntry.pTags)->len;
    return metaSaveJsonVarToIdx(pMeta, pCtbEntry, pTagColumn);
wmmhello's avatar
wmmhello 已提交
1007
  }
1008 1009
  if (metaCreateTagIdxKey(pCtbEntry->ctbEntry.suid, pTagColumn->colId, pTagData, nTagData, pTagColumn->type,
                          pCtbEntry->uid, &pTagIdxKey, &nTagIdxKey) < 0) {
H
Hongze Cheng 已提交
1010 1011
    return -1;
  }
H
Hongze Cheng 已提交
1012
  tdbTbInsert(pMeta->pTagIdx, pTagIdxKey, nTagIdxKey, NULL, 0, &pMeta->txn);
H
Hongze Cheng 已提交
1013 1014 1015
  metaDestroyTagIdxKey(pTagIdxKey);
  tDecoderClear(&dc);
  tdbFree(pData);
H
Hongze Cheng 已提交
1016 1017 1018
  return 0;
}

H
Hongze Cheng 已提交
1019
static int metaSaveToSkmDb(SMeta *pMeta, const SMetaEntry *pME) {
H
Hongze Cheng 已提交
1020
  SEncoder              coder = {0};
dengyihao's avatar
dengyihao 已提交
1021
  void *                pVal = NULL;
H
Hongze Cheng 已提交
1022 1023 1024 1025 1026 1027
  int                   vLen = 0;
  int                   rcode = 0;
  SSkmDbKey             skmDbKey = {0};
  const SSchemaWrapper *pSW;

  if (pME->type == TSDB_SUPER_TABLE) {
1028
    pSW = &pME->stbEntry.schemaRow;
H
Hongze Cheng 已提交
1029
  } else if (pME->type == TSDB_NORMAL_TABLE) {
1030
    pSW = &pME->ntbEntry.schemaRow;
H
Hongze Cheng 已提交
1031 1032
  } else {
    ASSERT(0);
H
Hongze Cheng 已提交
1033 1034
  }

H
Hongze Cheng 已提交
1035
  skmDbKey.uid = pME->uid;
1036
  skmDbKey.sver = pSW->version;
H
Hongze Cheng 已提交
1037 1038

  // encode schema
wafwerar's avatar
wafwerar 已提交
1039 1040 1041
  int32_t ret = 0;
  tEncodeSize(tEncodeSSchemaWrapper, pSW, vLen, ret);
  if (ret < 0) return -1;
H
Hongze Cheng 已提交
1042
  pVal = taosMemoryMalloc(vLen);
H
Hongze Cheng 已提交
1043 1044 1045 1046 1047 1048
  if (pVal == NULL) {
    rcode = -1;
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

H
Hongze Cheng 已提交
1049
  tEncoderInit(&coder, pVal, vLen);
H
Hongze Cheng 已提交
1050 1051
  tEncodeSSchemaWrapper(&coder, pSW);

H
Hongze Cheng 已提交
1052
  if (tdbTbInsert(pMeta->pSkmDb, &skmDbKey, sizeof(skmDbKey), pVal, vLen, &pMeta->txn) < 0) {
H
Hongze Cheng 已提交
1053 1054 1055 1056 1057
    rcode = -1;
    goto _exit;
  }

_exit:
H
Hongze Cheng 已提交
1058
  taosMemoryFree(pVal);
H
Hongze Cheng 已提交
1059
  tEncoderClear(&coder);
H
Hongze Cheng 已提交
1060 1061 1062 1063
  return rcode;
}

static int metaHandleEntry(SMeta *pMeta, const SMetaEntry *pME) {
H
Hongze Cheng 已提交
1064 1065
  metaWLock(pMeta);

H
Hongze Cheng 已提交
1066
  // save to table.db
H
Hongze Cheng 已提交
1067
  if (metaSaveToTbDb(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1068 1069

  // update uid.idx
H
Hongze Cheng 已提交
1070
  if (metaUpdateUidIdx(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1071 1072

  // update name.idx
H
Hongze Cheng 已提交
1073
  if (metaUpdateNameIdx(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1074 1075 1076

  if (pME->type == TSDB_CHILD_TABLE) {
    // update ctb.idx
H
Hongze Cheng 已提交
1077
    if (metaUpdateCtbIdx(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1078 1079

    // update tag.idx
H
Hongze Cheng 已提交
1080
    if (metaUpdateTagIdx(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1081 1082
  } else {
    // update schema.db
H
Hongze Cheng 已提交
1083
    if (metaSaveToSkmDb(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1084 1085 1086
  }

  if (pME->type != TSDB_SUPER_TABLE) {
H
Hongze Cheng 已提交
1087
    if (metaUpdateTtlIdx(pMeta, pME) < 0) goto _err;
H
Hongze Cheng 已提交
1088 1089
  }

H
Hongze Cheng 已提交
1090
  metaULock(pMeta);
H
Hongze Cheng 已提交
1091
  return 0;
H
Hongze Cheng 已提交
1092 1093 1094 1095

_err:
  metaULock(pMeta);
  return -1;
1096
}
dengyihao's avatar
dengyihao 已提交
1097
// refactor later
dengyihao's avatar
dengyihao 已提交
1098 1099
void *metaGetIdx(SMeta *pMeta) { return pMeta->pTagIdx; }
void *metaGetIvtIdx(SMeta *pMeta) { return pMeta->pTagIvtIdx; }