dCDAstProcess.c 34.2 KB
Newer Older
H
Haojun Liao 已提交
1 2
#include "tmsg.h"
#include "tglobal.h"
3
#include "parserInt.h"
H
Haojun Liao 已提交
4 5 6
#include "ttime.h"
#include "astToMsg.h"
#include "astGenerator.h"
H
Haojun Liao 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19 20
#include "parserUtil.h"
#include "queryInfoUtil.h"

/* is contained in pFieldList or not */
static bool has(SArray* pFieldList, int32_t startIndex, const char* name) {
  size_t numOfCols = taosArrayGetSize(pFieldList);
  for (int32_t j = startIndex; j < numOfCols; ++j) {
    TAOS_FIELD* field = taosArrayGet(pFieldList, j);
    if (strncasecmp(name, field->name, sizeof(field->name) - 1) == 0) return true;
  }

  return false;
}

H
Haojun Liao 已提交
21
static int32_t setShowInfo(SShowInfo* pShowInfo, SParseContext* pCtx, void** output, int32_t* outputLen,
22
                           SEpSet* pEpSet, void** pExtension, SMsgBuf* pMsgBuf) {
H
Haojun Liao 已提交
23 24 25 26 27 28
  const char* msg1 = "invalid name";
  const char* msg2 = "wildcard string should be less than %d characters";
  const char* msg3 = "database name too long";
  const char* msg4 = "pattern is invalid";
  const char* msg5 = "database name is empty";
  const char* msg6 = "pattern string is empty";
H
Haojun Liao 已提交
29
  const char* msg7 = "database not specified";
H
Haojun Liao 已提交
30 31 32 33
  /*
   * database prefix in pInfo->pMiscInfo->a[0]
   * wildcard in like clause in pInfo->pMiscInfo->a[1]
   */
H
Hongze Cheng 已提交
34
  int16_t showType = pShowInfo->showType;
H
Haojun Liao 已提交
35
  if (showType == TSDB_MGMT_TABLE_TABLE) {
36
    SArray* array = NULL;
H
Hongze Cheng 已提交
37
    SName   name = {0};
H
Haojun Liao 已提交
38 39 40 41 42 43 44 45 46 47 48

    if (pCtx->db == NULL && pShowInfo->prefix.n == 0) {
      return buildInvalidOperationMsg(pMsgBuf, msg7);
    }

    SVShowTablesReq* pShowReq = calloc(1, sizeof(SVShowTablesReq));
    if (pShowInfo->prefix.n > 0) {
      tNameSetDbName(&name, pCtx->acctId, pShowInfo->prefix.z, pShowInfo->prefix.n);
    } else {
      tNameSetDbName(&name, pCtx->acctId, pCtx->db, strlen(pCtx->db));
    }
49 50 51 52

    char dbFname[TSDB_DB_FNAME_LEN] = {0};
    tNameGetFullDbName(&name, dbFname);

H
Haojun Liao 已提交
53 54 55 56 57
    int32_t code = catalogGetDBVgroup(pCtx->pCatalog, pCtx->pTransporter, &pCtx->mgmtEpSet, dbFname, false, &array);
    if (code != TSDB_CODE_SUCCESS) {
      terrno = code;
      return code;
    }
58 59 60 61 62 63

    SVgroupInfo* info = taosArrayGet(array, 0);
    pShowReq->head.vgId = htonl(info->vgId);
    pEpSet->numOfEps = info->numOfEps;
    pEpSet->inUse = info->inUse;

H
Hongze Cheng 已提交
64
    for (int32_t i = 0; i < pEpSet->numOfEps; ++i) {
65 66 67 68
      strncpy(pEpSet->fqdn[i], info->epAddr[i].fqdn, tListLen(pEpSet->fqdn[i]));
      pEpSet->port[i] = info->epAddr[i].port;
    }

H
Haojun Liao 已提交
69 70
    *outputLen  = sizeof(SVShowTablesReq);
    *output     = pShowReq;
71
    *pExtension = array;
H
Haojun Liao 已提交
72 73 74 75 76 77 78
  } else {
    if (showType == TSDB_MGMT_TABLE_STB || showType == TSDB_MGMT_TABLE_VGROUP) {
      SToken* pDbPrefixToken = &pShowInfo->prefix;
      if (pDbPrefixToken->type != 0) {
        if (pDbPrefixToken->n >= TSDB_DB_NAME_LEN) {  // db name is too long
          return buildInvalidOperationMsg(pMsgBuf, msg3);
        }
H
Haojun Liao 已提交
79

H
Haojun Liao 已提交
80 81 82
        if (pDbPrefixToken->n <= 0) {
          return buildInvalidOperationMsg(pMsgBuf, msg5);
        }
H
Haojun Liao 已提交
83

H
Haojun Liao 已提交
84 85 86 87 88 89 90 91
        if (parserValidateIdToken(pDbPrefixToken) != TSDB_CODE_SUCCESS) {
          return buildInvalidOperationMsg(pMsgBuf, msg1);
        }

        //      int32_t ret = tNameSetDbName(&pTableMetaInfo->name, getAccountId(pRequest->pTsc), pDbPrefixToken);
        //      if (ret != TSDB_CODE_SUCCESS) {
        //        return buildInvalidOperationMsg(pMsgBuf, msg1);
        //      }
H
Haojun Liao 已提交
92 93
      }

H
Haojun Liao 已提交
94 95 96 97 98 99
      // show table/stable like 'xxxx', set the like pattern for show tables
      SToken* pPattern = &pShowInfo->pattern;
      if (pPattern->type != 0) {
        if (pPattern->type == TK_ID && pPattern->z[0] == TS_ESCAPE_CHAR) {
          return buildInvalidOperationMsg(pMsgBuf, msg4);
        }
H
Haojun Liao 已提交
100

H
Haojun Liao 已提交
101 102 103 104
        pPattern->n = strdequote(pPattern->z);
        if (pPattern->n <= 0) {
          return buildInvalidOperationMsg(pMsgBuf, msg6);
        }
H
Haojun Liao 已提交
105

H
Haojun Liao 已提交
106 107 108 109 110 111 112 113 114
        if (pPattern->n > tsMaxWildCardsLen) {
          char tmp[64] = {0};
          sprintf(tmp, msg2, tsMaxWildCardsLen);
          return buildInvalidOperationMsg(pMsgBuf, tmp);
        }
      }
    } else if (showType == TSDB_MGMT_TABLE_VNODES) {
      if (pShowInfo->prefix.type == 0) {
        return buildInvalidOperationMsg(pMsgBuf, "No specified dnode ep");
H
Haojun Liao 已提交
115 116
      }

H
Haojun Liao 已提交
117 118
      if (pShowInfo->prefix.type == TK_STRING) {
        pShowInfo->prefix.n = strdequote(pShowInfo->prefix.z);
H
Haojun Liao 已提交
119 120 121
      }
    }

H
Haojun Liao 已提交
122
    *pEpSet = pCtx->mgmtEpSet;
123 124 125 126 127
    *output = buildShowMsg(pShowInfo, pCtx, pMsgBuf);
    if (*output == NULL) {
      return terrno;
    }

S
Shengliang Guan 已提交
128
    *outputLen = sizeof(SShowReq) /* + htons(pShowMsg->payloadLen)*/;
H
Haojun Liao 已提交
129 130 131 132 133 134
  }

  return TSDB_CODE_SUCCESS;
}

// can only perform the parameters based on the macro definitation
S
Shengliang Guan 已提交
135
static int32_t doCheckDbOptions(SCreateDbReq* pCreate, SMsgBuf* pMsgBuf) {
H
Haojun Liao 已提交
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
  char msg[512] = {0};

  if (pCreate->walLevel != -1 && (pCreate->walLevel < TSDB_MIN_WAL_LEVEL || pCreate->walLevel > TSDB_MAX_WAL_LEVEL)) {
    snprintf(msg, tListLen(msg), "invalid db option walLevel: %d, only 1-2 allowed", pCreate->walLevel);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  if (pCreate->replications != -1 &&
      (pCreate->replications < TSDB_MIN_DB_REPLICA_OPTION || pCreate->replications > TSDB_MAX_DB_REPLICA_OPTION)) {
    snprintf(msg, tListLen(msg), "invalid db option replications: %d valid range: [%d, %d]", pCreate->replications,
             TSDB_MIN_DB_REPLICA_OPTION, TSDB_MAX_DB_REPLICA_OPTION);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  int32_t blocks = ntohl(pCreate->totalBlocks);
  if (blocks != -1 && (blocks < TSDB_MIN_TOTAL_BLOCKS || blocks > TSDB_MAX_TOTAL_BLOCKS)) {
    snprintf(msg, tListLen(msg), "invalid db option totalBlocks: %d valid range: [%d, %d]", blocks,
             TSDB_MIN_TOTAL_BLOCKS, TSDB_MAX_TOTAL_BLOCKS);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  if (pCreate->quorum != -1 &&
      (pCreate->quorum < TSDB_MIN_DB_QUORUM_OPTION || pCreate->quorum > TSDB_MAX_DB_QUORUM_OPTION)) {
    snprintf(msg, tListLen(msg), "invalid db option quorum: %d valid range: [%d, %d]", pCreate->quorum,
             TSDB_MIN_DB_QUORUM_OPTION, TSDB_MAX_DB_QUORUM_OPTION);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  int32_t val = htonl(pCreate->daysPerFile);
  if (val != -1 && (val < TSDB_MIN_DAYS_PER_FILE || val > TSDB_MAX_DAYS_PER_FILE)) {
H
Hongze Cheng 已提交
166 167
    snprintf(msg, tListLen(msg), "invalid db option daysPerFile: %d valid range: [%d, %d]", val, TSDB_MIN_DAYS_PER_FILE,
             TSDB_MAX_DAYS_PER_FILE);
H
Haojun Liao 已提交
168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  val = htonl(pCreate->cacheBlockSize);
  if (val != -1 && (val < TSDB_MIN_CACHE_BLOCK_SIZE || val > TSDB_MAX_CACHE_BLOCK_SIZE)) {
    snprintf(msg, tListLen(msg), "invalid db option cacheBlockSize: %d valid range: [%d, %d]", val,
             TSDB_MIN_CACHE_BLOCK_SIZE, TSDB_MAX_CACHE_BLOCK_SIZE);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  if (pCreate->precision != TSDB_TIME_PRECISION_MILLI && pCreate->precision != TSDB_TIME_PRECISION_MICRO &&
      pCreate->precision != TSDB_TIME_PRECISION_NANO) {
    snprintf(msg, tListLen(msg), "invalid db option timePrecision: %d valid value: [%d, %d, %d]", pCreate->precision,
             TSDB_TIME_PRECISION_MILLI, TSDB_TIME_PRECISION_MICRO, TSDB_TIME_PRECISION_NANO);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  val = htonl(pCreate->commitTime);
  if (val != -1 && (val < TSDB_MIN_COMMIT_TIME || val > TSDB_MAX_COMMIT_TIME)) {
H
Hongze Cheng 已提交
187 188
    snprintf(msg, tListLen(msg), "invalid db option commitTime: %d valid range: [%d, %d]", val, TSDB_MIN_COMMIT_TIME,
             TSDB_MAX_COMMIT_TIME);
H
Haojun Liao 已提交
189 190 191 192 193
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  val = htonl(pCreate->fsyncPeriod);
  if (val != -1 && (val < TSDB_MIN_FSYNC_PERIOD || val > TSDB_MAX_FSYNC_PERIOD)) {
H
Hongze Cheng 已提交
194 195
    snprintf(msg, tListLen(msg), "invalid db option fsyncPeriod: %d valid range: [%d, %d]", val, TSDB_MIN_FSYNC_PERIOD,
             TSDB_MAX_FSYNC_PERIOD);
H
Haojun Liao 已提交
196 197 198 199 200 201 202 203 204 205
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

  if (pCreate->compression != -1 &&
      (pCreate->compression < TSDB_MIN_COMP_LEVEL || pCreate->compression > TSDB_MAX_COMP_LEVEL)) {
    snprintf(msg, tListLen(msg), "invalid db option compression: %d valid range: [%d, %d]", pCreate->compression,
             TSDB_MIN_COMP_LEVEL, TSDB_MAX_COMP_LEVEL);
    return buildInvalidOperationMsg(pMsgBuf, msg);
  }

H
Haojun Liao 已提交
206 207 208
  val = htonl(pCreate->numOfVgroups);
  if (val < TSDB_MIN_VNODES_PER_DB || val > TSDB_MAX_VNODES_PER_DB) {
    snprintf(msg, tListLen(msg), "invalid number of vgroups for DB:%d valid range: [%d, %d]", val,
H
Hongze Cheng 已提交
209
             TSDB_MIN_VNODES_PER_DB, TSDB_MAX_VNODES_PER_DB);
H
Haojun Liao 已提交
210 211
  }

H
Haojun Liao 已提交
212 213 214 215 216 217 218 219 220 221 222 223
  val = htonl(pCreate->maxRows);
  if (val < TSDB_MIN_MAX_ROW_FBLOCK || val > TSDB_MAX_MAX_ROW_FBLOCK) {
    snprintf(msg, tListLen(msg), "invalid number of max rows in file block for DB:%d valid range: [%d, %d]", val,
             TSDB_MIN_MAX_ROW_FBLOCK, TSDB_MAX_MAX_ROW_FBLOCK);
  }

  val = htonl(pCreate->minRows);
  if (val < TSDB_MIN_MIN_ROW_FBLOCK || val > TSDB_MAX_MIN_ROW_FBLOCK) {
    snprintf(msg, tListLen(msg), "invalid number of min rows in file block for DB:%d valid range: [%d, %d]", val,
             TSDB_MIN_MIN_ROW_FBLOCK, TSDB_MAX_MIN_ROW_FBLOCK);
  }

H
Haojun Liao 已提交
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
  return TSDB_CODE_SUCCESS;
}

static int32_t validateTableColumns(SArray* pFieldList, int32_t maxRowLength, int32_t maxColumns, SMsgBuf* pMsgBuf) {
  const char* msg2 = "row length exceeds max length";
  const char* msg3 = "duplicated column names";
  const char* msg4 = "invalid data type";
  const char* msg5 = "invalid binary/nchar column length";
  const char* msg6 = "invalid column name";
  const char* msg7 = "too many columns";
  const char* msg8 = "illegal number of columns";

  size_t numOfCols = taosArrayGetSize(pFieldList);
  if (numOfCols > maxColumns) {
    return buildInvalidOperationMsg(pMsgBuf, msg7);
  }

  int32_t rowLen = 0;
  for (int32_t i = 0; i < numOfCols; ++i) {
    TAOS_FIELD* pField = taosArrayGet(pFieldList, i);
    if (!isValidDataType(pField->type)) {
      return buildInvalidOperationMsg(pMsgBuf, msg4);
    }

    if (pField->bytes == 0) {
      return buildInvalidOperationMsg(pMsgBuf, msg5);
    }

    if ((pField->type == TSDB_DATA_TYPE_BINARY && (pField->bytes <= 0 || pField->bytes > TSDB_MAX_BINARY_LEN)) ||
        (pField->type == TSDB_DATA_TYPE_NCHAR && (pField->bytes <= 0 || pField->bytes > TSDB_MAX_NCHAR_LEN))) {
      return buildInvalidOperationMsg(pMsgBuf, msg5);
    }

    SToken nameToken = {.z = pField->name, .n = strlen(pField->name), .type = TK_ID};
    if (parserValidateNameToken(&nameToken) != TSDB_CODE_SUCCESS) {
      return buildInvalidOperationMsg(pMsgBuf, msg6);
    }

    // field name must be unique
    if (has(pFieldList, i + 1, pField->name) == true) {
      return buildInvalidOperationMsg(pMsgBuf, msg3);
    }

    rowLen += pField->bytes;
  }

  // max row length must be less than TSDB_MAX_BYTES_PER_ROW
  if (rowLen > maxRowLength) {
    return buildInvalidOperationMsg(pMsgBuf, msg2);
  }

  return TSDB_CODE_SUCCESS;
}

static int32_t validateTableColumnInfo(SArray* pFieldList, SMsgBuf* pMsgBuf) {
279
  assert(pFieldList != NULL && pMsgBuf != NULL);
H
Haojun Liao 已提交
280 281

  const char* msg1 = "first column must be timestamp";
282
  const char* msg2 = "illegal number of columns";
H
Haojun Liao 已提交
283 284 285 286 287 288 289 290 291 292

  // first column must be timestamp
  SField* pField = taosArrayGet(pFieldList, 0);
  if (pField->type != TSDB_DATA_TYPE_TIMESTAMP) {
    return buildInvalidOperationMsg(pMsgBuf, msg1);
  }

  // number of fields no less than 2
  size_t numOfCols = taosArrayGetSize(pFieldList);
  if (numOfCols <= 1) {
293
    return buildInvalidOperationMsg(pMsgBuf, msg2);
H
Haojun Liao 已提交
294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321
  }

  return validateTableColumns(pFieldList, TSDB_MAX_BYTES_PER_ROW, TSDB_MAX_COLUMNS, pMsgBuf);
}

static int32_t validateTagParams(SArray* pTagsList, SArray* pFieldList, SMsgBuf* pMsgBuf) {
  assert(pTagsList != NULL);

  const char* msg1 = "invalid number of tag columns";
  const char* msg3 = "duplicated column names";

  // number of fields at least 1
  size_t numOfTags = taosArrayGetSize(pTagsList);
  if (numOfTags < 1) {
    return buildInvalidOperationMsg(pMsgBuf, msg1);
  }

  // field name must be unique
  for (int32_t i = 0; i < numOfTags; ++i) {
    SField* p = taosArrayGet(pTagsList, i);
    if (has(pFieldList, 0, p->name) == true) {
      return buildInvalidOperationMsg(pMsgBuf, msg3);
    }
  }

  return validateTableColumns(pFieldList, TSDB_MAX_TAGS_LEN, TSDB_MAX_TAGS, pMsgBuf);
}

322
int32_t doCheckForCreateTable(SCreateTableSql* pCreateTable, SMsgBuf* pMsgBuf) {
H
Haojun Liao 已提交
323 324 325 326 327 328 329
  const char* msg1 = "invalid table name";

  SArray* pFieldList = pCreateTable->colInfo.pColumns;
  SArray* pTagList = pCreateTable->colInfo.pTagColumns;
  assert(pFieldList != NULL);

  // if sql specifies db, use it, otherwise use default db
330
  SToken* pNameToken = &(pCreateTable->name);
H
Haojun Liao 已提交
331

332
  if (parserValidateIdToken(pNameToken) != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
333 334 335 336 337 338 339 340 341 342 343
    return buildInvalidOperationMsg(pMsgBuf, msg1);
  }

  if (validateTableColumnInfo(pFieldList, pMsgBuf) != TSDB_CODE_SUCCESS ||
      (pTagList != NULL && validateTagParams(pTagList, pFieldList, pMsgBuf) != TSDB_CODE_SUCCESS)) {
    return TSDB_CODE_TSC_INVALID_OPERATION;
  }

  return TSDB_CODE_SUCCESS;
}

344 345 346 347 348
typedef struct SVgroupTablesBatch {
  SVCreateTbBatchReq req;
  SVgroupInfo        info;
} SVgroupTablesBatch;

349 350
static SArray* doSerializeVgroupCreateTableInfo(SHashObj* pVgroupHashmap);

H
Haojun Liao 已提交
351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373
static int32_t doParseSerializeTagValue(SSchema* pTagSchema, int32_t numOfInputTag, SKVRowBuilder* pKvRowBuilder,
                                        SArray* pTagValList, int32_t tsPrecision, SMsgBuf* pMsgBuf) {
  const char* msg1 = "illegal value or data overflow";
  int32_t     code = TSDB_CODE_SUCCESS;

  for (int32_t i = 0; i < numOfInputTag; ++i) {
    SSchema* pSchema = &pTagSchema[i];

    char* endPtr = NULL;
    char  tmpTokenBuf[TSDB_MAX_TAGS_LEN] = {0};
    SKvParam param = {.builder = pKvRowBuilder, .schema = pSchema};

    SToken* pItem = taosArrayGet(pTagValList, i);
    code = parseValueToken(&endPtr, pItem, pSchema, tsPrecision, tmpTokenBuf, KvRowAppend, &param, pMsgBuf);

    if (code != TSDB_CODE_SUCCESS) {
      return buildInvalidOperationMsg(pMsgBuf, msg1);
    }
  }

  return code;
}

374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
static void addCreateTbReqIntoVgroup(SHashObj* pVgroupHashmap, const SName* pTableName, SKVRow row, uint64_t suid, SVgroupInfo* pVgInfo) {
  struct SVCreateTbReq req = {0};
  req.type        = TD_CHILD_TABLE;
  req.name        = strdup(tNameGetTableName(pTableName));
  req.ctbCfg.suid = suid;
  req.ctbCfg.pTag = row;

  SVgroupTablesBatch* pTableBatch = taosHashGet(pVgroupHashmap, &pVgInfo->vgId, sizeof(pVgInfo->vgId));
  if (pTableBatch == NULL) {
    SVgroupTablesBatch tBatch = {0};
    tBatch.info = *pVgInfo;

    tBatch.req.pArray = taosArrayInit(4, sizeof(struct SVCreateTbReq));
    taosArrayPush(tBatch.req.pArray, &req);

    taosHashPut(pVgroupHashmap, &pVgInfo->vgId, sizeof(pVgInfo->vgId), &tBatch, sizeof(tBatch));
  } else {  // add to the correct vgroup
    assert(pVgInfo->vgId == pTableBatch->info.vgId);
    taosArrayPush(pTableBatch->req.pArray, &req);
  }
}

static void destroyCreateTbReqBatch(SVgroupTablesBatch* pTbBatch) {
  size_t size = taosArrayGetSize(pTbBatch->req.pArray);
  for(int32_t i = 0; i < size; ++i) {
    SVCreateTbReq* pTableReq = taosArrayGet(pTbBatch->req.pArray, i);
    tfree(pTableReq->name);
401 402 403 404 405 406 407 408

    if (pTableReq->type == TSDB_NORMAL_TABLE) {
      tfree(pTableReq->ntbCfg.pSchema);
    } else if (pTableReq->type == TSDB_CHILD_TABLE) {
      tfree(pTableReq->ctbCfg.pTag);
    } else {
      assert(0);
    }
409 410 411 412 413
  }

  taosArrayDestroy(pTbBatch->req.pArray);
}

H
Haojun Liao 已提交
414
static int32_t doCheckAndBuildCreateCTableReq(SCreateTableSql* pCreateTable, SParseContext* pCtx, SMsgBuf* pMsgBuf, SArray** pBufArray) {
H
Haojun Liao 已提交
415 416 417 418 419
  const char* msg1 = "invalid table name";
  const char* msg2 = "tags number not matched";
  const char* msg3 = "tag value too long";
  const char* msg4 = "illegal value or data overflow";

420 421 422
  int32_t code = 0;
  STableMeta* pSuperTableMeta = NULL;

423 424
  SHashObj* pVgroupHashmap = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_NO_LOCK);

H
Haojun Liao 已提交
425
  // super table name, create table by using dst
426
  size_t numOfTables = taosArrayGetSize(pCreateTable->childTableInfo);
H
Hongze Cheng 已提交
427
  for (int32_t j = 0; j < numOfTables; ++j) {
H
Haojun Liao 已提交
428 429 430
    SCreatedTableInfo* pCreateTableInfo = taosArrayGet(pCreateTable->childTableInfo, j);

    SToken* pSTableNameToken = &pCreateTableInfo->stbName;
431
    code = parserValidateNameToken(pSTableNameToken);
H
Haojun Liao 已提交
432
    if (code != TSDB_CODE_SUCCESS) {
433 434
      code = buildInvalidOperationMsg(pMsgBuf, msg1);
      goto _error;
H
Haojun Liao 已提交
435 436 437 438 439
    }

    SName name = {0};
    code = createSName(&name, pSTableNameToken, pCtx, pMsgBuf);
    if (code != TSDB_CODE_SUCCESS) {
440
      goto _error;
H
Haojun Liao 已提交
441 442
    }

H
Haojun Liao 已提交
443 444
    SKVRowBuilder kvRowBuilder = {0};
    if (tdInitKVRowBuilder(&kvRowBuilder) < 0) {
445 446
      code = TSDB_CODE_TSC_OUT_OF_MEMORY;
      goto _error;
H
Haojun Liao 已提交
447 448
    }

H
Haojun Liao 已提交
449
    SArray* pValList = pCreateTableInfo->pTagVals;
H
Haojun Liao 已提交
450
    size_t  numOfInputTag = taosArrayGetSize(pValList);
H
Haojun Liao 已提交
451

452 453
    code = catalogGetTableMeta(pCtx->pCatalog, pCtx->pTransporter, &pCtx->mgmtEpSet, &name, &pSuperTableMeta);
    if (code != TSDB_CODE_SUCCESS) {
454
      goto _error;
455 456
    }

457
    assert(pSuperTableMeta != NULL);
458

H
Haojun Liao 已提交
459
    // too long tag values will return invalid sql, not be truncated automatically
H
Hongze Cheng 已提交
460
    SSchema*      pTagSchema = getTableTagSchema(pSuperTableMeta);
H
Haojun Liao 已提交
461 462 463
    STableComInfo tinfo = getTableInfo(pSuperTableMeta);

    SArray* pNameList = NULL;
H
Haojun Liao 已提交
464
    size_t  numOfBoundTags = 0;
H
Haojun Liao 已提交
465 466 467 468
    int32_t schemaSize = getNumOfTags(pSuperTableMeta);

    if (pCreateTableInfo->pTagNames) {
      pNameList = pCreateTableInfo->pTagNames;
H
Haojun Liao 已提交
469
      numOfBoundTags = taosArrayGetSize(pNameList);
H
Haojun Liao 已提交
470

H
Haojun Liao 已提交
471
      if (numOfInputTag != numOfBoundTags || schemaSize < numOfInputTag) {
H
Haojun Liao 已提交
472
        tdDestroyKVRowBuilder(&kvRowBuilder);
473 474
        code = buildInvalidOperationMsg(pMsgBuf, msg2);
        goto _error;
H
Haojun Liao 已提交
475 476 477
      }

      bool findColumnIndex = false;
H
Haojun Liao 已提交
478
      for (int32_t i = 0; i < numOfBoundTags; ++i) {
H
Haojun Liao 已提交
479 480
        SToken* sToken = taosArrayGet(pNameList, i);

H
Hongze Cheng 已提交
481
        char tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0};  // create tmp buf to avoid alter orginal sqlstr
H
Haojun Liao 已提交
482 483 484
        strncpy(tmpTokenBuf, sToken->z, sToken->n);
        sToken->z = tmpTokenBuf;

H
Hongze Cheng 已提交
485 486 487
        //        if (TK_STRING == sToken->type) {
        //          tscDequoteAndTrimToken(sToken);
        //        }
H
Haojun Liao 已提交
488

H
Hongze Cheng 已提交
489 490 491
        //        if (TK_ID == sToken->type) {
        //          tscRmEscapeAndTrimToken(sToken);
        //        }
H
Haojun Liao 已提交
492

493
        SListItem* pItem = taosArrayGet(pValList, i);
H
Haojun Liao 已提交
494 495 496 497 498 499

        findColumnIndex = false;

        // todo speedup by using hash list
        for (int32_t t = 0; t < schemaSize; ++t) {
          if (strncmp(sToken->z, pTagSchema[t].name, sToken->n) == 0 && strlen(pTagSchema[t].name) == sToken->n) {
H
Hongze Cheng 已提交
500
            SSchema* pSchema = &pTagSchema[t];
H
Haojun Liao 已提交
501 502 503 504 505

            char tagVal[TSDB_MAX_TAGS_LEN] = {0};
            if (pSchema->type == TSDB_DATA_TYPE_BINARY || pSchema->type == TSDB_DATA_TYPE_NCHAR) {
              if (pItem->pVar.nLen > pSchema->bytes) {
                tdDestroyKVRowBuilder(&kvRowBuilder);
506 507
                code = buildInvalidOperationMsg(pMsgBuf, msg3);
                goto _error;
H
Haojun Liao 已提交
508 509 510
              }
            } else if (pSchema->type == TSDB_DATA_TYPE_TIMESTAMP) {
              if (pItem->pVar.nType == TSDB_DATA_TYPE_BINARY) {
H
Hongze Cheng 已提交
511 512 513 514
                //                code = convertTimestampStrToInt64(&(pItem->pVar), tinfo.precision);
                //                if (code != TSDB_CODE_SUCCESS) {
                //                  return buildInvalidOperationMsg(pMsgBuf, msg4);
                //                }
H
Haojun Liao 已提交
515
              } else if (pItem->pVar.nType == TSDB_DATA_TYPE_TIMESTAMP) {
516
                pItem->pVar.i = convertTimePrecision(pItem->pVar.i, TSDB_TIME_PRECISION_NANO, tinfo.precision);
H
Haojun Liao 已提交
517 518 519
              }
            }

520
            code = taosVariantDump(&(pItem->pVar), tagVal, pSchema->type, true);
H
Haojun Liao 已提交
521 522

            // check again after the convert since it may be converted from binary to nchar.
523
            if (IS_VAR_DATA_TYPE(pSchema->type)) {
H
Haojun Liao 已提交
524 525 526
              int16_t len = varDataTLen(tagVal);
              if (len > pSchema->bytes) {
                tdDestroyKVRowBuilder(&kvRowBuilder);
527 528
                code = buildInvalidOperationMsg(pMsgBuf, msg3);
                goto _error;
H
Haojun Liao 已提交
529 530 531 532 533
              }
            }

            if (code != TSDB_CODE_SUCCESS) {
              tdDestroyKVRowBuilder(&kvRowBuilder);
534 535
              code = buildInvalidOperationMsg(pMsgBuf, msg4);
              goto _error;
H
Haojun Liao 已提交
536 537 538 539 540 541 542 543 544 545 546
            }

            tdAddColToKVRow(&kvRowBuilder, pSchema->colId, pSchema->type, tagVal);

            findColumnIndex = true;
            break;
          }
        }

        if (!findColumnIndex) {
          tdDestroyKVRowBuilder(&kvRowBuilder);
H
Hongze Cheng 已提交
547
          //          return buildInvalidOperationMsg(pMsgBuf, "invalid tag name", sToken->z);
H
Haojun Liao 已提交
548 549 550
        }
      }
    } else {
551
      if (schemaSize != numOfInputTag) {
H
Haojun Liao 已提交
552
        tdDestroyKVRowBuilder(&kvRowBuilder);
553 554
        code = buildInvalidOperationMsg(pMsgBuf, msg2);
        goto _error;
H
Haojun Liao 已提交
555 556
      }

H
Haojun Liao 已提交
557 558
      code = doParseSerializeTagValue(pTagSchema, numOfInputTag, &kvRowBuilder, pValList, tinfo.precision, pMsgBuf);
      if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
559
        tdDestroyKVRowBuilder(&kvRowBuilder);
560
        goto _error;
H
Haojun Liao 已提交
561 562 563 564 565 566
      }
    }

    SKVRow row = tdGetKVRowFromBuilder(&kvRowBuilder);
    tdDestroyKVRowBuilder(&kvRowBuilder);
    if (row == NULL) {
567 568
      code = TSDB_CODE_QRY_OUT_OF_MEMORY;
      goto _error;
H
Haojun Liao 已提交
569
    }
570

H
Haojun Liao 已提交
571 572
    tdSortKVRowByColIdx(row);

573 574 575
    SName tableName = {0};
    code = createSName(&tableName, &pCreateTableInfo->name, pCtx, pMsgBuf);
    if (code != TSDB_CODE_SUCCESS) {
576
      goto _error;
H
Haojun Liao 已提交
577 578
    }

579
    // Find a appropriate vgroup to accommodate this table , according to the table name
580
    SVgroupInfo info = {0};
581 582 583 584
    code = catalogGetTableHashVgroup(pCtx->pCatalog, pCtx->pTransporter, &pCtx->mgmtEpSet, &tableName, &info);
    if (code != TSDB_CODE_SUCCESS) {
      goto _error;
    }
585

586 587 588
    addCreateTbReqIntoVgroup(pVgroupHashmap, &tableName, row, pSuperTableMeta->uid, &info);
    tfree(pSuperTableMeta);
  }
589

590
  *pBufArray = doSerializeVgroupCreateTableInfo(pVgroupHashmap);
591 592 593 594
  if (*pBufArray == NULL) {
    code = terrno;
    goto _error;
  }
595 596 597

  taosHashCleanup(pVgroupHashmap);
  return TSDB_CODE_SUCCESS;
598 599 600 601 602 603

  _error:
  taosHashCleanup(pVgroupHashmap);
  tfree(pSuperTableMeta);
  terrno = code;
  return code;
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 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651
}

static int32_t serializeVgroupTablesBatchImpl(SVgroupTablesBatch* pTbBatch, SArray* pBufArray) {
  int   tlen = sizeof(SMsgHead) + tSVCreateTbBatchReqSerialize(NULL, &(pTbBatch->req));
  void* buf = malloc(tlen);
  if (buf == NULL) {
    // TODO: handle error
  }

  ((SMsgHead*)buf)->vgId = htonl(pTbBatch->info.vgId);
  ((SMsgHead*)buf)->contLen = htonl(tlen);

  void* pBuf = POINTER_SHIFT(buf, sizeof(SMsgHead));
  tSVCreateTbBatchReqSerialize(&pBuf, &(pTbBatch->req));

  SVgDataBlocks* pVgData = calloc(1, sizeof(SVgDataBlocks));
  pVgData->vg    = pTbBatch->info;
  pVgData->pData = buf;
  pVgData->size  = tlen;
  pVgData->numOfTables = (int32_t) taosArrayGetSize(pTbBatch->req.pArray);

  taosArrayPush(pBufArray, &pVgData);
}

static int32_t doBuildSingleTableBatchReq(SName* pTableName, SArray* pColumns, SVgroupInfo* pVgroupInfo, SVgroupTablesBatch* pBatch) {
  struct SVCreateTbReq req = {0};
  req.type = TD_NORMAL_TABLE;
  req.name = strdup(tNameGetTableName(pTableName));

  req.ntbCfg.nCols = taosArrayGetSize(pColumns);
  int32_t num = req.ntbCfg.nCols;

  req.ntbCfg.pSchema = calloc(num, sizeof(SSchema));
  for(int32_t i = 0; i < num; ++i) {
    SSchema* pSchema = taosArrayGet(pColumns, i);
    memcpy(&req.ntbCfg.pSchema[i], pSchema, sizeof(SSchema));
  }

  pBatch->info = *pVgroupInfo;
  pBatch->req.pArray = taosArrayInit(1, sizeof(struct SVCreateTbReq));
  if (pBatch->req.pArray == NULL) {
    return TSDB_CODE_QRY_OUT_OF_MEMORY;
  }

  taosArrayPush(pBatch->req.pArray, &req);
  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
652
int32_t doCheckAndBuildCreateTableReq(SCreateTableSql* pCreateTable, SParseContext* pCtx, SMsgBuf* pMsgBuf, char** pOutput, int32_t* len) {
653
  SArray* pBufArray = NULL;
H
Haojun Liao 已提交
654
  int32_t code = 0;
655 656 657 658

  // it is a sql statement to create a normal table
  if (pCreateTable->childTableInfo == NULL) {
    assert(taosArrayGetSize(pCreateTable->colInfo.pColumns) > 0 && pCreateTable->colInfo.pTagColumns == NULL);
H
Haojun Liao 已提交
659
    code = doCheckForCreateTable(pCreateTable, pMsgBuf);
660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }

    SName tableName = {0};
    code = createSName(&tableName, &pCreateTable->name, pCtx, pMsgBuf);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }

    SVgroupInfo info = {0};
    catalogGetTableHashVgroup(pCtx->pCatalog, pCtx->pTransporter, &pCtx->mgmtEpSet, &tableName, &info);

    SVgroupTablesBatch tbatch = {0};
    code = doBuildSingleTableBatchReq(&tableName, pCreateTable->colInfo.pColumns, &info, &tbatch);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }

    pBufArray = taosArrayInit(1, POINTER_BYTES);
    if (pBufArray == NULL) {
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }

    serializeVgroupTablesBatchImpl(&tbatch, pBufArray);
    destroyCreateTbReqBatch(&tbatch);
  } else { // it is a child table, created according to a super table
H
Haojun Liao 已提交
687
    code = doCheckAndBuildCreateCTableReq(pCreateTable, pCtx, pMsgBuf, &pBufArray);
688
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
689 690
      return code;
    }
691
  }
692

693 694 695
  SVnodeModifOpStmtInfo* pStmtInfo = calloc(1, sizeof(SVnodeModifOpStmtInfo));
  pStmtInfo->nodeType    = TSDB_SQL_CREATE_TABLE;
  pStmtInfo->pDataBlocks = pBufArray;
696

697 698
  *pOutput = (char*) pStmtInfo;
  *len     = sizeof(SVnodeModifOpStmtInfo);
H
Haojun Liao 已提交
699

700 701
  return TSDB_CODE_SUCCESS;
}
H
Haojun Liao 已提交
702

703
SArray* doSerializeVgroupCreateTableInfo(SHashObj* pVgroupHashmap) {
704 705 706
  SArray* pBufArray = taosArrayInit(taosHashGetSize(pVgroupHashmap), sizeof(void*));

  SVgroupTablesBatch* pTbBatch = NULL;
H
Hongze Cheng 已提交
707
  do {
708
    pTbBatch = taosHashIterate(pVgroupHashmap, pTbBatch);
709 710 711
    if (pTbBatch == NULL) {
      break;
    }
H
Hongze Cheng 已提交
712

713
    /*int32_t code = */serializeVgroupTablesBatchImpl(pTbBatch, pBufArray);
714
    destroyCreateTbReqBatch(pTbBatch);
H
Hongze Cheng 已提交
715
  } while (true);
716

717
  return pBufArray;
H
Haojun Liao 已提交
718 719
}

H
Haojun Liao 已提交
720
SDclStmtInfo* qParserValidateDclSqlNode(SSqlInfo* pInfo, SParseContext* pCtx, char* msgBuf, int32_t msgBufLen) {
H
Haojun Liao 已提交
721 722
  int32_t code = 0;

723 724
  SDclStmtInfo* pDcl = calloc(1, sizeof(SDclStmtInfo));

H
Hongze Cheng 已提交
725 726
  SMsgBuf  m = {.buf = msgBuf, .len = msgBufLen};
  SMsgBuf* pMsgBuf = &m;
H
Haojun Liao 已提交
727

H
Haojun Liao 已提交
728 729
  pDcl->epSet = pCtx->mgmtEpSet;

H
Haojun Liao 已提交
730 731 732 733 734 735 736 737 738 739 740 741 742
  switch (pInfo->type) {
    case TSDB_SQL_CREATE_USER:
    case TSDB_SQL_ALTER_USER: {
      const char* msg1 = "not support options";
      const char* msg2 = "invalid user/account name";
      const char* msg3 = "name too long";
      const char* msg4 = "invalid user rights";

      SUserInfo* pUser = &pInfo->pMiscInfo->user;
      SToken*    pName = &pUser->user;
      SToken*    pPwd = &pUser->passwd;

      if (pName->n >= TSDB_USER_LEN) {
743 744
        code = buildInvalidOperationMsg(pMsgBuf, msg3);
        goto _error;
H
Haojun Liao 已提交
745 746 747
      }

      if (parserValidateIdToken(pName) != TSDB_CODE_SUCCESS) {
748 749
        code = buildInvalidOperationMsg(pMsgBuf, msg2);
        goto _error;
H
Haojun Liao 已提交
750 751 752 753
      }

      if (pInfo->type == TSDB_SQL_CREATE_USER) {
        if (parserValidatePassword(pPwd, pMsgBuf) != TSDB_CODE_SUCCESS) {
754 755
          code = TSDB_CODE_TSC_INVALID_OPERATION;
          goto _error;
H
Haojun Liao 已提交
756 757 758 759
        }
      } else {
        if (pUser->type == TSDB_ALTER_USER_PASSWD) {
          if (parserValidatePassword(pPwd, pMsgBuf) != TSDB_CODE_SUCCESS) {
760 761
            code = TSDB_CODE_TSC_INVALID_OPERATION;
            goto _error;
H
Haojun Liao 已提交
762 763 764 765 766 767 768 769 770 771
          }
        } else if (pUser->type == TSDB_ALTER_USER_PRIVILEGES) {
          assert(pPwd->type == TSDB_DATA_TYPE_NULL);

          SToken* pPrivilege = &pUser->privilege;
          if (strncasecmp(pPrivilege->z, "super", 5) == 0 && pPrivilege->n == 5) {
            //            pCmd->count = 1;
          } else if (strncasecmp(pPrivilege->z, "normal", 4) == 0 && pPrivilege->n == 4) {
            //            pCmd->count = 2;
          } else {
772 773
            code = buildInvalidOperationMsg(pMsgBuf, msg4);
            goto _error;
H
Haojun Liao 已提交
774 775
          }
        } else {
776 777
          code = buildInvalidOperationMsg(pMsgBuf, msg1);
          goto _error;
H
Haojun Liao 已提交
778 779 780 781
        }
      }

      pDcl->pMsg = (char*)buildUserManipulationMsg(pInfo, &pDcl->msgLen, pCtx->requestId, msgBuf, msgBufLen);
H
Hongze Cheng 已提交
782
      pDcl->msgType = (pInfo->type == TSDB_SQL_CREATE_USER) ? TDMT_MND_CREATE_USER : TDMT_MND_ALTER_USER;
H
Haojun Liao 已提交
783 784 785 786 787 788 789 790 791 792 793 794 795
      break;
    }

    case TSDB_SQL_CREATE_ACCT:
    case TSDB_SQL_ALTER_ACCT: {
      const char* msg1 = "invalid state option, available options[no, r, w, all]";
      const char* msg2 = "invalid user/account name";
      const char* msg3 = "name too long";

      SToken* pName = &pInfo->pMiscInfo->user.user;
      SToken* pPwd = &pInfo->pMiscInfo->user.passwd;

      if (parserValidatePassword(pPwd, pMsgBuf) != TSDB_CODE_SUCCESS) {
796 797
        code = TSDB_CODE_TSC_INVALID_OPERATION;
        goto _error;
H
Haojun Liao 已提交
798 799 800
      }

      if (pName->n >= TSDB_USER_LEN) {
801 802
        code = buildInvalidOperationMsg(pMsgBuf, msg3);
        goto _error;
H
Haojun Liao 已提交
803 804 805
      }

      if (parserValidateNameToken(pName) != TSDB_CODE_SUCCESS) {
806 807
        code = buildInvalidOperationMsg(pMsgBuf, msg2);
        goto _error;
H
Haojun Liao 已提交
808 809 810 811 812 813 814 815 816
      }

      SCreateAcctInfo* pAcctOpt = &pInfo->pMiscInfo->acctOpt;
      if (pAcctOpt->stat.n > 0) {
        if (pAcctOpt->stat.z[0] == 'r' && pAcctOpt->stat.n == 1) {
        } else if (pAcctOpt->stat.z[0] == 'w' && pAcctOpt->stat.n == 1) {
        } else if (strncmp(pAcctOpt->stat.z, "all", 3) == 0 && pAcctOpt->stat.n == 3) {
        } else if (strncmp(pAcctOpt->stat.z, "no", 2) == 0 && pAcctOpt->stat.n == 2) {
        } else {
817 818
          code = buildInvalidOperationMsg(pMsgBuf, msg1);
          goto _error;
H
Haojun Liao 已提交
819 820 821 822
        }
      }

      pDcl->pMsg = (char*)buildAcctManipulationMsg(pInfo, &pDcl->msgLen, pCtx->requestId, msgBuf, msgBufLen);
H
Hongze Cheng 已提交
823
      pDcl->msgType = (pInfo->type == TSDB_SQL_CREATE_ACCT) ? TDMT_MND_CREATE_ACCT : TDMT_MND_ALTER_ACCT;
H
Haojun Liao 已提交
824 825 826 827 828 829
      break;
    }

    case TSDB_SQL_DROP_ACCT:
    case TSDB_SQL_DROP_USER: {
      pDcl->pMsg = (char*)buildDropUserMsg(pInfo, &pDcl->msgLen, pCtx->requestId, msgBuf, msgBufLen);
H
Hongze Cheng 已提交
830
      pDcl->msgType = (pInfo->type == TSDB_SQL_DROP_ACCT) ? TDMT_MND_DROP_ACCT : TDMT_MND_DROP_USER;
H
Haojun Liao 已提交
831 832 833 834
      break;
    }

    case TSDB_SQL_SHOW: {
H
Haojun Liao 已提交
835
      SShowInfo* pShowInfo = &pInfo->pMiscInfo->showOpt;
836
      code = setShowInfo(pShowInfo, pCtx, (void**)&pDcl->pMsg, &pDcl->msgLen, &pDcl->epSet, &pDcl->pExtension, pMsgBuf);
837 838 839 840
      if (code != TSDB_CODE_SUCCESS) {
        goto _error;
      }
      
H
Hongze Cheng 已提交
841
      pDcl->msgType = (pShowInfo->showType == TSDB_MGMT_TABLE_TABLE) ? TDMT_VND_SHOW_TABLES : TDMT_MND_SHOW;
H
Haojun Liao 已提交
842 843 844 845 846 847 848 849
      break;
    }

    case TSDB_SQL_USE_DB: {
      const char* msg = "invalid db name";

      SToken* pToken = taosArrayGet(pInfo->pMiscInfo->a, 0);
      if (parserValidateNameToken(pToken) != TSDB_CODE_SUCCESS) {
850 851
        code = buildInvalidOperationMsg(pMsgBuf, msg);
        goto _error;
H
Haojun Liao 已提交
852 853
      }

H
Hongze Cheng 已提交
854
      SName   n = {0};
H
Haojun Liao 已提交
855 856
      int32_t ret = tNameSetDbName(&n, pCtx->acctId, pToken->z, pToken->n);
      if (ret != TSDB_CODE_SUCCESS) {
857 858
        code = buildInvalidOperationMsg(pMsgBuf, msg);
        goto _error;
H
Haojun Liao 已提交
859 860
      }

S
Shengliang Guan 已提交
861
      SUseDbReq* pUseDbMsg = (SUseDbReq*)calloc(1, sizeof(SUseDbReq));
H
Haojun Liao 已提交
862 863 864
      tNameExtractFullName(&n, pUseDbMsg->db);

      pDcl->pMsg = (char*)pUseDbMsg;
S
Shengliang Guan 已提交
865
      pDcl->msgLen = sizeof(SUseDbReq);
H
Haojun Liao 已提交
866 867 868 869 870 871 872 873 874 875 876
      pDcl->msgType = TDMT_MND_USE_DB;
      break;
    }

    case TSDB_SQL_ALTER_DB:
    case TSDB_SQL_CREATE_DB: {
      const char* msg1 = "invalid db name";
      const char* msg2 = "name too long";

      SCreateDbInfo* pCreateDB = &(pInfo->pMiscInfo->dbOpt);
      if (pCreateDB->dbname.n >= TSDB_DB_NAME_LEN) {
877 878
        code = buildInvalidOperationMsg(pMsgBuf, msg2);
        goto _error;
H
Haojun Liao 已提交
879 880
      }

H
Hongze Cheng 已提交
881
      char   buf[TSDB_DB_NAME_LEN] = {0};
H
Haojun Liao 已提交
882 883 884
      SToken token = taosTokenDup(&pCreateDB->dbname, buf, tListLen(buf));

      if (parserValidateNameToken(&token) != TSDB_CODE_SUCCESS) {
885 886
        code = buildInvalidOperationMsg(pMsgBuf, msg1);
        goto _error;
H
Haojun Liao 已提交
887 888
      }

S
Shengliang Guan 已提交
889
      SCreateDbReq* pCreateMsg = buildCreateDbMsg(pCreateDB, pCtx, pMsgBuf);
H
Haojun Liao 已提交
890
      if (doCheckDbOptions(pCreateMsg, pMsgBuf) != TSDB_CODE_SUCCESS) {
891 892
        code = TSDB_CODE_TSC_INVALID_OPERATION;
        goto _error;
H
Haojun Liao 已提交
893 894 895
      }

      pDcl->pMsg = (char*)pCreateMsg;
S
Shengliang Guan 已提交
896
      pDcl->msgLen = sizeof(SCreateDbReq);
H
Hongze Cheng 已提交
897
      pDcl->msgType = (pInfo->type == TSDB_SQL_CREATE_DB) ? TDMT_MND_CREATE_DB : TDMT_MND_ALTER_DB;
H
Haojun Liao 已提交
898 899 900 901 902 903 904 905 906 907 908 909
      break;
    }

    case TSDB_SQL_DROP_DB: {
      const char* msg1 = "invalid database name";

      assert(taosArrayGetSize(pInfo->pMiscInfo->a) == 1);
      SToken* dbName = taosArrayGet(pInfo->pMiscInfo->a, 0);

      SName name = {0};
      code = tNameSetDbName(&name, pCtx->acctId, dbName->z, dbName->n);
      if (code != TSDB_CODE_SUCCESS) {
910 911
        code = buildInvalidOperationMsg(pMsgBuf, msg1);
        goto _error;
H
Haojun Liao 已提交
912 913
      }

S
Shengliang Guan 已提交
914
      SDropDbReq* pDropDbMsg = (SDropDbReq*)calloc(1, sizeof(SDropDbReq));
H
Haojun Liao 已提交
915 916 917 918 919 920

      code = tNameExtractFullName(&name, pDropDbMsg->db);
      pDropDbMsg->ignoreNotExists = pInfo->pMiscInfo->existsCheck ? 1 : 0;
      assert(code == TSDB_CODE_SUCCESS && name.type == TSDB_DB_NAME_T);

      pDcl->msgType = TDMT_MND_DROP_DB;
S
Shengliang Guan 已提交
921
      pDcl->msgLen = sizeof(SDropDbReq);
H
Haojun Liao 已提交
922
      pDcl->pMsg = (char*)pDropDbMsg;
923
      break;
H
Haojun Liao 已提交
924 925
    }

H
Haojun Liao 已提交
926
    case TSDB_SQL_CREATE_STABLE: {
H
Haojun Liao 已提交
927
      SCreateTableSql* pCreateTable = pInfo->pCreateTableInfo;
928 929 930
      if ((code = doCheckForCreateTable(pCreateTable, pMsgBuf)) != TSDB_CODE_SUCCESS) {
        terrno = code;
        goto _error;
H
Haojun Liao 已提交
931 932
      }

933 934
      pDcl->pMsg = (char*)buildCreateStbMsg(pCreateTable, &pDcl->msgLen, pCtx, pMsgBuf);
      pDcl->msgType = TDMT_MND_CREATE_STB;
H
Haojun Liao 已提交
935 936 937 938
      break;
    }

    case TSDB_SQL_DROP_TABLE: {
H
Haojun Liao 已提交
939
      pDcl->pMsg = (char*)buildDropStableMsg(pInfo, &pDcl->msgLen, pCtx, pMsgBuf);
H
Haojun Liao 已提交
940
      if (pDcl->pMsg == NULL) {
941
        goto _error;
H
Haojun Liao 已提交
942 943 944
      }

      pDcl->msgType = TDMT_MND_DROP_STB;
945 946 947 948
      break;
    }

    case TSDB_SQL_CREATE_DNODE: {
H
Hongze Cheng 已提交
949
      pDcl->pMsg = (char*)buildCreateDnodeMsg(pInfo, &pDcl->msgLen, pMsgBuf);
950
      if (pDcl->pMsg == NULL) {
951
        goto _error;
952 953 954 955 956 957 958
      }

      pDcl->msgType = TDMT_MND_CREATE_DNODE;
      break;
    }

    case TSDB_SQL_DROP_DNODE: {
H
Hongze Cheng 已提交
959
      pDcl->pMsg = (char*)buildDropDnodeMsg(pInfo, &pDcl->msgLen, pMsgBuf);
960
      if (pDcl->pMsg == NULL) {
961
        goto _error;
962 963 964
      }

      pDcl->msgType = TDMT_MND_DROP_DNODE;
H
Haojun Liao 已提交
965 966 967 968 969 970 971
      break;
    }

    default:
      break;
  }

972 973 974 975 976 977
  return pDcl;

  _error:
    terrno = code;
    tfree(pDcl);
    return NULL;
H
Haojun Liao 已提交
978
}
979

H
Haojun Liao 已提交
980
SVnodeModifOpStmtInfo* qParserValidateCreateTbSqlNode(SSqlInfo* pInfo, SParseContext* pCtx, char* msgBuf, int32_t msgBufLen) {
981
  SCreateTableSql* pCreateTable = pInfo->pCreateTableInfo;
982
  assert(pCreateTable->type == TSDB_SQL_CREATE_TABLE);
983 984 985 986

  SMsgBuf  m = {.buf = msgBuf, .len = msgBufLen};
  SMsgBuf* pMsgBuf = &m;

987
  SVnodeModifOpStmtInfo* pModifSqlStmt = NULL;
988 989

  int32_t msgLen = 0;
990
  int32_t code = doCheckAndBuildCreateTableReq(pCreateTable, pCtx, pMsgBuf, (char**) &pModifSqlStmt, &msgLen);
991
  if (code != TSDB_CODE_SUCCESS) {
992
    terrno = code;
993
    tfree(pModifSqlStmt);
994 995 996
    return NULL;
  }

997
  return pModifSqlStmt;
998
}