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

wafwerar's avatar
wafwerar 已提交
16
#include "os.h"
X
Xiaoyu Wang 已提交
17 18 19
#include "parInsertData.h"
#include "parInt.h"
#include "parToken.h"
H
refact  
Hongze Cheng 已提交
20
#include "parUtil.h"
X
Xiaoyu Wang 已提交
21
#include "query.h"
22 23 24 25
#include "tglobal.h"
#include "ttime.h"
#include "ttypes.h"

H
refact  
Hongze Cheng 已提交
26 27 28
#define NEXT_TOKEN(pSql, sToken)                \
  do {                                          \
    int32_t index = 0;                          \
29
    sToken = tStrGetToken(pSql, &index, false); \
H
refact  
Hongze Cheng 已提交
30
    pSql += index;                              \
31 32
  } while (0)

H
refact  
Hongze Cheng 已提交
33 34 35
#define NEXT_TOKEN_WITH_PREV(pSql, sToken)     \
  do {                                         \
    int32_t index = 0;                         \
X
Xiaoyu Wang 已提交
36
    sToken = tStrGetToken(pSql, &index, true); \
H
refact  
Hongze Cheng 已提交
37
    pSql += index;                             \
X
Xiaoyu Wang 已提交
38 39
  } while (0)

40
#define NEXT_TOKEN_KEEP_SQL(pSql, sToken, index) \
H
refact  
Hongze Cheng 已提交
41 42
  do {                                           \
    sToken = tStrGetToken(pSql, &index, false);  \
43 44
  } while (0)

45 46 47 48 49 50 51
#define NEXT_VALID_TOKEN(pSql, sToken)        \
  do {                                        \
    sToken.n = tGetToken(pSql, &sToken.type); \
    sToken.z = pSql;                          \
    pSql += sToken.n;                         \
  } while (TK_NK_SPACE == sToken.type)

52 53 54 55 56 57
typedef struct SInsertParseBaseContext {
  SParseContext* pComCxt;
  char*          pSql;
  SMsgBuf        msg;
} SInsertParseBaseContext;

58
typedef struct SInsertParseContext {
X
Xiaoyu Wang 已提交
59 60 61 62 63 64 65 66 67 68
  SParseContext*     pComCxt;             // input
  char*              pSql;                // input
  SMsgBuf            msg;                 // input
  STableMeta*        pTableMeta;          // each table
  SParsedDataColInfo tags;                // each table
  SVCreateTbReq      createTblReq;        // each table
  SHashObj*          pVgroupsHashObj;     // global
  SHashObj*          pTableBlockHashObj;  // global
  SHashObj*          pSubTableHashObj;    // global
  SArray*            pVgDataBlocks;       // global
X
Xiaoyu Wang 已提交
69
  SHashObj*          pTableNameHashObj;   // global
D
dapan1121 已提交
70
  SHashObj*          pDbFNameHashObj;     // global
X
Xiaoyu Wang 已提交
71
  int32_t            totalNum;
X
Xiaoyu Wang 已提交
72
  SVnodeModifOpStmt* pOutput;
X
Xiaoyu Wang 已提交
73
  SStmtCallback*     pStmtCb;
74
  SParseMetaCache*   pMetaCache;
wmmhello's avatar
wmmhello 已提交
75
  char               sTableName[TSDB_TABLE_NAME_LEN];
X
Xiaoyu Wang 已提交
76 77 78
  char               tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW];
  int64_t            memElapsed;
  int64_t            parRowElapsed;
79 80
} SInsertParseContext;

81 82 83 84 85 86 87
typedef struct SInsertParseSyntaxCxt {
  SParseContext*   pComCxt;
  char*            pSql;
  SMsgBuf          msg;
  SParseMetaCache* pMetaCache;
} SInsertParseSyntaxCxt;

H
refact  
Hongze Cheng 已提交
88
typedef int32_t (*_row_append_fn_t)(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param);
X
Xiaoyu Wang 已提交
89 90 91 92

static uint8_t TRUE_VALUE = (uint8_t)TSDB_TRUE;
static uint8_t FALSE_VALUE = (uint8_t)TSDB_FALSE;

D
stmt  
dapan1121 已提交
93
typedef struct SKvParam {
C
Cary Xu 已提交
94
  int16_t  pos;
C
Cary Xu 已提交
95
  SArray*  pTagVals;
C
Cary Xu 已提交
96 97
  SSchema* schema;
  char     buf[TSDB_MAX_TAGS_LEN];
D
stmt  
dapan1121 已提交
98 99 100 101 102 103 104 105 106
} SKvParam;

typedef struct SMemParam {
  SRowBuilder* rb;
  SSchema*     schema;
  int32_t      toffset;
  col_id_t     colIdx;
} SMemParam;

X
Xiaoyu Wang 已提交
107 108 109
#define CHECK_CODE(expr)             \
  do {                               \
    int32_t code = expr;             \
D
stmt  
dapan1121 已提交
110
    if (TSDB_CODE_SUCCESS != code) { \
X
Xiaoyu Wang 已提交
111 112
      return code;                   \
    }                                \
D
stmt  
dapan1121 已提交
113 114
  } while (0)

115
static int32_t skipInsertInto(char** pSql, SMsgBuf* pMsg) {
116
  SToken sToken;
117
  NEXT_TOKEN(*pSql, sToken);
118
  if (TK_INSERT != sToken.type && TK_IMPORT != sToken.type) {
119
    return buildSyntaxErrMsg(pMsg, "keyword INSERT is expected", sToken.z);
120
  }
121
  NEXT_TOKEN(*pSql, sToken);
122
  if (TK_INTO != sToken.type) {
123
    return buildSyntaxErrMsg(pMsg, "keyword INTO is expected", sToken.z);
124 125 126 127
  }
  return TSDB_CODE_SUCCESS;
}

128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
static char* tableNameGetPosition(SToken* pToken, char target) {
  bool inEscape = false;
  bool inQuote = false;
  char quotaStr = 0;

  for (uint32_t i = 0; i < pToken->n; ++i) {
    if (*(pToken->z + i) == target && (!inEscape) && (!inQuote)) {
      return pToken->z + i;
    }

    if (*(pToken->z + i) == TS_ESCAPE_CHAR) {
      if (!inQuote) {
        inEscape = !inEscape;
      }
    }

    if (*(pToken->z + i) == '\'' || *(pToken->z + i) == '"') {
      if (!inEscape) {
        if (!inQuote) {
          quotaStr = *(pToken->z + i);
          inQuote = !inQuote;
        } else if (quotaStr == *(pToken->z + i)) {
          inQuote = !inQuote;
        }
      }
    }
  }

  return NULL;
}

D
stmt  
dapan1121 已提交
159
static int32_t createSName(SName* pName, SToken* pTableName, int32_t acctId, const char* dbName, SMsgBuf* pMsgBuf) {
X
Xiaoyu Wang 已提交
160 161 162
  const char* msg1 = "name too long";
  const char* msg2 = "invalid database name";
  const char* msg3 = "db is not specified";
163
  const char* msg4 = "invalid table name";
X
Xiaoyu Wang 已提交
164

H
refact  
Hongze Cheng 已提交
165
  int32_t code = TSDB_CODE_SUCCESS;
166
  char*   p = tableNameGetPosition(pTableName, TS_PATH_DELIMITER[0]);
X
Xiaoyu Wang 已提交
167

H
refact  
Hongze Cheng 已提交
168
  if (p != NULL) {  // db has been specified in sql string so we ignore current db path
X
Xiaoyu Wang 已提交
169 170 171
    assert(*p == TS_PATH_DELIMITER[0]);

    int32_t dbLen = p - pTableName->z;
X
Xiaoyu Wang 已提交
172 173 174 175
    if (dbLen <= 0) {
      return buildInvalidOperationMsg(pMsgBuf, msg2);
    }
    char name[TSDB_DB_FNAME_LEN] = {0};
X
Xiaoyu Wang 已提交
176
    strncpy(name, pTableName->z, dbLen);
177
    int32_t actualDbLen = strdequote(name);
X
Xiaoyu Wang 已提交
178

179
    code = tNameSetDbName(pName, acctId, name, actualDbLen);
X
Xiaoyu Wang 已提交
180 181 182 183 184
    if (code != TSDB_CODE_SUCCESS) {
      return buildInvalidOperationMsg(pMsgBuf, msg1);
    }

    int32_t tbLen = pTableName->n - dbLen - 1;
185 186 187 188
    if (tbLen <= 0) {
      return buildInvalidOperationMsg(pMsgBuf, msg4);
    }

189
    char tbname[TSDB_TABLE_FNAME_LEN] = {0};
X
Xiaoyu Wang 已提交
190
    strncpy(tbname, p + 1, tbLen);
H
refact  
Hongze Cheng 已提交
191
    /*tbLen = */ strdequote(tbname);
X
Xiaoyu Wang 已提交
192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207

    code = tNameFromString(pName, tbname, T_NAME_TABLE);
    if (code != 0) {
      return buildInvalidOperationMsg(pMsgBuf, msg1);
    }
  } else {  // get current DB name first, and then set it into path
    if (pTableName->n >= TSDB_TABLE_NAME_LEN) {
      return buildInvalidOperationMsg(pMsgBuf, msg1);
    }

    assert(pTableName->n < TSDB_TABLE_FNAME_LEN);

    char name[TSDB_TABLE_FNAME_LEN] = {0};
    strncpy(name, pTableName->z, pTableName->n);
    strdequote(name);

D
stmt  
dapan1121 已提交
208
    if (dbName == NULL) {
X
Xiaoyu Wang 已提交
209 210 211
      return buildInvalidOperationMsg(pMsgBuf, msg3);
    }

D
stmt  
dapan1121 已提交
212
    code = tNameSetDbName(pName, acctId, dbName, strlen(dbName));
X
Xiaoyu Wang 已提交
213 214 215 216 217 218 219 220 221 222 223 224 225 226
    if (code != TSDB_CODE_SUCCESS) {
      code = buildInvalidOperationMsg(pMsgBuf, msg2);
      return code;
    }

    code = tNameFromString(pName, name, T_NAME_TABLE);
    if (code != 0) {
      code = buildInvalidOperationMsg(pMsgBuf, msg1);
    }
  }

  return code;
}

227 228
static int32_t checkAuth(SInsertParseContext* pCxt, char* pDbFname, bool* pPass) {
  SParseContext* pBasicCtx = pCxt->pComCxt;
X
Xiaoyu Wang 已提交
229
  if (pBasicCtx->async) {
230 231
    return getUserAuthFromCache(pCxt->pMetaCache, pBasicCtx->pUser, pDbFname, AUTH_TYPE_WRITE, pPass);
  }
X
Xiaoyu Wang 已提交
232
  SRequestConnInfo conn = {.pTrans = pBasicCtx->pTransporter,
D
dapan1121 已提交
233 234 235 236 237
                           .requestId = pBasicCtx->requestId,
                           .requestObjRefId = pBasicCtx->requestRid,
                           .mgmtEps = pBasicCtx->mgmtEpSet};

  return catalogChkAuth(pBasicCtx->pCatalog, &conn, pBasicCtx->pUser, pDbFname, AUTH_TYPE_WRITE, pPass);
238 239
}

X
Xiaoyu Wang 已提交
240 241
static int32_t getTableSchema(SInsertParseContext* pCxt, int32_t tbNo, SName* pTbName, bool isStb,
                              STableMeta** pTableMeta) {
242
  SParseContext* pBasicCtx = pCxt->pComCxt;
X
Xiaoyu Wang 已提交
243
  if (pBasicCtx->async) {
X
Xiaoyu Wang 已提交
244
    return getTableMetaFromCacheForInsert(pBasicCtx->pTableMetaPos, pCxt->pMetaCache, tbNo, pTableMeta);
245
  }
X
Xiaoyu Wang 已提交
246
  SRequestConnInfo conn = {.pTrans = pBasicCtx->pTransporter,
D
dapan1121 已提交
247 248 249
                           .requestId = pBasicCtx->requestId,
                           .requestObjRefId = pBasicCtx->requestRid,
                           .mgmtEps = pBasicCtx->mgmtEpSet};
X
Xiaoyu Wang 已提交
250

251
  if (isStb) {
D
dapan1121 已提交
252
    return catalogGetSTableMeta(pBasicCtx->pCatalog, &conn, pTbName, pTableMeta);
253
  }
D
dapan1121 已提交
254
  return catalogGetTableMeta(pBasicCtx->pCatalog, &conn, pTbName, pTableMeta);
255 256
}

X
Xiaoyu Wang 已提交
257
static int32_t getTableVgroup(SInsertParseContext* pCxt, int32_t tbNo, SName* pTbName, SVgroupInfo* pVg) {
H
Haojun Liao 已提交
258
  SParseContext* pBasicCtx = pCxt->pComCxt;
X
Xiaoyu Wang 已提交
259
  if (pBasicCtx->async) {
X
Xiaoyu Wang 已提交
260
    return getTableVgroupFromCacheForInsert(pBasicCtx->pTableVgroupPos, pCxt->pMetaCache, tbNo, pVg);
261
  }
X
Xiaoyu Wang 已提交
262
  SRequestConnInfo conn = {.pTrans = pBasicCtx->pTransporter,
D
dapan1121 已提交
263 264 265 266
                           .requestId = pBasicCtx->requestId,
                           .requestObjRefId = pBasicCtx->requestRid,
                           .mgmtEps = pBasicCtx->mgmtEpSet};
  return catalogGetTableHashVgroup(pBasicCtx->pCatalog, &conn, pTbName, pVg);
267
}
D
dapan 已提交
268

X
Xiaoyu Wang 已提交
269 270
static int32_t getTableMetaImpl(SInsertParseContext* pCxt, int32_t tbNo, SName* name, char* dbFname, bool isStb) {
  CHECK_CODE(getTableSchema(pCxt, tbNo, name, isStb, &pCxt->pTableMeta));
271
  if (!isStb) {
X
Xiaoyu Wang 已提交
272
    SVgroupInfo vg;
X
Xiaoyu Wang 已提交
273
    CHECK_CODE(getTableVgroup(pCxt, tbNo, name, &vg));
X
Xiaoyu Wang 已提交
274
    CHECK_CODE(taosHashPut(pCxt->pVgroupsHashObj, (const char*)&vg.vgId, sizeof(vg.vgId), (char*)&vg, sizeof(vg)));
D
stmt  
dapan1121 已提交
275
  }
H
refact  
Hongze Cheng 已提交
276
  return TSDB_CODE_SUCCESS;
D
stmt  
dapan1121 已提交
277 278
}

X
Xiaoyu Wang 已提交
279 280
static int32_t getTableMeta(SInsertParseContext* pCxt, int32_t tbNo, SName* name, char* dbFname) {
  return getTableMetaImpl(pCxt, tbNo, name, dbFname, false);
X
Xiaoyu Wang 已提交
281
}
D
stmt  
dapan1121 已提交
282

X
Xiaoyu Wang 已提交
283 284
static int32_t getSTableMeta(SInsertParseContext* pCxt, int32_t tbNo, SName* name, char* dbFname) {
  return getTableMetaImpl(pCxt, tbNo, name, dbFname, true);
X
Xiaoyu Wang 已提交
285
}
D
stmt  
dapan1121 已提交
286

X
Xiaoyu Wang 已提交
287 288 289 290 291
static int32_t getDBCfg(SInsertParseContext* pCxt, const char* pDbFName, SDbCfgInfo* pInfo) {
  SParseContext* pBasicCtx = pCxt->pComCxt;
  if (pBasicCtx->async) {
    CHECK_CODE(getDbCfgFromCache(pCxt->pMetaCache, pDbFName, pInfo));
  } else {
X
Xiaoyu Wang 已提交
292
    SRequestConnInfo conn = {.pTrans = pBasicCtx->pTransporter,
D
dapan1121 已提交
293 294 295 296
                             .requestId = pBasicCtx->requestId,
                             .requestObjRefId = pBasicCtx->requestRid,
                             .mgmtEps = pBasicCtx->mgmtEpSet};
    CHECK_CODE(catalogGetDBCfg(pBasicCtx->pCatalog, &conn, pDbFName, pInfo));
X
Xiaoyu Wang 已提交
297 298 299 300
  }
  return TSDB_CODE_SUCCESS;
}

301 302 303 304 305 306 307 308 309 310
static int32_t findCol(SToken* pColname, int32_t start, int32_t end, SSchema* pSchema) {
  while (start < end) {
    if (strlen(pSchema[start].name) == pColname->n && strncmp(pColname->z, pSchema[start].name, pColname->n) == 0) {
      return start;
    }
    ++start;
  }
  return -1;
}

D
dapan1121 已提交
311
static void buildMsgHeader(STableDataBlocks* src, SVgDataBlocks* blocks) {
H
refact  
Hongze Cheng 已提交
312 313 314 315 316 317 318 319 320
  SSubmitReq* submit = (SSubmitReq*)blocks->pData;
  submit->header.vgId = htonl(blocks->vg.vgId);
  submit->header.contLen = htonl(blocks->size);
  submit->length = submit->header.contLen;
  submit->numOfBlocks = htonl(blocks->numOfTables);
  SSubmitBlk* blk = (SSubmitBlk*)(submit + 1);
  int32_t     numOfBlocks = blocks->numOfTables;
  while (numOfBlocks--) {
    int32_t dataLen = blk->dataLen;
321
    int32_t schemaLen = blk->schemaLen;
H
refact  
Hongze Cheng 已提交
322 323 324 325 326
    blk->uid = htobe64(blk->uid);
    blk->suid = htobe64(blk->suid);
    blk->sversion = htonl(blk->sversion);
    blk->dataLen = htonl(blk->dataLen);
    blk->schemaLen = htonl(blk->schemaLen);
327
    blk->numOfRows = htonl(blk->numOfRows);
328
    blk = (SSubmitBlk*)(blk->data + schemaLen + dataLen);
H
refact  
Hongze Cheng 已提交
329
  }
330 331 332 333 334 335 336 337 338 339
}

static int32_t buildOutput(SInsertParseContext* pCxt) {
  size_t numOfVg = taosArrayGetSize(pCxt->pVgDataBlocks);
  pCxt->pOutput->pDataBlocks = taosArrayInit(numOfVg, POINTER_BYTES);
  if (NULL == pCxt->pOutput->pDataBlocks) {
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }
  for (size_t i = 0; i < numOfVg; ++i) {
    STableDataBlocks* src = taosArrayGetP(pCxt->pVgDataBlocks, i);
H
refact  
Hongze Cheng 已提交
340
    SVgDataBlocks*    dst = taosMemoryCalloc(1, sizeof(SVgDataBlocks));
341 342 343
    if (NULL == dst) {
      return TSDB_CODE_TSC_OUT_OF_MEMORY;
    }
H
Haojun Liao 已提交
344
    taosHashGetDup(pCxt->pVgroupsHashObj, (const char*)&src->vgId, sizeof(src->vgId), &dst->vg);
345 346
    dst->numOfTables = src->numOfTables;
    dst->size = src->size;
wafwerar's avatar
wafwerar 已提交
347
    TSWAP(dst->pData, src->pData);
D
dapan1121 已提交
348
    buildMsgHeader(src, dst);
349 350 351 352 353
    taosArrayPush(pCxt->pOutput->pDataBlocks, &dst);
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
354
int32_t checkTimestamp(STableDataBlocks* pDataBlocks, const char* start) {
355 356 357 358 359
  // once the data block is disordered, we do NOT keep previous timestamp any more
  if (!pDataBlocks->ordered) {
    return TSDB_CODE_SUCCESS;
  }

H
refact  
Hongze Cheng 已提交
360
  TSKEY k = *(TSKEY*)start;
361
  if (k <= pDataBlocks->prevTS) {
362 363 364 365 366 367 368
    pDataBlocks->ordered = false;
  }

  pDataBlocks->prevTS = k;
  return TSDB_CODE_SUCCESS;
}

H
refact  
Hongze Cheng 已提交
369 370 371 372 373 374
static int parseTime(char** end, SToken* pToken, int16_t timePrec, int64_t* time, SMsgBuf* pMsgBuf) {
  int32_t index = 0;
  SToken  sToken;
  int64_t interval;
  int64_t ts = 0;
  char*   pTokenEnd = *end;
375 376

  if (pToken->type == TK_NOW) {
377
    ts = taosGetTimestamp(timePrec);
378 379
  } else if (pToken->type == TK_TODAY) {
    ts = taosGetTimestampToday(timePrec);
380
  } else if (pToken->type == TK_NK_INTEGER) {
X
Xiaoyu Wang 已提交
381
    toInteger(pToken->z, pToken->n, 10, &ts);
H
refact  
Hongze Cheng 已提交
382
  } else {  // parse the RFC-3339/ISO-8601 timestamp format string
S
os env  
Shengliang Guan 已提交
383
    if (taosParseTime(pToken->z, time, pToken->n, timePrec, tsDaylight) != TSDB_CODE_SUCCESS) {
384
      return buildSyntaxErrMsg(pMsgBuf, "invalid timestamp format", pToken->z);
385 386 387 388 389 390 391
    }

    return TSDB_CODE_SUCCESS;
  }

  for (int k = pToken->n; pToken->z[k] != '\0'; k++) {
    if (pToken->z[k] == ' ' || pToken->z[k] == '\t') continue;
H
refact  
Hongze Cheng 已提交
392
    if (pToken->z[k] == '(' && pToken->z[k + 1] == ')') {  // for insert NOW()/TODAY()
393 394 395 396
      *end = pTokenEnd = &pToken->z[k + 2];
      k++;
      continue;
    }
397
    if (pToken->z[k] == ',') {
398 399
      *end = pTokenEnd;
      *time = ts;
400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
      return 0;
    }

    break;
  }

  /*
   * time expression:
   * e.g., now+12a, now-5h
   */
  SToken valueToken;
  index = 0;
  sToken = tStrGetToken(pTokenEnd, &index, false);
  pTokenEnd += index;

X
Xiaoyu Wang 已提交
415
  if (sToken.type == TK_NK_MINUS || sToken.type == TK_NK_PLUS) {
416 417 418 419 420
    index = 0;
    valueToken = tStrGetToken(pTokenEnd, &index, false);
    pTokenEnd += index;

    if (valueToken.n < 2) {
421
      return buildSyntaxErrMsg(pMsgBuf, "value expected in timestamp", sToken.z);
422 423 424 425 426 427 428
    }

    char unit = 0;
    if (parseAbsoluteDuration(valueToken.z, valueToken.n, &interval, &unit, timePrec) != TSDB_CODE_SUCCESS) {
      return TSDB_CODE_TSC_INVALID_OPERATION;
    }

429
    if (sToken.type == TK_NK_PLUS) {
430
      ts += interval;
431
    } else {
432
      ts = ts - interval;
433 434
    }

435
    *end = pTokenEnd;
436 437
  }

438
  *time = ts;
439 440
  return TSDB_CODE_SUCCESS;
}
441

wmmhello's avatar
wmmhello 已提交
442
static FORCE_INLINE int32_t checkAndTrimValue(SToken* pToken, char* tmpTokenBuf, SMsgBuf* pMsgBuf) {
H
refact  
Hongze Cheng 已提交
443 444 445 446
  if ((pToken->type != TK_NOW && pToken->type != TK_TODAY && pToken->type != TK_NK_INTEGER &&
       pToken->type != TK_NK_STRING && pToken->type != TK_NK_FLOAT && pToken->type != TK_NK_BOOL &&
       pToken->type != TK_NULL && pToken->type != TK_NK_HEX && pToken->type != TK_NK_OCT &&
       pToken->type != TK_NK_BIN) ||
447
      (pToken->n == 0) || (pToken->type == TK_NK_RP)) {
X
Xiaoyu Wang 已提交
448 449 450 451
    return buildSyntaxErrMsg(pMsgBuf, "invalid data or symbol", pToken->z);
  }

  // Remove quotation marks
X
Xiaoyu Wang 已提交
452
  if (TK_NK_STRING == pToken->type) {
X
Xiaoyu Wang 已提交
453 454 455 456
    if (pToken->n >= TSDB_MAX_BYTES_PER_ROW) {
      return buildSyntaxErrMsg(pMsgBuf, "too long string", pToken->z);
    }

457
    int32_t len = trimString(pToken->z, pToken->n, tmpTokenBuf, TSDB_MAX_BYTES_PER_ROW);
X
Xiaoyu Wang 已提交
458
    pToken->z = tmpTokenBuf;
459
    pToken->n = len;
X
Xiaoyu Wang 已提交
460 461 462 463 464
  }

  return TSDB_CODE_SUCCESS;
}

H
refact  
Hongze Cheng 已提交
465
static bool isNullStr(SToken* pToken) {
466
  return ((pToken->type == TK_NK_STRING) && (strlen(TSDB_DATA_NULL_STR_L) == pToken->n) &&
467
          (strncasecmp(TSDB_DATA_NULL_STR_L, pToken->z, pToken->n) == 0));
X
Xiaoyu Wang 已提交
468 469
}

470 471
static bool isNullValue(int8_t dataType, SToken* pToken) {
  return TK_NULL == pToken->type || (!IS_STR_DATA_TYPE(dataType) && isNullStr(pToken));
X
Xiaoyu Wang 已提交
472 473
}

H
refact  
Hongze Cheng 已提交
474
static FORCE_INLINE int32_t toDouble(SToken* pToken, double* value, char** endPtr) {
X
Xiaoyu Wang 已提交
475
  errno = 0;
wafwerar's avatar
wafwerar 已提交
476
  *value = taosStr2Double(pToken->z, endPtr);
X
Xiaoyu Wang 已提交
477 478 479

  // not a valid integer number, return error
  if ((*endPtr - pToken->z) != pToken->n) {
480
    return TK_NK_ILLEGAL;
X
Xiaoyu Wang 已提交
481 482 483 484 485
  }

  return pToken->type;
}

H
refact  
Hongze Cheng 已提交
486 487
static int32_t parseValueToken(char** end, SToken* pToken, SSchema* pSchema, int16_t timePrec, char* tmpTokenBuf,
                               _row_append_fn_t func, void* param, SMsgBuf* pMsgBuf) {
X
Xiaoyu Wang 已提交
488 489 490
  int64_t  iv;
  uint64_t uv;
  char*    endptr = NULL;
X
Xiaoyu Wang 已提交
491

wmmhello's avatar
wmmhello 已提交
492
  int32_t code = checkAndTrimValue(pToken, tmpTokenBuf, pMsgBuf);
X
Xiaoyu Wang 已提交
493 494 495 496
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

497
  if (isNullValue(pSchema->type, pToken)) {
X
Xiaoyu Wang 已提交
498
    if (TSDB_DATA_TYPE_TIMESTAMP == pSchema->type && PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
D
stmt  
dapan1121 已提交
499
      return buildSyntaxErrMsg(pMsgBuf, "primary timestamp should not be null", pToken->z);
X
Xiaoyu Wang 已提交
500 501
    }

X
Xiaoyu Wang 已提交
502
    return func(pMsgBuf, NULL, 0, param);
X
Xiaoyu Wang 已提交
503 504 505 506
  }

  switch (pSchema->type) {
    case TSDB_DATA_TYPE_BOOL: {
507
      if ((pToken->type == TK_NK_BOOL || pToken->type == TK_NK_STRING) && (pToken->n != 0)) {
X
Xiaoyu Wang 已提交
508
        if (strncmp(pToken->z, "true", pToken->n) == 0) {
X
Xiaoyu Wang 已提交
509
          return func(pMsgBuf, &TRUE_VALUE, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
510
        } else if (strncmp(pToken->z, "false", pToken->n) == 0) {
X
Xiaoyu Wang 已提交
511
          return func(pMsgBuf, &FALSE_VALUE, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
512 513 514
        } else {
          return buildSyntaxErrMsg(pMsgBuf, "invalid bool data", pToken->z);
        }
515
      } else if (pToken->type == TK_NK_INTEGER) {
516 517
        return func(pMsgBuf, ((taosStr2Int64(pToken->z, NULL, 10) == 0) ? &FALSE_VALUE : &TRUE_VALUE), pSchema->bytes,
                    param);
518
      } else if (pToken->type == TK_NK_FLOAT) {
519 520
        return func(pMsgBuf, ((taosStr2Double(pToken->z, NULL) == 0) ? &FALSE_VALUE : &TRUE_VALUE), pSchema->bytes,
                    param);
X
Xiaoyu Wang 已提交
521 522 523 524 525 526
      } else {
        return buildSyntaxErrMsg(pMsgBuf, "invalid bool data", pToken->z);
      }
    }

    case TSDB_DATA_TYPE_TINYINT: {
X
Xiaoyu Wang 已提交
527
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
X
Xiaoyu Wang 已提交
528 529 530 531 532 533
        return buildSyntaxErrMsg(pMsgBuf, "invalid tinyint data", pToken->z);
      } else if (!IS_VALID_TINYINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "tinyint data overflow", pToken->z);
      }

      uint8_t tmpVal = (uint8_t)iv;
X
Xiaoyu Wang 已提交
534
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
535 536
    }

H
refact  
Hongze Cheng 已提交
537
    case TSDB_DATA_TYPE_UTINYINT: {
X
Xiaoyu Wang 已提交
538
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
X
Xiaoyu Wang 已提交
539
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned tinyint data", pToken->z);
X
Xiaoyu Wang 已提交
540
      } else if (!IS_VALID_UTINYINT(uv)) {
X
Xiaoyu Wang 已提交
541 542
        return buildSyntaxErrMsg(pMsgBuf, "unsigned tinyint data overflow", pToken->z);
      }
X
Xiaoyu Wang 已提交
543
      uint8_t tmpVal = (uint8_t)uv;
X
Xiaoyu Wang 已提交
544
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
545 546 547
    }

    case TSDB_DATA_TYPE_SMALLINT: {
X
Xiaoyu Wang 已提交
548
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
X
Xiaoyu Wang 已提交
549 550 551 552 553
        return buildSyntaxErrMsg(pMsgBuf, "invalid smallint data", pToken->z);
      } else if (!IS_VALID_SMALLINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "smallint data overflow", pToken->z);
      }
      int16_t tmpVal = (int16_t)iv;
X
Xiaoyu Wang 已提交
554
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
555 556 557
    }

    case TSDB_DATA_TYPE_USMALLINT: {
X
Xiaoyu Wang 已提交
558
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
X
Xiaoyu Wang 已提交
559
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned smallint data", pToken->z);
X
Xiaoyu Wang 已提交
560
      } else if (!IS_VALID_USMALLINT(uv)) {
X
Xiaoyu Wang 已提交
561 562
        return buildSyntaxErrMsg(pMsgBuf, "unsigned smallint data overflow", pToken->z);
      }
X
Xiaoyu Wang 已提交
563
      uint16_t tmpVal = (uint16_t)uv;
X
Xiaoyu Wang 已提交
564
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
565 566 567
    }

    case TSDB_DATA_TYPE_INT: {
X
Xiaoyu Wang 已提交
568
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
X
Xiaoyu Wang 已提交
569 570 571 572 573
        return buildSyntaxErrMsg(pMsgBuf, "invalid int data", pToken->z);
      } else if (!IS_VALID_INT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "int data overflow", pToken->z);
      }
      int32_t tmpVal = (int32_t)iv;
X
Xiaoyu Wang 已提交
574
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
575 576 577
    }

    case TSDB_DATA_TYPE_UINT: {
X
Xiaoyu Wang 已提交
578
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
X
Xiaoyu Wang 已提交
579
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned int data", pToken->z);
X
Xiaoyu Wang 已提交
580
      } else if (!IS_VALID_UINT(uv)) {
X
Xiaoyu Wang 已提交
581 582
        return buildSyntaxErrMsg(pMsgBuf, "unsigned int data overflow", pToken->z);
      }
X
Xiaoyu Wang 已提交
583
      uint32_t tmpVal = (uint32_t)uv;
X
Xiaoyu Wang 已提交
584
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
585 586 587
    }

    case TSDB_DATA_TYPE_BIGINT: {
X
Xiaoyu Wang 已提交
588
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
X
Xiaoyu Wang 已提交
589 590 591 592
        return buildSyntaxErrMsg(pMsgBuf, "invalid bigint data", pToken->z);
      } else if (!IS_VALID_BIGINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "bigint data overflow", pToken->z);
      }
X
Xiaoyu Wang 已提交
593
      return func(pMsgBuf, &iv, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
594 595 596
    }

    case TSDB_DATA_TYPE_UBIGINT: {
X
Xiaoyu Wang 已提交
597
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
X
Xiaoyu Wang 已提交
598
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned bigint data", pToken->z);
X
Xiaoyu Wang 已提交
599
      } else if (!IS_VALID_UBIGINT(uv)) {
X
Xiaoyu Wang 已提交
600 601
        return buildSyntaxErrMsg(pMsgBuf, "unsigned bigint data overflow", pToken->z);
      }
X
Xiaoyu Wang 已提交
602
      return func(pMsgBuf, &uv, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
603 604 605 606
    }

    case TSDB_DATA_TYPE_FLOAT: {
      double dv;
607
      if (TK_NK_ILLEGAL == toDouble(pToken, &dv, &endptr)) {
X
Xiaoyu Wang 已提交
608 609
        return buildSyntaxErrMsg(pMsgBuf, "illegal float data", pToken->z);
      }
H
refact  
Hongze Cheng 已提交
610 611
      if (((dv == HUGE_VAL || dv == -HUGE_VAL) && errno == ERANGE) || dv > FLT_MAX || dv < -FLT_MAX || isinf(dv) ||
          isnan(dv)) {
X
Xiaoyu Wang 已提交
612 613 614
        return buildSyntaxErrMsg(pMsgBuf, "illegal float data", pToken->z);
      }
      float tmpVal = (float)dv;
X
Xiaoyu Wang 已提交
615
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
616 617 618 619
    }

    case TSDB_DATA_TYPE_DOUBLE: {
      double dv;
620
      if (TK_NK_ILLEGAL == toDouble(pToken, &dv, &endptr)) {
X
Xiaoyu Wang 已提交
621 622 623 624 625
        return buildSyntaxErrMsg(pMsgBuf, "illegal double data", pToken->z);
      }
      if (((dv == HUGE_VAL || dv == -HUGE_VAL) && errno == ERANGE) || isinf(dv) || isnan(dv)) {
        return buildSyntaxErrMsg(pMsgBuf, "illegal double data", pToken->z);
      }
X
Xiaoyu Wang 已提交
626
      return func(pMsgBuf, &dv, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
627 628 629 630 631
    }

    case TSDB_DATA_TYPE_BINARY: {
      // Too long values will raise the invalid sql error message
      if (pToken->n + VARSTR_HEADER_SIZE > pSchema->bytes) {
D
dapan1121 已提交
632
        return generateSyntaxErrMsg(pMsgBuf, TSDB_CODE_PAR_VALUE_TOO_LONG, pSchema->name);
X
Xiaoyu Wang 已提交
633 634
      }

X
Xiaoyu Wang 已提交
635
      return func(pMsgBuf, pToken->z, pToken->n, param);
X
Xiaoyu Wang 已提交
636 637 638
    }

    case TSDB_DATA_TYPE_NCHAR: {
X
Xiaoyu Wang 已提交
639
      return func(pMsgBuf, pToken->z, pToken->n, param);
X
Xiaoyu Wang 已提交
640
    }
641
    case TSDB_DATA_TYPE_JSON: {
X
Xiaoyu Wang 已提交
642
      if (pToken->n > (TSDB_MAX_JSON_TAG_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
643 644 645 646
        return buildSyntaxErrMsg(pMsgBuf, "json string too long than 4095", pToken->z);
      }
      return func(pMsgBuf, pToken->z, pToken->n, param);
    }
X
Xiaoyu Wang 已提交
647 648 649 650 651 652
    case TSDB_DATA_TYPE_TIMESTAMP: {
      int64_t tmpVal;
      if (parseTime(end, pToken, timePrec, &tmpVal, pMsgBuf) != TSDB_CODE_SUCCESS) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid timestamp", pToken->z);
      }

X
Xiaoyu Wang 已提交
653
      return func(pMsgBuf, &tmpVal, pSchema->bytes, param);
X
Xiaoyu Wang 已提交
654 655 656 657 658 659
    }
  }

  return TSDB_CODE_FAILED;
}

X
Xiaoyu Wang 已提交
660
static FORCE_INLINE int32_t MemRowAppend(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param) {
C
Cary Xu 已提交
661 662
  SMemParam*   pa = (SMemParam*)param;
  SRowBuilder* rb = pa->rb;
663 664 665 666 667 668

  if (value == NULL) {  // it is a null data
    tdAppendColValToRow(rb, pa->schema->colId, pa->schema->type, TD_VTYPE_NULL, value, false, pa->toffset, pa->colIdx);
    return TSDB_CODE_SUCCESS;
  }

669
  if (TSDB_DATA_TYPE_BINARY == pa->schema->type) {
C
Cary Xu 已提交
670
    const char* rowEnd = tdRowEnd(rb->pBuf);
671
    STR_WITH_SIZE_TO_VARSTR(rowEnd, value, len);
672
    tdAppendColValToRow(rb, pa->schema->colId, pa->schema->type, TD_VTYPE_NORM, rowEnd, false, pa->toffset, pa->colIdx);
673 674
  } else if (TSDB_DATA_TYPE_NCHAR == pa->schema->type) {
    // if the converted output len is over than pColumnModel->bytes, return error: 'Argument list too long'
C
Cary Xu 已提交
675 676
    int32_t     output = 0;
    const char* rowEnd = tdRowEnd(rb->pBuf);
wafwerar's avatar
wafwerar 已提交
677
    if (!taosMbsToUcs4(value, len, (TdUcs4*)varDataVal(rowEnd), pa->schema->bytes - VARSTR_HEADER_SIZE, &output)) {
D
dapan1121 已提交
678 679 680
      if (errno == E2BIG) {
        return generateSyntaxErrMsg(pMsgBuf, TSDB_CODE_PAR_VALUE_TOO_LONG, pa->schema->name);
      }
X
Xiaoyu Wang 已提交
681 682 683
      char buf[512] = {0};
      snprintf(buf, tListLen(buf), "%s", strerror(errno));
      return buildSyntaxErrMsg(pMsgBuf, buf, value);
684
    }
685
    varDataSetLen(rowEnd, output);
C
Cary Xu 已提交
686
    tdAppendColValToRow(rb, pa->schema->colId, pa->schema->type, TD_VTYPE_NORM, rowEnd, false, pa->toffset, pa->colIdx);
687
  } else {
688
    tdAppendColValToRow(rb, pa->schema->colId, pa->schema->type, TD_VTYPE_NORM, value, false, pa->toffset, pa->colIdx);
689
  }
690

691 692
  return TSDB_CODE_SUCCESS;
}
693 694 695

// pSql -> tag1_name, ...)
static int32_t parseBoundColumns(SInsertParseContext* pCxt, SParsedDataColInfo* pColList, SSchema* pSchema) {
C
Cary Xu 已提交
696
  col_id_t nCols = pColList->numOfCols;
697

H
refact  
Hongze Cheng 已提交
698
  pColList->numOfBound = 0;
C
Cary Xu 已提交
699
  pColList->boundNullLen = 0;
C
Cary Xu 已提交
700
  memset(pColList->boundColumns, 0, sizeof(col_id_t) * nCols);
C
Cary Xu 已提交
701
  for (col_id_t i = 0; i < nCols; ++i) {
702 703 704
    pColList->cols[i].valStat = VAL_STAT_NONE;
  }

H
refact  
Hongze Cheng 已提交
705 706
  SToken   sToken;
  bool     isOrdered = true;
C
Cary Xu 已提交
707
  col_id_t lastColIdx = -1;  // last column found
708 709 710
  while (1) {
    NEXT_TOKEN(pCxt->pSql, sToken);

711
    if (TK_NK_RP == sToken.type) {
712 713 714
      break;
    }

715 716 717 718 719
    char tmpTokenBuf[TSDB_COL_NAME_LEN + 2] = {0};  // used for deleting Escape character backstick(`)
    strncpy(tmpTokenBuf, sToken.z, sToken.n);
    sToken.z = tmpTokenBuf;
    sToken.n = strdequote(sToken.z);

C
Cary Xu 已提交
720 721
    col_id_t t = lastColIdx + 1;
    col_id_t index = findCol(&sToken, t, nCols, pSchema);
722 723 724 725 726
    if (index < 0 && t > 0) {
      index = findCol(&sToken, 0, t, pSchema);
      isOrdered = false;
    }
    if (index < 0) {
727
      return generateSyntaxErrMsg(&pCxt->msg, TSDB_CODE_PAR_INVALID_COLUMN, sToken.z);
728 729 730 731 732 733
    }
    if (pColList->cols[index].valStat == VAL_STAT_HAS) {
      return buildSyntaxErrMsg(&pCxt->msg, "duplicated column name", sToken.z);
    }
    lastColIdx = index;
    pColList->cols[index].valStat = VAL_STAT_HAS;
X
Xiaoyu Wang 已提交
734
    pColList->boundColumns[pColList->numOfBound] = index;
735
    ++pColList->numOfBound;
C
Cary Xu 已提交
736 737 738 739 740 741 742 743 744 745 746
    switch (pSchema[t].type) {
      case TSDB_DATA_TYPE_BINARY:
        pColList->boundNullLen += (sizeof(VarDataOffsetT) + VARSTR_HEADER_SIZE + CHAR_BYTES);
        break;
      case TSDB_DATA_TYPE_NCHAR:
        pColList->boundNullLen += (sizeof(VarDataOffsetT) + VARSTR_HEADER_SIZE + TSDB_NCHAR_SIZE);
        break;
      default:
        pColList->boundNullLen += TYPE_BYTES[pSchema[t].type];
        break;
    }
747 748 749 750 751
  }

  pColList->orderStatus = isOrdered ? ORDER_STATUS_ORDERED : ORDER_STATUS_DISORDERED;

  if (!isOrdered) {
wafwerar's avatar
wafwerar 已提交
752
    pColList->colIdxInfo = taosMemoryCalloc(pColList->numOfBound, sizeof(SBoundIdxInfo));
753 754 755 756
    if (NULL == pColList->colIdxInfo) {
      return TSDB_CODE_TSC_OUT_OF_MEMORY;
    }
    SBoundIdxInfo* pColIdx = pColList->colIdxInfo;
C
Cary Xu 已提交
757
    for (col_id_t i = 0; i < pColList->numOfBound; ++i) {
758
      pColIdx[i].schemaColIdx = pColList->boundColumns[i];
759 760
      pColIdx[i].boundIdx = i;
    }
wafwerar's avatar
wafwerar 已提交
761
    taosSort(pColIdx, pColList->numOfBound, sizeof(SBoundIdxInfo), schemaIdxCompar);
C
Cary Xu 已提交
762
    for (col_id_t i = 0; i < pColList->numOfBound; ++i) {
763 764
      pColIdx[i].finalIdx = i;
    }
wafwerar's avatar
wafwerar 已提交
765
    taosSort(pColIdx, pColList->numOfBound, sizeof(SBoundIdxInfo), boundIdxCompar);
766 767
  }

X
Xiaoyu Wang 已提交
768
  if (pColList->numOfCols > pColList->numOfBound) {
769 770 771
    memset(&pColList->boundColumns[pColList->numOfBound], 0,
           sizeof(col_id_t) * (pColList->numOfCols - pColList->numOfBound));
  }
772 773 774 775

  return TSDB_CODE_SUCCESS;
}

776 777
static void buildCreateTbReq(SVCreateTbReq* pTbReq, const char* tname, STag* pTag, int64_t suid, const char* sname,
                             SArray* tagName, uint8_t tagNum) {
wmmhello's avatar
wmmhello 已提交
778 779 780
  pTbReq->type = TD_CHILD_TABLE;
  pTbReq->name = strdup(tname);
  pTbReq->ctb.suid = suid;
781
  pTbReq->ctb.tagNum = tagNum;
782
  if (sname) pTbReq->ctb.name = strdup(sname);
wmmhello's avatar
wmmhello 已提交
783
  pTbReq->ctb.pTag = (uint8_t*)pTag;
wmmhello's avatar
wmmhello 已提交
784
  pTbReq->ctb.tagName = taosArrayDup(tagName);
wmmhello's avatar
wmmhello 已提交
785
  pTbReq->commentLen = -1;
786

wmmhello's avatar
wmmhello 已提交
787 788
  return;
}
789

790 791
static int32_t parseTagToken(char** end, SToken* pToken, SSchema* pSchema, int16_t timePrec, STagVal* val,
                             SMsgBuf* pMsgBuf) {
wmmhello's avatar
wmmhello 已提交
792 793 794 795
  int64_t  iv;
  uint64_t uv;
  char*    endptr = NULL;

796
  if (isNullValue(pSchema->type, pToken)) {
wmmhello's avatar
wmmhello 已提交
797 798 799
    if (TSDB_DATA_TYPE_TIMESTAMP == pSchema->type && PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
      return buildSyntaxErrMsg(pMsgBuf, "primary timestamp should not be null", pToken->z);
    }
800 801 802 803

    return TSDB_CODE_SUCCESS;
  }

804
  //  strcpy(val->colName, pSchema->name);
wmmhello's avatar
wmmhello 已提交
805
  val->cid = pSchema->colId;
wmmhello's avatar
wmmhello 已提交
806
  val->type = pSchema->type;
C
Cary Xu 已提交
807

wmmhello's avatar
wmmhello 已提交
808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823
  switch (pSchema->type) {
    case TSDB_DATA_TYPE_BOOL: {
      if ((pToken->type == TK_NK_BOOL || pToken->type == TK_NK_STRING) && (pToken->n != 0)) {
        if (strncmp(pToken->z, "true", pToken->n) == 0) {
          *(int8_t*)(&val->i64) = TRUE_VALUE;
        } else if (strncmp(pToken->z, "false", pToken->n) == 0) {
          *(int8_t*)(&val->i64) = FALSE_VALUE;
        } else {
          return buildSyntaxErrMsg(pMsgBuf, "invalid bool data", pToken->z);
        }
      } else if (pToken->type == TK_NK_INTEGER) {
        *(int8_t*)(&val->i64) = ((taosStr2Int64(pToken->z, NULL, 10) == 0) ? FALSE_VALUE : TRUE_VALUE);
      } else if (pToken->type == TK_NK_FLOAT) {
        *(int8_t*)(&val->i64) = ((taosStr2Double(pToken->z, NULL) == 0) ? FALSE_VALUE : TRUE_VALUE);
      } else {
        return buildSyntaxErrMsg(pMsgBuf, "invalid bool data", pToken->z);
D
dapan1121 已提交
824
      }
wmmhello's avatar
wmmhello 已提交
825 826
      break;
    }
827

wmmhello's avatar
wmmhello 已提交
828 829 830 831 832
    case TSDB_DATA_TYPE_TINYINT: {
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid tinyint data", pToken->z);
      } else if (!IS_VALID_TINYINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "tinyint data overflow", pToken->z);
D
dapan1121 已提交
833
      }
wmmhello's avatar
wmmhello 已提交
834 835 836

      *(int8_t*)(&val->i64) = iv;
      break;
837 838
    }

wmmhello's avatar
wmmhello 已提交
839 840 841 842 843 844 845 846 847
    case TSDB_DATA_TYPE_UTINYINT: {
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned tinyint data", pToken->z);
      } else if (!IS_VALID_UTINYINT(uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "unsigned tinyint data overflow", pToken->z);
      }
      *(uint8_t*)(&val->i64) = uv;
      break;
    }
848

wmmhello's avatar
wmmhello 已提交
849 850 851 852 853 854 855 856 857
    case TSDB_DATA_TYPE_SMALLINT: {
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid smallint data", pToken->z);
      } else if (!IS_VALID_SMALLINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "smallint data overflow", pToken->z);
      }
      *(int16_t*)(&val->i64) = iv;
      break;
    }
858

wmmhello's avatar
wmmhello 已提交
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940
    case TSDB_DATA_TYPE_USMALLINT: {
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned smallint data", pToken->z);
      } else if (!IS_VALID_USMALLINT(uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "unsigned smallint data overflow", pToken->z);
      }
      *(uint16_t*)(&val->i64) = uv;
      break;
    }

    case TSDB_DATA_TYPE_INT: {
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid int data", pToken->z);
      } else if (!IS_VALID_INT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "int data overflow", pToken->z);
      }
      *(int32_t*)(&val->i64) = iv;
      break;
    }

    case TSDB_DATA_TYPE_UINT: {
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned int data", pToken->z);
      } else if (!IS_VALID_UINT(uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "unsigned int data overflow", pToken->z);
      }
      *(uint32_t*)(&val->i64) = uv;
      break;
    }

    case TSDB_DATA_TYPE_BIGINT: {
      if (TSDB_CODE_SUCCESS != toInteger(pToken->z, pToken->n, 10, &iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid bigint data", pToken->z);
      } else if (!IS_VALID_BIGINT(iv)) {
        return buildSyntaxErrMsg(pMsgBuf, "bigint data overflow", pToken->z);
      }

      val->i64 = iv;
      break;
    }

    case TSDB_DATA_TYPE_UBIGINT: {
      if (TSDB_CODE_SUCCESS != toUInteger(pToken->z, pToken->n, 10, &uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid unsigned bigint data", pToken->z);
      } else if (!IS_VALID_UBIGINT(uv)) {
        return buildSyntaxErrMsg(pMsgBuf, "unsigned bigint data overflow", pToken->z);
      }
      *(uint64_t*)(&val->i64) = uv;
      break;
    }

    case TSDB_DATA_TYPE_FLOAT: {
      double dv;
      if (TK_NK_ILLEGAL == toDouble(pToken, &dv, &endptr)) {
        return buildSyntaxErrMsg(pMsgBuf, "illegal float data", pToken->z);
      }
      if (((dv == HUGE_VAL || dv == -HUGE_VAL) && errno == ERANGE) || dv > FLT_MAX || dv < -FLT_MAX || isinf(dv) ||
          isnan(dv)) {
        return buildSyntaxErrMsg(pMsgBuf, "illegal float data", pToken->z);
      }
      *(float*)(&val->i64) = dv;
      break;
    }

    case TSDB_DATA_TYPE_DOUBLE: {
      double dv;
      if (TK_NK_ILLEGAL == toDouble(pToken, &dv, &endptr)) {
        return buildSyntaxErrMsg(pMsgBuf, "illegal double data", pToken->z);
      }
      if (((dv == HUGE_VAL || dv == -HUGE_VAL) && errno == ERANGE) || isinf(dv) || isnan(dv)) {
        return buildSyntaxErrMsg(pMsgBuf, "illegal double data", pToken->z);
      }

      *(double*)(&val->i64) = dv;
      break;
    }

    case TSDB_DATA_TYPE_BINARY: {
      // Too long values will raise the invalid sql error message
      if (pToken->n + VARSTR_HEADER_SIZE > pSchema->bytes) {
        return generateSyntaxErrMsg(pMsgBuf, TSDB_CODE_PAR_VALUE_TOO_LONG, pSchema->name);
      }
941
      val->pData = strdup(pToken->z);
wmmhello's avatar
wmmhello 已提交
942 943 944 945 946 947
      val->nData = pToken->n;
      break;
    }

    case TSDB_DATA_TYPE_NCHAR: {
      int32_t output = 0;
948
      void*   p = taosMemoryCalloc(1, pSchema->bytes - VARSTR_HEADER_SIZE);
949
      if (p == NULL) {
wmmhello's avatar
wmmhello 已提交
950 951
        return TSDB_CODE_OUT_OF_MEMORY;
      }
952
      if (!taosMbsToUcs4(pToken->z, pToken->n, (TdUcs4*)(p), pSchema->bytes - VARSTR_HEADER_SIZE, &output)) {
wmmhello's avatar
wmmhello 已提交
953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974
        if (errno == E2BIG) {
          taosMemoryFree(p);
          return generateSyntaxErrMsg(pMsgBuf, TSDB_CODE_PAR_VALUE_TOO_LONG, pSchema->name);
        }
        char buf[512] = {0};
        snprintf(buf, tListLen(buf), " taosMbsToUcs4 error:%s", strerror(errno));
        taosMemoryFree(p);
        return buildSyntaxErrMsg(pMsgBuf, buf, pToken->z);
      }
      val->pData = p;
      val->nData = output;
      break;
    }
    case TSDB_DATA_TYPE_TIMESTAMP: {
      if (parseTime(end, pToken, timePrec, &iv, pMsgBuf) != TSDB_CODE_SUCCESS) {
        return buildSyntaxErrMsg(pMsgBuf, "invalid timestamp", pToken->z);
      }

      val->i64 = iv;
      break;
    }
  }
X
Xiaoyu Wang 已提交
975 976 977 978

  return TSDB_CODE_SUCCESS;
}

979
// pSql -> tag1_value, ...)
wmmhello's avatar
wmmhello 已提交
980
static int32_t parseTagsClause(SInsertParseContext* pCxt, SSchema* pSchema, uint8_t precision, const char* tName) {
wmmhello's avatar
wmmhello 已提交
981
  int32_t code = TSDB_CODE_SUCCESS;
982
  SArray* pTagVals = taosArrayInit(pCxt->tags.numOfBound, sizeof(STagVal));
wmmhello's avatar
wmmhello 已提交
983
  SArray* tagName = taosArrayInit(8, TSDB_COL_NAME_LEN);
984 985 986 987
  SToken  sToken;
  bool    isParseBindParam = false;
  bool    isJson = false;
  STag*   pTag = NULL;
988
  for (int i = 0; i < pCxt->tags.numOfBound; ++i) {
X
Xiaoyu Wang 已提交
989
    NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
D
stmt  
dapan1121 已提交
990 991 992 993

    if (sToken.type == TK_NK_QUESTION) {
      isParseBindParam = true;
      if (NULL == pCxt->pStmtCb) {
wmmhello's avatar
wmmhello 已提交
994 995
        code = buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
        goto end;
D
stmt  
dapan1121 已提交
996 997 998 999 1000 1001
      }

      continue;
    }

    if (isParseBindParam) {
wmmhello's avatar
wmmhello 已提交
1002 1003
      code = buildInvalidOperationMsg(&pCxt->msg, "no mix usage for ? and tag values");
      goto end;
D
stmt  
dapan1121 已提交
1004
    }
X
Xiaoyu Wang 已提交
1005

X
Xiaoyu Wang 已提交
1006
    SSchema* pTagSchema = &pSchema[pCxt->tags.boundColumns[i]];
1007
    char     tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0};  // todo this can be optimize with parse column
wmmhello's avatar
wmmhello 已提交
1008 1009 1010 1011
    code = checkAndTrimValue(&sToken, tmpTokenBuf, &pCxt->msg);
    if (code != TSDB_CODE_SUCCESS) {
      goto end;
    }
wmmhello's avatar
wmmhello 已提交
1012

1013
    if (!isNullValue(pTagSchema->type, &sToken)) {
wmmhello's avatar
wmmhello 已提交
1014 1015
      taosArrayPush(tagName, pTagSchema->name);
    }
1016
    if (pTagSchema->type == TSDB_DATA_TYPE_JSON) {
wmmhello's avatar
wmmhello 已提交
1017 1018 1019 1020
      if (sToken.n > (TSDB_MAX_JSON_TAG_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
        code = buildSyntaxErrMsg(&pCxt->msg, "json string too long than 4095", sToken.z);
        goto end;
      }
1021
      if (isNullValue(pTagSchema->type, &sToken)) {
1022 1023
        code = tTagNew(pTagVals, 1, true, &pTag);
      } else {
wmmhello's avatar
wmmhello 已提交
1024
        code = parseJsontoTagData(sToken.z, pTagVals, &pTag, &pCxt->msg);
1025
      }
1026
      if (code != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1027 1028
        goto end;
      }
wmmhello's avatar
wmmhello 已提交
1029
      isJson = true;
1030
    } else {
wmmhello's avatar
wmmhello 已提交
1031
      STagVal val = {0};
wmmhello's avatar
wmmhello 已提交
1032
      code = parseTagToken(&pCxt->pSql, &sToken, pTagSchema, precision, &val, &pCxt->msg);
wmmhello's avatar
wmmhello 已提交
1033 1034 1035
      if (TSDB_CODE_SUCCESS != code) {
        goto end;
      }
1036

wmmhello's avatar
wmmhello 已提交
1037 1038
      taosArrayPush(pTagVals, &val);
    }
1039 1040
  }

D
stmt  
dapan1121 已提交
1041
  if (isParseBindParam) {
wmmhello's avatar
wmmhello 已提交
1042 1043
    code = TSDB_CODE_SUCCESS;
    goto end;
D
stmt  
dapan1121 已提交
1044 1045
  }

1046
  if (!isJson && (code = tTagNew(pTagVals, 1, false, &pTag)) != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1047
    goto end;
1048 1049
  }

1050 1051
  buildCreateTbReq(&pCxt->createTblReq, tName, pTag, pCxt->pTableMeta->suid, pCxt->sTableName, tagName,
                   pCxt->pTableMeta->tableInfo.numOfTags);
wmmhello's avatar
wmmhello 已提交
1052 1053 1054

end:
  for (int i = 0; i < taosArrayGetSize(pTagVals); ++i) {
1055
    STagVal* p = (STagVal*)taosArrayGet(pTagVals, i);
1056
    if (IS_VAR_DATA_TYPE(p->type)) {
wmmhello's avatar
wmmhello 已提交
1057 1058 1059 1060
      taosMemoryFree(p->pData);
    }
  }
  taosArrayDestroy(pTagVals);
wmmhello's avatar
wmmhello 已提交
1061
  taosArrayDestroy(tagName);
wmmhello's avatar
wmmhello 已提交
1062
  return code;
X
Xiaoyu Wang 已提交
1063
}
1064

X
Xiaoyu Wang 已提交
1065 1066
static int32_t storeTableMeta(SInsertParseContext* pCxt, SHashObj* pHash, int32_t tbNo, SName* pTableName,
                              const char* pName, int32_t len, STableMeta* pMeta) {
1067
  SVgroupInfo vg;
X
Xiaoyu Wang 已提交
1068
  CHECK_CODE(getTableVgroup(pCxt, tbNo, pTableName, &vg));
X
Xiaoyu Wang 已提交
1069 1070
  CHECK_CODE(taosHashPut(pCxt->pVgroupsHashObj, (const char*)&vg.vgId, sizeof(vg.vgId), (char*)&vg, sizeof(vg)));

X
Xiaoyu Wang 已提交
1071
  pMeta->uid = tbNo;
X
Xiaoyu Wang 已提交
1072
  pMeta->vgId = vg.vgId;
D
dapan 已提交
1073
  pMeta->tableType = TSDB_CHILD_TABLE;
X
Xiaoyu Wang 已提交
1074

X
Xiaoyu Wang 已提交
1075 1076 1077 1078 1079
  STableMeta* pBackup = NULL;
  if (TSDB_CODE_SUCCESS != cloneTableMeta(pMeta, &pBackup)) {
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }
  return taosHashPut(pHash, pName, len, &pBackup, POINTER_BYTES);
1080 1081
}

1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
static int32_t skipParentheses(SInsertParseSyntaxCxt* pCxt) {
  SToken  sToken;
  int32_t expectRightParenthesis = 1;
  while (1) {
    NEXT_TOKEN(pCxt->pSql, sToken);
    if (TK_NK_LP == sToken.type) {
      ++expectRightParenthesis;
    } else if (TK_NK_RP == sToken.type && 0 == --expectRightParenthesis) {
      break;
    }
    if (0 == sToken.n) {
      return buildSyntaxErrMsg(&pCxt->msg, ") expected", NULL);
    }
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t skipBoundColumns(SInsertParseSyntaxCxt* pCxt) { return skipParentheses(pCxt); }

static int32_t ignoreBoundColumns(SInsertParseContext* pCxt) {
  SInsertParseSyntaxCxt cxt = {.pComCxt = pCxt->pComCxt, .pSql = pCxt->pSql, .msg = pCxt->msg, .pMetaCache = NULL};
  int32_t               code = skipBoundColumns(&cxt);
  pCxt->pSql = cxt.pSql;
  return code;
}

1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
static int32_t skipUsingClause(SInsertParseSyntaxCxt* pCxt);

// pSql -> stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)
static int32_t ignoreAutoCreateTableClause(SInsertParseContext* pCxt) {
  SToken sToken;
  NEXT_TOKEN(pCxt->pSql, sToken);
  SInsertParseSyntaxCxt cxt = {.pComCxt = pCxt->pComCxt, .pSql = pCxt->pSql, .msg = pCxt->msg, .pMetaCache = NULL};
  int32_t               code = skipUsingClause(&cxt);
  pCxt->pSql = cxt.pSql;
  return code;
}

1120
// pSql -> stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)
X
Xiaoyu Wang 已提交
1121
static int32_t parseUsingClause(SInsertParseContext* pCxt, int32_t tbNo, SName* name, char* tbFName) {
H
refact  
Hongze Cheng 已提交
1122
  int32_t      len = strlen(tbFName);
X
Xiaoyu Wang 已提交
1123 1124
  STableMeta** pMeta = taosHashGet(pCxt->pSubTableHashObj, tbFName, len);
  if (NULL != pMeta) {
1125
    CHECK_CODE(ignoreAutoCreateTableClause(pCxt));
X
Xiaoyu Wang 已提交
1126 1127
    return cloneTableMeta(*pMeta, &pCxt->pTableMeta);
  }
1128

X
Xiaoyu Wang 已提交
1129
  SToken sToken;
1130 1131
  // pSql -> stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)
  NEXT_TOKEN(pCxt->pSql, sToken);
D
dapan 已提交
1132 1133 1134

  SName sname;
  createSName(&sname, &sToken, pCxt->pComCxt->acctId, pCxt->pComCxt->db, &pCxt->msg);
X
Xiaoyu Wang 已提交
1135 1136
  char dbFName[TSDB_DB_FNAME_LEN];
  tNameGetFullDbName(&sname, dbFName);
wmmhello's avatar
wmmhello 已提交
1137
  strcpy(pCxt->sTableName, sname.tname);
X
Xiaoyu Wang 已提交
1138

X
Xiaoyu Wang 已提交
1139
  CHECK_CODE(getSTableMeta(pCxt, tbNo, &sname, dbFName));
1140 1141 1142
  if (TSDB_SUPER_TABLE != pCxt->pTableMeta->tableType) {
    return buildInvalidOperationMsg(&pCxt->msg, "create table only from super table is allowed");
  }
X
Xiaoyu Wang 已提交
1143
  CHECK_CODE(storeTableMeta(pCxt, pCxt->pSubTableHashObj, tbNo, name, tbFName, len, pCxt->pTableMeta));
1144 1145

  SSchema* pTagsSchema = getTableTagSchema(pCxt->pTableMeta);
1146
  setBoundColumnInfo(&pCxt->tags, pTagsSchema, getNumOfTags(pCxt->pTableMeta));
1147 1148 1149

  // pSql -> [(tag1_name, ...)] TAGS (tag1_value, ...)
  NEXT_TOKEN(pCxt->pSql, sToken);
1150
  if (TK_NK_LP == sToken.type) {
1151
    CHECK_CODE(parseBoundColumns(pCxt, &pCxt->tags, pTagsSchema));
1152 1153 1154 1155 1156 1157 1158 1159
    NEXT_TOKEN(pCxt->pSql, sToken);
  }

  if (TK_TAGS != sToken.type) {
    return buildSyntaxErrMsg(&pCxt->msg, "TAGS is expected", sToken.z);
  }
  // pSql -> (tag1_value, ...)
  NEXT_TOKEN(pCxt->pSql, sToken);
1160
  if (TK_NK_LP != sToken.type) {
1161 1162
    return buildSyntaxErrMsg(&pCxt->msg, "( is expected", sToken.z);
  }
X
Xiaoyu Wang 已提交
1163
  CHECK_CODE(parseTagsClause(pCxt, pTagsSchema, getTableInfo(pCxt->pTableMeta).precision, name->tname));
1164 1165 1166 1167
  NEXT_VALID_TOKEN(pCxt->pSql, sToken);
  if (TK_NK_COMMA == sToken.type) {
    return generateSyntaxErrMsg(&pCxt->msg, TSDB_CODE_PAR_TAGS_NOT_MATCHED);
  } else if (TK_NK_RP != sToken.type) {
X
Xiaoyu Wang 已提交
1168 1169
    return buildSyntaxErrMsg(&pCxt->msg, ") is expected", sToken.z);
  }
1170 1171 1172 1173

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1174 1175
static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks, int16_t timePrec, bool* gotRow,
                       char* tmpTokenBuf) {
1176
  SParsedDataColInfo* spd = &pDataBlocks->boundColumnInfo;
C
Cary Xu 已提交
1177 1178 1179 1180
  SRowBuilder*        pBuilder = &pDataBlocks->rowBuilder;
  STSRow*             row = (STSRow*)(pDataBlocks->pData + pDataBlocks->size);  // skip the SSubmitBlk header

  tdSRowResetBuf(pBuilder, row);
1181

H
refact  
Hongze Cheng 已提交
1182 1183
  bool      isParseBindParam = false;
  SSchema*  schema = getTableColumnSchema(pDataBlocks->pTableMeta);
C
Cary Xu 已提交
1184
  SMemParam param = {.rb = pBuilder};
H
refact  
Hongze Cheng 已提交
1185
  SToken    sToken = {0};
1186 1187
  // 1. set the parsed value from sql string
  for (int i = 0; i < spd->numOfBound; ++i) {
X
Xiaoyu Wang 已提交
1188
    NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
X
Xiaoyu Wang 已提交
1189
    SSchema* pSchema = &schema[spd->boundColumns[i]];
D
stmt  
dapan1121 已提交
1190 1191 1192 1193 1194 1195 1196 1197 1198 1199

    if (sToken.type == TK_NK_QUESTION) {
      isParseBindParam = true;
      if (NULL == pCxt->pStmtCb) {
        return buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
      }

      continue;
    }

D
dapan1121 已提交
1200 1201 1202 1203
    if (TK_NK_RP == sToken.type) {
      return generateSyntaxErrMsg(&pCxt->msg, TSDB_CODE_PAR_INVALID_COLUMNS_NUM);
    }

D
stmt  
dapan1121 已提交
1204 1205 1206
    if (isParseBindParam) {
      return buildInvalidOperationMsg(&pCxt->msg, "no mix usage for ? and values");
    }
X
Xiaoyu Wang 已提交
1207

1208
    param.schema = pSchema;
D
stmt  
dapan1121 已提交
1209
    getSTSRowAppendInfo(pBuilder->rowType, spd, i, &param.toffset, &param.colIdx);
1210
    CHECK_CODE(parseValueToken(&pCxt->pSql, &sToken, pSchema, timePrec, tmpTokenBuf, MemRowAppend, &param, &pCxt->msg));
1211

X
Xiaoyu Wang 已提交
1212 1213 1214 1215 1216 1217
    if (i < spd->numOfBound - 1) {
      NEXT_VALID_TOKEN(pCxt->pSql, sToken);
      if (TK_NK_COMMA != sToken.type) {
        return buildSyntaxErrMsg(&pCxt->msg, ", expected", sToken.z);
      }
    }
1218 1219
  }

1220 1221 1222
  TSKEY tsKey = TD_ROW_KEY(row);
  checkTimestamp(pDataBlocks, (const char*)&tsKey);

1223
  if (!isParseBindParam) {
C
Cary Xu 已提交
1224
    // set the null value for the columns that do not assign values
C
Cary Xu 已提交
1225
    if ((spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
1226
      pBuilder->hasNone = true;
1227
    }
D
stmt  
dapan1121 已提交
1228

C
Cary Xu 已提交
1229 1230
    tdSRowEnd(pBuilder);

D
stmt  
dapan1121 已提交
1231
    *gotRow = true;
X
Xiaoyu Wang 已提交
1232

C
Cary Xu 已提交
1233
#ifdef TD_DEBUG_PRINT_ROW
C
Cary Xu 已提交
1234
    STSchema* pSTSchema = tdGetSTSChemaFromSSChema(schema, spd->numOfCols, 1);
C
Cary Xu 已提交
1235
    tdSRowPrint(row, pSTSchema, __func__);
C
Cary Xu 已提交
1236 1237
    taosMemoryFree(pSTSchema);
#endif
1238 1239
  }

C
Cary Xu 已提交
1240
  // *len = pBuilder->extendedRowSize;
1241 1242 1243 1244
  return TSDB_CODE_SUCCESS;
}

// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
1245
static int32_t parseValues(SInsertParseContext* pCxt, STableDataBlocks* pDataBlock, int maxRows, int32_t* numOfRows) {
1246
  STableComInfo tinfo = getTableInfo(pDataBlock->pTableMeta);
H
refact  
Hongze Cheng 已提交
1247
  int32_t       extendedRowSize = getExtendedRowSize(pDataBlock);
C
Cary Xu 已提交
1248
  CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));
1249 1250

  (*numOfRows) = 0;
X
Xiaoyu Wang 已提交
1251
  // char   tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0};  // used for deleting Escape character: \\, \', \"
1252 1253
  SToken sToken;
  while (1) {
1254 1255
    int32_t index = 0;
    NEXT_TOKEN_KEEP_SQL(pCxt->pSql, sToken, index);
1256
    if (TK_NK_LP != sToken.type) {
1257 1258
      break;
    }
1259
    pCxt->pSql += index;
1260 1261 1262 1263 1264 1265 1266 1267

    if ((*numOfRows) >= maxRows || pDataBlock->size + extendedRowSize >= pDataBlock->nAllocSize) {
      int32_t tSize;
      CHECK_CODE(allocateMemIfNeed(pDataBlock, extendedRowSize, &tSize));
      ASSERT(tSize >= maxRows);
      maxRows = tSize;
    }

D
stmt  
dapan1121 已提交
1268
    bool gotRow = false;
X
Xiaoyu Wang 已提交
1269
    CHECK_CODE(parseOneRow(pCxt, pDataBlock, tinfo.precision, &gotRow, pCxt->tmpTokenBuf));
D
stmt  
dapan1121 已提交
1270
    if (gotRow) {
X
Xiaoyu Wang 已提交
1271
      pDataBlock->size += extendedRowSize;  // len;
D
stmt  
dapan1121 已提交
1272
    }
1273

1274 1275 1276 1277
    NEXT_VALID_TOKEN(pCxt->pSql, sToken);
    if (TK_NK_COMMA == sToken.type) {
      return generateSyntaxErrMsg(&pCxt->msg, TSDB_CODE_PAR_INVALID_COLUMNS_NUM);
    } else if (TK_NK_RP != sToken.type) {
1278 1279 1280
      return buildSyntaxErrMsg(&pCxt->msg, ") expected", sToken.z);
    }

D
stmt  
dapan1121 已提交
1281 1282 1283
    if (gotRow) {
      (*numOfRows)++;
    }
1284 1285
  }

D
stmt  
dapan1121 已提交
1286
  if (0 == (*numOfRows) && (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT))) {
X
Xiaoyu Wang 已提交
1287
    return buildSyntaxErrMsg(&pCxt->msg, "no any data points", NULL);
1288 1289 1290 1291
  }
  return TSDB_CODE_SUCCESS;
}

H
refact  
Hongze Cheng 已提交
1292
static int32_t parseValuesClause(SInsertParseContext* pCxt, STableDataBlocks* dataBuf) {
1293 1294 1295 1296
  int32_t maxNumOfRows;
  CHECK_CODE(allocateMemIfNeed(dataBuf, getExtendedRowSize(dataBuf), &maxNumOfRows));

  int32_t numOfRows = 0;
1297
  CHECK_CODE(parseValues(pCxt, dataBuf, maxNumOfRows, &numOfRows));
1298

H
refact  
Hongze Cheng 已提交
1299
  SSubmitBlk* pBlocks = (SSubmitBlk*)(dataBuf->pData);
D
dapan1121 已提交
1300
  if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, dataBuf, numOfRows)) {
X
Xiaoyu Wang 已提交
1301 1302
    return buildInvalidOperationMsg(&pCxt->msg,
                                    "too many rows in sql, total number of rows should be less than INT32_MAX");
1303 1304 1305 1306 1307 1308 1309
  }

  dataBuf->numOfTables = 1;
  pCxt->totalNum += numOfRows;
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355
static int32_t parseCsvFile(SInsertParseContext* pCxt, TdFilePtr fp, STableDataBlocks* pDataBlock, int maxRows,
                            int32_t* numOfRows) {
  STableComInfo tinfo = getTableInfo(pDataBlock->pTableMeta);
  int32_t       extendedRowSize = getExtendedRowSize(pDataBlock);
  CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));

  (*numOfRows) = 0;
  char    tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0};  // used for deleting Escape character: \\, \', \"
  char*   pLine = NULL;
  int64_t readLen = 0;
  while ((readLen = taosGetLineFile(fp, &pLine)) != -1) {
    if (('\r' == pLine[readLen - 1]) || ('\n' == pLine[readLen - 1])) {
      pLine[--readLen] = '\0';
    }

    if (readLen == 0) {
      continue;
    }

    if ((*numOfRows) >= maxRows || pDataBlock->size + extendedRowSize >= pDataBlock->nAllocSize) {
      int32_t tSize;
      CHECK_CODE(allocateMemIfNeed(pDataBlock, extendedRowSize, &tSize));
      ASSERT(tSize >= maxRows);
      maxRows = tSize;
    }

    strtolower(pLine, pLine);
    char* pRawSql = pCxt->pSql;
    pCxt->pSql = pLine;
    bool gotRow = false;
    CHECK_CODE(parseOneRow(pCxt, pDataBlock, tinfo.precision, &gotRow, tmpTokenBuf));
    if (gotRow) {
      pDataBlock->size += extendedRowSize;  // len;
      (*numOfRows)++;
    }
    pCxt->pSql = pRawSql;
  }

  if (0 == (*numOfRows) && (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT))) {
    return buildSyntaxErrMsg(&pCxt->msg, "no any data points", NULL);
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t parseDataFromFile(SInsertParseContext* pCxt, SToken filePath, STableDataBlocks* dataBuf) {
  char filePathStr[TSDB_FILENAME_LEN] = {0};
1356 1357 1358 1359 1360
  if (TK_NK_STRING == filePath.type) {
    trimString(filePath.z, filePath.n, filePathStr, sizeof(filePathStr));
  } else {
    strncpy(filePathStr, filePath.z, filePath.n);
  }
X
Xiaoyu Wang 已提交
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373
  TdFilePtr fp = taosOpenFile(filePathStr, TD_FILE_READ | TD_FILE_STREAM);
  if (NULL == fp) {
    return TAOS_SYSTEM_ERROR(errno);
  }

  int32_t maxNumOfRows;
  CHECK_CODE(allocateMemIfNeed(dataBuf, getExtendedRowSize(dataBuf), &maxNumOfRows));

  int32_t numOfRows = 0;
  CHECK_CODE(parseCsvFile(pCxt, fp, dataBuf, maxNumOfRows, &numOfRows));

  SSubmitBlk* pBlocks = (SSubmitBlk*)(dataBuf->pData);
  if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, dataBuf, numOfRows)) {
X
Xiaoyu Wang 已提交
1374 1375
    return buildInvalidOperationMsg(&pCxt->msg,
                                    "too many rows in sql, total number of rows should be less than INT32_MAX");
X
Xiaoyu Wang 已提交
1376 1377 1378 1379 1380 1381 1382
  }

  dataBuf->numOfTables = 1;
  pCxt->totalNum += numOfRows;
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1383
static void destroyInsertParseContextForTable(SInsertParseContext* pCxt) {
X
Xiaoyu Wang 已提交
1384 1385 1386
  if (!pCxt->pComCxt->async) {
    taosMemoryFreeClear(pCxt->pTableMeta);
  }
X
Xiaoyu Wang 已提交
1387
  destroyBoundColumnInfo(&pCxt->tags);
wmmhello's avatar
wmmhello 已提交
1388
  tdDestroySVCreateTbReq(&pCxt->createTblReq);
X
Xiaoyu Wang 已提交
1389 1390
}

1391 1392
static void destroySubTableHashElem(void* p) { taosMemoryFree(*(STableMeta**)p); }

X
Xiaoyu Wang 已提交
1393 1394 1395
static void destroyInsertParseContext(SInsertParseContext* pCxt) {
  destroyInsertParseContextForTable(pCxt);
  taosHashCleanup(pCxt->pVgroupsHashObj);
1396
  taosHashCleanup(pCxt->pSubTableHashObj);
D
dapan1121 已提交
1397
  taosHashCleanup(pCxt->pTableNameHashObj);
D
dapan1121 已提交
1398
  taosHashCleanup(pCxt->pDbFNameHashObj);
1399 1400

  destroyBlockHashmap(pCxt->pTableBlockHashObj);
X
Xiaoyu Wang 已提交
1401 1402 1403
  destroyBlockArrayList(pCxt->pVgDataBlocks);
}

X
Xiaoyu Wang 已提交
1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417
static int32_t parseTableName(SInsertParseContext* pCxt, SToken* pTbnameToken, SName* pName, char* pDbFName,
                              char* pTbFName) {
  int32_t code = createSName(pName, pTbnameToken, pCxt->pComCxt->acctId, pCxt->pComCxt->db, &pCxt->msg);
  if (TSDB_CODE_SUCCESS == code) {
    tNameExtractFullName(pName, pTbFName);
    code = taosHashPut(pCxt->pTableNameHashObj, pTbFName, strlen(pTbFName), pName, sizeof(SName));
  }
  if (TSDB_CODE_SUCCESS == code) {
    tNameGetFullDbName(pName, pDbFName);
    code = taosHashPut(pCxt->pDbFNameHashObj, pDbFName, strlen(pDbFName), pDbFName, TSDB_DB_FNAME_LEN);
  }
  return code;
}

1418 1419 1420 1421 1422 1423
//   tb_name
//       [USING stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)]
//       [(field1_name, ...)]
//       VALUES (field1_value, ...) [(field1_value2, ...) ...] | FILE csv_file_path
//   [...];
static int32_t parseInsertBody(SInsertParseContext* pCxt) {
1424
  int32_t tbNum = 0;
X
Xiaoyu Wang 已提交
1425
  SName   name;
1426
  char    tbFName[TSDB_TABLE_FNAME_LEN];
X
Xiaoyu Wang 已提交
1427
  char    dbFName[TSDB_DB_FNAME_LEN];
1428
  bool    autoCreateTbl = false;
X
Xiaoyu Wang 已提交
1429

X
Xiaoyu Wang 已提交
1430
  // for each table
1431 1432
  while (1) {
    SToken sToken;
X
Xiaoyu Wang 已提交
1433
    char*  tbName = NULL;
D
stmt  
dapan1121 已提交
1434

1435 1436 1437 1438 1439
    // pSql -> tb_name ...
    NEXT_TOKEN(pCxt->pSql, sToken);

    // no data in the sql string anymore.
    if (sToken.n == 0) {
D
dapan1121 已提交
1440 1441 1442
      if (sToken.type && pCxt->pSql[0]) {
        return buildSyntaxErrMsg(&pCxt->msg, "invalid charactor in SQL", sToken.z);
      }
X
Xiaoyu Wang 已提交
1443

D
stmt  
dapan1121 已提交
1444
      if (0 == pCxt->totalNum && (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT))) {
H
refact  
Hongze Cheng 已提交
1445
        return buildInvalidOperationMsg(&pCxt->msg, "no data in sql");
1446 1447 1448 1449
      }
      break;
    }

D
stmt  
dapan1121 已提交
1450
    if (TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT) && tbNum > 0) {
X
Xiaoyu Wang 已提交
1451
      return buildInvalidOperationMsg(&pCxt->msg, "single table allowed in one stmt");
D
stmt  
dapan1121 已提交
1452 1453
    }

D
stmt  
dapan1121 已提交
1454 1455
    destroyInsertParseContextForTable(pCxt);

D
stmt  
dapan1121 已提交
1456 1457 1458
    if (TK_NK_QUESTION == sToken.type) {
      if (pCxt->pStmtCb) {
        CHECK_CODE((*pCxt->pStmtCb->getTbNameFn)(pCxt->pStmtCb->pStmt, &tbName));
X
Xiaoyu Wang 已提交
1459

D
stmt  
dapan1121 已提交
1460 1461 1462 1463 1464 1465
        sToken.z = tbName;
        sToken.n = strlen(tbName);
      } else {
        return buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
      }
    }
X
Xiaoyu Wang 已提交
1466

1467 1468 1469
    SToken tbnameToken = sToken;
    NEXT_TOKEN(pCxt->pSql, sToken);

X
Xiaoyu Wang 已提交
1470
    if (!pCxt->pComCxt->async || TK_USING == sToken.type) {
X
Xiaoyu Wang 已提交
1471
      CHECK_CODE(parseTableName(pCxt, &tbnameToken, &name, dbFName, tbFName));
X
Xiaoyu Wang 已提交
1472
    }
X
Xiaoyu Wang 已提交
1473

1474
    bool existedUsing = false;
1475
    // USING clause
1476
    if (TK_USING == sToken.type) {
1477
      existedUsing = true;
X
Xiaoyu Wang 已提交
1478
      CHECK_CODE(parseUsingClause(pCxt, tbNum, &name, tbFName));
1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
      NEXT_TOKEN(pCxt->pSql, sToken);
      autoCreateTbl = true;
    }

    char* pBoundColsStart = NULL;
    if (TK_NK_LP == sToken.type) {
      // pSql -> field1_name, ...)
      pBoundColsStart = pCxt->pSql;
      CHECK_CODE(ignoreBoundColumns(pCxt));
      NEXT_TOKEN(pCxt->pSql, sToken);
    }

1491
    if (TK_USING == sToken.type) {
X
Xiaoyu Wang 已提交
1492 1493 1494
      if (pCxt->pComCxt->async) {
        CHECK_CODE(parseTableName(pCxt, &tbnameToken, &name, dbFName, tbFName));
      }
X
Xiaoyu Wang 已提交
1495
      CHECK_CODE(parseUsingClause(pCxt, tbNum, &name, tbFName));
1496
      NEXT_TOKEN(pCxt->pSql, sToken);
D
dapan 已提交
1497
      autoCreateTbl = true;
1498
    } else if (!existedUsing) {
X
Xiaoyu Wang 已提交
1499
      CHECK_CODE(getTableMeta(pCxt, tbNum, &name, dbFName));
1500 1501
    }

H
refact  
Hongze Cheng 已提交
1502
    STableDataBlocks* dataBuf = NULL;
X
Xiaoyu Wang 已提交
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
    if (pCxt->pComCxt->async) {
      CHECK_CODE(getDataBlockFromList(pCxt->pTableBlockHashObj, &pCxt->pTableMeta->uid, sizeof(pCxt->pTableMeta->uid),
                                      TSDB_DEFAULT_PAYLOAD_SIZE, sizeof(SSubmitBlk),
                                      getTableInfo(pCxt->pTableMeta).rowSize, pCxt->pTableMeta, &dataBuf, NULL,
                                      &pCxt->createTblReq));
    } else {
      CHECK_CODE(getDataBlockFromList(pCxt->pTableBlockHashObj, tbFName, strlen(tbFName), TSDB_DEFAULT_PAYLOAD_SIZE,
                                      sizeof(SSubmitBlk), getTableInfo(pCxt->pTableMeta).rowSize, pCxt->pTableMeta,
                                      &dataBuf, NULL, &pCxt->createTblReq));
    }
1513

1514 1515 1516
    if (NULL != pBoundColsStart) {
      char* pCurrPos = pCxt->pSql;
      pCxt->pSql = pBoundColsStart;
D
dapan1121 已提交
1517
      CHECK_CODE(parseBoundColumns(pCxt, &dataBuf->boundColumnInfo, getTableColumnSchema(pCxt->pTableMeta)));
1518
      pCxt->pSql = pCurrPos;
1519 1520 1521 1522 1523
    }

    if (TK_VALUES == sToken.type) {
      // pSql -> (field1_value, ...) [(field1_value2, ...) ...]
      CHECK_CODE(parseValuesClause(pCxt, dataBuf));
D
stmt  
dapan1121 已提交
1524
      TSDB_QUERY_SET_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_INSERT);
D
stmt  
dapan1121 已提交
1525 1526

      tbNum++;
1527 1528 1529 1530
      continue;
    }

    // FILE csv_file_path
X
Xiaoyu Wang 已提交
1531
    if (TK_FILE == sToken.type) {
1532 1533
      // pSql -> csv_file_path
      NEXT_TOKEN(pCxt->pSql, sToken);
1534
      if (0 == sToken.n || (TK_NK_STRING != sToken.type && TK_NK_ID != sToken.type)) {
1535 1536
        return buildSyntaxErrMsg(&pCxt->msg, "file path is required following keyword FILE", sToken.z);
      }
X
Xiaoyu Wang 已提交
1537
      CHECK_CODE(parseDataFromFile(pCxt, sToken, dataBuf));
1538
      pCxt->pOutput->insertType = TSDB_QUERY_TYPE_FILE_INSERT;
D
stmt  
dapan1121 已提交
1539 1540

      tbNum++;
1541 1542 1543 1544 1545
      continue;
    }

    return buildSyntaxErrMsg(&pCxt->msg, "keyword VALUES or FILE is expected", sToken.z);
  }
X
Xiaoyu Wang 已提交
1546

D
dapan1121 已提交
1547 1548
  qDebug("0x%" PRIx64 " insert input rows: %d", pCxt->pComCxt->requestId, pCxt->totalNum);

D
stmt  
dapan1121 已提交
1549
  if (TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT)) {
X
Xiaoyu Wang 已提交
1550
    SParsedDataColInfo* tags = taosMemoryMalloc(sizeof(pCxt->tags));
D
stmt  
dapan1121 已提交
1551 1552 1553 1554
    if (NULL == tags) {
      return TSDB_CODE_TSC_OUT_OF_MEMORY;
    }
    memcpy(tags, &pCxt->tags, sizeof(pCxt->tags));
1555
    (*pCxt->pStmtCb->setInfoFn)(pCxt->pStmtCb->pStmt, pCxt->pTableMeta, tags, tbFName, autoCreateTbl,
wmmhello's avatar
wmmhello 已提交
1556
                                pCxt->pVgroupsHashObj, pCxt->pTableBlockHashObj, pCxt->sTableName);
D
stmt  
dapan1121 已提交
1557

D
dapan 已提交
1558
    memset(&pCxt->tags, 0, sizeof(pCxt->tags));
D
stmt  
dapan1121 已提交
1559 1560
    pCxt->pVgroupsHashObj = NULL;
    pCxt->pTableBlockHashObj = NULL;
X
Xiaoyu Wang 已提交
1561

D
stmt  
dapan1121 已提交
1562 1563
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
1564

1565
  // merge according to vgId
D
stmt  
dapan1121 已提交
1566
  if (taosHashGetSize(pCxt->pTableBlockHashObj) > 0) {
X
Xiaoyu Wang 已提交
1567
    CHECK_CODE(mergeTableDataBlocks(pCxt->pTableBlockHashObj, pCxt->pOutput->payloadType, &pCxt->pVgDataBlocks));
1568
  }
1569
  return buildOutput(pCxt);
1570 1571 1572 1573 1574 1575 1576 1577
}

// INSERT INTO
//   tb_name
//       [USING stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)]
//       [(field1_name, ...)]
//       VALUES (field1_value, ...) [(field1_value2, ...) ...] | FILE csv_file_path
//   [...];
1578
int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery, SParseMetaCache* pMetaCache) {
1579
  SInsertParseContext context = {
X
Xiaoyu Wang 已提交
1580 1581 1582 1583
      .pComCxt = pContext,
      .pSql = (char*)pContext->pSql,
      .msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
      .pTableMeta = NULL,
wmmhello's avatar
wmmhello 已提交
1584
      .createTblReq = {0},
X
Xiaoyu Wang 已提交
1585 1586
      .pSubTableHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, HASH_NO_LOCK),
      .pTableNameHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, HASH_NO_LOCK),
D
dapan1121 已提交
1587
      .pDbFNameHashObj = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, HASH_NO_LOCK),
X
Xiaoyu Wang 已提交
1588 1589
      .totalNum = 0,
      .pOutput = (SVnodeModifOpStmt*)nodesMakeNode(QUERY_NODE_VNODE_MODIF_STMT),
1590
      .pStmtCb = pContext->pStmtCb,
X
Xiaoyu Wang 已提交
1591 1592 1593
      .pMetaCache = pMetaCache,
      .memElapsed = 0,
      .parRowElapsed = 0};
1594

D
stmt  
dapan1121 已提交
1595
  if (pContext->pStmtCb && *pQuery) {
X
Xiaoyu Wang 已提交
1596 1597
    (*pContext->pStmtCb->getExecInfoFn)(pContext->pStmtCb->pStmt, &context.pVgroupsHashObj,
                                        &context.pTableBlockHashObj);
D
dapan1121 已提交
1598 1599 1600 1601 1602 1603 1604
    if (NULL == context.pVgroupsHashObj) {
      context.pVgroupsHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_NO_LOCK);
    }
    if (NULL == context.pTableBlockHashObj) {
      context.pTableBlockHashObj =
          taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
    }
D
stmt  
dapan1121 已提交
1605
  } else {
X
Xiaoyu Wang 已提交
1606 1607
    context.pVgroupsHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_NO_LOCK);
    context.pTableBlockHashObj =
X
Xiaoyu Wang 已提交
1608
        taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
D
stmt  
dapan1121 已提交
1609
  }
X
Xiaoyu Wang 已提交
1610 1611

  if (NULL == context.pVgroupsHashObj || NULL == context.pTableBlockHashObj || NULL == context.pSubTableHashObj ||
D
dapan1121 已提交
1612
      NULL == context.pTableNameHashObj || NULL == context.pDbFNameHashObj || NULL == context.pOutput) {
X
Xiaoyu Wang 已提交
1613
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
1614
  }
1615
  taosHashSetFreeFp(context.pSubTableHashObj, destroySubTableHashElem);
1616

D
stmt  
dapan1121 已提交
1617 1618 1619 1620
  if (pContext->pStmtCb) {
    TSDB_QUERY_SET_TYPE(context.pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT);
  }

1621
  if (NULL == *pQuery) {
1622
    *pQuery = (SQuery*)nodesMakeNode(QUERY_NODE_QUERY);
D
stmt  
dapan1121 已提交
1623 1624 1625
    if (NULL == *pQuery) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
D
dapan1121 已提交
1626 1627
  } else {
    nodesDestroyNode((*pQuery)->pRoot);
1628
  }
1629

1630 1631 1632 1633
  (*pQuery)->execMode = QUERY_EXEC_MODE_SCHEDULE;
  (*pQuery)->haveResultSet = false;
  (*pQuery)->msgType = TDMT_VND_SUBMIT;
  (*pQuery)->pRoot = (SNode*)context.pOutput;
X
Xiaoyu Wang 已提交
1634

D
dapan1121 已提交
1635 1636 1637 1638 1639 1640
  if (NULL == (*pQuery)->pTableList) {
    (*pQuery)->pTableList = taosArrayInit(taosHashGetSize(context.pTableNameHashObj), sizeof(SName));
    if (NULL == (*pQuery)->pTableList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
1641

D
dapan1121 已提交
1642 1643 1644 1645 1646 1647 1648
  if (NULL == (*pQuery)->pDbList) {
    (*pQuery)->pDbList = taosArrayInit(taosHashGetSize(context.pDbFNameHashObj), TSDB_DB_FNAME_LEN);
    if (NULL == (*pQuery)->pDbList) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }

1649
  context.pOutput->payloadType = PAYLOAD_TYPE_KV;
1650

1651
  int32_t code = skipInsertInto(&context.pSql, &context.msg);
1652 1653 1654
  if (TSDB_CODE_SUCCESS == code) {
    code = parseInsertBody(&context);
  }
1655
  if (TSDB_CODE_SUCCESS == code || NEED_CLIENT_HANDLE_ERROR(code)) {
X
Xiaoyu Wang 已提交
1656 1657 1658 1659 1660
    SName* pTable = taosHashIterate(context.pTableNameHashObj, NULL);
    while (NULL != pTable) {
      taosArrayPush((*pQuery)->pTableList, pTable);
      pTable = taosHashIterate(context.pTableNameHashObj, pTable);
    }
D
dapan1121 已提交
1661 1662 1663 1664 1665 1666

    char* pDb = taosHashIterate(context.pDbFNameHashObj, NULL);
    while (NULL != pDb) {
      taosArrayPush((*pQuery)->pDbList, pDb);
      pDb = taosHashIterate(context.pDbFNameHashObj, pDb);
    }
X
Xiaoyu Wang 已提交
1667
  }
1668
  destroyInsertParseContext(&context);
X
Xiaoyu Wang 已提交
1669
  return code;
1670
}
D
stmt  
dapan1121 已提交
1671

1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
static int32_t skipValuesClause(SInsertParseSyntaxCxt* pCxt) {
  int32_t numOfRows = 0;
  SToken  sToken;
  while (1) {
    int32_t index = 0;
    NEXT_TOKEN_KEEP_SQL(pCxt->pSql, sToken, index);
    if (TK_NK_LP != sToken.type) {
      break;
    }
    pCxt->pSql += index;

    CHECK_CODE(skipParentheses(pCxt));
    ++numOfRows;
  }
  if (0 == numOfRows) {
    return buildSyntaxErrMsg(&pCxt->msg, "no any data points", NULL);
  }
  return TSDB_CODE_SUCCESS;
}

static int32_t skipTagsClause(SInsertParseSyntaxCxt* pCxt) { return skipParentheses(pCxt); }

// pSql -> [(tag1_name, ...)] TAGS (tag1_value, ...)
static int32_t skipUsingClause(SInsertParseSyntaxCxt* pCxt) {
  SToken sToken;
  NEXT_TOKEN(pCxt->pSql, sToken);
  if (TK_NK_LP == sToken.type) {
    CHECK_CODE(skipBoundColumns(pCxt));
    NEXT_TOKEN(pCxt->pSql, sToken);
  }

  if (TK_TAGS != sToken.type) {
    return buildSyntaxErrMsg(&pCxt->msg, "TAGS is expected", sToken.z);
  }
  // pSql -> (tag1_value, ...)
  NEXT_TOKEN(pCxt->pSql, sToken);
  if (TK_NK_LP != sToken.type) {
    return buildSyntaxErrMsg(&pCxt->msg, "( is expected", sToken.z);
  }
  CHECK_CODE(skipTagsClause(pCxt));

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1717
static int32_t collectTableMetaKey(SInsertParseSyntaxCxt* pCxt, bool isStable, int32_t tableNo, SToken* pTbToken) {
1718 1719
  SName name;
  CHECK_CODE(createSName(&name, pTbToken, pCxt->pComCxt->acctId, pCxt->pComCxt->db, &pCxt->msg));
X
Xiaoyu Wang 已提交
1720 1721
  CHECK_CODE(reserveTableMetaInCacheForInsert(&name, isStable ? CATALOG_REQ_TYPE_META : CATALOG_REQ_TYPE_BOTH, tableNo,
                                              pCxt->pMetaCache));
1722 1723 1724
  return TSDB_CODE_SUCCESS;
}

1725 1726 1727 1728 1729 1730 1731
static int32_t checkTableName(const char* pTableName, SMsgBuf* pMsgBuf) {
  if (NULL != strchr(pTableName, '.')) {
    return generateSyntaxErrMsgExt(pMsgBuf, TSDB_CODE_PAR_INVALID_IDENTIFIER_NAME, "The table name cannot contain '.'");
  }
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1732
static int32_t collectAutoCreateTableMetaKey(SInsertParseSyntaxCxt* pCxt, int32_t tableNo, SToken* pTbToken) {
X
Xiaoyu Wang 已提交
1733 1734
  SName name;
  CHECK_CODE(createSName(&name, pTbToken, pCxt->pComCxt->acctId, pCxt->pComCxt->db, &pCxt->msg));
1735
  CHECK_CODE(checkTableName(name.tname, &pCxt->msg));
X
Xiaoyu Wang 已提交
1736
  CHECK_CODE(reserveTableMetaInCacheForInsert(&name, CATALOG_REQ_TYPE_VGROUP, tableNo, pCxt->pMetaCache));
X
Xiaoyu Wang 已提交
1737 1738 1739
  return TSDB_CODE_SUCCESS;
}

1740
static int32_t parseInsertBodySyntax(SInsertParseSyntaxCxt* pCxt) {
X
Xiaoyu Wang 已提交
1741 1742
  bool    hasData = false;
  int32_t tableNo = 0;
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766
  // for each table
  while (1) {
    SToken sToken;

    // pSql -> tb_name ...
    NEXT_TOKEN(pCxt->pSql, sToken);

    // no data in the sql string anymore.
    if (sToken.n == 0) {
      if (sToken.type && pCxt->pSql[0]) {
        return buildSyntaxErrMsg(&pCxt->msg, "invalid charactor in SQL", sToken.z);
      }

      if (!hasData) {
        return buildInvalidOperationMsg(&pCxt->msg, "no data in sql");
      }
      break;
    }

    hasData = false;

    SToken tbnameToken = sToken;
    NEXT_TOKEN(pCxt->pSql, sToken);

1767
    bool existedUsing = false;
1768 1769
    // USING clause
    if (TK_USING == sToken.type) {
1770
      existedUsing = true;
X
Xiaoyu Wang 已提交
1771
      CHECK_CODE(collectAutoCreateTableMetaKey(pCxt, tableNo, &tbnameToken));
1772
      NEXT_TOKEN(pCxt->pSql, sToken);
X
Xiaoyu Wang 已提交
1773
      CHECK_CODE(collectTableMetaKey(pCxt, true, tableNo, &sToken));
1774 1775 1776 1777 1778 1779 1780 1781 1782 1783
      CHECK_CODE(skipUsingClause(pCxt));
      NEXT_TOKEN(pCxt->pSql, sToken);
    }

    if (TK_NK_LP == sToken.type) {
      // pSql -> field1_name, ...)
      CHECK_CODE(skipBoundColumns(pCxt));
      NEXT_TOKEN(pCxt->pSql, sToken);
    }

1784 1785
    if (TK_USING == sToken.type && !existedUsing) {
      existedUsing = true;
X
Xiaoyu Wang 已提交
1786
      CHECK_CODE(collectAutoCreateTableMetaKey(pCxt, tableNo, &tbnameToken));
1787
      NEXT_TOKEN(pCxt->pSql, sToken);
X
Xiaoyu Wang 已提交
1788
      CHECK_CODE(collectTableMetaKey(pCxt, true, tableNo, &sToken));
1789 1790
      CHECK_CODE(skipUsingClause(pCxt));
      NEXT_TOKEN(pCxt->pSql, sToken);
X
Xiaoyu Wang 已提交
1791
    } else if (!existedUsing) {
X
Xiaoyu Wang 已提交
1792
      CHECK_CODE(collectTableMetaKey(pCxt, false, tableNo, &tbnameToken));
1793 1794
    }

X
Xiaoyu Wang 已提交
1795 1796
    ++tableNo;

1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820
    if (TK_VALUES == sToken.type) {
      // pSql -> (field1_value, ...) [(field1_value2, ...) ...]
      CHECK_CODE(skipValuesClause(pCxt));
      hasData = true;
      continue;
    }

    // FILE csv_file_path
    if (TK_FILE == sToken.type) {
      // pSql -> csv_file_path
      NEXT_TOKEN(pCxt->pSql, sToken);
      if (0 == sToken.n || (TK_NK_STRING != sToken.type && TK_NK_ID != sToken.type)) {
        return buildSyntaxErrMsg(&pCxt->msg, "file path is required following keyword FILE", sToken.z);
      }
      hasData = true;
      continue;
    }

    return buildSyntaxErrMsg(&pCxt->msg, "keyword VALUES or FILE is expected", sToken.z);
  }

  return TSDB_CODE_SUCCESS;
}

1821
int32_t parseInsertSyntax(SParseContext* pContext, SQuery** pQuery, SParseMetaCache* pMetaCache) {
1822 1823 1824
  SInsertParseSyntaxCxt context = {.pComCxt = pContext,
                                   .pSql = (char*)pContext->pSql,
                                   .msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
1825 1826
                                   .pMetaCache = pMetaCache};
  int32_t               code = skipInsertInto(&context.pSql, &context.msg);
1827 1828 1829 1830
  if (TSDB_CODE_SUCCESS == code) {
    code = parseInsertBodySyntax(&context);
  }
  if (TSDB_CODE_SUCCESS == code) {
1831
    *pQuery = (SQuery*)nodesMakeNode(QUERY_NODE_QUERY);
1832 1833 1834 1835 1836 1837 1838
    if (NULL == *pQuery) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  return code;
}

X
Xiaoyu Wang 已提交
1839 1840 1841 1842
int32_t qCreateSName(SName* pName, const char* pTableName, int32_t acctId, char* dbName, char* msgBuf,
                     int32_t msgBufLen) {
  SMsgBuf msg = {.buf = msgBuf, .len = msgBufLen};
  SToken  sToken;
D
stmt  
dapan1121 已提交
1843
  int32_t code = 0;
X
Xiaoyu Wang 已提交
1844 1845
  char*   tbName = NULL;

D
stmt  
dapan1121 已提交
1846
  NEXT_TOKEN(pTableName, sToken);
X
Xiaoyu Wang 已提交
1847

D
stmt  
dapan1121 已提交
1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858
  if (sToken.n == 0) {
    return buildInvalidOperationMsg(&msg, "empty table name");
  }

  code = createSName(pName, &sToken, acctId, dbName, &msg);
  if (code) {
    return code;
  }

  NEXT_TOKEN(pTableName, sToken);

D
stmt  
dapan1121 已提交
1859
  if (sToken.n > 0) {
D
stmt  
dapan1121 已提交
1860 1861 1862 1863 1864 1865 1866
    return buildInvalidOperationMsg(&msg, "table name format is wrong");
  }

  return TSDB_CODE_SUCCESS;
}

int32_t qBuildStmtOutput(SQuery* pQuery, SHashObj* pVgHash, SHashObj* pBlockHash) {
X
Xiaoyu Wang 已提交
1867 1868
  SVnodeModifOpStmt*  modifyNode = (SVnodeModifOpStmt*)pQuery->pRoot;
  int32_t             code = 0;
D
stmt  
dapan1121 已提交
1869
  SInsertParseContext insertCtx = {
X
Xiaoyu Wang 已提交
1870 1871 1872
      .pVgroupsHashObj = pVgHash,
      .pTableBlockHashObj = pBlockHash,
      .pOutput = (SVnodeModifOpStmt*)pQuery->pRoot,
D
stmt  
dapan1121 已提交
1873
  };
X
Xiaoyu Wang 已提交
1874

D
stmt  
dapan1121 已提交
1875 1876
  // merge according to vgId
  if (taosHashGetSize(insertCtx.pTableBlockHashObj) > 0) {
D
stmt  
dapan1121 已提交
1877
    CHECK_CODE(mergeTableDataBlocks(insertCtx.pTableBlockHashObj, modifyNode->payloadType, &insertCtx.pVgDataBlocks));
D
stmt  
dapan1121 已提交
1878 1879 1880 1881
  }

  CHECK_CODE(buildOutput(&insertCtx));

wmmhello's avatar
wmmhello 已提交
1882
  destroyBlockArrayList(insertCtx.pVgDataBlocks);
D
stmt  
dapan1121 已提交
1883 1884 1885
  return TSDB_CODE_SUCCESS;
}

1886 1887
int32_t qBindStmtTagsValue(void* pBlock, void* boundTags, int64_t suid, const char* sTableName, char* tName,
                           TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen) {
X
Xiaoyu Wang 已提交
1888 1889
  STableDataBlocks*   pDataBlock = (STableDataBlocks*)pBlock;
  SMsgBuf             pBuf = {.buf = msgBuf, .len = msgBufLen};
D
stmt  
dapan1121 已提交
1890 1891 1892 1893 1894
  SParsedDataColInfo* tags = (SParsedDataColInfo*)boundTags;
  if (NULL == tags) {
    return TSDB_CODE_QRY_APP_ERROR;
  }

C
Cary Xu 已提交
1895 1896
  SArray* pTagArray = taosArrayInit(tags->numOfBound, sizeof(STagVal));
  if (!pTagArray) {
C
Cary Xu 已提交
1897
    return buildInvalidOperationMsg(&pBuf, "out of memory");
D
stmt  
dapan1121 已提交
1898 1899
  }

wmmhello's avatar
wmmhello 已提交
1900 1901 1902 1903 1904
  SArray* tagName = taosArrayInit(8, TSDB_COL_NAME_LEN);
  if (!tagName) {
    return buildInvalidOperationMsg(&pBuf, "out of memory");
  }

1905
  int32_t  code = TSDB_CODE_SUCCESS;
D
dapan1121 已提交
1906
  SSchema* pSchema = getTableTagSchema(pDataBlock->pTableMeta);
D
stmt  
dapan1121 已提交
1907

1908
  bool  isJson = false;
wmmhello's avatar
wmmhello 已提交
1909
  STag* pTag = NULL;
D
dapan1121 已提交
1910

D
stmt  
dapan1121 已提交
1911 1912 1913 1914
  for (int c = 0; c < tags->numOfBound; ++c) {
    if (bind[c].is_null && bind[c].is_null[0]) {
      continue;
    }
X
Xiaoyu Wang 已提交
1915

X
Xiaoyu Wang 已提交
1916
    SSchema* pTagSchema = &pSchema[tags->boundColumns[c]];
1917
    int32_t  colLen = pTagSchema->bytes;
D
stmt  
dapan1121 已提交
1918 1919 1920
    if (IS_VAR_DATA_TYPE(pTagSchema->type)) {
      colLen = bind[c].length[0];
    }
wmmhello's avatar
wmmhello 已提交
1921
    taosArrayPush(tagName, pTagSchema->name);
wmmhello's avatar
wmmhello 已提交
1922 1923 1924 1925 1926
    if (pTagSchema->type == TSDB_DATA_TYPE_JSON) {
      if (colLen > (TSDB_MAX_JSON_TAG_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
        code = buildSyntaxErrMsg(&pBuf, "json string too long than 4095", bind[c].buffer);
        goto end;
      }
X
Xiaoyu Wang 已提交
1927

wmmhello's avatar
wmmhello 已提交
1928
      isJson = true;
1929
      char* tmp = taosMemoryCalloc(1, colLen + 1);
wmmhello's avatar
wmmhello 已提交
1930
      memcpy(tmp, bind[c].buffer, colLen);
wmmhello's avatar
wmmhello 已提交
1931
      code = parseJsontoTagData(tmp, pTagArray, &pTag, &pBuf);
wmmhello's avatar
wmmhello 已提交
1932
      taosMemoryFree(tmp);
1933
      if (code != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1934 1935
        goto end;
      }
1936
    } else {
wmmhello's avatar
wmmhello 已提交
1937
      STagVal val = {.cid = pTagSchema->colId, .type = pTagSchema->type};
1938
      //      strcpy(val.colName, pTagSchema->name);
1939
      if (pTagSchema->type == TSDB_DATA_TYPE_BINARY) {
wmmhello's avatar
wmmhello 已提交
1940 1941
        val.pData = (uint8_t*)bind[c].buffer;
        val.nData = colLen;
1942
      } else if (pTagSchema->type == TSDB_DATA_TYPE_NCHAR) {
wmmhello's avatar
wmmhello 已提交
1943
        int32_t output = 0;
1944 1945
        void*   p = taosMemoryCalloc(1, colLen * TSDB_NCHAR_SIZE);
        if (p == NULL) {
wmmhello's avatar
wmmhello 已提交
1946 1947 1948
          code = TSDB_CODE_OUT_OF_MEMORY;
          goto end;
        }
wmmhello's avatar
wmmhello 已提交
1949
        if (!taosMbsToUcs4(bind[c].buffer, colLen, (TdUcs4*)(p), colLen * TSDB_NCHAR_SIZE, &output)) {
wmmhello's avatar
wmmhello 已提交
1950 1951
          if (errno == E2BIG) {
            taosMemoryFree(p);
wmmhello's avatar
wmmhello 已提交
1952
            code = generateSyntaxErrMsg(&pBuf, TSDB_CODE_PAR_VALUE_TOO_LONG, pTagSchema->name);
wmmhello's avatar
wmmhello 已提交
1953 1954 1955 1956
            goto end;
          }
          char buf[512] = {0};
          snprintf(buf, tListLen(buf), " taosMbsToUcs4 error:%s", strerror(errno));
wmmhello's avatar
wmmhello 已提交
1957
          taosMemoryFree(p);
wmmhello's avatar
wmmhello 已提交
1958
          code = buildSyntaxErrMsg(&pBuf, buf, bind[c].buffer);
wmmhello's avatar
wmmhello 已提交
1959 1960
          goto end;
        }
wmmhello's avatar
wmmhello 已提交
1961 1962
        val.pData = p;
        val.nData = output;
1963
      } else {
wmmhello's avatar
wmmhello 已提交
1964
        memcpy(&val.i64, bind[c].buffer, colLen);
wmmhello's avatar
wmmhello 已提交
1965
      }
wmmhello's avatar
wmmhello 已提交
1966
      taosArrayPush(pTagArray, &val);
wmmhello's avatar
wmmhello 已提交
1967
    }
D
stmt  
dapan1121 已提交
1968 1969
  }

wmmhello's avatar
wmmhello 已提交
1970
  if (!isJson && (code = tTagNew(pTagArray, 1, false, &pTag)) != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1971
    goto end;
D
stmt  
dapan1121 已提交
1972 1973 1974
  }

  SVCreateTbReq tbReq = {0};
1975
  buildCreateTbReq(&tbReq, tName, pTag, suid, sTableName, tagName, pDataBlock->pTableMeta->tableInfo.numOfTags);
wmmhello's avatar
wmmhello 已提交
1976
  code = buildCreateTbMsg(pDataBlock, &tbReq);
wmmhello's avatar
wmmhello 已提交
1977
  tdDestroySVCreateTbReq(&tbReq);
D
stmt  
dapan1121 已提交
1978

wmmhello's avatar
wmmhello 已提交
1979 1980
end:
  for (int i = 0; i < taosArrayGetSize(pTagArray); ++i) {
1981 1982
    STagVal* p = (STagVal*)taosArrayGet(pTagArray, i);
    if (p->type == TSDB_DATA_TYPE_NCHAR) {
wmmhello's avatar
wmmhello 已提交
1983 1984 1985
      taosMemoryFree(p->pData);
    }
  }
C
Cary Xu 已提交
1986
  taosArrayDestroy(pTagArray);
wmmhello's avatar
wmmhello 已提交
1987
  taosArrayDestroy(tagName);
D
stmt  
dapan1121 已提交
1988

wmmhello's avatar
wmmhello 已提交
1989
  return code;
D
stmt  
dapan1121 已提交
1990 1991
}

X
Xiaoyu Wang 已提交
1992 1993 1994 1995
int32_t qBindStmtColsValue(void* pBlock, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen) {
  STableDataBlocks*   pDataBlock = (STableDataBlocks*)pBlock;
  SSchema*            pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
  int32_t             extendedRowSize = getExtendedRowSize(pDataBlock);
D
stmt  
dapan1121 已提交
1996 1997
  SParsedDataColInfo* spd = &pDataBlock->boundColumnInfo;
  SRowBuilder*        pBuilder = &pDataBlock->rowBuilder;
X
Xiaoyu Wang 已提交
1998 1999 2000 2001
  SMemParam           param = {.rb = pBuilder};
  SMsgBuf             pBuf = {.buf = msgBuf, .len = msgBufLen};
  int32_t             rowNum = bind->num;

D
stmt  
dapan1121 已提交
2002 2003
  CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));

D
stmt  
dapan1121 已提交
2004
  CHECK_CODE(allocateMemForSize(pDataBlock, extendedRowSize * bind->num));
X
Xiaoyu Wang 已提交
2005

D
stmt  
dapan1121 已提交
2006 2007 2008
  for (int32_t r = 0; r < bind->num; ++r) {
    STSRow* row = (STSRow*)(pDataBlock->pData + pDataBlock->size);  // skip the SSubmitBlk header
    tdSRowResetBuf(pBuilder, row);
X
Xiaoyu Wang 已提交
2009

D
stmt  
dapan1121 已提交
2010
    for (int c = 0; c < spd->numOfBound; ++c) {
X
Xiaoyu Wang 已提交
2011
      SSchema* pColSchema = &pSchema[spd->boundColumns[c]];
D
stmt  
dapan1121 已提交
2012 2013 2014 2015

      if (bind[c].num != rowNum) {
        return buildInvalidOperationMsg(&pBuf, "row number in each bind param should be the same");
      }
X
Xiaoyu Wang 已提交
2016

D
stmt  
dapan1121 已提交
2017 2018 2019 2020
      param.schema = pColSchema;
      getSTSRowAppendInfo(pBuilder->rowType, spd, c, &param.toffset, &param.colIdx);

      if (bind[c].is_null && bind[c].is_null[r]) {
D
stmt  
dapan1121 已提交
2021 2022 2023
        if (pColSchema->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
          return buildInvalidOperationMsg(&pBuf, "primary timestamp should not be NULL");
        }
X
Xiaoyu Wang 已提交
2024

D
stmt  
dapan1121 已提交
2025 2026
        CHECK_CODE(MemRowAppend(&pBuf, NULL, 0, &param));
      } else {
D
dapan1121 已提交
2027 2028 2029 2030
        if (bind[c].buffer_type != pColSchema->type) {
          return buildInvalidOperationMsg(&pBuf, "column type mis-match with buffer type");
        }

D
stmt  
dapan1121 已提交
2031 2032 2033 2034
        int32_t colLen = pColSchema->bytes;
        if (IS_VAR_DATA_TYPE(pColSchema->type)) {
          colLen = bind[c].length[r];
        }
X
Xiaoyu Wang 已提交
2035 2036

        CHECK_CODE(MemRowAppend(&pBuf, (char*)bind[c].buffer + bind[c].buffer_length * r, colLen, &param));
D
stmt  
dapan1121 已提交
2037
      }
X
Xiaoyu Wang 已提交
2038

D
stmt  
dapan1121 已提交
2039 2040
      if (PRIMARYKEY_TIMESTAMP_COL_ID == pColSchema->colId) {
        TSKEY tsKey = TD_ROW_KEY(row);
X
Xiaoyu Wang 已提交
2041
        checkTimestamp(pDataBlock, (const char*)&tsKey);
D
stmt  
dapan1121 已提交
2042 2043 2044 2045
      }
    }
    // set the null value for the columns that do not assign values
    if ((spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
C
Cary Xu 已提交
2046
      pBuilder->hasNone = true;
D
stmt  
dapan1121 已提交
2047
    }
2048
    tdSRowEnd(pBuilder);
C
Cary Xu 已提交
2049
#ifdef TD_DEBUG_PRINT_ROW
C
Cary Xu 已提交
2050
    STSchema* pSTSchema = tdGetSTSChemaFromSSChema(pSchema, spd->numOfCols, 1);
C
Cary Xu 已提交
2051
    tdSRowPrint(row, pSTSchema, __func__);
C
Cary Xu 已提交
2052 2053
    taosMemoryFree(pSTSchema);
#endif
D
stmt  
dapan1121 已提交
2054 2055
    pDataBlock->size += extendedRowSize;
  }
D
stmt  
dapan1121 已提交
2056

X
Xiaoyu Wang 已提交
2057
  SSubmitBlk* pBlocks = (SSubmitBlk*)(pDataBlock->pData);
D
stmt  
dapan1121 已提交
2058
  if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, pDataBlock, bind->num)) {
2059
    return buildInvalidOperationMsg(&pBuf, "too many rows in sql, total number of rows should be less than INT32_MAX");
D
stmt  
dapan1121 已提交
2060 2061 2062 2063 2064
  }

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2065 2066 2067 2068 2069
int32_t qBindStmtSingleColValue(void* pBlock, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen, int32_t colIdx,
                                int32_t rowNum) {
  STableDataBlocks*   pDataBlock = (STableDataBlocks*)pBlock;
  SSchema*            pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
  int32_t             extendedRowSize = getExtendedRowSize(pDataBlock);
D
stmt  
dapan1121 已提交
2070 2071
  SParsedDataColInfo* spd = &pDataBlock->boundColumnInfo;
  SRowBuilder*        pBuilder = &pDataBlock->rowBuilder;
X
Xiaoyu Wang 已提交
2072 2073 2074 2075
  SMemParam           param = {.rb = pBuilder};
  SMsgBuf             pBuf = {.buf = msgBuf, .len = msgBufLen};
  bool                rowStart = (0 == colIdx);
  bool                rowEnd = ((colIdx + 1) == spd->numOfBound);
D
stmt  
dapan1121 已提交
2076 2077 2078 2079 2080

  if (rowStart) {
    CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));
    CHECK_CODE(allocateMemForSize(pDataBlock, extendedRowSize * bind->num));
  }
X
Xiaoyu Wang 已提交
2081

D
stmt  
dapan1121 已提交
2082 2083 2084 2085 2086 2087 2088
  for (int32_t r = 0; r < bind->num; ++r) {
    STSRow* row = (STSRow*)(pDataBlock->pData + pDataBlock->size + extendedRowSize * r);  // skip the SSubmitBlk header
    if (rowStart) {
      tdSRowResetBuf(pBuilder, row);
    } else {
      tdSRowGetBuf(pBuilder, row);
    }
D
stmt  
dapan1121 已提交
2089

X
Xiaoyu Wang 已提交
2090
    SSchema* pColSchema = &pSchema[spd->boundColumns[colIdx]];
D
stmt  
dapan1121 已提交
2091 2092 2093 2094

    if (bind->num != rowNum) {
      return buildInvalidOperationMsg(&pBuf, "row number in each bind param should be the same");
    }
X
Xiaoyu Wang 已提交
2095

D
stmt  
dapan1121 已提交
2096 2097 2098 2099 2100 2101 2102
    param.schema = pColSchema;
    getSTSRowAppendInfo(pBuilder->rowType, spd, colIdx, &param.toffset, &param.colIdx);

    if (bind->is_null && bind->is_null[r]) {
      if (pColSchema->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
        return buildInvalidOperationMsg(&pBuf, "primary timestamp should not be NULL");
      }
X
Xiaoyu Wang 已提交
2103

D
stmt  
dapan1121 已提交
2104 2105
      CHECK_CODE(MemRowAppend(&pBuf, NULL, 0, &param));
    } else {
D
dapan1121 已提交
2106 2107 2108 2109
      if (bind->buffer_type != pColSchema->type) {
        return buildInvalidOperationMsg(&pBuf, "column type mis-match with buffer type");
      }

D
stmt  
dapan1121 已提交
2110 2111 2112 2113
      int32_t colLen = pColSchema->bytes;
      if (IS_VAR_DATA_TYPE(pColSchema->type)) {
        colLen = bind->length[r];
      }
X
Xiaoyu Wang 已提交
2114 2115

      CHECK_CODE(MemRowAppend(&pBuf, (char*)bind->buffer + bind->buffer_length * r, colLen, &param));
D
stmt  
dapan1121 已提交
2116
    }
X
Xiaoyu Wang 已提交
2117

D
stmt  
dapan1121 已提交
2118 2119
    if (PRIMARYKEY_TIMESTAMP_COL_ID == pColSchema->colId) {
      TSKEY tsKey = TD_ROW_KEY(row);
X
Xiaoyu Wang 已提交
2120
      checkTimestamp(pDataBlock, (const char*)&tsKey);
D
stmt  
dapan1121 已提交
2121
    }
X
Xiaoyu Wang 已提交
2122

D
stmt  
dapan1121 已提交
2123 2124
    // set the null value for the columns that do not assign values
    if (rowEnd && (spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
C
Cary Xu 已提交
2125
      pBuilder->hasNone = true;
X
Xiaoyu Wang 已提交
2126
    }
2127
    if (rowEnd) {
2128
      tdSRowEnd(pBuilder);
X
Xiaoyu Wang 已提交
2129
    }
C
Cary Xu 已提交
2130
#ifdef TD_DEBUG_PRINT_ROW
X
Xiaoyu Wang 已提交
2131
    if (rowEnd) {
C
Cary Xu 已提交
2132
      STSchema* pSTSchema = tdGetSTSChemaFromSSChema(pSchema, spd->numOfCols, 1);
C
Cary Xu 已提交
2133
      tdSRowPrint(row, pSTSchema, __func__);
C
Cary Xu 已提交
2134 2135 2136
      taosMemoryFree(pSTSchema);
    }
#endif
D
stmt  
dapan1121 已提交
2137 2138
  }

D
stmt  
dapan1121 已提交
2139 2140 2141
  if (rowEnd) {
    pDataBlock->size += extendedRowSize * bind->num;

X
Xiaoyu Wang 已提交
2142
    SSubmitBlk* pBlocks = (SSubmitBlk*)(pDataBlock->pData);
D
stmt  
dapan1121 已提交
2143
    if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, pDataBlock, bind->num)) {
X
Xiaoyu Wang 已提交
2144 2145
      return buildInvalidOperationMsg(&pBuf,
                                      "too many rows in sql, total number of rows should be less than INT32_MAX");
D
stmt  
dapan1121 已提交
2146 2147 2148 2149 2150 2151
    }
  }

  return TSDB_CODE_SUCCESS;
}

2152 2153
int32_t buildBoundFields(SParsedDataColInfo* boundInfo, SSchema* pSchema, int32_t* fieldNum, TAOS_FIELD_E** fields,
                         uint8_t timePrec) {
D
stmt  
dapan1121 已提交
2154 2155 2156 2157 2158 2159
  if (fields) {
    *fields = taosMemoryCalloc(boundInfo->numOfBound, sizeof(TAOS_FIELD));
    if (NULL == *fields) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }

D
dapan1121 已提交
2160 2161
    SSchema* schema = &pSchema[boundInfo->boundColumns[0]];
    if (TSDB_DATA_TYPE_TIMESTAMP == schema->type) {
D
dapan1121 已提交
2162 2163
      (*fields)[0].precision = timePrec;
    }
2164

D
stmt  
dapan1121 已提交
2165
    for (int32_t i = 0; i < boundInfo->numOfBound; ++i) {
D
dapan1121 已提交
2166 2167 2168 2169
      schema = &pSchema[boundInfo->boundColumns[i]];
      strcpy((*fields)[i].name, schema->name);
      (*fields)[i].type = schema->type;
      (*fields)[i].bytes = schema->bytes;
D
stmt  
dapan1121 已提交
2170
    }
D
stmt  
dapan1121 已提交
2171 2172 2173 2174 2175 2176 2177
  }

  *fieldNum = boundInfo->numOfBound;

  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
2178
int32_t qBuildStmtTagFields(void* pBlock, void* boundTags, int32_t* fieldNum, TAOS_FIELD_E** fields) {
X
Xiaoyu Wang 已提交
2179
  STableDataBlocks*   pDataBlock = (STableDataBlocks*)pBlock;
D
stmt  
dapan1121 已提交
2180 2181 2182 2183
  SParsedDataColInfo* tags = (SParsedDataColInfo*)boundTags;
  if (NULL == tags) {
    return TSDB_CODE_QRY_APP_ERROR;
  }
X
Xiaoyu Wang 已提交
2184

D
dapan1121 已提交
2185 2186 2187 2188
  if (pDataBlock->pTableMeta->tableType != TSDB_SUPER_TABLE && pDataBlock->pTableMeta->tableType != TSDB_CHILD_TABLE) {
    return TSDB_CODE_TSC_STMT_API_ERROR;
  }

X
Xiaoyu Wang 已提交
2189
  SSchema* pSchema = getTableTagSchema(pDataBlock->pTableMeta);
D
stmt  
dapan1121 已提交
2190 2191 2192 2193 2194 2195 2196
  if (tags->numOfBound <= 0) {
    *fieldNum = 0;
    *fields = NULL;

    return TSDB_CODE_SUCCESS;
  }

D
dapan1121 已提交
2197
  CHECK_CODE(buildBoundFields(tags, pSchema, fieldNum, fields, 0));
X
Xiaoyu Wang 已提交
2198

D
stmt  
dapan1121 已提交
2199 2200 2201
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
2202
int32_t qBuildStmtColFields(void* pBlock, int32_t* fieldNum, TAOS_FIELD_E** fields) {
X
Xiaoyu Wang 已提交
2203 2204
  STableDataBlocks* pDataBlock = (STableDataBlocks*)pBlock;
  SSchema*          pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
D
stmt  
dapan1121 已提交
2205 2206
  if (pDataBlock->boundColumnInfo.numOfBound <= 0) {
    *fieldNum = 0;
D
stmt  
dapan1121 已提交
2207 2208 2209
    if (fields) {
      *fields = NULL;
    }
D
stmt  
dapan1121 已提交
2210 2211 2212 2213

    return TSDB_CODE_SUCCESS;
  }

2214 2215
  CHECK_CODE(buildBoundFields(&pDataBlock->boundColumnInfo, pSchema, fieldNum, fields,
                              pDataBlock->pTableMeta->tableInfo.precision));
X
Xiaoyu Wang 已提交
2216

D
stmt  
dapan1121 已提交
2217 2218 2219
  return TSDB_CODE_SUCCESS;
}

wmmhello's avatar
wmmhello 已提交
2220
// schemaless logic start
D
stmt  
dapan1121 已提交
2221

wmmhello's avatar
wmmhello 已提交
2222
typedef struct SmlExecTableHandle {
X
Xiaoyu Wang 已提交
2223 2224
  SParsedDataColInfo tags;          // each table
  SVCreateTbReq      createTblReq;  // each table
wmmhello's avatar
wmmhello 已提交
2225
} SmlExecTableHandle;
wmmhello's avatar
wmmhello 已提交
2226

wmmhello's avatar
wmmhello 已提交
2227
typedef struct SmlExecHandle {
2228 2229 2230
  SHashObj*          pBlockHash;
  SmlExecTableHandle tableExecHandle;
  SQuery*            pQuery;
wmmhello's avatar
wmmhello 已提交
2231
} SSmlExecHandle;
wmmhello's avatar
wmmhello 已提交
2232

wmmhello's avatar
wmmhello 已提交
2233 2234 2235
static void smlDestroyTableHandle(void* pHandle) {
  SmlExecTableHandle* handle = (SmlExecTableHandle*)pHandle;
  destroyBoundColumnInfo(&handle->tags);
wmmhello's avatar
wmmhello 已提交
2236
  tdDestroySVCreateTbReq(&handle->createTblReq);
wmmhello's avatar
wmmhello 已提交
2237 2238
}

2239
static int32_t smlBoundColumnData(SArray* cols, SParsedDataColInfo* pColList, SSchema* pSchema, bool isTag) {
wmmhello's avatar
wmmhello 已提交
2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251
  col_id_t nCols = pColList->numOfCols;

  pColList->numOfBound = 0;
  pColList->boundNullLen = 0;
  memset(pColList->boundColumns, 0, sizeof(col_id_t) * nCols);
  for (col_id_t i = 0; i < nCols; ++i) {
    pColList->cols[i].valStat = VAL_STAT_NONE;
  }

  bool     isOrdered = true;
  col_id_t lastColIdx = -1;  // last column found
  for (int i = 0; i < taosArrayGetSize(cols); ++i) {
X
Xiaoyu Wang 已提交
2252 2253
    SSmlKv*  kv = taosArrayGetP(cols, i);
    SToken   sToken = {.n = kv->keyLen, .z = (char*)kv->key};
wmmhello's avatar
wmmhello 已提交
2254
    col_id_t t = lastColIdx + 1;
2255 2256
    col_id_t index = ((t == 0 && !isTag) ? 0 : findCol(&sToken, t, nCols, pSchema));
    uDebug("SML, index:%d, t:%d, ncols:%d, kv->name:%s", index, t, nCols, kv->key);
wmmhello's avatar
wmmhello 已提交
2257 2258 2259 2260 2261
    if (index < 0 && t > 0) {
      index = findCol(&sToken, 0, t, pSchema);
      isOrdered = false;
    }
    if (index < 0) {
2262
      uError("smlBoundColumnData. index:%d", index);
wmmhello's avatar
wmmhello 已提交
2263 2264 2265
      return TSDB_CODE_SML_INVALID_DATA;
    }
    if (pColList->cols[index].valStat == VAL_STAT_HAS) {
2266
      uError("smlBoundColumnData. already set. index:%d", index);
wmmhello's avatar
wmmhello 已提交
2267 2268 2269 2270
      return TSDB_CODE_SML_INVALID_DATA;
    }
    lastColIdx = index;
    pColList->cols[index].valStat = VAL_STAT_HAS;
X
Xiaoyu Wang 已提交
2271
    pColList->boundColumns[pColList->numOfBound] = index;
wmmhello's avatar
wmmhello 已提交
2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297
    ++pColList->numOfBound;
    switch (pSchema[t].type) {
      case TSDB_DATA_TYPE_BINARY:
        pColList->boundNullLen += (sizeof(VarDataOffsetT) + VARSTR_HEADER_SIZE + CHAR_BYTES);
        break;
      case TSDB_DATA_TYPE_NCHAR:
        pColList->boundNullLen += (sizeof(VarDataOffsetT) + VARSTR_HEADER_SIZE + TSDB_NCHAR_SIZE);
        break;
      default:
        pColList->boundNullLen += TYPE_BYTES[pSchema[t].type];
        break;
    }
  }

  pColList->orderStatus = isOrdered ? ORDER_STATUS_ORDERED : ORDER_STATUS_DISORDERED;

  if (!isOrdered) {
    pColList->colIdxInfo = taosMemoryCalloc(pColList->numOfBound, sizeof(SBoundIdxInfo));
    if (NULL == pColList->colIdxInfo) {
      return TSDB_CODE_TSC_OUT_OF_MEMORY;
    }
    SBoundIdxInfo* pColIdx = pColList->colIdxInfo;
    for (col_id_t i = 0; i < pColList->numOfBound; ++i) {
      pColIdx[i].schemaColIdx = pColList->boundColumns[i];
      pColIdx[i].boundIdx = i;
    }
wafwerar's avatar
wafwerar 已提交
2298
    taosSort(pColIdx, pColList->numOfBound, sizeof(SBoundIdxInfo), schemaIdxCompar);
wmmhello's avatar
wmmhello 已提交
2299 2300 2301
    for (col_id_t i = 0; i < pColList->numOfBound; ++i) {
      pColIdx[i].finalIdx = i;
    }
wafwerar's avatar
wafwerar 已提交
2302
    taosSort(pColIdx, pColList->numOfBound, sizeof(SBoundIdxInfo), boundIdxCompar);
wmmhello's avatar
wmmhello 已提交
2303 2304
  }

X
Xiaoyu Wang 已提交
2305
  if (pColList->numOfCols > pColList->numOfBound) {
wmmhello's avatar
wmmhello 已提交
2306 2307 2308 2309 2310 2311 2312
    memset(&pColList->boundColumns[pColList->numOfBound], 0,
           sizeof(col_id_t) * (pColList->numOfCols - pColList->numOfBound));
  }

  return TSDB_CODE_SUCCESS;
}

C
Cary Xu 已提交
2313 2314 2315 2316 2317 2318 2319 2320 2321 2322
/**
 * @brief No json tag for schemaless
 *
 * @param cols
 * @param tags
 * @param pSchema
 * @param ppTag
 * @param msg
 * @return int32_t
 */
2323 2324
static int32_t smlBuildTagRow(SArray* cols, SParsedDataColInfo* tags, SSchema* pSchema, STag** ppTag, SArray** tagName,
                              SMsgBuf* msg) {
C
Cary Xu 已提交
2325 2326
  SArray* pTagArray = taosArrayInit(tags->numOfBound, sizeof(STagVal));
  if (!pTagArray) {
wmmhello's avatar
wmmhello 已提交
2327 2328
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }
wmmhello's avatar
wmmhello 已提交
2329 2330 2331 2332
  *tagName = taosArrayInit(8, TSDB_COL_NAME_LEN);
  if (!*tagName) {
    return TSDB_CODE_TSC_OUT_OF_MEMORY;
  }
wmmhello's avatar
wmmhello 已提交
2333

wmmhello's avatar
wmmhello 已提交
2334
  int32_t code = TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
2335
  for (int i = 0; i < tags->numOfBound; ++i) {
X
Xiaoyu Wang 已提交
2336
    SSchema* pTagSchema = &pSchema[tags->boundColumns[i]];
2337
    SSmlKv*  kv = taosArrayGetP(cols, i);
wmmhello's avatar
wmmhello 已提交
2338

wmmhello's avatar
wmmhello 已提交
2339
    taosArrayPush(*tagName, pTagSchema->name);
wmmhello's avatar
wmmhello 已提交
2340
    STagVal val = {.cid = pTagSchema->colId, .type = pTagSchema->type};
2341
    //    strcpy(val.colName, pTagSchema->name);
2342 2343
    if (pTagSchema->type == TSDB_DATA_TYPE_BINARY) {
      val.pData = (uint8_t*)kv->value;
wmmhello's avatar
wmmhello 已提交
2344
      val.nData = kv->length;
2345
    } else if (pTagSchema->type == TSDB_DATA_TYPE_NCHAR) {
wmmhello's avatar
wmmhello 已提交
2346
      int32_t output = 0;
X
Xiaoyu Wang 已提交
2347 2348
      void*   p = taosMemoryCalloc(1, kv->length * TSDB_NCHAR_SIZE);
      if (p == NULL) {
wmmhello's avatar
wmmhello 已提交
2349 2350 2351
        code = TSDB_CODE_OUT_OF_MEMORY;
        goto end;
      }
wmmhello's avatar
wmmhello 已提交
2352
      if (!taosMbsToUcs4(kv->value, kv->length, (TdUcs4*)(p), kv->length * TSDB_NCHAR_SIZE, &output)) {
wmmhello's avatar
wmmhello 已提交
2353 2354
        if (errno == E2BIG) {
          taosMemoryFree(p);
wmmhello's avatar
wmmhello 已提交
2355
          code = generateSyntaxErrMsg(msg, TSDB_CODE_PAR_VALUE_TOO_LONG, pTagSchema->name);
wmmhello's avatar
wmmhello 已提交
2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
          goto end;
        }
        char buf[512] = {0};
        snprintf(buf, tListLen(buf), " taosMbsToUcs4 error:%s", strerror(errno));
        taosMemoryFree(p);
        code = buildSyntaxErrMsg(msg, buf, kv->value);
        goto end;
      }
      val.pData = p;
      val.nData = output;
2366
    } else {
wmmhello's avatar
wmmhello 已提交
2367
      memcpy(&val.i64, &(kv->value), kv->length);
wmmhello's avatar
wmmhello 已提交
2368
    }
wmmhello's avatar
wmmhello 已提交
2369
    taosArrayPush(pTagArray, &val);
wmmhello's avatar
wmmhello 已提交
2370 2371
  }

wmmhello's avatar
wmmhello 已提交
2372 2373 2374
  code = tTagNew(pTagArray, 1, false, ppTag);
end:
  for (int i = 0; i < taosArrayGetSize(pTagArray); ++i) {
2375 2376
    STagVal* p = (STagVal*)taosArrayGet(pTagArray, i);
    if (p->type == TSDB_DATA_TYPE_NCHAR) {
wmmhello's avatar
wmmhello 已提交
2377 2378
      taosMemoryFree(p->pData);
    }
wmmhello's avatar
wmmhello 已提交
2379
  }
C
Cary Xu 已提交
2380
  taosArrayDestroy(pTagArray);
wmmhello's avatar
wmmhello 已提交
2381
  return code;
wmmhello's avatar
wmmhello 已提交
2382 2383
}

2384
int32_t smlBindData(void* handle, SArray* tags, SArray* colsSchema, SArray* cols, bool format, STableMeta* pTableMeta,
wmmhello's avatar
wmmhello 已提交
2385
                    char* tableName, const char* sTableName, int32_t sTableNameLen, char* msgBuf, int16_t msgBufLen) {
wmmhello's avatar
wmmhello 已提交
2386 2387
  SMsgBuf pBuf = {.buf = msgBuf, .len = msgBufLen};

X
Xiaoyu Wang 已提交
2388
  SSmlExecHandle* smlHandle = (SSmlExecHandle*)handle;
2389 2390
  smlDestroyTableHandle(&smlHandle->tableExecHandle);  // free for each table
  SSchema* pTagsSchema = getTableTagSchema(pTableMeta);
wmmhello's avatar
wmmhello 已提交
2391
  setBoundColumnInfo(&smlHandle->tableExecHandle.tags, pTagsSchema, getNumOfTags(pTableMeta));
2392
  int ret = smlBoundColumnData(tags, &smlHandle->tableExecHandle.tags, pTagsSchema, true);
X
Xiaoyu Wang 已提交
2393
  if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
2394 2395 2396
    buildInvalidOperationMsg(&pBuf, "bound tags error");
    return ret;
  }
2397
  STag*   pTag = NULL;
wmmhello's avatar
wmmhello 已提交
2398 2399
  SArray* tagName = NULL;
  ret = smlBuildTagRow(tags, &smlHandle->tableExecHandle.tags, pTagsSchema, &pTag, &tagName, &pBuf);
X
Xiaoyu Wang 已提交
2400
  if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
2401
    taosArrayDestroy(tagName);
wmmhello's avatar
wmmhello 已提交
2402 2403
    return ret;
  }
wmmhello's avatar
wmmhello 已提交
2404

2405 2406
  buildCreateTbReq(&smlHandle->tableExecHandle.createTblReq, tableName, pTag, pTableMeta->suid, NULL, tagName,
                   pTableMeta->tableInfo.numOfTags);
wmmhello's avatar
wmmhello 已提交
2407 2408
  taosArrayDestroy(tagName);

wmmhello's avatar
wmmhello 已提交
2409 2410 2411
  smlHandle->tableExecHandle.createTblReq.ctb.name = taosMemoryMalloc(sTableNameLen + 1);
  memcpy(smlHandle->tableExecHandle.createTblReq.ctb.name, sTableName, sTableNameLen);
  smlHandle->tableExecHandle.createTblReq.ctb.name[sTableNameLen] = 0;
wmmhello's avatar
wmmhello 已提交
2412

wmmhello's avatar
wmmhello 已提交
2413
  STableDataBlocks* pDataBlock = NULL;
X
Xiaoyu Wang 已提交
2414 2415
  ret = getDataBlockFromList(smlHandle->pBlockHash, &pTableMeta->uid, sizeof(pTableMeta->uid),
                             TSDB_DEFAULT_PAYLOAD_SIZE, sizeof(SSubmitBlk), getTableInfo(pTableMeta).rowSize,
wmmhello's avatar
wmmhello 已提交
2416
                             pTableMeta, &pDataBlock, NULL, &smlHandle->tableExecHandle.createTblReq);
X
Xiaoyu Wang 已提交
2417
  if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
2418 2419 2420
    buildInvalidOperationMsg(&pBuf, "create data block error");
    return ret;
  }
wmmhello's avatar
wmmhello 已提交
2421 2422 2423

  SSchema* pSchema = getTableColumnSchema(pTableMeta);

2424
  ret = smlBoundColumnData(colsSchema, &pDataBlock->boundColumnInfo, pSchema, false);
X
Xiaoyu Wang 已提交
2425
  if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
2426 2427 2428
    buildInvalidOperationMsg(&pBuf, "bound cols error");
    return ret;
  }
X
Xiaoyu Wang 已提交
2429
  int32_t             extendedRowSize = getExtendedRowSize(pDataBlock);
wmmhello's avatar
wmmhello 已提交
2430 2431
  SParsedDataColInfo* spd = &pDataBlock->boundColumnInfo;
  SRowBuilder*        pBuilder = &pDataBlock->rowBuilder;
X
Xiaoyu Wang 已提交
2432
  SMemParam           param = {.rb = pBuilder};
wmmhello's avatar
wmmhello 已提交
2433 2434 2435

  initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo);

2436
  int32_t rowNum = taosArrayGetSize(cols);
2437
  if (rowNum <= 0) {
wmmhello's avatar
wmmhello 已提交
2438 2439
    return buildInvalidOperationMsg(&pBuf, "cols size <= 0");
  }
wmmhello's avatar
wmmhello 已提交
2440
  ret = allocateMemForSize(pDataBlock, extendedRowSize * rowNum);
X
Xiaoyu Wang 已提交
2441
  if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
2442 2443 2444
    buildInvalidOperationMsg(&pBuf, "allocate memory error");
    return ret;
  }
wmmhello's avatar
wmmhello 已提交
2445 2446 2447
  for (int32_t r = 0; r < rowNum; ++r) {
    STSRow* row = (STSRow*)(pDataBlock->pData + pDataBlock->size);  // skip the SSubmitBlk header
    tdSRowResetBuf(pBuilder, row);
2448
    void*  rowData = taosArrayGetP(cols, r);
2449
    size_t rowDataSize = 0;
2450
    if (format) {
2451
      rowDataSize = taosArrayGetSize(rowData);
wmmhello's avatar
wmmhello 已提交
2452
    }
wmmhello's avatar
wmmhello 已提交
2453 2454

    // 1. set the parsed value from sql string
2455
    for (int c = 0, j = 0; c < spd->numOfBound; ++c) {
2456
      SSchema* pColSchema = &pSchema[spd->boundColumns[c]];
wmmhello's avatar
wmmhello 已提交
2457 2458 2459 2460

      param.schema = pColSchema;
      getSTSRowAppendInfo(pBuilder->rowType, spd, c, &param.toffset, &param.colIdx);

X
Xiaoyu Wang 已提交
2461 2462 2463
      SSmlKv* kv = NULL;
      if (format) {
        if (j < rowDataSize) {
2464
          kv = taosArrayGetP(rowData, j);
2465
          if (rowDataSize != spd->numOfBound && j != 0 &&
X
Xiaoyu Wang 已提交
2466
              (kv->keyLen != strlen(pColSchema->name) || strncmp(kv->key, pColSchema->name, kv->keyLen) != 0)) {
2467
            kv = NULL;
X
Xiaoyu Wang 已提交
2468
          } else {
2469
            j++;
2470
          }
wmmhello's avatar
wmmhello 已提交
2471
        }
X
Xiaoyu Wang 已提交
2472 2473 2474
      } else {
        void** p = taosHashGet(rowData, pColSchema->name, strlen(pColSchema->name));
        if (p) kv = *p;
wmmhello's avatar
wmmhello 已提交
2475
      }
wmmhello's avatar
wmmhello 已提交
2476

2477
      if (!kv || kv->length == 0) {
wmmhello's avatar
wmmhello 已提交
2478 2479
        MemRowAppend(&pBuf, NULL, 0, &param);
      } else {
wmmhello's avatar
wmmhello 已提交
2480 2481
        int32_t colLen = kv->length;
        if (pColSchema->type == TSDB_DATA_TYPE_TIMESTAMP) {
S
Shengliang Guan 已提交
2482
          //          uError("SML:data before:%" PRId64 ", precision:%d", kv->i, pTableMeta->tableInfo.precision);
2483
          kv->i = convertTimePrecision(kv->i, TSDB_TIME_PRECISION_NANO, pTableMeta->tableInfo.precision);
S
Shengliang Guan 已提交
2484
          //          uError("SML:data after:%" PRId64 ", precision:%d", kv->i, pTableMeta->tableInfo.precision);
wmmhello's avatar
wmmhello 已提交
2485 2486
        }

2487
        if (IS_VAR_DATA_TYPE(kv->type)) {
wmmhello's avatar
wmmhello 已提交
2488
          MemRowAppend(&pBuf, kv->value, colLen, &param);
2489
        } else {
wmmhello's avatar
wmmhello 已提交
2490 2491
          MemRowAppend(&pBuf, &(kv->value), colLen, &param);
        }
wmmhello's avatar
wmmhello 已提交
2492 2493 2494 2495
      }

      if (PRIMARYKEY_TIMESTAMP_COL_ID == pColSchema->colId) {
        TSKEY tsKey = TD_ROW_KEY(row);
X
Xiaoyu Wang 已提交
2496
        checkTimestamp(pDataBlock, (const char*)&tsKey);
wmmhello's avatar
wmmhello 已提交
2497 2498 2499 2500 2501
      }
    }

    // set the null value for the columns that do not assign values
    if ((spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
C
Cary Xu 已提交
2502
      pBuilder->hasNone = true;
wmmhello's avatar
wmmhello 已提交
2503 2504
    }

2505
    tdSRowEnd(pBuilder);
wmmhello's avatar
wmmhello 已提交
2506 2507 2508
    pDataBlock->size += extendedRowSize;
  }

X
Xiaoyu Wang 已提交
2509
  SSubmitBlk* pBlocks = (SSubmitBlk*)(pDataBlock->pData);
wmmhello's avatar
wmmhello 已提交
2510
  if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, pDataBlock, rowNum)) {
2511
    return buildInvalidOperationMsg(&pBuf, "too many rows in sql, total number of rows should be less than INT32_MAX");
wmmhello's avatar
wmmhello 已提交
2512 2513 2514 2515 2516
  }

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
2517 2518 2519
void* smlInitHandle(SQuery* pQuery) {
  SSmlExecHandle* handle = taosMemoryCalloc(1, sizeof(SSmlExecHandle));
  if (!handle) return NULL;
wmmhello's avatar
wmmhello 已提交
2520 2521 2522 2523 2524 2525
  handle->pBlockHash = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, false);
  handle->pQuery = pQuery;

  return handle;
}

X
Xiaoyu Wang 已提交
2526 2527 2528
void smlDestroyHandle(void* pHandle) {
  if (!pHandle) return;
  SSmlExecHandle* handle = (SSmlExecHandle*)pHandle;
wmmhello's avatar
wmmhello 已提交
2529
  destroyBlockHashmap(handle->pBlockHash);
wmmhello's avatar
wmmhello 已提交
2530
  smlDestroyTableHandle(&handle->tableExecHandle);
wmmhello's avatar
wmmhello 已提交
2531 2532 2533 2534
  taosMemoryFree(handle);
}

int32_t smlBuildOutput(void* handle, SHashObj* pVgHash) {
X
Xiaoyu Wang 已提交
2535
  SSmlExecHandle* smlHandle = (SSmlExecHandle*)handle;
wmmhello's avatar
wmmhello 已提交
2536 2537
  return qBuildStmtOutput(smlHandle->pQuery, pVgHash, smlHandle->pBlockHash);
}
wmmhello's avatar
wmmhello 已提交
2538
// schemaless logic end