parTranslater.c 334.1 KB
Newer Older
1
/*
wmmhello's avatar
wmmhello 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14
 * 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/>.
 */
15

X
Xiaoyu Wang 已提交
16
#include "parInt.h"
17

X
Xiaoyu Wang 已提交
18
#include "catalog.h"
19
#include "cmdnodes.h"
X
Xiaoyu Wang 已提交
20
#include "filter.h"
21
#include "functionMgt.h"
X
Xiaoyu Wang 已提交
22
#include "parUtil.h"
23
#include "scalar.h"
24
#include "systable.h"
X
Xiaoyu Wang 已提交
25
#include "tglobal.h"
26
#include "ttime.h"
27

X
Xiaoyu Wang 已提交
28
#define generateDealNodeErrMsg(pCxt, code, ...) \
wmmhello's avatar
wmmhello 已提交
29
  (pCxt->errCode = generateSyntaxErrMsg(&pCxt->msgBuf, code, ##__VA_ARGS__), DEAL_RES_ERROR)
X
Xiaoyu Wang 已提交
30

X
Xiaoyu Wang 已提交
31 32
#define SYSTABLE_SHOW_TYPE_OFFSET QUERY_NODE_SHOW_DNODES_STMT

D
dapan1121 已提交
33 34 35 36 37
typedef struct SRewriteTbNameContext {
  int32_t          errCode;
  char*            pTbName;
} SRewriteTbNameContext;

38
typedef struct STranslateContext {
wmmhello's avatar
wmmhello 已提交
39 40 41 42 43
  SParseContext*   pParseCxt;
  int32_t          errCode;
  SMsgBuf          msgBuf;
  SArray*          pNsLevel;  // element is SArray*, the element of this subarray is STableNode*
  int32_t          currLevel;
44
  int32_t          levelNo;
wmmhello's avatar
wmmhello 已提交
45
  ESqlClause       currClause;
46
  SNode*           pCurrStmt;
wmmhello's avatar
wmmhello 已提交
47 48 49
  SCmdMsgInfo*     pCmdMsg;
  SHashObj*        pDbs;
  SHashObj*        pTables;
D
dapan1121 已提交
50
  SHashObj*        pTargetTables;
wmmhello's avatar
wmmhello 已提交
51
  SExplainOptions* pExplainOpt;
52
  SParseMetaCache* pMetaCache;
X
Xiaoyu Wang 已提交
53
  bool             createStream;
54
  bool             stableQuery;
X
Xiaoyu Wang 已提交
55
  bool             showRewrite;
56 57
} STranslateContext;

58 59 60 61 62 63 64
typedef struct SBuildTopicContext {
  bool        colExists;
  bool        colNotFound;
  STableMeta* pMeta;
  SNodeList*  pTags;
} SBuildTopicContext;

X
Xiaoyu Wang 已提交
65
typedef struct SFullDatabaseName {
wmmhello's avatar
wmmhello 已提交
66
  char fullDbName[TSDB_DB_FNAME_LEN];
X
Xiaoyu Wang 已提交
67 68
} SFullDatabaseName;

X
Xiaoyu Wang 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
typedef struct SSysTableShowAdapter {
  ENodeType   showType;
  const char* pDbName;
  const char* pTableName;
  int32_t     numOfShowCols;
  const char* pShowCols[2];
} SSysTableShowAdapter;

// clang-format off
static const SSysTableShowAdapter sysTableShowAdapter[] = {
  {
    .showType = QUERY_NODE_SHOW_DNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_DNODES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
84
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
85 86 87 88 89 90
  },
  {
    .showType = QUERY_NODE_SHOW_MNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_MNODES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
91
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
92 93 94 95 96 97
  },
  {
    .showType = QUERY_NODE_SHOW_MODULES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_MODULES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
98
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
99 100 101 102 103 104
  },
  {
    .showType = QUERY_NODE_SHOW_QNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_QNODES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
105
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
106 107 108 109 110 111
  },
  {
    .showType = QUERY_NODE_SHOW_SNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_SNODES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
112
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
113 114 115 116 117 118
  },
  {
    .showType = QUERY_NODE_SHOW_BNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_BNODES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
119
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
120 121 122 123 124 125
  },
  {
    .showType = QUERY_NODE_SHOW_CLUSTER_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_CLUSTER,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
126
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
  },
  {
    .showType = QUERY_NODE_SHOW_DATABASES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_DATABASES,
    .numOfShowCols = 1,
    .pShowCols = {"name"}
  },
  {
    .showType = QUERY_NODE_SHOW_FUNCTIONS_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_FUNCTIONS,
    .numOfShowCols = 1,
    .pShowCols = {"name"}
  },
  {
    .showType = QUERY_NODE_SHOW_INDEXES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_INDEXES,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
147
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
148 149 150 151 152 153 154 155 156 157
  },
  {
    .showType = QUERY_NODE_SHOW_STABLES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_STABLES,
    .numOfShowCols = 1,
    .pShowCols = {"stable_name"}
  },
  {
    .showType = QUERY_NODE_SHOW_STREAMS_STMT,
158 159
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_STREAMS,
X
Xiaoyu Wang 已提交
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
    .numOfShowCols = 1,
    .pShowCols = {"stream_name"}
  },
  {
    .showType = QUERY_NODE_SHOW_TABLES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_TABLES,
    .numOfShowCols = 1,
    .pShowCols = {"table_name"}
  },
  {
    .showType = QUERY_NODE_SHOW_TAGS_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_TAGS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_USERS_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_USERS,
    .numOfShowCols = 1,
X
Xiaoyu Wang 已提交
182
    .pShowCols = {"*"}
X
Xiaoyu Wang 已提交
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
  },
  {
    .showType = QUERY_NODE_SHOW_LICENCES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_LICENCES,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_VGROUPS_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_VGROUPS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_TOPICS_STMT,
200 201
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_TOPICS,
X
Xiaoyu Wang 已提交
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 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
    .numOfShowCols = 1,
    .pShowCols = {"topic_name"}
  },
  {
    .showType = QUERY_NODE_SHOW_CONSUMERS_STMT,
    .pDbName = TSDB_PERFORMANCE_SCHEMA_DB,
    .pTableName = TSDB_PERFS_TABLE_CONSUMERS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_CONNECTIONS_STMT,
    .pDbName = TSDB_PERFORMANCE_SCHEMA_DB,
    .pTableName = TSDB_PERFS_TABLE_CONNECTIONS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_QUERIES_STMT,
    .pDbName = TSDB_PERFORMANCE_SCHEMA_DB,
    .pTableName = TSDB_PERFS_TABLE_QUERIES,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_APPS_STMT,
    .pDbName = TSDB_PERFORMANCE_SCHEMA_DB,
    .pTableName = TSDB_PERFS_TABLE_APPS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_VARIABLES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_CONFIGS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_DNODE_VARIABLES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_DNODE_VARIABLES,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_TRANSACTIONS_STMT,
    .pDbName = TSDB_PERFORMANCE_SCHEMA_DB,
    .pTableName = TSDB_PERFS_TABLE_TRANS,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
  {
    .showType = QUERY_NODE_SHOW_SUBSCRIPTIONS_STMT,
256 257
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_SUBSCRIPTIONS,
X
Xiaoyu Wang 已提交
258 259 260
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
261 262 263 264 265 266
  { .showType = QUERY_NODE_SHOW_VNODES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_VNODES,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
267 268 269 270 271 272
  { .showType = QUERY_NODE_SHOW_USER_PRIVILEGES_STMT,
    .pDbName = TSDB_INFORMATION_SCHEMA_DB,
    .pTableName = TSDB_INS_TABLE_USER_PRIVILEGES,
    .numOfShowCols = 1,
    .pShowCols = {"*"}
  },
X
Xiaoyu Wang 已提交
273 274 275
};
// clang-format on

X
Xiaoyu Wang 已提交
276 277 278
static int32_t  translateSubquery(STranslateContext* pCxt, SNode* pNode);
static int32_t  translateQuery(STranslateContext* pCxt, SNode* pNode);
static EDealRes translateValue(STranslateContext* pCxt, SValueNode* pVal);
279

dengyihao's avatar
dengyihao 已提交
280
static bool afterGroupBy(ESqlClause clause) { return clause > SQL_CLAUSE_GROUP_BY; }
281

dengyihao's avatar
dengyihao 已提交
282
static bool beforeHaving(ESqlClause clause) { return clause < SQL_CLAUSE_HAVING; }
283

284 285
static bool afterHaving(ESqlClause clause) { return clause > SQL_CLAUSE_HAVING; }

X
Xiaoyu Wang 已提交
286 287
static bool beforeWindow(ESqlClause clause) { return clause < SQL_CLAUSE_WINDOW; }

288 289 290 291 292 293 294 295 296 297 298 299 300 301
static bool hasSameTableAlias(SArray* pTables) {
  if (taosArrayGetSize(pTables) < 2) {
    return false;
  }
  STableNode* pTable0 = taosArrayGetP(pTables, 0);
  for (int32_t i = 1; i < taosArrayGetSize(pTables); ++i) {
    STableNode* pTable = taosArrayGetP(pTables, i);
    if (0 == strcmp(pTable0->tableAlias, pTable->tableAlias)) {
      return true;
    }
  }
  return false;
}

302
static int32_t addNamespace(STranslateContext* pCxt, void* pTable) {
wmmhello's avatar
wmmhello 已提交
303 304 305 306
  size_t currTotalLevel = taosArrayGetSize(pCxt->pNsLevel);
  if (currTotalLevel > pCxt->currLevel) {
    SArray* pTables = taosArrayGetP(pCxt->pNsLevel, pCxt->currLevel);
    taosArrayPush(pTables, &pTable);
307
    if (hasSameTableAlias(pTables)) {
308 309
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_UNIQUE_TABLE_ALIAS,
                                     "Not unique table/alias: '%s'", ((STableNode*)pTable)->tableAlias);
310
    }
wmmhello's avatar
wmmhello 已提交
311 312 313
  } else {
    do {
      SArray* pTables = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
X
Xiaoyu Wang 已提交
314 315 316
      if (NULL == pTables) {
        return TSDB_CODE_OUT_OF_MEMORY;
      }
wmmhello's avatar
wmmhello 已提交
317 318
      if (pCxt->currLevel == currTotalLevel) {
        taosArrayPush(pTables, &pTable);
319
        if (hasSameTableAlias(pTables)) {
X
Xiaoyu Wang 已提交
320
          taosArrayDestroy(pTables);
321 322
          return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_UNIQUE_TABLE_ALIAS,
                                         "Not unique table/alias: '%s'", ((STableNode*)pTable)->tableAlias);
323
        }
wmmhello's avatar
wmmhello 已提交
324 325 326 327 328 329
      }
      taosArrayPush(pCxt->pNsLevel, &pTables);
      ++currTotalLevel;
    } while (currTotalLevel <= pCxt->currLevel);
  }
  return TSDB_CODE_SUCCESS;
330 331
}

D
dapan1121 已提交
332
static int32_t collectUseDatabaseImpl(const char* pFullDbName, SHashObj* pDbs) {
wmmhello's avatar
wmmhello 已提交
333
  SFullDatabaseName name = {0};
X
Xiaoyu Wang 已提交
334
  snprintf(name.fullDbName, sizeof(name.fullDbName), "%s", pFullDbName);
wmmhello's avatar
wmmhello 已提交
335
  return taosHashPut(pDbs, pFullDbName, strlen(pFullDbName), &name, sizeof(SFullDatabaseName));
X
Xiaoyu Wang 已提交
336 337
}

D
dapan1121 已提交
338
static int32_t collectUseDatabase(const SName* pName, SHashObj* pDbs) {
wmmhello's avatar
wmmhello 已提交
339 340 341
  char dbFName[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(pName, dbFName);
  return collectUseDatabaseImpl(dbFName, pDbs);
D
dapan1121 已提交
342 343
}

D
dapan1121 已提交
344
static int32_t collectUseTable(const SName* pName, SHashObj* pTable) {
wmmhello's avatar
wmmhello 已提交
345 346
  char fullName[TSDB_TABLE_FNAME_LEN];
  tNameExtractFullName(pName, fullName);
D
dapan1121 已提交
347
  return taosHashPut(pTable, fullName, strlen(fullName), pName, sizeof(SName));
X
Xiaoyu Wang 已提交
348 349 350
}

static int32_t getTableMetaImpl(STranslateContext* pCxt, const SName* pName, STableMeta** pMeta) {
wmmhello's avatar
wmmhello 已提交
351
  SParseContext* pParCxt = pCxt->pParseCxt;
D
dapan1121 已提交
352 353 354 355 356 357 358 359
  int32_t        code = collectUseDatabase(pName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(pName, pCxt->pTables);
  }
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getTableMetaFromCache(pCxt->pMetaCache, pName, pMeta);
    } else {
X
Xiaoyu Wang 已提交
360
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
361 362 363 364
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
      code = catalogGetTableMeta(pParCxt->pCatalog, &conn, pName, pMeta);
365
    }
wmmhello's avatar
wmmhello 已提交
366
  }
367
  if (TSDB_CODE_SUCCESS != code && TSDB_CODE_PAR_TABLE_NOT_EXIST != code) {
368 369
    parserError("0x%" PRIx64 " catalogGetTableMeta error, code:%s, dbName:%s, tbName:%s", pCxt->pParseCxt->requestId,
                tstrerror(code), pName->dbname, pName->tname);
wmmhello's avatar
wmmhello 已提交
370 371
  }
  return code;
X
Xiaoyu Wang 已提交
372 373
}

X
Xiaoyu Wang 已提交
374
static int32_t getTableMeta(STranslateContext* pCxt, const char* pDbName, const char* pTableName, STableMeta** pMeta) {
wmmhello's avatar
wmmhello 已提交
375 376
  SName name;
  return getTableMetaImpl(pCxt, toName(pCxt->pParseCxt->acctId, pDbName, pTableName, &name), pMeta);
X
Xiaoyu Wang 已提交
377 378
}

D
dapan1121 已提交
379 380 381 382 383 384 385 386 387 388 389 390 391 392
static int32_t getTableCfg(STranslateContext* pCxt, const SName* pName, STableCfg** pCfg) {
  SParseContext* pParCxt = pCxt->pParseCxt;
  int32_t        code = collectUseDatabase(pName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(pName, pCxt->pTables);
  }
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getTableCfgFromCache(pCxt->pMetaCache, pName, pCfg);
    } else {
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
D
dapan1121 已提交
393
      code = catalogRefreshGetTableCfg(pParCxt->pCatalog, &conn, pName, pCfg);
D
dapan1121 已提交
394 395 396
    }
  }
  if (TSDB_CODE_SUCCESS != code) {
397 398
    parserError("0x%" PRIx64 " catalogRefreshGetTableCfg error, code:%s, dbName:%s, tbName:%s",
                pCxt->pParseCxt->requestId, tstrerror(code), pName->dbname, pName->tname);
D
dapan1121 已提交
399 400 401 402
  }
  return code;
}

403
static int32_t refreshGetTableMeta(STranslateContext* pCxt, const char* pDbName, const char* pTableName,
wmmhello's avatar
wmmhello 已提交
404 405 406 407
                                   STableMeta** pMeta) {
  SParseContext* pParCxt = pCxt->pParseCxt;
  SName          name;
  toName(pCxt->pParseCxt->acctId, pDbName, pTableName, &name);
408 409 410 411
  int32_t code = TSDB_CODE_SUCCESS;
  if (pParCxt->async) {
    code = getTableMetaFromCache(pCxt->pMetaCache, &name, pMeta);
  } else {
X
Xiaoyu Wang 已提交
412
    SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
413 414 415
                             .requestId = pParCxt->requestId,
                             .requestObjRefId = pParCxt->requestRid,
                             .mgmtEps = pParCxt->mgmtEpSet};
X
Xiaoyu Wang 已提交
416 417

    code = catalogRefreshGetTableMeta(pParCxt->pCatalog, &conn, &name, pMeta, false);
418
  }
wmmhello's avatar
wmmhello 已提交
419
  if (TSDB_CODE_SUCCESS != code) {
420 421
    parserError("0x%" PRIx64 " catalogRefreshGetTableMeta error, code:%s, dbName:%s, tbName:%s",
                pCxt->pParseCxt->requestId, tstrerror(code), pDbName, pTableName);
wmmhello's avatar
wmmhello 已提交
422 423
  }
  return code;
424 425
}

X
Xiaoyu Wang 已提交
426
static int32_t getDBVgInfoImpl(STranslateContext* pCxt, const SName* pName, SArray** pVgInfo) {
wmmhello's avatar
wmmhello 已提交
427 428 429
  SParseContext* pParCxt = pCxt->pParseCxt;
  char           fullDbName[TSDB_DB_FNAME_LEN];
  tNameGetFullDbName(pName, fullDbName);
D
dapan1121 已提交
430 431 432 433 434
  int32_t code = collectUseDatabaseImpl(fullDbName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getDbVgInfoFromCache(pCxt->pMetaCache, fullDbName, pVgInfo);
    } else {
X
Xiaoyu Wang 已提交
435
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
436 437 438
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
439
      code = catalogGetDBVgList(pParCxt->pCatalog, &conn, fullDbName, pVgInfo);
440
    }
wmmhello's avatar
wmmhello 已提交
441 442
  }
  if (TSDB_CODE_SUCCESS != code) {
443
    parserError("0x%" PRIx64 " catalogGetDBVgList error, code:%s, dbFName:%s", pCxt->pParseCxt->requestId,
444
                tstrerror(code), fullDbName);
wmmhello's avatar
wmmhello 已提交
445 446
  }
  return code;
X
Xiaoyu Wang 已提交
447 448
}

X
Xiaoyu Wang 已提交
449
static int32_t getDBVgInfo(STranslateContext* pCxt, const char* pDbName, SArray** pVgInfo) {
wmmhello's avatar
wmmhello 已提交
450 451 452 453 454
  SName name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pDbName, strlen(pDbName));
  char dbFname[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(&name, dbFname);
  return getDBVgInfoImpl(pCxt, &name, pVgInfo);
X
Xiaoyu Wang 已提交
455 456
}

X
Xiaoyu Wang 已提交
457
static int32_t getTableHashVgroupImpl(STranslateContext* pCxt, const SName* pName, SVgroupInfo* pInfo) {
wmmhello's avatar
wmmhello 已提交
458
  SParseContext* pParCxt = pCxt->pParseCxt;
D
dapan1121 已提交
459 460 461 462 463 464 465 466
  int32_t        code = collectUseDatabase(pName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(pName, pCxt->pTables);
  }
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getTableVgroupFromCache(pCxt->pMetaCache, pName, pInfo);
    } else {
X
Xiaoyu Wang 已提交
467
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
468 469 470 471
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
      code = catalogGetTableHashVgroup(pParCxt->pCatalog, &conn, pName, pInfo);
472
    }
wmmhello's avatar
wmmhello 已提交
473 474
  }
  if (TSDB_CODE_SUCCESS != code) {
475 476
    parserError("0x%" PRIx64 " catalogGetTableHashVgroup error, code:%s, dbName:%s, tbName:%s",
                pCxt->pParseCxt->requestId, tstrerror(code), pName->dbname, pName->tname);
wmmhello's avatar
wmmhello 已提交
477 478
  }
  return code;
X
Xiaoyu Wang 已提交
479 480
}

dengyihao's avatar
dengyihao 已提交
481
static int32_t getTableHashVgroup(STranslateContext* pCxt, const char* pDbName, const char* pTableName,
wmmhello's avatar
wmmhello 已提交
482 483 484
                                  SVgroupInfo* pInfo) {
  SName name;
  return getTableHashVgroupImpl(pCxt, toName(pCxt->pParseCxt->acctId, pDbName, pTableName, &name), pInfo);
X
Xiaoyu Wang 已提交
485 486
}

dengyihao's avatar
dengyihao 已提交
487
static int32_t getDBVgVersion(STranslateContext* pCxt, const char* pDbFName, int32_t* pVersion, int64_t* pDbId,
D
dapan1121 已提交
488
                              int32_t* pTableNum, int64_t* pStateTs) {
wmmhello's avatar
wmmhello 已提交
489
  SParseContext* pParCxt = pCxt->pParseCxt;
D
dapan1121 已提交
490 491 492
  int32_t        code = collectUseDatabaseImpl(pDbFName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
D
dapan1121 已提交
493
      code = getDbVgVersionFromCache(pCxt->pMetaCache, pDbFName, pVersion, pDbId, pTableNum, pStateTs);
D
dapan1121 已提交
494
    } else {
D
dapan1121 已提交
495
      code = catalogGetDBVgVersion(pParCxt->pCatalog, pDbFName, pVersion, pDbId, pTableNum, pStateTs);
496
    }
wmmhello's avatar
wmmhello 已提交
497 498
  }
  if (TSDB_CODE_SUCCESS != code) {
499 500
    parserError("0x%" PRIx64 " catalogGetDBVgVersion error, code:%s, dbFName:%s", pCxt->pParseCxt->requestId,
                tstrerror(code), pDbFName);
wmmhello's avatar
wmmhello 已提交
501 502
  }
  return code;
X
Xiaoyu Wang 已提交
503 504
}

505
static int32_t getDBCfg(STranslateContext* pCxt, const char* pDbName, SDbCfgInfo* pInfo) {
wmmhello's avatar
wmmhello 已提交
506 507 508 509 510
  SParseContext* pParCxt = pCxt->pParseCxt;
  SName          name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pDbName, strlen(pDbName));
  char dbFname[TSDB_DB_FNAME_LEN] = {0};
  tNameGetFullDbName(&name, dbFname);
D
dapan1121 已提交
511 512 513 514 515
  int32_t code = collectUseDatabaseImpl(dbFname, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getDbCfgFromCache(pCxt->pMetaCache, dbFname, pInfo);
    } else {
X
Xiaoyu Wang 已提交
516
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
517 518 519 520 521
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};

      code = catalogGetDBCfg(pParCxt->pCatalog, &conn, dbFname, pInfo);
522
    }
wmmhello's avatar
wmmhello 已提交
523 524
  }
  if (TSDB_CODE_SUCCESS != code) {
525 526
    parserError("0x%" PRIx64 " catalogGetDBCfg error, code:%s, dbFName:%s", pCxt->pParseCxt->requestId, tstrerror(code),
                dbFname);
wmmhello's avatar
wmmhello 已提交
527 528
  }
  return code;
529 530
}

531 532 533 534 535 536 537
static int32_t getUdfInfo(STranslateContext* pCxt, SFunctionNode* pFunc) {
  SParseContext* pParCxt = pCxt->pParseCxt;
  SFuncInfo      funcInfo = {0};
  int32_t        code = TSDB_CODE_SUCCESS;
  if (pParCxt->async) {
    code = getUdfInfoFromCache(pCxt->pMetaCache, pFunc->functionName, &funcInfo);
  } else {
X
Xiaoyu Wang 已提交
538
    SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
D
dapan1121 已提交
539 540 541 542
                             .requestId = pParCxt->requestId,
                             .requestObjRefId = pParCxt->requestRid,
                             .mgmtEps = pParCxt->mgmtEpSet};

X
Xiaoyu Wang 已提交
543
    code = catalogGetUdfInfo(pParCxt->pCatalog, &conn, pFunc->functionName, &funcInfo);
544 545 546 547 548 549 550 551 552
  }
  if (TSDB_CODE_SUCCESS == code) {
    pFunc->funcType = FUNCTION_TYPE_UDF;
    pFunc->funcId = TSDB_FUNC_TYPE_AGGREGATE == funcInfo.funcType ? FUNC_AGGREGATE_UDF_ID : FUNC_SCALAR_UDF_ID;
    pFunc->node.resType.type = funcInfo.outputType;
    pFunc->node.resType.bytes = funcInfo.outputLen;
    pFunc->udfBufSize = funcInfo.bufSize;
    tFreeSFuncInfo(&funcInfo);
  }
553 554 555 556
  if (TSDB_CODE_SUCCESS != code) {
    parserError("0x%" PRIx64 " catalogGetUdfInfo error, code:%s, funcName:%s", pCxt->pParseCxt->requestId,
                tstrerror(code), pFunc->functionName);
  }
557 558 559
  return code;
}

X
Xiaoyu Wang 已提交
560
static int32_t getTableIndex(STranslateContext* pCxt, const SName* pName, SArray** pIndexes) {
X
Xiaoyu Wang 已提交
561
  SParseContext* pParCxt = pCxt->pParseCxt;
X
Xiaoyu Wang 已提交
562 563 564 565
  int32_t        code = collectUseDatabase(pName, pCxt->pDbs);
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(pName, pCxt->pTables);
  }
X
Xiaoyu Wang 已提交
566 567 568 569 570 571 572 573
  if (TSDB_CODE_SUCCESS == code) {
    if (pParCxt->async) {
      code = getTableIndexFromCache(pCxt->pMetaCache, pName, pIndexes);
    } else {
      SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
                               .requestId = pParCxt->requestId,
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
X
Xiaoyu Wang 已提交
574

X
Xiaoyu Wang 已提交
575 576
      code = catalogGetTableIndex(pParCxt->pCatalog, &conn, pName, pIndexes);
    }
X
Xiaoyu Wang 已提交
577 578
  }
  if (TSDB_CODE_SUCCESS != code) {
579 580
    parserError("0x%" PRIx64 " getTableIndex error, code:%s, dbName:%s, tbName:%s", pCxt->pParseCxt->requestId,
                tstrerror(code), pName->dbname, pName->tname);
X
Xiaoyu Wang 已提交
581 582 583 584
  }
  return code;
}

585 586 587 588 589 590 591 592 593 594 595 596 597
static int32_t getDnodeList(STranslateContext* pCxt, SArray** pDnodes) {
  SParseContext* pParCxt = pCxt->pParseCxt;
  int32_t        code = TSDB_CODE_SUCCESS;
  if (pParCxt->async) {
    code = getDnodeListFromCache(pCxt->pMetaCache, pDnodes);
  } else {
    SRequestConnInfo conn = {.pTrans = pParCxt->pTransporter,
                             .requestId = pParCxt->requestId,
                             .requestObjRefId = pParCxt->requestRid,
                             .mgmtEps = pParCxt->mgmtEpSet};
    code = catalogGetDnodeList(pParCxt->pCatalog, &conn, pDnodes);
  }
  if (TSDB_CODE_SUCCESS != code) {
598
    parserError("0x%" PRIx64 " getDnodeList error, code:%s", pCxt->pParseCxt->requestId, tstrerror(code));
599 600 601 602
  }
  return code;
}

603
static int32_t initTranslateContext(SParseContext* pParseCxt, SParseMetaCache* pMetaCache, STranslateContext* pCxt) {
wmmhello's avatar
wmmhello 已提交
604 605 606 607 608 609
  pCxt->pParseCxt = pParseCxt;
  pCxt->errCode = TSDB_CODE_SUCCESS;
  pCxt->msgBuf.buf = pParseCxt->pMsg;
  pCxt->msgBuf.len = pParseCxt->msgLen;
  pCxt->pNsLevel = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
  pCxt->currLevel = 0;
610
  pCxt->levelNo = 0;
wmmhello's avatar
wmmhello 已提交
611
  pCxt->currClause = 0;
612
  pCxt->pMetaCache = pMetaCache;
wmmhello's avatar
wmmhello 已提交
613 614
  pCxt->pDbs = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
  pCxt->pTables = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
D
dapan1121 已提交
615 616
  pCxt->pTargetTables = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
  if (NULL == pCxt->pNsLevel || NULL == pCxt->pDbs || NULL == pCxt->pTables || NULL == pCxt->pTargetTables) {
wmmhello's avatar
wmmhello 已提交
617 618 619
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  return TSDB_CODE_SUCCESS;
620 621
}

X
Xiaoyu Wang 已提交
622
static int32_t resetTranslateNamespace(STranslateContext* pCxt) {
wmmhello's avatar
wmmhello 已提交
623 624 625 626 627 628 629 630 631 632 633 634
  if (NULL != pCxt->pNsLevel) {
    size_t size = taosArrayGetSize(pCxt->pNsLevel);
    for (size_t i = 0; i < size; ++i) {
      taosArrayDestroy(taosArrayGetP(pCxt->pNsLevel, i));
    }
    taosArrayDestroy(pCxt->pNsLevel);
  }
  pCxt->pNsLevel = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
  if (NULL == pCxt->pNsLevel) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
635 636
}

637
static void destroyTranslateContext(STranslateContext* pCxt) {
wmmhello's avatar
wmmhello 已提交
638 639 640 641 642 643 644
  if (NULL != pCxt->pNsLevel) {
    size_t size = taosArrayGetSize(pCxt->pNsLevel);
    for (size_t i = 0; i < size; ++i) {
      taosArrayDestroy(taosArrayGetP(pCxt->pNsLevel, i));
    }
    taosArrayDestroy(pCxt->pNsLevel);
  }
645

wmmhello's avatar
wmmhello 已提交
646 647 648 649
  if (NULL != pCxt->pCmdMsg) {
    taosMemoryFreeClear(pCxt->pCmdMsg->pMsg);
    taosMemoryFreeClear(pCxt->pCmdMsg);
  }
650

wmmhello's avatar
wmmhello 已提交
651 652
  taosHashCleanup(pCxt->pDbs);
  taosHashCleanup(pCxt->pTables);
D
dapan1121 已提交
653
  taosHashCleanup(pCxt->pTargetTables);
654 655
}

656 657 658 659
static bool isSelectStmt(SNode* pCurrStmt) {
  return NULL != pCurrStmt && QUERY_NODE_SELECT_STMT == nodeType(pCurrStmt);
}

K
kailixu 已提交
660 661 662 663
static bool isDeleteStmt(SNode* pCurrStmt) {
  return NULL != pCurrStmt && QUERY_NODE_DELETE_STMT == nodeType(pCurrStmt);
}

664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684
static bool isSetOperator(SNode* pCurrStmt) {
  return NULL != pCurrStmt && QUERY_NODE_SET_OPERATOR == nodeType(pCurrStmt);
}

static SNodeList* getProjectListFromCurrStmt(SNode* pCurrStmt) {
  if (isSelectStmt(pCurrStmt)) {
    return ((SSelectStmt*)pCurrStmt)->pProjectionList;
  }
  if (isSetOperator(pCurrStmt)) {
    return ((SSetOperator*)pCurrStmt)->pProjectionList;
  }
  return NULL;
}

static uint8_t getPrecisionFromCurrStmt(SNode* pCurrStmt, uint8_t defaultVal) {
  if (isSelectStmt(pCurrStmt)) {
    return ((SSelectStmt*)pCurrStmt)->precision;
  }
  if (isSetOperator(pCurrStmt)) {
    return ((SSetOperator*)pCurrStmt)->precision;
  }
685 686 687
  if (NULL != pCurrStmt && QUERY_NODE_CREATE_STREAM_STMT == nodeType(pCurrStmt)) {
    return getPrecisionFromCurrStmt(((SCreateStreamStmt*)pCurrStmt)->pQuery, defaultVal);
  }
K
kailixu 已提交
688 689 690
  if (isDeleteStmt(pCurrStmt)) {
    return ((SDeleteStmt*)pCurrStmt)->precision;
  }
691 692 693
  return defaultVal;
}

694
static bool isAliasColumn(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
695
  return (QUERY_NODE_COLUMN == nodeType(pNode) && ('\0' == ((SColumnNode*)pNode)->tableAlias[0]));
696 697 698
}

static bool isAggFunc(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
699
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsAggFunc(((SFunctionNode*)pNode)->funcId));
700 701 702
}

static bool isSelectFunc(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
703
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsSelectFunc(((SFunctionNode*)pNode)->funcId));
704 705
}

5
54liuyao 已提交
706 707 708 709
static bool isWindowPseudoColumnFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsWindowPseudoColumnFunc(((SFunctionNode*)pNode)->funcId));
}

710 711 712 713
static bool isInterpFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsInterpFunc(((SFunctionNode*)pNode)->funcId));
}

714 715 716 717
static bool isInterpPseudoColumnFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsInterpPseudoColumnFunc(((SFunctionNode*)pNode)->funcId));
}

718
static bool isTimelineFunc(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
719
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsTimelineFunc(((SFunctionNode*)pNode)->funcId));
720 721
}

722 723 724 725
static bool isImplicitTsFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsImplicitTsFunc(((SFunctionNode*)pNode)->funcId));
}

726
static bool isScanPseudoColumnFunc(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
727
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsScanPseudoColumnFunc(((SFunctionNode*)pNode)->funcId));
728 729
}

wmmhello's avatar
wmmhello 已提交
730 731
static bool isIndefiniteRowsFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsIndefiniteRowsFunc(((SFunctionNode*)pNode)->funcId));
732 733
}

X
Xiaoyu Wang 已提交
734 735 736 737
static bool isVectorFunc(const SNode* pNode) {
  return (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsVectorFunc(((SFunctionNode*)pNode)->funcId));
}

738
static bool isDistinctOrderBy(STranslateContext* pCxt) {
739 740
  return (SQL_CLAUSE_ORDER_BY == pCxt->currClause && isSelectStmt(pCxt->pCurrStmt) &&
          ((SSelectStmt*)pCxt->pCurrStmt)->isDistinct);
741 742
}

743
static bool belongTable(const SColumnNode* pCol, const STableNode* pTable) {
wmmhello's avatar
wmmhello 已提交
744 745 746 747 748 749 750 751
  int cmp = 0;
  if ('\0' != pCol->dbName[0]) {
    cmp = strcmp(pCol->dbName, pTable->dbName);
  }
  if (0 == cmp) {
    cmp = strcmp(pCol->tableAlias, pTable->tableAlias);
  }
  return (0 == cmp);
752 753
}

X
Xiaoyu Wang 已提交
754
static SNodeList* getProjectList(const SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
755 756 757 758 759 760
  if (QUERY_NODE_SELECT_STMT == nodeType(pNode)) {
    return ((SSelectStmt*)pNode)->pProjectionList;
  } else if (QUERY_NODE_SET_OPERATOR == nodeType(pNode)) {
    return ((SSetOperator*)pNode)->pProjectionList;
  }
  return NULL;
761 762
}

763 764
static bool isTimeLineQuery(SNode* pStmt) {
  if (QUERY_NODE_SELECT_STMT == nodeType(pStmt)) {
765
    return (TIME_LINE_MULTI == ((SSelectStmt*)pStmt)->timeLineResMode) || (TIME_LINE_GLOBAL == ((SSelectStmt*)pStmt)->timeLineResMode);
766 767
  } else if (QUERY_NODE_SET_OPERATOR == nodeType(pStmt)) {
    return TIME_LINE_GLOBAL == ((SSetOperator*)pStmt)->timeLineResMode;
768 769 770 771 772
  } else {
    return false;
  }
}

X
Xiaoyu Wang 已提交
773
static bool isGlobalTimeLineQuery(SNode* pStmt) {
774 775
  if (QUERY_NODE_SELECT_STMT == nodeType(pStmt)) {
    return TIME_LINE_GLOBAL == ((SSelectStmt*)pStmt)->timeLineResMode;
776 777
  } else if (QUERY_NODE_SET_OPERATOR == nodeType(pStmt)) {
    return TIME_LINE_GLOBAL == ((SSetOperator*)pStmt)->timeLineResMode;
778
  } else {
X
Xiaoyu Wang 已提交
779 780 781 782
    return false;
  }
}

783 784
static bool isTimeLineAlignedQuery(SNode* pStmt) {
  SSelectStmt *pSelect = (SSelectStmt *)pStmt;
785
  if (isGlobalTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
786 787
    return true;
  }
788 789 790 791
  if (!isTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
    return false;
  }
  if (QUERY_NODE_SELECT_STMT != nodeType(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
792 793
    return false;
  }
794
  SSelectStmt *pSub = (SSelectStmt *)((STempTableNode*)pSelect->pFromTable)->pSubquery;
795 796 797 798 799 800
  if (nodesListMatch(pSelect->pPartitionByList, pSub->pPartitionByList)) {
    return true;
  }
  return false;
}

801 802 803 804 805
static bool isPrimaryKeyImpl(SNode* pExpr) {
  if (QUERY_NODE_COLUMN == nodeType(pExpr)) {
    return (PRIMARYKEY_TIMESTAMP_COL_ID == ((SColumnNode*)pExpr)->colId);
  } else if (QUERY_NODE_FUNCTION == nodeType(pExpr)) {
    SFunctionNode* pFunc = (SFunctionNode*)pExpr;
806
    if (FUNCTION_TYPE_SELECT_VALUE == pFunc->funcType || FUNCTION_TYPE_GROUP_KEY == pFunc->funcType ||
807 808
        FUNCTION_TYPE_FIRST == pFunc->funcType || FUNCTION_TYPE_LAST == pFunc->funcType ||
        FUNCTION_TYPE_LAST_ROW == pFunc->funcType) {
809
      return isPrimaryKeyImpl(nodesListGetNode(pFunc->pParameterList, 0));
810 811
    } else if (FUNCTION_TYPE_WSTART == pFunc->funcType || FUNCTION_TYPE_WEND == pFunc->funcType ||
               FUNCTION_TYPE_IROWTS == pFunc->funcType) {
812 813 814 815 816 817 818 819 820 821 822 823 824
      return true;
    }
  }
  return false;
}

static bool isPrimaryKey(STempTableNode* pTable, SNode* pExpr) {
  if (!isTimeLineQuery(pTable->pSubquery)) {
    return false;
  }
  return isPrimaryKeyImpl(pExpr);
}

X
Xiaoyu Wang 已提交
825
static void setColumnInfoBySchema(const SRealTableNode* pTable, const SSchema* pColSchema, int32_t tagFlag,
wmmhello's avatar
wmmhello 已提交
826 827 828 829 830 831 832 833
                                  SColumnNode* pCol) {
  strcpy(pCol->dbName, pTable->table.dbName);
  strcpy(pCol->tableAlias, pTable->table.tableAlias);
  strcpy(pCol->tableName, pTable->table.tableName);
  strcpy(pCol->colName, pColSchema->name);
  if ('\0' == pCol->node.aliasName[0]) {
    strcpy(pCol->node.aliasName, pColSchema->name);
  }
X
Xiaoyu Wang 已提交
834 835 836
  if ('\0' == pCol->node.userAlias[0]) {
    strcpy(pCol->node.userAlias, pColSchema->name);
  }
wmmhello's avatar
wmmhello 已提交
837 838 839
  pCol->tableId = pTable->pMeta->uid;
  pCol->tableType = pTable->pMeta->tableType;
  pCol->colId = pColSchema->colId;
X
Xiaoyu Wang 已提交
840
  pCol->colType = (tagFlag >= 0 ? COLUMN_TYPE_TAG : COLUMN_TYPE_COLUMN);
dengyihao's avatar
dengyihao 已提交
841
  pCol->hasIndex = ((0 == tagFlag) || (pColSchema != NULL && IS_IDX_ON(pColSchema)));
wmmhello's avatar
wmmhello 已提交
842 843 844 845 846
  pCol->node.resType.type = pColSchema->type;
  pCol->node.resType.bytes = pColSchema->bytes;
  if (TSDB_DATA_TYPE_TIMESTAMP == pCol->node.resType.type) {
    pCol->node.resType.precision = pTable->pMeta->tableInfo.precision;
  }
847 848
}

849
static void setColumnInfoByExpr(STempTableNode* pTable, SExprNode* pExpr, SColumnNode** pColRef) {
X
Xiaoyu Wang 已提交
850 851
  SColumnNode* pCol = *pColRef;

wmmhello's avatar
wmmhello 已提交
852 853 854
  if (NULL == pExpr->pAssociation) {
    pExpr->pAssociation = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
  }
X
Xiaoyu Wang 已提交
855
  taosArrayPush(pExpr->pAssociation, &pColRef);
856 857
  strcpy(pCol->tableAlias, pTable->table.tableAlias);
  pCol->colId = isPrimaryKey(pTable, (SNode*)pExpr) ? PRIMARYKEY_TIMESTAMP_COL_ID : 0;
wmmhello's avatar
wmmhello 已提交
858 859 860 861
  strcpy(pCol->colName, pExpr->aliasName);
  if ('\0' == pCol->node.aliasName[0]) {
    strcpy(pCol->node.aliasName, pCol->colName);
  }
X
Xiaoyu Wang 已提交
862
  if ('\0' == pCol->node.userAlias[0]) {
863
    strcpy(pCol->node.userAlias, pExpr->userAlias);
X
Xiaoyu Wang 已提交
864
  }
wmmhello's avatar
wmmhello 已提交
865
  pCol->node.resType = pExpr->resType;
866 867
}

868
static int32_t createColumnsByTable(STranslateContext* pCxt, const STableNode* pTable, bool igTags, SNodeList* pList) {
wmmhello's avatar
wmmhello 已提交
869 870
  if (QUERY_NODE_REAL_TABLE == nodeType(pTable)) {
    const STableMeta* pMeta = ((SRealTableNode*)pTable)->pMeta;
871 872
    int32_t           nums = pMeta->tableInfo.numOfColumns +
                   (igTags ? 0 : ((TSDB_SUPER_TABLE == pMeta->tableType) ? pMeta->tableInfo.numOfTags : 0));
wmmhello's avatar
wmmhello 已提交
873
    for (int32_t i = 0; i < nums; ++i) {
874 875 876
      if (invisibleColumn(pCxt->pParseCxt->enableSysInfo, pMeta->tableType, pMeta->schema[i].flags)) {
        continue;
      }
wmmhello's avatar
wmmhello 已提交
877 878 879 880
      SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
      if (NULL == pCol) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
      }
X
Xiaoyu Wang 已提交
881
      setColumnInfoBySchema((SRealTableNode*)pTable, pMeta->schema + i, (i - pMeta->tableInfo.numOfColumns), pCol);
wmmhello's avatar
wmmhello 已提交
882 883 884
      nodesListAppend(pList, (SNode*)pCol);
    }
  } else {
885 886 887
    STempTableNode* pTempTable = (STempTableNode*)pTable;
    SNodeList*      pProjectList = getProjectList(pTempTable->pSubquery);
    SNode*          pNode;
wmmhello's avatar
wmmhello 已提交
888 889 890 891 892 893
    FOREACH(pNode, pProjectList) {
      SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
      if (NULL == pCol) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
      }
      nodesListAppend(pList, (SNode*)pCol);
X
Xiaoyu Wang 已提交
894
      SListCell* pCell = nodesListGetCell(pList, LIST_LENGTH(pList) - 1);
895
      setColumnInfoByExpr(pTempTable, (SExprNode*)pNode, (SColumnNode**)&pCell->pNode);
wmmhello's avatar
wmmhello 已提交
896 897 898
    }
  }
  return TSDB_CODE_SUCCESS;
899 900
}

X
Xiaoyu Wang 已提交
901
static bool isInternalPrimaryKey(const SColumnNode* pCol) {
902 903
  return PRIMARYKEY_TIMESTAMP_COL_ID == pCol->colId &&
         (0 == strcmp(pCol->colName, ROWTS_PSEUDO_COLUMN_NAME) || 0 == strcmp(pCol->colName, C0_PSEUDO_COLUMN_NAME));
X
Xiaoyu Wang 已提交
904 905
}

906 907
static int32_t findAndSetColumn(STranslateContext* pCxt, SColumnNode** pColRef, const STableNode* pTable,
                                bool* pFound) {
X
Xiaoyu Wang 已提交
908
  SColumnNode* pCol = *pColRef;
909
  *pFound = false;
wmmhello's avatar
wmmhello 已提交
910 911 912
  if (QUERY_NODE_REAL_TABLE == nodeType(pTable)) {
    const STableMeta* pMeta = ((SRealTableNode*)pTable)->pMeta;
    if (isInternalPrimaryKey(pCol)) {
X
Xiaoyu Wang 已提交
913
      setColumnInfoBySchema((SRealTableNode*)pTable, pMeta->schema, -1, pCol);
914 915
      *pFound = true;
      return TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
916 917 918
    }
    int32_t nums = pMeta->tableInfo.numOfTags + pMeta->tableInfo.numOfColumns;
    for (int32_t i = 0; i < nums; ++i) {
919 920
      if (0 == strcmp(pCol->colName, pMeta->schema[i].name) &&
          !invisibleColumn(pCxt->pParseCxt->enableSysInfo, pMeta->tableType, pMeta->schema[i].flags)) {
X
Xiaoyu Wang 已提交
921
        setColumnInfoBySchema((SRealTableNode*)pTable, pMeta->schema + i, (i - pMeta->tableInfo.numOfColumns), pCol);
922
        *pFound = true;
wmmhello's avatar
wmmhello 已提交
923 924 925 926
        break;
      }
    }
  } else {
927 928 929
    STempTableNode* pTempTable = (STempTableNode*)pTable;
    SNodeList*      pProjectList = getProjectList(pTempTable->pSubquery);
    SNode*          pNode;
wmmhello's avatar
wmmhello 已提交
930 931
    FOREACH(pNode, pProjectList) {
      SExprNode* pExpr = (SExprNode*)pNode;
X
Xiaoyu Wang 已提交
932
      if (0 == strcmp(pCol->colName, pExpr->aliasName)) {
933 934 935
        if (*pFound) {
          return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AMBIGUOUS_COLUMN, pCol->colName);
        }
936
        setColumnInfoByExpr(pTempTable, pExpr, pColRef);
937
        *pFound = true;
938 939
      } else if (isPrimaryKey(pTempTable, pNode) && isInternalPrimaryKey(pCol)) {
        setColumnInfoByExpr(pTempTable, pExpr, pColRef);
X
Xiaoyu Wang 已提交
940
        *pFound = true;
wmmhello's avatar
wmmhello 已提交
941 942 943
      }
    }
  }
944
  return TSDB_CODE_SUCCESS;
945 946
}

X
Xiaoyu Wang 已提交
947
static EDealRes translateColumnWithPrefix(STranslateContext* pCxt, SColumnNode** pCol) {
wmmhello's avatar
wmmhello 已提交
948 949 950 951 952
  SArray* pTables = taosArrayGetP(pCxt->pNsLevel, pCxt->currLevel);
  size_t  nums = taosArrayGetSize(pTables);
  bool    foundTable = false;
  for (size_t i = 0; i < nums; ++i) {
    STableNode* pTable = taosArrayGetP(pTables, i);
953
    if (belongTable((*pCol), pTable)) {
wmmhello's avatar
wmmhello 已提交
954
      foundTable = true;
955 956 957 958 959 960
      bool foundCol = false;
      pCxt->errCode = findAndSetColumn(pCxt, pCol, pTable, &foundCol);
      if (TSDB_CODE_SUCCESS != pCxt->errCode) {
        return DEAL_RES_ERROR;
      }
      if (foundCol) {
wmmhello's avatar
wmmhello 已提交
961 962
        break;
      }
X
Xiaoyu Wang 已提交
963
      return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_INVALID_COLUMN, (*pCol)->colName);
wmmhello's avatar
wmmhello 已提交
964 965 966
    }
  }
  if (!foundTable) {
X
Xiaoyu Wang 已提交
967
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_TABLE_NOT_EXIST, (*pCol)->tableAlias);
wmmhello's avatar
wmmhello 已提交
968 969
  }
  return DEAL_RES_CONTINUE;
970
}
971

X
Xiaoyu Wang 已提交
972
static EDealRes translateColumnWithoutPrefix(STranslateContext* pCxt, SColumnNode** pCol) {
wmmhello's avatar
wmmhello 已提交
973 974 975
  SArray* pTables = taosArrayGetP(pCxt->pNsLevel, pCxt->currLevel);
  size_t  nums = taosArrayGetSize(pTables);
  bool    found = false;
X
Xiaoyu Wang 已提交
976
  bool    isInternalPk = isInternalPrimaryKey(*pCol);
wmmhello's avatar
wmmhello 已提交
977 978
  for (size_t i = 0; i < nums; ++i) {
    STableNode* pTable = taosArrayGetP(pTables, i);
979 980 981 982 983 984
    bool        foundCol = false;
    pCxt->errCode = findAndSetColumn(pCxt, pCol, pTable, &foundCol);
    if (TSDB_CODE_SUCCESS != pCxt->errCode) {
      return DEAL_RES_ERROR;
    }
    if (foundCol) {
wmmhello's avatar
wmmhello 已提交
985
      if (found) {
X
Xiaoyu Wang 已提交
986
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_AMBIGUOUS_COLUMN, (*pCol)->colName);
wmmhello's avatar
wmmhello 已提交
987 988
      }
      found = true;
989 990 991
    }
    if (isInternalPk) {
      break;
wmmhello's avatar
wmmhello 已提交
992 993 994 995
    }
  }
  if (!found) {
    if (isInternalPk) {
996
      if (isSelectStmt(pCxt->pCurrStmt) && NULL != ((SSelectStmt*)pCxt->pCurrStmt)->pWindow) {
wmmhello's avatar
wmmhello 已提交
997 998 999 1000
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_NOT_ALLOWED_WIN_QUERY);
      }
      return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_INVALID_INTERNAL_PK);
    } else {
X
Xiaoyu Wang 已提交
1001
      return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_INVALID_COLUMN, (*pCol)->colName);
wmmhello's avatar
wmmhello 已提交
1002 1003 1004
    }
  }
  return DEAL_RES_CONTINUE;
1005 1006
}

1007
static EDealRes translateColumnUseAlias(STranslateContext* pCxt, SColumnNode** pCol, bool* pFound) {
1008
  SNodeList* pProjectionList = getProjectListFromCurrStmt(pCxt->pCurrStmt);
wmmhello's avatar
wmmhello 已提交
1009 1010 1011
  SNode*     pNode;
  FOREACH(pNode, pProjectionList) {
    SExprNode* pExpr = (SExprNode*)pNode;
X
Xiaoyu Wang 已提交
1012
    if (0 == strcmp((*pCol)->colName, pExpr->userAlias)) {
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
      SColumnRefNode* pColRef = (SColumnRefNode*)nodesMakeNode(QUERY_NODE_COLUMN_REF);
      if (NULL == pColRef) {
        pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
        return DEAL_RES_ERROR;
      }
      strcpy(pColRef->colName, pExpr->aliasName);
      nodesDestroyNode(*(SNode**)pCol);
      *(SNode**)pCol = (SNode*)pColRef;
      *pFound = true;
      return DEAL_RES_CONTINUE;
wmmhello's avatar
wmmhello 已提交
1023 1024
    }
  }
1025 1026
  *pFound = false;
  return DEAL_RES_CONTINUE;
1027 1028
}

X
Xiaoyu Wang 已提交
1029
static EDealRes translateColumn(STranslateContext* pCxt, SColumnNode** pCol) {
X
Xiaoyu Wang 已提交
1030 1031
  if (NULL == pCxt->pCurrStmt ||
      (isSelectStmt(pCxt->pCurrStmt) && NULL == ((SSelectStmt*)pCxt->pCurrStmt)->pFromTable)) {
1032 1033 1034
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_INVALID_COLUMN, (*pCol)->colName);
  }

wmmhello's avatar
wmmhello 已提交
1035
  // count(*)/first(*)/last(*) and so on
X
Xiaoyu Wang 已提交
1036
  if (0 == strcmp((*pCol)->colName, "*")) {
wmmhello's avatar
wmmhello 已提交
1037 1038 1039 1040
    return DEAL_RES_CONTINUE;
  }

  EDealRes res = DEAL_RES_CONTINUE;
X
Xiaoyu Wang 已提交
1041
  if ('\0' != (*pCol)->tableAlias[0]) {
wmmhello's avatar
wmmhello 已提交
1042 1043 1044 1045
    res = translateColumnWithPrefix(pCxt, pCol);
  } else {
    bool found = false;
    if (SQL_CLAUSE_ORDER_BY == pCxt->currClause) {
1046 1047 1048
      res = translateColumnUseAlias(pCxt, pCol, &found);
    }
    if (DEAL_RES_ERROR != res && !found) {
1049
      if (isSetOperator(pCxt->pCurrStmt)) {
1050 1051 1052 1053
        res = generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_INVALID_COLUMN, (*pCol)->colName);
      } else {
        res = translateColumnWithoutPrefix(pCxt, pCol);
      }
wmmhello's avatar
wmmhello 已提交
1054 1055 1056
    }
  }
  return res;
1057 1058
}

X
Xiaoyu Wang 已提交
1059 1060 1061 1062 1063
static int32_t parseTimeFromValueNode(STranslateContext* pCxt, SValueNode* pVal) {
  if (IS_NUMERIC_TYPE(pVal->node.resType.type) || TSDB_DATA_TYPE_BOOL == pVal->node.resType.type) {
    if (DEAL_RES_ERROR == translateValue(pCxt, pVal)) {
      return pCxt->errCode;
    }
X
Xiaoyu Wang 已提交
1064
    int64_t value = 0;
X
Xiaoyu Wang 已提交
1065
    if (IS_UNSIGNED_NUMERIC_TYPE(pVal->node.resType.type)) {
X
Xiaoyu Wang 已提交
1066
      value = pVal->datum.u;
X
Xiaoyu Wang 已提交
1067
    } else if (IS_FLOAT_TYPE(pVal->node.resType.type)) {
X
Xiaoyu Wang 已提交
1068
      value = pVal->datum.d;
X
Xiaoyu Wang 已提交
1069
    } else if (TSDB_DATA_TYPE_BOOL == pVal->node.resType.type) {
X
Xiaoyu Wang 已提交
1070 1071 1072
      value = pVal->datum.b;
    } else {
      value = pVal->datum.i;
X
Xiaoyu Wang 已提交
1073
    }
X
Xiaoyu Wang 已提交
1074
    pVal->datum.i = value;
wmmhello's avatar
wmmhello 已提交
1075 1076 1077 1078 1079 1080 1081 1082
    return TSDB_CODE_SUCCESS;
  } else if (IS_VAR_DATA_TYPE(pVal->node.resType.type) || TSDB_DATA_TYPE_TIMESTAMP == pVal->node.resType.type) {
    if (TSDB_CODE_SUCCESS == taosParseTime(pVal->literal, &pVal->datum.i, pVal->node.resType.bytes,
                                           pVal->node.resType.precision, tsDaylight)) {
      return TSDB_CODE_SUCCESS;
    }
    char* pEnd = NULL;
    pVal->datum.i = taosStr2Int64(pVal->literal, &pEnd, 10);
X
Xiaoyu Wang 已提交
1083
    return (NULL != pEnd && '\0' == *pEnd) ? TSDB_CODE_SUCCESS : TSDB_CODE_PAR_WRONG_VALUE_TYPE;
wmmhello's avatar
wmmhello 已提交
1084
  } else {
X
Xiaoyu Wang 已提交
1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100
    return TSDB_CODE_PAR_WRONG_VALUE_TYPE;
  }
}

static int32_t parseBoolFromValueNode(STranslateContext* pCxt, SValueNode* pVal) {
  if (IS_VAR_DATA_TYPE(pVal->node.resType.type) || TSDB_DATA_TYPE_BOOL == pVal->node.resType.type) {
    pVal->datum.b = (0 == strcasecmp(pVal->literal, "true"));
    return TSDB_CODE_SUCCESS;
  } else if (IS_INTEGER_TYPE(pVal->node.resType.type)) {
    pVal->datum.b = (0 != taosStr2Int64(pVal->literal, NULL, 10));
    return TSDB_CODE_SUCCESS;
  } else if (IS_FLOAT_TYPE(pVal->node.resType.type)) {
    pVal->datum.b = (0 != taosStr2Double(pVal->literal, NULL));
    return TSDB_CODE_SUCCESS;
  } else {
    return TSDB_CODE_PAR_WRONG_VALUE_TYPE;
wmmhello's avatar
wmmhello 已提交
1101
  }
1102 1103
}

1104 1105 1106 1107
static EDealRes translateDurationValue(STranslateContext* pCxt, SValueNode* pVal) {
  if (parseNatualDuration(pVal->literal, strlen(pVal->literal), &pVal->datum.i, &pVal->unit,
                          pVal->node.resType.precision) != TSDB_CODE_SUCCESS) {
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
1108
  }
1109 1110 1111 1112 1113 1114 1115 1116 1117 1118
  *(int64_t*)&pVal->typeData = pVal->datum.i;
  return DEAL_RES_CONTINUE;
}

static EDealRes translateNormalValue(STranslateContext* pCxt, SValueNode* pVal, SDataType targetDt, bool strict) {
  int32_t code = TSDB_CODE_SUCCESS;
  switch (targetDt.type) {
    case TSDB_DATA_TYPE_BOOL:
      if (TSDB_CODE_SUCCESS != parseBoolFromValueNode(pCxt, pVal)) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1119
      }
1120 1121 1122 1123 1124 1125
      *(bool*)&pVal->typeData = pVal->datum.b;
      break;
    case TSDB_DATA_TYPE_TINYINT: {
      code = toInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.i);
      if (strict && (TSDB_CODE_SUCCESS != code || !IS_VALID_TINYINT(pVal->datum.i))) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1126
      }
1127 1128 1129 1130 1131 1132 1133
      *(int8_t*)&pVal->typeData = pVal->datum.i;
      break;
    }
    case TSDB_DATA_TYPE_SMALLINT: {
      code = toInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.i);
      if (strict && (TSDB_CODE_SUCCESS != code || !IS_VALID_SMALLINT(pVal->datum.i))) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1134
      }
1135 1136 1137 1138 1139 1140 1141
      *(int16_t*)&pVal->typeData = pVal->datum.i;
      break;
    }
    case TSDB_DATA_TYPE_INT: {
      code = toInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.i);
      if (strict && (TSDB_CODE_SUCCESS != code || !IS_VALID_INT(pVal->datum.i))) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1142
      }
1143 1144 1145 1146 1147
      *(int32_t*)&pVal->typeData = pVal->datum.i;
      break;
    }
    case TSDB_DATA_TYPE_BIGINT: {
      code = toInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.i);
X
Xiaoyu Wang 已提交
1148
      if (strict && TSDB_CODE_SUCCESS != code) {
1149
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1150
      }
1151 1152 1153 1154 1155
      *(int64_t*)&pVal->typeData = pVal->datum.i;
      break;
    }
    case TSDB_DATA_TYPE_UTINYINT: {
      code = toUInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.u);
X
Xiaoyu Wang 已提交
1156
      if (strict && (TSDB_CODE_SUCCESS != code || pVal->datum.u > UINT8_MAX)) {
1157
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1158
      }
1159 1160 1161 1162 1163
      *(uint8_t*)&pVal->typeData = pVal->datum.u;
      break;
    }
    case TSDB_DATA_TYPE_USMALLINT: {
      code = toUInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.u);
X
Xiaoyu Wang 已提交
1164
      if (strict && (TSDB_CODE_SUCCESS != code || pVal->datum.u > UINT16_MAX)) {
1165
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
1166
      }
1167 1168 1169 1170 1171
      *(uint16_t*)&pVal->typeData = pVal->datum.u;
      break;
    }
    case TSDB_DATA_TYPE_UINT: {
      code = toUInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.u);
X
Xiaoyu Wang 已提交
1172
      if (strict && (TSDB_CODE_SUCCESS != code || pVal->datum.u > UINT32_MAX)) {
1173
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
D
dapan1121 已提交
1174
      }
1175 1176 1177 1178 1179
      *(uint32_t*)&pVal->typeData = pVal->datum.u;
      break;
    }
    case TSDB_DATA_TYPE_UBIGINT: {
      code = toUInteger(pVal->literal, strlen(pVal->literal), 10, &pVal->datum.u);
X
Xiaoyu Wang 已提交
1180
      if (strict && TSDB_CODE_SUCCESS != code) {
1181
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
1182
      }
1183 1184 1185 1186 1187
      *(uint64_t*)&pVal->typeData = pVal->datum.u;
      break;
    }
    case TSDB_DATA_TYPE_FLOAT: {
      pVal->datum.d = taosStr2Double(pVal->literal, NULL);
X
Xiaoyu Wang 已提交
1188 1189 1190
      if (strict && !IS_VALID_FLOAT(pVal->datum.d)) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
      }
1191 1192 1193 1194 1195
      *(float*)&pVal->typeData = pVal->datum.d;
      break;
    }
    case TSDB_DATA_TYPE_DOUBLE: {
      pVal->datum.d = taosStr2Double(pVal->literal, NULL);
X
Xiaoyu Wang 已提交
1196 1197 1198 1199
      if (strict && (((pVal->datum.d == HUGE_VAL || pVal->datum.d == -HUGE_VAL) && errno == ERANGE) ||
                     isinf(pVal->datum.d) || isnan(pVal->datum.d))) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
      }
1200 1201 1202 1203
      *(double*)&pVal->typeData = pVal->datum.d;
      break;
    }
    case TSDB_DATA_TYPE_VARCHAR:
D
Dingle Zhang 已提交
1204 1205
    case TSDB_DATA_TYPE_VARBINARY:
    case TSDB_DATA_TYPE_GEOMETRY: {
1206
      if (strict && (pVal->node.resType.bytes > targetDt.bytes - VARSTR_HEADER_SIZE)) {
1207 1208
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
      }
1209 1210 1211
      pVal->datum.p = taosMemoryCalloc(1, targetDt.bytes + 1);
      if (NULL == pVal->datum.p) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_OUT_OF_MEMORY);
1212
      }
1213 1214 1215 1216 1217 1218 1219 1220
      int32_t len = TMIN(targetDt.bytes - VARSTR_HEADER_SIZE, pVal->node.resType.bytes);
      varDataSetLen(pVal->datum.p, len);
      strncpy(varDataVal(pVal->datum.p), pVal->literal, len);
      break;
    }
    case TSDB_DATA_TYPE_TIMESTAMP: {
      if (TSDB_CODE_SUCCESS != parseTimeFromValueNode(pCxt, pVal)) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
1221
      }
1222 1223 1224 1225 1226 1227 1228
      *(int64_t*)&pVal->typeData = pVal->datum.i;
      break;
    }
    case TSDB_DATA_TYPE_NCHAR: {
      pVal->datum.p = taosMemoryCalloc(1, targetDt.bytes + 1);
      if (NULL == pVal->datum.p) {
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_OUT_OF_MEMORY);
1229
      }
1230

1231 1232 1233
      int32_t len = 0;
      if (!taosMbsToUcs4(pVal->literal, strlen(pVal->literal), (TdUcs4*)varDataVal(pVal->datum.p),
                         targetDt.bytes - VARSTR_HEADER_SIZE, &len)) {
1234
        return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
1235 1236 1237
      }
      varDataSetLen(pVal->datum.p, len);
      break;
1238
    }
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
    case TSDB_DATA_TYPE_DECIMAL:
    case TSDB_DATA_TYPE_BLOB:
      return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pVal->literal);
    default:
      break;
  }
  return DEAL_RES_CONTINUE;
}

static EDealRes translateValueImpl(STranslateContext* pCxt, SValueNode* pVal, SDataType targetDt, bool strict) {
  if (pVal->placeholderNo > 0 || pVal->isNull) {
    return DEAL_RES_CONTINUE;
  }

  if (TSDB_DATA_TYPE_NULL == pVal->node.resType.type) {
    pVal->translate = true;
    pVal->isNull = true;
    return DEAL_RES_CONTINUE;
  }

  pVal->node.resType.precision = getPrecisionFromCurrStmt(pCxt->pCurrStmt, targetDt.precision);

  EDealRes res = DEAL_RES_CONTINUE;
  if (pVal->isDuration) {
    res = translateDurationValue(pCxt, pVal);
  } else {
    res = translateNormalValue(pCxt, pVal, targetDt, strict);
1266
  }
5
54liuyao 已提交
1267 1268
  pVal->node.resType.type = targetDt.type;
  pVal->node.resType.bytes = targetDt.bytes;
D
dapan1121 已提交
1269
  pVal->node.resType.scale = pVal->unit;
1270
  pVal->translate = true;
1271 1272 1273
  if (!strict && TSDB_DATA_TYPE_UBIGINT == pVal->node.resType.type && pVal->datum.u <= INT64_MAX) {
    pVal->node.resType.type = TSDB_DATA_TYPE_BIGINT;
  }
1274
  return res;
1275
}
wmmhello's avatar
wmmhello 已提交
1276

1277
static int32_t calcTypeBytes(SDataType dt) {
D
Dingle Zhang 已提交
1278
  if (TSDB_DATA_TYPE_BINARY == dt.type || TSDB_DATA_TYPE_GEOMETRY == dt.type) {
1279 1280 1281 1282 1283 1284
    return dt.bytes + VARSTR_HEADER_SIZE;
  } else if (TSDB_DATA_TYPE_NCHAR == dt.type) {
    return dt.bytes * TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE;
  } else {
    return dt.bytes;
  }
1285 1286
}

1287
static EDealRes translateValue(STranslateContext* pCxt, SValueNode* pVal) {
1288 1289 1290 1291
  if (pVal->translate) {
    return TSDB_CODE_SUCCESS;
  }

1292 1293
  SDataType dt = pVal->node.resType;
  dt.bytes = calcTypeBytes(dt);
1294
  return translateValueImpl(pCxt, pVal, dt, false);
1295 1296
}

1297 1298 1299 1300
static int32_t doTranslateValue(STranslateContext* pCxt, SValueNode* pVal) {
  return DEAL_RES_ERROR == translateValue(pCxt, pVal) ? pCxt->errCode : TSDB_CODE_SUCCESS;
}

1301
static bool isMultiResFunc(SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
1302 1303 1304 1305 1306 1307
  if (NULL == pNode) {
    return false;
  }
  if (QUERY_NODE_FUNCTION != nodeType(pNode) || !fmIsMultiResFunc(((SFunctionNode*)pNode)->funcId)) {
    return false;
  }
1308 1309 1310
  if (FUNCTION_TYPE_TAGS == ((SFunctionNode*)pNode)->funcType) {
    return true;
  }
wmmhello's avatar
wmmhello 已提交
1311 1312 1313 1314 1315 1316
  SNodeList* pParameterList = ((SFunctionNode*)pNode)->pParameterList;
  if (LIST_LENGTH(pParameterList) > 1) {
    return true;
  }
  SNode* pParam = nodesListGetNode(pParameterList, 0);
  return (QUERY_NODE_COLUMN == nodeType(pParam) ? 0 == strcmp(((SColumnNode*)pParam)->colName, "*") : false);
1317 1318
}

D
dapan1121 已提交
1319 1320 1321 1322
static bool dataTypeEqual(const SDataType* l, const SDataType* r) {
  return (l->type == r->type && l->bytes == r->bytes && l->precision == r->precision && l->scale == r->scale);
}

X
Xiaoyu Wang 已提交
1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335
// 0 means equal, 1 means the left shall prevail, -1 means the right shall prevail
static int32_t dataTypeComp(const SDataType* l, const SDataType* r) {
  if (l->type != r->type) {
    return 1;
  }

  if (l->bytes != r->bytes) {
    return l->bytes > r->bytes ? 1 : -1;
  }

  return (l->precision == r->precision && l->scale == r->scale) ? 0 : 1;
}

1336
static EDealRes translateOperator(STranslateContext* pCxt, SOperatorNode* pOp) {
wmmhello's avatar
wmmhello 已提交
1337 1338 1339 1340 1341 1342 1343
  if (isMultiResFunc(pOp->pLeft)) {
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, ((SExprNode*)(pOp->pLeft))->aliasName);
  }
  if (isMultiResFunc(pOp->pRight)) {
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, ((SExprNode*)(pOp->pRight))->aliasName);
  }

1344 1345
  if (TSDB_CODE_SUCCESS != scalarGetOperatorResultType(pOp)) {
    return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, pOp->node.aliasName);
wmmhello's avatar
wmmhello 已提交
1346
  }
1347

wmmhello's avatar
wmmhello 已提交
1348
  return DEAL_RES_CONTINUE;
1349 1350
}

1351
static EDealRes haveVectorFunction(SNode* pNode, void* pContext) {
1352 1353
  if (isAggFunc(pNode) || isIndefiniteRowsFunc(pNode) || isWindowPseudoColumnFunc(pNode) ||
      isInterpPseudoColumnFunc(pNode)) {
wmmhello's avatar
wmmhello 已提交
1354 1355 1356 1357
    *((bool*)pContext) = true;
    return DEAL_RES_END;
  }
  return DEAL_RES_CONTINUE;
X
Xiaoyu Wang 已提交
1358 1359
}

1360
static int32_t findTable(STranslateContext* pCxt, const char* pTableAlias, STableNode** pOutput) {
wmmhello's avatar
wmmhello 已提交
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
  SArray* pTables = taosArrayGetP(pCxt->pNsLevel, pCxt->currLevel);
  size_t  nums = taosArrayGetSize(pTables);
  for (size_t i = 0; i < nums; ++i) {
    STableNode* pTable = taosArrayGetP(pTables, i);
    if (NULL == pTableAlias || 0 == strcmp(pTable->tableAlias, pTableAlias)) {
      *pOutput = pTable;
      return TSDB_CODE_SUCCESS;
    }
  }
  return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TABLE_NOT_EXIST, pTableAlias);
1371 1372 1373
}

static bool isCountStar(SFunctionNode* pFunc) {
wmmhello's avatar
wmmhello 已提交
1374 1375 1376 1377 1378
  if (FUNCTION_TYPE_COUNT != pFunc->funcType || 1 != LIST_LENGTH(pFunc->pParameterList)) {
    return false;
  }
  SNode* pPara = nodesListGetNode(pFunc->pParameterList, 0);
  return (QUERY_NODE_COLUMN == nodeType(pPara) && 0 == strcmp(((SColumnNode*)pPara)->colName, "*"));
1379 1380 1381 1382
}

// count(*) is rewritten as count(ts) for scannning optimization
static int32_t rewriteCountStar(STranslateContext* pCxt, SFunctionNode* pCount) {
1383
  SColumnNode* pCol = (SColumnNode*)nodesListGetNode(pCount->pParameterList, 0);
wmmhello's avatar
wmmhello 已提交
1384 1385 1386
  STableNode*  pTable = NULL;
  int32_t      code = findTable(pCxt, ('\0' == pCol->tableAlias[0] ? NULL : pCol->tableAlias), &pTable);
  if (TSDB_CODE_SUCCESS == code && QUERY_NODE_REAL_TABLE == nodeType(pTable)) {
X
Xiaoyu Wang 已提交
1387
    setColumnInfoBySchema((SRealTableNode*)pTable, ((SRealTableNode*)pTable)->pMeta->schema, -1, pCol);
wmmhello's avatar
wmmhello 已提交
1388 1389
  }
  return code;
1390 1391
}

D
dapan1121 已提交
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
static bool isCountNotNullValue(SFunctionNode* pFunc) {
  if (FUNCTION_TYPE_COUNT != pFunc->funcType || 1 != LIST_LENGTH(pFunc->pParameterList)) {
    return false;
  }
  SNode* pPara = nodesListGetNode(pFunc->pParameterList, 0);
  return (QUERY_NODE_VALUE == nodeType(pPara) && !((SValueNode*)pPara)->isNull);
}

// count(1) is rewritten as count(ts) for scannning optimization
static int32_t rewriteCountNotNullValue(STranslateContext* pCxt, SFunctionNode* pCount) {
  SValueNode* pValue = (SValueNode*)nodesListGetNode(pCount->pParameterList, 0);
1403 1404
  STableNode* pTable = NULL;
  int32_t     code = findTable(pCxt, NULL, &pTable);
D
dapan1121 已提交
1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417
  if (TSDB_CODE_SUCCESS == code && QUERY_NODE_REAL_TABLE == nodeType(pTable)) {
    SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
    if (NULL == pCol) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    } else {
      setColumnInfoBySchema((SRealTableNode*)pTable, ((SRealTableNode*)pTable)->pMeta->schema, -1, pCol);
      NODES_DESTORY_LIST(pCount->pParameterList);
      code = nodesListMakeAppend(&pCount->pParameterList, (SNode*)pCol);
    }
  }
  return code;
}

X
Xiaoyu Wang 已提交
1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
static bool isCountTbname(SFunctionNode* pFunc) {
  if (FUNCTION_TYPE_COUNT != pFunc->funcType || 1 != LIST_LENGTH(pFunc->pParameterList)) {
    return false;
  }
  SNode* pPara = nodesListGetNode(pFunc->pParameterList, 0);
  return (QUERY_NODE_FUNCTION == nodeType(pPara) && FUNCTION_TYPE_TBNAME == ((SFunctionNode*)pPara)->funcType);
}

// count(tbname) is rewritten as count(ts) for scannning optimization
static int32_t rewriteCountTbname(STranslateContext* pCxt, SFunctionNode* pCount) {
  SFunctionNode* pTbname = (SFunctionNode*)nodesListGetNode(pCount->pParameterList, 0);
  const char*    pTableAlias = NULL;
  if (LIST_LENGTH(pTbname->pParameterList) > 0) {
    pTableAlias = ((SValueNode*)nodesListGetNode(pTbname->pParameterList, 0))->literal;
  }
  STableNode* pTable = NULL;
  int32_t     code = findTable(pCxt, pTableAlias, &pTable);
  if (TSDB_CODE_SUCCESS == code) {
    SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
    if (NULL == pCol) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    } else {
      setColumnInfoBySchema((SRealTableNode*)pTable, ((SRealTableNode*)pTable)->pMeta->schema, -1, pCol);
      NODES_DESTORY_LIST(pCount->pParameterList);
      code = nodesListMakeAppend(&pCount->pParameterList, (SNode*)pCol);
    }
  }
  return code;
}

1448
static bool hasInvalidFuncNesting(SNodeList* pParameterList) {
wmmhello's avatar
wmmhello 已提交
1449
  bool hasInvalidFunc = false;
1450
  nodesWalkExprs(pParameterList, haveVectorFunction, &hasInvalidFunc);
wmmhello's avatar
wmmhello 已提交
1451 1452 1453 1454
  return hasInvalidFunc;
}

static int32_t getFuncInfo(STranslateContext* pCxt, SFunctionNode* pFunc) {
1455 1456
  // the time precision of the function execution environment
  pFunc->node.resType.precision = getPrecisionFromCurrStmt(pCxt->pCurrStmt, TSDB_TIME_PRECISION_MILLI);
1457 1458 1459 1460 1461
  int32_t code = fmGetFuncInfo(pFunc, pCxt->msgBuf.buf, pCxt->msgBuf.len);
  if (TSDB_CODE_FUNC_NOT_BUILTIN_FUNTION == code) {
    code = getUdfInfo(pCxt, pFunc);
  }
  return code;
1462 1463
}

X
Xiaoyu Wang 已提交
1464
static int32_t translateAggFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
X
Xiaoyu Wang 已提交
1465 1466 1467
  if (!fmIsAggFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
1468 1469 1470 1471 1472 1473
  if (beforeHaving(pCxt->currClause)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_ILLEGAL_USE_AGG_FUNCTION);
  }
  if (hasInvalidFuncNesting(pFunc->pParameterList)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AGG_FUNC_NESTING);
  }
1474 1475 1476
  // The auto-generated COUNT function in the DELETE statement is legal
  if (isSelectStmt(pCxt->pCurrStmt) &&
      (((SSelectStmt*)pCxt->pCurrStmt)->hasIndefiniteRowsFunc || ((SSelectStmt*)pCxt->pCurrStmt)->hasMultiRowsFunc)) {
X
Xiaoyu Wang 已提交
1477 1478 1479 1480 1481 1482
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }

  if (isCountStar(pFunc)) {
    return rewriteCountStar(pCxt, pFunc);
  }
D
dapan1121 已提交
1483 1484
  if (isCountNotNullValue(pFunc)) {
    return rewriteCountNotNullValue(pCxt, pFunc);
1485
  }
X
Xiaoyu Wang 已提交
1486 1487 1488
  if (isCountTbname(pFunc)) {
    return rewriteCountTbname(pCxt, pFunc);
  }
X
Xiaoyu Wang 已提交
1489 1490 1491 1492
  return TSDB_CODE_SUCCESS;
}

static int32_t translateScanPseudoColumnFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
X
Xiaoyu Wang 已提交
1493 1494 1495
  if (!fmIsScanPseudoColumnFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
1496
  if (0 == LIST_LENGTH(pFunc->pParameterList)) {
1497
    if (!isSelectStmt(pCxt->pCurrStmt) || NULL == ((SSelectStmt*)pCxt->pCurrStmt)->pFromTable ||
1498
        QUERY_NODE_REAL_TABLE != nodeType(((SSelectStmt*)pCxt->pCurrStmt)->pFromTable)) {
X
Xiaoyu Wang 已提交
1499 1500 1501
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TBNAME);
    }
  } else {
1502
    SValueNode* pVal = (SValueNode*)nodesListGetNode(pFunc->pParameterList, 0);
X
Xiaoyu Wang 已提交
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
    STableNode* pTable = NULL;
    pCxt->errCode = findTable(pCxt, pVal->literal, &pTable);
    if (TSDB_CODE_SUCCESS == pCxt->errCode && (NULL == pTable || QUERY_NODE_REAL_TABLE != nodeType(pTable))) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TBNAME);
    }
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t translateIndefiniteRowsFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
X
Xiaoyu Wang 已提交
1513 1514 1515
  if (!fmIsIndefiniteRowsFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
1516
  if (!isSelectStmt(pCxt->pCurrStmt) || SQL_CLAUSE_SELECT != pCxt->currClause) {
X
Xiaoyu Wang 已提交
1517 1518
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }
1519 1520
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
  if (pSelect->hasAggFuncs || pSelect->hasMultiRowsFunc ||
1521 1522
      (pSelect->hasIndefiniteRowsFunc &&
       (FUNC_RETURN_ROWS_INDEFINITE == pSelect->returnRows || pSelect->returnRows != fmGetFuncReturnRows(pFunc)))) {
1523 1524 1525
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }
  if (NULL != pSelect->pWindow || NULL != pSelect->pGroupByList) {
1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                   "%s function is not supported in window query or group query", pFunc->functionName);
  }
  if (hasInvalidFuncNesting(pFunc->pParameterList)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AGG_FUNC_NESTING);
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t translateMultiRowsFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsMultiRowsFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
  if (!isSelectStmt(pCxt->pCurrStmt) || SQL_CLAUSE_SELECT != pCxt->currClause ||
1540 1541
      ((SSelectStmt*)pCxt->pCurrStmt)->hasIndefiniteRowsFunc || ((SSelectStmt*)pCxt->pCurrStmt)->hasAggFuncs ||
      ((SSelectStmt*)pCxt->pCurrStmt)->hasMultiRowsFunc) {
X
Xiaoyu Wang 已提交
1542 1543 1544 1545 1546 1547 1548 1549
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }
  if (hasInvalidFuncNesting(pFunc->pParameterList)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AGG_FUNC_NESTING);
  }
  return TSDB_CODE_SUCCESS;
}

1550 1551 1552 1553 1554 1555 1556 1557
static int32_t translateInterpFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsInterpFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
  if (!isSelectStmt(pCxt->pCurrStmt) || SQL_CLAUSE_SELECT != pCxt->currClause) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
1558 1559
  SNode*       pTable = pSelect->pFromTable;

1560 1561 1562
  if (pSelect->hasAggFuncs || pSelect->hasMultiRowsFunc || pSelect->hasIndefiniteRowsFunc) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC);
  }
1563 1564 1565 1566 1567 1568

  if (pSelect->hasInterpFunc && (FUNC_RETURN_ROWS_INDEFINITE == pSelect->returnRows || pSelect->returnRows != fmGetFuncReturnRows(pFunc))) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                "%s ignoring null value options cannot be used when applying to multiple columns", pFunc->functionName);
  }

1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
  if (NULL != pSelect->pWindow || NULL != pSelect->pGroupByList) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                   "%s function is not supported in window query or group query", pFunc->functionName);
  }
  if (hasInvalidFuncNesting(pFunc->pParameterList)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AGG_FUNC_NESTING);
  }
  return TSDB_CODE_SUCCESS;
}

1579 1580 1581 1582
static int32_t translateInterpPseudoColumnFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsInterpPseudoColumnFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
1583 1584 1585
  if (!isSelectStmt(pCxt->pCurrStmt)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                   "%s must be used in select statements", pFunc->functionName);
1586
  }
1587 1588 1589
  if (pCxt->currClause == SQL_CLAUSE_WHERE) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_INTERP_CLAUSE,
                                   "%s is not allowed in where clause", pFunc->functionName);
1590 1591 1592 1593
  }
  return TSDB_CODE_SUCCESS;
}

1594 1595 1596 1597 1598 1599 1600 1601 1602
static int32_t translateTimelineFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsTimelineFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
  if (!isSelectStmt(pCxt->pCurrStmt)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                   "%s function must be used in select statements", pFunc->functionName);
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
X
Xiaoyu Wang 已提交
1603
  if (NULL != pSelect->pFromTable && QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&
1604
      !isGlobalTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery) && !isTimeLineAlignedQuery(pCxt->pCurrStmt)) {
1605 1606 1607 1608 1609 1610
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                   "%s function requires valid time series input", pFunc->functionName);
  }
  return TSDB_CODE_SUCCESS;
}

1611 1612 1613 1614
static int32_t translateDateTimeFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsDateTimeFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
G
Ganlin Zhao 已提交
1615 1616 1617 1618 1619

  if (!isSelectStmt(pCxt->pCurrStmt)) {
    return TSDB_CODE_SUCCESS;
  }

1620 1621 1622 1623 1624 1625
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
  pFunc->node.resType.precision = pSelect->precision;

  return TSDB_CODE_SUCCESS;
}

1626 1627 1628 1629 1630 1631 1632 1633 1634
static bool hasFillClause(SNode* pCurrStmt) {
  if (!isSelectStmt(pCurrStmt)) {
    return false;
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCurrStmt;
  return NULL != pSelect->pWindow && QUERY_NODE_INTERVAL_WINDOW == nodeType(pSelect->pWindow) &&
         NULL != ((SIntervalWindowNode*)pSelect->pWindow)->pFill;
}

X
Xiaoyu Wang 已提交
1635 1636 1637 1638
static int32_t translateForbidFillFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsForbidFillFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
1639
  if (hasFillClause(pCxt->pCurrStmt)) {
X
Xiaoyu Wang 已提交
1640 1641 1642 1643 1644
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_FILL_NOT_ALLOWED_FUNC, pFunc->functionName);
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1645 1646 1647 1648
static int32_t translateWindowPseudoColumnFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsWindowPseudoColumnFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
1649
  if (!isSelectStmt(pCxt->pCurrStmt) || NULL == ((SSelectStmt*)pCxt->pCurrStmt)->pWindow) {
X
Xiaoyu Wang 已提交
1650 1651
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_WINDOW_PC);
  }
X
Xiaoyu Wang 已提交
1652
  if (beforeWindow(pCxt->currClause)) {
X
Xiaoyu Wang 已提交
1653 1654 1655
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_WINDOW_PC, "There mustn't be %s",
                                   pFunc->functionName);
  }
X
Xiaoyu Wang 已提交
1656 1657 1658
  return TSDB_CODE_SUCCESS;
}

1659 1660 1661 1662 1663 1664 1665 1666 1667 1668
static int32_t translateForbidStreamFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsForbidStreamFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
  if (pCxt->createStream) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_STREAM_NOT_ALLOWED_FUNC, pFunc->functionName);
  }
  return TSDB_CODE_SUCCESS;
}

1669 1670 1671 1672
static int32_t translateRepeatScanFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsRepeatScanFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
1673
  if (!isSelectStmt(pCxt->pCurrStmt)) {
X
Xiaoyu Wang 已提交
1674 1675 1676 1677 1678 1679 1680 1681
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_ONLY_SUPPORT_SINGLE_TABLE,
                                   "%s is only supported in single table query", pFunc->functionName);
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
  SNode*       pTable = pSelect->pFromTable;
  // select percentile() without from clause is also valid
  if ((NULL != pTable && (QUERY_NODE_REAL_TABLE != nodeType(pTable) ||
                          (TSDB_CHILD_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType &&
1682
                           TSDB_NORMAL_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType)))) {
X
Xiaoyu Wang 已提交
1683 1684
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_ONLY_SUPPORT_SINGLE_TABLE,
                                   "%s is only supported in single table query", pFunc->functionName);
1685
  }
1686 1687
  if (NULL != pSelect->pPartitionByList) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
X
Xiaoyu Wang 已提交
1688
                                   "%s function is not supported in partition query", pFunc->functionName);
1689
  }
X
Xiaoyu Wang 已提交
1690
  return TSDB_CODE_SUCCESS;
1691 1692
}

1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707
static int32_t translateBlockDistFunc(STranslateContext* pCtx, SFunctionNode* pFunc) {
  if (!fmIsBlockDistFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
  if (!isSelectStmt(pCtx->pCurrStmt)) {
    return generateSyntaxErrMsgExt(&pCtx->msgBuf, TSDB_CODE_PAR_ONLY_SUPPORT_SINGLE_TABLE,
                                   "%s is only supported in single table query", pFunc->functionName);
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCtx->pCurrStmt;
  SNode*       pTable = pSelect->pFromTable;
  if (NULL != pTable && (QUERY_NODE_REAL_TABLE != nodeType(pTable) ||
                         (TSDB_SUPER_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType &&
                          TSDB_CHILD_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType &&
                          TSDB_NORMAL_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType))) {
    return generateSyntaxErrMsgExt(&pCtx->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
1708 1709
                                   "%s is only supported on super table, child table or normal table",
                                   pFunc->functionName);
1710
  }
X
Xiaoyu Wang 已提交
1711
  return TSDB_CODE_SUCCESS;
1712 1713
}

1714 1715 1716 1717 1718 1719 1720 1721 1722 1723
static bool isStar(SNode* pNode) {
  return (QUERY_NODE_COLUMN == nodeType(pNode)) && ('\0' == ((SColumnNode*)pNode)->tableAlias[0]) &&
         (0 == strcmp(((SColumnNode*)pNode)->colName, "*"));
}

static bool isTableStar(SNode* pNode) {
  return (QUERY_NODE_COLUMN == nodeType(pNode)) && ('\0' != ((SColumnNode*)pNode)->tableAlias[0]) &&
         (0 == strcmp(((SColumnNode*)pNode)->colName, "*"));
}

1724 1725
static bool isStarParam(SNode* pNode) { return isStar(pNode) || isTableStar(pNode); }

1726 1727 1728 1729
static int32_t translateMultiResFunc(STranslateContext* pCxt, SFunctionNode* pFunc) {
  if (!fmIsMultiResFunc(pFunc->funcId)) {
    return TSDB_CODE_SUCCESS;
  }
1730
  if (SQL_CLAUSE_SELECT != pCxt->currClause) {
1731
    SNode* pPara = nodesListGetNode(pFunc->pParameterList, 0);
1732
    if (isStarParam(pPara)) {
1733 1734 1735 1736
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                     "%s(*) is only supported in SELECTed list", pFunc->functionName);
    }
  }
1737
  if (tsKeepColumnName && 1 == LIST_LENGTH(pFunc->pParameterList) && !pFunc->node.asAlias) {
1738
    strcpy(pFunc->node.userAlias, ((SExprNode*)nodesListGetNode(pFunc->pParameterList, 0))->userAlias);
1739
    strcpy(pFunc->node.aliasName, pFunc->node.userAlias);
1740
  }
1741 1742
  return TSDB_CODE_SUCCESS;
}
1743 1744 1745

static int32_t getMultiResFuncNum(SNodeList* pParameterList) {
  if (1 == LIST_LENGTH(pParameterList)) {
1746
    return isStarParam(nodesListGetNode(pParameterList, 0)) ? 2 : 1;
1747 1748 1749 1750
  }
  return LIST_LENGTH(pParameterList);
}

1751 1752 1753 1754 1755 1756 1757 1758 1759
static int32_t calcSelectFuncNum(SFunctionNode* pFunc, int32_t currSelectFuncNum) {
  if (fmIsCumulativeFunc(pFunc->funcId)) {
    return currSelectFuncNum > 0 ? currSelectFuncNum : 1;
  }
  return currSelectFuncNum + ((fmIsMultiResFunc(pFunc->funcId) && !fmIsLastRowFunc(pFunc->funcId))
                                  ? getMultiResFuncNum(pFunc->pParameterList)
                                  : 1);
}

1760 1761 1762
static void setFuncClassification(SNode* pCurrStmt, SFunctionNode* pFunc) {
  if (NULL != pCurrStmt && QUERY_NODE_SELECT_STMT == nodeType(pCurrStmt)) {
    SSelectStmt* pSelect = (SSelectStmt*)pCurrStmt;
X
Xiaoyu Wang 已提交
1763 1764
    pSelect->hasAggFuncs = pSelect->hasAggFuncs ? true : fmIsAggFunc(pFunc->funcId);
    pSelect->hasRepeatScanFuncs = pSelect->hasRepeatScanFuncs ? true : fmIsRepeatScanFunc(pFunc->funcId);
1765

1766 1767 1768
    if (fmIsIndefiniteRowsFunc(pFunc->funcId)) {
      pSelect->hasIndefiniteRowsFunc = true;
      pSelect->returnRows = fmGetFuncReturnRows(pFunc);
1769 1770
    } else if (fmIsInterpFunc(pFunc->funcId)) {
      pSelect->returnRows = fmGetFuncReturnRows(pFunc);
1771
    }
1772

1773
    pSelect->hasMultiRowsFunc = pSelect->hasMultiRowsFunc ? true : fmIsMultiRowsFunc(pFunc->funcId);
1774 1775
    if (fmIsSelectFunc(pFunc->funcId)) {
      pSelect->hasSelectFunc = true;
1776
      pSelect->selectFuncNum = calcSelectFuncNum(pFunc, pSelect->selectFuncNum);
1777
    } else if (fmIsVectorFunc(pFunc->funcId)) {
1778 1779
      pSelect->hasOtherVectorFunc = true;
    }
1780 1781
    pSelect->hasUniqueFunc = pSelect->hasUniqueFunc ? true : (FUNCTION_TYPE_UNIQUE == pFunc->funcType);
    pSelect->hasTailFunc = pSelect->hasTailFunc ? true : (FUNCTION_TYPE_TAIL == pFunc->funcType);
X
Xiaoyu Wang 已提交
1782
    pSelect->hasInterpFunc = pSelect->hasInterpFunc ? true : (FUNCTION_TYPE_INTERP == pFunc->funcType);
1783 1784
    pSelect->hasInterpPseudoColFunc =
        pSelect->hasInterpPseudoColFunc ? true : fmIsInterpPseudoColumnFunc(pFunc->funcId);
X
Xiaoyu Wang 已提交
1785
    pSelect->hasLastRowFunc = pSelect->hasLastRowFunc ? true : (FUNCTION_TYPE_LAST_ROW == pFunc->funcType);
X
Xiaoyu Wang 已提交
1786
    pSelect->hasLastFunc = pSelect->hasLastFunc ? true : (FUNCTION_TYPE_LAST == pFunc->funcType);
1787
    pSelect->hasTimeLineFunc = pSelect->hasTimeLineFunc ? true : fmIsTimelineFunc(pFunc->funcId);
X
Xiaoyu Wang 已提交
1788
    pSelect->hasUdaf = pSelect->hasUdaf ? true : fmIsUserDefinedFunc(pFunc->funcId) && fmIsAggFunc(pFunc->funcId);
1789
    pSelect->onlyHasKeepOrderFunc = pSelect->onlyHasKeepOrderFunc ? fmIsKeepOrderFunc(pFunc->funcId) : false;
X
Xiaoyu Wang 已提交
1790
  }
X
Xiaoyu Wang 已提交
1791 1792
}

1793
static int32_t rewriteFuncToValue(STranslateContext* pCxt, char* pLiteral, SNode** pNode) {
1794 1795 1796 1797 1798
  SValueNode* pVal = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
  if (NULL == pVal) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  strcpy(pVal->node.aliasName, ((SExprNode*)*pNode)->aliasName);
1799
  strcpy(pVal->node.userAlias, ((SExprNode*)*pNode)->userAlias);
1800 1801 1802 1803 1804
  pVal->node.resType = ((SExprNode*)*pNode)->resType;
  if (NULL == pLiteral) {
    pVal->isNull = true;
  } else {
    pVal->literal = pLiteral;
1805 1806 1807
    if (IS_VAR_DATA_TYPE(pVal->node.resType.type)) {
      pVal->node.resType.bytes = strlen(pLiteral);
    }
1808 1809
  }
  if (DEAL_RES_ERROR != translateValue(pCxt, pVal)) {
1810
    nodesDestroyNode(*pNode);
1811 1812 1813 1814 1815 1816 1817 1818 1819 1820
    *pNode = (SNode*)pVal;
  } else {
    nodesDestroyNode((SNode*)pVal);
  }
  return pCxt->errCode;
}

static int32_t rewriteDatabaseFunc(STranslateContext* pCxt, SNode** pNode) {
  char* pCurrDb = NULL;
  if (NULL != pCxt->pParseCxt->db) {
1821
    pCurrDb = taosStrdup((void*)pCxt->pParseCxt->db);
1822 1823
    if (NULL == pCurrDb) {
      return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
1824 1825
    }
  }
1826
  return rewriteFuncToValue(pCxt, pCurrDb, pNode);
1827
}
wmmhello's avatar
wmmhello 已提交
1828

1829
static int32_t rewriteClentVersionFunc(STranslateContext* pCxt, SNode** pNode) {
1830
  char* pVer = taosStrdup((void*)version);
1831 1832
  if (NULL == pVer) {
    return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
1833
  }
1834
  return rewriteFuncToValue(pCxt, pVer, pNode);
1835 1836 1837
}

static int32_t rewriteServerVersionFunc(STranslateContext* pCxt, SNode** pNode) {
1838
  char* pVer = taosStrdup((void*)pCxt->pParseCxt->svrVer);
1839 1840
  if (NULL == pVer) {
    return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
1841
  }
1842
  return rewriteFuncToValue(pCxt, pVer, pNode);
1843 1844 1845 1846 1847
}

static int32_t rewriteServerStatusFunc(STranslateContext* pCxt, SNode** pNode) {
  if (pCxt->pParseCxt->nodeOffline) {
    return TSDB_CODE_RPC_NETWORK_UNAVAIL;
X
Xiaoyu Wang 已提交
1848
  }
1849
  char* pStatus = taosStrdup((void*)"1");
1850
  return rewriteFuncToValue(pCxt, pStatus, pNode);
1851 1852 1853 1854 1855 1856
}

static int32_t rewriteUserFunc(STranslateContext* pCxt, SNode** pNode) {
  char    userConn[TSDB_USER_LEN + 1 + TSDB_FQDN_LEN] = {0};  // format 'user@host'
  int32_t len = snprintf(userConn, sizeof(userConn), "%s@", pCxt->pParseCxt->pUser);
  taosGetFqdn(userConn + len);
1857
  char* pUserConn = taosStrdup((void*)userConn);
1858 1859
  if (NULL == pUserConn) {
    return TSDB_CODE_OUT_OF_MEMORY;
X
Xiaoyu Wang 已提交
1860
  }
1861
  return rewriteFuncToValue(pCxt, pUserConn, pNode);
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878
}

static int32_t rewriteSystemInfoFunc(STranslateContext* pCxt, SNode** pNode) {
  switch (((SFunctionNode*)*pNode)->funcType) {
    case FUNCTION_TYPE_DATABASE:
      return rewriteDatabaseFunc(pCxt, pNode);
    case FUNCTION_TYPE_CLIENT_VERSION:
      return rewriteClentVersionFunc(pCxt, pNode);
    case FUNCTION_TYPE_SERVER_VERSION:
      return rewriteServerVersionFunc(pCxt, pNode);
    case FUNCTION_TYPE_SERVER_STATUS:
      return rewriteServerStatusFunc(pCxt, pNode);
    case FUNCTION_TYPE_CURRENT_USER:
    case FUNCTION_TYPE_USER:
      return rewriteUserFunc(pCxt, pNode);
    default:
      break;
X
Xiaoyu Wang 已提交
1879
  }
1880 1881 1882
  return TSDB_CODE_PAR_INTERNAL_ERROR;
}

1883
static int32_t translateNormalFunction(STranslateContext* pCxt, SFunctionNode* pFunc) {
1884 1885 1886
  int32_t code = translateAggFunc(pCxt, pFunc);
  if (TSDB_CODE_SUCCESS == code) {
    code = translateScanPseudoColumnFunc(pCxt, pFunc);
1887
  }
1888 1889
  if (TSDB_CODE_SUCCESS == code) {
    code = translateIndefiniteRowsFunc(pCxt, pFunc);
1890
  }
1891 1892 1893 1894 1895
  if (TSDB_CODE_SUCCESS == code) {
    code = translateForbidFillFunc(pCxt, pFunc);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateWindowPseudoColumnFunc(pCxt, pFunc);
1896
  }
1897 1898 1899
  if (TSDB_CODE_SUCCESS == code) {
    code = translateForbidStreamFunc(pCxt, pFunc);
  }
1900 1901 1902
  if (TSDB_CODE_SUCCESS == code) {
    code = translateRepeatScanFunc(pCxt, pFunc);
  }
1903 1904 1905
  if (TSDB_CODE_SUCCESS == code) {
    code = translateMultiResFunc(pCxt, pFunc);
  }
1906 1907 1908
  if (TSDB_CODE_SUCCESS == code) {
    code = translateMultiRowsFunc(pCxt, pFunc);
  }
1909 1910 1911
  if (TSDB_CODE_SUCCESS == code) {
    code = translateInterpFunc(pCxt, pFunc);
  }
1912 1913 1914
  if (TSDB_CODE_SUCCESS == code) {
    code = translateInterpPseudoColumnFunc(pCxt, pFunc);
  }
1915 1916 1917
  if (TSDB_CODE_SUCCESS == code) {
    code = translateTimelineFunc(pCxt, pFunc);
  }
1918 1919 1920
  if (TSDB_CODE_SUCCESS == code) {
    code = translateDateTimeFunc(pCxt, pFunc);
  }
1921 1922 1923
  if (TSDB_CODE_SUCCESS == code) {
    code = translateBlockDistFunc(pCxt, pFunc);
  }
1924 1925 1926 1927 1928 1929
  if (TSDB_CODE_SUCCESS == code) {
    setFuncClassification(pCxt->pCurrStmt, pFunc);
  }
  return code;
}

1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976
static int32_t rewriteQueryTimeFunc(STranslateContext* pCxt, int64_t val, SNode** pNode) {
  if (INT64_MIN == val || INT64_MAX == val) {
    return rewriteFuncToValue(pCxt, NULL, pNode);
  }

  char* pStr = taosMemoryCalloc(1, 20);
  if (NULL == pStr) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  snprintf(pStr, 20, "%" PRId64 "", val);
  return rewriteFuncToValue(pCxt, pStr, pNode);
}

static int32_t rewriteQstartFunc(STranslateContext* pCxt, SNode** pNode) {
  return rewriteQueryTimeFunc(pCxt, ((SSelectStmt*)pCxt->pCurrStmt)->timeRange.skey, pNode);
}

static int32_t rewriteQendFunc(STranslateContext* pCxt, SNode** pNode) {
  return rewriteQueryTimeFunc(pCxt, ((SSelectStmt*)pCxt->pCurrStmt)->timeRange.ekey, pNode);
}

static int32_t rewriteQdurationFunc(STranslateContext* pCxt, SNode** pNode) {
  STimeWindow range = ((SSelectStmt*)pCxt->pCurrStmt)->timeRange;
  if (INT64_MIN == range.skey || INT64_MAX == range.ekey) {
    return rewriteQueryTimeFunc(pCxt, INT64_MIN, pNode);
  }
  return rewriteQueryTimeFunc(pCxt, range.ekey - range.skey + 1, pNode);
}

static int32_t rewriteClientPseudoColumnFunc(STranslateContext* pCxt, SNode** pNode) {
  if (NULL == pCxt->pCurrStmt || QUERY_NODE_SELECT_STMT != nodeType(pCxt->pCurrStmt) ||
      pCxt->currClause <= SQL_CLAUSE_WHERE) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC, "Illegal pseudo column");
  }
  switch (((SFunctionNode*)*pNode)->funcType) {
    case FUNCTION_TYPE_QSTART:
      return rewriteQstartFunc(pCxt, pNode);
    case FUNCTION_TYPE_QEND:
      return rewriteQendFunc(pCxt, pNode);
    case FUNCTION_TYPE_QDURATION:
      return rewriteQdurationFunc(pCxt, pNode);
    default:
      break;
  }
  return TSDB_CODE_PAR_INTERNAL_ERROR;
}

1977 1978 1979 1980
static int32_t translateFunctionImpl(STranslateContext* pCxt, SFunctionNode** pFunc) {
  if (fmIsSystemInfoFunc((*pFunc)->funcId)) {
    return rewriteSystemInfoFunc(pCxt, (SNode**)pFunc);
  }
1981 1982 1983
  if (fmIsClientPseudoColumnFunc((*pFunc)->funcId)) {
    return rewriteClientPseudoColumnFunc(pCxt, (SNode**)pFunc);
  }
1984
  return translateNormalFunction(pCxt, *pFunc);
1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995
}

static EDealRes translateFunction(STranslateContext* pCxt, SFunctionNode** pFunc) {
  SNode* pParam = NULL;
  FOREACH(pParam, (*pFunc)->pParameterList) {
    if (isMultiResFunc(pParam)) {
      return generateDealNodeErrMsg(pCxt, TSDB_CODE_PAR_WRONG_VALUE_TYPE, ((SExprNode*)pParam)->aliasName);
    }
  }

  pCxt->errCode = getFuncInfo(pCxt, *pFunc);
X
Xiaoyu Wang 已提交
1996
  if (TSDB_CODE_SUCCESS == pCxt->errCode) {
1997
    pCxt->errCode = translateFunctionImpl(pCxt, pFunc);
wmmhello's avatar
wmmhello 已提交
1998 1999
  }
  return TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_CONTINUE : DEAL_RES_ERROR;
2000 2001 2002
}

static EDealRes translateExprSubquery(STranslateContext* pCxt, SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
2003
  return (TSDB_CODE_SUCCESS == translateSubquery(pCxt, pNode) ? DEAL_RES_CONTINUE : DEAL_RES_ERROR);
2004 2005
}

2006
static EDealRes translateLogicCond(STranslateContext* pCxt, SLogicConditionNode* pCond) {
wmmhello's avatar
wmmhello 已提交
2007 2008 2009
  pCond->node.resType.type = TSDB_DATA_TYPE_BOOL;
  pCond->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;
  return DEAL_RES_CONTINUE;
2010 2011
}

X
Xiaoyu Wang 已提交
2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037
static int32_t createCastFunc(STranslateContext* pCxt, SNode* pExpr, SDataType dt, SNode** pCast) {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  strcpy(pFunc->functionName, "cast");
  pFunc->node.resType = dt;
  if (TSDB_CODE_SUCCESS != nodesListMakeAppend(&pFunc->pParameterList, pExpr)) {
    nodesDestroyNode((SNode*)pFunc);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  if (TSDB_CODE_SUCCESS != getFuncInfo(pCxt, pFunc)) {
    nodesClearList(pFunc->pParameterList);
    pFunc->pParameterList = NULL;
    nodesDestroyNode((SNode*)pFunc);
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, ((SExprNode*)pExpr)->aliasName);
  }
  *pCast = (SNode*)pFunc;
  return TSDB_CODE_SUCCESS;
}

static EDealRes translateWhenThen(STranslateContext* pCxt, SWhenThenNode* pWhenThen) {
  pWhenThen->node.resType = ((SExprNode*)pWhenThen->pThen)->resType;
  return DEAL_RES_CONTINUE;
}

2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057
static bool isCondition(const SNode* pNode) {
  if (QUERY_NODE_OPERATOR == nodeType(pNode)) {
    return nodesIsComparisonOp((const SOperatorNode*)pNode);
  }
  return (QUERY_NODE_LOGIC_CONDITION == nodeType(pNode));
}

static int32_t rewriteIsTrue(SNode* pSrc, SNode** pIsTrue) {
  SOperatorNode* pOp = (SOperatorNode*)nodesMakeNode(QUERY_NODE_OPERATOR);
  if (NULL == pOp) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pOp->opType = OP_TYPE_IS_TRUE;
  pOp->pLeft = pSrc;
  pOp->node.resType.type = TSDB_DATA_TYPE_BOOL;
  pOp->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;
  *pIsTrue = (SNode*)pOp;
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2058 2059
static EDealRes translateCaseWhen(STranslateContext* pCxt, SCaseWhenNode* pCaseWhen) {
  bool   first = true;
2060
  bool   allNullThen = true;
X
Xiaoyu Wang 已提交
2061 2062
  SNode* pNode = NULL;
  FOREACH(pNode, pCaseWhen->pWhenThenList) {
2063 2064 2065 2066 2067 2068 2069 2070 2071
    SWhenThenNode* pWhenThen = (SWhenThenNode*)pNode;
    if (NULL == pCaseWhen->pCase && !isCondition(pWhenThen->pWhen)) {
      SNode* pIsTrue = NULL;
      pCxt->errCode = rewriteIsTrue(pWhenThen->pWhen, &pIsTrue);
      if (TSDB_CODE_SUCCESS != pCxt->errCode) {
        return DEAL_RES_ERROR;
      }
      pWhenThen->pWhen = pIsTrue;
    }
2072 2073 2074 2075 2076 2077 2078 2079

    SExprNode* pThenExpr = (SExprNode*)pNode;
    if (TSDB_DATA_TYPE_NULL == pThenExpr->resType.type) {
      continue;
    }
    allNullThen = false;
    if (first || dataTypeComp(&pCaseWhen->node.resType, &pThenExpr->resType) < 0) {
      pCaseWhen->node.resType = pThenExpr->resType;
X
Xiaoyu Wang 已提交
2080 2081 2082 2083
    }
    first = false;
  }

2084 2085 2086 2087 2088 2089 2090 2091 2092 2093
  if (allNullThen) {
    if (NULL != pCaseWhen->pElse) {
      pCaseWhen->node.resType = ((SExprNode*)pCaseWhen->pElse)->resType;
    } else {
      pCaseWhen->node.resType.type = TSDB_DATA_TYPE_NULL;
      pCaseWhen->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_NULL].bytes;
      return DEAL_RES_CONTINUE;
    }
  }

X
Xiaoyu Wang 已提交
2094 2095 2096
  FOREACH(pNode, pCaseWhen->pWhenThenList) {
    SWhenThenNode* pWhenThen = (SWhenThenNode*)pNode;
    if (!dataTypeEqual(&pCaseWhen->node.resType, &((SExprNode*)pNode)->resType)) {
2097 2098 2099 2100
      SNode* pCastFunc = NULL;
      pCxt->errCode = createCastFunc(pCxt, pWhenThen->pThen, pCaseWhen->node.resType, &pCastFunc);
      if (TSDB_CODE_SUCCESS != pCxt->errCode) {
        return DEAL_RES_ERROR;
X
Xiaoyu Wang 已提交
2101 2102 2103 2104 2105
      }
      pWhenThen->pThen = pCastFunc;
      pWhenThen->node.resType = pCaseWhen->node.resType;
    }
  }
X
Xiaoyu Wang 已提交
2106

X
Xiaoyu Wang 已提交
2107 2108
  if (NULL != pCaseWhen->pElse && !dataTypeEqual(&pCaseWhen->node.resType, &((SExprNode*)pCaseWhen->pElse)->resType)) {
    SNode* pCastFunc = NULL;
2109 2110 2111
    pCxt->errCode = createCastFunc(pCxt, pCaseWhen->pElse, pCaseWhen->node.resType, &pCastFunc);
    if (TSDB_CODE_SUCCESS != pCxt->errCode) {
      return DEAL_RES_ERROR;
X
Xiaoyu Wang 已提交
2112 2113 2114 2115 2116 2117 2118
    }
    pCaseWhen->pElse = pCastFunc;
    ((SExprNode*)pCaseWhen->pElse)->resType = pCaseWhen->node.resType;
  }
  return DEAL_RES_CONTINUE;
}

X
Xiaoyu Wang 已提交
2119
static EDealRes doTranslateExpr(SNode** pNode, void* pContext) {
wmmhello's avatar
wmmhello 已提交
2120
  STranslateContext* pCxt = (STranslateContext*)pContext;
X
Xiaoyu Wang 已提交
2121
  switch (nodeType(*pNode)) {
wmmhello's avatar
wmmhello 已提交
2122
    case QUERY_NODE_COLUMN:
X
Xiaoyu Wang 已提交
2123
      return translateColumn(pCxt, (SColumnNode**)pNode);
wmmhello's avatar
wmmhello 已提交
2124
    case QUERY_NODE_VALUE:
X
Xiaoyu Wang 已提交
2125
      return translateValue(pCxt, (SValueNode*)*pNode);
wmmhello's avatar
wmmhello 已提交
2126
    case QUERY_NODE_OPERATOR:
2127
      return translateOperator(pCxt, (SOperatorNode*)*pNode);
wmmhello's avatar
wmmhello 已提交
2128
    case QUERY_NODE_FUNCTION:
2129
      return translateFunction(pCxt, (SFunctionNode**)pNode);
wmmhello's avatar
wmmhello 已提交
2130
    case QUERY_NODE_LOGIC_CONDITION:
X
Xiaoyu Wang 已提交
2131
      return translateLogicCond(pCxt, (SLogicConditionNode*)*pNode);
wmmhello's avatar
wmmhello 已提交
2132
    case QUERY_NODE_TEMP_TABLE:
X
Xiaoyu Wang 已提交
2133
      return translateExprSubquery(pCxt, ((STempTableNode*)*pNode)->pSubquery);
X
Xiaoyu Wang 已提交
2134 2135 2136 2137
    case QUERY_NODE_WHEN_THEN:
      return translateWhenThen(pCxt, (SWhenThenNode*)*pNode);
    case QUERY_NODE_CASE_WHEN:
      return translateCaseWhen(pCxt, (SCaseWhenNode*)*pNode);
wmmhello's avatar
wmmhello 已提交
2138 2139 2140 2141
    default:
      break;
  }
  return DEAL_RES_CONTINUE;
2142 2143
}

X
Xiaoyu Wang 已提交
2144 2145
static int32_t translateExpr(STranslateContext* pCxt, SNode** pNode) {
  nodesRewriteExprPostOrder(pNode, doTranslateExpr, pCxt);
wmmhello's avatar
wmmhello 已提交
2146
  return pCxt->errCode;
2147 2148 2149
}

static int32_t translateExprList(STranslateContext* pCxt, SNodeList* pList) {
X
Xiaoyu Wang 已提交
2150
  nodesRewriteExprsPostOrder(pList, doTranslateExpr, pCxt);
wmmhello's avatar
wmmhello 已提交
2151
  return pCxt->errCode;
2152 2153
}

2154
static SNodeList* getGroupByList(STranslateContext* pCxt) {
wmmhello's avatar
wmmhello 已提交
2155
  if (isDistinctOrderBy(pCxt)) {
2156
    return ((SSelectStmt*)pCxt->pCurrStmt)->pProjectionList;
wmmhello's avatar
wmmhello 已提交
2157
  }
2158
  return ((SSelectStmt*)pCxt->pCurrStmt)->pGroupByList;
2159 2160 2161
}

static SNode* getGroupByNode(SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
2162 2163 2164 2165
  if (QUERY_NODE_GROUPING_SET == nodeType(pNode)) {
    return nodesListGetNode(((SGroupingSetNode*)pNode)->pParameterList, 0);
  }
  return pNode;
2166 2167 2168
}

static int32_t getGroupByErrorCode(STranslateContext* pCxt) {
wmmhello's avatar
wmmhello 已提交
2169 2170 2171
  if (isDistinctOrderBy(pCxt)) {
    return TSDB_CODE_PAR_NOT_SELECTED_EXPRESSION;
  }
2172 2173 2174
  if (isSelectStmt(pCxt->pCurrStmt) && NULL != ((SSelectStmt*)pCxt->pCurrStmt)->pGroupByList) {
    return TSDB_CODE_PAR_GROUPBY_LACK_EXPRESSION;
  }
D
dapan1121 已提交
2175
  return TSDB_CODE_PAR_INVALID_OPTR_USAGE;
2176 2177
}

2178
static EDealRes rewriteColToSelectValFunc(STranslateContext* pCxt, SNode** pNode) {
2179
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
wmmhello's avatar
wmmhello 已提交
2180 2181 2182 2183 2184 2185
  if (NULL == pFunc) {
    pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
    return DEAL_RES_ERROR;
  }
  strcpy(pFunc->functionName, "_select_value");
  strcpy(pFunc->node.aliasName, ((SExprNode*)*pNode)->aliasName);
X
Xiaoyu Wang 已提交
2186
  strcpy(pFunc->node.userAlias, ((SExprNode*)*pNode)->userAlias);
wmmhello's avatar
wmmhello 已提交
2187 2188
  pCxt->errCode = nodesListMakeAppend(&pFunc->pParameterList, *pNode);
  if (TSDB_CODE_SUCCESS == pCxt->errCode) {
2189
    pCxt->errCode = getFuncInfo(pCxt, pFunc);
wmmhello's avatar
wmmhello 已提交
2190 2191 2192
  }
  if (TSDB_CODE_SUCCESS == pCxt->errCode) {
    *pNode = (SNode*)pFunc;
2193
    ((SSelectStmt*)pCxt->pCurrStmt)->hasSelectValFunc = true;
wmmhello's avatar
wmmhello 已提交
2194
  } else {
2195
    nodesDestroyNode((SNode*)pFunc);
wmmhello's avatar
wmmhello 已提交
2196 2197
  }
  return TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR;
2198 2199
}

2200 2201 2202 2203 2204 2205 2206 2207 2208
static EDealRes rewriteExprToGroupKeyFunc(STranslateContext* pCxt, SNode** pNode) {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
    return DEAL_RES_ERROR;
  }

  strcpy(pFunc->functionName, "_group_key");
  strcpy(pFunc->node.aliasName, ((SExprNode*)*pNode)->aliasName);
2209
  strcpy(pFunc->node.userAlias, ((SExprNode*)*pNode)->userAlias);
2210 2211 2212 2213 2214 2215 2216 2217 2218
  pCxt->errCode = nodesListMakeAppend(&pFunc->pParameterList, *pNode);
  if (TSDB_CODE_SUCCESS == pCxt->errCode) {
    *pNode = (SNode*)pFunc;
    pCxt->errCode = fmGetFuncInfo(pFunc, pCxt->msgBuf.buf, pCxt->msgBuf.len);
  }

  return (TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR);
}

2219
static EDealRes doCheckExprForGroupBy(SNode** pNode, void* pContext) {
2220 2221
  STranslateContext* pCxt = (STranslateContext*)pContext;
  SSelectStmt*       pSelect = (SSelectStmt*)pCxt->pCurrStmt;
wmmhello's avatar
wmmhello 已提交
2222 2223 2224
  if (!nodesIsExprNode(*pNode) || isAliasColumn(*pNode)) {
    return DEAL_RES_CONTINUE;
  }
2225
  if (isVectorFunc(*pNode) && !isDistinctOrderBy(pCxt)) {
wmmhello's avatar
wmmhello 已提交
2226 2227
    return DEAL_RES_IGNORE_CHILD;
  }
2228
  SNode* pGroupNode = NULL;
2229
  FOREACH(pGroupNode, getGroupByList(pCxt)) {
wmmhello's avatar
wmmhello 已提交
2230
    if (nodesEqualNode(getGroupByNode(pGroupNode), *pNode)) {
2231 2232 2233 2234
      return DEAL_RES_IGNORE_CHILD;
    }
  }
  SNode* pPartKey = NULL;
2235
  FOREACH(pPartKey, pSelect->pPartitionByList) {
2236
    if (nodesEqualNode(pPartKey, *pNode)) {
2237
      return rewriteExprToGroupKeyFunc(pCxt, pNode);
wmmhello's avatar
wmmhello 已提交
2238 2239
    }
  }
2240 2241 2242 2243 2244 2245
  if (NULL != pSelect->pWindow && QUERY_NODE_STATE_WINDOW == nodeType(pSelect->pWindow)) {
    if (nodesEqualNode(((SStateWindowNode*)pSelect->pWindow)->pExpr, *pNode)) {
      pSelect->hasStateKey = true;
      return rewriteExprToGroupKeyFunc(pCxt, pNode);
    }
  }
wmmhello's avatar
wmmhello 已提交
2246
  if (isScanPseudoColumnFunc(*pNode) || QUERY_NODE_COLUMN == nodeType(*pNode)) {
2247
    if (pSelect->selectFuncNum > 1 || pSelect->hasOtherVectorFunc || !pSelect->hasSelectFunc) {
D
dapan1121 已提交
2248
      return generateDealNodeErrMsg(pCxt, getGroupByErrorCode(pCxt), ((SExprNode*)(*pNode))->userAlias);
wmmhello's avatar
wmmhello 已提交
2249
    } else {
2250
      return rewriteColToSelectValFunc(pCxt, pNode);
wmmhello's avatar
wmmhello 已提交
2251 2252
    }
  }
2253
  if (isVectorFunc(*pNode) && isDistinctOrderBy(pCxt)) {
D
dapan1121 已提交
2254
    return generateDealNodeErrMsg(pCxt, getGroupByErrorCode(pCxt), ((SExprNode*)(*pNode))->userAlias);
wmmhello's avatar
wmmhello 已提交
2255 2256
  }
  return DEAL_RES_CONTINUE;
2257 2258
}

2259
static int32_t checkExprForGroupBy(STranslateContext* pCxt, SNode** pNode) {
2260
  nodesRewriteExpr(pNode, doCheckExprForGroupBy, pCxt);
wmmhello's avatar
wmmhello 已提交
2261
  return pCxt->errCode;
2262 2263
}

2264 2265
static int32_t checkExprListForGroupBy(STranslateContext* pCxt, SSelectStmt* pSelect, SNodeList* pList) {
  if (NULL == getGroupByList(pCxt) && NULL == pSelect->pWindow) {
wmmhello's avatar
wmmhello 已提交
2266 2267
    return TSDB_CODE_SUCCESS;
  }
2268
  nodesRewriteExprs(pList, doCheckExprForGroupBy, pCxt);
wmmhello's avatar
wmmhello 已提交
2269
  return pCxt->errCode;
2270 2271 2272
}

static EDealRes rewriteColsToSelectValFuncImpl(SNode** pNode, void* pContext) {
wmmhello's avatar
wmmhello 已提交
2273 2274 2275 2276 2277 2278 2279
  if (isAggFunc(*pNode) || isIndefiniteRowsFunc(*pNode)) {
    return DEAL_RES_IGNORE_CHILD;
  }
  if (isScanPseudoColumnFunc(*pNode) || QUERY_NODE_COLUMN == nodeType(*pNode)) {
    return rewriteColToSelectValFunc((STranslateContext*)pContext, pNode);
  }
  return DEAL_RES_CONTINUE;
2280 2281 2282
}

static int32_t rewriteColsToSelectValFunc(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
2283 2284 2285 2286 2287
  nodesRewriteExprs(pSelect->pProjectionList, rewriteColsToSelectValFuncImpl, pCxt);
  if (TSDB_CODE_SUCCESS == pCxt->errCode && !pSelect->isDistinct) {
    nodesRewriteExprs(pSelect->pOrderByList, rewriteColsToSelectValFuncImpl, pCxt);
  }
  return pCxt->errCode;
2288 2289
}

2290
typedef struct CheckAggColCoexistCxt {
wmmhello's avatar
wmmhello 已提交
2291 2292
  STranslateContext* pTranslateCxt;
  bool               existCol;
2293 2294
} CheckAggColCoexistCxt;

2295
static EDealRes doCheckAggColCoexist(SNode** pNode, void* pContext) {
wmmhello's avatar
wmmhello 已提交
2296
  CheckAggColCoexistCxt* pCxt = (CheckAggColCoexistCxt*)pContext;
2297
  if (isVectorFunc(*pNode)) {
wmmhello's avatar
wmmhello 已提交
2298 2299
    return DEAL_RES_IGNORE_CHILD;
  }
2300
  SNode* pPartKey = NULL;
2301
  FOREACH(pPartKey, ((SSelectStmt*)pCxt->pTranslateCxt->pCurrStmt)->pPartitionByList) {
2302 2303
    if (nodesEqualNode(pPartKey, *pNode)) {
      return rewriteExprToGroupKeyFunc(pCxt->pTranslateCxt, pNode);
2304 2305
    }
  }
2306
  if (isScanPseudoColumnFunc(*pNode) || QUERY_NODE_COLUMN == nodeType(*pNode)) {
wmmhello's avatar
wmmhello 已提交
2307 2308 2309
    pCxt->existCol = true;
  }
  return DEAL_RES_CONTINUE;
2310 2311 2312
}

static int32_t checkAggColCoexist(STranslateContext* pCxt, SSelectStmt* pSelect) {
2313
  if (NULL != pSelect->pGroupByList || NULL != pSelect->pWindow ||
2314
      (!pSelect->hasAggFuncs && !pSelect->hasIndefiniteRowsFunc && !pSelect->hasInterpFunc)) {
wmmhello's avatar
wmmhello 已提交
2315 2316
    return TSDB_CODE_SUCCESS;
  }
2317
  if (!pSelect->onlyHasKeepOrderFunc) {
2318
    pSelect->timeLineResMode = TIME_LINE_NONE;
2319
  }
2320
  CheckAggColCoexistCxt cxt = {.pTranslateCxt = pCxt, .existCol = false};
2321
  nodesRewriteExprs(pSelect->pProjectionList, doCheckAggColCoexist, &cxt);
wmmhello's avatar
wmmhello 已提交
2322
  if (!pSelect->isDistinct) {
2323
    nodesRewriteExprs(pSelect->pOrderByList, doCheckAggColCoexist, &cxt);
wmmhello's avatar
wmmhello 已提交
2324
  }
2325
  if (1 == pSelect->selectFuncNum && !pSelect->hasOtherVectorFunc) {
wmmhello's avatar
wmmhello 已提交
2326 2327
    return rewriteColsToSelectValFunc(pCxt, pSelect);
  }
2328
  if (cxt.existCol) {
wmmhello's avatar
wmmhello 已提交
2329 2330 2331
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_SINGLE_GROUP);
  }
  return TSDB_CODE_SUCCESS;
2332 2333
}

D
dapan1121 已提交
2334
static int32_t checkHavingGroupBy(STranslateContext* pCxt, SSelectStmt* pSelect) {
D
dapan1121 已提交
2335
  int32_t code = TSDB_CODE_SUCCESS;
D
dapan1121 已提交
2336
  if (NULL == getGroupByList(pCxt) && NULL == pSelect->pPartitionByList && NULL == pSelect->pWindow) {
D
dapan1121 已提交
2337 2338 2339 2340 2341
    return code;
  }  
  if (NULL != pSelect->pHaving) {
    code = checkExprForGroupBy(pCxt, &pSelect->pHaving);
  }
D
dapan1121 已提交
2342
/*
D
dapan1121 已提交
2343 2344 2345 2346 2347 2348
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pProjectionList) {
    code = checkExprListForGroupBy(pCxt, pSelect, pSelect->pProjectionList);
  }
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pOrderByList) {
    code = checkExprListForGroupBy(pCxt, pSelect, pSelect->pOrderByList);
  }
D
dapan1121 已提交
2349
*/  
D
dapan1121 已提交
2350 2351 2352
  return code;
}

2353 2354 2355 2356
static int32_t checkWindowFuncCoexist(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (NULL == pSelect->pWindow) {
    return TSDB_CODE_SUCCESS;
  }
2357
  if (NULL != pSelect->pWindow && !pSelect->hasAggFuncs && !pSelect->hasStateKey) {
2358 2359 2360 2361 2362
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_NO_VALID_FUNC_IN_WIN);
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2363
static int32_t toVgroupsInfo(SArray* pVgs, SVgroupsInfo** pVgsInfo) {
wmmhello's avatar
wmmhello 已提交
2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374
  size_t vgroupNum = taosArrayGetSize(pVgs);
  *pVgsInfo = taosMemoryCalloc(1, sizeof(SVgroupsInfo) + sizeof(SVgroupInfo) * vgroupNum);
  if (NULL == *pVgsInfo) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  (*pVgsInfo)->numOfVgroups = vgroupNum;
  for (int32_t i = 0; i < vgroupNum; ++i) {
    SVgroupInfo* vg = taosArrayGet(pVgs, i);
    (*pVgsInfo)->vgroups[i] = *vg;
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
2375 2376
}

2377
static int32_t addMnodeToVgroupList(const SEpSet* pEpSet, SArray** pVgroupList) {
wmmhello's avatar
wmmhello 已提交
2378 2379 2380 2381 2382 2383 2384 2385 2386 2387
  if (NULL == *pVgroupList) {
    *pVgroupList = taosArrayInit(TARRAY_MIN_SIZE, sizeof(SVgroupInfo));
    if (NULL == *pVgroupList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  SVgroupInfo vg = {.vgId = MNODE_HANDLE};
  memcpy(&vg.epSet, pEpSet, sizeof(SEpSet));
  taosArrayPush(*pVgroupList, &vg);
  return TSDB_CODE_SUCCESS;
2388 2389
}

2390 2391 2392 2393 2394
static int32_t dnodeToVgroupsInfo(SArray* pDnodes, SVgroupsInfo** pVgsInfo) {
  size_t ndnode = taosArrayGetSize(pDnodes);
  *pVgsInfo = taosMemoryCalloc(1, sizeof(SVgroupsInfo) + sizeof(SVgroupInfo) * ndnode);
  if (NULL == *pVgsInfo) {
    return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
2395
  }
2396 2397 2398 2399 2400 2401 2402 2403
  (*pVgsInfo)->numOfVgroups = ndnode;
  for (int32_t i = 0; i < ndnode; ++i) {
    memcpy(&((*pVgsInfo)->vgroups[i].epSet), taosArrayGet(pDnodes, i), sizeof(SEpSet));
  }
  return TSDB_CODE_SUCCESS;
}

static bool sysTableFromVnode(const char* pTable) {
2404 2405
  return ((0 == strcmp(pTable, TSDB_INS_TABLE_TABLES)) || (0 == strcmp(pTable, TSDB_INS_TABLE_TAGS)) ||
          (0 == strcmp(pTable, TSDB_INS_TABLE_COLS)));
2406 2407 2408
}

static bool sysTableFromDnode(const char* pTable) { return 0 == strcmp(pTable, TSDB_INS_TABLE_DNODE_VARIABLES); }
wmmhello's avatar
wmmhello 已提交
2409

X
Xiaoyu Wang 已提交
2410 2411
static int32_t getVnodeSysTableVgroupListImpl(STranslateContext* pCxt, SName* pTargetName, SName* pName,
                                              SArray** pVgroupList) {
2412
  if (0 == pTargetName->type) {
2413 2414 2415
    return getDBVgInfoImpl(pCxt, pName, pVgroupList);
  }

X
Xiaoyu Wang 已提交
2416 2417 2418 2419 2420 2421
  if (0 == strcmp(pTargetName->dbname, TSDB_INFORMATION_SCHEMA_DB) ||
      0 == strcmp(pTargetName->dbname, TSDB_PERFORMANCE_SCHEMA_DB)) {
    pTargetName->type = 0;
    return TSDB_CODE_SUCCESS;
  }

2422
  if (TSDB_DB_NAME_T == pTargetName->type) {
2423
    int32_t code = getDBVgInfoImpl(pCxt, pTargetName, pVgroupList);
X
Xiaoyu Wang 已提交
2424 2425
    if (!pCxt->showRewrite && (TSDB_CODE_MND_DB_NOT_EXIST == code || TSDB_CODE_MND_DB_IN_CREATING == code ||
                               TSDB_CODE_MND_DB_IN_DROPPING == code)) {
X
Xiaoyu Wang 已提交
2426
      // system table query should not report errors
2427 2428 2429
      code = TSDB_CODE_SUCCESS;
    }
    return code;
2430 2431 2432 2433 2434 2435 2436 2437
  }

  SVgroupInfo vgInfo = {0};
  int32_t     code = getTableHashVgroupImpl(pCxt, pTargetName, &vgInfo);
  if (TSDB_CODE_SUCCESS == code) {
    *pVgroupList = taosArrayInit(1, sizeof(SVgroupInfo));
    if (NULL == *pVgroupList) {
      code = TSDB_CODE_OUT_OF_MEMORY;
2438 2439
    } else {
      taosArrayPush(*pVgroupList, &vgInfo);
2440
    }
2441 2442
  } else if (TSDB_CODE_MND_DB_NOT_EXIST == code || TSDB_CODE_MND_DB_IN_CREATING == code ||
             TSDB_CODE_MND_DB_IN_DROPPING == code) {
X
Xiaoyu Wang 已提交
2443
    // system table query should not report errors
2444
    code = TSDB_CODE_SUCCESS;
2445 2446 2447 2448
  }
  return code;
}

X
Xiaoyu Wang 已提交
2449
static int32_t getVnodeSysTableVgroupList(STranslateContext* pCxt, SName* pName, SArray** pVgs, bool* pHasUserDbCond) {
2450 2451 2452 2453 2454
  if (!isSelectStmt(pCxt->pCurrStmt)) {
    return TSDB_CODE_SUCCESS;
  }
  SSelectStmt* pSelect = (SSelectStmt*)pCxt->pCurrStmt;
  SName        targetName = {0};
X
Xiaoyu Wang 已提交
2455
  int32_t      code = getVnodeSysTableTargetName(pCxt->pParseCxt->acctId, pSelect->pWhere, &targetName);
2456
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
2457
    code = getVnodeSysTableVgroupListImpl(pCxt, &targetName, pName, pVgs);
2458
  }
X
Xiaoyu Wang 已提交
2459
  *pHasUserDbCond = (0 != targetName.type && taosArrayGetSize(*pVgs) > 0);
2460 2461 2462
  return code;
}

2463
static int32_t setVnodeSysTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
X
Xiaoyu Wang 已提交
2464 2465 2466
  bool    hasUserDbCond = false;
  SArray* pVgs = NULL;
  int32_t code = getVnodeSysTableVgroupList(pCxt, pName, &pVgs, &hasUserDbCond);
wmmhello's avatar
wmmhello 已提交
2467

2468
  if (TSDB_CODE_SUCCESS == code && 0 == strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_TAGS) &&
X
Xiaoyu Wang 已提交
2469
      isSelectStmt(pCxt->pCurrStmt) && 0 == taosArrayGetSize(pVgs)) {
2470 2471 2472
    ((SSelectStmt*)pCxt->pCurrStmt)->isEmptyResult = true;
  }

2473 2474 2475
  if (TSDB_CODE_SUCCESS == code &&
          (0 == strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_TABLES) && !hasUserDbCond) ||
      0 == strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_COLS)) {
X
Xiaoyu Wang 已提交
2476
    code = addMnodeToVgroupList(&pCxt->pParseCxt->mgmtEpSet, &pVgs);
wmmhello's avatar
wmmhello 已提交
2477 2478 2479
  }

  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
2480
    code = toVgroupsInfo(pVgs, &pRealTable->pVgroupList);
wmmhello's avatar
wmmhello 已提交
2481
  }
X
Xiaoyu Wang 已提交
2482
  taosArrayDestroy(pVgs);
wmmhello's avatar
wmmhello 已提交
2483 2484

  return code;
X
Xiaoyu Wang 已提交
2485 2486
}

2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506
static int32_t setDnodeSysTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
  SArray* pDnodes = NULL;
  int32_t code = getDnodeList(pCxt, &pDnodes);
  if (TSDB_CODE_SUCCESS == code) {
    code = dnodeToVgroupsInfo(pDnodes, &pRealTable->pVgroupList);
  }
  taosArrayDestroy(pDnodes);
  return code;
}

static int32_t setSysTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
  if (sysTableFromVnode(pRealTable->table.tableName)) {
    return setVnodeSysTableVgroupList(pCxt, pName, pRealTable);
  } else if (sysTableFromDnode(pRealTable->table.tableName)) {
    return setDnodeSysTableVgroupList(pCxt, pName, pRealTable);
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

X
Xiaoyu Wang 已提交
2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525
static int32_t setSuperTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
  SArray* vgroupList = NULL;
  int32_t code = getDBVgInfoImpl(pCxt, pName, &vgroupList);
  if (TSDB_CODE_SUCCESS == code) {
    code = toVgroupsInfo(vgroupList, &pRealTable->pVgroupList);
  }
  taosArrayDestroy(vgroupList);
  return code;
}

static int32_t setNormalTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
  pRealTable->pVgroupList = taosMemoryCalloc(1, sizeof(SVgroupsInfo) + sizeof(SVgroupInfo));
  if (NULL == pRealTable->pVgroupList) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pRealTable->pVgroupList->numOfVgroups = 1;
  return getTableHashVgroupImpl(pCxt, pName, pRealTable->pVgroupList->vgroups);
}

X
Xiaoyu Wang 已提交
2526
static int32_t setTableVgroupList(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
wmmhello's avatar
wmmhello 已提交
2527 2528 2529 2530 2531
  if (pCxt->pParseCxt->topicQuery) {
    return TSDB_CODE_SUCCESS;
  }

  if (TSDB_SUPER_TABLE == pRealTable->pMeta->tableType) {
X
Xiaoyu Wang 已提交
2532
    return setSuperTableVgroupList(pCxt, pName, pRealTable);
wmmhello's avatar
wmmhello 已提交
2533
  }
X
Xiaoyu Wang 已提交
2534 2535 2536 2537 2538 2539

  if (TSDB_SYSTEM_TABLE == pRealTable->pMeta->tableType) {
    return setSysTableVgroupList(pCxt, pName, pRealTable);
  }

  return setNormalTableVgroupList(pCxt, pName, pRealTable);
2540 2541
}

X
Xiaoyu Wang 已提交
2542
static uint8_t getStmtPrecision(SNode* pStmt) {
wmmhello's avatar
wmmhello 已提交
2543 2544
  if (QUERY_NODE_SELECT_STMT == nodeType(pStmt)) {
    return ((SSelectStmt*)pStmt)->precision;
X
Xiaoyu Wang 已提交
2545 2546
  } else if (QUERY_NODE_SET_OPERATOR == nodeType(pStmt)) {
    return ((SSetOperator*)pStmt)->precision;
wmmhello's avatar
wmmhello 已提交
2547 2548
  }
  return 0;
X
Xiaoyu Wang 已提交
2549 2550
}

X
Xiaoyu Wang 已提交
2551 2552
static bool stmtIsSingleTable(SNode* pStmt) {
  if (QUERY_NODE_SELECT_STMT == nodeType(pStmt)) {
2553 2554
    SSelectStmt* pSelect = (SSelectStmt*)pStmt;
    return NULL == pSelect->pFromTable || ((STableNode*)pSelect->pFromTable)->singleTable;
X
Xiaoyu Wang 已提交
2555 2556 2557 2558
  }
  return false;
}

X
Xiaoyu Wang 已提交
2559 2560 2561 2562
static uint8_t calcPrecision(uint8_t lp, uint8_t rp) { return (lp > rp ? rp : lp); }

static uint8_t calcJoinTablePrecision(SJoinTableNode* pJoinTable) {
  return calcPrecision(((STableNode*)pJoinTable->pLeft)->precision, ((STableNode*)pJoinTable->pRight)->precision);
X
Xiaoyu Wang 已提交
2563 2564
}

X
Xiaoyu Wang 已提交
2565 2566 2567 2568 2569 2570 2571
static bool joinTableIsSingleTable(SJoinTableNode* pJoinTable) {
  return (((STableNode*)pJoinTable->pLeft)->singleTable && ((STableNode*)pJoinTable->pRight)->singleTable);
}

static bool isSingleTable(SRealTableNode* pRealTable) {
  int8_t tableType = pRealTable->pMeta->tableType;
  if (TSDB_SYSTEM_TABLE == tableType) {
D
dapan1121 已提交
2572
    return 0 != strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_TABLES) &&
2573 2574
           0 != strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_TAGS) &&
           0 != strcmp(pRealTable->table.tableName, TSDB_INS_TABLE_COLS);
X
Xiaoyu Wang 已提交
2575 2576 2577 2578
  }
  return (TSDB_CHILD_TABLE == tableType || TSDB_NORMAL_TABLE == tableType);
}

X
Xiaoyu Wang 已提交
2579
static int32_t setTableIndex(STranslateContext* pCxt, SName* pName, SRealTableNode* pRealTable) {
X
Xiaoyu Wang 已提交
2580
  if (pCxt->createStream || QUERY_SMA_OPTIMIZE_DISABLE == tsQuerySmaOptimize) {
X
Xiaoyu Wang 已提交
2581 2582
    return TSDB_CODE_SUCCESS;
  }
2583 2584
  if (isSelectStmt(pCxt->pCurrStmt) && NULL != ((SSelectStmt*)pCxt->pCurrStmt)->pWindow &&
      QUERY_NODE_INTERVAL_WINDOW == nodeType(((SSelectStmt*)pCxt->pCurrStmt)->pWindow)) {
X
Xiaoyu Wang 已提交
2585 2586 2587 2588 2589
    return getTableIndex(pCxt, pName, &pRealTable->pSmaIndexes);
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2590
static int32_t setTableCacheLastMode(STranslateContext* pCxt, SSelectStmt* pSelect) {
2591 2592
  if ((!pSelect->hasLastRowFunc && !pSelect->hasLastFunc) || QUERY_NODE_REAL_TABLE != nodeType(pSelect->pFromTable) ||
      TSDB_SYSTEM_TABLE == ((SRealTableNode*)pSelect->pFromTable)->pMeta->tableType) {
2593 2594 2595
    return TSDB_CODE_SUCCESS;
  }

X
Xiaoyu Wang 已提交
2596 2597 2598
  SRealTableNode* pTable = (SRealTableNode*)pSelect->pFromTable;
  SDbCfgInfo      dbCfg = {0};
  int32_t         code = getDBCfg(pCxt, pTable->table.dbName, &dbCfg);
2599
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
2600
    pTable->cacheLastMode = dbCfg.cacheLast;
2601 2602 2603 2604
  }
  return code;
}

D
dapan1121 已提交
2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662
static EDealRes doTranslateTbName(SNode** pNode, void* pContext) {
  switch (nodeType(*pNode)) {
    case QUERY_NODE_FUNCTION: {
      SFunctionNode *pFunc = (SFunctionNode *)*pNode;
      if (FUNCTION_TYPE_TBNAME == pFunc->funcType) {
        SRewriteTbNameContext *pCxt = (SRewriteTbNameContext*)pContext;
        SValueNode* pVal = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
        if (NULL == pVal) {
          pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
          return DEAL_RES_ERROR;
        }

        int32_t tbLen = strlen(pCxt->pTbName);
        pVal->literal = taosStrdup(pCxt->pTbName);
        if (NULL == pVal->literal) {
          pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
          return DEAL_RES_ERROR;
        }
        pVal->isDuration = false;
        pVal->translate = true;
        pVal->node.resType.type = TSDB_DATA_TYPE_BINARY;
        pVal->node.resType.bytes = tbLen + VARSTR_HEADER_SIZE;
        pVal->datum.p = taosMemoryCalloc(1, tbLen + VARSTR_HEADER_SIZE + 1);
        varDataSetLen(pVal->datum.p, tbLen);
        strncpy(varDataVal(pVal->datum.p), pVal->literal, tbLen);
        strcpy(pVal->node.userAlias, pFunc->node.userAlias);
        strcpy(pVal->node.aliasName, pFunc->node.aliasName);

        nodesDestroyNode(*pNode);
        *pNode = (SNode*)pVal;
      }
      break;
    }
    default:
      break;
  }
  return DEAL_RES_CONTINUE;
}

static int32_t replaceTbName(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (QUERY_NODE_REAL_TABLE != nodeType(pSelect->pFromTable)) {
    return TSDB_CODE_SUCCESS;
  }

  SRealTableNode* pTable = (SRealTableNode*)pSelect->pFromTable;
  if (TSDB_CHILD_TABLE != pTable->pMeta->tableType && TSDB_NORMAL_TABLE != pTable->pMeta->tableType && TSDB_SYSTEM_TABLE != pTable->pMeta->tableType) {
    return TSDB_CODE_SUCCESS;
  }

  SNode** pNode = NULL;
  SRewriteTbNameContext pRewriteCxt = {0};
  pRewriteCxt.pTbName = pTable->table.tableName;
  
  nodesRewriteExprPostOrder(&pSelect->pWhere, doTranslateTbName, &pRewriteCxt);

  return pRewriteCxt.errCode;
}

X
Xiaoyu Wang 已提交
2663 2664
static int32_t checkJoinTable(STranslateContext* pCxt, SJoinTableNode* pJoinTable) {
  if ((QUERY_NODE_TEMP_TABLE == nodeType(pJoinTable->pLeft) &&
2665
       !isGlobalTimeLineQuery(((STempTableNode*)pJoinTable->pLeft)->pSubquery)) ||
X
Xiaoyu Wang 已提交
2666
      (QUERY_NODE_TEMP_TABLE == nodeType(pJoinTable->pRight) &&
2667
       !isGlobalTimeLineQuery(((STempTableNode*)pJoinTable->pRight)->pSubquery))) {
X
Xiaoyu Wang 已提交
2668 2669 2670 2671 2672 2673
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_SUPPORT_JOIN,
                                   "Join requires valid time series input");
  }
  return TSDB_CODE_SUCCESS;
}

2674
static int32_t translateTable(STranslateContext* pCxt, SNode* pTable) {
wmmhello's avatar
wmmhello 已提交
2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
  int32_t code = TSDB_CODE_SUCCESS;
  switch (nodeType(pTable)) {
    case QUERY_NODE_REAL_TABLE: {
      SRealTableNode* pRealTable = (SRealTableNode*)pTable;
      pRealTable->ratio = (NULL != pCxt->pExplainOpt ? pCxt->pExplainOpt->ratio : 1.0);
      // The SRealTableNode created through ROLLUP already has STableMeta.
      if (NULL == pRealTable->pMeta) {
        SName name;
        code = getTableMetaImpl(
            pCxt, toName(pCxt->pParseCxt->acctId, pRealTable->table.dbName, pRealTable->table.tableName, &name),
            &(pRealTable->pMeta));
        if (TSDB_CODE_SUCCESS != code) {
2687
          return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_GET_META_ERROR, tstrerror(code));
wmmhello's avatar
wmmhello 已提交
2688 2689
        }
        code = setTableVgroupList(pCxt, &name, pRealTable);
X
Xiaoyu Wang 已提交
2690
        if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
2691
          code = setTableIndex(pCxt, &name, pRealTable);
X
Xiaoyu Wang 已提交
2692
        }
wmmhello's avatar
wmmhello 已提交
2693 2694
      }
      if (TSDB_CODE_SUCCESS == code) {
2695 2696 2697 2698 2699
        pRealTable->table.precision = pRealTable->pMeta->tableInfo.precision;
        pRealTable->table.singleTable = isSingleTable(pRealTable);
        if (TSDB_SUPER_TABLE == pRealTable->pMeta->tableType) {
          pCxt->stableQuery = true;
        }
G
Ganlin Zhao 已提交
2700 2701
        if (TSDB_SYSTEM_TABLE == pRealTable->pMeta->tableType) {
          if (isSelectStmt(pCxt->pCurrStmt)) {
2702
            ((SSelectStmt*)pCxt->pCurrStmt)->timeLineResMode = TIME_LINE_NONE;
G
Ganlin Zhao 已提交
2703 2704 2705 2706
          } else if (isDeleteStmt(pCxt->pCurrStmt)) {
            code = TSDB_CODE_TSC_INVALID_OPERATION;
            break;
          }
X
Xiaoyu Wang 已提交
2707
        }
wmmhello's avatar
wmmhello 已提交
2708 2709 2710 2711 2712 2713 2714 2715
        code = addNamespace(pCxt, pRealTable);
      }
      break;
    }
    case QUERY_NODE_TEMP_TABLE: {
      STempTableNode* pTempTable = (STempTableNode*)pTable;
      code = translateSubquery(pCxt, pTempTable->pSubquery);
      if (TSDB_CODE_SUCCESS == code) {
2716 2717
        if (QUERY_NODE_SELECT_STMT == nodeType(pTempTable->pSubquery) &&
            ((SSelectStmt*)pTempTable->pSubquery)->isEmptyResult && isSelectStmt(pCxt->pCurrStmt)) {
D
dapan1121 已提交
2718 2719 2720
          ((SSelectStmt*)pCxt->pCurrStmt)->isEmptyResult = true;
        }

wmmhello's avatar
wmmhello 已提交
2721
        pTempTable->table.precision = getStmtPrecision(pTempTable->pSubquery);
X
Xiaoyu Wang 已提交
2722
        pTempTable->table.singleTable = stmtIsSingleTable(pTempTable->pSubquery);
wmmhello's avatar
wmmhello 已提交
2723 2724 2725 2726 2727 2728 2729 2730 2731 2732
        code = addNamespace(pCxt, pTempTable);
      }
      break;
    }
    case QUERY_NODE_JOIN_TABLE: {
      SJoinTableNode* pJoinTable = (SJoinTableNode*)pTable;
      code = translateTable(pCxt, pJoinTable->pLeft);
      if (TSDB_CODE_SUCCESS == code) {
        code = translateTable(pCxt, pJoinTable->pRight);
      }
X
Xiaoyu Wang 已提交
2733 2734 2735
      if (TSDB_CODE_SUCCESS == code) {
        code = checkJoinTable(pCxt, pJoinTable);
      }
wmmhello's avatar
wmmhello 已提交
2736
      if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
2737
        pJoinTable->table.precision = calcJoinTablePrecision(pJoinTable);
X
Xiaoyu Wang 已提交
2738 2739
        pJoinTable->table.singleTable = joinTableIsSingleTable(pJoinTable);
        code = translateExpr(pCxt, &pJoinTable->pOnCond);
wmmhello's avatar
wmmhello 已提交
2740 2741 2742 2743 2744 2745 2746
      }
      break;
    }
    default:
      break;
  }
  return code;
2747 2748
}

2749
static int32_t createAllColumns(STranslateContext* pCxt, bool igTags, SNodeList** pCols) {
wmmhello's avatar
wmmhello 已提交
2750 2751 2752 2753 2754 2755 2756 2757
  *pCols = nodesMakeList();
  if (NULL == *pCols) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
  }
  SArray* pTables = taosArrayGetP(pCxt->pNsLevel, pCxt->currLevel);
  size_t  nums = taosArrayGetSize(pTables);
  for (size_t i = 0; i < nums; ++i) {
    STableNode* pTable = taosArrayGetP(pTables, i);
2758
    int32_t     code = createColumnsByTable(pCxt, pTable, igTags, *pCols);
wmmhello's avatar
wmmhello 已提交
2759 2760 2761 2762 2763
    if (TSDB_CODE_SUCCESS != code) {
      return code;
    }
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
2764 2765
}

2766
static SNode* createMultiResFunc(SFunctionNode* pSrcFunc, SExprNode* pExpr) {
2767
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
wmmhello's avatar
wmmhello 已提交
2768 2769 2770 2771 2772
  if (NULL == pFunc) {
    return NULL;
  }
  pFunc->pParameterList = nodesMakeList();
  if (NULL == pFunc->pParameterList ||
2773 2774
      TSDB_CODE_SUCCESS != nodesListStrictAppend(pFunc->pParameterList, nodesCloneNode((SNode*)pExpr))) {
    nodesDestroyNode((SNode*)pFunc);
wmmhello's avatar
wmmhello 已提交
2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
    return NULL;
  }

  pFunc->node.resType = pExpr->resType;
  pFunc->funcId = pSrcFunc->funcId;
  pFunc->funcType = pSrcFunc->funcType;
  strcpy(pFunc->functionName, pSrcFunc->functionName);
  char    buf[TSDB_FUNC_NAME_LEN + TSDB_TABLE_NAME_LEN + TSDB_COL_NAME_LEN];
  int32_t len = 0;
  if (QUERY_NODE_COLUMN == nodeType(pExpr)) {
    SColumnNode* pCol = (SColumnNode*)pExpr;
2786 2787
    if (tsKeepColumnName) {
      strcpy(pFunc->node.userAlias, pCol->colName);
2788
      strcpy(pFunc->node.aliasName, pCol->colName);
2789
    } else {
2790 2791
      len = snprintf(buf, sizeof(buf), "%s(%s.%s)", pSrcFunc->functionName, pCol->tableAlias, pCol->colName);
      strncpy(pFunc->node.aliasName, buf, TMIN(len, sizeof(pFunc->node.aliasName) - 1));
2792 2793 2794
      len = snprintf(buf, sizeof(buf), "%s(%s)", pSrcFunc->functionName, pCol->colName);
      strncpy(pFunc->node.userAlias, buf, TMIN(len, sizeof(pFunc->node.userAlias) - 1));
    }
wmmhello's avatar
wmmhello 已提交
2795 2796
  } else {
    len = snprintf(buf, sizeof(buf), "%s(%s)", pSrcFunc->functionName, pExpr->aliasName);
X
Xiaoyu Wang 已提交
2797
    strncpy(pFunc->node.aliasName, buf, TMIN(len, sizeof(pFunc->node.aliasName) - 1));
2798
    len = snprintf(buf, sizeof(buf), "%s(%s)", pSrcFunc->functionName, pExpr->userAlias);
X
Xiaoyu Wang 已提交
2799
    strncpy(pFunc->node.userAlias, buf, TMIN(len, sizeof(pFunc->node.userAlias) - 1));
wmmhello's avatar
wmmhello 已提交
2800 2801 2802
  }

  return (SNode*)pFunc;
2803 2804
}

2805
static int32_t createTableAllCols(STranslateContext* pCxt, SColumnNode* pCol, bool igTags, SNodeList** pOutput) {
wmmhello's avatar
wmmhello 已提交
2806 2807 2808 2809 2810 2811 2812 2813 2814
  STableNode* pTable = NULL;
  int32_t     code = findTable(pCxt, pCol->tableAlias, &pTable);
  if (TSDB_CODE_SUCCESS == code && NULL == *pOutput) {
    *pOutput = nodesMakeList();
    if (NULL == *pOutput) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
    }
  }
  if (TSDB_CODE_SUCCESS == code) {
2815
    code = createColumnsByTable(pCxt, pTable, igTags, *pOutput);
wmmhello's avatar
wmmhello 已提交
2816 2817
  }
  return code;
2818 2819
}

2820
static int32_t createMultiResFuncsParas(STranslateContext* pCxt, SNodeList* pSrcParas, SNodeList** pOutput) {
wmmhello's avatar
wmmhello 已提交
2821 2822 2823 2824 2825 2826
  int32_t code = TSDB_CODE_SUCCESS;

  SNodeList* pExprs = NULL;
  SNode*     pPara = NULL;
  FOREACH(pPara, pSrcParas) {
    if (isStar(pPara)) {
2827
      code = createAllColumns(pCxt, true, &pExprs);
wmmhello's avatar
wmmhello 已提交
2828
    } else if (isTableStar(pPara)) {
2829
      code = createTableAllCols(pCxt, (SColumnNode*)pPara, true, &pExprs);
wmmhello's avatar
wmmhello 已提交
2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844
    } else {
      code = nodesListMakeStrictAppend(&pExprs, nodesCloneNode(pPara));
    }
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pOutput = pExprs;
  } else {
    nodesDestroyList(pExprs);
  }

  return code;
2845 2846 2847
}

static int32_t createMultiResFuncs(SFunctionNode* pSrcFunc, SNodeList* pExprs, SNodeList** pOutput) {
wmmhello's avatar
wmmhello 已提交
2848 2849 2850 2851
  SNodeList* pFuncs = nodesMakeList();
  if (NULL == pFuncs) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
2852

wmmhello's avatar
wmmhello 已提交
2853 2854 2855 2856 2857 2858 2859 2860
  int32_t code = TSDB_CODE_SUCCESS;
  SNode*  pExpr = NULL;
  FOREACH(pExpr, pExprs) {
    code = nodesListStrictAppend(pFuncs, createMultiResFunc(pSrcFunc, (SExprNode*)pExpr));
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
  }
2861

wmmhello's avatar
wmmhello 已提交
2862 2863 2864 2865 2866
  if (TSDB_CODE_SUCCESS == code) {
    *pOutput = pFuncs;
  } else {
    nodesDestroyList(pFuncs);
  }
2867

wmmhello's avatar
wmmhello 已提交
2868
  return code;
2869 2870 2871
}

static int32_t createMultiResFuncsFromStar(STranslateContext* pCxt, SFunctionNode* pSrcFunc, SNodeList** pOutput) {
wmmhello's avatar
wmmhello 已提交
2872 2873 2874 2875 2876
  SNodeList* pExprs = NULL;
  int32_t    code = createMultiResFuncsParas(pCxt, pSrcFunc->pParameterList, &pExprs);
  if (TSDB_CODE_SUCCESS == code) {
    code = createMultiResFuncs(pSrcFunc, pExprs, pOutput);
  }
2877

X
Xiaoyu Wang 已提交
2878
  nodesDestroyList(pExprs);
wmmhello's avatar
wmmhello 已提交
2879
  return code;
2880 2881
}

2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909
static int32_t createTags(STranslateContext* pCxt, SNodeList** pOutput) {
  if (QUERY_NODE_REAL_TABLE != nodeType(((SSelectStmt*)pCxt->pCurrStmt)->pFromTable)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_PC,
                                   "The _TAGS pseudo column can only be used for subtable and supertable queries");
  }

  SRealTableNode*   pTable = (SRealTableNode*)(((SSelectStmt*)pCxt->pCurrStmt)->pFromTable);
  const STableMeta* pMeta = pTable->pMeta;
  if (TSDB_SUPER_TABLE != pMeta->tableType && TSDB_CHILD_TABLE != pMeta->tableType) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_PC,
                                "The _TAGS pseudo column can only be used for subtable and supertable queries");
  }

  SSchema* pTagsSchema = getTableTagSchema(pMeta);
  for (int32_t i = 0; i < pMeta->tableInfo.numOfTags; ++i) {
    SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
    if (NULL == pCol) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
    }
    setColumnInfoBySchema(pTable, pTagsSchema + i, 1, pCol);
    if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(pOutput, (SNode*)pCol)) {
      NODES_DESTORY_LIST(*pOutput);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2910
static int32_t translateStar(STranslateContext* pCxt, SSelectStmt* pSelect) {
X
Xiaoyu Wang 已提交
2911 2912 2913 2914 2915
  SNode* pNode = NULL;
  WHERE_EACH(pNode, pSelect->pProjectionList) {
    int32_t code = TSDB_CODE_SUCCESS;
    if (isStar(pNode)) {
      SNodeList* pCols = NULL;
2916
      code = createAllColumns(pCxt, false, &pCols);
X
Xiaoyu Wang 已提交
2917 2918 2919 2920
      if (TSDB_CODE_SUCCESS == code) {
        INSERT_LIST(pSelect->pProjectionList, pCols);
        ERASE_NODE(pSelect->pProjectionList);
        continue;
wmmhello's avatar
wmmhello 已提交
2921
      }
X
Xiaoyu Wang 已提交
2922
    } else if (isMultiResFunc(pNode)) {
2923 2924 2925 2926 2927 2928
      SNodeList* pNodeList = NULL;
      if (FUNCTION_TYPE_TAGS == ((SFunctionNode*)pNode)->funcType) {
        code = createTags(pCxt, &pNodeList);
      } else {
        code = createMultiResFuncsFromStar(pCxt, (SFunctionNode*)pNode, &pNodeList);
      }
X
Xiaoyu Wang 已提交
2929
      if (TSDB_CODE_SUCCESS == code) {
2930
        INSERT_LIST(pSelect->pProjectionList, pNodeList);
X
Xiaoyu Wang 已提交
2931 2932 2933 2934 2935
        ERASE_NODE(pSelect->pProjectionList);
        continue;
      }
    } else if (isTableStar(pNode)) {
      SNodeList* pCols = NULL;
2936
      code = createTableAllCols(pCxt, (SColumnNode*)pNode, false, &pCols);
X
Xiaoyu Wang 已提交
2937 2938 2939 2940
      if (TSDB_CODE_SUCCESS == code) {
        INSERT_LIST(pSelect->pProjectionList, pCols);
        ERASE_NODE(pSelect->pProjectionList);
        continue;
wmmhello's avatar
wmmhello 已提交
2941 2942
      }
    }
X
Xiaoyu Wang 已提交
2943 2944 2945 2946
    if (TSDB_CODE_SUCCESS != code) {
      return code;
    }
    WHERE_NEXT;
wmmhello's avatar
wmmhello 已提交
2947 2948
  }
  return TSDB_CODE_SUCCESS;
2949 2950
}

2951
static int32_t getPositionValue(const SValueNode* pVal) {
wmmhello's avatar
wmmhello 已提交
2952 2953 2954 2955 2956 2957 2958
  switch (pVal->node.resType.type) {
    case TSDB_DATA_TYPE_NULL:
    case TSDB_DATA_TYPE_TIMESTAMP:
    case TSDB_DATA_TYPE_NCHAR:
    case TSDB_DATA_TYPE_VARCHAR:
    case TSDB_DATA_TYPE_VARBINARY:
    case TSDB_DATA_TYPE_JSON:
D
Dingle Zhang 已提交
2959
    case TSDB_DATA_TYPE_GEOMETRY:
wmmhello's avatar
wmmhello 已提交
2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979
      return -1;
    case TSDB_DATA_TYPE_BOOL:
      return (pVal->datum.b ? 1 : 0);
    case TSDB_DATA_TYPE_TINYINT:
    case TSDB_DATA_TYPE_SMALLINT:
    case TSDB_DATA_TYPE_INT:
    case TSDB_DATA_TYPE_BIGINT:
      return pVal->datum.i;
    case TSDB_DATA_TYPE_FLOAT:
    case TSDB_DATA_TYPE_DOUBLE:
      return pVal->datum.d;
    case TSDB_DATA_TYPE_UTINYINT:
    case TSDB_DATA_TYPE_USMALLINT:
    case TSDB_DATA_TYPE_UINT:
    case TSDB_DATA_TYPE_UBIGINT:
      return pVal->datum.u;
    default:
      break;
  }
  return -1;
2980 2981
}

dengyihao's avatar
dengyihao 已提交
2982
static int32_t translateOrderByPosition(STranslateContext* pCxt, SNodeList* pProjectionList, SNodeList* pOrderByList,
wmmhello's avatar
wmmhello 已提交
2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999
                                        bool* pOther) {
  *pOther = false;
  SNode* pNode = NULL;
  WHERE_EACH(pNode, pOrderByList) {
    SNode* pExpr = ((SOrderByExprNode*)pNode)->pExpr;
    if (QUERY_NODE_VALUE == nodeType(pExpr)) {
      SValueNode* pVal = (SValueNode*)pExpr;
      if (DEAL_RES_ERROR == translateValue(pCxt, pVal)) {
        return pCxt->errCode;
      }
      int32_t pos = getPositionValue(pVal);
      if (pos < 0) {
        ERASE_NODE(pOrderByList);
        continue;
      } else if (0 == pos || pos > LIST_LENGTH(pProjectionList)) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_NUMBER_OF_SELECT);
      } else {
3000
        SColumnRefNode* pCol = (SColumnRefNode*)nodesMakeNode(QUERY_NODE_COLUMN_REF);
wmmhello's avatar
wmmhello 已提交
3001 3002 3003
        if (NULL == pCol) {
          return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_OUT_OF_MEMORY);
        }
3004
        strcpy(pCol->colName, ((SExprNode*)nodesListGetNode(pProjectionList, pos - 1))->aliasName);
wmmhello's avatar
wmmhello 已提交
3005 3006 3007 3008 3009 3010 3011 3012 3013
        ((SOrderByExprNode*)pNode)->pExpr = (SNode*)pCol;
        nodesDestroyNode(pExpr);
      }
    } else {
      *pOther = true;
    }
    WHERE_NEXT;
  }
  return TSDB_CODE_SUCCESS;
3014 3015
}

3016
static int32_t translateOrderBy(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3017 3018 3019
  bool    other;
  int32_t code = translateOrderByPosition(pCxt, pSelect->pProjectionList, pSelect->pOrderByList, &other);
  if (TSDB_CODE_SUCCESS == code) {
3020 3021 3022 3023
    if (0 == LIST_LENGTH(pSelect->pOrderByList)) {
      NODES_DESTORY_LIST(pSelect->pOrderByList);
      return TSDB_CODE_SUCCESS;
    }
3024 3025 3026 3027 3028
    if (!other) {
      return TSDB_CODE_SUCCESS;
    }
    pCxt->currClause = SQL_CLAUSE_ORDER_BY;
    code = translateExprList(pCxt, pSelect->pOrderByList);
3029 3030 3031
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkExprListForGroupBy(pCxt, pSelect, pSelect->pOrderByList);
wmmhello's avatar
wmmhello 已提交
3032 3033
  }
  return code;
3034 3035
}

X
Xiaoyu Wang 已提交
3036
static EDealRes needFillImpl(SNode* pNode, void* pContext) {
3037
  if ((isAggFunc(pNode) || isInterpFunc(pNode)) && FUNCTION_TYPE_GROUP_KEY != ((SFunctionNode*)pNode)->funcType) {
X
Xiaoyu Wang 已提交
3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049
    *(bool*)pContext = true;
    return DEAL_RES_END;
  }
  return DEAL_RES_CONTINUE;
}

static bool needFill(SNode* pNode) {
  bool hasFillFunc = false;
  nodesWalkExpr(pNode, needFillImpl, &hasFillFunc);
  return hasFillFunc;
}

3050 3051 3052 3053
static int32_t convertFillValue(STranslateContext* pCxt, SDataType dt, SNodeList* pValues, int32_t index) {
  SListCell* pCell = nodesListGetCell(pValues, index);
  if (dataTypeEqual(&dt, &((SExprNode*)pCell->pNode)->resType)) {
    return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3054
  }
3055 3056 3057
  SNode*  pCaseFunc = NULL;
  int32_t code = createCastFunc(pCxt, pCell->pNode, dt, &pCaseFunc);
  if (TSDB_CODE_SUCCESS == code) {
3058
    code = scalarCalculateConstants(pCaseFunc, &pCell->pNode);
X
Xiaoyu Wang 已提交
3059
  }
3060
  if (TSDB_CODE_SUCCESS == code && QUERY_NODE_VALUE != nodeType(pCell->pNode)) {
G
Ganlin Zhao 已提交
3061
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, "Fill value can only accept constant");
3062 3063 3064
  } else if (TSDB_CODE_SUCCESS != code) {
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, "Filled data type mismatch");
  }
3065
  return code;
X
Xiaoyu Wang 已提交
3066 3067 3068
}

static int32_t checkFillValues(STranslateContext* pCxt, SFillNode* pFill, SNodeList* pProjectionList) {
D
dapan1121 已提交
3069
  if (FILL_MODE_VALUE != pFill->mode && FILL_MODE_VALUE_F != pFill->mode) {
X
Xiaoyu Wang 已提交
3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080
    return TSDB_CODE_SUCCESS;
  }

  int32_t        fillNo = 0;
  SNodeListNode* pFillValues = (SNodeListNode*)pFill->pValues;
  SNode*         pProject = NULL;
  FOREACH(pProject, pProjectionList) {
    if (needFill(pProject)) {
      if (fillNo >= LIST_LENGTH(pFillValues->pNodeList)) {
        return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, "Filled values number mismatch");
      }
3081 3082 3083
      int32_t code = convertFillValue(pCxt, ((SExprNode*)pProject)->resType, pFillValues->pNodeList, fillNo);
      if (TSDB_CODE_SUCCESS != code) {
        return code;
X
Xiaoyu Wang 已提交
3084
      }
G
Ganlin Zhao 已提交
3085

X
Xiaoyu Wang 已提交
3086 3087 3088 3089 3090 3091 3092 3093 3094
      ++fillNo;
    }
  }
  if (fillNo != LIST_LENGTH(pFillValues->pNodeList)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, "Filled values number mismatch");
  }
  return TSDB_CODE_SUCCESS;
}

3095 3096 3097 3098 3099
static int32_t translateFillValues(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (NULL == pSelect->pWindow || QUERY_NODE_INTERVAL_WINDOW != nodeType(pSelect->pWindow) ||
      NULL == ((SIntervalWindowNode*)pSelect->pWindow)->pFill) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
3100
  return checkFillValues(pCxt, (SFillNode*)((SIntervalWindowNode*)pSelect->pWindow)->pFill, pSelect->pProjectionList);
3101 3102
}

X
Xiaoyu Wang 已提交
3103 3104 3105
static int32_t rewriteProjectAlias(SNodeList* pProjectionList) {
  int32_t no = 1;
  SNode*  pProject = NULL;
X
Xiaoyu Wang 已提交
3106 3107 3108 3109 3110
  FOREACH(pProject, pProjectionList) {
    SExprNode* pExpr = (SExprNode*)pProject;
    if ('\0' == pExpr->userAlias[0]) {
      strcpy(pExpr->userAlias, pExpr->aliasName);
    }
K
kailixu 已提交
3111
    if (QUERY_NODE_FUNCTION == nodeType(pProject) && fmIsWindowPseudoColumnFunc(((SFunctionNode*)pProject)->funcId) &&
K
kailixu 已提交
3112 3113 3114
        ((0 == strcasecmp("_wstart", pExpr->userAlias) || 0 == strcasecmp("_wend", pExpr->userAlias) ||
          0 == strcasecmp("_wduration", pExpr->userAlias)) &&
         '\0' != pExpr->aliasName[0])) {
K
kailixu 已提交
3115 3116
        continue;
      }
X
Xiaoyu Wang 已提交
3117 3118
    sprintf(pExpr->aliasName, "#expr_%d", no++);
  }
X
Xiaoyu Wang 已提交
3119 3120 3121
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
3122
static int32_t checkProjectAlias(STranslateContext* pCxt, SNodeList* pProjectionList, SHashObj** pOutput) {
X
Xiaoyu Wang 已提交
3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133
  SHashObj* pUserAliasSet = taosHashInit(LIST_LENGTH(pProjectionList),
                                         taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  SNode*    pProject = NULL;
  FOREACH(pProject, pProjectionList) {
    SExprNode* pExpr = (SExprNode*)pProject;
    if (NULL != taosHashGet(pUserAliasSet, pExpr->userAlias, strlen(pExpr->userAlias))) {
      taosHashCleanup(pUserAliasSet);
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_AMBIGUOUS_COLUMN, pExpr->userAlias);
    }
    taosHashPut(pUserAliasSet, pExpr->userAlias, strlen(pExpr->userAlias), &pExpr, POINTER_BYTES);
  }
X
Xiaoyu Wang 已提交
3134 3135 3136 3137 3138
  if (NULL == pOutput) {
    taosHashCleanup(pUserAliasSet);
  } else {
    *pOutput = pUserAliasSet;
  }
X
Xiaoyu Wang 已提交
3139 3140 3141 3142 3143
  return TSDB_CODE_SUCCESS;
}

static int32_t translateProjectionList(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (pSelect->isSubquery) {
X
Xiaoyu Wang 已提交
3144
    return checkProjectAlias(pCxt, pSelect->pProjectionList, NULL);
X
Xiaoyu Wang 已提交
3145 3146 3147 3148
  }
  return rewriteProjectAlias(pSelect->pProjectionList);
}

3149
static int32_t translateSelectList(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3150 3151 3152 3153 3154
  pCxt->currClause = SQL_CLAUSE_SELECT;
  int32_t code = translateExprList(pCxt, pSelect->pProjectionList);
  if (TSDB_CODE_SUCCESS == code) {
    code = translateStar(pCxt, pSelect);
  }
X
Xiaoyu Wang 已提交
3155 3156 3157
  if (TSDB_CODE_SUCCESS == code) {
    code = translateProjectionList(pCxt, pSelect);
  }
wmmhello's avatar
wmmhello 已提交
3158
  if (TSDB_CODE_SUCCESS == code) {
3159
    code = checkExprListForGroupBy(pCxt, pSelect, pSelect->pProjectionList);
D
dapan1121 已提交
3160
  }  
3161 3162 3163
  if (TSDB_CODE_SUCCESS == code) {
    code = translateFillValues(pCxt, pSelect);
  }
3164 3165 3166
  if (NULL == pSelect->pProjectionList || 0 >= pSelect->pProjectionList->length) {
    code = TSDB_CODE_PAR_INVALID_SELECTED_EXPR;
  }
wmmhello's avatar
wmmhello 已提交
3167
  return code;
3168 3169
}

3170
static int32_t translateHaving(STranslateContext* pCxt, SSelectStmt* pSelect) {
3171 3172
  if (NULL == pSelect->pGroupByList && NULL == pSelect->pPartitionByList && NULL == pSelect->pWindow &&
      NULL != pSelect->pHaving) {
wmmhello's avatar
wmmhello 已提交
3173 3174 3175
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_GROUPBY_LACK_EXPRESSION);
  }
  pCxt->currClause = SQL_CLAUSE_HAVING;
X
Xiaoyu Wang 已提交
3176
  int32_t code = translateExpr(pCxt, &pSelect->pHaving);
wmmhello's avatar
wmmhello 已提交
3177
  return code;
3178 3179
}

X
Xiaoyu Wang 已提交
3180
static int32_t translateGroupBy(STranslateContext* pCxt, SSelectStmt* pSelect) {
3181 3182
  if (NULL == pSelect->pGroupByList) {
    return TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
3183
  }
3184 3185
  if (NULL != pSelect->pWindow) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_GROUPBY_WINDOW_COEXIST);
wmmhello's avatar
wmmhello 已提交
3186
  }
3187
  pCxt->currClause = SQL_CLAUSE_GROUP_BY;
3188
  pSelect->timeLineResMode = TIME_LINE_NONE;
3189
  return translateExprList(pCxt, pSelect->pGroupByList);
3190 3191
}

X
Xiaoyu Wang 已提交
3192 3193 3194 3195 3196 3197
static int32_t getTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* pIsStrict) {
  SNode*  pNew = NULL;
  int32_t code = scalarCalculateConstants(*pPrimaryKeyCond, &pNew);
  if (TSDB_CODE_SUCCESS == code) {
    *pPrimaryKeyCond = pNew;
    code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, pIsStrict);
wmmhello's avatar
wmmhello 已提交
3198 3199 3200
  }
  return code;
}
X
Xiaoyu Wang 已提交
3201

3202
static int32_t getQueryTimeRange(STranslateContext* pCxt, SNode* pWhere, STimeWindow* pTimeRange) {
wmmhello's avatar
wmmhello 已提交
3203 3204 3205 3206
  if (NULL == pWhere) {
    *pTimeRange = TSWINDOW_INITIALIZER;
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
3207

X
Xiaoyu Wang 已提交
3208 3209 3210
  SNode* pCond = nodesCloneNode(pWhere);
  if (NULL == pCond) {
    return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
3211
  }
X
Xiaoyu Wang 已提交
3212

X
Xiaoyu Wang 已提交
3213
  SNode* pPrimaryKeyCond = NULL;
X
Xiaoyu Wang 已提交
3214
  filterPartitionCond(&pCond, &pPrimaryKeyCond, NULL, NULL, NULL);
X
Xiaoyu Wang 已提交
3215 3216 3217

  int32_t code = TSDB_CODE_SUCCESS;
  if (NULL != pPrimaryKeyCond) {
wmmhello's avatar
wmmhello 已提交
3218
    bool isStrict = false;
X
Xiaoyu Wang 已提交
3219
    code = getTimeRange(&pPrimaryKeyCond, pTimeRange, &isStrict);
wmmhello's avatar
wmmhello 已提交
3220 3221 3222
  } else {
    *pTimeRange = TSWINDOW_INITIALIZER;
  }
X
Xiaoyu Wang 已提交
3223 3224 3225
  nodesDestroyNode(pCond);
  nodesDestroyNode(pPrimaryKeyCond);
  return code;
X
Xiaoyu Wang 已提交
3226 3227
}

3228
static int32_t checkFill(STranslateContext* pCxt, SFillNode* pFill, SValueNode* pInterval, bool isInterpFill) {
X
Xiaoyu Wang 已提交
3229
  if (FILL_MODE_NONE == pFill->mode) {
3230 3231 3232 3233
    if (isInterpFill) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, "Unsupported fill type");
    }

X
Xiaoyu Wang 已提交
3234 3235 3236
    return TSDB_CODE_SUCCESS;
  }

X
Xiaoyu Wang 已提交
3237
  if (!pCxt->createStream && TSWINDOW_IS_EQUAL(pFill->timeRange, TSWINDOW_INITIALIZER)) {
wmmhello's avatar
wmmhello 已提交
3238 3239 3240
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_FILL_TIME_RANGE);
  }

X
Xiaoyu Wang 已提交
3241 3242 3243 3244
  if (TSWINDOW_IS_EQUAL(pFill->timeRange, TSWINDOW_DESC_INITIALIZER)) {
    return TSDB_CODE_SUCCESS;
  }

X
Xiaoyu Wang 已提交
3245 3246 3247 3248 3249 3250 3251
  // interp FILL clause
  if (NULL == pInterval) {
    return TSDB_CODE_SUCCESS;
  }

  int64_t timeRange = TABS(pFill->timeRange.skey - pFill->timeRange.ekey);
  int64_t intervalRange = 0;
H
Haojun Liao 已提交
3252
  if (IS_CALENDAR_TIME_DURATION(pInterval->unit)) {
wmmhello's avatar
wmmhello 已提交
3253
    int64_t f = 1;
X
Xiaoyu Wang 已提交
3254
    if (pInterval->unit == 'n') {
wafwerar's avatar
wafwerar 已提交
3255
      f = 30LL * MILLISECOND_PER_DAY;
X
Xiaoyu Wang 已提交
3256
    } else if (pInterval->unit == 'y') {
wafwerar's avatar
wafwerar 已提交
3257
      f = 365LL * MILLISECOND_PER_DAY;
wmmhello's avatar
wmmhello 已提交
3258
    }
X
Xiaoyu Wang 已提交
3259
    intervalRange = pInterval->datum.i * f;
wmmhello's avatar
wmmhello 已提交
3260
  } else {
X
Xiaoyu Wang 已提交
3261
    intervalRange = pInterval->datum.i;
wmmhello's avatar
wmmhello 已提交
3262
  }
3263

X
Xiaoyu Wang 已提交
3264
  if ((timeRange / intervalRange) >= MAX_INTERVAL_TIME_WINDOW) {
wmmhello's avatar
wmmhello 已提交
3265 3266 3267 3268
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_FILL_TIME_RANGE);
  }

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3269 3270
}

3271
static int32_t translateFill(STranslateContext* pCxt, SSelectStmt* pSelect, SIntervalWindowNode* pInterval) {
wmmhello's avatar
wmmhello 已提交
3272 3273 3274
  if (NULL == pInterval->pFill) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
3275

3276
  ((SFillNode*)pInterval->pFill)->timeRange = pSelect->timeRange;
3277
  return checkFill(pCxt, (SFillNode*)pInterval->pFill, (SValueNode*)pInterval->pInterval, false);
X
Xiaoyu Wang 已提交
3278 3279
}

X
Xiaoyu Wang 已提交
3280
static int64_t getMonthsFromTimeVal(int64_t val, int32_t fromPrecision, char unit) {
wmmhello's avatar
wmmhello 已提交
3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299
  int64_t days = convertTimeFromPrecisionToUnit(val, fromPrecision, 'd');
  switch (unit) {
    case 'b':
    case 'u':
    case 'a':
    case 's':
    case 'm':
    case 'h':
    case 'd':
    case 'w':
      return days / 28;
    case 'n':
      return val;
    case 'y':
      return val * 12;
    default:
      break;
  }
  return -1;
X
Xiaoyu Wang 已提交
3300 3301
}

X
Xiaoyu Wang 已提交
3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315
static const char* getPrecisionStr(uint8_t precision) {
  switch (precision) {
    case TSDB_TIME_PRECISION_MILLI:
      return TSDB_TIME_PRECISION_MILLI_STR;
    case TSDB_TIME_PRECISION_MICRO:
      return TSDB_TIME_PRECISION_MICRO_STR;
    case TSDB_TIME_PRECISION_NANO:
      return TSDB_TIME_PRECISION_NANO_STR;
    default:
      break;
  }
  return "unknown";
}

3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328
static void convertVarDuration(SValueNode* pOffset, uint8_t precision) {
  const int64_t factors[3] = {NANOSECOND_PER_MSEC, NANOSECOND_PER_USEC, 1};
  const int8_t  units[3] = {TIME_UNIT_MILLISECOND, TIME_UNIT_MICROSECOND, TIME_UNIT_NANOSECOND};

  if (pOffset->unit == 'n') {
    pOffset->datum.i = pOffset->datum.i * 31 * (NANOSECOND_PER_DAY / factors[precision]);
  } else {
    pOffset->datum.i = pOffset->datum.i * 365 * (NANOSECOND_PER_DAY / factors[precision]);
  }

  pOffset->unit = units[precision];
}

X
Xiaoyu Wang 已提交
3329
static int32_t checkIntervalWindow(STranslateContext* pCxt, SIntervalWindowNode* pInterval) {
wmmhello's avatar
wmmhello 已提交
3330 3331 3332
  uint8_t precision = ((SColumnNode*)pInterval->pCol)->node.resType.precision;

  SValueNode* pInter = (SValueNode*)pInterval->pInterval;
H
Haojun Liao 已提交
3333
  bool        valInter = IS_CALENDAR_TIME_DURATION(pInter->unit);
X
Xiaoyu Wang 已提交
3334 3335 3336
  if (pInter->datum.i <= 0 || (!valInter && pInter->datum.i < tsMinIntervalTime)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_VALUE_TOO_SMALL, tsMinIntervalTime,
                                getPrecisionStr(precision));
wmmhello's avatar
wmmhello 已提交
3337 3338 3339 3340 3341 3342 3343 3344 3345 3346
  }

  if (NULL != pInterval->pOffset) {
    SValueNode* pOffset = (SValueNode*)pInterval->pOffset;
    if (pOffset->datum.i <= 0) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_OFFSET_NEGATIVE);
    }
    if (pInter->unit == 'n' && pOffset->unit == 'y') {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_OFFSET_UNIT);
    }
H
Haojun Liao 已提交
3347
    bool fixed = !IS_CALENDAR_TIME_DURATION(pOffset->unit) && !valInter;
wmmhello's avatar
wmmhello 已提交
3348 3349 3350 3351 3352
    if ((fixed && pOffset->datum.i >= pInter->datum.i) ||
        (!fixed && getMonthsFromTimeVal(pOffset->datum.i, precision, pOffset->unit) >=
                       getMonthsFromTimeVal(pInter->datum.i, precision, pInter->unit))) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_OFFSET_TOO_BIG);
    }
3353 3354 3355 3356

    if (pOffset->unit == 'n' || pOffset->unit == 'y') {
      convertVarDuration(pOffset, precision);
    }
wmmhello's avatar
wmmhello 已提交
3357 3358 3359 3360 3361 3362
  }

  if (NULL != pInterval->pSliding) {
    const static int32_t INTERVAL_SLIDING_FACTOR = 100;

    SValueNode* pSliding = (SValueNode*)pInterval->pSliding;
H
Haojun Liao 已提交
3363
    if (IS_CALENDAR_TIME_DURATION(pSliding->unit)) {
wmmhello's avatar
wmmhello 已提交
3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_SLIDING_UNIT);
    }
    if ((pSliding->datum.i < convertTimePrecision(tsMinSlidingTime, TSDB_TIME_PRECISION_MILLI, precision)) ||
        (pInter->datum.i / pSliding->datum.i > INTERVAL_SLIDING_FACTOR)) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_SLIDING_TOO_SMALL);
    }
    if (pSliding->datum.i > pInter->datum.i) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_SLIDING_TOO_BIG);
    }
  }

X
Xiaoyu Wang 已提交
3375 3376 3377
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
3378 3379 3380
static int32_t translateIntervalWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
  SIntervalWindowNode* pInterval = (SIntervalWindowNode*)pSelect->pWindow;
  int32_t              code = checkIntervalWindow(pCxt, pInterval);
X
Xiaoyu Wang 已提交
3381
  if (TSDB_CODE_SUCCESS == code) {
3382
    code = translateFill(pCxt, pSelect, pInterval);
X
Xiaoyu Wang 已提交
3383 3384
  }
  return code;
X
Xiaoyu Wang 已提交
3385 3386
}

X
Xiaoyu Wang 已提交
3387 3388 3389 3390 3391
static int32_t checkStateExpr(STranslateContext* pCxt, SNode* pNode) {
  int32_t type = ((SExprNode*)pNode)->resType.type;
  if (!IS_INTEGER_TYPE(type) && type != TSDB_DATA_TYPE_BOOL && !IS_VAR_DATA_TYPE(type)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STATE_WIN_TYPE);
  }
wmmhello's avatar
wmmhello 已提交
3392

X
Xiaoyu Wang 已提交
3393 3394
  if (QUERY_NODE_COLUMN == nodeType(pNode) && COLUMN_TYPE_TAG == ((SColumnNode*)pNode)->colType) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STATE_WIN_COL);
wmmhello's avatar
wmmhello 已提交
3395
  }
X
Xiaoyu Wang 已提交
3396 3397

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3398 3399
}

3400 3401 3402 3403 3404 3405
static bool hasPartitionByTbname(SNodeList* pPartitionByList) {
  SNode* pPartKey = NULL;
  FOREACH(pPartKey, pPartitionByList) {
    if (QUERY_NODE_FUNCTION == nodeType(pPartKey) && FUNCTION_TYPE_TBNAME == ((SFunctionNode*)pPartKey)->funcType) {
      return true;
    }
X
Xiaoyu Wang 已提交
3406
  }
3407
  return false;
X
Xiaoyu Wang 已提交
3408 3409 3410 3411 3412 3413 3414
}

static int32_t checkStateWindowForStream(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (!pCxt->createStream) {
    return TSDB_CODE_SUCCESS;
  }
  if (TSDB_SUPER_TABLE == ((SRealTableNode*)pSelect->pFromTable)->pMeta->tableType &&
3415
      !hasPartitionByTbname(pSelect->pPartitionByList)) {
X
Xiaoyu Wang 已提交
3416 3417 3418 3419 3420 3421
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY, "Unsupported stream query");
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t translateStateWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
X
Xiaoyu Wang 已提交
3422 3423 3424 3425 3426 3427
  if (QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&
      !isGlobalTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TIMELINE_QUERY,
                                   "STATE_WINDOW requires valid time series input");
  }

X
Xiaoyu Wang 已提交
3428
  SStateWindowNode* pState = (SStateWindowNode*)pSelect->pWindow;
X
Xiaoyu Wang 已提交
3429 3430 3431
  int32_t           code = checkStateExpr(pCxt, pState->pExpr);
  if (TSDB_CODE_SUCCESS == code) {
    code = checkStateWindowForStream(pCxt, pSelect);
X
Xiaoyu Wang 已提交
3432
  }
X
Xiaoyu Wang 已提交
3433
  return code;
X
Xiaoyu Wang 已提交
3434 3435
}

X
Xiaoyu Wang 已提交
3436 3437 3438 3439 3440 3441 3442 3443
static int32_t translateSessionWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&
      !isGlobalTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TIMELINE_QUERY,
                                   "SESSION requires valid time series input");
  }

  SSessionWindowNode* pSession = (SSessionWindowNode*)pSelect->pWindow;
wmmhello's avatar
wmmhello 已提交
3444 3445 3446 3447 3448 3449 3450
  if ('y' == pSession->pGap->unit || 'n' == pSession->pGap->unit || 0 == pSession->pGap->datum.i) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_SESSION_GAP);
  }
  if (PRIMARYKEY_TIMESTAMP_COL_ID != pSession->pCol->colId) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INTER_SESSION_COL);
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3451 3452
}

X
Xiaoyu Wang 已提交
3453 3454 3455 3456 3457 3458 3459 3460 3461
static int32_t translateEventWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&
      !isGlobalTimeLineQuery(((STempTableNode*)pSelect->pFromTable)->pSubquery)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TIMELINE_QUERY,
                                   "EVENT_WINDOW requires valid time series input");
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
3462
static int32_t translateSpecificWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3463 3464
  switch (nodeType(pSelect->pWindow)) {
    case QUERY_NODE_STATE_WINDOW:
X
Xiaoyu Wang 已提交
3465
      return translateStateWindow(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3466
    case QUERY_NODE_SESSION_WINDOW:
X
Xiaoyu Wang 已提交
3467
      return translateSessionWindow(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3468
    case QUERY_NODE_INTERVAL_WINDOW:
X
Xiaoyu Wang 已提交
3469
      return translateIntervalWindow(pCxt, pSelect);
X
Xiaoyu Wang 已提交
3470 3471
    case QUERY_NODE_EVENT_WINDOW:
      return translateEventWindow(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3472 3473 3474 3475
    default:
      break;
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3476 3477
}

X
Xiaoyu Wang 已提交
3478
static int32_t translateWindow(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3479 3480 3481 3482
  if (NULL == pSelect->pWindow) {
    return TSDB_CODE_SUCCESS;
  }
  pCxt->currClause = SQL_CLAUSE_WINDOW;
X
Xiaoyu Wang 已提交
3483
  int32_t code = translateExpr(pCxt, &pSelect->pWindow);
wmmhello's avatar
wmmhello 已提交
3484
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
3485
    code = translateSpecificWindow(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3486 3487
  }
  return code;
3488 3489
}

X
Xiaoyu Wang 已提交
3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503
static int32_t createDefaultFillNode(STranslateContext* pCxt, SNode** pOutput) {
  SFillNode* pFill = (SFillNode*)nodesMakeNode(QUERY_NODE_FILL);
  if (NULL == pFill) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pFill->mode = FILL_MODE_NONE;

  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
  if (NULL == pCol) {
    nodesDestroyNode((SNode*)pFill);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pCol->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
3504
  strcpy(pCol->colName, ROWTS_PSEUDO_COLUMN_NAME);
X
Xiaoyu Wang 已提交
3505 3506 3507 3508 3509 3510
  pFill->pWStartTs = (SNode*)pCol;

  *pOutput = (SNode*)pFill;
  return TSDB_CODE_SUCCESS;
}

3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526
static int32_t createDefaultEveryNode(STranslateContext* pCxt, SNode** pOutput) {
  SValueNode* pEvery = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
  if (NULL == pEvery) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pEvery->node.resType.type = TSDB_DATA_TYPE_BIGINT;
  pEvery->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BIGINT].bytes;
  pEvery->isDuration = true;
  pEvery->literal = taosStrdup("1s");


  *pOutput = (SNode*)pEvery;
  return TSDB_CODE_SUCCESS;
}

3527 3528 3529
static int32_t checkEvery(STranslateContext* pCxt, SValueNode* pInterval) {
  int32_t len = strlen(pInterval->literal);

3530
  char* unit = &pInterval->literal[len - 1];
3531 3532
  if (*unit == 'n' || *unit == 'y') {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE,
G
Ganlin Zhao 已提交
3533
                                   "Unsupported time unit in EVERY clause");
3534 3535 3536 3537 3538 3539 3540 3541
  }

  return TSDB_CODE_SUCCESS;
}

static int32_t translateInterpEvery(STranslateContext* pCxt, SNode** pEvery) {
  int32_t code = TSDB_CODE_SUCCESS;

3542 3543 3544 3545 3546 3547
  if (NULL == *pEvery) {
    code = createDefaultEveryNode(pCxt, pEvery);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkEvery(pCxt, (SValueNode*)(*pEvery));
  }
3548 3549 3550 3551
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, pEvery);
  }

3552 3553 3554 3555 3556 3557
  int64_t interval = ((SValueNode*)(*pEvery))->datum.i;
  if (interval == 0) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE,
                                   "Unsupported time unit in EVERY clause");
  }

3558 3559 3560
  return code;
}

X
Xiaoyu Wang 已提交
3561 3562 3563 3564 3565 3566 3567 3568 3569 3570
static int32_t translateInterpFill(STranslateContext* pCxt, SSelectStmt* pSelect) {
  int32_t code = TSDB_CODE_SUCCESS;

  if (NULL == pSelect->pFill) {
    code = createDefaultFillNode(pCxt, &pSelect->pFill);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pSelect->pFill);
  }
  if (TSDB_CODE_SUCCESS == code) {
3571
    code = getQueryTimeRange(pCxt, pSelect->pRange, &(((SFillNode*)pSelect->pFill)->timeRange));
X
Xiaoyu Wang 已提交
3572 3573
  }
  if (TSDB_CODE_SUCCESS == code) {
3574
    code = checkFill(pCxt, (SFillNode*)pSelect->pFill, (SValueNode*)pSelect->pEvery, true);
X
Xiaoyu Wang 已提交
3575
  }
3576
  if (TSDB_CODE_SUCCESS == code) {
3577
    code = checkFillValues(pCxt, (SFillNode*)pSelect->pFill, pSelect->pProjectionList);
3578
  }
X
Xiaoyu Wang 已提交
3579 3580 3581 3582 3583 3584 3585 3586 3587

  return code;
}

static int32_t translateInterp(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (!pSelect->hasInterpFunc) {
    if (NULL != pSelect->pRange || NULL != pSelect->pEvery || NULL != pSelect->pFill) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_INTERP_CLAUSE);
    }
3588
    if (pSelect->hasInterpPseudoColFunc) {
3589 3590
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_NOT_ALLOWED_FUNC,
                                     "Has Interp pseudo column(s) but missing interp function");
3591
    }
X
Xiaoyu Wang 已提交
3592 3593 3594
    return TSDB_CODE_SUCCESS;
  }

X
Xiaoyu Wang 已提交
3595
  if (NULL == pSelect->pRange || NULL == pSelect->pEvery || NULL == pSelect->pFill) {
G
fix bug  
Ganlin Zhao 已提交
3596
    if (pSelect->pRange != NULL && QUERY_NODE_OPERATOR == nodeType(pSelect->pRange) && pSelect->pEvery == NULL) {
3597 3598 3599 3600 3601
      // single point interp every can be omitted
    } else {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_INTERP_CLAUSE,
                                     "Missing RANGE clause, EVERY clause or FILL clause");
    }
X
Xiaoyu Wang 已提交
3602 3603
  }

X
Xiaoyu Wang 已提交
3604 3605
  int32_t code = translateExpr(pCxt, &pSelect->pRange);
  if (TSDB_CODE_SUCCESS == code) {
3606
    code = translateInterpEvery(pCxt, &pSelect->pEvery);
X
Xiaoyu Wang 已提交
3607 3608 3609 3610 3611 3612 3613
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateInterpFill(pCxt, pSelect);
  }
  return code;
}

3614
static int32_t translatePartitionBy(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3615
  pCxt->currClause = SQL_CLAUSE_PARTITION_BY;
3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627
  int32_t code = TSDB_CODE_SUCCESS;
  
  if (pSelect->pPartitionByList) {
    int8_t typeType = getTableTypeFromTableNode(pSelect->pFromTable);
    SNode* pPar = nodesListGetNode(pSelect->pPartitionByList, 0);
    if (!((TSDB_NORMAL_TABLE == typeType || TSDB_CHILD_TABLE == typeType) && 
      1 == pSelect->pPartitionByList->length && (QUERY_NODE_FUNCTION == nodeType(pPar) && FUNCTION_TYPE_TBNAME == ((SFunctionNode*)pPar)->funcType))) {
      pSelect->timeLineResMode = TIME_LINE_MULTI;
    }
    
    code = translateExprList(pCxt, pSelect->pPartitionByList);
  }
3628 3629 3630 3631 3632 3633 3634
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExprList(pCxt, pSelect->pTags);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pSelect->pSubtable);
  }
  return code;
3635 3636
}

3637
static int32_t translateWhere(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3638
  pCxt->currClause = SQL_CLAUSE_WHERE;
3639
  int32_t code = translateExpr(pCxt, &pSelect->pWhere);
3640
  if (TSDB_CODE_SUCCESS == code) {
3641 3642
    code = getQueryTimeRange(pCxt, pSelect->pWhere, &pSelect->timeRange);
  }
D
dapan1121 已提交
3643 3644 3645
  if (TSDB_CODE_SUCCESS == code && pSelect->timeRange.skey > pSelect->timeRange.ekey) {
    pSelect->isEmptyResult = true;
  }
3646
  return code;
3647 3648
}

X
Xiaoyu Wang 已提交
3649
static int32_t translateFrom(STranslateContext* pCxt, SNode* pTable) {
wmmhello's avatar
wmmhello 已提交
3650
  pCxt->currClause = SQL_CLAUSE_FROM;
X
Xiaoyu Wang 已提交
3651
  return translateTable(pCxt, pTable);
3652 3653
}

X
Xiaoyu Wang 已提交
3654
static int32_t checkLimit(STranslateContext* pCxt, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
3655 3656 3657 3658
  if ((NULL != pSelect->pLimit && pSelect->pLimit->offset < 0) ||
      (NULL != pSelect->pSlimit && pSelect->pSlimit->offset < 0)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_OFFSET_LESS_ZERO);
  }
X
Xiaoyu Wang 已提交
3659

D
dapan1121 已提交
3660 3661
  if (NULL != pSelect->pSlimit && (NULL == pSelect->pPartitionByList && NULL == pSelect->pGroupByList)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_SLIMIT_LEAK_PARTITION_GROUP_BY);
wmmhello's avatar
wmmhello 已提交
3662
  }
X
Xiaoyu Wang 已提交
3663

wmmhello's avatar
wmmhello 已提交
3664
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
3665 3666
}

3667
static int32_t createPrimaryKeyColByTable(STranslateContext* pCxt, STableNode* pTable, SNode** pPrimaryKey) {
3668
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
wmmhello's avatar
wmmhello 已提交
3669 3670 3671 3672
  if (NULL == pCol) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pCol->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
3673
  strcpy(pCol->colName, ROWTS_PSEUDO_COLUMN_NAME);
3674 3675 3676
  bool    found = false;
  int32_t code = findAndSetColumn(pCxt, &pCol, pTable, &found);
  if (TSDB_CODE_SUCCESS != code || !found) {
wmmhello's avatar
wmmhello 已提交
3677 3678 3679 3680
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TIMELINE_FUNC);
  }
  *pPrimaryKey = (SNode*)pCol;
  return TSDB_CODE_SUCCESS;
3681 3682 3683
}

static int32_t createPrimaryKeyCol(STranslateContext* pCxt, SNode** pPrimaryKey) {
wmmhello's avatar
wmmhello 已提交
3684 3685 3686 3687 3688 3689
  STableNode* pTable = NULL;
  int32_t     code = findTable(pCxt, NULL, &pTable);
  if (TSDB_CODE_SUCCESS == code) {
    code = createPrimaryKeyColByTable(pCxt, pTable, pPrimaryKey);
  }
  return code;
3690 3691
}

3692
static EDealRes appendTsForImplicitTsFuncImpl(SNode* pNode, void* pContext) {
wmmhello's avatar
wmmhello 已提交
3693
  STranslateContext* pCxt = pContext;
3694
  if (isImplicitTsFunc(pNode)) {
wmmhello's avatar
wmmhello 已提交
3695 3696 3697 3698 3699 3700 3701 3702 3703
    SFunctionNode* pFunc = (SFunctionNode*)pNode;
    SNode*         pPrimaryKey = NULL;
    pCxt->errCode = createPrimaryKeyCol(pCxt, &pPrimaryKey);
    if (TSDB_CODE_SUCCESS == pCxt->errCode) {
      pCxt->errCode = nodesListMakeStrictAppend(&pFunc->pParameterList, pPrimaryKey);
    }
    return TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR;
  }
  return DEAL_RES_CONTINUE;
3704 3705
}

3706 3707
static int32_t appendTsForImplicitTsFunc(STranslateContext* pCxt, SSelectStmt* pSelect) {
  nodesWalkSelectStmt(pSelect, SQL_CLAUSE_FROM, appendTsForImplicitTsFuncImpl, pCxt);
wmmhello's avatar
wmmhello 已提交
3708
  return pCxt->errCode;
3709
}
X
Xiaoyu Wang 已提交
3710

3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726 3727 3728
typedef struct SReplaceOrderByAliasCxt {
  STranslateContext* pTranslateCxt;
  SNodeList*         pProjectionList;
} SReplaceOrderByAliasCxt;

static EDealRes replaceOrderByAliasImpl(SNode** pNode, void* pContext) {
  SReplaceOrderByAliasCxt* pCxt = pContext;
  if (QUERY_NODE_COLUMN_REF == nodeType(*pNode)) {
    SNodeList* pProjectionList = pCxt->pProjectionList;
    SNode*     pProject = NULL;
    FOREACH(pProject, pProjectionList) {
      SExprNode* pExpr = (SExprNode*)pProject;
      if (0 == strcmp(((SColumnRefNode*)*pNode)->colName, pExpr->aliasName)) {
        SNode* pNew = nodesCloneNode(pProject);
        if (NULL == pNew) {
          pCxt->pTranslateCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
          return DEAL_RES_ERROR;
        }
3729
        ((SExprNode*)pNew)->orderAlias = true;
3730 3731 3732 3733 3734 3735 3736 3737 3738 3739
        nodesDestroyNode(*pNode);
        *pNode = pNew;
        return DEAL_RES_CONTINUE;
      }
    }
  }
  return DEAL_RES_CONTINUE;
}

static int32_t replaceOrderByAlias(STranslateContext* pCxt, SNodeList* pProjectionList, SNodeList* pOrderByList) {
3740 3741 3742
  if (NULL == pOrderByList) {
    return TSDB_CODE_SUCCESS;
  }
3743 3744 3745 3746 3747
  SReplaceOrderByAliasCxt cxt = {.pTranslateCxt = pCxt, .pProjectionList = pProjectionList};
  nodesRewriteExprsPostOrder(pOrderByList, replaceOrderByAliasImpl, &cxt);
  return pCxt->errCode;
}

3748 3749 3750 3751 3752 3753 3754 3755
static void resetResultTimeline(SSelectStmt* pSelect) {
  if (NULL == pSelect->pOrderByList) {
    return;
  }
  SNode* pOrder = ((SOrderByExprNode*)nodesListGetNode(pSelect->pOrderByList, 0))->pExpr;
  if ((QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&
       isPrimaryKey((STempTableNode*)pSelect->pFromTable, pOrder)) ||
      (QUERY_NODE_TEMP_TABLE != nodeType(pSelect->pFromTable) && isPrimaryKeyImpl(pOrder))) {
3756
    pSelect->timeLineResMode = TIME_LINE_GLOBAL;
3757
  } else {
3758
    pSelect->timeLineResMode = TIME_LINE_NONE;
3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769
  }
}

static int32_t replaceOrderByAliasForSelect(STranslateContext* pCxt, SSelectStmt* pSelect) {
  int32_t code = replaceOrderByAlias(pCxt, pSelect->pProjectionList, pSelect->pOrderByList);
  if (TSDB_CODE_SUCCESS == pCxt->errCode) {
    resetResultTimeline(pSelect);
  }
  return code;
}

3770
static int32_t translateSelectWithoutFrom(STranslateContext* pCxt, SSelectStmt* pSelect) {
3771
  pCxt->pCurrStmt = (SNode*)pSelect;
3772 3773 3774 3775 3776
  pCxt->currClause = SQL_CLAUSE_SELECT;
  return translateExprList(pCxt, pSelect->pProjectionList);
}

static int32_t translateSelectFrom(STranslateContext* pCxt, SSelectStmt* pSelect) {
3777
  pCxt->pCurrStmt = (SNode*)pSelect;
X
Xiaoyu Wang 已提交
3778
  int32_t code = translateFrom(pCxt, pSelect->pFromTable);
wmmhello's avatar
wmmhello 已提交
3779
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
3780
    pSelect->precision = ((STableNode*)pSelect->pFromTable)->precision;
3781
    code = translateWhere(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3782 3783
  }
  if (TSDB_CODE_SUCCESS == code) {
3784
    code = translatePartitionBy(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateWindow(pCxt, pSelect);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateGroupBy(pCxt, pSelect);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateHaving(pCxt, pSelect);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateSelectList(pCxt, pSelect);
  }
  if (TSDB_CODE_SUCCESS == code) {
D
dapan1121 已提交
3799
    code = checkHavingGroupBy(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3800 3801 3802 3803 3804 3805 3806
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateOrderBy(pCxt, pSelect);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkAggColCoexist(pCxt, pSelect);
  }
3807 3808 3809
  if (TSDB_CODE_SUCCESS == code) {
    code = checkWindowFuncCoexist(pCxt, pSelect);
  }
wmmhello's avatar
wmmhello 已提交
3810 3811 3812
  if (TSDB_CODE_SUCCESS == code) {
    code = checkLimit(pCxt, pSelect);
  }
X
Xiaoyu Wang 已提交
3813 3814 3815
  if (TSDB_CODE_SUCCESS == code) {
    code = translateInterp(pCxt, pSelect);
  }
3816
  if (TSDB_CODE_SUCCESS == code) {
3817
    code = appendTsForImplicitTsFunc(pCxt, pSelect);
3818
  }
wmmhello's avatar
wmmhello 已提交
3819
  if (TSDB_CODE_SUCCESS == code) {
3820
    code = replaceOrderByAliasForSelect(pCxt, pSelect);
wmmhello's avatar
wmmhello 已提交
3821
  }
X
Xiaoyu Wang 已提交
3822 3823 3824
  if (TSDB_CODE_SUCCESS == code) {
    code = setTableCacheLastMode(pCxt, pSelect);
  }
D
dapan1121 已提交
3825 3826 3827
  if (TSDB_CODE_SUCCESS == code) {
    code = replaceTbName(pCxt, pSelect);
  }
3828

wmmhello's avatar
wmmhello 已提交
3829
  return code;
3830 3831
}

3832 3833 3834 3835 3836 3837 3838 3839
static int32_t translateSelect(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (NULL == pSelect->pFromTable) {
    return translateSelectWithoutFrom(pCxt, pSelect);
  } else {
    return translateSelectFrom(pCxt, pSelect);
  }
}

3840
static SNode* createSetOperProject(const char* pTableAlias, SNode* pNode) {
3841
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
wmmhello's avatar
wmmhello 已提交
3842 3843 3844 3845
  if (NULL == pCol) {
    return NULL;
  }
  pCol->node.resType = ((SExprNode*)pNode)->resType;
X
Xiaoyu Wang 已提交
3846 3847 3848 3849
  snprintf(pCol->tableAlias, sizeof(pCol->tableAlias), "%s", pTableAlias);
  snprintf(pCol->colName, sizeof(pCol->colName), "%s", ((SExprNode*)pNode)->aliasName);
  snprintf(pCol->node.aliasName, sizeof(pCol->node.aliasName), "%s", pCol->colName);
  snprintf(pCol->node.userAlias, sizeof(pCol->node.userAlias), "%s", ((SExprNode*)pNode)->userAlias);
wmmhello's avatar
wmmhello 已提交
3850
  return (SNode*)pCol;
X
Xiaoyu Wang 已提交
3851 3852
}

3853
static int32_t translateSetOperProject(STranslateContext* pCxt, SSetOperator* pSetOperator) {
wmmhello's avatar
wmmhello 已提交
3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864
  SNodeList* pLeftProjections = getProjectList(pSetOperator->pLeft);
  SNodeList* pRightProjections = getProjectList(pSetOperator->pRight);
  if (LIST_LENGTH(pLeftProjections) != LIST_LENGTH(pRightProjections)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INCORRECT_NUM_OF_COL);
  }

  SNode* pLeft = NULL;
  SNode* pRight = NULL;
  FORBOTH(pLeft, pLeftProjections, pRight, pRightProjections) {
    SExprNode* pLeftExpr = (SExprNode*)pLeft;
    SExprNode* pRightExpr = (SExprNode*)pRight;
X
Xiaoyu Wang 已提交
3865 3866
    int32_t    comp = dataTypeComp(&pLeftExpr->resType, &pRightExpr->resType);
    if (comp > 0) {
wmmhello's avatar
wmmhello 已提交
3867 3868
      SNode*  pRightFunc = NULL;
      int32_t code = createCastFunc(pCxt, pRight, pLeftExpr->resType, &pRightFunc);
X
Xiaoyu Wang 已提交
3869
      if (TSDB_CODE_SUCCESS != code || NULL == pRightFunc) {  // deal scan coverity
wmmhello's avatar
wmmhello 已提交
3870 3871 3872 3873
        return code;
      }
      REPLACE_LIST2_NODE(pRightFunc);
      pRightExpr = (SExprNode*)pRightFunc;
X
Xiaoyu Wang 已提交
3874 3875 3876
    } else if (comp < 0) {
      SNode*  pLeftFunc = NULL;
      int32_t code = createCastFunc(pCxt, pLeft, pRightExpr->resType, &pLeftFunc);
X
Xiaoyu Wang 已提交
3877
      if (TSDB_CODE_SUCCESS != code || NULL == pLeftFunc) {  // deal scan coverity
X
Xiaoyu Wang 已提交
3878 3879 3880 3881 3882 3883 3884 3885 3886 3887
        return code;
      }
      REPLACE_LIST1_NODE(pLeftFunc);
      SExprNode* pLeftFuncExpr = (SExprNode*)pLeftFunc;
      snprintf(pLeftFuncExpr->aliasName, sizeof(pLeftFuncExpr->aliasName), "%s", pLeftExpr->aliasName);
      snprintf(pLeftFuncExpr->userAlias, sizeof(pLeftFuncExpr->userAlias), "%s", pLeftExpr->userAlias);
      pLeft = pLeftFunc;
      pLeftExpr = pLeftFuncExpr;
    }
    snprintf(pRightExpr->aliasName, sizeof(pRightExpr->aliasName), "%s", pLeftExpr->aliasName);
3888 3889 3890 3891 3892 3893 3894
    SNode* pProj = createSetOperProject(pSetOperator->stmtName, pLeft);
    if (QUERY_NODE_COLUMN == nodeType(pLeft) && QUERY_NODE_COLUMN == nodeType(pRight) 
      && ((SColumnNode*)pLeft)->colId == PRIMARYKEY_TIMESTAMP_COL_ID
      && ((SColumnNode*)pRight)->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
      ((SColumnNode*)pProj)->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
    }
    if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(&pSetOperator->pProjectionList, pProj)) {
wmmhello's avatar
wmmhello 已提交
3895 3896 3897 3898
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  return TSDB_CODE_SUCCESS;
3899 3900
}

X
Xiaoyu Wang 已提交
3901 3902 3903 3904
static uint8_t calcSetOperatorPrecision(SSetOperator* pSetOperator) {
  return calcPrecision(getStmtPrecision(pSetOperator->pLeft), getStmtPrecision(pSetOperator->pRight));
}

3905
static int32_t translateSetOperOrderBy(STranslateContext* pCxt, SSetOperator* pSetOperator) {
3906 3907 3908 3909
  if (NULL == pSetOperator->pOrderByList || pSetOperator->pOrderByList->length <= 0) {
    return TSDB_CODE_SUCCESS;
  }

3910 3911 3912 3913 3914
  bool    other;
  int32_t code = translateOrderByPosition(pCxt, pSetOperator->pProjectionList, pSetOperator->pOrderByList, &other);
  if (TSDB_CODE_SUCCESS == code) {
    if (other) {
      pCxt->currClause = SQL_CLAUSE_ORDER_BY;
3915
      pCxt->pCurrStmt = (SNode*)pSetOperator;
3916 3917 3918 3919 3920 3921
      code = translateExprList(pCxt, pSetOperator->pOrderByList);
    }
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = replaceOrderByAlias(pCxt, pSetOperator->pProjectionList, pSetOperator->pOrderByList);
  }
3922 3923 3924 3925 3926 3927 3928 3929
  if (TSDB_CODE_SUCCESS == code) {
    SNode* pOrder = ((SOrderByExprNode*)nodesListGetNode(pSetOperator->pOrderByList, 0))->pExpr;
    if (isPrimaryKeyImpl(pOrder)) {
      pSetOperator->timeLineResMode = TIME_LINE_GLOBAL;
    } else {
      pSetOperator->timeLineResMode = TIME_LINE_NONE;
    }
  }
3930 3931 3932
  return code;
}

3933 3934 3935 3936 3937 3938 3939
static int32_t checkSetOperLimit(STranslateContext* pCxt, SLimitNode* pLimit) {
  if ((NULL != pLimit && pLimit->offset < 0)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_OFFSET_LESS_ZERO);
  }
  return TSDB_CODE_SUCCESS;
}

3940
static int32_t translateSetOperator(STranslateContext* pCxt, SSetOperator* pSetOperator) {
wmmhello's avatar
wmmhello 已提交
3941 3942 3943 3944 3945 3946 3947 3948
  int32_t code = translateQuery(pCxt, pSetOperator->pLeft);
  if (TSDB_CODE_SUCCESS == code) {
    code = resetTranslateNamespace(pCxt);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateQuery(pCxt, pSetOperator->pRight);
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
3949
    pSetOperator->precision = calcSetOperatorPrecision(pSetOperator);
3950 3951 3952 3953
    code = translateSetOperProject(pCxt, pSetOperator);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateSetOperOrderBy(pCxt, pSetOperator);
wmmhello's avatar
wmmhello 已提交
3954
  }
3955 3956 3957
  if (TSDB_CODE_SUCCESS == code) {
    code = checkSetOperLimit(pCxt, (SLimitNode*)pSetOperator->pLimit);
  }
wmmhello's avatar
wmmhello 已提交
3958
  return code;
3959 3960
}

X
Xiaoyu Wang 已提交
3961 3962 3963 3964 3965 3966 3967 3968
static int32_t partitionDeleteWhere(STranslateContext* pCxt, SDeleteStmt* pDelete) {
  if (NULL == pDelete->pWhere) {
    pDelete->timeRange = TSWINDOW_INITIALIZER;
    return TSDB_CODE_SUCCESS;
  }

  SNode*  pPrimaryKeyCond = NULL;
  SNode*  pOtherCond = NULL;
X
Xiaoyu Wang 已提交
3969
  int32_t code = filterPartitionCond(&pDelete->pWhere, &pPrimaryKeyCond, NULL, &pDelete->pTagCond, &pOtherCond);
X
Xiaoyu Wang 已提交
3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985
  if (TSDB_CODE_SUCCESS == code && NULL != pOtherCond) {
    code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DELETE_WHERE);
  }
  if (TSDB_CODE_SUCCESS == code) {
    bool isStrict = false;
    code = getTimeRange(&pPrimaryKeyCond, &pDelete->timeRange, &isStrict);
    if (TSDB_CODE_SUCCESS == code && !isStrict) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DELETE_WHERE);
    }
  }
  nodesDestroyNode(pPrimaryKeyCond);
  nodesDestroyNode(pOtherCond);
  return code;
}

static int32_t translateDeleteWhere(STranslateContext* pCxt, SDeleteStmt* pDelete) {
3986 3987
  pCxt->currClause = SQL_CLAUSE_WHERE;
  int32_t code = translateExpr(pCxt, &pDelete->pWhere);
X
Xiaoyu Wang 已提交
3988 3989 3990 3991 3992 3993 3994
  if (TSDB_CODE_SUCCESS == code) {
    code = partitionDeleteWhere(pCxt, pDelete);
  }
  return code;
}

static int32_t translateDelete(STranslateContext* pCxt, SDeleteStmt* pDelete) {
3995
  pCxt->pCurrStmt = (SNode*)pDelete;
X
Xiaoyu Wang 已提交
3996 3997
  int32_t code = translateFrom(pCxt, pDelete->pFromTable);
  if (TSDB_CODE_SUCCESS == code) {
K
kailixu 已提交
3998
    pDelete->precision = ((STableNode*)pDelete->pFromTable)->precision;
X
Xiaoyu Wang 已提交
3999 4000
    code = translateDeleteWhere(pCxt, pDelete);
  }
4001
  pCxt->currClause = SQL_CLAUSE_SELECT;
X
Xiaoyu Wang 已提交
4002 4003 4004
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pDelete->pCountFunc);
  }
4005 4006 4007 4008 4009 4010
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pDelete->pFirstFunc);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pDelete->pLastFunc);
  }
X
Xiaoyu Wang 已提交
4011 4012 4013
  return code;
}

4014 4015 4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078 4079 4080
static int32_t translateInsertCols(STranslateContext* pCxt, SInsertStmt* pInsert) {
  if (NULL == pInsert->pCols) {
    return createAllColumns(pCxt, false, &pInsert->pCols);
  }
  return translateExprList(pCxt, pInsert->pCols);
}

static int32_t translateInsertQuery(STranslateContext* pCxt, SInsertStmt* pInsert) {
  int32_t code = resetTranslateNamespace(pCxt);
  if (TSDB_CODE_SUCCESS == code) {
    code = translateQuery(pCxt, pInsert->pQuery);
  }
  return code;
}

static int32_t addOrderByPrimaryKeyToQueryImpl(STranslateContext* pCxt, SNode* pPrimaryKeyExpr,
                                               SNodeList** pOrderByList) {
  SOrderByExprNode* pOrderByExpr = (SOrderByExprNode*)nodesMakeNode(QUERY_NODE_ORDER_BY_EXPR);
  if (NULL == pOrderByExpr) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pOrderByExpr->nullOrder = NULL_ORDER_FIRST;
  pOrderByExpr->order = ORDER_ASC;
  pOrderByExpr->pExpr = nodesCloneNode(pPrimaryKeyExpr);
  if (NULL == pOrderByExpr->pExpr) {
    nodesDestroyNode((SNode*)pOrderByExpr);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  ((SExprNode*)pOrderByExpr->pExpr)->orderAlias = true;
  NODES_DESTORY_LIST(*pOrderByList);
  return nodesListMakeStrictAppend(pOrderByList, (SNode*)pOrderByExpr);
}

static int32_t addOrderByPrimaryKeyToQuery(STranslateContext* pCxt, SNode* pPrimaryKeyExpr, SNode* pStmt) {
  if (QUERY_NODE_SELECT_STMT == nodeType(pStmt)) {
    return addOrderByPrimaryKeyToQueryImpl(pCxt, pPrimaryKeyExpr, &((SSelectStmt*)pStmt)->pOrderByList);
  }
  return addOrderByPrimaryKeyToQueryImpl(pCxt, pPrimaryKeyExpr, &((SSetOperator*)pStmt)->pOrderByList);
}

static int32_t translateInsertProject(STranslateContext* pCxt, SInsertStmt* pInsert) {
  SNodeList* pProjects = getProjectList(pInsert->pQuery);
  if (LIST_LENGTH(pInsert->pCols) != LIST_LENGTH(pProjects)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM, "Illegal number of columns");
  }

  SNode* pPrimaryKeyExpr = NULL;
  SNode* pBoundCol = NULL;
  SNode* pProj = NULL;
  FORBOTH(pBoundCol, pInsert->pCols, pProj, pProjects) {
    SColumnNode* pCol = (SColumnNode*)pBoundCol;
    SExprNode*   pExpr = (SExprNode*)pProj;
    if (!dataTypeEqual(&pCol->node.resType, &pExpr->resType)) {
      SNode*  pFunc = NULL;
      int32_t code = createCastFunc(pCxt, pProj, pCol->node.resType, &pFunc);
      if (TSDB_CODE_SUCCESS != code) {
        return code;
      }
      REPLACE_LIST2_NODE(pFunc);
      pExpr = (SExprNode*)pFunc;
    }
    snprintf(pExpr->aliasName, sizeof(pExpr->aliasName), "%s", pCol->colName);
    if (PRIMARYKEY_TIMESTAMP_COL_ID == pCol->colId) {
      pPrimaryKeyExpr = pProj;
    }
  }

X
Xiaoyu Wang 已提交
4081 4082 4083 4084 4085
  if (NULL == pPrimaryKeyExpr) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM,
                                   "Primary timestamp column can not be null");
  }

4086 4087 4088
  return addOrderByPrimaryKeyToQuery(pCxt, pPrimaryKeyExpr, pInsert->pQuery);
}

X
Xiaoyu Wang 已提交
4089 4090 4091 4092 4093 4094 4095 4096 4097 4098
static int32_t translateInsertTable(STranslateContext* pCxt, SNode* pTable) {
  int32_t code = translateFrom(pCxt, pTable);
  if (TSDB_CODE_SUCCESS == code && TSDB_CHILD_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType &&
      TSDB_NORMAL_TABLE != ((SRealTableNode*)pTable)->pMeta->tableType) {
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR,
                                   "insert data into super table is not supported");
  }
  return code;
}

4099 4100
static int32_t translateInsert(STranslateContext* pCxt, SInsertStmt* pInsert) {
  pCxt->pCurrStmt = (SNode*)pInsert;
X
Xiaoyu Wang 已提交
4101
  int32_t code = translateInsertTable(pCxt, pInsert->pTable);
4102
  if (TSDB_CODE_SUCCESS == code) {
4103
    code = translateInsertCols(pCxt, pInsert);
4104
  }
4105
  if (TSDB_CODE_SUCCESS == code) {
4106
    code = translateInsertQuery(pCxt, pInsert);
4107
  }
4108
  if (TSDB_CODE_SUCCESS == code) {
4109
    code = translateInsertProject(pCxt, pInsert);
4110 4111 4112 4113
  }
  return code;
}

4114
static int64_t getUnitPerMinute(uint8_t precision) {
wmmhello's avatar
wmmhello 已提交
4115 4116 4117 4118
  switch (precision) {
    case TSDB_TIME_PRECISION_MILLI:
      return MILLISECOND_PER_MINUTE;
    case TSDB_TIME_PRECISION_MICRO:
wafwerar's avatar
wafwerar 已提交
4119
      return MILLISECOND_PER_MINUTE * 1000LL;
wmmhello's avatar
wmmhello 已提交
4120 4121 4122 4123 4124 4125
    case TSDB_TIME_PRECISION_NANO:
      return NANOSECOND_PER_MINUTE;
    default:
      break;
  }
  return MILLISECOND_PER_MINUTE;
4126 4127 4128
}

static int64_t getBigintFromValueNode(SValueNode* pVal) {
wmmhello's avatar
wmmhello 已提交
4129 4130 4131 4132
  if (pVal->isDuration) {
    return pVal->datum.i / getUnitPerMinute(pVal->node.resType.precision);
  }
  return pVal->datum.i;
4133 4134
}

X
Xiaoyu Wang 已提交
4135
static int32_t buildCreateDbRetentions(const SNodeList* pRetentions, SCreateDbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154
  if (NULL != pRetentions) {
    pReq->pRetensions = taosArrayInit(LIST_LENGTH(pRetentions), sizeof(SRetention));
    if (NULL == pReq->pRetensions) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    SValueNode* pFreq = NULL;
    SValueNode* pKeep = NULL;
    SNode*      pNode = NULL;
    int32_t     index = 0;
    FOREACH(pNode, pRetentions) {
      pFreq = (SValueNode*)nodesListGetNode(((SNodeListNode*)pNode)->pNodeList, 0);
      pKeep = (SValueNode*)nodesListGetNode(((SNodeListNode*)pNode)->pNodeList, 1);
      SRetention retention = {
          .freq = pFreq->datum.i, .freqUnit = pFreq->unit, .keep = pKeep->datum.i, .keepUnit = pKeep->unit};
      taosArrayPush(pReq->pRetensions, &retention);
    }
    pReq->numOfRetensions = taosArrayGetSize(pReq->pRetensions);
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4155 4156 4157
}

static int32_t buildCreateDbReq(STranslateContext* pCxt, SCreateDatabaseStmt* pStmt, SCreateDbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171
  SName name = {0};
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, pReq->db);
  pReq->numOfVgroups = pStmt->pOptions->numOfVgroups;
  pReq->numOfStables = pStmt->pOptions->singleStable;
  pReq->buffer = pStmt->pOptions->buffer;
  pReq->pageSize = pStmt->pOptions->pagesize;
  pReq->pages = pStmt->pOptions->pages;
  pReq->daysPerFile = pStmt->pOptions->daysPerFile;
  pReq->daysToKeep0 = pStmt->pOptions->keep[0];
  pReq->daysToKeep1 = pStmt->pOptions->keep[1];
  pReq->daysToKeep2 = pStmt->pOptions->keep[2];
  pReq->minRows = pStmt->pOptions->minRowsPerBlock;
  pReq->maxRows = pStmt->pOptions->maxRowsPerBlock;
4172
  pReq->walFsyncPeriod = pStmt->pOptions->fsyncPeriod;
wmmhello's avatar
wmmhello 已提交
4173 4174 4175 4176 4177
  pReq->walLevel = pStmt->pOptions->walLevel;
  pReq->precision = pStmt->pOptions->precision;
  pReq->compression = pStmt->pOptions->compressionLevel;
  pReq->replications = pStmt->pOptions->replica;
  pReq->strict = pStmt->pOptions->strict;
4178
  pReq->cacheLast = pStmt->pOptions->cacheModel;
4179
  pReq->cacheLastSize = pStmt->pOptions->cacheLastSize;
X
Xiaoyu Wang 已提交
4180
  pReq->schemaless = pStmt->pOptions->schemaless;
X
Xiaoyu Wang 已提交
4181 4182 4183 4184
  pReq->walRetentionPeriod = pStmt->pOptions->walRetentionPeriod;
  pReq->walRetentionSize = pStmt->pOptions->walRetentionSize;
  pReq->walRollPeriod = pStmt->pOptions->walRollPeriod;
  pReq->walSegmentSize = pStmt->pOptions->walSegmentSize;
4185
  pReq->sstTrigger = pStmt->pOptions->sstTrigger;
4186 4187
  pReq->hashPrefix = pStmt->pOptions->tablePrefix;
  pReq->hashSuffix = pStmt->pOptions->tableSuffix;
4188
  pReq->tsdbPageSize = pStmt->pOptions->tsdbPageSize;
wmmhello's avatar
wmmhello 已提交
4189 4190
  pReq->ignoreExist = pStmt->ignoreExists;
  return buildCreateDbRetentions(pStmt->pOptions->pRetentions, pReq);
X
Xiaoyu Wang 已提交
4191 4192
}

4193 4194
static int32_t checkRangeOption(STranslateContext* pCxt, int32_t code, const char* pName, int64_t val, int64_t minVal,
                                int64_t maxVal) {
wmmhello's avatar
wmmhello 已提交
4195
  if (val >= 0 && (val < minVal || val > maxVal)) {
4196 4197
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, code,
                                   "Invalid option %s: %" PRId64 " valid range: [%" PRId64 ", %" PRId64 "]", pName, val,
X
Xiaoyu Wang 已提交
4198
                                   minVal, maxVal);
wmmhello's avatar
wmmhello 已提交
4199 4200
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4201 4202
}

4203 4204 4205 4206 4207
static int32_t checkDbRangeOption(STranslateContext* pCxt, const char* pName, int32_t val, int32_t minVal,
                                  int32_t maxVal) {
  return checkRangeOption(pCxt, TSDB_CODE_PAR_INVALID_DB_OPTION, pName, val, minVal, maxVal);
}

4208 4209
static int32_t checkTableRangeOption(STranslateContext* pCxt, const char* pName, int64_t val, int64_t minVal,
                                     int64_t maxVal) {
4210 4211 4212
  return checkRangeOption(pCxt, TSDB_CODE_PAR_INVALID_TABLE_OPTION, pName, val, minVal, maxVal);
}

X
Xiaoyu Wang 已提交
4213
static int32_t checkDbDaysOption(STranslateContext* pCxt, SDatabaseOptions* pOptions) {
wmmhello's avatar
wmmhello 已提交
4214 4215 4216 4217 4218 4219
  if (NULL != pOptions->pDaysPerFile) {
    if (DEAL_RES_ERROR == translateValue(pCxt, pOptions->pDaysPerFile)) {
      return pCxt->errCode;
    }
    if (TIME_UNIT_MINUTE != pOptions->pDaysPerFile->unit && TIME_UNIT_HOUR != pOptions->pDaysPerFile->unit &&
        TIME_UNIT_DAY != pOptions->pDaysPerFile->unit) {
4220 4221 4222
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                     "Invalid option duration unit: %c, only %c, %c, %c allowed",
                                     pOptions->pDaysPerFile->unit, TIME_UNIT_MINUTE, TIME_UNIT_HOUR, TIME_UNIT_DAY);
wmmhello's avatar
wmmhello 已提交
4223 4224 4225
    }
    pOptions->daysPerFile = getBigintFromValueNode(pOptions->pDaysPerFile);
  }
4226
  return checkDbRangeOption(pCxt, "daysPerFile", pOptions->daysPerFile, TSDB_MIN_DAYS_PER_FILE, TSDB_MAX_DAYS_PER_FILE);
X
Xiaoyu Wang 已提交
4227 4228
}

X
Xiaoyu Wang 已提交
4229
static int32_t checkDbKeepOption(STranslateContext* pCxt, SDatabaseOptions* pOptions) {
wmmhello's avatar
wmmhello 已提交
4230 4231 4232 4233 4234 4235
  if (NULL == pOptions->pKeep) {
    return TSDB_CODE_SUCCESS;
  }

  int32_t numOfKeep = LIST_LENGTH(pOptions->pKeep);
  if (numOfKeep > 3 || numOfKeep < 1) {
4236
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid number of keep options");
wmmhello's avatar
wmmhello 已提交
4237 4238 4239 4240 4241 4242 4243 4244 4245 4246
  }

  SNode* pNode = NULL;
  FOREACH(pNode, pOptions->pKeep) {
    SValueNode* pVal = (SValueNode*)pNode;
    if (DEAL_RES_ERROR == translateValue(pCxt, pVal)) {
      return pCxt->errCode;
    }
    if (pVal->isDuration && TIME_UNIT_MINUTE != pVal->unit && TIME_UNIT_HOUR != pVal->unit &&
        TIME_UNIT_DAY != pVal->unit) {
4247 4248
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                     "Invalid option keep unit: %c, only m, h, d allowed", pVal->unit);
wmmhello's avatar
wmmhello 已提交
4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267
    }
    if (!pVal->isDuration) {
      pVal->datum.i = pVal->datum.i * 1440;
    }
  }

  pOptions->keep[0] = getBigintFromValueNode((SValueNode*)nodesListGetNode(pOptions->pKeep, 0));

  if (numOfKeep < 2) {
    pOptions->keep[1] = pOptions->keep[0];
  } else {
    pOptions->keep[1] = getBigintFromValueNode((SValueNode*)nodesListGetNode(pOptions->pKeep, 1));
  }
  if (numOfKeep < 3) {
    pOptions->keep[2] = pOptions->keep[1];
  } else {
    pOptions->keep[2] = getBigintFromValueNode((SValueNode*)nodesListGetNode(pOptions->pKeep, 2));
  }

4268 4269
  int64_t tsdbMaxKeep = TSDB_MAX_KEEP;
  if (pOptions->precision == TSDB_TIME_PRECISION_NANO) {
4270
    tsdbMaxKeep = TSDB_MAX_KEEP_NS;
4271 4272
  }

wmmhello's avatar
wmmhello 已提交
4273
  if (pOptions->keep[0] < TSDB_MIN_KEEP || pOptions->keep[1] < TSDB_MIN_KEEP || pOptions->keep[2] < TSDB_MIN_KEEP ||
4274
      pOptions->keep[0] > tsdbMaxKeep || pOptions->keep[1] > tsdbMaxKeep || pOptions->keep[2] > tsdbMaxKeep) {
4275
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
4276 4277
                                   "Invalid option keep: %" PRId64 ", %" PRId64 ", %" PRId64
                                   " valid range: [%dm, %" PRId64 "m]",
X
Xiaoyu Wang 已提交
4278
                                   pOptions->keep[0], pOptions->keep[1], pOptions->keep[2], TSDB_MIN_KEEP, tsdbMaxKeep);
wmmhello's avatar
wmmhello 已提交
4279 4280 4281
  }

  if (!((pOptions->keep[0] <= pOptions->keep[1]) && (pOptions->keep[1] <= pOptions->keep[2]))) {
4282 4283
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                   "Invalid keep value, should be keep0 <= keep1 <= keep2");
wmmhello's avatar
wmmhello 已提交
4284 4285 4286
  }

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4287 4288
}

4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299
static int32_t checkDbCacheModelOption(STranslateContext* pCxt, SDatabaseOptions* pOptions) {
  if ('\0' != pOptions->cacheModelStr[0]) {
    if (0 == strcasecmp(pOptions->cacheModelStr, TSDB_CACHE_MODEL_NONE_STR)) {
      pOptions->cacheModel = TSDB_CACHE_MODEL_NONE;
    } else if (0 == strcasecmp(pOptions->cacheModelStr, TSDB_CACHE_MODEL_LAST_ROW_STR)) {
      pOptions->cacheModel = TSDB_CACHE_MODEL_LAST_ROW;
    } else if (0 == strcasecmp(pOptions->cacheModelStr, TSDB_CACHE_MODEL_LAST_VALUE_STR)) {
      pOptions->cacheModel = TSDB_CACHE_MODEL_LAST_VALUE;
    } else if (0 == strcasecmp(pOptions->cacheModelStr, TSDB_CACHE_MODEL_BOTH_STR)) {
      pOptions->cacheModel = TSDB_CACHE_MODEL_BOTH;
    } else {
4300 4301
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid option cacheModel: %s",
                                     pOptions->cacheModelStr);
4302 4303 4304 4305 4306
    }
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
4307
static int32_t checkDbPrecisionOption(STranslateContext* pCxt, SDatabaseOptions* pOptions) {
wmmhello's avatar
wmmhello 已提交
4308
  if ('\0' != pOptions->precisionStr[0]) {
4309
    if (0 == strcasecmp(pOptions->precisionStr, TSDB_TIME_PRECISION_MILLI_STR)) {
wmmhello's avatar
wmmhello 已提交
4310
      pOptions->precision = TSDB_TIME_PRECISION_MILLI;
4311
    } else if (0 == strcasecmp(pOptions->precisionStr, TSDB_TIME_PRECISION_MICRO_STR)) {
wmmhello's avatar
wmmhello 已提交
4312
      pOptions->precision = TSDB_TIME_PRECISION_MICRO;
4313
    } else if (0 == strcasecmp(pOptions->precisionStr, TSDB_TIME_PRECISION_NANO_STR)) {
wmmhello's avatar
wmmhello 已提交
4314 4315
      pOptions->precision = TSDB_TIME_PRECISION_NANO;
    } else {
4316 4317
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid option precision: %s",
                                     pOptions->precisionStr);
wmmhello's avatar
wmmhello 已提交
4318 4319 4320
    }
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4321
}
X
Xiaoyu Wang 已提交
4322

4323 4324 4325 4326 4327 4328 4329
static int32_t checkDbStrictOption(STranslateContext* pCxt, SDatabaseOptions* pOptions) {
  if ('\0' != pOptions->strictStr[0]) {
    if (0 == strcasecmp(pOptions->strictStr, TSDB_DB_STRICT_OFF_STR)) {
      pOptions->strict = TSDB_DB_STRICT_OFF;
    } else if (0 == strcasecmp(pOptions->strictStr, TSDB_DB_STRICT_ON_STR)) {
      pOptions->strict = TSDB_DB_STRICT_ON;
    } else {
4330 4331
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid option strict: %s",
                                     pOptions->strictStr);
4332 4333 4334 4335 4336
    }
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
4337
static int32_t checkDbEnumOption(STranslateContext* pCxt, const char* pName, int32_t val, int32_t v1, int32_t v2) {
wmmhello's avatar
wmmhello 已提交
4338
  if (val >= 0 && val != v1 && val != v2) {
4339
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
X
Xiaoyu Wang 已提交
4340
                                   "Invalid option %s: %d, only %d, %d allowed", pName, val, v1, v2);
wmmhello's avatar
wmmhello 已提交
4341 4342
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4343 4344 4345
}

static int32_t checkDbRetentionsOption(STranslateContext* pCxt, SNodeList* pRetentions) {
wmmhello's avatar
wmmhello 已提交
4346 4347 4348
  if (NULL == pRetentions) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
4349

wmmhello's avatar
wmmhello 已提交
4350
  if (LIST_LENGTH(pRetentions) > 3) {
4351
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid option retentions");
wmmhello's avatar
wmmhello 已提交
4352
  }
X
Xiaoyu Wang 已提交
4353

X
Xiaoyu Wang 已提交
4354 4355 4356
  SValueNode* pPrevFreq = NULL;
  SValueNode* pPrevKeep = NULL;
  SNode*      pRetention = NULL;
wmmhello's avatar
wmmhello 已提交
4357 4358 4359 4360 4361 4362 4363 4364
  FOREACH(pRetention, pRetentions) {
    SNode* pNode = NULL;
    FOREACH(pNode, ((SNodeListNode*)pRetention)->pNodeList) {
      SValueNode* pVal = (SValueNode*)pNode;
      if (DEAL_RES_ERROR == translateValue(pCxt, pVal)) {
        return pCxt->errCode;
      }
    }
X
Xiaoyu Wang 已提交
4365 4366 4367 4368 4369 4370

    SValueNode* pFreq = (SValueNode*)nodesListGetNode(((SNodeListNode*)pRetention)->pNodeList, 0);
    SValueNode* pKeep = (SValueNode*)nodesListGetNode(((SNodeListNode*)pRetention)->pNodeList, 1);
    if (pFreq->datum.i <= 0 || 'n' == pFreq->unit || 'y' == pFreq->unit || pFreq->datum.i >= pKeep->datum.i ||
        (NULL != pPrevFreq && pPrevFreq->datum.i >= pFreq->datum.i) ||
        (NULL != pPrevKeep && pPrevKeep->datum.i > pKeep->datum.i)) {
4371
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION, "Invalid option retentions");
X
Xiaoyu Wang 已提交
4372 4373 4374
    }
    pPrevFreq = pFreq;
    pPrevKeep = pKeep;
wmmhello's avatar
wmmhello 已提交
4375
  }
X
Xiaoyu Wang 已提交
4376

wmmhello's avatar
wmmhello 已提交
4377
  return TSDB_CODE_SUCCESS;
4378 4379
}

4380 4381 4382
static int32_t checkDbTbPrefixSuffixOptions(STranslateContext* pCxt, int32_t tbPrefix, int32_t tbSuffix) {
  if (tbPrefix < TSDB_MIN_HASH_PREFIX || tbPrefix > TSDB_MAX_HASH_PREFIX) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
4383
                                   "Invalid option table_prefix: %d valid range: [%d, %d]", tbPrefix,
4384 4385 4386 4387 4388
                                   TSDB_MIN_HASH_PREFIX, TSDB_MAX_HASH_PREFIX);
  }

  if (tbSuffix < TSDB_MIN_HASH_SUFFIX || tbSuffix > TSDB_MAX_HASH_SUFFIX) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
4389
                                   "Invalid option table_suffix: %d valid range: [%d, %d]", tbSuffix,
4390 4391 4392 4393 4394 4395 4396 4397 4398 4399 4400 4401 4402 4403 4404 4405
                                   TSDB_MIN_HASH_SUFFIX, TSDB_MAX_HASH_SUFFIX);
  }

  if ((tbPrefix * tbSuffix) < 0) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                   "Invalid option table_prefix & table_suffix: mixed usage not allowed");
  }

  if ((tbPrefix + tbSuffix) >= (TSDB_TABLE_NAME_LEN - 1)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                   "Invalid option table_prefix & table_suffix: exceed max table name length");
  }

  return TSDB_CODE_SUCCESS;
}

4406
static int32_t checkOptionsDependency(STranslateContext* pCxt, const char* pDbName, SDatabaseOptions* pOptions) {
wmmhello's avatar
wmmhello 已提交
4407
  int32_t daysPerFile = pOptions->daysPerFile;
4408
  int64_t daysToKeep0 = pOptions->keep[0];
wmmhello's avatar
wmmhello 已提交
4409 4410 4411
  if (-1 == daysPerFile && -1 == daysToKeep0) {
    return TSDB_CODE_SUCCESS;
  } else if (-1 == daysPerFile || -1 == daysToKeep0) {
4412
    SDbCfgInfo dbCfg = {0};
wmmhello's avatar
wmmhello 已提交
4413 4414 4415 4416 4417 4418 4419 4420
    int32_t    code = getDBCfg(pCxt, pDbName, &dbCfg);
    if (TSDB_CODE_SUCCESS != code) {
      return code;
    }
    daysPerFile = (-1 == daysPerFile ? dbCfg.daysPerFile : daysPerFile);
    daysToKeep0 = (-1 == daysToKeep0 ? dbCfg.daysToKeep0 : daysToKeep0);
  }
  if (daysPerFile > daysToKeep0) {
X
Xiaoyu Wang 已提交
4421 4422
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DB_OPTION,
                                   "Invalid duration value, should be keep2 >= keep1 >= keep0 >= duration");
wmmhello's avatar
wmmhello 已提交
4423 4424
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4425 4426
}

4427
static int32_t checkDatabaseOptions(STranslateContext* pCxt, const char* pDbName, SDatabaseOptions* pOptions) {
wmmhello's avatar
wmmhello 已提交
4428
  int32_t code =
4429
      checkDbRangeOption(pCxt, "buffer", pOptions->buffer, TSDB_MIN_BUFFER_PER_VNODE, TSDB_MAX_BUFFER_PER_VNODE);
wmmhello's avatar
wmmhello 已提交
4430
  if (TSDB_CODE_SUCCESS == code) {
4431
    code = checkDbCacheModelOption(pCxt, pOptions);
wmmhello's avatar
wmmhello 已提交
4432
  }
4433
  if (TSDB_CODE_SUCCESS == code) {
4434 4435
    code =
        checkDbRangeOption(pCxt, "cacheSize", pOptions->cacheLastSize, TSDB_MIN_DB_CACHE_SIZE, TSDB_MAX_DB_CACHE_SIZE);
4436
  }
wmmhello's avatar
wmmhello 已提交
4437
  if (TSDB_CODE_SUCCESS == code) {
4438 4439
    code =
        checkDbRangeOption(pCxt, "compression", pOptions->compressionLevel, TSDB_MIN_COMP_LEVEL, TSDB_MAX_COMP_LEVEL);
wmmhello's avatar
wmmhello 已提交
4440 4441 4442 4443 4444
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbDaysOption(pCxt, pOptions);
  }
  if (TSDB_CODE_SUCCESS == code) {
4445
    code = checkDbRangeOption(pCxt, "fsyncPeriod", pOptions->fsyncPeriod, TSDB_MIN_FSYNC_PERIOD, TSDB_MAX_FSYNC_PERIOD);
wmmhello's avatar
wmmhello 已提交
4446 4447
  }
  if (TSDB_CODE_SUCCESS == code) {
4448 4449
    code = checkDbRangeOption(pCxt, "maxRowsPerBlock", pOptions->maxRowsPerBlock, TSDB_MIN_MAXROWS_FBLOCK,
                              TSDB_MAX_MAXROWS_FBLOCK);
wmmhello's avatar
wmmhello 已提交
4450 4451
  }
  if (TSDB_CODE_SUCCESS == code) {
4452 4453
    code = checkDbRangeOption(pCxt, "minRowsPerBlock", pOptions->minRowsPerBlock, TSDB_MIN_MINROWS_FBLOCK,
                              TSDB_MAX_MINROWS_FBLOCK);
wmmhello's avatar
wmmhello 已提交
4454 4455
  }
  if (TSDB_CODE_SUCCESS == code) {
4456 4457 4458
    code = checkDbPrecisionOption(pCxt, pOptions);
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
4459
    code = checkDbKeepOption(pCxt, pOptions);  // use precision
wmmhello's avatar
wmmhello 已提交
4460 4461
  }
  if (TSDB_CODE_SUCCESS == code) {
4462
    code = checkDbRangeOption(pCxt, "pages", pOptions->pages, TSDB_MIN_PAGES_PER_VNODE, TSDB_MAX_PAGES_PER_VNODE);
wmmhello's avatar
wmmhello 已提交
4463 4464
  }
  if (TSDB_CODE_SUCCESS == code) {
4465 4466
    code = checkDbRangeOption(pCxt, "pagesize", pOptions->pagesize, TSDB_MIN_PAGESIZE_PER_VNODE,
                              TSDB_MAX_PAGESIZE_PER_VNODE);
wmmhello's avatar
wmmhello 已提交
4467
  }
4468 4469 4470 4471
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbRangeOption(pCxt, "tsdbPagesize", pOptions->tsdbPageSize, TSDB_MIN_TSDB_PAGESIZE,
                              TSDB_MAX_TSDB_PAGESIZE);
  }
wmmhello's avatar
wmmhello 已提交
4472 4473 4474 4475
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbEnumOption(pCxt, "replications", pOptions->replica, TSDB_MIN_DB_REPLICA, TSDB_MAX_DB_REPLICA);
  }
  if (TSDB_CODE_SUCCESS == code) {
4476
    code = checkDbStrictOption(pCxt, pOptions);
wmmhello's avatar
wmmhello 已提交
4477 4478 4479 4480 4481
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbEnumOption(pCxt, "walLevel", pOptions->walLevel, TSDB_MIN_WAL_LEVEL, TSDB_MAX_WAL_LEVEL);
  }
  if (TSDB_CODE_SUCCESS == code) {
4482
    code = checkDbRangeOption(pCxt, "vgroups", pOptions->numOfVgroups, TSDB_MIN_VNODES_PER_DB, TSDB_MAX_VNODES_PER_DB);
wmmhello's avatar
wmmhello 已提交
4483 4484 4485 4486 4487 4488 4489 4490
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbEnumOption(pCxt, "singleStable", pOptions->singleStable, TSDB_DB_SINGLE_STABLE_ON,
                             TSDB_DB_SINGLE_STABLE_OFF);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbRetentionsOption(pCxt, pOptions->pRetentions);
  }
X
Xiaoyu Wang 已提交
4491 4492 4493
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbEnumOption(pCxt, "schemaless", pOptions->schemaless, TSDB_DB_SCHEMALESS_ON, TSDB_DB_SCHEMALESS_OFF);
  }
X
Xiaoyu Wang 已提交
4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbRangeOption(pCxt, "walRetentionPeriod", pOptions->walRetentionPeriod,
                              TSDB_DB_MIN_WAL_RETENTION_PERIOD, INT32_MAX);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbRangeOption(pCxt, "walRetentionSize", pOptions->walRetentionSize, TSDB_DB_MIN_WAL_RETENTION_SIZE,
                              INT32_MAX);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkDbRangeOption(pCxt, "walRollPeriod", pOptions->walRollPeriod, TSDB_DB_MIN_WAL_ROLL_PERIOD, INT32_MAX);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code =
        checkDbRangeOption(pCxt, "walSegmentSize", pOptions->walSegmentSize, TSDB_DB_MIN_WAL_SEGMENT_SIZE, INT32_MAX);
  }
4509
  if (TSDB_CODE_SUCCESS == code) {
H
Hongze Cheng 已提交
4510
    code = checkDbRangeOption(pCxt, "sstTrigger", pOptions->sstTrigger, TSDB_MIN_STT_TRIGGER, TSDB_MAX_STT_TRIGGER);
4511
  }
4512
  if (TSDB_CODE_SUCCESS == code) {
4513
    code = checkDbTbPrefixSuffixOptions(pCxt, pOptions->tablePrefix, pOptions->tableSuffix);
4514
  }
wmmhello's avatar
wmmhello 已提交
4515 4516 4517 4518
  if (TSDB_CODE_SUCCESS == code) {
    code = checkOptionsDependency(pCxt, pDbName, pOptions);
  }
  return code;
X
Xiaoyu Wang 已提交
4519 4520 4521
}

static int32_t checkCreateDatabase(STranslateContext* pCxt, SCreateDatabaseStmt* pStmt) {
4522 4523 4524 4525
  if (NULL != strchr(pStmt->dbName, '.')) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_IDENTIFIER_NAME,
                                   "The database name cannot contain '.'");
  }
wmmhello's avatar
wmmhello 已提交
4526
  return checkDatabaseOptions(pCxt, pStmt->dbName, pStmt->pOptions);
X
Xiaoyu Wang 已提交
4527 4528
}

H
Hongze Cheng 已提交
4529
typedef int32_t (*FSerializeFunc)(void* pBuf, int32_t bufLen, void* pReq);
4530 4531

static int32_t buildCmdMsg(STranslateContext* pCxt, int16_t msgType, FSerializeFunc func, void* pReq) {
wmmhello's avatar
wmmhello 已提交
4532 4533 4534 4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545
  pCxt->pCmdMsg = taosMemoryMalloc(sizeof(SCmdMsgInfo));
  if (NULL == pCxt->pCmdMsg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pCxt->pCmdMsg->epSet = pCxt->pParseCxt->mgmtEpSet;
  pCxt->pCmdMsg->msgType = msgType;
  pCxt->pCmdMsg->msgLen = func(NULL, 0, pReq);
  pCxt->pCmdMsg->pMsg = taosMemoryMalloc(pCxt->pCmdMsg->msgLen);
  if (NULL == pCxt->pCmdMsg->pMsg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  func(pCxt->pCmdMsg->pMsg, pCxt->pCmdMsg->msgLen, pReq);

  return TSDB_CODE_SUCCESS;
4546 4547
}

4548
static int32_t translateCreateDatabase(STranslateContext* pCxt, SCreateDatabaseStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
4549
  SCreateDbReq createReq = {0};
4550
  int32_t      code = checkCreateDatabase(pCxt, pStmt);
wmmhello's avatar
wmmhello 已提交
4551 4552 4553 4554 4555 4556
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateDbReq(pCxt, pStmt, &createReq);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_DB, (FSerializeFunc)tSerializeSCreateDbReq, &createReq);
  }
4557
  tFreeSCreateDbReq(&createReq);
wmmhello's avatar
wmmhello 已提交
4558
  return code;
4559 4560
}

4561
static int32_t translateDropDatabase(STranslateContext* pCxt, SDropDatabaseStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
4562 4563 4564 4565 4566
  SDropDbReq dropReq = {0};
  SName      name = {0};
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, dropReq.db);
  dropReq.ignoreNotExists = pStmt->ignoreNotExists;
4567

wmmhello's avatar
wmmhello 已提交
4568
  return buildCmdMsg(pCxt, TDMT_MND_DROP_DB, (FSerializeFunc)tSerializeSDropDbReq, &dropReq);
4569 4570
}

4571
static void buildAlterDbReq(STranslateContext* pCxt, SAlterDatabaseStmt* pStmt, SAlterDbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
4572 4573 4574 4575 4576 4577 4578 4579 4580 4581
  SName name = {0};
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, pReq->db);
  pReq->buffer = pStmt->pOptions->buffer;
  pReq->pageSize = -1;
  pReq->pages = pStmt->pOptions->pages;
  pReq->daysPerFile = -1;
  pReq->daysToKeep0 = pStmt->pOptions->keep[0];
  pReq->daysToKeep1 = pStmt->pOptions->keep[1];
  pReq->daysToKeep2 = pStmt->pOptions->keep[2];
4582
  pReq->walFsyncPeriod = pStmt->pOptions->fsyncPeriod;
wmmhello's avatar
wmmhello 已提交
4583 4584
  pReq->walLevel = pStmt->pOptions->walLevel;
  pReq->strict = pStmt->pOptions->strict;
4585
  pReq->cacheLast = pStmt->pOptions->cacheModel;
4586
  pReq->cacheLastSize = pStmt->pOptions->cacheLastSize;
wmmhello's avatar
wmmhello 已提交
4587
  pReq->replications = pStmt->pOptions->replica;
4588
  pReq->sstTrigger = pStmt->pOptions->sstTrigger;
4589
  pReq->minRows = pStmt->pOptions->minRowsPerBlock;
4590 4591
  pReq->walRetentionPeriod = pStmt->pOptions->walRetentionPeriod;
  pReq->walRetentionSize = pStmt->pOptions->walRetentionSize;
wmmhello's avatar
wmmhello 已提交
4592
  return;
4593 4594 4595
}

static int32_t translateAlterDatabase(STranslateContext* pCxt, SAlterDatabaseStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
4596 4597 4598 4599
  int32_t code = checkDatabaseOptions(pCxt, pStmt->dbName, pStmt->pOptions);
  if (TSDB_CODE_SUCCESS != code) {
    return code;
  }
X
Xiaoyu Wang 已提交
4600

wmmhello's avatar
wmmhello 已提交
4601 4602
  SAlterDbReq alterReq = {0};
  buildAlterDbReq(pCxt, pStmt, &alterReq);
4603

wmmhello's avatar
wmmhello 已提交
4604
  return buildCmdMsg(pCxt, TDMT_MND_ALTER_DB, (FSerializeFunc)tSerializeSAlterDbReq, &alterReq);
4605 4606
}

X
Xiaoyu Wang 已提交
4607
static int32_t translateTrimDatabase(STranslateContext* pCxt, STrimDatabaseStmt* pStmt) {
4608
  STrimDbReq req = {.maxSpeed = pStmt->maxSpeed};
X
Xiaoyu Wang 已提交
4609 4610 4611 4612 4613 4614
  SName      name = {0};
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, req.db);
  return buildCmdMsg(pCxt, TDMT_MND_TRIM_DB, (FSerializeFunc)tSerializeSTrimDbReq, &req);
}

X
Xiaoyu Wang 已提交
4615
static int32_t columnDefNodeToField(SNodeList* pList, SArray** pArray) {
wmmhello's avatar
wmmhello 已提交
4616 4617 4618 4619 4620 4621 4622 4623 4624 4625 4626 4627
  *pArray = taosArrayInit(LIST_LENGTH(pList), sizeof(SField));
  SNode* pNode;
  FOREACH(pNode, pList) {
    SColumnDefNode* pCol = (SColumnDefNode*)pNode;
    SField          field = {.type = pCol->dataType.type, .bytes = calcTypeBytes(pCol->dataType)};
    strcpy(field.name, pCol->colName);
    if (pCol->sma) {
      field.flags |= COL_SMA_ON;
    }
    taosArrayPush(*pArray, &field);
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4628 4629
}

4630
static SColumnDefNode* findColDef(SNodeList* pCols, const SColumnNode* pCol) {
wmmhello's avatar
wmmhello 已提交
4631 4632 4633 4634 4635 4636 4637
  SNode* pColDef = NULL;
  FOREACH(pColDef, pCols) {
    if (0 == strcmp(pCol->colName, ((SColumnDefNode*)pColDef)->colName)) {
      return (SColumnDefNode*)pColDef;
    }
  }
  return NULL;
X
Xiaoyu Wang 已提交
4638 4639
}

X
Xiaoyu Wang 已提交
4640
static int32_t checkTableSmaOption(STranslateContext* pCxt, SCreateTableStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
4641 4642 4643 4644 4645 4646 4647 4648 4649 4650 4651 4652 4653 4654
  if (NULL != pStmt->pOptions->pSma) {
    SNode* pNode = NULL;
    FOREACH(pNode, pStmt->pCols) { ((SColumnDefNode*)pNode)->sma = false; }
    FOREACH(pNode, pStmt->pOptions->pSma) {
      SColumnNode*    pSmaCol = (SColumnNode*)pNode;
      SColumnDefNode* pColDef = findColDef(pStmt->pCols, pSmaCol);
      if (NULL == pColDef) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, pSmaCol->colName);
      }
      pSmaCol->node.resType = pColDef->dataType;
      pColDef->sma = true;
    }
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4655 4656
}

X
Xiaoyu Wang 已提交
4657 4658 4659 4660 4661 4662 4663 4664 4665 4666 4667
static bool validRollupFunc(const char* pFunc) {
  static const char*   rollupFuncs[] = {"avg", "sum", "min", "max", "last", "first"};
  static const int32_t numOfRollupFuncs = (sizeof(rollupFuncs) / sizeof(char*));
  for (int i = 0; i < numOfRollupFuncs; ++i) {
    if (0 == strcmp(rollupFuncs[i], pFunc)) {
      return true;
    }
  }
  return false;
}

4668 4669
static int32_t checkTableRollupOption(STranslateContext* pCxt, SNodeList* pFuncs, bool createStable,
                                      SDbCfgInfo* pDbCfg) {
wmmhello's avatar
wmmhello 已提交
4670
  if (NULL == pFuncs) {
4671 4672
    if (NULL != pDbCfg->pRetensions) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
4673
                                     "To create a super table in databases configured with the 'RETENTIONS' option, "
4674 4675
                                     "the 'ROLLUP' option must be present");
    }
wmmhello's avatar
wmmhello 已提交
4676 4677
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
4678

4679 4680 4681 4682 4683 4684
  if (!createStable || NULL == pDbCfg->pRetensions) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
                                   "Invalid option rollup: Only supported for create super table in databases "
                                   "configured with the 'RETENTIONS' option");
  }
  if (1 != LIST_LENGTH(pFuncs)) {
4685
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
4686 4687 4688 4689
                                   "Invalid option rollup: only one function is allowed");
  }
  const char* pFunc = ((SFunctionNode*)nodesListGetNode(pFuncs, 0))->functionName;
  if (!validRollupFunc(pFunc)) {
4690
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
4691
                                   "Invalid option rollup: %s function is not supported", pFunc);
wmmhello's avatar
wmmhello 已提交
4692 4693
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
4694 4695
}

4696
static int32_t checkTableTagsSchema(STranslateContext* pCxt, SHashObj* pHash, SNodeList* pTags) {
wmmhello's avatar
wmmhello 已提交
4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708 4709 4710 4711 4712 4713 4714 4715 4716
  int32_t ntags = LIST_LENGTH(pTags);
  if (0 == ntags) {
    return TSDB_CODE_SUCCESS;
  } else if (ntags > TSDB_MAX_TAGS) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_NUM);
  }

  int32_t code = TSDB_CODE_SUCCESS;
  int32_t tagsSize = 0;
  SNode*  pNode = NULL;
  FOREACH(pNode, pTags) {
    SColumnDefNode* pTag = (SColumnDefNode*)pNode;
    int32_t         len = strlen(pTag->colName);
    if (NULL != taosHashGet(pHash, pTag->colName, len)) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_DUPLICATED_COLUMN);
    }
    if (TSDB_CODE_SUCCESS == code && pTag->dataType.type == TSDB_DATA_TYPE_JSON && ntags > 1) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_ONLY_ONE_JSON_TAG);
    }
    if (TSDB_CODE_SUCCESS == code) {
K
kailixu 已提交
4717
      if ((TSDB_DATA_TYPE_VARCHAR == pTag->dataType.type && calcTypeBytes(pTag->dataType) > TSDB_MAX_TAGS_LEN) ||
D
Dingle Zhang 已提交
4718 4719
          (TSDB_DATA_TYPE_NCHAR == pTag->dataType.type && calcTypeBytes(pTag->dataType) > TSDB_MAX_TAGS_LEN) ||
          (TSDB_DATA_TYPE_GEOMETRY == pTag->dataType.type && calcTypeBytes(pTag->dataType) > TSDB_MAX_TAGS_LEN)) {
4720
        code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
wmmhello's avatar
wmmhello 已提交
4721 4722 4723 4724 4725 4726
      }
    }
    if (TSDB_CODE_SUCCESS == code) {
      code = taosHashPut(pHash, pTag->colName, len, &pTag, POINTER_BYTES);
    }
    if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
4727
      tagsSize += calcTypeBytes(pTag->dataType);
wmmhello's avatar
wmmhello 已提交
4728 4729 4730 4731 4732 4733 4734 4735 4736 4737
    } else {
      break;
    }
  }

  if (TSDB_CODE_SUCCESS == code && tagsSize > TSDB_MAX_TAGS_LEN) {
    code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_LENGTH, TSDB_MAX_TAGS_LEN);
  }

  return code;
4738 4739
}

4740
static int32_t checkTableColsSchema(STranslateContext* pCxt, SHashObj* pHash, int32_t ntags, SNodeList* pCols) {
wmmhello's avatar
wmmhello 已提交
4741 4742 4743
  int32_t ncols = LIST_LENGTH(pCols);
  if (ncols < TSDB_MIN_COLUMNS) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM);
4744
  } else if (ncols + ntags > TSDB_MAX_COLUMNS) {
wmmhello's avatar
wmmhello 已提交
4745 4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764 4765 4766 4767 4768
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TOO_MANY_COLUMNS);
  }

  int32_t code = TSDB_CODE_SUCCESS;

  bool    first = true;
  int32_t rowSize = 0;
  SNode*  pNode = NULL;
  FOREACH(pNode, pCols) {
    SColumnDefNode* pCol = (SColumnDefNode*)pNode;
    if (first) {
      first = false;
      if (TSDB_DATA_TYPE_TIMESTAMP != pCol->dataType.type) {
        code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_FIRST_COLUMN);
      }
    }
    if (TSDB_CODE_SUCCESS == code && pCol->dataType.type == TSDB_DATA_TYPE_JSON) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COL_JSON);
    }
    int32_t len = strlen(pCol->colName);
    if (TSDB_CODE_SUCCESS == code && NULL != taosHashGet(pHash, pCol->colName, len)) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_DUPLICATED_COLUMN);
    }
    if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
4769
      if ((TSDB_DATA_TYPE_VARCHAR == pCol->dataType.type && calcTypeBytes(pCol->dataType) > TSDB_MAX_BINARY_LEN) ||
D
Dingle Zhang 已提交
4770 4771
          (TSDB_DATA_TYPE_NCHAR == pCol->dataType.type && calcTypeBytes(pCol->dataType) > TSDB_MAX_NCHAR_LEN) ||
          (TSDB_DATA_TYPE_GEOMETRY == pCol->dataType.type && calcTypeBytes(pCol->dataType) > TSDB_MAX_GEOMETRY_LEN)) {
wmmhello's avatar
wmmhello 已提交
4772 4773 4774 4775 4776 4777 4778
        code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
      }
    }
    if (TSDB_CODE_SUCCESS == code) {
      code = taosHashPut(pHash, pCol->colName, len, &pCol, POINTER_BYTES);
    }
    if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
4779
      rowSize += calcTypeBytes(pCol->dataType);
wmmhello's avatar
wmmhello 已提交
4780 4781 4782 4783 4784 4785 4786 4787 4788 4789
    } else {
      break;
    }
  }

  if (TSDB_CODE_SUCCESS == code && rowSize > TSDB_MAX_BYTES_PER_ROW) {
    code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ROW_LENGTH, TSDB_MAX_BYTES_PER_ROW);
  }

  return code;
4790 4791 4792
}

static int32_t checkTableSchema(STranslateContext* pCxt, SCreateTableStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
4793 4794 4795 4796 4797
  SHashObj* pHash = taosHashInit(LIST_LENGTH(pStmt->pTags) + LIST_LENGTH(pStmt->pCols),
                                 taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  if (NULL == pHash) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
4798

wmmhello's avatar
wmmhello 已提交
4799 4800
  int32_t code = checkTableTagsSchema(pCxt, pHash, pStmt->pTags);
  if (TSDB_CODE_SUCCESS == code) {
4801
    code = checkTableColsSchema(pCxt, pHash, LIST_LENGTH(pStmt->pTags), pStmt->pCols);
wmmhello's avatar
wmmhello 已提交
4802
  }
4803

wmmhello's avatar
wmmhello 已提交
4804 4805
  taosHashCleanup(pHash);
  return code;
4806 4807
}

X
Xiaoyu Wang 已提交
4808 4809 4810 4811 4812
static int32_t getTableDelayOrWatermarkOption(STranslateContext* pCxt, const char* pName, int64_t minVal,
                                              int64_t maxVal, SValueNode* pVal, int64_t* pMaxDelay) {
  int32_t code = (DEAL_RES_ERROR == translateValue(pCxt, pVal) ? pCxt->errCode : TSDB_CODE_SUCCESS);
  if (TSDB_CODE_SUCCESS == code && TIME_UNIT_MILLISECOND != pVal->unit && TIME_UNIT_SECOND != pVal->unit &&
      TIME_UNIT_MINUTE != pVal->unit) {
4813 4814 4815
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
                                   "Invalid option %s unit: %c, only %c, %c, %c allowed", pName, pVal->unit,
                                   TIME_UNIT_MILLISECOND, TIME_UNIT_SECOND, TIME_UNIT_MINUTE);
X
Xiaoyu Wang 已提交
4816 4817
  }
  if (TSDB_CODE_SUCCESS == code) {
4818
    code = checkTableRangeOption(pCxt, pName, pVal->datum.i, minVal, maxVal);
X
Xiaoyu Wang 已提交
4819 4820 4821 4822 4823 4824 4825 4826 4827 4828 4829 4830
  }
  if (TSDB_CODE_SUCCESS == code) {
    *pMaxDelay = pVal->datum.i;
  }
  return code;
}

static int32_t getTableMaxDelayOption(STranslateContext* pCxt, SValueNode* pVal, int64_t* pMaxDelay) {
  return getTableDelayOrWatermarkOption(pCxt, "maxDelay", TSDB_MIN_ROLLUP_MAX_DELAY, TSDB_MAX_ROLLUP_MAX_DELAY, pVal,
                                        pMaxDelay);
}

4831 4832
static int32_t checkTableMaxDelayOption(STranslateContext* pCxt, STableOptions* pOptions, bool createStable,
                                        SDbCfgInfo* pDbCfg) {
X
Xiaoyu Wang 已提交
4833 4834 4835 4836
  if (NULL == pOptions->pMaxDelay) {
    return TSDB_CODE_SUCCESS;
  }

4837 4838 4839 4840 4841 4842
  if (!createStable || NULL == pDbCfg->pRetensions) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
                                   "Invalid option maxdelay: Only supported for create super table in databases "
                                   "configured with the 'RETENTIONS' option");
  }

X
Xiaoyu Wang 已提交
4843
  if (LIST_LENGTH(pOptions->pMaxDelay) > 2) {
4844
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION, "Invalid option maxdelay");
X
Xiaoyu Wang 已提交
4845 4846 4847 4848 4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859 4860
  }

  int32_t code =
      getTableMaxDelayOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pMaxDelay, 0), &pOptions->maxDelay1);
  if (TSDB_CODE_SUCCESS == code && 2 == LIST_LENGTH(pOptions->pMaxDelay)) {
    code = getTableMaxDelayOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pMaxDelay, 1), &pOptions->maxDelay2);
  }

  return code;
}

static int32_t getTableWatermarkOption(STranslateContext* pCxt, SValueNode* pVal, int64_t* pMaxDelay) {
  return getTableDelayOrWatermarkOption(pCxt, "watermark", TSDB_MIN_ROLLUP_WATERMARK, TSDB_MAX_ROLLUP_WATERMARK, pVal,
                                        pMaxDelay);
}

4861 4862
static int32_t checkTableWatermarkOption(STranslateContext* pCxt, STableOptions* pOptions, bool createStable,
                                         SDbCfgInfo* pDbCfg) {
X
Xiaoyu Wang 已提交
4863 4864 4865 4866
  if (NULL == pOptions->pWatermark) {
    return TSDB_CODE_SUCCESS;
  }

4867 4868 4869 4870 4871 4872
  if (!createStable || NULL == pDbCfg->pRetensions) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
                                   "Invalid option watermark: Only supported for create super table in databases "
                                   "configured with the 'RETENTIONS' option");
  }

X
Xiaoyu Wang 已提交
4873
  if (LIST_LENGTH(pOptions->pWatermark) > 2) {
4874
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION, "Invalid option watermark");
X
Xiaoyu Wang 已提交
4875
  }
X
Xiaoyu Wang 已提交
4876 4877 4878 4879 4880 4881 4882 4883

  int32_t code =
      getTableWatermarkOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pWatermark, 0), &pOptions->watermark1);
  if (TSDB_CODE_SUCCESS == code && 2 == LIST_LENGTH(pOptions->pWatermark)) {
    code = getTableWatermarkOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pWatermark, 1), &pOptions->watermark2);
  }

  return code;
X
Xiaoyu Wang 已提交
4884 4885
}

4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916
static int32_t getTableDeleteMarkOption(STranslateContext* pCxt, SValueNode* pVal, int64_t* pMaxDelay) {
  return getTableDelayOrWatermarkOption(pCxt, "delete_mark", TSDB_MIN_ROLLUP_DELETE_MARK, TSDB_MAX_ROLLUP_DELETE_MARK,
                                        pVal, pMaxDelay);
}

static int32_t checkTableDeleteMarkOption(STranslateContext* pCxt, STableOptions* pOptions, bool createStable,
                                          SDbCfgInfo* pDbCfg) {
  if (NULL == pOptions->pDeleteMark) {
    return TSDB_CODE_SUCCESS;
  }

  if (!createStable || NULL == pDbCfg->pRetensions) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
                                   "Invalid option delete_mark: Only supported for create super table in databases "
                                   "configured with the 'RETENTIONS' option");
  }

  if (LIST_LENGTH(pOptions->pDeleteMark) > 2) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION, "Invalid option delete_mark");
  }

  int32_t code =
      getTableDeleteMarkOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pDeleteMark, 0), &pOptions->deleteMark1);
  if (TSDB_CODE_SUCCESS == code && 2 == LIST_LENGTH(pOptions->pDeleteMark)) {
    code =
        getTableDeleteMarkOption(pCxt, (SValueNode*)nodesListGetNode(pOptions->pDeleteMark, 1), &pOptions->deleteMark2);
  }

  return code;
}

4917
static int32_t checkCreateTable(STranslateContext* pCxt, SCreateTableStmt* pStmt, bool createStable) {
X
Xiaoyu Wang 已提交
4918 4919 4920 4921 4922
  if (NULL != strchr(pStmt->tableName, '.')) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_IDENTIFIER_NAME,
                                   "The table name cannot contain '.'");
  }

4923
  SDbCfgInfo dbCfg = {0};
4924 4925 4926 4927 4928
  int32_t    code = getDBCfg(pCxt, pStmt->dbName, &dbCfg);
  if (TSDB_CODE_SUCCESS == code && !createStable && NULL != dbCfg.pRetensions) {
    code = generateSyntaxErrMsgExt(
        &pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TABLE_OPTION,
        "Only super table creation is supported in databases configured with the 'RETENTIONS' option");
4929 4930 4931 4932
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkTableMaxDelayOption(pCxt, pStmt->pOptions, createStable, &dbCfg);
  }
X
Xiaoyu Wang 已提交
4933
  if (TSDB_CODE_SUCCESS == code) {
4934
    code = checkTableWatermarkOption(pCxt, pStmt->pOptions, createStable, &dbCfg);
X
Xiaoyu Wang 已提交
4935
  }
4936 4937 4938
  if (TSDB_CODE_SUCCESS == code) {
    code = checkTableDeleteMarkOption(pCxt, pStmt->pOptions, createStable, &dbCfg);
  }
wmmhello's avatar
wmmhello 已提交
4939
  if (TSDB_CODE_SUCCESS == code) {
4940
    code = checkTableRollupOption(pCxt, pStmt->pOptions->pRollupFuncs, createStable, &dbCfg);
wmmhello's avatar
wmmhello 已提交
4941 4942 4943 4944 4945 4946 4947 4948
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkTableSmaOption(pCxt, pStmt);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkTableSchema(pCxt, pStmt);
  }
  return code;
X
Xiaoyu Wang 已提交
4949 4950
}

4951
static void toSchema(const SColumnDefNode* pCol, col_id_t colId, SSchema* pSchema) {
wmmhello's avatar
wmmhello 已提交
4952 4953 4954 4955 4956 4957 4958 4959 4960
  int8_t flags = 0;
  if (pCol->sma) {
    flags |= COL_SMA_ON;
  }
  pSchema->colId = colId;
  pSchema->type = pCol->dataType.type;
  pSchema->bytes = calcTypeBytes(pCol->dataType);
  pSchema->flags = flags;
  strcpy(pSchema->name, pCol->colName);
4961 4962 4963
}

typedef struct SSampleAstInfo {
wmmhello's avatar
wmmhello 已提交
4964 4965 4966 4967 4968 4969
  const char* pDbName;
  const char* pTableName;
  SNodeList*  pFuncs;
  SNode*      pInterval;
  SNode*      pOffset;
  SNode*      pSliding;
4970
  SNodeList*  pPartitionByList;
wmmhello's avatar
wmmhello 已提交
4971
  STableMeta* pRollupTableMeta;
4972
  bool        createSmaIndex;
4973 4974
} SSampleAstInfo;

4975
static int32_t buildTableForSampleAst(SSampleAstInfo* pInfo, SNode** pOutput) {
4976
  SRealTableNode* pTable = (SRealTableNode*)nodesMakeNode(QUERY_NODE_REAL_TABLE);
wmmhello's avatar
wmmhello 已提交
4977 4978 4979
  if (NULL == pTable) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
4980 4981
  snprintf(pTable->table.dbName, sizeof(pTable->table.dbName), "%s", pInfo->pDbName);
  snprintf(pTable->table.tableName, sizeof(pTable->table.tableName), "%s", pInfo->pTableName);
4982
  snprintf(pTable->table.tableAlias, sizeof(pTable->table.tableAlias), "%s", pInfo->pTableName);
wmmhello's avatar
wmmhello 已提交
4983
  TSWAP(pTable->pMeta, pInfo->pRollupTableMeta);
4984 4985 4986
  *pOutput = (SNode*)pTable;
  return TSDB_CODE_SUCCESS;
}
wmmhello's avatar
wmmhello 已提交
4987

4988
static int32_t addWstartToSampleProjects(SNodeList* pProjectionList) {
4989
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
4990
  if (NULL == pFunc) {
wmmhello's avatar
wmmhello 已提交
4991 4992
    return TSDB_CODE_OUT_OF_MEMORY;
  }
4993
  strcpy(pFunc->functionName, "_wstart");
4994 4995 4996 4997 4998 4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009 5010 5011 5012 5013 5014 5015 5016 5017
  return nodesListPushFront(pProjectionList, (SNode*)pFunc);
}

static int32_t addWendToSampleProjects(SNodeList* pProjectionList) {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  strcpy(pFunc->functionName, "_wend");
  return nodesListAppend(pProjectionList, (SNode*)pFunc);
}

static int32_t addWdurationToSampleProjects(SNodeList* pProjectionList) {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  strcpy(pFunc->functionName, "_wduration");
  return nodesListAppend(pProjectionList, (SNode*)pFunc);
}

static int32_t buildProjectsForSampleAst(SSampleAstInfo* pInfo, SNodeList** pList) {
  SNodeList* pProjectionList = pInfo->pFuncs;
  pInfo->pFuncs = NULL;
wmmhello's avatar
wmmhello 已提交
5018

5019 5020 5021 5022 5023 5024 5025
  int32_t code = addWstartToSampleProjects(pProjectionList);
  if (TSDB_CODE_SUCCESS == code && pInfo->createSmaIndex) {
    code = addWendToSampleProjects(pProjectionList);
    if (TSDB_CODE_SUCCESS == code) {
      code = addWdurationToSampleProjects(pProjectionList);
    }
  }
5026

5027 5028 5029 5030 5031 5032 5033 5034 5035 5036 5037
  if (TSDB_CODE_SUCCESS == code) {
    SNode* pProject = NULL;
    FOREACH(pProject, pProjectionList) { sprintf(((SExprNode*)pProject)->aliasName, "#%p", pProject); }
    *pList = pProjectionList;
  } else {
    nodesDestroyList(pProjectionList);
  }
  return code;
}

static int32_t buildIntervalForSampleAst(SSampleAstInfo* pInfo, SNode** pOutput) {
5038
  SIntervalWindowNode* pInterval = (SIntervalWindowNode*)nodesMakeNode(QUERY_NODE_INTERVAL_WINDOW);
wmmhello's avatar
wmmhello 已提交
5039 5040 5041 5042 5043 5044 5045 5046
  if (NULL == pInterval) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  TSWAP(pInterval->pInterval, pInfo->pInterval);
  TSWAP(pInterval->pOffset, pInfo->pOffset);
  TSWAP(pInterval->pSliding, pInfo->pSliding);
  pInterval->pCol = nodesMakeNode(QUERY_NODE_COLUMN);
  if (NULL == pInterval->pCol) {
5047
    nodesDestroyNode((SNode*)pInterval);
wmmhello's avatar
wmmhello 已提交
5048 5049 5050
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  ((SColumnNode*)pInterval->pCol)->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
5051
  strcpy(((SColumnNode*)pInterval->pCol)->colName, ROWTS_PSEUDO_COLUMN_NAME);
5052 5053 5054
  *pOutput = (SNode*)pInterval;
  return TSDB_CODE_SUCCESS;
}
wmmhello's avatar
wmmhello 已提交
5055

5056 5057 5058 5059 5060 5061 5062 5063 5064 5065 5066 5067 5068 5069 5070 5071 5072 5073 5074 5075
static int32_t buildSampleAst(STranslateContext* pCxt, SSampleAstInfo* pInfo, char** pAst, int32_t* pLen, char** pExpr,
                              int32_t* pExprLen) {
  SSelectStmt* pSelect = (SSelectStmt*)nodesMakeNode(QUERY_NODE_SELECT_STMT);
  if (NULL == pSelect) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  sprintf(pSelect->stmtName, "%p", pSelect);

  int32_t code = buildTableForSampleAst(pInfo, &pSelect->pFromTable);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildProjectsForSampleAst(pInfo, &pSelect->pProjectionList);
  }
  if (TSDB_CODE_SUCCESS == code) {
    TSWAP(pSelect->pPartitionByList, pInfo->pPartitionByList);
    code = buildIntervalForSampleAst(pInfo, &pSelect->pWindow);
  }
  if (TSDB_CODE_SUCCESS == code) {
    pCxt->createStream = true;
    code = translateQuery(pCxt, (SNode*)pSelect);
  }
wmmhello's avatar
wmmhello 已提交
5076
  if (TSDB_CODE_SUCCESS == code) {
5077
    code = nodesNodeToString((SNode*)pSelect, false, pAst, pLen);
wmmhello's avatar
wmmhello 已提交
5078
  }
X
Xiaoyu Wang 已提交
5079 5080 5081
  if (TSDB_CODE_SUCCESS == code && NULL != pExpr) {
    code = nodesListToString(pSelect->pProjectionList, false, pExpr, pExprLen);
  }
5082
  nodesDestroyNode((SNode*)pSelect);
wmmhello's avatar
wmmhello 已提交
5083
  return code;
5084 5085 5086
}

static void clearSampleAstInfo(SSampleAstInfo* pInfo) {
wmmhello's avatar
wmmhello 已提交
5087 5088 5089 5090
  nodesDestroyList(pInfo->pFuncs);
  nodesDestroyNode(pInfo->pInterval);
  nodesDestroyNode(pInfo->pOffset);
  nodesDestroyNode(pInfo->pSliding);
5091 5092 5093
}

static SNode* makeIntervalVal(SRetention* pRetension, int8_t precision) {
5094
  SValueNode* pVal = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
wmmhello's avatar
wmmhello 已提交
5095 5096 5097 5098 5099 5100 5101 5102
  if (NULL == pVal) {
    return NULL;
  }
  int64_t timeVal = convertTimeFromPrecisionToUnit(pRetension->freq, precision, pRetension->freqUnit);
  char    buf[20] = {0};
  int32_t len = snprintf(buf, sizeof(buf), "%" PRId64 "%c", timeVal, pRetension->freqUnit);
  pVal->literal = strndup(buf, len);
  if (NULL == pVal->literal) {
5103
    nodesDestroyNode((SNode*)pVal);
wmmhello's avatar
wmmhello 已提交
5104 5105 5106 5107 5108 5109 5110
    return NULL;
  }
  pVal->isDuration = true;
  pVal->node.resType.type = TSDB_DATA_TYPE_BIGINT;
  pVal->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BIGINT].bytes;
  pVal->node.resType.precision = precision;
  return (SNode*)pVal;
5111 5112 5113
}

static SNode* createColumnFromDef(SColumnDefNode* pDef) {
5114
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
wmmhello's avatar
wmmhello 已提交
5115 5116 5117 5118 5119
  if (NULL == pCol) {
    return NULL;
  }
  strcpy(pCol->colName, pDef->colName);
  return (SNode*)pCol;
5120 5121 5122
}

static SNode* createRollupFunc(SNode* pSrcFunc, SColumnDefNode* pColDef) {
5123
  SFunctionNode* pFunc = (SFunctionNode*)nodesCloneNode(pSrcFunc);
wmmhello's avatar
wmmhello 已提交
5124 5125 5126 5127
  if (NULL == pFunc) {
    return NULL;
  }
  if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(&pFunc->pParameterList, createColumnFromDef(pColDef))) {
5128
    nodesDestroyNode((SNode*)pFunc);
wmmhello's avatar
wmmhello 已提交
5129 5130 5131
    return NULL;
  }
  return (SNode*)pFunc;
5132 5133 5134
}

static SNodeList* createRollupFuncs(SCreateTableStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5135 5136 5137 5138 5139 5140 5141 5142 5143 5144 5145 5146 5147 5148 5149 5150 5151 5152 5153 5154 5155 5156
  SNodeList* pFuncs = nodesMakeList();
  if (NULL == pFuncs) {
    return NULL;
  }

  SNode* pFunc = NULL;
  FOREACH(pFunc, pStmt->pOptions->pRollupFuncs) {
    SNode* pCol = NULL;
    bool   primaryKey = true;
    FOREACH(pCol, pStmt->pCols) {
      if (primaryKey) {
        primaryKey = false;
        continue;
      }
      if (TSDB_CODE_SUCCESS != nodesListStrictAppend(pFuncs, createRollupFunc(pFunc, (SColumnDefNode*)pCol))) {
        nodesDestroyList(pFuncs);
        return NULL;
      }
    }
  }

  return pFuncs;
5157 5158 5159
}

static STableMeta* createRollupTableMeta(SCreateTableStmt* pStmt, int8_t precision) {
wmmhello's avatar
wmmhello 已提交
5160 5161 5162 5163 5164 5165 5166 5167 5168 5169 5170 5171 5172 5173 5174 5175 5176 5177 5178 5179 5180 5181 5182
  int32_t     numOfField = LIST_LENGTH(pStmt->pCols) + LIST_LENGTH(pStmt->pTags);
  STableMeta* pMeta = taosMemoryCalloc(1, sizeof(STableMeta) + numOfField * sizeof(SSchema));
  if (NULL == pMeta) {
    return NULL;
  }
  pMeta->tableType = TSDB_SUPER_TABLE;
  pMeta->tableInfo.numOfTags = LIST_LENGTH(pStmt->pTags);
  pMeta->tableInfo.precision = precision;
  pMeta->tableInfo.numOfColumns = LIST_LENGTH(pStmt->pCols);

  int32_t index = 0;
  SNode*  pCol = NULL;
  FOREACH(pCol, pStmt->pCols) {
    toSchema((SColumnDefNode*)pCol, index + 1, pMeta->schema + index);
    ++index;
  }
  SNode* pTag = NULL;
  FOREACH(pTag, pStmt->pTags) {
    toSchema((SColumnDefNode*)pTag, index + 1, pMeta->schema + index);
    ++index;
  }

  return pMeta;
5183 5184
}

5185 5186 5187 5188 5189 5190
static SNode* createTbnameFunction() {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return NULL;
  }
  strcpy(pFunc->functionName, "tbname");
5191 5192
  strcpy(pFunc->node.aliasName, "tbname");
  strcpy(pFunc->node.userAlias, "tbname");
5193 5194 5195
  return (SNode*)pFunc;
}

L
Liu Jicong 已提交
5196
static int32_t buildSampleAstInfoByTable(STranslateContext* pCxt, SCreateTableStmt* pStmt, SRetention* pRetension,
wmmhello's avatar
wmmhello 已提交
5197 5198 5199 5200 5201 5202 5203 5204 5205
                                         int8_t precision, SSampleAstInfo* pInfo) {
  pInfo->pDbName = pStmt->dbName;
  pInfo->pTableName = pStmt->tableName;
  pInfo->pFuncs = createRollupFuncs(pStmt);
  pInfo->pInterval = makeIntervalVal(pRetension, precision);
  pInfo->pRollupTableMeta = createRollupTableMeta(pStmt, precision);
  if (NULL == pInfo->pFuncs || NULL == pInfo->pInterval || NULL == pInfo->pRollupTableMeta) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
5206
  return nodesListMakeStrictAppend(&pInfo->pPartitionByList, createTbnameFunction());
5207 5208
}

L
Liu Jicong 已提交
5209
static int32_t getRollupAst(STranslateContext* pCxt, SCreateTableStmt* pStmt, SRetention* pRetension, int8_t precision,
wmmhello's avatar
wmmhello 已提交
5210 5211 5212 5213
                            char** pAst, int32_t* pLen) {
  SSampleAstInfo info = {0};
  int32_t        code = buildSampleAstInfoByTable(pCxt, pStmt, pRetension, precision, &info);
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
5214
    code = buildSampleAst(pCxt, &info, pAst, pLen, NULL, NULL);
wmmhello's avatar
wmmhello 已提交
5215 5216 5217
  }
  clearSampleAstInfo(&info);
  return code;
5218 5219 5220
}

static int32_t buildRollupAst(STranslateContext* pCxt, SCreateTableStmt* pStmt, SMCreateStbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
5221 5222 5223 5224 5225 5226 5227 5228 5229
  SDbCfgInfo dbCfg = {0};
  int32_t    code = getDBCfg(pCxt, pStmt->dbName, &dbCfg);
  int32_t    num = taosArrayGetSize(dbCfg.pRetensions);
  if (TSDB_CODE_SUCCESS != code || num < 2) {
    return code;
  }
  for (int32_t i = 1; i < num; ++i) {
    SRetention*       pRetension = taosArrayGet(dbCfg.pRetensions, i);
    STranslateContext cxt = {0};
5230
    initTranslateContext(pCxt->pParseCxt, pCxt->pMetaCache, &cxt);
wmmhello's avatar
wmmhello 已提交
5231 5232 5233 5234 5235 5236 5237
    code = getRollupAst(&cxt, pStmt, pRetension, dbCfg.precision, 1 == i ? &pReq->pAst1 : &pReq->pAst2,
                        1 == i ? &pReq->ast1Len : &pReq->ast2Len);
    destroyTranslateContext(&cxt);
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
  }
D
dapan1121 已提交
5238

wmmhello's avatar
wmmhello 已提交
5239
  return code;
5240
}
X
Xiaoyu Wang 已提交
5241

D
dapan1121 已提交
5242 5243 5244 5245 5246 5247
static int32_t buildRollupFuncs(SNodeList* pFuncs, SArray** pArray) {
  if (NULL == pFuncs) {
    return TSDB_CODE_SUCCESS;
  }
  *pArray = taosArrayInit(LIST_LENGTH(pFuncs), TSDB_FUNC_NAME_LEN);
  SNode* pNode;
5248
  FOREACH(pNode, pFuncs) { taosArrayPush(*pArray, ((SFunctionNode*)pNode)->functionName); }
D
dapan1121 已提交
5249 5250 5251
  return TSDB_CODE_SUCCESS;
}

5252
static int32_t buildCreateStbReq(STranslateContext* pCxt, SCreateTableStmt* pStmt, SMCreateStbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
5253
  pReq->igExists = pStmt->ignoreExists;
X
Xiaoyu Wang 已提交
5254 5255 5256 5257
  pReq->delay1 = pStmt->pOptions->maxDelay1;
  pReq->delay2 = pStmt->pOptions->maxDelay2;
  pReq->watermark1 = pStmt->pOptions->watermark1;
  pReq->watermark2 = pStmt->pOptions->watermark2;
5258 5259
  pReq->deleteMark1 = pStmt->pOptions->deleteMark1;
  pReq->deleteMark2 = pStmt->pOptions->deleteMark2;
5260 5261 5262
  pReq->colVer = 1;
  pReq->tagVer = 1;
  pReq->source = TD_REQ_FROM_APP;
wmmhello's avatar
wmmhello 已提交
5263 5264 5265 5266
  columnDefNodeToField(pStmt->pCols, &pReq->pColumns);
  columnDefNodeToField(pStmt->pTags, &pReq->pTags);
  pReq->numOfColumns = LIST_LENGTH(pStmt->pCols);
  pReq->numOfTags = LIST_LENGTH(pStmt->pTags);
X
Xiaoyu Wang 已提交
5267
  if (pStmt->pOptions->commentNull == false) {
5268
    pReq->pComment = taosStrdup(pStmt->pOptions->comment);
S
Shengliang Guan 已提交
5269
    if (NULL == pReq->pComment) {
wmmhello's avatar
wmmhello 已提交
5270 5271
      return TSDB_CODE_OUT_OF_MEMORY;
    }
wmmhello's avatar
wmmhello 已提交
5272
    pReq->commentLen = strlen(pStmt->pOptions->comment);
X
Xiaoyu Wang 已提交
5273
  } else {
wmmhello's avatar
wmmhello 已提交
5274
    pReq->commentLen = -1;
wmmhello's avatar
wmmhello 已提交
5275
  }
D
dapan1121 已提交
5276
  buildRollupFuncs(pStmt->pOptions->pRollupFuncs, &pReq->pFuncs);
S
Shengliang Guan 已提交
5277
  pReq->numOfFuncs = taosArrayGetSize(pReq->pFuncs);
wmmhello's avatar
wmmhello 已提交
5278 5279 5280 5281

  SName tableName;
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &tableName), pReq->name);
  int32_t code = collectUseTable(&tableName, pCxt->pTables);
D
dapan1121 已提交
5282 5283 5284
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(&tableName, pCxt->pTargetTables);
  }
wmmhello's avatar
wmmhello 已提交
5285 5286 5287 5288
  if (TSDB_CODE_SUCCESS == code) {
    code = buildRollupAst(pCxt, pStmt, pReq);
  }
  return code;
5289
}
5290

5291
static int32_t translateCreateSuperTable(STranslateContext* pCxt, SCreateTableStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5292
  SMCreateStbReq createReq = {0};
5293
  int32_t        code = checkCreateTable(pCxt, pStmt, true);
wmmhello's avatar
wmmhello 已提交
5294 5295 5296 5297 5298 5299 5300 5301
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateStbReq(pCxt, pStmt, &createReq);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_STB, (FSerializeFunc)tSerializeSMCreateStbReq, &createReq);
  }
  tFreeSMCreateStbReq(&createReq);
  return code;
5302 5303
}

5304
static int32_t doTranslateDropSuperTable(STranslateContext* pCxt, const SName* pTableName, bool ignoreNotExists) {
D
dapan1121 已提交
5305 5306 5307 5308 5309 5310 5311 5312
  int32_t code = collectUseTable(pTableName, pCxt->pTargetTables);
  if (TSDB_CODE_SUCCESS == code) {
    SMDropStbReq dropReq = {0};
    tNameExtractFullName(pTableName, dropReq.name);
    dropReq.igNotExists = ignoreNotExists;
    code = buildCmdMsg(pCxt, TDMT_MND_DROP_STB, (FSerializeFunc)tSerializeSMDropStbReq, &dropReq);
  }
  return code;
5313 5314 5315
}

static int32_t translateDropTable(STranslateContext* pCxt, SDropTableStmt* pStmt) {
5316
  SDropTableClause* pClause = (SDropTableClause*)nodesListGetNode(pStmt->pTables, 0);
X
Xiaoyu Wang 已提交
5317 5318 5319
  SName             tableName;
  return doTranslateDropSuperTable(
      pCxt, toName(pCxt->pParseCxt->acctId, pClause->dbName, pClause->tableName, &tableName), pClause->ignoreNotExists);
5320 5321 5322
}

static int32_t translateDropSuperTable(STranslateContext* pCxt, SDropSuperTableStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5323 5324 5325
  SName tableName;
  return doTranslateDropSuperTable(pCxt, toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &tableName),
                                   pStmt->ignoreNotExists);
5326 5327
}

X
Xiaoyu Wang 已提交
5328 5329 5330 5331 5332
static int32_t buildAlterSuperTableReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, SMAlterStbReq* pAlterReq) {
  SName tableName;
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &tableName), pAlterReq->name);
  pAlterReq->alterType = pStmt->alterType;

wmmhello's avatar
wmmhello 已提交
5333
  if (TSDB_ALTER_TABLE_UPDATE_OPTIONS == pStmt->alterType) {
X
Xiaoyu Wang 已提交
5334
    //    pAlterReq->ttl = pStmt->pOptions->ttl;
wmmhello's avatar
wmmhello 已提交
5335
    if (pStmt->pOptions->commentNull == false) {
5336
      pAlterReq->comment = taosStrdup(pStmt->pOptions->comment);
wmmhello's avatar
wmmhello 已提交
5337 5338 5339
      if (NULL == pAlterReq->comment) {
        return TSDB_CODE_OUT_OF_MEMORY;
      }
wmmhello's avatar
wmmhello 已提交
5340
      pAlterReq->commentLen = strlen(pStmt->pOptions->comment);
X
Xiaoyu Wang 已提交
5341
    } else {
wmmhello's avatar
wmmhello 已提交
5342
      pAlterReq->commentLen = -1;
wmmhello's avatar
wmmhello 已提交
5343
    }
wmmhello's avatar
wmmhello 已提交
5344

wmmhello's avatar
wmmhello 已提交
5345 5346 5347 5348 5349 5350 5351 5352 5353 5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364 5365 5366 5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377 5378 5379 5380
    return TSDB_CODE_SUCCESS;
  }

  pAlterReq->pFields = taosArrayInit(2, sizeof(TAOS_FIELD));
  if (NULL == pAlterReq->pFields) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  switch (pStmt->alterType) {
    case TSDB_ALTER_TABLE_ADD_TAG:
    case TSDB_ALTER_TABLE_DROP_TAG:
    case TSDB_ALTER_TABLE_ADD_COLUMN:
    case TSDB_ALTER_TABLE_DROP_COLUMN:
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES:
    case TSDB_ALTER_TABLE_UPDATE_TAG_BYTES: {
      TAOS_FIELD field = {.type = pStmt->dataType.type, .bytes = calcTypeBytes(pStmt->dataType)};
      strcpy(field.name, pStmt->colName);
      taosArrayPush(pAlterReq->pFields, &field);
      break;
    }
    case TSDB_ALTER_TABLE_UPDATE_TAG_NAME:
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_NAME: {
      TAOS_FIELD oldField = {0};
      strcpy(oldField.name, pStmt->colName);
      taosArrayPush(pAlterReq->pFields, &oldField);
      TAOS_FIELD newField = {0};
      strcpy(newField.name, pStmt->newColName);
      taosArrayPush(pAlterReq->pFields, &newField);
      break;
    }
    default:
      break;
  }

  pAlterReq->numOfFields = taosArrayGetSize(pAlterReq->pFields);
  return TSDB_CODE_SUCCESS;
5381 5382
}

5383
static const SSchema* getColSchema(const STableMeta* pTableMeta, const char* pColName) {
X
Xiaoyu Wang 已提交
5384 5385
  int32_t numOfFields = getNumOfTags(pTableMeta) + getNumOfColumns(pTableMeta);
  for (int32_t i = 0; i < numOfFields; ++i) {
5386
    const SSchema* pSchema = pTableMeta->schema + i;
5387 5388 5389 5390 5391 5392 5393
    if (0 == strcmp(pColName, pSchema->name)) {
      return pSchema;
    }
  }
  return NULL;
}

5394 5395 5396 5397 5398 5399 5400 5401 5402 5403 5404 5405
static const SSchema* getNormalColSchema(const STableMeta* pTableMeta, const char* pColName) {
  int32_t  numOfCols = getNumOfColumns(pTableMeta);
  SSchema* pColsSchema = getTableColumnSchema(pTableMeta);
  for (int32_t i = 0; i < numOfCols; ++i) {
    const SSchema* pSchema = pColsSchema + i;
    if (0 == strcmp(pColName, pSchema->name)) {
      return pSchema;
    }
  }
  return NULL;
}

5406
static SSchema* getTagSchema(const STableMeta* pTableMeta, const char* pTagName) {
5407 5408 5409 5410 5411 5412
  int32_t  numOfTags = getNumOfTags(pTableMeta);
  SSchema* pTagsSchema = getTableTagSchema(pTableMeta);
  for (int32_t i = 0; i < numOfTags; ++i) {
    SSchema* pSchema = pTagsSchema + i;
    if (0 == strcmp(pTagName, pSchema->name)) {
      return pSchema;
X
Xiaoyu Wang 已提交
5413 5414 5415 5416 5417
    }
  }
  return NULL;
}

5418 5419
static int32_t checkAlterSuperTableBySchema(STranslateContext* pCxt, SAlterTableStmt* pStmt,
                                            const STableMeta* pTableMeta) {
5420 5421 5422 5423 5424 5425 5426
  SSchema* pTagsSchema = getTableTagSchema(pTableMeta);
  if (getNumOfTags(pTableMeta) == 1 && pTagsSchema->type == TSDB_DATA_TYPE_JSON &&
      (pStmt->alterType == TSDB_ALTER_TABLE_ADD_TAG || pStmt->alterType == TSDB_ALTER_TABLE_DROP_TAG ||
       pStmt->alterType == TSDB_ALTER_TABLE_UPDATE_TAG_BYTES)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_ONLY_ONE_JSON_TAG);
  }

X
Xiaoyu Wang 已提交
5427 5428 5429 5430 5431
  int32_t tagsLen = 0;
  for (int32_t i = 0; i < pTableMeta->tableInfo.numOfTags; ++i) {
    tagsLen += pTagsSchema[i].bytes;
  }

X
Xiaoyu Wang 已提交
5432
  if (TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES == pStmt->alterType ||
5433 5434
      TSDB_ALTER_TABLE_UPDATE_TAG_BYTES == pStmt->alterType || TSDB_ALTER_TABLE_DROP_COLUMN == pStmt->alterType ||
      TSDB_ALTER_TABLE_DROP_TAG == pStmt->alterType) {
5435 5436 5437
    if (TSDB_SUPER_TABLE != pTableMeta->tableType) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE, "Table is not super table");
    }
5438

5439
    const SSchema* pSchema = getColSchema(pTableMeta, pStmt->colName);
5440
    if (NULL == pSchema) {
5441 5442 5443 5444 5445 5446 5447 5448 5449
      return generateSyntaxErrMsg(
          &pCxt->msgBuf,
          (TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES == pStmt->alterType || TSDB_ALTER_TABLE_DROP_COLUMN == pStmt->alterType)
              ? TSDB_CODE_PAR_INVALID_COLUMN
              : TSDB_CODE_PAR_INVALID_TAG_NAME,
          pStmt->colName);
    }

    if ((TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES == pStmt->alterType ||
5450
         TSDB_ALTER_TABLE_UPDATE_TAG_BYTES == pStmt->alterType) &&
5451 5452
        (!IS_VAR_DATA_TYPE(pSchema->type) || pSchema->type != pStmt->dataType.type ||
         pSchema->bytes >= calcTypeBytes(pStmt->dataType))) {
5453
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_MODIFY_COL);
X
Xiaoyu Wang 已提交
5454
    }
X
Xiaoyu Wang 已提交
5455

5456
    if (TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES == pStmt->alterType) {
5457 5458
      if ((TSDB_DATA_TYPE_VARCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_BINARY_LEN) ||
          (TSDB_DATA_TYPE_NCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_NCHAR_LEN)) {
5459 5460 5461 5462 5463 5464
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
      }

      if (pTableMeta->tableInfo.rowSize + calcTypeBytes(pStmt->dataType) - pSchema->bytes > TSDB_MAX_BYTES_PER_ROW) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ROW_LENGTH, TSDB_MAX_BYTES_PER_ROW);
      }
X
Xiaoyu Wang 已提交
5465 5466
    }

5467
    if (TSDB_ALTER_TABLE_UPDATE_TAG_BYTES == pStmt->alterType) {
K
kailixu 已提交
5468
      if (calcTypeBytes(pStmt->dataType) > TSDB_MAX_TAGS_LEN) {
5469 5470 5471 5472 5473 5474
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
      }

      if (tagsLen + calcTypeBytes(pStmt->dataType) - pSchema->bytes > TSDB_MAX_TAGS_LEN) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_LENGTH, TSDB_MAX_TAGS_LEN);
      }
X
Xiaoyu Wang 已提交
5475 5476 5477 5478 5479 5480 5481 5482
    }
  }

  if (TSDB_ALTER_TABLE_ADD_COLUMN == pStmt->alterType) {
    if (TSDB_MAX_COLUMNS == pTableMeta->tableInfo.numOfColumns) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TOO_MANY_COLUMNS);
    }

5483 5484 5485 5486 5487
    if ((TSDB_DATA_TYPE_VARCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_BINARY_LEN) ||
        (TSDB_DATA_TYPE_NCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_NCHAR_LEN)) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
    }

X
Xiaoyu Wang 已提交
5488 5489 5490 5491 5492 5493 5494 5495 5496 5497 5498 5499 5500
    if (pTableMeta->tableInfo.rowSize + calcTypeBytes(pStmt->dataType) > TSDB_MAX_BYTES_PER_ROW) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ROW_LENGTH, TSDB_MAX_BYTES_PER_ROW);
    }
  }

  if (TSDB_ALTER_TABLE_ADD_TAG == pStmt->alterType) {
    if (TSDB_MAX_TAGS == pTableMeta->tableInfo.numOfTags) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_NUM);
    }

    if (tagsLen + calcTypeBytes(pStmt->dataType) > TSDB_MAX_TAGS_LEN) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAGS_LENGTH, TSDB_MAX_TAGS_LEN);
    }
X
Xiaoyu Wang 已提交
5501
  }
X
Xiaoyu Wang 已提交
5502

5503 5504 5505 5506
  if (getNumOfTags(pTableMeta) == 1 && pStmt->alterType == TSDB_ALTER_TABLE_DROP_TAG) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE, "the only tag cannot be dropped");
  }

X
Xiaoyu Wang 已提交
5507 5508 5509
  return TSDB_CODE_SUCCESS;
}

5510
static int32_t checkAlterSuperTable(STranslateContext* pCxt, SAlterTableStmt* pStmt) {
5511
  if (TSDB_ALTER_TABLE_UPDATE_TAG_VAL == pStmt->alterType) {
5512 5513 5514 5515
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE,
                                   "Set tag value only available for child table");
  }

5516 5517 5518 5519 5520
  if (TSDB_ALTER_TABLE_UPDATE_COLUMN_NAME == pStmt->alterType) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE,
                                   "Rename column only available for normal table");
  }

5521 5522 5523 5524 5525 5526 5527 5528 5529 5530 5531 5532
  if (pStmt->alterType == TSDB_ALTER_TABLE_UPDATE_OPTIONS && -1 != pStmt->pOptions->ttl) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE);
  }

  if (pStmt->dataType.type == TSDB_DATA_TYPE_JSON && pStmt->alterType == TSDB_ALTER_TABLE_ADD_TAG) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_ONLY_ONE_JSON_TAG);
  }

  if (pStmt->dataType.type == TSDB_DATA_TYPE_JSON && pStmt->alterType == TSDB_ALTER_TABLE_ADD_COLUMN) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COL_JSON);
  }

5533 5534 5535 5536 5537 5538 5539 5540 5541
  SDbCfgInfo dbCfg = {0};
  int32_t    code = getDBCfg(pCxt, pStmt->dbName, &dbCfg);
  if (TSDB_CODE_SUCCESS == code && NULL != dbCfg.pRetensions &&
      (TSDB_ALTER_TABLE_ADD_COLUMN == pStmt->alterType || TSDB_ALTER_TABLE_DROP_COLUMN == pStmt->alterType ||
       TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES == pStmt->alterType)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE,
                                   "Modifying the table schema is not supported in databases "
                                   "configured with the 'RETENTIONS' option");
  }
5542 5543
  STableMeta* pTableMeta = NULL;
  if (TSDB_CODE_SUCCESS == code) {
5544 5545 5546 5547
    code = getTableMeta(pCxt, pStmt->dbName, pStmt->tableName, &pTableMeta);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = checkAlterSuperTableBySchema(pCxt, pStmt, pTableMeta);
5548 5549 5550 5551 5552
  }
  taosMemoryFree(pTableMeta);
  return code;
}

X
Xiaoyu Wang 已提交
5553 5554 5555 5556 5557 5558
static int32_t translateAlterSuperTable(STranslateContext* pCxt, SAlterTableStmt* pStmt) {
  SMAlterStbReq alterReq = {0};
  int32_t       code = checkAlterSuperTable(pCxt, pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildAlterSuperTableReq(pCxt, pStmt, &alterReq);
  }
X
Xiaoyu Wang 已提交
5559 5560 5561
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_ALTER_STB, (FSerializeFunc)tSerializeSMAlterStbReq, &alterReq);
  }
5562
  tFreeSMAltertbReq(&alterReq);
X
Xiaoyu Wang 已提交
5563
  return code;
5564 5565
}

5566
static int32_t translateUseDatabase(STranslateContext* pCxt, SUseDatabaseStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5567 5568 5569 5570
  SUseDbReq usedbReq = {0};
  SName     name = {0};
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameExtractFullName(&name, usedbReq.db);
X
Xiaoyu Wang 已提交
5571 5572
  int32_t code =
      getDBVgVersion(pCxt, usedbReq.db, &usedbReq.vgVersion, &usedbReq.dbId, &usedbReq.numOfTable, &usedbReq.stateTs);
wmmhello's avatar
wmmhello 已提交
5573 5574 5575 5576
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_USE_DB, (FSerializeFunc)tSerializeSUseDbReq, &usedbReq);
  }
  return code;
5577 5578
}

5579
static int32_t translateCreateUser(STranslateContext* pCxt, SCreateUserStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5580
  SCreateUserReq createReq = {0};
X
Xiaoyu Wang 已提交
5581
  strcpy(createReq.user, pStmt->userName);
wmmhello's avatar
wmmhello 已提交
5582 5583
  createReq.createType = 0;
  createReq.superUser = 0;
X
Xiaoyu Wang 已提交
5584
  createReq.sysInfo = pStmt->sysinfo;
5585
  createReq.enable = 1;
wmmhello's avatar
wmmhello 已提交
5586
  strcpy(createReq.pass, pStmt->password);
5587

wmmhello's avatar
wmmhello 已提交
5588
  return buildCmdMsg(pCxt, TDMT_MND_CREATE_USER, (FSerializeFunc)tSerializeSCreateUserReq, &createReq);
5589 5590 5591
}

static int32_t translateAlterUser(STranslateContext* pCxt, SAlterUserStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5592
  SAlterUserReq alterReq = {0};
X
Xiaoyu Wang 已提交
5593
  strcpy(alterReq.user, pStmt->userName);
wmmhello's avatar
wmmhello 已提交
5594 5595
  alterReq.alterType = pStmt->alterType;
  alterReq.superUser = 0;
X
Xiaoyu Wang 已提交
5596 5597
  alterReq.enable = pStmt->enable;
  alterReq.sysInfo = pStmt->sysinfo;
X
Xiaoyu Wang 已提交
5598
  snprintf(alterReq.pass, sizeof(alterReq.pass), "%s", pStmt->password);
wmmhello's avatar
wmmhello 已提交
5599
  if (NULL != pCxt->pParseCxt->db) {
5600
    snprintf(alterReq.objname, sizeof(alterReq.objname), "%s", pCxt->pParseCxt->db);
wmmhello's avatar
wmmhello 已提交
5601
  }
5602

wmmhello's avatar
wmmhello 已提交
5603
  return buildCmdMsg(pCxt, TDMT_MND_ALTER_USER, (FSerializeFunc)tSerializeSAlterUserReq, &alterReq);
5604 5605 5606
}

static int32_t translateDropUser(STranslateContext* pCxt, SDropUserStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5607
  SDropUserReq dropReq = {0};
X
Xiaoyu Wang 已提交
5608
  strcpy(dropReq.user, pStmt->userName);
5609

wmmhello's avatar
wmmhello 已提交
5610
  return buildCmdMsg(pCxt, TDMT_MND_DROP_USER, (FSerializeFunc)tSerializeSDropUserReq, &dropReq);
5611 5612 5613
}

static int32_t translateCreateDnode(STranslateContext* pCxt, SCreateDnodeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5614 5615 5616
  SCreateDnodeReq createReq = {0};
  strcpy(createReq.fqdn, pStmt->fqdn);
  createReq.port = pStmt->port;
5617

wmmhello's avatar
wmmhello 已提交
5618
  return buildCmdMsg(pCxt, TDMT_MND_CREATE_DNODE, (FSerializeFunc)tSerializeSCreateDnodeReq, &createReq);
5619 5620 5621
}

static int32_t translateDropDnode(STranslateContext* pCxt, SDropDnodeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5622 5623 5624 5625
  SDropDnodeReq dropReq = {0};
  dropReq.dnodeId = pStmt->dnodeId;
  strcpy(dropReq.fqdn, pStmt->fqdn);
  dropReq.port = pStmt->port;
5626
  dropReq.force = pStmt->force;
D
dapan1121 已提交
5627
  dropReq.unsafe = pStmt->unsafe;
5628

wmmhello's avatar
wmmhello 已提交
5629
  return buildCmdMsg(pCxt, TDMT_MND_DROP_DNODE, (FSerializeFunc)tSerializeSDropDnodeReq, &dropReq);
5630 5631
}

5632
static int32_t translateAlterDnode(STranslateContext* pCxt, SAlterDnodeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5633 5634 5635 5636
  SMCfgDnodeReq cfgReq = {0};
  cfgReq.dnodeId = pStmt->dnodeId;
  strcpy(cfgReq.config, pStmt->config);
  strcpy(cfgReq.value, pStmt->value);
5637

wmmhello's avatar
wmmhello 已提交
5638
  return buildCmdMsg(pCxt, TDMT_MND_CONFIG_DNODE, (FSerializeFunc)tSerializeSMCfgDnodeReq, &cfgReq);
5639 5640
}

D
dapan1121 已提交
5641 5642 5643 5644 5645 5646 5647 5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663
static int32_t translateRestoreDnode(STranslateContext* pCxt, SRestoreComponentNodeStmt* pStmt) {
  SRestoreDnodeReq restoreReq = {0};
  restoreReq.dnodeId = pStmt->dnodeId;
  switch (nodeType((SNode*)pStmt)) {
    case QUERY_NODE_RESTORE_DNODE_STMT:
      restoreReq.restoreType = RESTORE_TYPE__ALL;
      break;
    case QUERY_NODE_RESTORE_QNODE_STMT:
      restoreReq.restoreType = RESTORE_TYPE__QNODE;
      break;
    case QUERY_NODE_RESTORE_MNODE_STMT:
      restoreReq.restoreType = RESTORE_TYPE__MNODE;
      break;
    case QUERY_NODE_RESTORE_VNODE_STMT:
      restoreReq.restoreType = RESTORE_TYPE__VNODE;
      break;
    default:
      return -1;
  }
  return buildCmdMsg(pCxt, TDMT_MND_RESTORE_DNODE, (FSerializeFunc)tSerializeSRestoreDnodeReq, &restoreReq);
}


X
Xiaoyu Wang 已提交
5664 5665
static int32_t getSmaIndexDstVgId(STranslateContext* pCxt, const char* pDbName, const char* pTableName,
                                  int32_t* pVgId) {
wmmhello's avatar
wmmhello 已提交
5666
  SVgroupInfo vg = {0};
X
Xiaoyu Wang 已提交
5667
  int32_t     code = getTableHashVgroup(pCxt, pDbName, pTableName, &vg);
wmmhello's avatar
wmmhello 已提交
5668 5669 5670 5671
  if (TSDB_CODE_SUCCESS == code) {
    *pVgId = vg.vgId;
  }
  return code;
X
Xiaoyu Wang 已提交
5672
}
X
Xiaoyu Wang 已提交
5673

X
Xiaoyu Wang 已提交
5674
static int32_t getSmaIndexSql(STranslateContext* pCxt, char** pSql, int32_t* pLen) {
5675
  *pSql = taosStrdup(pCxt->pParseCxt->pSql);
wmmhello's avatar
wmmhello 已提交
5676 5677 5678 5679 5680
  if (NULL == *pSql) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  *pLen = pCxt->pParseCxt->sqlLen + 1;
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
5681 5682
}

5683
static int32_t buildSampleAstInfoByIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt, SSampleAstInfo* pInfo) {
5684
  pInfo->createSmaIndex = true;
X
Xiaoyu Wang 已提交
5685
  pInfo->pDbName = pStmt->dbName;
wmmhello's avatar
wmmhello 已提交
5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696
  pInfo->pTableName = pStmt->tableName;
  pInfo->pFuncs = nodesCloneList(pStmt->pOptions->pFuncs);
  pInfo->pInterval = nodesCloneNode(pStmt->pOptions->pInterval);
  pInfo->pOffset = nodesCloneNode(pStmt->pOptions->pOffset);
  pInfo->pSliding = nodesCloneNode(pStmt->pOptions->pSliding);
  if (NULL == pInfo->pFuncs || NULL == pInfo->pInterval ||
      (NULL != pStmt->pOptions->pOffset && NULL == pInfo->pOffset) ||
      (NULL != pStmt->pOptions->pSliding && NULL == pInfo->pSliding)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  return TSDB_CODE_SUCCESS;
5697
}
X
Xiaoyu Wang 已提交
5698

X
Xiaoyu Wang 已提交
5699 5700
static int32_t getSmaIndexAst(STranslateContext* pCxt, SCreateIndexStmt* pStmt, char** pAst, int32_t* pLen,
                              char** pExpr, int32_t* pExprLen) {
wmmhello's avatar
wmmhello 已提交
5701 5702 5703
  SSampleAstInfo info = {0};
  int32_t        code = buildSampleAstInfoByIndex(pCxt, pStmt, &info);
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
5704
    code = buildSampleAst(pCxt, &info, pAst, pLen, pExpr, pExprLen);
wmmhello's avatar
wmmhello 已提交
5705 5706 5707
  }
  clearSampleAstInfo(&info);
  return code;
X
Xiaoyu Wang 已提交
5708 5709 5710
}

static int32_t buildCreateSmaReq(STranslateContext* pCxt, SCreateIndexStmt* pStmt, SMCreateSmaReq* pReq) {
wmmhello's avatar
wmmhello 已提交
5711
  SName name;
X
Xiaoyu Wang 已提交
5712
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->indexDbName, pStmt->indexName, &name), pReq->name);
5713 5714
  memset(&name, 0, sizeof(SName));
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &name), pReq->stb);
wmmhello's avatar
wmmhello 已提交
5715 5716 5717 5718 5719 5720 5721 5722
  pReq->igExists = pStmt->ignoreExists;
  pReq->interval = ((SValueNode*)pStmt->pOptions->pInterval)->datum.i;
  pReq->intervalUnit = ((SValueNode*)pStmt->pOptions->pInterval)->unit;
  pReq->offset = (NULL != pStmt->pOptions->pOffset ? ((SValueNode*)pStmt->pOptions->pOffset)->datum.i : 0);
  pReq->sliding =
      (NULL != pStmt->pOptions->pSliding ? ((SValueNode*)pStmt->pOptions->pSliding)->datum.i : pReq->interval);
  pReq->slidingUnit =
      (NULL != pStmt->pOptions->pSliding ? ((SValueNode*)pStmt->pOptions->pSliding)->unit : pReq->intervalUnit);
5723 5724

  int32_t code = TSDB_CODE_SUCCESS;
5725 5726
  if (NULL != pStmt->pOptions->pStreamOptions) {
    SStreamOptions* pStreamOpt = (SStreamOptions*)pStmt->pOptions->pStreamOptions;
5727 5728 5729 5730
    if (NULL != pStreamOpt->pDelay) {
      code = getTableMaxDelayOption(pCxt, (SValueNode*)pStreamOpt->pDelay, &pReq->maxDelay);
    } else {
      pReq->maxDelay = -1;
C
Cary Xu 已提交
5731
    }
5732 5733 5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744
    if (TSDB_CODE_SUCCESS == code) {
      if (NULL != pStreamOpt->pWatermark) {
        code = getTableWatermarkOption(pCxt, (SValueNode*)pStreamOpt->pWatermark, &pReq->watermark);
      } else {
        pReq->watermark = TSDB_DEFAULT_ROLLUP_WATERMARK;
      }
    }
    if (TSDB_CODE_SUCCESS == code) {
      if (NULL != pStreamOpt->pDeleteMark) {
        code = getTableDeleteMarkOption(pCxt, (SValueNode*)pStreamOpt->pDeleteMark, &pReq->deleteMark);
      } else {
        pReq->deleteMark = TSDB_DEFAULT_ROLLUP_DELETE_MARK;
      }
C
Cary Xu 已提交
5745
    }
5746
  }
wmmhello's avatar
wmmhello 已提交
5747

5748 5749 5750
  if (TSDB_CODE_SUCCESS == code) {
    code = getSmaIndexDstVgId(pCxt, pStmt->dbName, pStmt->tableName, &pReq->dstVgId);
  }
wmmhello's avatar
wmmhello 已提交
5751 5752 5753 5754
  if (TSDB_CODE_SUCCESS == code) {
    code = getSmaIndexSql(pCxt, &pReq->sql, &pReq->sqlLen);
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
5755
    code = getSmaIndexAst(pCxt, pStmt, &pReq->ast, &pReq->astLen, &pReq->expr, &pReq->exprLen);
wmmhello's avatar
wmmhello 已提交
5756 5757 5758
  }

  return code;
X
Xiaoyu Wang 已提交
5759 5760
}

5761 5762
static int32_t checkCreateSmaIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt) {
  SDbCfgInfo dbCfg = {0};
X
Xiaoyu Wang 已提交
5763
  int32_t    code = getDBCfg(pCxt, pStmt->dbName, &dbCfg);
5764 5765 5766 5767 5768 5769 5770 5771 5772 5773 5774 5775
  if (TSDB_CODE_SUCCESS == code && NULL != dbCfg.pRetensions) {
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_SMA_INDEX,
                                   "Tables configured with the 'ROLLUP' option do not support creating sma index");
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = doTranslateValue(pCxt, (SValueNode*)pStmt->pOptions->pInterval);
  }
  if (TSDB_CODE_SUCCESS == code && NULL != pStmt->pOptions->pOffset) {
    code = doTranslateValue(pCxt, (SValueNode*)pStmt->pOptions->pOffset);
  }
  if (TSDB_CODE_SUCCESS == code && NULL != pStmt->pOptions->pSliding) {
    code = doTranslateValue(pCxt, (SValueNode*)pStmt->pOptions->pSliding);
wmmhello's avatar
wmmhello 已提交
5776 5777
  }

5778 5779 5780 5781
  return code;
}

static int32_t translateCreateSmaIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5782
  SMCreateSmaReq createSmaReq = {0};
5783 5784 5785 5786
  int32_t        code = checkCreateSmaIndex(pCxt, pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateSmaReq(pCxt, pStmt, &createSmaReq);
  }
wmmhello's avatar
wmmhello 已提交
5787 5788 5789 5790 5791
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_SMA, (FSerializeFunc)tSerializeSMCreateSmaReq, &createSmaReq);
  }
  tFreeSMCreateSmaReq(&createSmaReq);
  return code;
X
Xiaoyu Wang 已提交
5792
}
5793

dengyihao's avatar
dengyihao 已提交
5794
static int32_t buildCreateFullTextReq(STranslateContext* pCxt, SCreateIndexStmt* pStmt, SMCreateFullTextReq* pReq) {
wmmhello's avatar
wmmhello 已提交
5795
  // impl later
5796
  return TSDB_CODE_SUCCESS;
dengyihao's avatar
dengyihao 已提交
5797
}
5798

dengyihao's avatar
dengyihao 已提交
5799 5800
static int32_t buildCreateTagIndexReq(STranslateContext* pCxt, SCreateIndexStmt* pStmt, SCreateTagIndexReq* pReq) {
  SName name;
dengyihao's avatar
dengyihao 已提交
5801
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->indexDbName, pStmt->indexName, &name), pReq->idxName);
dengyihao's avatar
dengyihao 已提交
5802 5803 5804
  memset(&name, 0, sizeof(SName));

  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &name), pReq->stbName);
dengyihao's avatar
dengyihao 已提交
5805 5806 5807 5808
  memset(&name, 0, sizeof(SName));

  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, pReq->dbFName);
dengyihao's avatar
dengyihao 已提交
5809 5810 5811 5812 5813 5814 5815 5816

  SNode* pNode = NULL;
  ASSERT(LIST_LENGTH(pStmt->pCols) == 1);
  FOREACH(pNode, pStmt->pCols) {
    SColumnNode* p = (SColumnNode*)pNode;
    memcpy(pReq->colName, p->colName, sizeof(p->colName));
  }

dengyihao's avatar
dengyihao 已提交
5817 5818 5819 5820
  // impl later
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
5821
static int32_t translateCreateFullTextIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5822 5823 5824 5825 5826 5827 5828
  SMCreateFullTextReq createFTReq = {0};
  int32_t             code = buildCreateFullTextReq(pCxt, pStmt, &createFTReq);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_INDEX, (FSerializeFunc)tSerializeSMCreateFullTextReq, &createFTReq);
  }
  tFreeSMCreateFullTextReq(&createFTReq);
  return code;
dengyihao's avatar
dengyihao 已提交
5829
}
X
Xiaoyu Wang 已提交
5830

dengyihao's avatar
dengyihao 已提交
5831 5832 5833 5834 5835 5836 5837 5838 5839
static int32_t translateCreateNormalIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt) {
  SCreateTagIndexReq createTagIdxReq = {0};
  int32_t            code = buildCreateTagIndexReq(pCxt, pStmt, &createTagIdxReq);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_INDEX, (FSerializeFunc)tSerializeSCreateTagIdxReq, &createTagIdxReq);
  }
  return code;
}

X
Xiaoyu Wang 已提交
5840
static int32_t translateCreateIndex(STranslateContext* pCxt, SCreateIndexStmt* pStmt) {
5841
  if (INDEX_TYPE_FULLTEXT == pStmt->indexType) {
wmmhello's avatar
wmmhello 已提交
5842
    return translateCreateFullTextIndex(pCxt, pStmt);
dengyihao's avatar
dengyihao 已提交
5843 5844
  } else if (INDEX_TYPE_NORMAL == pStmt->indexType) {
    return translateCreateNormalIndex(pCxt, pStmt);
wmmhello's avatar
wmmhello 已提交
5845
  }
5846
  return translateCreateSmaIndex(pCxt, pStmt);
X
Xiaoyu Wang 已提交
5847 5848
}

5849
static int32_t translateDropIndex(STranslateContext* pCxt, SDropIndexStmt* pStmt) {
X
Xiaoyu Wang 已提交
5850 5851
  SMDropSmaReq dropSmaReq = {0};
  SName        name;
X
Xiaoyu Wang 已提交
5852
  tNameExtractFullName(toName(pCxt->pParseCxt->acctId, pStmt->indexDbName, pStmt->indexName, &name), dropSmaReq.name);
X
Xiaoyu Wang 已提交
5853 5854
  dropSmaReq.igNotExists = pStmt->ignoreNotExists;
  return buildCmdMsg(pCxt, TDMT_MND_DROP_SMA, (FSerializeFunc)tSerializeSMDropSmaReq, &dropSmaReq);
wmmhello's avatar
wmmhello 已提交
5855 5856 5857 5858 5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869 5870 5871
}

static int16_t getCreateComponentNodeMsgType(ENodeType type) {
  switch (type) {
    case QUERY_NODE_CREATE_QNODE_STMT:
      return TDMT_MND_CREATE_QNODE;
    case QUERY_NODE_CREATE_BNODE_STMT:
      return TDMT_MND_CREATE_BNODE;
    case QUERY_NODE_CREATE_SNODE_STMT:
      return TDMT_MND_CREATE_SNODE;
    case QUERY_NODE_CREATE_MNODE_STMT:
      return TDMT_MND_CREATE_MNODE;
    default:
      break;
  }
  return -1;
}
5872 5873

static int32_t translateCreateComponentNode(STranslateContext* pCxt, SCreateComponentNodeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5874 5875
  SMCreateQnodeReq createReq = {.dnodeId = pStmt->dnodeId};
  return buildCmdMsg(pCxt, getCreateComponentNodeMsgType(nodeType(pStmt)),
S
Shengliang Guan 已提交
5876
                     (FSerializeFunc)tSerializeSCreateDropMQSNodeReq, &createReq);
X
Xiaoyu Wang 已提交
5877 5878
}

5879
static int16_t getDropComponentNodeMsgType(ENodeType type) {
wmmhello's avatar
wmmhello 已提交
5880 5881 5882 5883 5884 5885 5886 5887 5888 5889 5890 5891 5892
  switch (type) {
    case QUERY_NODE_DROP_QNODE_STMT:
      return TDMT_MND_DROP_QNODE;
    case QUERY_NODE_DROP_BNODE_STMT:
      return TDMT_MND_DROP_BNODE;
    case QUERY_NODE_DROP_SNODE_STMT:
      return TDMT_MND_DROP_SNODE;
    case QUERY_NODE_DROP_MNODE_STMT:
      return TDMT_MND_DROP_MNODE;
    default:
      break;
  }
  return -1;
5893 5894 5895
}

static int32_t translateDropComponentNode(STranslateContext* pCxt, SDropComponentNodeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
5896 5897
  SDDropQnodeReq dropReq = {.dnodeId = pStmt->dnodeId};
  return buildCmdMsg(pCxt, getDropComponentNodeMsgType(nodeType(pStmt)),
S
Shengliang Guan 已提交
5898
                     (FSerializeFunc)tSerializeSCreateDropMQSNodeReq, &dropReq);
5899 5900
}

X
Xiaoyu Wang 已提交
5901 5902 5903 5904 5905 5906 5907
static int32_t checkTopicQuery(STranslateContext* pCxt, SSelectStmt* pSelect) {
  if (pSelect->hasAggFuncs || pSelect->hasInterpFunc || pSelect->hasIndefiniteRowsFunc) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TOPIC_QUERY);
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
5908
static int32_t buildCreateTopicReq(STranslateContext* pCxt, SCreateTopicStmt* pStmt, SCMCreateTopicReq* pReq) {
X
Xiaoyu Wang 已提交
5909
  snprintf(pReq->name, sizeof(pReq->name), "%d.%s", pCxt->pParseCxt->acctId, pStmt->topicName);
wmmhello's avatar
wmmhello 已提交
5910
  pReq->igExists = pStmt->ignoreExists;
X
Xiaoyu Wang 已提交
5911
  pReq->withMeta = pStmt->withMeta;
wmmhello's avatar
wmmhello 已提交
5912

5913
  pReq->sql = taosStrdup(pCxt->pParseCxt->pSql);
wmmhello's avatar
wmmhello 已提交
5914 5915 5916 5917 5918
  if (NULL == pReq->sql) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
5919
  SName   name;
5920 5921 5922
  if ('\0' != pStmt->subSTbName[0]) {
    pReq->subType = TOPIC_SUB_TYPE__TABLE;
    toName(pCxt->pParseCxt->acctId, pStmt->subDbName, pStmt->subSTbName, &name);
L
Liu Jicong 已提交
5923
    tNameGetFullDbName(&name, pReq->subDbName);
5924
    tNameExtractFullName(&name, pReq->subStbName);
5925 5926 5927
    if(pStmt->pQuery != NULL) {
      code = nodesNodeToString(pStmt->pQuery, false, &pReq->ast, NULL);
    }
5928 5929 5930 5931 5932 5933
  } else if ('\0' != pStmt->subDbName[0]) {
    pReq->subType = TOPIC_SUB_TYPE__DB;
    tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->subDbName, strlen(pStmt->subDbName));
    tNameGetFullDbName(&name, pReq->subDbName);
  } else {
    pReq->subType = TOPIC_SUB_TYPE__COLUMN;
L
Liu Jicong 已提交
5934 5935
    char* dbName = ((SRealTableNode*)(((SSelectStmt*)pStmt->pQuery)->pFromTable))->table.dbName;
    tNameSetDbName(&name, pCxt->pParseCxt->acctId, dbName, strlen(dbName));
L
Liu Jicong 已提交
5936
    tNameGetFullDbName(&name, pReq->subDbName);
wmmhello's avatar
wmmhello 已提交
5937 5938
    pCxt->pParseCxt->topicQuery = true;
    code = translateQuery(pCxt, pStmt->pQuery);
X
Xiaoyu Wang 已提交
5939 5940 5941
    if (TSDB_CODE_SUCCESS == code) {
      code = checkTopicQuery(pCxt, (SSelectStmt*)pStmt->pQuery);
    }
wmmhello's avatar
wmmhello 已提交
5942 5943 5944 5945 5946 5947
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesNodeToString(pStmt->pQuery, false, &pReq->ast, NULL);
    }
  }

  return code;
X
Xiaoyu Wang 已提交
5948 5949
}

5950 5951 5952 5953 5954 5955 5956 5957 5958 5959 5960 5961 5962 5963 5964 5965 5966 5967 5968 5969 5970 5971 5972 5973 5974 5975 5976 5977 5978 5979 5980 5981 5982 5983 5984 5985 5986 5987 5988 5989 5990 5991 5992 5993
static int32_t addTagList(SNodeList** ppList, SNode* pNode) {
  if (NULL == *ppList) {
    *ppList = nodesMakeList();
  }

  nodesListStrictAppend(*ppList, pNode);

  return TSDB_CODE_SUCCESS;
}

static EDealRes checkColumnTagsInCond(SNode* pNode, void* pContext) {
  SBuildTopicContext* pCxt = (SBuildTopicContext*)pContext;
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
    ETableColumnType type;
    getColumnTypeFromMeta(pCxt->pMeta, ((SColumnNode*)pNode)->colName, &type);
    if (type == TCOL_TYPE_COLUMN) {
      pCxt->colExists = true;
      return DEAL_RES_ERROR;
    } else if (type == TCOL_TYPE_TAG) {
      addTagList(&pCxt->pTags, nodesCloneNode(pNode));
    } else {
      pCxt->colNotFound = true;
      return DEAL_RES_ERROR;
    }
  } else if (QUERY_NODE_FUNCTION == nodeType(pNode)) {
    SFunctionNode* pFunc = (SFunctionNode*)pNode;
    if (0 == strcasecmp(pFunc->functionName, "tbname")) {
      addTagList(&pCxt->pTags, nodesCloneNode(pNode));
    }
  }
  
  return DEAL_RES_CONTINUE;
}

static int32_t checkCollectTopicTags(STranslateContext* pCxt, SCreateTopicStmt* pStmt, STableMeta* pMeta, SNodeList** ppProjection) {
  SBuildTopicContext colCxt = {.colExists = false, .colNotFound = false, .pMeta = pMeta, .pTags = NULL};
  nodesWalkExprPostOrder(pStmt->pWhere, checkColumnTagsInCond, &colCxt);
  if (colCxt.colNotFound) {
    nodesDestroyList(colCxt.pTags);
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR, "Invalid column name");
  } else if (colCxt.colExists) {
    nodesDestroyList(colCxt.pTags);
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR, "Columns are forbidden in where clause");
  }
5994 5995 5996
  if (NULL == colCxt.pTags) {   // put one column to select
//    for (int32_t i = 0; i < pMeta->tableInfo.numOfColumns; ++i) {
      SSchema* column = &pMeta->schema[0];
5997
      SColumnNode* col = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
5998
      strcpy(col->colName, column->name);
5999 6000 6001
      strcpy(col->node.aliasName, col->colName);
      strcpy(col->node.userAlias, col->colName);
      addTagList(&colCxt.pTags, (SNode*)col);
6002
//    }
6003 6004 6005 6006 6007 6008 6009 6010 6011 6012 6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024 6025 6026 6027 6028 6029 6030 6031 6032 6033 6034 6035
  }

  *ppProjection = colCxt.pTags;
  return TSDB_CODE_SUCCESS;
}

static int32_t buildQueryForTableTopic(STranslateContext* pCxt, SCreateTopicStmt* pStmt, SNode** pSelect) {
  SParseContext* pParCxt = pCxt->pParseCxt;
  SRequestConnInfo connInfo = {.pTrans = pParCxt->pTransporter, 
                               .requestId = pParCxt->requestId, 
                               .requestObjRefId = pParCxt->requestRid,
                               .mgmtEps = pParCxt->mgmtEpSet};
  SName name;          
  STableMeta* pMeta = NULL;
  int32_t code = getTableMetaImpl(pCxt, toName(pParCxt->acctId, pStmt->subDbName, pStmt->subSTbName, &name), &pMeta);
  if (code) {
    taosMemoryFree(pMeta);
    return code;
  }
  if (TSDB_SUPER_TABLE != pMeta->tableType) {
    taosMemoryFree(pMeta);
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR, "Only supertable table can be used");
  }  

  SNodeList* pProjection = NULL;
  code = checkCollectTopicTags(pCxt, pStmt, pMeta, &pProjection);
  if (TSDB_CODE_SUCCESS == code) {
    SRealTableNode* realTable = (SRealTableNode*)nodesMakeNode(QUERY_NODE_REAL_TABLE);
    strcpy(realTable->table.dbName, pStmt->subDbName);
    strcpy(realTable->table.tableName, pStmt->subSTbName);
    strcpy(realTable->table.tableAlias, pStmt->subSTbName);
    *pSelect = createSelectStmtImpl(true, pProjection, (SNode*)realTable);
    ((SSelectStmt*)*pSelect)->pWhere = nodesCloneNode(pStmt->pWhere);
D
dapan1121 已提交
6036
    pCxt->pParseCxt->topicQuery = true;
6037 6038 6039 6040 6041 6042 6043
    code = translateQuery(pCxt, *pSelect);
  }

  taosMemoryFree(pMeta);
  return code;
}

X
Xiaoyu Wang 已提交
6044
static int32_t checkCreateTopic(STranslateContext* pCxt, SCreateTopicStmt* pStmt) {
6045
  if (NULL == pStmt->pQuery && NULL == pStmt->pWhere) {
wmmhello's avatar
wmmhello 已提交
6046 6047
    return TSDB_CODE_SUCCESS;
  }
6048

6049 6050 6051
  if (pStmt->pWhere) {
    return buildQueryForTableTopic(pCxt, pStmt, &pStmt->pQuery);
  } else if (QUERY_NODE_SELECT_STMT == nodeType(pStmt->pQuery)) {
wmmhello's avatar
wmmhello 已提交
6052
    SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
X
Xiaoyu Wang 已提交
6053 6054
    if (!pSelect->isDistinct &&
        (NULL != pSelect->pFromTable && QUERY_NODE_REAL_TABLE == nodeType(pSelect->pFromTable)) &&
wmmhello's avatar
wmmhello 已提交
6055 6056 6057 6058 6059
        NULL == pSelect->pGroupByList && NULL == pSelect->pLimit && NULL == pSelect->pSlimit &&
        NULL == pSelect->pOrderByList && NULL == pSelect->pPartitionByList) {
      return TSDB_CODE_SUCCESS;
    }
  }
X
Xiaoyu Wang 已提交
6060

wmmhello's avatar
wmmhello 已提交
6061
  return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TOPIC_QUERY);
X
Xiaoyu Wang 已提交
6062 6063 6064
}

static int32_t translateCreateTopic(STranslateContext* pCxt, SCreateTopicStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6065 6066 6067 6068 6069 6070
  SCMCreateTopicReq createReq = {0};
  int32_t           code = checkCreateTopic(pCxt, pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateTopicReq(pCxt, pStmt, &createReq);
  }
  if (TSDB_CODE_SUCCESS == code) {
L
Liu Jicong 已提交
6071
    code = buildCmdMsg(pCxt, TDMT_MND_TMQ_CREATE_TOPIC, (FSerializeFunc)tSerializeSCMCreateTopicReq, &createReq);
wmmhello's avatar
wmmhello 已提交
6072 6073 6074
  }
  tFreeSCMCreateTopicReq(&createReq);
  return code;
6075 6076 6077
}

static int32_t translateDropTopic(STranslateContext* pCxt, SDropTopicStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6078
  SMDropTopicReq dropReq = {0};
6079

wmmhello's avatar
wmmhello 已提交
6080 6081 6082 6083
  SName name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->topicName, strlen(pStmt->topicName));
  tNameGetFullDbName(&name, dropReq.name);
  dropReq.igNotExists = pStmt->ignoreNotExists;
6084

L
Liu Jicong 已提交
6085
  return buildCmdMsg(pCxt, TDMT_MND_TMQ_DROP_TOPIC, (FSerializeFunc)tSerializeSMDropTopicReq, &dropReq);
6086 6087
}

X
Xiaoyu Wang 已提交
6088 6089 6090 6091 6092 6093 6094 6095 6096
static int32_t translateDropCGroup(STranslateContext* pCxt, SDropCGroupStmt* pStmt) {
  SMDropCgroupReq dropReq = {0};

  SName name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->topicName, strlen(pStmt->topicName));
  tNameGetFullDbName(&name, dropReq.topic);
  dropReq.igNotExists = pStmt->ignoreNotExists;
  strcpy(dropReq.cgroup, pStmt->cgroup);

L
Liu Jicong 已提交
6097
  return buildCmdMsg(pCxt, TDMT_MND_TMQ_DROP_CGROUP, (FSerializeFunc)tSerializeSMDropCgroupReq, &dropReq);
X
Xiaoyu Wang 已提交
6098 6099
}

6100
static int32_t translateAlterLocal(STranslateContext* pCxt, SAlterLocalStmt* pStmt) {
X
Xiaoyu Wang 已提交
6101
  // The statement is executed directly on the client without constructing a message.
6102 6103 6104 6105 6106 6107 6108 6109
  if ('\0' != pStmt->value[0]) {
    return TSDB_CODE_SUCCESS;
  }
  char* p = strchr(pStmt->config, ' ');
  if (NULL != p) {
    *p = 0;
    strcpy(pStmt->value, p + 1);
  }
wmmhello's avatar
wmmhello 已提交
6110
  return TSDB_CODE_SUCCESS;
6111 6112
}

6113
static int32_t translateExplain(STranslateContext* pCxt, SExplainStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6114 6115 6116 6117
  if (pStmt->analyze) {
    pCxt->pExplainOpt = pStmt->pOptions;
  }
  return translateQuery(pCxt, pStmt->pQuery);
6118 6119
}

6120
static int32_t translateDescribe(STranslateContext* pCxt, SDescribeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6121
  return refreshGetTableMeta(pCxt, pStmt->dbName, pStmt->tableName, &pStmt->pMeta);
6122 6123
}

6124 6125 6126 6127 6128 6129 6130 6131 6132 6133 6134 6135 6136 6137 6138 6139 6140 6141 6142 6143
static int32_t translateCompactRange(STranslateContext* pCxt, SCompactDatabaseStmt* pStmt, SCompactDbReq* pReq) {
  SDbCfgInfo dbCfg = {0};
  int32_t    code = getDBCfg(pCxt, pStmt->dbName, &dbCfg);
  if (TSDB_CODE_SUCCESS == code && NULL != pStmt->pStart) {
    ((SValueNode*)pStmt->pStart)->node.resType.precision = dbCfg.precision;
    ((SValueNode*)pStmt->pStart)->node.resType.type = TSDB_DATA_TYPE_TIMESTAMP;
    code = doTranslateValue(pCxt, (SValueNode*)pStmt->pStart);
  }
  if (TSDB_CODE_SUCCESS == code && NULL != pStmt->pEnd) {
    ((SValueNode*)pStmt->pEnd)->node.resType.precision = dbCfg.precision;
    ((SValueNode*)pStmt->pEnd)->node.resType.type = TSDB_DATA_TYPE_TIMESTAMP;
    code = doTranslateValue(pCxt, (SValueNode*)pStmt->pEnd);
  }
  if (TSDB_CODE_SUCCESS == code) {
    pReq->timeRange.skey = NULL != pStmt->pStart ? ((SValueNode*)pStmt->pStart)->datum.i : INT64_MIN;
    pReq->timeRange.ekey = NULL != pStmt->pEnd ? ((SValueNode*)pStmt->pEnd)->datum.i : INT64_MAX;
  }
  return code;
}

X
Xiaoyu Wang 已提交
6144 6145 6146 6147 6148
static int32_t translateCompact(STranslateContext* pCxt, SCompactDatabaseStmt* pStmt) {
  SCompactDbReq compactReq = {0};
  SName         name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, compactReq.db);
6149 6150 6151 6152 6153
  int32_t code = translateCompactRange(pCxt, pStmt, &compactReq);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_COMPACT_DB, (FSerializeFunc)tSerializeSCompactDbReq, &compactReq);
  }
  return code;
X
Xiaoyu Wang 已提交
6154 6155
}

6156
static int32_t translateKillConnection(STranslateContext* pCxt, SKillStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6157 6158
  SKillConnReq killReq = {0};
  killReq.connId = pStmt->targetId;
D
dapan1121 已提交
6159
  return buildCmdMsg(pCxt, TDMT_MND_KILL_CONN, (FSerializeFunc)tSerializeSKillConnReq, &killReq);
6160 6161
}

X
Xiaoyu Wang 已提交
6162
static int32_t translateKillQuery(STranslateContext* pCxt, SKillQueryStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6163
  SKillQueryReq killReq = {0};
X
Xiaoyu Wang 已提交
6164
  strcpy(killReq.queryStrId, pStmt->queryId);
wmmhello's avatar
wmmhello 已提交
6165
  return buildCmdMsg(pCxt, TDMT_MND_KILL_QUERY, (FSerializeFunc)tSerializeSKillQueryReq, &killReq);
6166 6167
}

6168
static int32_t translateKillTransaction(STranslateContext* pCxt, SKillStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6169 6170 6171
  SKillTransReq killReq = {0};
  killReq.transId = pStmt->targetId;
  return buildCmdMsg(pCxt, TDMT_MND_KILL_TRANS, (FSerializeFunc)tSerializeSKillTransReq, &killReq);
6172 6173
}

X
Xiaoyu Wang 已提交
6174 6175 6176
static bool crossTableWithoutAggOper(SSelectStmt* pSelect) {
  return NULL == pSelect->pWindow && !pSelect->hasAggFuncs && !pSelect->hasIndefiniteRowsFunc &&
         !pSelect->hasInterpFunc && TSDB_SUPER_TABLE == ((SRealTableNode*)pSelect->pFromTable)->pMeta->tableType &&
6177
         !hasPartitionByTbname(pSelect->pPartitionByList);
X
Xiaoyu Wang 已提交
6178 6179
}

X
Xiaoyu Wang 已提交
6180 6181
static bool crossTableWithUdaf(SSelectStmt* pSelect) {
  return pSelect->hasUdaf && TSDB_SUPER_TABLE == ((SRealTableNode*)pSelect->pFromTable)->pMeta->tableType &&
6182
         !hasPartitionByTbname(pSelect->pPartitionByList);
X
Xiaoyu Wang 已提交
6183 6184
}

X
Xiaoyu Wang 已提交
6185
static int32_t checkCreateStream(STranslateContext* pCxt, SCreateStreamStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6186 6187 6188
  if (NULL == pStmt->pQuery) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
6189

6190
  if (QUERY_NODE_SELECT_STMT != nodeType(pStmt->pQuery) || NULL == ((SSelectStmt*)pStmt->pQuery)->pFromTable ||
X
Xiaoyu Wang 已提交
6191
      QUERY_NODE_REAL_TABLE != nodeType(((SSelectStmt*)pStmt->pQuery)->pFromTable)) {
X
Xiaoyu Wang 已提交
6192
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY, "Unsupported stream query");
wmmhello's avatar
wmmhello 已提交
6193
  }
6194

X
Xiaoyu Wang 已提交
6195
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
6196 6197 6198
}

static void getSourceDatabase(SNode* pStmt, int32_t acctId, char* pDbFName) {
wmmhello's avatar
wmmhello 已提交
6199 6200 6201
  SName name = {.type = TSDB_DB_NAME_T, .acctId = acctId};
  strcpy(name.dbname, ((SRealTableNode*)(((SSelectStmt*)pStmt)->pFromTable))->table.dbName);
  tNameGetFullDbName(&name, pDbFName);
X
Xiaoyu Wang 已提交
6202 6203
}

X
Xiaoyu Wang 已提交
6204 6205 6206 6207 6208 6209 6210 6211 6212
static void getStreamQueryFirstProjectAliasName(SHashObj* pUserAliasSet, char* aliasName, int32_t len) {
  if (NULL == taosHashGet(pUserAliasSet, "_wstart", strlen("_wstart"))) {
    snprintf(aliasName, len, "%s", "_wstart");
    return;
  }
  if (NULL == taosHashGet(pUserAliasSet, "ts", strlen("ts"))) {
    snprintf(aliasName, len, "%s", "ts");
    return;
  }
X
Xiaoyu Wang 已提交
6213 6214 6215 6216
  do {
    taosRandStr(aliasName, len - 1);
    aliasName[len - 1] = '\0';
  } while (NULL != taosHashGet(pUserAliasSet, aliasName, strlen(aliasName)));
X
Xiaoyu Wang 已提交
6217 6218 6219
  return;
}

6220 6221
static int32_t addWstartTsToCreateStreamQueryImpl(STranslateContext* pCxt, SSelectStmt* pSelect,
                                                  SHashObj* pUserAliasSet) {
X
Xiaoyu Wang 已提交
6222
  SNode* pProj = nodesListGetNode(pSelect->pProjectionList, 0);
6223
  if (NULL == pSelect->pWindow ||
6224
      (QUERY_NODE_FUNCTION == nodeType(pProj) && 0 == strcmp("_wstart", ((SFunctionNode*)pProj)->functionName))) {
6225 6226 6227 6228 6229 6230
    return TSDB_CODE_SUCCESS;
  }
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
6231
  strcpy(pFunc->functionName, "_wstart");
X
Xiaoyu Wang 已提交
6232
  getStreamQueryFirstProjectAliasName(pUserAliasSet, pFunc->node.aliasName, sizeof(pFunc->node.aliasName));
6233
  int32_t code = getFuncInfo(pCxt, pFunc);
6234
  if (TSDB_CODE_SUCCESS == code) {
6235 6236
    code = nodesListPushFront(pSelect->pProjectionList, (SNode*)pFunc);
  }
6237 6238 6239 6240 6241 6242
  if (TSDB_CODE_SUCCESS != code) {
    nodesDestroyNode((SNode*)pFunc);
  }
  return code;
}

X
Xiaoyu Wang 已提交
6243 6244 6245 6246 6247
static int32_t addWstartTsToCreateStreamQuery(STranslateContext* pCxt, SNode* pStmt) {
  SSelectStmt* pSelect = (SSelectStmt*)pStmt;
  SHashObj*    pUserAliasSet = NULL;
  int32_t      code = checkProjectAlias(pCxt, pSelect->pProjectionList, &pUserAliasSet);
  if (TSDB_CODE_SUCCESS == code) {
6248
    code = addWstartTsToCreateStreamQueryImpl(pCxt, pSelect, pUserAliasSet);
X
Xiaoyu Wang 已提交
6249 6250 6251 6252 6253
  }
  taosHashCleanup(pUserAliasSet);
  return code;
}

6254 6255 6256 6257 6258 6259 6260
static const char* getTagNameForCreateStreamTag(SNode* pTag) {
  if (QUERY_NODE_COLUMN_DEF == nodeType(pTag)) {
    return ((SColumnDefNode*)pTag)->colName;
  }
  return ((SColumnNode*)pTag)->colName;
}

6261 6262 6263 6264 6265 6266 6267 6268 6269 6270
static int32_t addTagsToCreateStreamQuery(STranslateContext* pCxt, SCreateStreamStmt* pStmt, SSelectStmt* pSelect) {
  if (NULL == pStmt->pTags) {
    return TSDB_CODE_SUCCESS;
  }

  SNode* pTag = NULL;
  FOREACH(pTag, pStmt->pTags) {
    bool   found = false;
    SNode* pPart = NULL;
    FOREACH(pPart, pSelect->pPartitionByList) {
6271
      if (0 == strcmp(getTagNameForCreateStreamTag(pTag), ((SExprNode*)pPart)->userAlias)) {
6272 6273 6274 6275 6276 6277 6278 6279
        if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(&pSelect->pTags, nodesCloneNode(pPart))) {
          return TSDB_CODE_OUT_OF_MEMORY;
        }
        found = true;
        break;
      }
    }
    if (!found) {
6280
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, ((SColumnDefNode*)pTag)->colName);
6281 6282 6283 6284 6285
    }
  }
  return TSDB_CODE_SUCCESS;
}

6286 6287 6288 6289 6290 6291 6292 6293 6294 6295 6296 6297 6298 6299 6300 6301 6302 6303 6304 6305
static SNode* createNullValue() {
  SValueNode* pValue = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
  if (NULL == pValue) {
    return NULL;
  }
  pValue->isNull = true;
  pValue->node.resType.type = TSDB_DATA_TYPE_NULL;
  pValue->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_NULL].bytes;
  return (SNode*)pValue;
}

static int32_t addNullTagsForExistTable(STranslateContext* pCxt, STableMeta* pMeta, SSelectStmt* pSelect) {
  int32_t numOfTags = getNumOfTags(pMeta);
  int32_t code = TSDB_CODE_SUCCESS;
  for (int32_t i = 0; TSDB_CODE_SUCCESS == code && i < numOfTags; ++i) {
    code = nodesListMakeStrictAppend(&pSelect->pTags, createNullValue());
  }
  return code;
}

6306 6307 6308 6309 6310 6311 6312 6313 6314 6315 6316 6317 6318 6319 6320 6321 6322 6323 6324 6325 6326 6327
typedef struct SRewriteSubtableCxt {
  STranslateContext* pCxt;
  SNodeList*         pPartitionList;
} SRewriteSubtableCxt;

static EDealRes rewriteSubtable(SNode** pNode, void* pContext) {
  if (QUERY_NODE_COLUMN == nodeType(*pNode)) {
    SRewriteSubtableCxt* pCxt = pContext;
    bool                 found = false;
    SNode*               pPart = NULL;
    FOREACH(pPart, pCxt->pPartitionList) {
      if (0 == strcmp(((SColumnNode*)*pNode)->colName, ((SExprNode*)pPart)->userAlias)) {
        SNode* pNew = nodesCloneNode(pPart);
        if (NULL == pNew) {
          pCxt->pCxt->errCode = TSDB_CODE_OUT_OF_MEMORY;
          return DEAL_RES_ERROR;
        }
        nodesDestroyNode(*pNode);
        *pNode = pNew;
        found = true;
        break;
      }
6328 6329 6330
    }
    if (!found) {
      return generateDealNodeErrMsg(pCxt->pCxt, TSDB_CODE_PAR_INVALID_COLUMN, ((SColumnNode*)*pNode)->colName);
6331 6332 6333 6334 6335 6336 6337 6338 6339 6340 6341 6342 6343 6344 6345 6346 6347 6348 6349 6350
    }
    return DEAL_RES_IGNORE_CHILD;
  }
  return DEAL_RES_CONTINUE;
}

static int32_t addSubtableNameToCreateStreamQuery(STranslateContext* pCxt, SCreateStreamStmt* pStmt,
                                                  SSelectStmt* pSelect) {
  if (NULL == pStmt->pSubtable) {
    return TSDB_CODE_SUCCESS;
  }
  pSelect->pSubtable = nodesCloneNode(pStmt->pSubtable);
  if (NULL == pSelect->pSubtable) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  SRewriteSubtableCxt cxt = {.pCxt = pCxt, .pPartitionList = pSelect->pPartitionByList};
  nodesRewriteExpr(&pSelect->pSubtable, rewriteSubtable, &cxt);
  return pCxt->errCode;
}

5
54liuyao 已提交
6351 6352 6353 6354 6355 6356 6357 6358 6359 6360 6361 6362 6363 6364 6365
static int32_t addNullTagsForCreateTable(STranslateContext* pCxt, SCreateStreamStmt* pStmt) {
  int32_t code = TSDB_CODE_SUCCESS;
  for (int32_t i = 0; TSDB_CODE_SUCCESS == code && i < LIST_LENGTH(pStmt->pTags); ++i) {
    code = nodesListMakeStrictAppend(&((SSelectStmt*)pStmt->pQuery)->pTags, createNullValue());
  }
  return code;
}

static int32_t addNullTagsToCreateStreamQuery(STranslateContext* pCxt, STableMeta* pMeta, SCreateStreamStmt* pStmt) {
  if (NULL == pMeta) {
    return addNullTagsForCreateTable(pCxt, pStmt);
  }
  return addNullTagsForExistTable(pCxt, pMeta, (SSelectStmt*)pStmt->pQuery);
}

6366 6367
static int32_t addSubtableInfoToCreateStreamQuery(STranslateContext* pCxt, STableMeta* pMeta,
                                                  SCreateStreamStmt* pStmt) {
6368
  int32_t      code = TSDB_CODE_SUCCESS;
6369 6370
  SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
  if (NULL == pSelect->pPartitionByList) {
5
54liuyao 已提交
6371
    code = addNullTagsToCreateStreamQuery(pCxt, pMeta, pStmt);
6372 6373
  } else {
    code = addTagsToCreateStreamQuery(pCxt, pStmt, pSelect);
6374 6375 6376 6377 6378 6379 6380 6381
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = addSubtableNameToCreateStreamQuery(pCxt, pStmt, pSelect);
  }

  return code;
}

6382 6383 6384 6385
static bool isEventWindowQuery(SSelectStmt* pSelect) {
  return NULL != pSelect->pWindow && QUERY_NODE_EVENT_WINDOW == nodeType(pSelect->pWindow);
}

6386 6387 6388 6389 6390 6391 6392 6393 6394 6395
static bool hasJsonTypeProjection(SSelectStmt* pSelect) {
  SNode* pProj = NULL;
  FOREACH(pProj, pSelect->pProjectionList) {
    if (TSDB_DATA_TYPE_JSON == ((SExprNode*)pProj)->resType.type) {
      return true;
    }
  }
  return false;
}

S
slzhou 已提交
6396
static EDealRes hasColumnOrPseudoColumn(SNode* pNode, void* pContext) {
6397 6398 6399 6400 6401 6402 6403 6404 6405 6406 6407
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
    *(bool*)pContext = true;
    return DEAL_RES_END;
  }
  if (QUERY_NODE_FUNCTION == nodeType(pNode) && fmIsPseudoColumnFunc(((SFunctionNode*)pNode)->funcId)) {
    *(bool*)pContext = true;
    return DEAL_RES_END;
  }
  return DEAL_RES_CONTINUE;
}

S
slzhou 已提交
6408 6409 6410 6411
static int32_t subtableExprHasColumnOrPseudoColumn(SNode* pNode) {
  bool hasColumn = false;
  nodesWalkExprPostOrder(pNode, hasColumnOrPseudoColumn, &hasColumn);
  return hasColumn;
6412 6413
}

6414 6415
static int32_t checkStreamQuery(STranslateContext* pCxt, SCreateStreamStmt* pStmt) {
  SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
X
Xiaoyu Wang 已提交
6416
  if (TSDB_DATA_TYPE_TIMESTAMP != ((SExprNode*)nodesListGetNode(pSelect->pProjectionList, 0))->resType.type ||
6417
      !isTimeLineQuery(pStmt->pQuery) || crossTableWithoutAggOper(pSelect) || NULL != pSelect->pOrderByList ||
6418
      crossTableWithUdaf(pSelect) || isEventWindowQuery(pSelect) || hasJsonTypeProjection(pSelect)) {
X
Xiaoyu Wang 已提交
6419 6420
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY, "Unsupported stream query");
  }
X
Xiaoyu Wang 已提交
6421 6422 6423 6424
  if (NULL != pSelect->pSubtable && TSDB_DATA_TYPE_VARCHAR != ((SExprNode*)pSelect->pSubtable)->resType.type) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY,
                                   "SUBTABLE expression must be of VARCHAR type");
  }
S
slzhou 已提交
6425
  if (NULL != pSelect->pSubtable && 0 == LIST_LENGTH(pSelect->pPartitionByList) && subtableExprHasColumnOrPseudoColumn(pSelect->pSubtable)) {
6426 6427 6428 6429
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY,
                                   "SUBTABLE expression must not has column when no partition by clause");
  }

6430 6431 6432 6433
  if (NULL == pSelect->pWindow && STREAM_TRIGGER_AT_ONCE != pStmt->pOptions->triggerType) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY,
                                   "The trigger mode of non window query can only be AT_ONCE");
  }
X
Xiaoyu Wang 已提交
6434 6435 6436
  return TSDB_CODE_SUCCESS;
}

6437 6438 6439 6440 6441 6442 6443 6444 6445
static int32_t adjustDataTypeOfProjections(STranslateContext* pCxt, const STableMeta* pMeta, SNodeList* pProjections) {
  if (getNumOfColumns(pMeta) != LIST_LENGTH(pProjections)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM, "Illegal number of columns");
  }

  SSchema* pSchemas = getTableColumnSchema(pMeta);
  int32_t  index = 0;
  SNode*   pProj = NULL;
  FOREACH(pProj, pProjections) {
6446
    SSchema*  pSchema = pSchemas + index++;
6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 6461 6462 6463 6464 6465 6466
    SDataType dt = {.type = pSchema->type, .bytes = pSchema->bytes};
    if (!dataTypeEqual(&dt, &((SExprNode*)pProj)->resType)) {
      SNode*  pFunc = NULL;
      int32_t code = createCastFunc(pCxt, pProj, dt, &pFunc);
      if (TSDB_CODE_SUCCESS != code) {
        return code;
      }
      REPLACE_NODE(pFunc);
    }
  }

  return TSDB_CODE_SUCCESS;
}

typedef struct SProjColPos {
  int32_t colId;
  SNode*  pProj;
} SProjColPos;

static int32_t projColPosCompar(const void* l, const void* r) {
6467
  return ((SProjColPos*)l)->colId > ((SProjColPos*)r)->colId;
6468 6469
}

X
Xiaoyu Wang 已提交
6470
static void projColPosDelete(void* p) { nodesDestroyNode(((SProjColPos*)p)->pProj); }
6471 6472 6473 6474 6475 6476 6477 6478 6479 6480 6481 6482 6483 6484 6485 6486 6487 6488 6489 6490 6491 6492 6493 6494

static int32_t addProjToProjColPos(STranslateContext* pCxt, const SSchema* pSchema, SNode* pProj, SArray* pProjColPos) {
  SNode* pNewProj = nodesCloneNode(pProj);
  if (NULL == pNewProj) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t   code = TSDB_CODE_SUCCESS;
  SDataType dt = {.type = pSchema->type, .bytes = pSchema->bytes};
  if (!dataTypeEqual(&dt, &((SExprNode*)pNewProj)->resType)) {
    SNode* pFunc = NULL;
    code = createCastFunc(pCxt, pNewProj, dt, &pFunc);
    pNewProj = pFunc;
  }
  if (TSDB_CODE_SUCCESS == code) {
    SProjColPos pos = {.colId = pSchema->colId, .pProj = pNewProj};
    code = (NULL == taosArrayPush(pProjColPos, &pos) ? TSDB_CODE_OUT_OF_MEMORY : TSDB_CODE_SUCCESS);
  }
  if (TSDB_CODE_SUCCESS != code) {
    nodesDestroyNode(pNewProj);
  }
  return code;
}

6495 6496 6497 6498 6499 6500 6501
static int32_t setFillNullCols(SArray* pProjColPos, const STableMeta* pMeta, SCMCreateStreamReq* pReq) {
  int32_t numOfBoundCols = taosArrayGetSize(pProjColPos);
  pReq->fillNullCols = taosArrayInit(pMeta->tableInfo.numOfColumns - numOfBoundCols, sizeof(SColLocation));
  if (NULL == pReq->fillNullCols) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  const SSchema* pSchemas = getTableColumnSchema(pMeta);
6502 6503 6504 6505 6506
  int32_t        indexOfBoundCols = 0;
  for (int32_t i = 0; i < pMeta->tableInfo.numOfColumns; ++i) {
    const SSchema* pSchema = pSchemas + i;
    if (indexOfBoundCols < numOfBoundCols) {
      SProjColPos* pPos = taosArrayGet(pProjColPos, indexOfBoundCols);
6507
      if (pSchema->colId == pPos->colId) {
6508 6509
        ++indexOfBoundCols;
        continue;
6510 6511
      }
    }
6512 6513
    SColLocation colLoc = {.colId = pSchema->colId, .slotId = i, .type = pSchema->type};
    taosArrayPush(pReq->fillNullCols, &colLoc);
6514 6515 6516 6517 6518 6519
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t adjustOrderOfProjections(STranslateContext* pCxt, SNodeList* pCols, const STableMeta* pMeta,
                                        SNodeList** pProjections, SCMCreateStreamReq* pReq) {
6520 6521 6522 6523 6524 6525 6526 6527 6528 6529
  if (LIST_LENGTH(pCols) != LIST_LENGTH(*pProjections)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM, "Illegal number of columns");
  }

  SArray* pProjColPos = taosArrayInit(LIST_LENGTH(pCols), sizeof(SProjColPos));
  if (NULL == pProjColPos) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
6530
  bool    hasPrimaryKey = false;
6531 6532 6533
  SNode*  pCol = NULL;
  SNode*  pProj = NULL;
  FORBOTH(pCol, pCols, pProj, *pProjections) {
6534
    const SSchema* pSchema = getNormalColSchema(pMeta, ((SColumnNode*)pCol)->colName);
6535 6536 6537 6538 6539 6540
    if (NULL == pSchema) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, ((SColumnNode*)pCol)->colName);
    }
    if (TSDB_CODE_SUCCESS == code) {
      code = addProjToProjColPos(pCxt, pSchema, pProj, pProjColPos);
    }
6541 6542 6543
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
X
Xiaoyu Wang 已提交
6544 6545 6546 6547 6548 6549 6550 6551
    if (PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
      hasPrimaryKey = true;
    }
  }

  if (TSDB_CODE_SUCCESS == code && !hasPrimaryKey) {
    code = generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM,
                                   "primary timestamp column can not be null");
6552 6553 6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565 6566 6567 6568
  }

  SNodeList* pNewProjections = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    taosArraySort(pProjColPos, projColPosCompar);
    int32_t num = taosArrayGetSize(pProjColPos);
    pNewProjections = nodesMakeList();
    if (NULL == pNewProjections) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
    for (int32_t i = 0; TSDB_CODE_SUCCESS == code && i < num; ++i) {
      SProjColPos* pPos = taosArrayGet(pProjColPos, i);
      code = nodesListStrictAppend(pNewProjections, pPos->pProj);
      pPos->pProj = NULL;
    }
  }

6569 6570 6571 6572
  if (TSDB_CODE_SUCCESS == code && pMeta->tableInfo.numOfColumns > LIST_LENGTH(pCols)) {
    code = setFillNullCols(pProjColPos, pMeta, pReq);
  }

6573 6574 6575 6576 6577 6578 6579 6580 6581 6582 6583 6584
  if (TSDB_CODE_SUCCESS == code) {
    taosArrayDestroy(pProjColPos);
    nodesDestroyList(*pProjections);
    *pProjections = pNewProjections;
  } else {
    taosArrayDestroyEx(pProjColPos, projColPosDelete);
    nodesDestroyList(pNewProjections);
  }

  return code;
}

6585 6586
static int32_t adjustProjectionsForExistTable(STranslateContext* pCxt, SCreateStreamStmt* pStmt,
                                              const STableMeta* pMeta, SCMCreateStreamReq* pReq) {
6587 6588 6589 6590
  SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
  if (NULL == pStmt->pCols) {
    return adjustDataTypeOfProjections(pCxt, pMeta, pSelect->pProjectionList);
  }
6591
  return adjustOrderOfProjections(pCxt, pStmt->pCols, pMeta, &pSelect->pProjectionList, pReq);
6592 6593
}

X
Xiaoyu Wang 已提交
6594 6595 6596 6597
static bool isGroupIdTagStream(const STableMeta* pMeta, SNodeList* pTags) {
  return (NULL == pTags && 1 == pMeta->tableInfo.numOfTags && TSDB_DATA_TYPE_UBIGINT == getTableTagSchema(pMeta)->type);
}

6598
static int32_t adjustDataTypeOfTags(STranslateContext* pCxt, const STableMeta* pMeta, SNodeList* pTags) {
X
Xiaoyu Wang 已提交
6599 6600 6601 6602
  if (isGroupIdTagStream(pMeta, pTags)) {
    return TSDB_CODE_SUCCESS;
  }

6603 6604 6605 6606 6607 6608 6609 6610 6611 6612 6613 6614 6615 6616 6617 6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631 6632 6633 6634 6635 6636 6637 6638 6639 6640 6641 6642 6643 6644 6645 6646 6647 6648 6649 6650 6651 6652 6653 6654 6655 6656 6657 6658 6659 6660 6661 6662 6663 6664 6665 6666 6667 6668 6669 6670 6671 6672 6673 6674 6675 6676 6677 6678 6679 6680 6681 6682 6683 6684 6685 6686 6687 6688 6689 6690 6691 6692 6693
  if (getNumOfTags(pMeta) != LIST_LENGTH(pTags)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM, "Illegal number of tags");
  }

  SSchema* pSchemas = getTableTagSchema(pMeta);
  int32_t  index = 0;
  SNode*   pTag = NULL;
  FOREACH(pTag, pTags) {
    SSchema*  pSchema = pSchemas + index++;
    SDataType dt = {.type = pSchema->type, .bytes = pSchema->bytes};
    if (!dataTypeEqual(&dt, &((SExprNode*)pTag)->resType)) {
      SNode*  pFunc = NULL;
      int32_t code = createCastFunc(pCxt, pTag, dt, &pFunc);
      if (TSDB_CODE_SUCCESS != code) {
        return code;
      }
      REPLACE_NODE(pFunc);
    }
  }

  return TSDB_CODE_SUCCESS;
}

static int32_t adjustOrderOfTags(STranslateContext* pCxt, SNodeList* pTags, const STableMeta* pMeta,
                                 SNodeList** pTagExprs, SCMCreateStreamReq* pReq) {
  if (LIST_LENGTH(pTags) != LIST_LENGTH(*pTagExprs)) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMNS_NUM, "Illegal number of tags");
  }

  SArray* pTagPos = taosArrayInit(LIST_LENGTH(pTags), sizeof(SProjColPos));
  if (NULL == pTagPos) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t code = TSDB_CODE_SUCCESS;
  SNode*  pTag = NULL;
  SNode*  pTagExpr = NULL;
  FORBOTH(pTag, pTags, pTagExpr, *pTagExprs) {
    const SSchema* pSchema = getTagSchema(pMeta, ((SColumnNode*)pTag)->colName);
    if (NULL == pSchema) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAG_NAME, ((SColumnNode*)pTag)->colName);
    }
    if (TSDB_CODE_SUCCESS == code) {
      code = addProjToProjColPos(pCxt, pSchema, pTagExpr, pTagPos);
    }
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
  }

  SNodeList* pNewTagExprs = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    taosArraySort(pTagPos, projColPosCompar);
    int32_t        indexOfBoundTags = 0;
    int32_t        numOfBoundTags = taosArrayGetSize(pTagPos);
    int32_t        numOfTags = getNumOfTags(pMeta);
    const SSchema* pTagsSchema = getTableTagSchema(pMeta);
    pNewTagExprs = nodesMakeList();
    if (NULL == pNewTagExprs) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
    for (int32_t i = 0; TSDB_CODE_SUCCESS == code && i < numOfTags; ++i) {
      const SSchema* pTagSchema = pTagsSchema + i;
      if (indexOfBoundTags < numOfBoundTags) {
        SProjColPos* pPos = taosArrayGet(pTagPos, indexOfBoundTags);
        if (pPos->colId == pTagSchema->colId) {
          ++indexOfBoundTags;
          code = nodesListStrictAppend(pNewTagExprs, pPos->pProj);
          pPos->pProj = NULL;
          continue;
        }
      }
      code = nodesListStrictAppend(pNewTagExprs, createNullValue());
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    taosArrayDestroy(pTagPos);
    nodesDestroyList(*pTagExprs);
    *pTagExprs = pNewTagExprs;
  } else {
    taosArrayDestroyEx(pTagPos, projColPosDelete);
    nodesDestroyList(pNewTagExprs);
  }

  return code;
}

static int32_t adjustTagsForExistTable(STranslateContext* pCxt, SCreateStreamStmt* pStmt, const STableMeta* pMeta,
                                       SCMCreateStreamReq* pReq) {
  SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
6694 6695 6696
  if (NULL == pSelect->pPartitionByList) {
    return TSDB_CODE_SUCCESS;
  }
6697 6698 6699 6700 6701 6702
  if (NULL == pStmt->pTags) {
    return adjustDataTypeOfTags(pCxt, pMeta, pSelect->pTags);
  }
  return adjustOrderOfTags(pCxt, pStmt->pTags, pMeta, &pSelect->pTags, pReq);
}

5
54liuyao 已提交
6703 6704 6705 6706 6707 6708 6709 6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723 6724 6725 6726 6727 6728 6729 6730 6731 6732 6733 6734 6735 6736 6737 6738 6739 6740 6741 6742 6743 6744 6745 6746 6747
static int32_t adjustTagsForCreateTable(STranslateContext* pCxt, SCreateStreamStmt* pStmt, SCMCreateStreamReq* pReq) {
  SSelectStmt* pSelect = (SSelectStmt*)pStmt->pQuery;
  if (NULL == pSelect->pPartitionByList || NULL == pSelect->pTags) {
    return TSDB_CODE_SUCCESS;
  }

  SNode* pTagDef = NULL;
  SNode* pTagExpr = NULL;
  FORBOTH(pTagDef, pStmt->pTags, pTagExpr, pSelect->pTags) {
    SColumnDefNode* pDef = (SColumnDefNode*)pTagDef;
    if (!dataTypeEqual(&pDef->dataType, &((SExprNode*)pTagExpr)->resType)) {
      SNode*  pFunc = NULL;
      int32_t code = createCastFunc(pCxt, pTagExpr, pDef->dataType, &pFunc);
      if (TSDB_CODE_SUCCESS != code) {
        return code;
      }
      REPLACE_LIST2_NODE(pFunc);
    }
  }

  return TSDB_CODE_SUCCESS;
}

static int32_t adjustTags(STranslateContext* pCxt, SCreateStreamStmt* pStmt, const STableMeta* pMeta,
                          SCMCreateStreamReq* pReq) {
  if (NULL == pMeta) {
    return adjustTagsForCreateTable(pCxt, pStmt, pReq);
  }
  return adjustTagsForExistTable(pCxt, pStmt, pMeta, pReq);
}

static bool isTagDef(SNodeList* pTags) {
  if (NULL == pTags) {
    return false;
  }
  return QUERY_NODE_COLUMN_DEF == nodeType(nodesListGetNode(pTags, 0));
}

static bool isTagBound(SNodeList* pTags) {
  if (NULL == pTags) {
    return false;
  }
  return QUERY_NODE_COLUMN == nodeType(nodesListGetNode(pTags, 0));
}

6748 6749 6750
static int32_t translateStreamTargetTable(STranslateContext* pCxt, SCreateStreamStmt* pStmt, SCMCreateStreamReq* pReq,
                                          STableMeta** pMeta) {
  int32_t code = getTableMeta(pCxt, pStmt->targetDbName, pStmt->targetTabName, pMeta);
6751
  if (TSDB_CODE_PAR_TABLE_NOT_EXIST == code) {
5
54liuyao 已提交
6752
    if (NULL != pStmt->pCols || isTagBound(pStmt->pTags)) {
6753 6754
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TABLE_NOT_EXIST, pStmt->targetTabName);
    }
6755
    pReq->createStb = STREAM_CREATE_STABLE_TRUE;
6756
    pReq->targetStbUid = 0;
6757
    return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
6758
  } else if (TSDB_CODE_SUCCESS == code) {
5
54liuyao 已提交
6759 6760 6761 6762
    if (isTagDef(pStmt->pTags)) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY, "Table already exist: %s",
                                     pStmt->targetTabName);
    }
X
Xiaoyu Wang 已提交
6763 6764 6765 6766
    if (TSDB_SUPER_TABLE != (*pMeta)->tableType) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_STREAM_QUERY,
                                     "Stream can only be written to super table");
    }
6767
    pReq->createStb = STREAM_CREATE_STABLE_FALSE;
6768
    pReq->targetStbUid = (*pMeta)->suid;
6769 6770 6771 6772
  }
  return code;
}

6773
static int32_t buildCreateStreamQuery(STranslateContext* pCxt, SCreateStreamStmt* pStmt, SCMCreateStreamReq* pReq) {
6774
  pCxt->createStream = true;
6775 6776
  STableMeta* pMeta = NULL;
  int32_t     code = translateStreamTargetTable(pCxt, pStmt, pReq, &pMeta);
6777
  if (TSDB_CODE_SUCCESS == code) {
6778
    code = addSubtableInfoToCreateStreamQuery(pCxt, pMeta, pStmt);
6779 6780 6781
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateQuery(pCxt, pStmt->pQuery);
6782
  }
X
Xiaoyu Wang 已提交
6783
  if (TSDB_CODE_SUCCESS == code) {
6784
    code = addWstartTsToCreateStreamQuery(pCxt, pStmt->pQuery);
X
Xiaoyu Wang 已提交
6785
  }
6786
  if (TSDB_CODE_SUCCESS == code) {
6787
    code = checkStreamQuery(pCxt, pStmt);
X
Xiaoyu Wang 已提交
6788
  }
6789 6790 6791
  if (TSDB_CODE_SUCCESS == code && NULL != pMeta) {
    code = adjustProjectionsForExistTable(pCxt, pStmt, pMeta, pReq);
  }
5
54liuyao 已提交
6792 6793
  if (TSDB_CODE_SUCCESS == code) {
    code = adjustTags(pCxt, pStmt, pMeta, pReq);
6794
  }
6795
  if (TSDB_CODE_SUCCESS == code) {
6796 6797
    getSourceDatabase(pStmt->pQuery, pCxt->pParseCxt->acctId, pReq->sourceDB);
    code = nodesNodeToString(pStmt->pQuery, false, &pReq->ast, NULL);
6798
  }
6799
  taosMemoryFree(pMeta);
6800 6801 6802
  return code;
}

6803 6804 6805 6806 6807 6808 6809 6810 6811 6812 6813
static int32_t translateStreamOptions(STranslateContext* pCxt, SCreateStreamStmt* pStmt) {
  pCxt->pCurrStmt = (SNode*)pStmt;
  SStreamOptions* pOptions = pStmt->pOptions;
  if ((NULL != pOptions->pWatermark && (DEAL_RES_ERROR == translateValue(pCxt, (SValueNode*)pOptions->pWatermark))) ||
      (NULL != pOptions->pDeleteMark && (DEAL_RES_ERROR == translateValue(pCxt, (SValueNode*)pOptions->pDeleteMark))) ||
      (NULL != pOptions->pDelay && (DEAL_RES_ERROR == translateValue(pCxt, (SValueNode*)pOptions->pDelay)))) {
    return pCxt->errCode;
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
6814
static int32_t buildCreateStreamReq(STranslateContext* pCxt, SCreateStreamStmt* pStmt, SCMCreateStreamReq* pReq) {
wmmhello's avatar
wmmhello 已提交
6815 6816 6817
  pReq->igExists = pStmt->ignoreExists;

  SName name;
L
Liu Jicong 已提交
6818 6819
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->streamName, strlen(pStmt->streamName));
  tNameGetFullDbName(&name, pReq->name);
wmmhello's avatar
wmmhello 已提交
6820 6821 6822 6823 6824 6825 6826

  if ('\0' != pStmt->targetTabName[0]) {
    strcpy(name.dbname, pStmt->targetDbName);
    strcpy(name.tname, pStmt->targetTabName);
    tNameExtractFullName(&name, pReq->targetStbFullName);
  }

6827
  int32_t code = buildCreateStreamQuery(pCxt, pStmt, pReq);
wmmhello's avatar
wmmhello 已提交
6828
  if (TSDB_CODE_SUCCESS == code) {
6829
    pReq->sql = taosStrdup(pCxt->pParseCxt->pSql);
wmmhello's avatar
wmmhello 已提交
6830 6831 6832 6833 6834
    if (NULL == pReq->sql) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

6835 6836 6837 6838
  if (TSDB_CODE_SUCCESS == code) {
    code = translateStreamOptions(pCxt, pStmt);
  }

wmmhello's avatar
wmmhello 已提交
6839 6840
  if (TSDB_CODE_SUCCESS == code) {
    pReq->triggerType = pStmt->pOptions->triggerType;
6841
    pReq->maxDelay = (NULL != pStmt->pOptions->pDelay ? ((SValueNode*)pStmt->pOptions->pDelay)->datum.i : 0);
wmmhello's avatar
wmmhello 已提交
6842
    pReq->watermark = (NULL != pStmt->pOptions->pWatermark ? ((SValueNode*)pStmt->pOptions->pWatermark)->datum.i : 0);
6843 6844
    pReq->deleteMark =
        (NULL != pStmt->pOptions->pDeleteMark ? ((SValueNode*)pStmt->pOptions->pDeleteMark)->datum.i : 0);
6845
    pReq->fillHistory = pStmt->pOptions->fillHistory;
6846
    pReq->igExpired = pStmt->pOptions->ignoreExpired;
6847 6848 6849 6850
    if (pReq->createStb) {
      columnDefNodeToField(pStmt->pTags, &pReq->pTags);
      pReq->numOfTags = LIST_LENGTH(pStmt->pTags);
    }
6851
    pReq->igUpdate = pStmt->pOptions->ignoreUpdate;
wmmhello's avatar
wmmhello 已提交
6852 6853 6854
  }

  return code;
X
Xiaoyu Wang 已提交
6855 6856 6857
}

static int32_t translateCreateStream(STranslateContext* pCxt, SCreateStreamStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6858
  SCMCreateStreamReq createReq = {0};
X
Xiaoyu Wang 已提交
6859

wmmhello's avatar
wmmhello 已提交
6860 6861 6862 6863 6864 6865 6866
  int32_t code = checkCreateStream(pCxt, pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateStreamReq(pCxt, pStmt, &createReq);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_STREAM, (FSerializeFunc)tSerializeSCMCreateStreamReq, &createReq);
  }
6867

wmmhello's avatar
wmmhello 已提交
6868 6869
  tFreeSCMCreateStreamReq(&createReq);
  return code;
6870 6871 6872
}

static int32_t translateDropStream(STranslateContext* pCxt, SDropStreamStmt* pStmt) {
X
Xiaoyu Wang 已提交
6873 6874 6875 6876 6877 6878
  SMDropStreamReq dropReq = {0};
  SName           name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->streamName, strlen(pStmt->streamName));
  tNameGetFullDbName(&name, dropReq.name);
  dropReq.igNotExists = pStmt->ignoreNotExists;
  return buildCmdMsg(pCxt, TDMT_MND_DROP_STREAM, (FSerializeFunc)tSerializeSMDropStreamReq, &dropReq);
6879
}
6880

6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894 6895 6896 6897 6898 6899
static int32_t translatePauseStream(STranslateContext* pCxt, SPauseStreamStmt* pStmt) {
  SMPauseStreamReq req = {0};
  SName            name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->streamName, strlen(pStmt->streamName));
  tNameGetFullDbName(&name, req.name);
  req.igNotExists = pStmt->ignoreNotExists;
  return buildCmdMsg(pCxt, TDMT_MND_PAUSE_STREAM, (FSerializeFunc)tSerializeSMPauseStreamReq, &req);
}

static int32_t translateResumeStream(STranslateContext* pCxt, SResumeStreamStmt* pStmt) {
  SMResumeStreamReq req = {0};
  SName             name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->streamName, strlen(pStmt->streamName));
  tNameGetFullDbName(&name, req.name);
  req.igNotExists = pStmt->ignoreNotExists;
  req.igUntreated = pStmt->ignoreUntreated;
  return buildCmdMsg(pCxt, TDMT_MND_RESUME_STREAM, (FSerializeFunc)tSerializeSMResumeStreamReq, &req);
}

L
Liu Jicong 已提交
6900
static int32_t readFromFile(char* pName, int32_t* len, char** buf) {
wmmhello's avatar
wmmhello 已提交
6901 6902 6903 6904 6905 6906 6907 6908 6909 6910 6911 6912 6913 6914 6915 6916 6917 6918 6919 6920 6921 6922 6923 6924 6925 6926
  int64_t filesize = 0;
  if (taosStatFile(pName, &filesize, NULL) < 0) {
    return TAOS_SYSTEM_ERROR(errno);
  }

  *len = filesize;

  if (*len <= 0) {
    return TSDB_CODE_TSC_FILE_EMPTY;
  }

  *buf = taosMemoryCalloc(1, *len);
  if (*buf == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  TdFilePtr tfile = taosOpenFile(pName, O_RDONLY | O_BINARY);
  if (NULL == tfile) {
    taosMemoryFreeClear(*buf);
    return TAOS_SYSTEM_ERROR(errno);
  }

  int64_t s = taosReadFile(tfile, *buf, *len);
  if (s != *len) {
    taosCloseFile(&tfile);
    taosMemoryFreeClear(*buf);
S
Shengliang Guan 已提交
6927
    return TSDB_CODE_APP_ERROR;
wmmhello's avatar
wmmhello 已提交
6928 6929 6930
  }
  taosCloseFile(&tfile);
  return TSDB_CODE_SUCCESS;
6931 6932 6933
}

static int32_t translateCreateFunction(STranslateContext* pCxt, SCreateFunctionStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6934 6935 6936
  if (fmIsBuiltinFunc(pStmt->funcName)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_FUNCTION_NAME);
  }
6937

X
Xiaoyu Wang 已提交
6938 6939
  if (TSDB_DATA_TYPE_JSON == pStmt->outputDt.type || TSDB_DATA_TYPE_VARBINARY == pStmt->outputDt.type ||
      TSDB_DATA_TYPE_DECIMAL == pStmt->outputDt.type || TSDB_DATA_TYPE_BLOB == pStmt->outputDt.type ||
6940 6941 6942 6943
      TSDB_DATA_TYPE_MEDIUMBLOB == pStmt->outputDt.type) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR, "Unsupported output type for UDF");
  }

X
Xiaoyu Wang 已提交
6944 6945 6946 6947
  if (!pStmt->isAgg && pStmt->bufSize > 0) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR, "BUFSIZE can only be used with UDAF");
  }

wmmhello's avatar
wmmhello 已提交
6948 6949
  SCreateFuncReq req = {0};
  strcpy(req.name, pStmt->funcName);
6950
  req.orReplace = pStmt->orReplace;
wmmhello's avatar
wmmhello 已提交
6951 6952
  req.igExists = pStmt->ignoreExists;
  req.funcType = pStmt->isAgg ? TSDB_FUNC_TYPE_AGGREGATE : TSDB_FUNC_TYPE_SCALAR;
6953
  req.scriptType = pStmt->language;
wmmhello's avatar
wmmhello 已提交
6954
  req.outputType = pStmt->outputDt.type;
S
slzhou 已提交
6955
  pStmt->outputDt.bytes = calcTypeBytes(pStmt->outputDt);
wmmhello's avatar
wmmhello 已提交
6956 6957 6958 6959 6960 6961
  req.outputLen = pStmt->outputDt.bytes;
  req.bufSize = pStmt->bufSize;
  int32_t code = readFromFile(pStmt->libraryPath, &req.codeLen, &req.pCode);
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_CREATE_FUNC, (FSerializeFunc)tSerializeSCreateFuncReq, &req);
  }
X
Xiaoyu Wang 已提交
6962
  tFreeSCreateFuncReq(&req);
wmmhello's avatar
wmmhello 已提交
6963
  return code;
6964 6965
}

S
slzhou 已提交
6966
static int32_t translateDropFunction(STranslateContext* pCxt, SDropFunctionStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
6967 6968 6969 6970
  SDropFuncReq req = {0};
  strcpy(req.name, pStmt->funcName);
  req.igNotExists = pStmt->ignoreNotExists;
  return buildCmdMsg(pCxt, TDMT_MND_DROP_FUNC, (FSerializeFunc)tSerializeSDropFuncReq, &req);
S
slzhou 已提交
6971 6972
}

X
Xiaoyu Wang 已提交
6973 6974 6975 6976 6977 6978 6979 6980 6981 6982 6983 6984 6985
static int32_t createRealTableForGrantTable(SGrantStmt* pStmt, SRealTableNode** pTable) {
  SRealTableNode* pRealTable = (SRealTableNode*)nodesMakeNode(QUERY_NODE_REAL_TABLE);
  if (NULL == pRealTable) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  strcpy(pRealTable->table.dbName, pStmt->objName);
  strcpy(pRealTable->table.tableName, pStmt->tabName);
  strcpy(pRealTable->table.tableAlias, pStmt->tabName);
  *pTable = pRealTable;
  return TSDB_CODE_SUCCESS;
}

static int32_t translateGrantTagCond(STranslateContext* pCxt, SGrantStmt* pStmt, SAlterUserReq* pReq) {
D
dapan1121 已提交
6986
  SRealTableNode* pTable = NULL;
X
Xiaoyu Wang 已提交
6987
  if ('\0' == pStmt->tabName[0] || '*' == pStmt->tabName[0]) {
D
dapan1121 已提交
6988 6989 6990 6991 6992 6993
    if (pStmt->pTagCond) {
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR,
                                     "The With clause can only be used for table level privilege");
    } else {
      return TSDB_CODE_SUCCESS;
    }
X
Xiaoyu Wang 已提交
6994 6995 6996 6997 6998 6999 7000
  }

  int32_t         code = createRealTableForGrantTable(pStmt, &pTable);
  if (TSDB_CODE_SUCCESS == code) {
    SName name;
    code = getTableMetaImpl(pCxt, toName(pCxt->pParseCxt->acctId, pTable->table.dbName, pTable->table.tableName, &name),
                            &(pTable->pMeta));
D
dapan1121 已提交
7001 7002 7003 7004 7005 7006 7007 7008 7009 7010
    if (code) {
      nodesDestroyNode((SNode*)pTable);
      return code;
    }

    if (TSDB_SUPER_TABLE != pTable->pMeta->tableType && TSDB_NORMAL_TABLE != pTable->pMeta->tableType) {
      nodesDestroyNode((SNode*)pTable);
      return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_SYNTAX_ERROR,
                                     "Only supertable and normal table can be granted");
    }
X
Xiaoyu Wang 已提交
7011
  }
D
dapan1121 已提交
7012 7013 7014

  if (TSDB_CODE_SUCCESS == code && NULL == pStmt->pTagCond) {
    nodesDestroyNode((SNode*)pTable);
X
Xiaoyu Wang 已提交
7015 7016 7017 7018
    return TSDB_CODE_SUCCESS;
  }

  pCxt->pCurrStmt = (SNode*)pStmt;
D
dapan1121 已提交
7019

X
Xiaoyu Wang 已提交
7020 7021 7022 7023 7024 7025 7026 7027 7028 7029 7030 7031 7032
  if (TSDB_CODE_SUCCESS == code) {
    code = addNamespace(pCxt, pTable);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateExpr(pCxt, &pStmt->pTagCond);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesNodeToString(pStmt->pTagCond, false, &pReq->tagCond, &pReq->tagCondLen);
  }
  nodesDestroyNode((SNode*)pTable);
  return code;
}

7033
static int32_t translateGrant(STranslateContext* pCxt, SGrantStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
7034
  SAlterUserReq req = {0};
X
Xiaoyu Wang 已提交
7035 7036 7037
  if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_ALL) ||
      (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_READ) &&
       BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_WRITE))) {
7038
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_ADD_ALL_DB : TSDB_ALTER_USER_ADD_ALL_TABLE);
X
Xiaoyu Wang 已提交
7039
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_READ)) {
7040
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_ADD_READ_DB : TSDB_ALTER_USER_ADD_READ_TABLE);
X
Xiaoyu Wang 已提交
7041
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_WRITE)) {
7042
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_ADD_WRITE_DB : TSDB_ALTER_USER_ADD_WRITE_TABLE);
X
Xiaoyu Wang 已提交
7043
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_SUBSCRIBE)) {
7044
    req.alterType = TSDB_ALTER_USER_ADD_SUBSCRIBE_TOPIC;
wmmhello's avatar
wmmhello 已提交
7045 7046
  }
  strcpy(req.user, pStmt->userName);
7047
  sprintf(req.objname, "%d.%s", pCxt->pParseCxt->acctId, pStmt->objName);
X
Xiaoyu Wang 已提交
7048
  sprintf(req.tabName, "%s", pStmt->tabName);
X
Xiaoyu Wang 已提交
7049
  int32_t code = translateGrantTagCond(pCxt, pStmt, &req);
X
Xiaoyu Wang 已提交
7050 7051 7052
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCmdMsg(pCxt, TDMT_MND_ALTER_USER, (FSerializeFunc)tSerializeSAlterUserReq, &req);
  }
X
Xiaoyu Wang 已提交
7053
  tFreeSAlterUserReq(&req);
X
Xiaoyu Wang 已提交
7054
  return code;
7055 7056 7057
}

static int32_t translateRevoke(STranslateContext* pCxt, SRevokeStmt* pStmt) {
wmmhello's avatar
wmmhello 已提交
7058
  SAlterUserReq req = {0};
X
Xiaoyu Wang 已提交
7059 7060 7061
  if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_ALL) ||
      (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_READ) &&
       BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_WRITE))) {
7062
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_REMOVE_ALL_DB : TSDB_ALTER_USER_REMOVE_ALL_TABLE);
X
Xiaoyu Wang 已提交
7063
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_READ)) {
7064
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_REMOVE_READ_DB : TSDB_ALTER_USER_REMOVE_READ_TABLE);
X
Xiaoyu Wang 已提交
7065
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_WRITE)) {
7066
    req.alterType = ('\0' == pStmt->tabName[0] ? TSDB_ALTER_USER_REMOVE_WRITE_DB : TSDB_ALTER_USER_REMOVE_WRITE_TABLE);
X
Xiaoyu Wang 已提交
7067
  } else if (BIT_FLAG_TEST_MASK(pStmt->privileges, PRIVILEGE_TYPE_SUBSCRIBE)) {
7068
    req.alterType = TSDB_ALTER_USER_REMOVE_SUBSCRIBE_TOPIC;
wmmhello's avatar
wmmhello 已提交
7069 7070
  }
  strcpy(req.user, pStmt->userName);
7071
  sprintf(req.objname, "%d.%s", pCxt->pParseCxt->acctId, pStmt->objName);
X
Xiaoyu Wang 已提交
7072
  sprintf(req.tabName, "%s", pStmt->tabName);
wmmhello's avatar
wmmhello 已提交
7073
  return buildCmdMsg(pCxt, TDMT_MND_ALTER_USER, (FSerializeFunc)tSerializeSAlterUserReq, &req);
7074 7075
}

X
Xiaoyu Wang 已提交
7076 7077 7078 7079 7080
static int32_t translateBalanceVgroup(STranslateContext* pCxt, SBalanceVgroupStmt* pStmt) {
  SBalanceVgroupReq req = {0};
  return buildCmdMsg(pCxt, TDMT_MND_BALANCE_VGROUP, (FSerializeFunc)tSerializeSBalanceVgroupReq, &req);
}

C
cadem 已提交
7081 7082 7083 7084 7085
static int32_t translateBalanceVgroupLeader(STranslateContext* pCxt, SBalanceVgroupLeaderStmt* pStmt) {
  SBalanceVgroupLeaderReq req = {0};
  return buildCmdMsg(pCxt, TDMT_MND_BALANCE_VGROUP_LEADER, (FSerializeFunc)tSerializeSBalanceVgroupLeaderReq, &req);
}

X
Xiaoyu Wang 已提交
7086 7087 7088 7089 7090
static int32_t translateMergeVgroup(STranslateContext* pCxt, SMergeVgroupStmt* pStmt) {
  SMergeVgroupReq req = {.vgId1 = pStmt->vgId1, .vgId2 = pStmt->vgId2};
  return buildCmdMsg(pCxt, TDMT_MND_MERGE_VGROUP, (FSerializeFunc)tSerializeSMergeVgroupReq, &req);
}

X
Xiaoyu Wang 已提交
7091 7092 7093 7094 7095 7096 7097 7098 7099 7100 7101 7102 7103 7104 7105 7106 7107 7108 7109 7110 7111 7112 7113 7114 7115 7116 7117
static int32_t checkDnodeIds(STranslateContext* pCxt, SRedistributeVgroupStmt* pStmt) {
  int32_t numOfDnodes = LIST_LENGTH(pStmt->pDnodes);
  if (numOfDnodes > 3 || numOfDnodes < 1) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_REDISTRIBUTE_VG);
  }

  SNode* pNode = NULL;
  FOREACH(pNode, pStmt->pDnodes) {
    SValueNode* pVal = (SValueNode*)pNode;
    if (DEAL_RES_ERROR == translateValue(pCxt, pVal)) {
      return pCxt->errCode;
    }
  }

  pStmt->dnodeId1 = getBigintFromValueNode((SValueNode*)nodesListGetNode(pStmt->pDnodes, 0));
  pStmt->dnodeId2 = -1;
  pStmt->dnodeId3 = -1;
  if (numOfDnodes > 1) {
    pStmt->dnodeId2 = getBigintFromValueNode((SValueNode*)nodesListGetNode(pStmt->pDnodes, 1));
  }
  if (numOfDnodes > 2) {
    pStmt->dnodeId3 = getBigintFromValueNode((SValueNode*)nodesListGetNode(pStmt->pDnodes, 2));
  }

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
7118 7119
static int32_t translateRedistributeVgroup(STranslateContext* pCxt, SRedistributeVgroupStmt* pStmt) {
  SRedistributeVgroupReq req = {.vgId = pStmt->vgId};
X
Xiaoyu Wang 已提交
7120 7121 7122 7123 7124 7125 7126 7127
  int32_t                code = checkDnodeIds(pCxt, pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    req.dnodeId1 = pStmt->dnodeId1;
    req.dnodeId2 = pStmt->dnodeId2;
    req.dnodeId3 = pStmt->dnodeId3;
    code = buildCmdMsg(pCxt, TDMT_MND_REDISTRIBUTE_VGROUP, (FSerializeFunc)tSerializeSRedistributeVgroupReq, &req);
  }
  return code;
X
Xiaoyu Wang 已提交
7128 7129
}

7130 7131 7132 7133 7134
static int32_t translateSplitVgroup(STranslateContext* pCxt, SSplitVgroupStmt* pStmt) {
  SSplitVgroupReq req = {.vgId = pStmt->vgId};
  return buildCmdMsg(pCxt, TDMT_MND_SPLIT_VGROUP, (FSerializeFunc)tSerializeSSplitVgroupReq, &req);
}

X
Xiaoyu Wang 已提交
7135 7136 7137 7138 7139
static int32_t translateShowVariables(STranslateContext* pCxt, SShowStmt* pStmt) {
  SShowVariablesReq req = {0};
  return buildCmdMsg(pCxt, TDMT_MND_SHOW_VARIABLES, (FSerializeFunc)tSerializeSShowVariablesReq, &req);
}

7140 7141 7142 7143 7144
static int32_t translateShowCreateDatabase(STranslateContext* pCxt, SShowCreateDatabaseStmt* pStmt) {
  pStmt->pCfg = taosMemoryCalloc(1, sizeof(SDbCfgInfo));
  if (NULL == pStmt->pCfg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
7145

7146 7147 7148
  SName name;
  tNameSetDbName(&name, pCxt->pParseCxt->acctId, pStmt->dbName, strlen(pStmt->dbName));
  tNameGetFullDbName(&name, pStmt->dbFName);
X
Xiaoyu Wang 已提交
7149

7150 7151 7152 7153
  return getDBCfg(pCxt, pStmt->dbName, (SDbCfgInfo*)pStmt->pCfg);
}

static int32_t translateShowCreateTable(STranslateContext* pCxt, SShowCreateTableStmt* pStmt) {
X
Xiaoyu Wang 已提交
7154 7155 7156 7157 7158 7159 7160 7161 7162 7163 7164
  pStmt->pDbCfg = taosMemoryCalloc(1, sizeof(SDbCfgInfo));
  if (NULL == pStmt->pDbCfg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  int32_t code = getDBCfg(pCxt, pStmt->dbName, (SDbCfgInfo*)pStmt->pDbCfg);
  if (TSDB_CODE_SUCCESS == code) {
    SName name;
    toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &name);
    code = getTableCfg(pCxt, &name, (STableCfg**)&pStmt->pTableCfg);
  }
  return code;
7165 7166
}

7167
static int32_t translateQuery(STranslateContext* pCxt, SNode* pNode) {
wmmhello's avatar
wmmhello 已提交
7168 7169 7170 7171 7172 7173 7174 7175
  int32_t code = TSDB_CODE_SUCCESS;
  switch (nodeType(pNode)) {
    case QUERY_NODE_SELECT_STMT:
      code = translateSelect(pCxt, (SSelectStmt*)pNode);
      break;
    case QUERY_NODE_SET_OPERATOR:
      code = translateSetOperator(pCxt, (SSetOperator*)pNode);
      break;
X
Xiaoyu Wang 已提交
7176 7177 7178
    case QUERY_NODE_DELETE_STMT:
      code = translateDelete(pCxt, (SDeleteStmt*)pNode);
      break;
7179 7180 7181
    case QUERY_NODE_INSERT_STMT:
      code = translateInsert(pCxt, (SInsertStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7182 7183 7184 7185 7186 7187 7188 7189 7190
    case QUERY_NODE_CREATE_DATABASE_STMT:
      code = translateCreateDatabase(pCxt, (SCreateDatabaseStmt*)pNode);
      break;
    case QUERY_NODE_DROP_DATABASE_STMT:
      code = translateDropDatabase(pCxt, (SDropDatabaseStmt*)pNode);
      break;
    case QUERY_NODE_ALTER_DATABASE_STMT:
      code = translateAlterDatabase(pCxt, (SAlterDatabaseStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7191 7192 7193
    case QUERY_NODE_TRIM_DATABASE_STMT:
      code = translateTrimDatabase(pCxt, (STrimDatabaseStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7194 7195 7196 7197 7198 7199 7200 7201 7202 7203
    case QUERY_NODE_CREATE_TABLE_STMT:
      code = translateCreateSuperTable(pCxt, (SCreateTableStmt*)pNode);
      break;
    case QUERY_NODE_DROP_TABLE_STMT:
      code = translateDropTable(pCxt, (SDropTableStmt*)pNode);
      break;
    case QUERY_NODE_DROP_SUPER_TABLE_STMT:
      code = translateDropSuperTable(pCxt, (SDropSuperTableStmt*)pNode);
      break;
    case QUERY_NODE_ALTER_TABLE_STMT:
7204
    case QUERY_NODE_ALTER_SUPER_TABLE_STMT:
X
Xiaoyu Wang 已提交
7205
      code = translateAlterSuperTable(pCxt, (SAlterTableStmt*)pNode);
wmmhello's avatar
wmmhello 已提交
7206 7207 7208 7209 7210 7211 7212 7213 7214 7215 7216 7217 7218 7219 7220 7221 7222 7223 7224 7225 7226 7227 7228 7229 7230 7231 7232 7233 7234 7235 7236 7237 7238 7239 7240 7241 7242 7243 7244 7245 7246 7247 7248 7249 7250 7251
      break;
    case QUERY_NODE_CREATE_USER_STMT:
      code = translateCreateUser(pCxt, (SCreateUserStmt*)pNode);
      break;
    case QUERY_NODE_ALTER_USER_STMT:
      code = translateAlterUser(pCxt, (SAlterUserStmt*)pNode);
      break;
    case QUERY_NODE_DROP_USER_STMT:
      code = translateDropUser(pCxt, (SDropUserStmt*)pNode);
      break;
    case QUERY_NODE_USE_DATABASE_STMT:
      code = translateUseDatabase(pCxt, (SUseDatabaseStmt*)pNode);
      break;
    case QUERY_NODE_CREATE_DNODE_STMT:
      code = translateCreateDnode(pCxt, (SCreateDnodeStmt*)pNode);
      break;
    case QUERY_NODE_DROP_DNODE_STMT:
      code = translateDropDnode(pCxt, (SDropDnodeStmt*)pNode);
      break;
    case QUERY_NODE_ALTER_DNODE_STMT:
      code = translateAlterDnode(pCxt, (SAlterDnodeStmt*)pNode);
      break;
    case QUERY_NODE_CREATE_INDEX_STMT:
      code = translateCreateIndex(pCxt, (SCreateIndexStmt*)pNode);
      break;
    case QUERY_NODE_DROP_INDEX_STMT:
      code = translateDropIndex(pCxt, (SDropIndexStmt*)pNode);
      break;
    case QUERY_NODE_CREATE_QNODE_STMT:
    case QUERY_NODE_CREATE_BNODE_STMT:
    case QUERY_NODE_CREATE_SNODE_STMT:
    case QUERY_NODE_CREATE_MNODE_STMT:
      code = translateCreateComponentNode(pCxt, (SCreateComponentNodeStmt*)pNode);
      break;
    case QUERY_NODE_DROP_QNODE_STMT:
    case QUERY_NODE_DROP_BNODE_STMT:
    case QUERY_NODE_DROP_SNODE_STMT:
    case QUERY_NODE_DROP_MNODE_STMT:
      code = translateDropComponentNode(pCxt, (SDropComponentNodeStmt*)pNode);
      break;
    case QUERY_NODE_CREATE_TOPIC_STMT:
      code = translateCreateTopic(pCxt, (SCreateTopicStmt*)pNode);
      break;
    case QUERY_NODE_DROP_TOPIC_STMT:
      code = translateDropTopic(pCxt, (SDropTopicStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7252 7253 7254
    case QUERY_NODE_DROP_CGROUP_STMT:
      code = translateDropCGroup(pCxt, (SDropCGroupStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7255 7256 7257 7258 7259 7260 7261 7262 7263
    case QUERY_NODE_ALTER_LOCAL_STMT:
      code = translateAlterLocal(pCxt, (SAlterLocalStmt*)pNode);
      break;
    case QUERY_NODE_EXPLAIN_STMT:
      code = translateExplain(pCxt, (SExplainStmt*)pNode);
      break;
    case QUERY_NODE_DESCRIBE_STMT:
      code = translateDescribe(pCxt, (SDescribeStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7264 7265 7266
    case QUERY_NODE_COMPACT_DATABASE_STMT:
      code = translateCompact(pCxt, (SCompactDatabaseStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7267 7268 7269 7270
    case QUERY_NODE_KILL_CONNECTION_STMT:
      code = translateKillConnection(pCxt, (SKillStmt*)pNode);
      break;
    case QUERY_NODE_KILL_QUERY_STMT:
X
Xiaoyu Wang 已提交
7271
      code = translateKillQuery(pCxt, (SKillQueryStmt*)pNode);
wmmhello's avatar
wmmhello 已提交
7272 7273 7274 7275 7276 7277 7278 7279 7280 7281
      break;
    case QUERY_NODE_KILL_TRANSACTION_STMT:
      code = translateKillTransaction(pCxt, (SKillStmt*)pNode);
      break;
    case QUERY_NODE_CREATE_STREAM_STMT:
      code = translateCreateStream(pCxt, (SCreateStreamStmt*)pNode);
      break;
    case QUERY_NODE_DROP_STREAM_STMT:
      code = translateDropStream(pCxt, (SDropStreamStmt*)pNode);
      break;
7282 7283 7284 7285 7286 7287
    case QUERY_NODE_PAUSE_STREAM_STMT:
      code = translatePauseStream(pCxt, (SPauseStreamStmt*)pNode);
      break;
    case QUERY_NODE_RESUME_STREAM_STMT:
      code = translateResumeStream(pCxt, (SResumeStreamStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7288 7289 7290 7291 7292 7293 7294 7295 7296 7297 7298 7299
    case QUERY_NODE_CREATE_FUNCTION_STMT:
      code = translateCreateFunction(pCxt, (SCreateFunctionStmt*)pNode);
      break;
    case QUERY_NODE_DROP_FUNCTION_STMT:
      code = translateDropFunction(pCxt, (SDropFunctionStmt*)pNode);
      break;
    case QUERY_NODE_GRANT_STMT:
      code = translateGrant(pCxt, (SGrantStmt*)pNode);
      break;
    case QUERY_NODE_REVOKE_STMT:
      code = translateRevoke(pCxt, (SRevokeStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7300 7301 7302
    case QUERY_NODE_BALANCE_VGROUP_STMT:
      code = translateBalanceVgroup(pCxt, (SBalanceVgroupStmt*)pNode);
      break;
C
cadem 已提交
7303 7304 7305
    case QUERY_NODE_BALANCE_VGROUP_LEADER_STMT:
      code = translateBalanceVgroupLeader(pCxt, (SBalanceVgroupLeaderStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7306 7307 7308 7309 7310 7311
    case QUERY_NODE_MERGE_VGROUP_STMT:
      code = translateMergeVgroup(pCxt, (SMergeVgroupStmt*)pNode);
      break;
    case QUERY_NODE_REDISTRIBUTE_VGROUP_STMT:
      code = translateRedistributeVgroup(pCxt, (SRedistributeVgroupStmt*)pNode);
      break;
7312 7313 7314
    case QUERY_NODE_SPLIT_VGROUP_STMT:
      code = translateSplitVgroup(pCxt, (SSplitVgroupStmt*)pNode);
      break;
X
Xiaoyu Wang 已提交
7315 7316 7317
    case QUERY_NODE_SHOW_VARIABLES_STMT:
      code = translateShowVariables(pCxt, (SShowStmt*)pNode);
      break;
7318 7319 7320 7321 7322 7323 7324
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
      code = translateShowCreateDatabase(pCxt, (SShowCreateDatabaseStmt*)pNode);
      break;
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
      code = translateShowCreateTable(pCxt, (SShowCreateTableStmt*)pNode);
      break;
D
dapan1121 已提交
7325 7326 7327 7328 7329 7330
    case QUERY_NODE_RESTORE_DNODE_STMT:
    case QUERY_NODE_RESTORE_QNODE_STMT:
    case QUERY_NODE_RESTORE_MNODE_STMT:
    case QUERY_NODE_RESTORE_VNODE_STMT:
      code = translateRestoreDnode(pCxt, (SRestoreComponentNodeStmt*)pNode);
      break;
wmmhello's avatar
wmmhello 已提交
7331 7332 7333 7334
    default:
      break;
  }
  return code;
7335 7336 7337
}

static int32_t translateSubquery(STranslateContext* pCxt, SNode* pNode) {
7338 7339
  ESqlClause currClause = pCxt->currClause;
  SNode*     pCurrStmt = pCxt->pCurrStmt;
7340 7341
  int32_t    currLevel = pCxt->currLevel;
  pCxt->currLevel = ++(pCxt->levelNo);
X
Xiaoyu Wang 已提交
7342
  int32_t code = translateQuery(pCxt, pNode);
wmmhello's avatar
wmmhello 已提交
7343
  pCxt->currClause = currClause;
7344
  pCxt->pCurrStmt = pCurrStmt;
7345
  pCxt->currLevel = currLevel;
wmmhello's avatar
wmmhello 已提交
7346
  return code;
7347 7348
}

X
Xiaoyu Wang 已提交
7349
static int32_t extractQueryResultSchema(const SNodeList* pProjections, int32_t* numOfCols, SSchema** pSchema) {
wmmhello's avatar
wmmhello 已提交
7350 7351 7352 7353 7354 7355 7356 7357 7358 7359
  *numOfCols = LIST_LENGTH(pProjections);
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  SNode*  pNode;
  int32_t index = 0;
  FOREACH(pNode, pProjections) {
    SExprNode* pExpr = (SExprNode*)pNode;
7360 7361 7362 7363 7364 7365 7366
    if (TSDB_DATA_TYPE_NULL == pExpr->resType.type) {
      (*pSchema)[index].type = TSDB_DATA_TYPE_VARCHAR;
      (*pSchema)[index].bytes = 0;
    } else {
      (*pSchema)[index].type = pExpr->resType.type;
      (*pSchema)[index].bytes = pExpr->resType.bytes;
    }
wmmhello's avatar
wmmhello 已提交
7367 7368 7369 7370 7371 7372 7373 7374 7375 7376
    (*pSchema)[index].colId = index + 1;
    if ('\0' != pExpr->userAlias[0]) {
      strcpy((*pSchema)[index].name, pExpr->userAlias);
    } else {
      strcpy((*pSchema)[index].name, pExpr->aliasName);
    }
    index += 1;
  }

  return TSDB_CODE_SUCCESS;
7377 7378
}

H
Hongze Cheng 已提交
7379
static int8_t extractResultTsPrecision(const SSelectStmt* pSelect) { return pSelect->precision; }
7380

7381
static int32_t extractExplainResultSchema(int32_t* numOfCols, SSchema** pSchema) {
wmmhello's avatar
wmmhello 已提交
7382 7383 7384 7385 7386 7387 7388 7389 7390
  *numOfCols = 1;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  (*pSchema)[0].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[0].bytes = TSDB_EXPLAIN_RESULT_ROW_SIZE;
  strcpy((*pSchema)[0].name, TSDB_EXPLAIN_RESULT_COLUMN_NAME);
  return TSDB_CODE_SUCCESS;
7391 7392 7393
}

static int32_t extractDescribeResultSchema(int32_t* numOfCols, SSchema** pSchema) {
wmmhello's avatar
wmmhello 已提交
7394 7395 7396 7397 7398
  *numOfCols = DESCRIBE_RESULT_COLS;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
7399

wmmhello's avatar
wmmhello 已提交
7400 7401 7402
  (*pSchema)[0].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[0].bytes = DESCRIBE_RESULT_FIELD_LEN;
  strcpy((*pSchema)[0].name, "field");
7403

wmmhello's avatar
wmmhello 已提交
7404 7405 7406
  (*pSchema)[1].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[1].bytes = DESCRIBE_RESULT_TYPE_LEN;
  strcpy((*pSchema)[1].name, "type");
7407

wmmhello's avatar
wmmhello 已提交
7408 7409 7410
  (*pSchema)[2].type = TSDB_DATA_TYPE_INT;
  (*pSchema)[2].bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
  strcpy((*pSchema)[2].name, "length");
7411

wmmhello's avatar
wmmhello 已提交
7412 7413 7414
  (*pSchema)[3].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[3].bytes = DESCRIBE_RESULT_NOTE_LEN;
  strcpy((*pSchema)[3].name, "note");
7415

wmmhello's avatar
wmmhello 已提交
7416
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
7417 7418
}

7419 7420 7421 7422 7423 7424 7425 7426 7427 7428 7429 7430 7431 7432 7433 7434 7435 7436
static int32_t extractShowCreateDatabaseResultSchema(int32_t* numOfCols, SSchema** pSchema) {
  *numOfCols = 2;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  (*pSchema)[0].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[0].bytes = TSDB_DB_NAME_LEN;
  strcpy((*pSchema)[0].name, "Database");

  (*pSchema)[1].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[1].bytes = TSDB_MAX_BINARY_LEN;
  strcpy((*pSchema)[1].name, "Create Database");

  return TSDB_CODE_SUCCESS;
}

7437 7438 7439 7440 7441 7442 7443 7444 7445 7446 7447 7448 7449 7450
static int32_t extractShowAliveResultSchema(int32_t* numOfCols, SSchema** pSchema) {
  *numOfCols = 1;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  (*pSchema)[0].type = TSDB_DATA_TYPE_INT;
  (*pSchema)[0].bytes = sizeof(int32_t);
  strcpy((*pSchema)[0].name, "status");

  return TSDB_CODE_SUCCESS;
}

7451 7452 7453 7454 7455 7456 7457 7458
static int32_t extractShowCreateTableResultSchema(int32_t* numOfCols, SSchema** pSchema) {
  *numOfCols = 2;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  (*pSchema)[0].type = TSDB_DATA_TYPE_BINARY;
7459
  (*pSchema)[0].bytes = SHOW_CREATE_TB_RESULT_FIELD1_LEN;
7460 7461 7462
  strcpy((*pSchema)[0].name, "Table");

  (*pSchema)[1].type = TSDB_DATA_TYPE_BINARY;
7463
  (*pSchema)[1].bytes = SHOW_CREATE_TB_RESULT_FIELD2_LEN;
7464 7465 7466 7467 7468
  strcpy((*pSchema)[1].name, "Create Table");

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
7469
static int32_t extractShowVariablesResultSchema(int32_t* numOfCols, SSchema** pSchema) {
D
dapan1121 已提交
7470 7471 7472 7473 7474 7475 7476 7477 7478 7479 7480 7481 7482 7483 7484 7485 7486
  *numOfCols = 2;
  *pSchema = taosMemoryCalloc((*numOfCols), sizeof(SSchema));
  if (NULL == (*pSchema)) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  (*pSchema)[0].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[0].bytes = TSDB_CONFIG_OPTION_LEN;
  strcpy((*pSchema)[0].name, "name");

  (*pSchema)[1].type = TSDB_DATA_TYPE_BINARY;
  (*pSchema)[1].bytes = TSDB_CONFIG_VALUE_LEN;
  strcpy((*pSchema)[1].name, "value");

  return TSDB_CODE_SUCCESS;
}

7487
int32_t extractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema) {
wmmhello's avatar
wmmhello 已提交
7488 7489 7490
  if (NULL == pRoot) {
    return TSDB_CODE_SUCCESS;
  }
7491

wmmhello's avatar
wmmhello 已提交
7492 7493 7494 7495 7496 7497 7498 7499
  switch (nodeType(pRoot)) {
    case QUERY_NODE_SELECT_STMT:
    case QUERY_NODE_SET_OPERATOR:
      return extractQueryResultSchema(getProjectList(pRoot), numOfCols, pSchema);
    case QUERY_NODE_EXPLAIN_STMT:
      return extractExplainResultSchema(numOfCols, pSchema);
    case QUERY_NODE_DESCRIBE_STMT:
      return extractDescribeResultSchema(numOfCols, pSchema);
7500 7501
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
      return extractShowCreateDatabaseResultSchema(numOfCols, pSchema);
7502 7503
    case QUERY_NODE_SHOW_DB_ALIVE_STMT:
    case QUERY_NODE_SHOW_CLUSTER_ALIVE_STMT:
7504
      return extractShowAliveResultSchema(numOfCols, pSchema);
7505 7506 7507
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
      return extractShowCreateTableResultSchema(numOfCols, pSchema);
D
dapan1121 已提交
7508
    case QUERY_NODE_SHOW_LOCAL_VARIABLES_STMT:
D
dapan1121 已提交
7509 7510
    case QUERY_NODE_SHOW_VARIABLES_STMT:
      return extractShowVariablesResultSchema(numOfCols, pSchema);
wmmhello's avatar
wmmhello 已提交
7511 7512 7513
    default:
      break;
  }
7514

wmmhello's avatar
wmmhello 已提交
7515
  return TSDB_CODE_FAILED;
7516 7517
}

X
Xiaoyu Wang 已提交
7518 7519 7520 7521
static SNode* createStarCol() {
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
  if (NULL == pCol) {
    return NULL;
wmmhello's avatar
wmmhello 已提交
7522
  }
X
Xiaoyu Wang 已提交
7523 7524
  strcpy(pCol->colName, "*");
  return (SNode*)pCol;
X
Xiaoyu Wang 已提交
7525 7526
}

X
Xiaoyu Wang 已提交
7527
static SNode* createProjectCol(const char* pProjCol) {
7528 7529 7530 7531
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
  if (NULL == pCol) {
    return NULL;
  }
X
Xiaoyu Wang 已提交
7532
  snprintf(pCol->colName, sizeof(pCol->colName), "%s", pProjCol);
7533 7534 7535
  return (SNode*)pCol;
}

X
Xiaoyu Wang 已提交
7536 7537
static SNodeList* createProjectCols(int32_t ncols, const char* const pCols[]) {
  SNodeList* pProjections = NULL;
7538
  if (0 == ncols) {
X
Xiaoyu Wang 已提交
7539 7540 7541 7542 7543 7544 7545 7546 7547 7548 7549 7550 7551 7552 7553
    nodesListMakeStrictAppend(&pProjections, createStarCol());
    return pProjections;
  }
  for (int32_t i = 0; i < ncols; ++i) {
    int32_t code = nodesListMakeStrictAppend(&pProjections, createProjectCol(pCols[i]));
    if (TSDB_CODE_SUCCESS != code) {
      nodesDestroyList(pProjections);
      return NULL;
    }
  }
  return pProjections;
}

static int32_t createSimpleSelectStmt(const char* pDb, const char* pTable, int32_t numOfProjs,
                                      const char* const pProjCol[], SSelectStmt** pStmt) {
7554
  SSelectStmt* pSelect = (SSelectStmt*)nodesMakeNode(QUERY_NODE_SELECT_STMT);
wmmhello's avatar
wmmhello 已提交
7555 7556 7557 7558
  if (NULL == pSelect) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  sprintf(pSelect->stmtName, "%p", pSelect);
X
Xiaoyu Wang 已提交
7559

7560 7561
  SRealTableNode* pRealTable = (SRealTableNode*)nodesMakeNode(QUERY_NODE_REAL_TABLE);
  if (NULL == pRealTable) {
7562
    nodesDestroyNode((SNode*)pSelect);
wmmhello's avatar
wmmhello 已提交
7563 7564
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
7565 7566 7567
  snprintf(pRealTable->table.dbName, sizeof(pRealTable->table.dbName), "%s", pDb);
  snprintf(pRealTable->table.tableName, sizeof(pRealTable->table.tableName), "%s", pTable);
  snprintf(pRealTable->table.tableAlias, sizeof(pRealTable->table.tableAlias), "%s", pTable);
7568
  pSelect->pFromTable = (SNode*)pRealTable;
X
Xiaoyu Wang 已提交
7569

7570 7571 7572 7573 7574 7575
  if (numOfProjs >= 0) {
    pSelect->pProjectionList = createProjectCols(numOfProjs, pProjCol);
    if (NULL == pSelect->pProjectionList) {
      nodesDestroyNode((SNode*)pSelect);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
7576 7577
  }

wmmhello's avatar
wmmhello 已提交
7578
  *pStmt = pSelect;
X
Xiaoyu Wang 已提交
7579

wmmhello's avatar
wmmhello 已提交
7580
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
7581 7582
}

7583
static int32_t createSelectStmtForShow(ENodeType showType, SSelectStmt** pStmt) {
X
Xiaoyu Wang 已提交
7584 7585
  const SSysTableShowAdapter* pShow = &sysTableShowAdapter[showType - SYSTABLE_SHOW_TYPE_OFFSET];
  return createSimpleSelectStmt(pShow->pDbName, pShow->pTableName, pShow->numOfShowCols, pShow->pShowCols, pStmt);
7586 7587 7588
}

static int32_t createSelectStmtForShowTableDist(SShowTableDistributedStmt* pStmt, SSelectStmt** pOutput) {
X
Xiaoyu Wang 已提交
7589
  return createSimpleSelectStmt(pStmt->dbName, pStmt->tableName, 0, NULL, pOutput);
7590 7591
}

X
Xiaoyu Wang 已提交
7592
static int32_t createOperatorNode(EOperatorType opType, const char* pColName, SNode* pRight, SNode** pOp) {
wmmhello's avatar
wmmhello 已提交
7593 7594 7595
  if (NULL == pRight) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
7596

7597
  SOperatorNode* pOper = (SOperatorNode*)nodesMakeNode(QUERY_NODE_OPERATOR);
wmmhello's avatar
wmmhello 已提交
7598 7599 7600
  if (NULL == pOper) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
7601

wmmhello's avatar
wmmhello 已提交
7602 7603 7604 7605
  pOper->opType = opType;
  pOper->pLeft = nodesMakeNode(QUERY_NODE_COLUMN);
  pOper->pRight = nodesCloneNode(pRight);
  if (NULL == pOper->pLeft || NULL == pOper->pRight) {
7606
    nodesDestroyNode((SNode*)pOper);
wmmhello's avatar
wmmhello 已提交
7607 7608
    return TSDB_CODE_OUT_OF_MEMORY;
  }
7609
  snprintf(((SColumnNode*)pOper->pLeft)->colName, sizeof(((SColumnNode*)pOper->pLeft)->colName), "%s", pColName);
X
Xiaoyu Wang 已提交
7610

wmmhello's avatar
wmmhello 已提交
7611 7612
  *pOp = (SNode*)pOper;
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
7613 7614 7615
}

static const char* getTbNameColName(ENodeType type) {
wmmhello's avatar
wmmhello 已提交
7616
  return (QUERY_NODE_SHOW_STABLES_STMT == type ? "stable_name" : "table_name");
X
Xiaoyu Wang 已提交
7617 7618 7619
}

static int32_t createLogicCondNode(SNode* pCond1, SNode* pCond2, SNode** pCond) {
7620
  SLogicConditionNode* pCondition = (SLogicConditionNode*)nodesMakeNode(QUERY_NODE_LOGIC_CONDITION);
wmmhello's avatar
wmmhello 已提交
7621 7622 7623 7624 7625 7626
  if (NULL == pCondition) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pCondition->condType = LOGIC_COND_TYPE_AND;
  pCondition->pParameterList = nodesMakeList();
  if (NULL == pCondition->pParameterList) {
7627
    nodesDestroyNode((SNode*)pCondition);
wmmhello's avatar
wmmhello 已提交
7628 7629 7630 7631
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  if (TSDB_CODE_SUCCESS != nodesListAppend(pCondition->pParameterList, pCond1) ||
      TSDB_CODE_SUCCESS != nodesListAppend(pCondition->pParameterList, pCond2)) {
7632
    nodesDestroyNode((SNode*)pCondition);
wmmhello's avatar
wmmhello 已提交
7633 7634 7635 7636 7637
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  *pCond = (SNode*)pCondition;
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
7638 7639 7640
}

static int32_t createShowCondition(const SShowStmt* pShow, SSelectStmt* pSelect) {
wmmhello's avatar
wmmhello 已提交
7641 7642 7643 7644
  SNode* pDbCond = NULL;
  SNode* pTbCond = NULL;
  if (TSDB_CODE_SUCCESS != createOperatorNode(OP_TYPE_EQUAL, "db_name", pShow->pDbName, &pDbCond) ||
      TSDB_CODE_SUCCESS !=
7645
          createOperatorNode(pShow->tableCondType, getTbNameColName(nodeType(pShow)), pShow->pTbName, &pTbCond)) {
wmmhello's avatar
wmmhello 已提交
7646 7647 7648 7649 7650 7651 7652 7653 7654 7655 7656 7657 7658 7659 7660 7661 7662 7663 7664 7665
    nodesDestroyNode(pDbCond);
    nodesDestroyNode(pTbCond);
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  if (NULL != pDbCond && NULL != pTbCond) {
    if (TSDB_CODE_SUCCESS != createLogicCondNode(pDbCond, pTbCond, &pSelect->pWhere)) {
      nodesDestroyNode(pDbCond);
      nodesDestroyNode(pTbCond);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  } else {
    pSelect->pWhere = (NULL == pDbCond ? pTbCond : pDbCond);
  }

  if (NULL != pShow->pDbName) {
    strcpy(((SRealTableNode*)pSelect->pFromTable)->qualDbName, ((SValueNode*)pShow->pDbName)->literal);
  }

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
7666 7667 7668
}

static int32_t rewriteShow(STranslateContext* pCxt, SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
7669 7670 7671 7672 7673 7674
  SSelectStmt* pStmt = NULL;
  int32_t      code = createSelectStmtForShow(nodeType(pQuery->pRoot), &pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    code = createShowCondition((SShowStmt*)pQuery->pRoot, pStmt);
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
7675
    pCxt->showRewrite = true;
wmmhello's avatar
wmmhello 已提交
7676 7677 7678
    pQuery->showRewrite = true;
    nodesDestroyNode(pQuery->pRoot);
    pQuery->pRoot = (SNode*)pStmt;
7679 7680 7681 7682 7683 7684
  } else {
    nodesDestroyNode((SNode*)pStmt);
  }
  return code;
}

X
Xiaoyu Wang 已提交
7685 7686 7687 7688 7689 7690 7691 7692 7693 7694 7695 7696 7697 7698
static int32_t checkShowVgroups(STranslateContext* pCxt, SShowStmt* pShow) {
  // just to verify whether the database exists
  SDbCfgInfo dbCfg = {0};
  return getDBCfg(pCxt, ((SValueNode*)pShow->pDbName)->literal, &dbCfg);
}

static int32_t rewriteShowVgroups(STranslateContext* pCxt, SQuery* pQuery) {
  int32_t code = checkShowVgroups(pCxt, (SShowStmt*)pQuery->pRoot);
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteShow(pCxt, pQuery);
  }
  return code;
}

7699 7700 7701 7702 7703 7704 7705 7706 7707
static SNode* createTagsFunction() {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return NULL;
  }
  strcpy(pFunc->functionName, "_tags");
  return (SNode*)pFunc;
}

7708 7709 7710 7711 7712 7713 7714 7715 7716 7717 7718 7719
static int32_t createShowTableTagsProjections(SNodeList** pProjections, SNodeList** pTags) {
  if (NULL != *pTags) {
    TSWAP(*pProjections, *pTags);
    return TSDB_CODE_SUCCESS;
  }
  int32_t code = nodesListMakeStrictAppend(pProjections, createTbnameFunction());
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesListStrictAppend(*pProjections, createTagsFunction());
  }
  return code;
}

7720
static int32_t rewriteShowStableTags(STranslateContext* pCxt, SQuery* pQuery) {
7721 7722
  SShowTableTagsStmt* pShow = (SShowTableTagsStmt*)pQuery->pRoot;
  SSelectStmt*        pSelect = NULL;
7723 7724 7725
  int32_t code = createSimpleSelectStmt(((SValueNode*)pShow->pDbName)->literal, ((SValueNode*)pShow->pTbName)->literal,
                                        -1, NULL, &pSelect);
  if (TSDB_CODE_SUCCESS == code) {
7726
    code = createShowTableTagsProjections(&pSelect->pProjectionList, &pShow->pTags);
7727 7728
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
7729
    pCxt->showRewrite = true;
7730
    pQuery->showRewrite = true;
7731
    pSelect->tagScan = true;
7732 7733 7734 7735
    nodesDestroyNode(pQuery->pRoot);
    pQuery->pRoot = (SNode*)pSelect;
  } else {
    nodesDestroyNode((SNode*)pSelect);
wmmhello's avatar
wmmhello 已提交
7736 7737
  }
  return code;
7738 7739
}

7740
static int32_t rewriteShowDnodeVariables(STranslateContext* pCxt, SQuery* pQuery) {
7741 7742 7743 7744 7745 7746 7747 7748 7749 7750 7751
  SShowDnodeVariablesStmt* pStmt = (SShowDnodeVariablesStmt*)pQuery->pRoot;
  SNode*                   pDnodeCond = NULL;
  SNode*                   pLikeCond = NULL;
  SSelectStmt*             pSelect = NULL;
  int32_t                  code = createSelectStmtForShow(nodeType(pQuery->pRoot), &pSelect);
  if (TSDB_CODE_SUCCESS == code) {
    code = createOperatorNode(OP_TYPE_EQUAL, "dnode_id", pStmt->pDnodeId, &pDnodeCond);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = createOperatorNode(OP_TYPE_LIKE, "name", pStmt->pLikePattern, &pLikeCond);
  }
7752
  if (TSDB_CODE_SUCCESS == code) {
7753 7754 7755 7756 7757
    if (NULL != pLikeCond) {
      code = createLogicCondNode(pDnodeCond, pLikeCond, &pSelect->pWhere);
    } else {
      pSelect->pWhere = pDnodeCond;
    }
7758 7759
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
7760
    pCxt->showRewrite = true;
7761 7762
    pQuery->showRewrite = true;
    nodesDestroyNode(pQuery->pRoot);
7763
    pQuery->pRoot = (SNode*)pSelect;
7764 7765 7766 7767
  }
  return code;
}

7768 7769 7770 7771 7772 7773 7774 7775
static int32_t rewriteShowVnodes(STranslateContext* pCxt, SQuery* pQuery) {
  SShowVnodesStmt* pShow = (SShowVnodesStmt*)(pQuery->pRoot);
  SSelectStmt*     pStmt = NULL;
  int32_t          code = createSelectStmtForShow(QUERY_NODE_SHOW_VNODES_STMT, &pStmt);
  if (TSDB_CODE_SUCCESS == code) {
    if (NULL != pShow->pDnodeId) {
      code = createOperatorNode(OP_TYPE_EQUAL, "dnode_id", pShow->pDnodeId, &pStmt->pWhere);
    } else {
7776
      code = createOperatorNode(OP_TYPE_EQUAL, "dnode_ep", pShow->pDnodeEndpoint, &pStmt->pWhere);
7777 7778 7779
    }
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
7780
    pCxt->showRewrite = true;
7781 7782 7783 7784 7785 7786 7787
    pQuery->showRewrite = true;
    nodesDestroyNode(pQuery->pRoot);
    pQuery->pRoot = (SNode*)pStmt;
  }
  return code;
}

7788 7789 7790 7791 7792 7793 7794 7795 7796 7797 7798 7799 7800 7801 7802 7803 7804 7805 7806 7807 7808 7809 7810 7811 7812 7813 7814 7815 7816 7817
static SNode* createBlockDistInfoFunc() {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return NULL;
  }

  strcpy(pFunc->functionName, "_block_dist_info");
  strcpy(pFunc->node.aliasName, "_block_dist_info");
  return (SNode*)pFunc;
}

static SNode* createBlockDistFunc() {
  SFunctionNode* pFunc = (SFunctionNode*)nodesMakeNode(QUERY_NODE_FUNCTION);
  if (NULL == pFunc) {
    return NULL;
  }

  strcpy(pFunc->functionName, "_block_dist");
  strcpy(pFunc->node.aliasName, "_block_dist");
  if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(&pFunc->pParameterList, createBlockDistInfoFunc())) {
    nodesDestroyNode((SNode*)pFunc);
    return NULL;
  }
  return (SNode*)pFunc;
}

static int32_t rewriteShowTableDist(STranslateContext* pCxt, SQuery* pQuery) {
  SSelectStmt* pStmt = NULL;
  int32_t      code = createSelectStmtForShowTableDist((SShowTableDistributedStmt*)pQuery->pRoot, &pStmt);
  if (TSDB_CODE_SUCCESS == code) {
7818
    NODES_DESTORY_LIST(pStmt->pProjectionList);
7819 7820 7821
    code = nodesListMakeStrictAppend(&pStmt->pProjectionList, createBlockDistFunc());
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
7822
    pCxt->showRewrite = true;
7823 7824 7825 7826 7827 7828 7829
    pQuery->showRewrite = true;
    nodesDestroyNode(pQuery->pRoot);
    pQuery->pRoot = (SNode*)pStmt;
  }
  return code;
}

X
Xiaoyu Wang 已提交
7830
typedef struct SVgroupCreateTableBatch {
wmmhello's avatar
wmmhello 已提交
7831 7832 7833
  SVCreateTbBatchReq req;
  SVgroupInfo        info;
  char               dbName[TSDB_DB_NAME_LEN];
X
Xiaoyu Wang 已提交
7834
} SVgroupCreateTableBatch;
7835

dengyihao's avatar
dengyihao 已提交
7836
static int32_t buildNormalTableBatchReq(int32_t acctId, const SCreateTableStmt* pStmt, const SVgroupInfo* pVgroupInfo,
wmmhello's avatar
wmmhello 已提交
7837 7838 7839 7840 7841 7842 7843 7844
                                        SVgroupCreateTableBatch* pBatch) {
  char  dbFName[TSDB_DB_FNAME_LEN] = {0};
  SName name = {.type = TSDB_DB_NAME_T, .acctId = acctId};
  strcpy(name.dbname, pStmt->dbName);
  tNameGetFullDbName(&name, dbFName);

  SVCreateTbReq req = {0};
  req.type = TD_NORMAL_TABLE;
7845
  req.name = taosStrdup(pStmt->tableName);
wmmhello's avatar
wmmhello 已提交
7846
  req.ttl = pStmt->pOptions->ttl;
wmmhello's avatar
wmmhello 已提交
7847
  if (pStmt->pOptions->commentNull == false) {
7848
    req.comment = taosStrdup(pStmt->pOptions->comment);
wmmhello's avatar
wmmhello 已提交
7849
    if (NULL == req.comment) {
7850
      tdDestroySVCreateTbReq(&req);
wmmhello's avatar
wmmhello 已提交
7851 7852
      return TSDB_CODE_OUT_OF_MEMORY;
    }
wmmhello's avatar
wmmhello 已提交
7853
    req.commentLen = strlen(pStmt->pOptions->comment);
X
Xiaoyu Wang 已提交
7854
  } else {
wmmhello's avatar
wmmhello 已提交
7855
    req.commentLen = -1;
wmmhello's avatar
wmmhello 已提交
7856
  }
7857 7858 7859 7860
  req.ntb.schemaRow.nCols = LIST_LENGTH(pStmt->pCols);
  req.ntb.schemaRow.version = 1;
  req.ntb.schemaRow.pSchema = taosMemoryCalloc(req.ntb.schemaRow.nCols, sizeof(SSchema));
  if (NULL == req.name || NULL == req.ntb.schemaRow.pSchema) {
7861
    tdDestroySVCreateTbReq(&req);
wmmhello's avatar
wmmhello 已提交
7862 7863 7864 7865 7866 7867 7868 7869
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  if (pStmt->ignoreExists) {
    req.flags |= TD_CREATE_IF_NOT_EXISTS;
  }
  SNode*   pCol;
  col_id_t index = 0;
  FOREACH(pCol, pStmt->pCols) {
7870
    toSchema((SColumnDefNode*)pCol, index + 1, req.ntb.schemaRow.pSchema + index);
wmmhello's avatar
wmmhello 已提交
7871 7872 7873 7874 7875 7876
    ++index;
  }
  pBatch->info = *pVgroupInfo;
  strcpy(pBatch->dbName, pStmt->dbName);
  pBatch->req.pArray = taosArrayInit(1, sizeof(struct SVCreateTbReq));
  if (NULL == pBatch->req.pArray) {
7877
    tdDestroySVCreateTbReq(&req);
wmmhello's avatar
wmmhello 已提交
7878 7879 7880 7881 7882
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  taosArrayPush(pBatch->req.pArray, &req);

  return TSDB_CODE_SUCCESS;
7883 7884
}

X
Xiaoyu Wang 已提交
7885
static int32_t serializeVgroupCreateTableBatch(SVgroupCreateTableBatch* pTbBatch, SArray* pBufArray) {
wmmhello's avatar
wmmhello 已提交
7886 7887 7888 7889 7890 7891 7892 7893 7894 7895 7896 7897 7898 7899 7900 7901 7902 7903 7904 7905 7906 7907 7908 7909 7910 7911 7912 7913 7914
  int      tlen;
  SEncoder coder = {0};

  int32_t ret = 0;
  tEncodeSize(tEncodeSVCreateTbBatchReq, &pTbBatch->req, tlen, ret);
  tlen += sizeof(SMsgHead);
  void* buf = taosMemoryMalloc(tlen);
  if (NULL == buf) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  ((SMsgHead*)buf)->vgId = htonl(pTbBatch->info.vgId);
  ((SMsgHead*)buf)->contLen = htonl(tlen);
  void* pBuf = POINTER_SHIFT(buf, sizeof(SMsgHead));

  tEncoderInit(&coder, pBuf, tlen - sizeof(SMsgHead));
  tEncodeSVCreateTbBatchReq(&coder, &pTbBatch->req);
  tEncoderClear(&coder);

  SVgDataBlocks* pVgData = taosMemoryCalloc(1, sizeof(SVgDataBlocks));
  if (NULL == pVgData) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pVgData->vg = pTbBatch->info;
  pVgData->pData = buf;
  pVgData->size = tlen;
  pVgData->numOfTables = (int32_t)taosArrayGetSize(pTbBatch->req.pArray);
  taosArrayPush(pBufArray, &pVgData);

  return TSDB_CODE_SUCCESS;
7915 7916
}

wmmhello's avatar
wmmhello 已提交
7917
static void destroyCreateTbReqBatch(void* data) {
7918 7919
  SVgroupCreateTableBatch* pTbBatch = (SVgroupCreateTableBatch*)data;
  size_t                   size = taosArrayGetSize(pTbBatch->req.pArray);
wmmhello's avatar
wmmhello 已提交
7920 7921
  for (int32_t i = 0; i < size; ++i) {
    SVCreateTbReq* pTableReq = taosArrayGet(pTbBatch->req.pArray, i);
7922
    tdDestroySVCreateTbReq(pTableReq);
wmmhello's avatar
wmmhello 已提交
7923
  }
7924

wmmhello's avatar
wmmhello 已提交
7925
  taosArrayDestroy(pTbBatch->req.pArray);
7926 7927
}

wmmhello's avatar
wmmhello 已提交
7928
int32_t rewriteToVnodeModifyOpStmt(SQuery* pQuery, SArray* pBufArray) {
X
Xiaoyu Wang 已提交
7929
  SVnodeModifyOpStmt* pNewStmt = (SVnodeModifyOpStmt*)nodesMakeNode(QUERY_NODE_VNODE_MODIFY_STMT);
wmmhello's avatar
wmmhello 已提交
7930 7931 7932 7933 7934 7935 7936 7937
  if (pNewStmt == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pNewStmt->sqlNodeType = nodeType(pQuery->pRoot);
  pNewStmt->pDataBlocks = pBufArray;
  nodesDestroyNode(pQuery->pRoot);
  pQuery->pRoot = (SNode*)pNewStmt;
  return TSDB_CODE_SUCCESS;
7938 7939
}

X
Xiaoyu Wang 已提交
7940
static void destroyCreateTbReqArray(SArray* pArray) {
wmmhello's avatar
wmmhello 已提交
7941 7942 7943 7944 7945 7946 7947
  size_t size = taosArrayGetSize(pArray);
  for (size_t i = 0; i < size; ++i) {
    SVgDataBlocks* pVg = taosArrayGetP(pArray, i);
    taosMemoryFreeClear(pVg->pData);
    taosMemoryFreeClear(pVg);
  }
  taosArrayDestroy(pArray);
X
Xiaoyu Wang 已提交
7948 7949
}

dengyihao's avatar
dengyihao 已提交
7950
static int32_t buildCreateTableDataBlock(int32_t acctId, const SCreateTableStmt* pStmt, const SVgroupInfo* pInfo,
wmmhello's avatar
wmmhello 已提交
7951 7952 7953 7954 7955
                                         SArray** pBufArray) {
  *pBufArray = taosArrayInit(1, POINTER_BYTES);
  if (NULL == *pBufArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
7956

wmmhello's avatar
wmmhello 已提交
7957 7958 7959 7960 7961
  SVgroupCreateTableBatch tbatch = {0};
  int32_t                 code = buildNormalTableBatchReq(acctId, pStmt, pInfo, &tbatch);
  if (TSDB_CODE_SUCCESS == code) {
    code = serializeVgroupCreateTableBatch(&tbatch, *pBufArray);
  }
X
Xiaoyu Wang 已提交
7962

wmmhello's avatar
wmmhello 已提交
7963 7964 7965 7966 7967
  destroyCreateTbReqBatch(&tbatch);
  if (TSDB_CODE_SUCCESS != code) {
    destroyCreateTbReqArray(*pBufArray);
  }
  return code;
7968 7969 7970
}

static int32_t rewriteCreateTable(STranslateContext* pCxt, SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
7971 7972
  SCreateTableStmt* pStmt = (SCreateTableStmt*)pQuery->pRoot;

7973
  int32_t     code = checkCreateTable(pCxt, pStmt, false);
wmmhello's avatar
wmmhello 已提交
7974
  SVgroupInfo info = {0};
7975
  SName       name;
D
dapan1121 已提交
7976
  toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &name);
wmmhello's avatar
wmmhello 已提交
7977
  if (TSDB_CODE_SUCCESS == code) {
D
dapan1121 已提交
7978 7979 7980 7981
    code = getTableHashVgroupImpl(pCxt, &name, &info);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(&name, pCxt->pTargetTables);
wmmhello's avatar
wmmhello 已提交
7982 7983 7984 7985 7986 7987 7988 7989 7990 7991 7992 7993 7994
  }
  SArray* pBufArray = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = buildCreateTableDataBlock(pCxt->pParseCxt->acctId, pStmt, &info, &pBufArray);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteToVnodeModifyOpStmt(pQuery, pBufArray);
    if (TSDB_CODE_SUCCESS != code) {
      destroyCreateTbReqArray(pBufArray);
    }
  }

  return code;
7995 7996
}

C
Cary Xu 已提交
7997
static void addCreateTbReqIntoVgroup(int32_t acctId, SHashObj* pVgroupHashmap, SCreateSubTableClause* pStmt,
7998
                                     const STag* pTag, uint64_t suid, const char* sTableNmae, SVgroupInfo* pVgInfo,
7999
                                     SArray* tagName, uint8_t tagNum) {
8000 8001 8002 8003
  //  char  dbFName[TSDB_DB_FNAME_LEN] = {0};
  //  SName name = {.type = TSDB_DB_NAME_T, .acctId = acctId};
  //  strcpy(name.dbname, pStmt->dbName);
  //  tNameGetFullDbName(&name, dbFName);
wmmhello's avatar
wmmhello 已提交
8004 8005 8006

  struct SVCreateTbReq req = {0};
  req.type = TD_CHILD_TABLE;
8007
  req.name = taosStrdup(pStmt->tableName);
X
Xiaoyu Wang 已提交
8008
  req.ttl = pStmt->pOptions->ttl;
wmmhello's avatar
wmmhello 已提交
8009
  if (pStmt->pOptions->commentNull == false) {
8010
    req.comment = taosStrdup(pStmt->pOptions->comment);
wmmhello's avatar
wmmhello 已提交
8011
    req.commentLen = strlen(pStmt->pOptions->comment);
X
Xiaoyu Wang 已提交
8012
  } else {
wmmhello's avatar
wmmhello 已提交
8013
    req.commentLen = -1;
wmmhello's avatar
wmmhello 已提交
8014
  }
wmmhello's avatar
wmmhello 已提交
8015
  req.ctb.suid = suid;
8016
  req.ctb.tagNum = tagNum;
8017
  req.ctb.stbName = taosStrdup(sTableNmae);
C
Cary Xu 已提交
8018
  req.ctb.pTag = (uint8_t*)pTag;
H
Haojun Liao 已提交
8019
  req.ctb.tagName = taosArrayDup(tagName, NULL);
wmmhello's avatar
wmmhello 已提交
8020 8021 8022 8023 8024 8025 8026 8027 8028 8029 8030 8031 8032 8033 8034 8035 8036
  if (pStmt->ignoreExists) {
    req.flags |= TD_CREATE_IF_NOT_EXISTS;
  }

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

    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
    taosArrayPush(pTableBatch->req.pArray, &req);
  }
8037 8038
}

X
Xiaoyu Wang 已提交
8039 8040
static SDataType schemaToDataType(uint8_t precision, SSchema* pSchema) {
  SDataType dt = {.type = pSchema->type, .bytes = pSchema->bytes, .precision = precision, .scale = 0};
8041 8042 8043
  return dt;
}

8044 8045 8046 8047 8048
static int32_t createCastFuncForTag(STranslateContext* pCxt, SNode* pNode, SDataType dt, SNode** pCast) {
  SNode* pExpr = nodesCloneNode(pNode);
  if (NULL == pExpr) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8049 8050 8051 8052
  int32_t code = translateExpr(pCxt, &pExpr);
  if (TSDB_CODE_SUCCESS == code) {
    code = createCastFunc(pCxt, pExpr, dt, pCast);
  }
8053 8054 8055 8056 8057 8058
  if (TSDB_CODE_SUCCESS != code) {
    nodesDestroyNode(pExpr);
  }
  return code;
}

8059
static int32_t createTagValFromExpr(STranslateContext* pCxt, SDataType targetDt, SNode* pNode, SValueNode** pVal) {
8060
  SNode*  pCast = NULL;
8061
  int32_t code = createCastFuncForTag(pCxt, pNode, targetDt, &pCast);
8062
  SNode*  pNew = NULL;
8063 8064 8065 8066 8067 8068 8069 8070 8071 8072 8073 8074
  if (TSDB_CODE_SUCCESS == code) {
    code = scalarCalculateConstants(pCast, &pNew);
  }
  if (TSDB_CODE_SUCCESS == code) {
    pCast = pNew;
    if (QUERY_NODE_VALUE != nodeType(pCast)) {
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_WRONG_VALUE_TYPE, ((SExprNode*)pNode)->aliasName);
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pVal = (SValueNode*)pCast;
wmmhello's avatar
wmmhello 已提交
8075
  } else {
8076
    nodesDestroyNode(pCast);
X
Xiaoyu Wang 已提交
8077
  }
8078
  return code;
X
Xiaoyu Wang 已提交
8079 8080
}

8081
static int32_t createTagValFromVal(STranslateContext* pCxt, SDataType targetDt, SNode* pNode, SValueNode** pVal) {
8082 8083
  SValueNode* pTempVal = (SValueNode*)nodesCloneNode(pNode);
  if (NULL == pTempVal) {
8084 8085
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8086 8087 8088 8089 8090 8091
  if (DEAL_RES_ERROR == translateValueImpl(pCxt, pTempVal, targetDt, true)) {
    nodesDestroyNode((SNode*)pTempVal);
    return pCxt->errCode;
  }
  *pVal = pTempVal;
  return TSDB_CODE_SUCCESS;
8092 8093 8094 8095 8096 8097 8098 8099 8100 8101 8102
}

static int32_t createTagVal(STranslateContext* pCxt, uint8_t precision, SSchema* pSchema, SNode* pNode,
                            SValueNode** pVal) {
  if (QUERY_NODE_VALUE == nodeType(pNode)) {
    return createTagValFromVal(pCxt, schemaToDataType(precision, pSchema), pNode, pVal);
  } else {
    return createTagValFromExpr(pCxt, schemaToDataType(precision, pSchema), pNode, pVal);
  }
}

X
Xiaoyu Wang 已提交
8103 8104 8105 8106
static int32_t buildJsonTagVal(STranslateContext* pCxt, SSchema* pTagSchema, SValueNode* pVal, SArray* pTagArray,
                               STag** ppTag) {
  if (pVal->literal && strlen(pVal->literal) > (TSDB_MAX_JSON_TAG_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
    return buildSyntaxErrMsg(&pCxt->msgBuf, "json string too long than 4095", pVal->literal);
wmmhello's avatar
wmmhello 已提交
8107
  }
X
Xiaoyu Wang 已提交
8108

wmmhello's avatar
wmmhello 已提交
8109
  return parseJsontoTagData(pVal->literal, pTagArray, ppTag, &pCxt->msgBuf);
X
Xiaoyu Wang 已提交
8110 8111 8112 8113 8114 8115
}

static int32_t buildNormalTagVal(STranslateContext* pCxt, SSchema* pTagSchema, SValueNode* pVal, SArray* pTagArray) {
  if (pVal->node.resType.type != TSDB_DATA_TYPE_NULL) {
    void*   nodeVal = nodesGetValueFromNode(pVal);
    STagVal val = {.cid = pTagSchema->colId, .type = pTagSchema->type};
8116
    //    strcpy(val.colName, pTagSchema->name);
X
Xiaoyu Wang 已提交
8117 8118 8119 8120 8121 8122 8123 8124 8125
    if (IS_VAR_DATA_TYPE(pTagSchema->type)) {
      val.pData = varDataVal(nodeVal);
      val.nData = varDataLen(nodeVal);
    } else {
      memcpy(&val.i64, nodeVal, pTagSchema->bytes);
    }
    taosArrayPush(pTagArray, &val);
  }
  return TSDB_CODE_SUCCESS;
8126 8127
}

dengyihao's avatar
dengyihao 已提交
8128
static int32_t buildKVRowForBindTags(STranslateContext* pCxt, SCreateSubTableClause* pStmt, STableMeta* pSuperTableMeta,
wmmhello's avatar
wmmhello 已提交
8129
                                     STag** ppTag, SArray* tagName) {
wmmhello's avatar
wmmhello 已提交
8130 8131 8132 8133 8134 8135
  int32_t numOfTags = getNumOfTags(pSuperTableMeta);
  if (LIST_LENGTH(pStmt->pValsOfTags) != LIST_LENGTH(pStmt->pSpecificTags) ||
      numOfTags < LIST_LENGTH(pStmt->pValsOfTags)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TAGS_NOT_MATCHED);
  }

C
Cary Xu 已提交
8136
  SArray* pTagArray = taosArrayInit(LIST_LENGTH(pStmt->pValsOfTags), sizeof(STagVal));
X
Xiaoyu Wang 已提交
8137 8138
  if (NULL == pTagArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
C
Cary Xu 已提交
8139
  }
8140 8141 8142 8143 8144 8145

  int32_t code = TSDB_CODE_SUCCESS;

  bool       isJson = false;
  SNodeList* pVals = NULL;
  SNode *    pTag = NULL, *pNode = NULL;
wmmhello's avatar
wmmhello 已提交
8146 8147
  FORBOTH(pTag, pStmt->pSpecificTags, pNode, pStmt->pValsOfTags) {
    SColumnNode* pCol = (SColumnNode*)pTag;
8148
    SSchema*     pSchema = getTagSchema(pSuperTableMeta, pCol->colName);
wmmhello's avatar
wmmhello 已提交
8149
    if (NULL == pSchema) {
wmmhello's avatar
wmmhello 已提交
8150
      code = generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_TAG_NAME, pCol->colName);
wmmhello's avatar
wmmhello 已提交
8151 8152
    }
    SValueNode* pVal = NULL;
8153 8154
    if (TSDB_CODE_SUCCESS == code) {
      code = createTagVal(pCxt, pSuperTableMeta->tableInfo.precision, pSchema, pNode, &pVal);
wmmhello's avatar
wmmhello 已提交
8155
    }
8156 8157 8158 8159
    if (TSDB_CODE_SUCCESS == code) {
      if (pSchema->type == TSDB_DATA_TYPE_JSON) {
        isJson = true;
        code = buildJsonTagVal(pCxt, pSchema, pVal, pTagArray, ppTag);
wmmhello's avatar
wmmhello 已提交
8160
        taosArrayPush(tagName, pCol->colName);
8161 8162
      } else if (pVal->node.resType.type != TSDB_DATA_TYPE_NULL) {
        code = buildNormalTagVal(pCxt, pSchema, pVal, pTagArray);
wmmhello's avatar
wmmhello 已提交
8163
        taosArrayPush(tagName, pCol->colName);
8164
      }
wmmhello's avatar
wmmhello 已提交
8165
    }
8166 8167 8168 8169 8170
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesListMakeAppend(&pVals, (SNode*)pVal);
    }
    if (TSDB_CODE_SUCCESS != code) {
      break;
wmmhello's avatar
wmmhello 已提交
8171 8172 8173
    }
  }

8174 8175
  if (TSDB_CODE_SUCCESS == code && !isJson) {
    code = tTagNew(pTagArray, 1, false, ppTag);
wmmhello's avatar
wmmhello 已提交
8176
  }
8177 8178

  nodesDestroyList(pVals);
C
Cary Xu 已提交
8179
  taosArrayDestroy(pTagArray);
8180
  return code;
8181
}
8182

dengyihao's avatar
dengyihao 已提交
8183
static int32_t buildKVRowForAllTags(STranslateContext* pCxt, SCreateSubTableClause* pStmt, STableMeta* pSuperTableMeta,
wmmhello's avatar
wmmhello 已提交
8184
                                    STag** ppTag, SArray* tagName) {
wmmhello's avatar
wmmhello 已提交
8185 8186 8187 8188
  if (getNumOfTags(pSuperTableMeta) != LIST_LENGTH(pStmt->pValsOfTags)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TAGS_NOT_MATCHED);
  }

8189 8190 8191
  SArray* pTagArray = taosArrayInit(LIST_LENGTH(pStmt->pValsOfTags), sizeof(STagVal));
  if (NULL == pTagArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
C
Cary Xu 已提交
8192 8193
  }

8194 8195 8196 8197 8198 8199 8200
  int32_t code = TSDB_CODE_SUCCESS;

  bool       isJson = false;
  int32_t    index = 0;
  SSchema*   pTagSchemas = getTableTagSchema(pSuperTableMeta);
  SNodeList* pVals = NULL;
  SNode*     pNode;
wmmhello's avatar
wmmhello 已提交
8201 8202
  FOREACH(pNode, pStmt->pValsOfTags) {
    SValueNode* pVal = NULL;
C
Cary Xu 已提交
8203
    SSchema*    pTagSchema = pTagSchemas + index;
8204 8205 8206 8207 8208
    code = createTagVal(pCxt, pSuperTableMeta->tableInfo.precision, pTagSchema, pNode, &pVal);
    if (TSDB_CODE_SUCCESS == code) {
      if (pTagSchema->type == TSDB_DATA_TYPE_JSON) {
        isJson = true;
        code = buildJsonTagVal(pCxt, pTagSchema, pVal, pTagArray, ppTag);
wmmhello's avatar
wmmhello 已提交
8209 8210 8211
        if (TSDB_CODE_SUCCESS != code) {
          nodesDestroyNode((SNode*)pVal);
        }
wmmhello's avatar
wmmhello 已提交
8212
        taosArrayPush(tagName, pTagSchema->name);
8213 8214 8215
      } else if (pVal->node.resType.type != TSDB_DATA_TYPE_NULL && !pVal->isNull) {
        char*   tmpVal = nodesGetValueFromNode(pVal);
        STagVal val = {.cid = pTagSchema->colId, .type = pTagSchema->type};
8216
        //        strcpy(val.colName, pTagSchema->name);
8217 8218 8219 8220 8221 8222 8223
        if (IS_VAR_DATA_TYPE(pTagSchema->type)) {
          val.pData = varDataVal(tmpVal);
          val.nData = varDataLen(tmpVal);
        } else {
          memcpy(&val.i64, tmpVal, pTagSchema->bytes);
        }
        taosArrayPush(pTagArray, &val);
wmmhello's avatar
wmmhello 已提交
8224
        taosArrayPush(tagName, pTagSchema->name);
8225
      }
wmmhello's avatar
wmmhello 已提交
8226
    }
8227 8228
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesListMakeAppend(&pVals, (SNode*)pVal);
wmmhello's avatar
wmmhello 已提交
8229
    }
8230 8231
    if (TSDB_CODE_SUCCESS != code) {
      break;
wmmhello's avatar
wmmhello 已提交
8232
    }
wmmhello's avatar
wmmhello 已提交
8233
    ++index;
wmmhello's avatar
wmmhello 已提交
8234
  }
wmmhello's avatar
wmmhello 已提交
8235

8236 8237
  if (TSDB_CODE_SUCCESS == code && !isJson) {
    code = tTagNew(pTagArray, 1, false, ppTag);
wmmhello's avatar
wmmhello 已提交
8238 8239
  }

8240
  nodesDestroyList(pVals);
C
Cary Xu 已提交
8241
  taosArrayDestroy(pTagArray);
wmmhello's avatar
wmmhello 已提交
8242
  return code;
8243 8244
}

X
Xiaoyu Wang 已提交
8245
static int32_t checkCreateSubTable(STranslateContext* pCxt, SCreateSubTableClause* pStmt) {
wmmhello's avatar
wmmhello 已提交
8246 8247 8248
  if (0 != strcmp(pStmt->dbName, pStmt->useDbName)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_CORRESPONDING_STABLE_ERR);
  }
X
Xiaoyu Wang 已提交
8249 8250 8251 8252
  if (NULL != strchr(pStmt->tableName, '.')) {
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_IDENTIFIER_NAME,
                                   "The table name cannot contain '.'");
  }
wmmhello's avatar
wmmhello 已提交
8253
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8254
}
8255
static int32_t rewriteCreateSubTable(STranslateContext* pCxt, SCreateSubTableClause* pStmt, SHashObj* pVgroupHashmap) {
wmmhello's avatar
wmmhello 已提交
8256 8257 8258 8259 8260 8261
  int32_t code = checkCreateSubTable(pCxt, pStmt);

  STableMeta* pSuperTableMeta = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = getTableMeta(pCxt, pStmt->useDbName, pStmt->useTableName, &pSuperTableMeta);
  }
D
dapan1121 已提交
8262 8263 8264 8265 8266
  if (TSDB_CODE_SUCCESS == code) {
    SName name;
    toName(pCxt->pParseCxt->acctId, pStmt->dbName, pStmt->tableName, &name);
    code = collectUseTable(&name, pCxt->pTargetTables);
  }
wmmhello's avatar
wmmhello 已提交
8267

8268
  STag*   pTag = NULL;
wmmhello's avatar
wmmhello 已提交
8269
  SArray* tagName = taosArrayInit(8, TSDB_COL_NAME_LEN);
wmmhello's avatar
wmmhello 已提交
8270 8271 8272

  if (TSDB_CODE_SUCCESS == code) {
    if (NULL != pStmt->pSpecificTags) {
wmmhello's avatar
wmmhello 已提交
8273
      code = buildKVRowForBindTags(pCxt, pStmt, pSuperTableMeta, &pTag, tagName);
wmmhello's avatar
wmmhello 已提交
8274
    } else {
wmmhello's avatar
wmmhello 已提交
8275
      code = buildKVRowForAllTags(pCxt, pStmt, pSuperTableMeta, &pTag, tagName);
wmmhello's avatar
wmmhello 已提交
8276 8277 8278 8279 8280 8281 8282 8283
    }
  }

  SVgroupInfo info = {0};
  if (TSDB_CODE_SUCCESS == code) {
    code = getTableHashVgroup(pCxt, pStmt->dbName, pStmt->tableName, &info);
  }
  if (TSDB_CODE_SUCCESS == code) {
8284
    addCreateTbReqIntoVgroup(pCxt->pParseCxt->acctId, pVgroupHashmap, pStmt, pTag, pSuperTableMeta->uid,
8285
                             pStmt->useTableName, &info, tagName, pSuperTableMeta->tableInfo.numOfTags);
8286 8287
  } else {
    taosMemoryFree(pTag);
wmmhello's avatar
wmmhello 已提交
8288 8289
  }

wmmhello's avatar
wmmhello 已提交
8290
  taosArrayDestroy(tagName);
wmmhello's avatar
wmmhello 已提交
8291 8292
  taosMemoryFreeClear(pSuperTableMeta);
  return code;
8293 8294
}

wmmhello's avatar
wmmhello 已提交
8295
SArray* serializeVgroupsCreateTableBatch(SHashObj* pVgroupHashmap) {
wmmhello's avatar
wmmhello 已提交
8296 8297 8298 8299
  SArray* pBufArray = taosArrayInit(taosHashGetSize(pVgroupHashmap), sizeof(void*));
  if (NULL == pBufArray) {
    return NULL;
  }
8300

wmmhello's avatar
wmmhello 已提交
8301 8302 8303 8304 8305 8306 8307
  int32_t                  code = TSDB_CODE_SUCCESS;
  SVgroupCreateTableBatch* pTbBatch = NULL;
  do {
    pTbBatch = taosHashIterate(pVgroupHashmap, pTbBatch);
    if (pTbBatch == NULL) {
      break;
    }
8308

wmmhello's avatar
wmmhello 已提交
8309 8310
    serializeVgroupCreateTableBatch(pTbBatch, pBufArray);
  } while (true);
8311

wmmhello's avatar
wmmhello 已提交
8312
  return pBufArray;
8313 8314 8315
}

static int32_t rewriteCreateMultiTable(STranslateContext* pCxt, SQuery* pQuery) {
X
Xiaoyu Wang 已提交
8316
  SCreateMultiTablesStmt* pStmt = (SCreateMultiTablesStmt*)pQuery->pRoot;
8317

wmmhello's avatar
wmmhello 已提交
8318 8319 8320 8321
  SHashObj* pVgroupHashmap = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_NO_LOCK);
  if (NULL == pVgroupHashmap) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8322

wmmhello's avatar
wmmhello 已提交
8323
  taosHashSetFreeFp(pVgroupHashmap, destroyCreateTbReqBatch);
wmmhello's avatar
wmmhello 已提交
8324 8325 8326
  int32_t code = TSDB_CODE_SUCCESS;
  SNode*  pNode;
  FOREACH(pNode, pStmt->pSubTables) {
X
Xiaoyu Wang 已提交
8327
    SCreateSubTableClause* pClause = (SCreateSubTableClause*)pNode;
X
Xiaoyu Wang 已提交
8328
    code = rewriteCreateSubTable(pCxt, pClause, pVgroupHashmap);
wmmhello's avatar
wmmhello 已提交
8329 8330 8331 8332 8333
    if (TSDB_CODE_SUCCESS != code) {
      taosHashCleanup(pVgroupHashmap);
      return code;
    }
  }
8334

wmmhello's avatar
wmmhello 已提交
8335
  SArray* pBufArray = serializeVgroupsCreateTableBatch(pVgroupHashmap);
wmmhello's avatar
wmmhello 已提交
8336 8337 8338 8339
  taosHashCleanup(pVgroupHashmap);
  if (NULL == pBufArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
8340

wmmhello's avatar
wmmhello 已提交
8341
  return rewriteToVnodeModifyOpStmt(pQuery, pBufArray);
X
Xiaoyu Wang 已提交
8342 8343 8344
}

typedef struct SVgroupDropTableBatch {
wmmhello's avatar
wmmhello 已提交
8345 8346 8347
  SVDropTbBatchReq req;
  SVgroupInfo      info;
  char             dbName[TSDB_DB_NAME_LEN];
X
Xiaoyu Wang 已提交
8348 8349
} SVgroupDropTableBatch;

8350 8351
static void addDropTbReqIntoVgroup(SHashObj* pVgroupHashmap, SDropTableClause* pClause, SVgroupInfo* pVgInfo,
                                   uint64_t suid) {
8352
  SVDropTbReq            req = {.name = pClause->tableName, .suid = suid, .igNotExists = pClause->ignoreNotExists};
wmmhello's avatar
wmmhello 已提交
8353 8354 8355 8356 8357 8358 8359 8360 8361 8362 8363
  SVgroupDropTableBatch* pTableBatch = taosHashGet(pVgroupHashmap, &pVgInfo->vgId, sizeof(pVgInfo->vgId));
  if (NULL == pTableBatch) {
    SVgroupDropTableBatch tBatch = {0};
    tBatch.info = *pVgInfo;
    tBatch.req.pArray = taosArrayInit(TARRAY_MIN_SIZE, sizeof(SVDropTbReq));
    taosArrayPush(tBatch.req.pArray, &req);

    taosHashPut(pVgroupHashmap, &pVgInfo->vgId, sizeof(pVgInfo->vgId), &tBatch, sizeof(tBatch));
  } else {  // add to the correct vgroup
    taosArrayPush(pTableBatch->req.pArray, &req);
  }
X
Xiaoyu Wang 已提交
8364 8365 8366
}

static int32_t buildDropTableVgroupHashmap(STranslateContext* pCxt, SDropTableClause* pClause, bool* pIsSuperTable,
wmmhello's avatar
wmmhello 已提交
8367
                                           SHashObj* pVgroupHashmap) {
D
dapan1121 已提交
8368 8369
  SName name;
  toName(pCxt->pParseCxt->acctId, pClause->dbName, pClause->tableName, &name);
wmmhello's avatar
wmmhello 已提交
8370
  STableMeta* pTableMeta = NULL;
D
dapan1121 已提交
8371 8372 8373 8374
  int32_t     code = getTableMetaImpl(pCxt, &name, &pTableMeta);
  if (TSDB_CODE_SUCCESS == code) {
    code = collectUseTable(&name, pCxt->pTargetTables);
  }
X
Xiaoyu Wang 已提交
8375

wmmhello's avatar
wmmhello 已提交
8376 8377 8378 8379
  if (TSDB_CODE_SUCCESS == code && TSDB_SUPER_TABLE == pTableMeta->tableType) {
    *pIsSuperTable = true;
    goto over;
  }
X
Xiaoyu Wang 已提交
8380

wmmhello's avatar
wmmhello 已提交
8381 8382
  if (TSDB_CODE_PAR_TABLE_NOT_EXIST == code && pClause->ignoreNotExists) {
    code = TSDB_CODE_SUCCESS;
8383
    goto over;
wmmhello's avatar
wmmhello 已提交
8384
  }
8385

wmmhello's avatar
wmmhello 已提交
8386
  *pIsSuperTable = false;
X
Xiaoyu Wang 已提交
8387

wmmhello's avatar
wmmhello 已提交
8388 8389 8390 8391 8392
  SVgroupInfo info = {0};
  if (TSDB_CODE_SUCCESS == code) {
    code = getTableHashVgroup(pCxt, pClause->dbName, pClause->tableName, &info);
  }
  if (TSDB_CODE_SUCCESS == code) {
8393
    addDropTbReqIntoVgroup(pVgroupHashmap, pClause, &info, pTableMeta->suid);
wmmhello's avatar
wmmhello 已提交
8394
  }
X
Xiaoyu Wang 已提交
8395 8396

over:
wmmhello's avatar
wmmhello 已提交
8397 8398
  taosMemoryFreeClear(pTableMeta);
  return code;
X
Xiaoyu Wang 已提交
8399 8400
}

wmmhello's avatar
wmmhello 已提交
8401 8402 8403 8404
static void destroyDropTbReqBatch(void* data) {
  SVgroupDropTableBatch* pTbBatch = (SVgroupDropTableBatch*)data;
  taosArrayDestroy(pTbBatch->req.pArray);
}
X
Xiaoyu Wang 已提交
8405 8406

static int32_t serializeVgroupDropTableBatch(SVgroupDropTableBatch* pTbBatch, SArray* pBufArray) {
wmmhello's avatar
wmmhello 已提交
8407 8408 8409 8410 8411 8412 8413 8414 8415 8416 8417 8418 8419 8420 8421 8422 8423 8424 8425 8426 8427 8428 8429 8430 8431 8432 8433 8434 8435
  int      tlen;
  SEncoder coder = {0};

  int32_t ret = 0;
  tEncodeSize(tEncodeSVDropTbBatchReq, &pTbBatch->req, tlen, ret);
  tlen += sizeof(SMsgHead);
  void* buf = taosMemoryMalloc(tlen);
  if (NULL == buf) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  ((SMsgHead*)buf)->vgId = htonl(pTbBatch->info.vgId);
  ((SMsgHead*)buf)->contLen = htonl(tlen);
  void* pBuf = POINTER_SHIFT(buf, sizeof(SMsgHead));

  tEncoderInit(&coder, pBuf, tlen - sizeof(SMsgHead));
  tEncodeSVDropTbBatchReq(&coder, &pTbBatch->req);
  tEncoderClear(&coder);

  SVgDataBlocks* pVgData = taosMemoryCalloc(1, sizeof(SVgDataBlocks));
  if (NULL == pVgData) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pVgData->vg = pTbBatch->info;
  pVgData->pData = buf;
  pVgData->size = tlen;
  pVgData->numOfTables = (int32_t)taosArrayGetSize(pTbBatch->req.pArray);
  taosArrayPush(pBufArray, &pVgData);

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8436 8437
}

wmmhello's avatar
wmmhello 已提交
8438
SArray* serializeVgroupsDropTableBatch(SHashObj* pVgroupHashmap) {
wmmhello's avatar
wmmhello 已提交
8439 8440 8441 8442
  SArray* pBufArray = taosArrayInit(taosHashGetSize(pVgroupHashmap), sizeof(void*));
  if (NULL == pBufArray) {
    return NULL;
  }
X
Xiaoyu Wang 已提交
8443

wmmhello's avatar
wmmhello 已提交
8444 8445 8446 8447 8448 8449 8450
  int32_t                code = TSDB_CODE_SUCCESS;
  SVgroupDropTableBatch* pTbBatch = NULL;
  do {
    pTbBatch = taosHashIterate(pVgroupHashmap, pTbBatch);
    if (pTbBatch == NULL) {
      break;
    }
X
Xiaoyu Wang 已提交
8451

wmmhello's avatar
wmmhello 已提交
8452 8453
    serializeVgroupDropTableBatch(pTbBatch, pBufArray);
  } while (true);
X
Xiaoyu Wang 已提交
8454

wmmhello's avatar
wmmhello 已提交
8455
  return pBufArray;
X
Xiaoyu Wang 已提交
8456 8457 8458
}

static int32_t rewriteDropTable(STranslateContext* pCxt, SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
8459 8460 8461 8462 8463 8464 8465
  SDropTableStmt* pStmt = (SDropTableStmt*)pQuery->pRoot;

  SHashObj* pVgroupHashmap = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_NO_LOCK);
  if (NULL == pVgroupHashmap) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

wmmhello's avatar
wmmhello 已提交
8466
  taosHashSetFreeFp(pVgroupHashmap, destroyDropTbReqBatch);
wmmhello's avatar
wmmhello 已提交
8467 8468 8469 8470 8471 8472 8473 8474 8475 8476 8477 8478 8479
  bool   isSuperTable = false;
  SNode* pNode;
  FOREACH(pNode, pStmt->pTables) {
    int32_t code = buildDropTableVgroupHashmap(pCxt, (SDropTableClause*)pNode, &isSuperTable, pVgroupHashmap);
    if (TSDB_CODE_SUCCESS != code) {
      taosHashCleanup(pVgroupHashmap);
      return code;
    }
    if (isSuperTable && LIST_LENGTH(pStmt->pTables) > 1) {
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DROP_STABLE);
    }
  }

8480
  if (isSuperTable || 0 == taosHashGetSize(pVgroupHashmap)) {
wmmhello's avatar
wmmhello 已提交
8481 8482 8483 8484
    taosHashCleanup(pVgroupHashmap);
    return TSDB_CODE_SUCCESS;
  }

wmmhello's avatar
wmmhello 已提交
8485
  SArray* pBufArray = serializeVgroupsDropTableBatch(pVgroupHashmap);
wmmhello's avatar
wmmhello 已提交
8486 8487 8488 8489 8490 8491
  taosHashCleanup(pVgroupHashmap);
  if (NULL == pBufArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  return rewriteToVnodeModifyOpStmt(pQuery, pBufArray);
X
Xiaoyu Wang 已提交
8492 8493
}

X
Xiaoyu Wang 已提交
8494
static int32_t buildUpdateTagValReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8495
                                    SVAlterTbReq* pReq) {
X
Xiaoyu Wang 已提交
8496
  SSchema* pSchema = getTagSchema(pTableMeta, pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8497
  if (NULL == pSchema) {
X
Xiaoyu Wang 已提交
8498 8499
    return generateSyntaxErrMsgExt(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE, "Invalid tag name: %s",
                                   pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8500 8501
  }

8502
  pReq->tagName = taosStrdup(pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8503 8504 8505
  if (NULL == pReq->tagName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8506
  pReq->colId = pSchema->colId;
wmmhello's avatar
wmmhello 已提交
8507

D
dapan1121 已提交
8508
  SDataType targetDt = schemaToDataType(pTableMeta->tableInfo.precision, pSchema);
wmmhello's avatar
wmmhello 已提交
8509

D
dapan1121 已提交
8510
  if (QUERY_NODE_VALUE != pStmt->pVal->node.type) {
H
Hongze Cheng 已提交
8511
    SValueNode* pVal = NULL;
D
dapan1121 已提交
8512
    pCxt->errCode = createTagValFromExpr(pCxt, targetDt, (SNode*)pStmt->pVal, &pVal);
D
dapan1121 已提交
8513 8514 8515
    if (pCxt->errCode) {
      return pCxt->errCode;
    }
D
dapan1121 已提交
8516 8517 8518

    nodesDestroyNode((SNode*)pStmt->pVal);
    pStmt->pVal = pVal;
D
dapan1121 已提交
8519 8520 8521 8522
  } else if (DEAL_RES_ERROR == translateValueImpl(pCxt, pStmt->pVal, targetDt, true)) {
    return pCxt->errCode;
  }

wmmhello's avatar
wmmhello 已提交
8523
  pReq->tagType = targetDt.type;
D
dapan1121 已提交
8524
  if (targetDt.type == TSDB_DATA_TYPE_JSON) {
8525 8526
    if (pStmt->pVal->literal &&
        strlen(pStmt->pVal->literal) > (TSDB_MAX_JSON_TAG_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
wmmhello's avatar
wmmhello 已提交
8527 8528
      return buildSyntaxErrMsg(&pCxt->msgBuf, "json string too long than 4095", pStmt->pVal->literal);
    }
X
Xiaoyu Wang 已提交
8529
    SArray* pTagVals = taosArrayInit(1, sizeof(STagVal));
wmmhello's avatar
wmmhello 已提交
8530
    int32_t code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8531
    STag*   pTag = NULL;
wmmhello's avatar
wmmhello 已提交
8532
    code = parseJsontoTagData(pStmt->pVal->literal, pTagVals, &pTag, &pCxt->msgBuf);
wmmhello's avatar
wmmhello 已提交
8533
    taosArrayDestroy(pTagVals);
X
Xiaoyu Wang 已提交
8534
    if (code != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
8535
      return code;
wmmhello's avatar
wmmhello 已提交
8536
    }
wmmhello's avatar
wmmhello 已提交
8537
    pReq->nTagVal = pTag->len;
X
Xiaoyu Wang 已提交
8538
    pReq->pTagVal = (uint8_t*)pTag;
wmmhello's avatar
wmmhello 已提交
8539
    pStmt->pVal->datum.p = (char*)pTag;  // for free
8540
  } else {
8541
    pReq->isNull = pStmt->pVal->isNull;
wmmhello's avatar
wmmhello 已提交
8542 8543
    pReq->nTagVal = pStmt->pVal->node.resType.bytes;
    pReq->pTagVal = nodesGetValueFromNode(pStmt->pVal);
wmmhello's avatar
wmmhello 已提交
8544 8545 8546 8547 8548 8549

    // data and length are seperated for new tag format STagVal
    if (IS_VAR_DATA_TYPE(pStmt->pVal->node.resType.type)) {
      pReq->nTagVal = varDataLen(pReq->pTagVal);
      pReq->pTagVal = varDataVal(pReq->pTagVal);
    }
wmmhello's avatar
wmmhello 已提交
8550 8551 8552
  }

  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8553 8554 8555
}

static int32_t buildAddColReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8556 8557 8558 8559
                              SVAlterTbReq* pReq) {
  if (NULL != getColSchema(pTableMeta, pStmt->colName)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_DUPLICATED_COLUMN);
  }
X
Xiaoyu Wang 已提交
8560

8561 8562 8563 8564 8565
  if ((TSDB_DATA_TYPE_VARCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_BINARY_LEN) ||
      (TSDB_DATA_TYPE_NCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_NCHAR_LEN)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
  }

X
Xiaoyu Wang 已提交
8566 8567 8568 8569 8570 8571 8572 8573
  if (TSDB_MAX_COLUMNS == pTableMeta->tableInfo.numOfColumns) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_TOO_MANY_COLUMNS);
  }

  if (pTableMeta->tableInfo.rowSize + calcTypeBytes(pStmt->dataType) > TSDB_MAX_BYTES_PER_ROW) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ROW_LENGTH, TSDB_MAX_BYTES_PER_ROW);
  }

8574
  pReq->colName = taosStrdup(pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8575 8576 8577
  if (NULL == pReq->colName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
8578

wmmhello's avatar
wmmhello 已提交
8579 8580
  pReq->type = pStmt->dataType.type;
  pReq->flags = COL_SMA_ON;
8581
  pReq->bytes = calcTypeBytes(pStmt->dataType);
wmmhello's avatar
wmmhello 已提交
8582
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8583 8584
}

8585
static int32_t buildDropColReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, const STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8586 8587 8588 8589
                               SVAlterTbReq* pReq) {
  if (2 == getNumOfColumns(pTableMeta)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_DROP_COL);
  }
8590
  const SSchema* pSchema = getColSchema(pTableMeta, pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8591 8592 8593 8594 8595
  if (NULL == pSchema) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, pStmt->colName);
  } else if (PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_CANNOT_DROP_PRIMARY_KEY);
  }
X
Xiaoyu Wang 已提交
8596

8597
  pReq->colName = taosStrdup(pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8598 8599 8600
  if (NULL == pReq->colName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8601
  pReq->colId = pSchema->colId;
X
Xiaoyu Wang 已提交
8602

wmmhello's avatar
wmmhello 已提交
8603
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8604 8605
}

8606
static int32_t buildUpdateColReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, const STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8607 8608
                                 SVAlterTbReq* pReq) {
  pReq->colModBytes = calcTypeBytes(pStmt->dataType);
wmmhello's avatar
wmmhello 已提交
8609
  pReq->colModType = pStmt->dataType.type;
8610
  const SSchema* pSchema = getColSchema(pTableMeta, pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8611 8612 8613 8614 8615 8616
  if (NULL == pSchema) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, pStmt->colName);
  } else if (!IS_VAR_DATA_TYPE(pSchema->type) || pSchema->type != pStmt->dataType.type ||
             pSchema->bytes >= pReq->colModBytes) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_MODIFY_COL);
  }
X
Xiaoyu Wang 已提交
8617

8618 8619 8620 8621 8622
  if ((TSDB_DATA_TYPE_VARCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_BINARY_LEN) ||
      (TSDB_DATA_TYPE_NCHAR == pStmt->dataType.type && calcTypeBytes(pStmt->dataType) > TSDB_MAX_NCHAR_LEN)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_VAR_COLUMN_LEN);
  }

X
Xiaoyu Wang 已提交
8623 8624 8625 8626
  if (pTableMeta->tableInfo.rowSize + pReq->colModBytes - pSchema->bytes > TSDB_MAX_BYTES_PER_ROW) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ROW_LENGTH, TSDB_MAX_BYTES_PER_ROW);
  }

8627
  pReq->colName = taosStrdup(pStmt->colName);
wmmhello's avatar
wmmhello 已提交
8628 8629 8630
  if (NULL == pReq->colName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
8631
  pReq->colId = pSchema->colId;
X
Xiaoyu Wang 已提交
8632

wmmhello's avatar
wmmhello 已提交
8633
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8634 8635 8636
}

static int32_t buildRenameColReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8637 8638 8639 8640 8641 8642 8643 8644
                                 SVAlterTbReq* pReq) {
  if (NULL == getColSchema(pTableMeta, pStmt->colName)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COLUMN, pStmt->colName);
  }
  if (NULL != getColSchema(pTableMeta, pStmt->newColName)) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_DUPLICATED_COLUMN);
  }

8645 8646
  pReq->colName = taosStrdup(pStmt->colName);
  pReq->colNewName = taosStrdup(pStmt->newColName);
wmmhello's avatar
wmmhello 已提交
8647 8648 8649 8650
  if (NULL == pReq->colName || NULL == pReq->colNewName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8651 8652 8653
}

static int32_t buildUpdateOptionsReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, SVAlterTbReq* pReq) {
wmmhello's avatar
wmmhello 已提交
8654
  int32_t code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
8655

wmmhello's avatar
wmmhello 已提交
8656
  if (-1 != pStmt->pOptions->ttl) {
8657 8658
    pReq->updateTTL = true;
    pReq->newTTL = pStmt->pOptions->ttl;
wmmhello's avatar
wmmhello 已提交
8659
  }
X
Xiaoyu Wang 已提交
8660

X
Xiaoyu Wang 已提交
8661 8662
  if (TSDB_CODE_SUCCESS == code) {
    if (pStmt->pOptions->commentNull == false) {
8663
      pReq->newComment = taosStrdup(pStmt->pOptions->comment);
wmmhello's avatar
wmmhello 已提交
8664 8665
      if (NULL == pReq->newComment) {
        code = TSDB_CODE_OUT_OF_MEMORY;
X
Xiaoyu Wang 已提交
8666 8667
      } else {
        pReq->newCommentLen = strlen(pReq->newComment);
wmmhello's avatar
wmmhello 已提交
8668
      }
X
Xiaoyu Wang 已提交
8669
    } else {
wmmhello's avatar
wmmhello 已提交
8670
      pReq->newCommentLen = -1;
wmmhello's avatar
wmmhello 已提交
8671 8672
    }
  }
X
Xiaoyu Wang 已提交
8673

wmmhello's avatar
wmmhello 已提交
8674
  return code;
X
Xiaoyu Wang 已提交
8675 8676 8677
}

static int32_t buildAlterTbReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
8678
                               SVAlterTbReq* pReq) {
8679
  pReq->tbName = taosStrdup(pStmt->tableName);
wmmhello's avatar
wmmhello 已提交
8680 8681 8682 8683 8684 8685 8686 8687 8688 8689 8690 8691 8692 8693 8694 8695 8696 8697 8698 8699 8700 8701
  if (NULL == pReq->tbName) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pReq->action = pStmt->alterType;

  switch (pStmt->alterType) {
    case TSDB_ALTER_TABLE_ADD_TAG:
    case TSDB_ALTER_TABLE_DROP_TAG:
    case TSDB_ALTER_TABLE_UPDATE_TAG_NAME:
    case TSDB_ALTER_TABLE_UPDATE_TAG_BYTES:
      return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE);
    case TSDB_ALTER_TABLE_UPDATE_TAG_VAL:
      return buildUpdateTagValReq(pCxt, pStmt, pTableMeta, pReq);
    case TSDB_ALTER_TABLE_ADD_COLUMN:
      return buildAddColReq(pCxt, pStmt, pTableMeta, pReq);
    case TSDB_ALTER_TABLE_DROP_COLUMN:
      return buildDropColReq(pCxt, pStmt, pTableMeta, pReq);
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_BYTES:
      return buildUpdateColReq(pCxt, pStmt, pTableMeta, pReq);
    case TSDB_ALTER_TABLE_UPDATE_OPTIONS:
      return buildUpdateOptionsReq(pCxt, pStmt, pReq);
    case TSDB_ALTER_TABLE_UPDATE_COLUMN_NAME:
X
Xiaoyu Wang 已提交
8702 8703 8704 8705 8706
      if (TSDB_CHILD_TABLE == pTableMeta->tableType) {
        return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE);
      } else {
        return buildRenameColReq(pCxt, pStmt, pTableMeta, pReq);
      }
wmmhello's avatar
wmmhello 已提交
8707 8708 8709 8710 8711
    default:
      break;
  }

  return TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
8712 8713 8714
}

static int32_t serializeAlterTbReq(STranslateContext* pCxt, SAlterTableStmt* pStmt, SVAlterTbReq* pReq,
wmmhello's avatar
wmmhello 已提交
8715 8716 8717 8718 8719 8720 8721 8722 8723 8724 8725 8726 8727 8728 8729 8730 8731 8732 8733 8734 8735 8736 8737 8738 8739 8740 8741 8742 8743 8744 8745 8746 8747 8748
                                   SArray* pArray) {
  SVgroupInfo vg = {0};
  int32_t     code = getTableHashVgroup(pCxt, pStmt->dbName, pStmt->tableName, &vg);
  int         tlen = 0;
  if (TSDB_CODE_SUCCESS == code) {
    tEncodeSize(tEncodeSVAlterTbReq, pReq, tlen, code);
  }
  if (TSDB_CODE_SUCCESS == code) {
    tlen += sizeof(SMsgHead);
    void* pMsg = taosMemoryMalloc(tlen);
    if (NULL == pMsg) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    ((SMsgHead*)pMsg)->vgId = htonl(vg.vgId);
    ((SMsgHead*)pMsg)->contLen = htonl(tlen);
    void*    pBuf = POINTER_SHIFT(pMsg, sizeof(SMsgHead));
    SEncoder coder = {0};
    tEncoderInit(&coder, pBuf, tlen - sizeof(SMsgHead));
    tEncodeSVAlterTbReq(&coder, pReq);
    tEncoderClear(&coder);

    SVgDataBlocks* pVgData = taosMemoryCalloc(1, sizeof(SVgDataBlocks));
    if (NULL == pVgData) {
      taosMemoryFree(pMsg);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    pVgData->vg = vg;
    pVgData->pData = pMsg;
    pVgData->size = tlen;
    pVgData->numOfTables = 1;
    taosArrayPush(pArray, &pVgData);
  }

  return code;
X
Xiaoyu Wang 已提交
8749 8750 8751
}

static int32_t buildModifyVnodeArray(STranslateContext* pCxt, SAlterTableStmt* pStmt, SVAlterTbReq* pReq,
wmmhello's avatar
wmmhello 已提交
8752 8753 8754 8755 8756
                                     SArray** pArray) {
  SArray* pTmpArray = taosArrayInit(1, sizeof(void*));
  if (NULL == pTmpArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
8757

wmmhello's avatar
wmmhello 已提交
8758 8759 8760 8761 8762 8763
  int32_t code = serializeAlterTbReq(pCxt, pStmt, pReq, pTmpArray);
  if (TSDB_CODE_SUCCESS == code) {
    *pArray = pTmpArray;
  } else {
    taosArrayDestroy(pTmpArray);
  }
X
Xiaoyu Wang 已提交
8764

wmmhello's avatar
wmmhello 已提交
8765
  return code;
8766 8767
}

8768 8769 8770 8771 8772 8773 8774 8775
static void destoryAlterTbReq(SVAlterTbReq* pReq) {
  taosMemoryFree(pReq->tbName);
  taosMemoryFree(pReq->colName);
  taosMemoryFree(pReq->colNewName);
  taosMemoryFree(pReq->tagName);
  taosMemoryFree(pReq->newComment);
}

8776 8777
static int32_t rewriteAlterTableImpl(STranslateContext* pCxt, SAlterTableStmt* pStmt, STableMeta* pTableMeta,
                                     SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
8778 8779 8780 8781 8782
  if (TSDB_SUPER_TABLE == pTableMeta->tableType) {
    return TSDB_CODE_SUCCESS;
  } else if (TSDB_CHILD_TABLE != pTableMeta->tableType && TSDB_NORMAL_TABLE != pTableMeta->tableType) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_ALTER_TABLE);
  }
X
Xiaoyu Wang 已提交
8783

wmmhello's avatar
wmmhello 已提交
8784
  SVAlterTbReq req = {0};
8785
  int32_t      code = buildAlterTbReq(pCxt, pStmt, pTableMeta, &req);
X
Xiaoyu Wang 已提交
8786

wmmhello's avatar
wmmhello 已提交
8787 8788 8789 8790 8791 8792 8793
  SArray* pArray = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = buildModifyVnodeArray(pCxt, pStmt, &req, &pArray);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteToVnodeModifyOpStmt(pQuery, pArray);
  }
8794
  destoryAlterTbReq(&req);
wmmhello's avatar
wmmhello 已提交
8795
  return code;
8796 8797
}

8798 8799 8800 8801 8802 8803 8804 8805 8806 8807 8808 8809 8810 8811 8812 8813
static int32_t rewriteAlterTable(STranslateContext* pCxt, SQuery* pQuery) {
  SAlterTableStmt* pStmt = (SAlterTableStmt*)pQuery->pRoot;

  if (pStmt->dataType.type == TSDB_DATA_TYPE_JSON && pStmt->alterType == TSDB_ALTER_TABLE_ADD_COLUMN) {
    return generateSyntaxErrMsg(&pCxt->msgBuf, TSDB_CODE_PAR_INVALID_COL_JSON);
  }

  STableMeta* pTableMeta = NULL;
  int32_t     code = getTableMeta(pCxt, pStmt->dbName, pStmt->tableName, &pTableMeta);
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteAlterTableImpl(pCxt, pStmt, pTableMeta, pQuery);
  }
  taosMemoryFree(pTableMeta);
  return code;
}

X
Xiaoyu Wang 已提交
8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828 8829 8830 8831 8832 8833 8834 8835 8836 8837 8838 8839 8840 8841 8842 8843 8844 8845 8846 8847 8848 8849 8850 8851 8852 8853 8854 8855 8856 8857 8858 8859 8860 8861 8862 8863 8864 8865 8866 8867 8868 8869 8870 8871 8872 8873 8874
static int32_t serializeFlushVgroup(SVgroupInfo* pVg, SArray* pBufArray) {
  int32_t len = sizeof(SMsgHead);
  void*   buf = taosMemoryMalloc(len);
  if (NULL == buf) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  ((SMsgHead*)buf)->vgId = htonl(pVg->vgId);
  ((SMsgHead*)buf)->contLen = htonl(len);

  SVgDataBlocks* pVgData = taosMemoryCalloc(1, sizeof(SVgDataBlocks));
  if (NULL == pVgData) {
    taosMemoryFree(buf);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  pVgData->vg = *pVg;
  pVgData->pData = buf;
  pVgData->size = len;
  taosArrayPush(pBufArray, &pVgData);

  return TSDB_CODE_SUCCESS;
}

static int32_t serializeFlushDb(SArray* pVgs, SArray** pOutput) {
  int32_t numOfVgs = taosArrayGetSize(pVgs);

  SArray* pBufArray = taosArrayInit(numOfVgs, sizeof(void*));
  if (NULL == pBufArray) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  for (int32_t i = 0; i < numOfVgs; ++i) {
    int32_t code = serializeFlushVgroup((SVgroupInfo*)taosArrayGet(pVgs, i), pBufArray);
    if (TSDB_CODE_SUCCESS != code) {
      taosArrayDestroy(pBufArray);
      return code;
    }
  }

  *pOutput = pBufArray;
  return TSDB_CODE_SUCCESS;
}

static int32_t rewriteFlushDatabase(STranslateContext* pCxt, SQuery* pQuery) {
  SFlushDatabaseStmt* pStmt = (SFlushDatabaseStmt*)pQuery->pRoot;

  SArray* pBufArray = NULL;
  SArray* pVgs = NULL;
  int32_t code = getDBVgInfo(pCxt, pStmt->dbName, &pVgs);
  if (TSDB_CODE_SUCCESS == code) {
    code = serializeFlushDb(pVgs, &pBufArray);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteToVnodeModifyOpStmt(pQuery, pBufArray);
  }
  if (TSDB_CODE_SUCCESS != code) {
    taosArrayDestroy(pBufArray);
  }
  taosArrayDestroy(pVgs);
  return code;
}

8875
static int32_t rewriteQuery(STranslateContext* pCxt, SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
8876 8877
  int32_t code = TSDB_CODE_SUCCESS;
  switch (nodeType(pQuery->pRoot)) {
X
Xiaoyu Wang 已提交
8878
    case QUERY_NODE_SHOW_LICENCES_STMT:
wmmhello's avatar
wmmhello 已提交
8879 8880 8881 8882 8883 8884 8885 8886 8887 8888 8889 8890 8891 8892 8893 8894 8895 8896
    case QUERY_NODE_SHOW_DATABASES_STMT:
    case QUERY_NODE_SHOW_TABLES_STMT:
    case QUERY_NODE_SHOW_STABLES_STMT:
    case QUERY_NODE_SHOW_USERS_STMT:
    case QUERY_NODE_SHOW_DNODES_STMT:
    case QUERY_NODE_SHOW_MNODES_STMT:
    case QUERY_NODE_SHOW_MODULES_STMT:
    case QUERY_NODE_SHOW_QNODES_STMT:
    case QUERY_NODE_SHOW_FUNCTIONS_STMT:
    case QUERY_NODE_SHOW_INDEXES_STMT:
    case QUERY_NODE_SHOW_STREAMS_STMT:
    case QUERY_NODE_SHOW_BNODES_STMT:
    case QUERY_NODE_SHOW_SNODES_STMT:
    case QUERY_NODE_SHOW_CONNECTIONS_STMT:
    case QUERY_NODE_SHOW_QUERIES_STMT:
    case QUERY_NODE_SHOW_CLUSTER_STMT:
    case QUERY_NODE_SHOW_TOPICS_STMT:
    case QUERY_NODE_SHOW_TRANSACTIONS_STMT:
D
dapan1121 已提交
8897
    case QUERY_NODE_SHOW_APPS_STMT:
8898 8899
    case QUERY_NODE_SHOW_CONSUMERS_STMT:
    case QUERY_NODE_SHOW_SUBSCRIPTIONS_STMT:
8900
    case QUERY_NODE_SHOW_TAGS_STMT:
8901
    case QUERY_NODE_SHOW_USER_PRIVILEGES_STMT:
wmmhello's avatar
wmmhello 已提交
8902 8903
      code = rewriteShow(pCxt, pQuery);
      break;
X
Xiaoyu Wang 已提交
8904 8905 8906
    case QUERY_NODE_SHOW_VGROUPS_STMT:
      code = rewriteShowVgroups(pCxt, pQuery);
      break;
8907 8908 8909
    case QUERY_NODE_SHOW_TABLE_TAGS_STMT:
      code = rewriteShowStableTags(pCxt, pQuery);
      break;
8910 8911 8912
    case QUERY_NODE_SHOW_DNODE_VARIABLES_STMT:
      code = rewriteShowDnodeVariables(pCxt, pQuery);
      break;
8913 8914 8915
    case QUERY_NODE_SHOW_VNODES_STMT:
      code = rewriteShowVnodes(pCxt, pQuery);
      break;
8916 8917 8918
    case QUERY_NODE_SHOW_TABLE_DISTRIBUTED_STMT:
      code = rewriteShowTableDist(pCxt, pQuery);
      break;
wmmhello's avatar
wmmhello 已提交
8919 8920 8921 8922 8923
    case QUERY_NODE_CREATE_TABLE_STMT:
      if (NULL == ((SCreateTableStmt*)pQuery->pRoot)->pTags) {
        code = rewriteCreateTable(pCxt, pQuery);
      }
      break;
X
Xiaoyu Wang 已提交
8924
    case QUERY_NODE_CREATE_MULTI_TABLES_STMT:
wmmhello's avatar
wmmhello 已提交
8925 8926 8927 8928 8929 8930 8931 8932
      code = rewriteCreateMultiTable(pCxt, pQuery);
      break;
    case QUERY_NODE_DROP_TABLE_STMT:
      code = rewriteDropTable(pCxt, pQuery);
      break;
    case QUERY_NODE_ALTER_TABLE_STMT:
      code = rewriteAlterTable(pCxt, pQuery);
      break;
X
Xiaoyu Wang 已提交
8933 8934 8935
    case QUERY_NODE_FLUSH_DATABASE_STMT:
      code = rewriteFlushDatabase(pCxt, pQuery);
      break;
wmmhello's avatar
wmmhello 已提交
8936 8937 8938 8939
    default:
      break;
  }
  return code;
8940 8941
}

D
dapan1121 已提交
8942 8943 8944 8945 8946 8947 8948 8949 8950 8951 8952 8953 8954 8955 8956 8957
static int32_t toMsgType(ENodeType type) {
  switch (type) {
    case QUERY_NODE_CREATE_TABLE_STMT:
      return TDMT_VND_CREATE_TABLE;
    case QUERY_NODE_ALTER_TABLE_STMT:
      return TDMT_VND_ALTER_TABLE;
    case QUERY_NODE_DROP_TABLE_STMT:
      return TDMT_VND_DROP_TABLE;
    default:
      break;
  }
  return TDMT_VND_CREATE_TABLE;
}

static int32_t setRefreshMate(STranslateContext* pCxt, SQuery* pQuery) {
  if (NULL != pCxt->pDbs) {
8958
    taosArrayDestroy(pQuery->pDbList);
D
dapan1121 已提交
8959 8960 8961 8962 8963 8964 8965 8966 8967 8968 8969 8970
    pQuery->pDbList = taosArrayInit(taosHashGetSize(pCxt->pDbs), TSDB_DB_FNAME_LEN);
    if (NULL == pQuery->pDbList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    SFullDatabaseName* pDb = taosHashIterate(pCxt->pDbs, NULL);
    while (NULL != pDb) {
      taosArrayPush(pQuery->pDbList, pDb->fullDbName);
      pDb = taosHashIterate(pCxt->pDbs, pDb);
    }
  }

  if (NULL != pCxt->pTables) {
8971
    taosArrayDestroy(pQuery->pTableList);
D
dapan1121 已提交
8972 8973 8974 8975 8976 8977 8978 8979 8980 8981
    pQuery->pTableList = taosArrayInit(taosHashGetSize(pCxt->pTables), sizeof(SName));
    if (NULL == pQuery->pTableList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    SName* pTable = taosHashIterate(pCxt->pTables, NULL);
    while (NULL != pTable) {
      taosArrayPush(pQuery->pTableList, pTable);
      pTable = taosHashIterate(pCxt->pTables, pTable);
    }
  }
D
dapan1121 已提交
8982 8983 8984 8985 8986 8987 8988 8989 8990 8991 8992 8993 8994 8995

  if (NULL != pCxt->pTargetTables) {
    taosArrayDestroy(pQuery->pTargetTableList);
    pQuery->pTargetTableList = taosArrayInit(taosHashGetSize(pCxt->pTargetTables), sizeof(SName));
    if (NULL == pQuery->pTargetTableList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    SName* pTable = taosHashIterate(pCxt->pTargetTables, NULL);
    while (NULL != pTable) {
      taosArrayPush(pQuery->pTargetTableList, pTable);
      pTable = taosHashIterate(pCxt->pTargetTables, pTable);
    }
  }

D
dapan1121 已提交
8996 8997 8998
  return TSDB_CODE_SUCCESS;
}

8999
static int32_t setQuery(STranslateContext* pCxt, SQuery* pQuery) {
wmmhello's avatar
wmmhello 已提交
9000 9001
  switch (nodeType(pQuery->pRoot)) {
    case QUERY_NODE_SELECT_STMT:
9002 9003 9004 9005 9006
      if (NULL == ((SSelectStmt*)pQuery->pRoot)->pFromTable) {
        pQuery->execMode = QUERY_EXEC_MODE_LOCAL;
        pQuery->haveResultSet = true;
        break;
      }
wmmhello's avatar
wmmhello 已提交
9007 9008 9009 9010
    case QUERY_NODE_SET_OPERATOR:
    case QUERY_NODE_EXPLAIN_STMT:
      pQuery->execMode = QUERY_EXEC_MODE_SCHEDULE;
      pQuery->haveResultSet = true;
D
dapan1121 已提交
9011
      pQuery->msgType = TDMT_SCH_QUERY;
wmmhello's avatar
wmmhello 已提交
9012
      break;
D
dapan1121 已提交
9013 9014 9015 9016
    case QUERY_NODE_DELETE_STMT:
      pQuery->execMode = QUERY_EXEC_MODE_SCHEDULE;
      pQuery->msgType = TDMT_VND_DELETE;
      break;
9017 9018 9019 9020
    case QUERY_NODE_INSERT_STMT:
      pQuery->execMode = QUERY_EXEC_MODE_SCHEDULE;
      pQuery->msgType = TDMT_VND_SUBMIT;
      break;
X
Xiaoyu Wang 已提交
9021
    case QUERY_NODE_VNODE_MODIFY_STMT:
wmmhello's avatar
wmmhello 已提交
9022
      pQuery->execMode = QUERY_EXEC_MODE_SCHEDULE;
X
Xiaoyu Wang 已提交
9023
      pQuery->msgType = toMsgType(((SVnodeModifyOpStmt*)pQuery->pRoot)->sqlNodeType);
wmmhello's avatar
wmmhello 已提交
9024 9025
      break;
    case QUERY_NODE_DESCRIBE_STMT:
9026
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
9027
    case QUERY_NODE_SHOW_DB_ALIVE_STMT:
9028
    case QUERY_NODE_SHOW_CLUSTER_ALIVE_STMT:
9029 9030
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
D
dapan1121 已提交
9031
    case QUERY_NODE_SHOW_LOCAL_VARIABLES_STMT:
wmmhello's avatar
wmmhello 已提交
9032 9033 9034 9035
      pQuery->execMode = QUERY_EXEC_MODE_LOCAL;
      pQuery->haveResultSet = true;
      break;
    case QUERY_NODE_RESET_QUERY_CACHE_STMT:
9036
    case QUERY_NODE_ALTER_LOCAL_STMT:
wmmhello's avatar
wmmhello 已提交
9037 9038
      pQuery->execMode = QUERY_EXEC_MODE_LOCAL;
      break;
D
dapan1121 已提交
9039 9040 9041 9042 9043 9044 9045
    case QUERY_NODE_SHOW_VARIABLES_STMT:
      pQuery->haveResultSet = true;
      pQuery->execMode = QUERY_EXEC_MODE_RPC;
      if (NULL != pCxt->pCmdMsg) {
        TSWAP(pQuery->pCmdMsg, pCxt->pCmdMsg);
        pQuery->msgType = pQuery->pCmdMsg->msgType;
      }
9046
      break;
wmmhello's avatar
wmmhello 已提交
9047 9048 9049 9050 9051 9052 9053 9054 9055
    default:
      pQuery->execMode = QUERY_EXEC_MODE_RPC;
      if (NULL != pCxt->pCmdMsg) {
        TSWAP(pQuery->pCmdMsg, pCxt->pCmdMsg);
        pQuery->msgType = pQuery->pCmdMsg->msgType;
      }
      break;
  }

9056 9057
  pQuery->stableQuery = pCxt->stableQuery;

wmmhello's avatar
wmmhello 已提交
9058
  if (pQuery->haveResultSet) {
9059
    taosMemoryFreeClear(pQuery->pResSchema);
wmmhello's avatar
wmmhello 已提交
9060 9061 9062 9063 9064 9065 9066 9067 9068 9069
    if (TSDB_CODE_SUCCESS != extractResultSchema(pQuery->pRoot, &pQuery->numOfResCols, &pQuery->pResSchema)) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }

    if (nodeType(pQuery->pRoot) == QUERY_NODE_SELECT_STMT) {
      pQuery->precision = extractResultTsPrecision((SSelectStmt*)pQuery->pRoot);
    }
  }

  return TSDB_CODE_SUCCESS;
9070 9071
}

9072
int32_t translate(SParseContext* pParseCxt, SQuery* pQuery, SParseMetaCache* pMetaCache) {
wmmhello's avatar
wmmhello 已提交
9073 9074
  STranslateContext cxt = {0};

9075
  int32_t code = initTranslateContext(pParseCxt, pMetaCache, &cxt);
wmmhello's avatar
wmmhello 已提交
9076 9077 9078 9079 9080 9081 9082 9083 9084
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteQuery(&cxt, pQuery);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = translateQuery(&cxt, pQuery->pRoot);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = setQuery(&cxt, pQuery);
  }
D
dapan1121 已提交
9085
  setRefreshMate(&cxt, pQuery);
wmmhello's avatar
wmmhello 已提交
9086 9087 9088
  destroyTranslateContext(&cxt);
  return code;
}