clientSml.c 67.3 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/>.
 */

wmmhello's avatar
wmmhello 已提交
16 17 18 19 20
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

21 22 23 24 25 26
#include "clientSml.h"

int64_t smlToMilli[3] = {3600000LL, 60000LL, 1000LL};
int64_t smlFactorNS[3] = {NANOSECOND_PER_MSEC, NANOSECOND_PER_USEC, 1};
int64_t smlFactorS[3] = {1000LL, 1000000LL, 1000000000LL};

wmmhello's avatar
wmmhello 已提交
27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
//void *nodeListGet(NodeList *list, const void *key, int32_t len, _equal_fn_sml fn) {
//  NodeList *tmp = list;
//  while (tmp) {
//    if (fn == NULL) {
//      if (tmp->data.used && tmp->data.keyLen == len && memcmp(tmp->data.key, key, len) == 0) {
//        return tmp->data.value;
//      }
//    } else {
//      if (tmp->data.used && fn(tmp->data.key, key) == 0) {
//        return tmp->data.value;
//      }
//    }
//
//    tmp = tmp->next;
//  }
//  return NULL;
//}
//
//int nodeListSet(NodeList **list, const void *key, int32_t len, void *value, _equal_fn_sml fn) {
//  NodeList *tmp = *list;
//  while (tmp) {
//    if (!tmp->data.used) break;
//    if (fn == NULL) {
//      if (tmp->data.keyLen == len && memcmp(tmp->data.key, key, len) == 0) {
//        return -1;
//      }
//    } else {
//      if (tmp->data.keyLen == len && fn(tmp->data.key, key) == 0) {
//        return -1;
//      }
//    }
//
//    tmp = tmp->next;
//  }
//  if (tmp) {
//    tmp->data.key = key;
//    tmp->data.keyLen = len;
//    tmp->data.value = value;
//    tmp->data.used = true;
//  } else {
//    NodeList *newNode = (NodeList *)taosMemoryCalloc(1, sizeof(NodeList));
//    if (newNode == NULL) {
//      return -1;
//    }
//    newNode->data.key = key;
//    newNode->data.keyLen = len;
//    newNode->data.value = value;
//    newNode->data.used = true;
//    newNode->next = *list;
//    *list = newNode;
//  }
//  return 0;
//}
//
//int nodeListSize(NodeList *list) {
//  int cnt = 0;
//  while (list) {
//    if (list->data.used)
//      cnt++;
//    else
//      break;
//    list = list->next;
//  }
//  return cnt;
//}

static int32_t smlCheckAuth(SSmlHandle *info,  SRequestConnInfo* conn, const char* pTabName, AUTH_TYPE type){
  SUserAuthInfo pAuth = {0};
  snprintf(pAuth.user, sizeof(pAuth.user), "%s", info->taos->user);
  if (NULL == pTabName) {
    tNameSetDbName(&pAuth.tbName, info->taos->acctId, info->pRequest->pDb, strlen(info->pRequest->pDb));
  } else {
    toName(info->taos->acctId, info->pRequest->pDb, pTabName, &pAuth.tbName);
wmmhello's avatar
wmmhello 已提交
100
  }
wmmhello's avatar
wmmhello 已提交
101
  pAuth.type = type;
wmmhello's avatar
wmmhello 已提交
102

wmmhello's avatar
wmmhello 已提交
103 104
  int32_t      code = TSDB_CODE_SUCCESS;
  SUserAuthRes authRes = {0};
105

wmmhello's avatar
wmmhello 已提交
106
  code = catalogChkAuth(info->pCatalog, conn, &pAuth, &authRes);
wmmhello's avatar
wmmhello 已提交
107

wmmhello's avatar
wmmhello 已提交
108

wmmhello's avatar
wmmhello 已提交
109 110 111
  return code ? (authRes.pass ? TSDB_CODE_SUCCESS : TSDB_CODE_PAR_PERMISSION_DENIED) : code;

}
112
inline bool smlDoubleToInt64OverFlow(double num) {
X
Xiaoyu Wang 已提交
113
  if (num >= (double)INT64_MAX || num <= (double)INT64_MIN) return true;
114 115 116
  return false;
}

117
int32_t smlBuildInvalidDataMsg(SSmlMsgBuf *pBuf, const char *msg1, const char *msg2) {
118
  if (pBuf->buf) {
119 120 121 122 123 124 125
    memset(pBuf->buf, 0, pBuf->len);
    if (msg1) strncat(pBuf->buf, msg1, pBuf->len);
    int32_t left = pBuf->len - strlen(pBuf->buf);
    if (left > 2 && msg2) {
      strncat(pBuf->buf, ":", left - 1);
      strncat(pBuf->buf, msg2, left - 2);
    }
wmmhello's avatar
wmmhello 已提交
126
  }
wmmhello's avatar
wmmhello 已提交
127 128 129
  return TSDB_CODE_SML_INVALID_DATA;
}

130 131 132 133 134
int64_t smlGetTimeValue(const char *value, int32_t len, uint8_t fromPrecision, uint8_t toPrecision) {
  char   *endPtr = NULL;
  int64_t tsInt64 = taosStr2Int64(value, &endPtr, 10);
  if (unlikely(value + len != endPtr)) {
    return -1;
135
  }
wmmhello's avatar
wmmhello 已提交
136

X
Xiaoyu Wang 已提交
137
  if (unlikely(fromPrecision >= TSDB_TIME_PRECISION_HOURS)) {
138
    int64_t unit = smlToMilli[fromPrecision - TSDB_TIME_PRECISION_HOURS];
D
dapan1121 已提交
139
    if (tsInt64 != 0 && unit > INT64_MAX / tsInt64) {
140 141 142 143
      return -1;
    }
    tsInt64 *= unit;
    fromPrecision = TSDB_TIME_PRECISION_MILLI;
wmmhello's avatar
wmmhello 已提交
144
  }
wmmhello's avatar
wmmhello 已提交
145

146
  return convertTimePrecision(tsInt64, fromPrecision, toPrecision);
wmmhello's avatar
wmmhello 已提交
147 148
}

149 150 151 152 153 154 155
int8_t smlGetTsTypeByLen(int32_t len) {
  if (len == TSDB_TIME_PRECISION_SEC_DIGITS) {
    return TSDB_TIME_PRECISION_SECONDS;
  } else if (len == TSDB_TIME_PRECISION_MILLI_DIGITS) {
    return TSDB_TIME_PRECISION_MILLI;
  } else {
    return -1;
156 157 158
  }
}

X
Xiaoyu Wang 已提交
159
SSmlTableInfo *smlBuildTableInfo(int numRows, const char *measure, int32_t measureLen) {
160 161 162
  SSmlTableInfo *tag = (SSmlTableInfo *)taosMemoryCalloc(sizeof(SSmlTableInfo), 1);
  if (!tag) {
    return NULL;
163 164
  }

165 166
  tag->sTableName = measure;
  tag->sTableNameLen = measureLen;
167

168 169 170 171
  tag->cols = taosArrayInit(numRows, POINTER_BYTES);
  if (tag->cols == NULL) {
    uError("SML:smlBuildTableInfo failed to allocate memory");
    goto cleanup;
172
  }
173

X
Xiaoyu Wang 已提交
174 175 176 177 178
  //  tag->tags = taosArrayInit(16, sizeof(SSmlKv));
  //  if (tag->tags == NULL) {
  //    uError("SML:smlBuildTableInfo failed to allocate memory");
  //    goto cleanup;
  //  }
179 180
  return tag;

X
Xiaoyu Wang 已提交
181
cleanup:
182 183
  taosMemoryFree(tag);
  return NULL;
184 185
}

186
static int32_t smlParseTableName(SArray *tags, char *childTableName) {
X
Xiaoyu Wang 已提交
187
  size_t childTableNameLen = strlen(tsSmlChildTableName);
188 189
  if (childTableNameLen <= 0) return TSDB_CODE_SUCCESS;

X
Xiaoyu Wang 已提交
190
  for (int i = 0; i < taosArrayGetSize(tags); i++) {
191 192 193 194 195
    SSmlKv *tag = (SSmlKv *)taosArrayGet(tags, i);
    // handle child table name
    if (childTableNameLen == tag->keyLen && strncmp(tag->key, tsSmlChildTableName, tag->keyLen) == 0) {
      memset(childTableName, 0, TSDB_TABLE_NAME_LEN);
      strncpy(childTableName, tag->value, (tag->length < TSDB_TABLE_NAME_LEN ? tag->length : TSDB_TABLE_NAME_LEN));
wmmhello's avatar
wmmhello 已提交
196
      taosArrayRemove(tags, i);
197
      break;
wmmhello's avatar
wmmhello 已提交
198 199
    }
  }
200

wmmhello's avatar
wmmhello 已提交
201 202 203
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
204
int32_t smlSetCTableName(SSmlTableInfo *oneTable) {
205
  smlParseTableName(oneTable->tags, oneTable->childTableName);
206

207
  if (strlen(oneTable->childTableName) == 0) {
X
Xiaoyu Wang 已提交
208
    SArray       *dst = taosArrayDup(oneTable->tags, NULL);
wmmhello's avatar
wmmhello 已提交
209
    RandTableName rName = {dst, oneTable->sTableName, (uint8_t)oneTable->sTableNameLen, oneTable->childTableName};
wmmhello's avatar
wmmhello 已提交
210

211 212
    buildChildTableName(&rName);
    taosArrayDestroy(dst);
213
  }
214 215
  return TSDB_CODE_SUCCESS;
}
216

217 218
void getTableUid(SSmlHandle *info, SSmlLineInfo *currElement, SSmlTableInfo *tinfo) {
  char   key[TSDB_TABLE_NAME_LEN * 2 + 1] = {0};
219 220 221
  size_t nLen = strlen(tinfo->childTableName);
  memcpy(key, currElement->measure, currElement->measureLen);
  memcpy(key + currElement->measureLen + 1, tinfo->childTableName, nLen);
222 223 224
  void *uid =
      taosHashGet(info->tableUids, key,
                  currElement->measureLen + 1 + nLen);  // use \0 as separator for stable name and child table name
225 226 227
  if (uid == NULL) {
    tinfo->uid = info->uid++;
    taosHashPut(info->tableUids, key, currElement->measureLen + 1 + nLen, &tinfo->uid, sizeof(uint64_t));
228 229
  } else {
    tinfo->uid = *(uint64_t *)uid;
230 231 232
  }
}

233 234 235 236
SSmlSTableMeta *smlBuildSTableMeta(bool isDataFormat) {
  SSmlSTableMeta *meta = (SSmlSTableMeta *)taosMemoryCalloc(sizeof(SSmlSTableMeta), 1);
  if (!meta) {
    return NULL;
237 238
  }

X
Xiaoyu Wang 已提交
239
  if (unlikely(!isDataFormat)) {
240 241 242 243 244
    meta->tagHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
    if (meta->tagHash == NULL) {
      uError("SML:smlBuildSTableMeta failed to allocate memory");
      goto cleanup;
    }
wmmhello's avatar
wmmhello 已提交
245

246 247 248 249 250
    meta->colHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
    if (meta->colHash == NULL) {
      uError("SML:smlBuildSTableMeta failed to allocate memory");
      goto cleanup;
    }
wmmhello's avatar
wmmhello 已提交
251 252
  }

253 254 255 256
  meta->tags = taosArrayInit(32, sizeof(SSmlKv));
  if (meta->tags == NULL) {
    uError("SML:smlBuildSTableMeta failed to allocate memory");
    goto cleanup;
257 258
  }

259 260 261 262 263 264
  meta->cols = taosArrayInit(32, sizeof(SSmlKv));
  if (meta->cols == NULL) {
    uError("SML:smlBuildSTableMeta failed to allocate memory");
    goto cleanup;
  }
  return meta;
265

X
Xiaoyu Wang 已提交
266
cleanup:
267 268
  taosMemoryFree(meta);
  return NULL;
wmmhello's avatar
wmmhello 已提交
269 270
}

X
Xiaoyu Wang 已提交
271 272 273 274 275 276 277
// uint16_t smlCalTypeSum(char* endptr, int32_t left){
//   uint16_t sum = 0;
//   for(int i = 0; i < left; i++){
//     sum += endptr[i];
//   }
//   return sum;
// }
wmmhello's avatar
wmmhello 已提交
278

X
Xiaoyu Wang 已提交
279 280 281
#define RETURN_FALSE                                 \
  smlBuildInvalidDataMsg(msg, "invalid data", pVal); \
  return false;
wmmhello's avatar
wmmhello 已提交
282

X
Xiaoyu Wang 已提交
283 284 285 286 287 288 289 290 291 292
#define SET_DOUBLE                     \
  kvVal->type = TSDB_DATA_TYPE_DOUBLE; \
  kvVal->d = result;

#define SET_FLOAT                                                                              \
  if (!IS_VALID_FLOAT(result)) {                                                               \
    smlBuildInvalidDataMsg(msg, "float out of range[-3.402823466e+38,3.402823466e+38]", pVal); \
    return false;                                                                              \
  }                                                                                            \
  kvVal->type = TSDB_DATA_TYPE_FLOAT;                                                          \
wmmhello's avatar
wmmhello 已提交
293 294
  kvVal->f = (float)result;

X
Xiaoyu Wang 已提交
295
#define SET_BIGINT                                                                                       \
wmmhello's avatar
wmmhello 已提交
296 297 298 299 300 301 302 303
  errno = 0;                                                                                             \
  int64_t tmp = taosStr2Int64(pVal, &endptr, 10);                                                        \
  if (errno == ERANGE) {                                                                                 \
    smlBuildInvalidDataMsg(msg, "big int out of range[-9223372036854775808,9223372036854775807]", pVal); \
    return false;                                                                                        \
  }                                                                                                      \
  kvVal->type = TSDB_DATA_TYPE_BIGINT;                                                                   \
  kvVal->i = tmp;
wmmhello's avatar
wmmhello 已提交
304

X
Xiaoyu Wang 已提交
305 306 307 308 309 310
#define SET_INT                                                                    \
  if (!IS_VALID_INT(result)) {                                                     \
    smlBuildInvalidDataMsg(msg, "int out of range[-2147483648,2147483647]", pVal); \
    return false;                                                                  \
  }                                                                                \
  kvVal->type = TSDB_DATA_TYPE_INT;                                                \
wmmhello's avatar
wmmhello 已提交
311 312
  kvVal->i = result;

X
Xiaoyu Wang 已提交
313 314 315 316 317 318
#define SET_SMALL_INT                                                          \
  if (!IS_VALID_SMALLINT(result)) {                                            \
    smlBuildInvalidDataMsg(msg, "small int our of range[-32768,32767]", pVal); \
    return false;                                                              \
  }                                                                            \
  kvVal->type = TSDB_DATA_TYPE_SMALLINT;                                       \
wmmhello's avatar
wmmhello 已提交
319 320
  kvVal->i = result;

X
Xiaoyu Wang 已提交
321
#define SET_UBIGINT                                                                             \
wmmhello's avatar
wmmhello 已提交
322 323 324 325 326 327 328 329
  errno = 0;                                                                                    \
  uint64_t tmp = taosStr2UInt64(pVal, &endptr, 10);                                             \
  if (errno == ERANGE || result < 0) {                                                          \
    smlBuildInvalidDataMsg(msg, "unsigned big int out of range[0,18446744073709551615]", pVal); \
    return false;                                                                               \
  }                                                                                             \
  kvVal->type = TSDB_DATA_TYPE_UBIGINT;                                                         \
  kvVal->u = tmp;
wmmhello's avatar
wmmhello 已提交
330

X
Xiaoyu Wang 已提交
331 332 333 334 335 336
#define SET_UINT                                                                  \
  if (!IS_VALID_UINT(result)) {                                                   \
    smlBuildInvalidDataMsg(msg, "unsigned int out of range[0,4294967295]", pVal); \
    return false;                                                                 \
  }                                                                               \
  kvVal->type = TSDB_DATA_TYPE_UINT;                                              \
wmmhello's avatar
wmmhello 已提交
337 338
  kvVal->u = result;

X
Xiaoyu Wang 已提交
339 340 341 342 343 344
#define SET_USMALL_INT                                                            \
  if (!IS_VALID_USMALLINT(result)) {                                              \
    smlBuildInvalidDataMsg(msg, "unsigned small int out of rang[0,65535]", pVal); \
    return false;                                                                 \
  }                                                                               \
  kvVal->type = TSDB_DATA_TYPE_USMALLINT;                                         \
wmmhello's avatar
wmmhello 已提交
345 346
  kvVal->u = result;

X
Xiaoyu Wang 已提交
347 348 349 350 351 352
#define SET_TINYINT                                                       \
  if (!IS_VALID_TINYINT(result)) {                                        \
    smlBuildInvalidDataMsg(msg, "tiny int out of range[-128,127]", pVal); \
    return false;                                                         \
  }                                                                       \
  kvVal->type = TSDB_DATA_TYPE_TINYINT;                                   \
wmmhello's avatar
wmmhello 已提交
353 354
  kvVal->i = result;

X
Xiaoyu Wang 已提交
355 356 357 358 359 360
#define SET_UTINYINT                                                            \
  if (!IS_VALID_UTINYINT(result)) {                                             \
    smlBuildInvalidDataMsg(msg, "unsigned tiny int out of range[0,255]", pVal); \
    return false;                                                               \
  }                                                                             \
  kvVal->type = TSDB_DATA_TYPE_UTINYINT;                                        \
wmmhello's avatar
wmmhello 已提交
361 362
  kvVal->u = result;

363
bool smlParseNumber(SSmlKv *kvVal, SSmlMsgBuf *msg) {
wmmhello's avatar
wmmhello 已提交
364 365
  const char *pVal = kvVal->value;
  int32_t     len = kvVal->length;
X
Xiaoyu Wang 已提交
366
  char       *endptr = NULL;
wmmhello's avatar
wmmhello 已提交
367 368 369 370 371 372 373 374 375
  double      result = taosStr2Double(pVal, &endptr);
  if (pVal == endptr) {
    RETURN_FALSE
  }

  int32_t left = len - (endptr - pVal);
  if (left == 0) {
    SET_DOUBLE
  } else if (left == 3) {
X
Xiaoyu Wang 已提交
376 377
    if (endptr[0] == 'f' || endptr[0] == 'F') {
      if (endptr[1] == '6' && endptr[2] == '4') {
wmmhello's avatar
wmmhello 已提交
378
        SET_DOUBLE
X
Xiaoyu Wang 已提交
379
      } else if (endptr[1] == '3' && endptr[2] == '2') {
wmmhello's avatar
wmmhello 已提交
380
        SET_FLOAT
X
Xiaoyu Wang 已提交
381
      } else {
wmmhello's avatar
wmmhello 已提交
382 383
        RETURN_FALSE
      }
X
Xiaoyu Wang 已提交
384 385
    } else if (endptr[0] == 'i' || endptr[0] == 'I') {
      if (endptr[1] == '6' && endptr[2] == '4') {
wmmhello's avatar
wmmhello 已提交
386
        SET_BIGINT
X
Xiaoyu Wang 已提交
387
      } else if (endptr[1] == '3' && endptr[2] == '2') {
wmmhello's avatar
wmmhello 已提交
388
        SET_INT
X
Xiaoyu Wang 已提交
389
      } else if (endptr[1] == '1' && endptr[2] == '6') {
wmmhello's avatar
wmmhello 已提交
390
        SET_SMALL_INT
X
Xiaoyu Wang 已提交
391
      } else {
wmmhello's avatar
wmmhello 已提交
392 393
        RETURN_FALSE
      }
X
Xiaoyu Wang 已提交
394 395
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
      if (endptr[1] == '6' && endptr[2] == '4') {
wmmhello's avatar
wmmhello 已提交
396
        SET_UBIGINT
X
Xiaoyu Wang 已提交
397
      } else if (endptr[1] == '3' && endptr[2] == '2') {
wmmhello's avatar
wmmhello 已提交
398
        SET_UINT
X
Xiaoyu Wang 已提交
399
      } else if (endptr[1] == '1' && endptr[2] == '6') {
wmmhello's avatar
wmmhello 已提交
400
        SET_USMALL_INT
X
Xiaoyu Wang 已提交
401
      } else {
wmmhello's avatar
wmmhello 已提交
402 403
        RETURN_FALSE
      }
wmmhello's avatar
wmmhello 已提交
404
    } else {
wmmhello's avatar
wmmhello 已提交
405 406
      RETURN_FALSE
    }
X
Xiaoyu Wang 已提交
407 408 409
  } else if (left == 2) {
    if (endptr[0] == 'i' || endptr[0] == 'I') {
      if (endptr[1] == '8') {
wmmhello's avatar
wmmhello 已提交
410
        SET_TINYINT
X
Xiaoyu Wang 已提交
411
      } else {
wmmhello's avatar
wmmhello 已提交
412 413
        RETURN_FALSE
      }
X
Xiaoyu Wang 已提交
414
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
wmmhello's avatar
wmmhello 已提交
415 416 417 418 419
      if (endptr[1] == '8') {
        SET_UTINYINT
      } else {
        RETURN_FALSE
      }
X
Xiaoyu Wang 已提交
420
    } else {
wmmhello's avatar
wmmhello 已提交
421 422
      RETURN_FALSE
    }
X
Xiaoyu Wang 已提交
423 424
  } else if (left == 1) {
    if (endptr[0] == 'i' || endptr[0] == 'I') {
wmmhello's avatar
wmmhello 已提交
425
      SET_BIGINT
X
Xiaoyu Wang 已提交
426
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
wmmhello's avatar
wmmhello 已提交
427
      SET_UBIGINT
X
Xiaoyu Wang 已提交
428
    } else {
wmmhello's avatar
wmmhello 已提交
429 430 431 432 433 434 435 436 437
      RETURN_FALSE
    }
  } else {
    RETURN_FALSE;
  }
  return true;
}

bool smlParseNumberOld(SSmlKv *kvVal, SSmlMsgBuf *msg) {
wmmhello's avatar
wmmhello 已提交
438
  const char *pVal = kvVal->value;
X
Xiaoyu Wang 已提交
439 440 441 442
  int32_t     len = kvVal->length;
  char       *endptr = NULL;
  double      result = taosStr2Double(pVal, &endptr);
  if (pVal == endptr) {
443
    smlBuildInvalidDataMsg(msg, "invalid data", pVal);
wmmhello's avatar
wmmhello 已提交
444 445 446
    return false;
  }

447
  int32_t left = len - (endptr - pVal);
X
Xiaoyu Wang 已提交
448
  if (left == 0 || (left == 3 && strncasecmp(endptr, "f64", left) == 0)) {
449 450
    kvVal->type = TSDB_DATA_TYPE_DOUBLE;
    kvVal->d = result;
X
Xiaoyu Wang 已提交
451 452
  } else if ((left == 3 && strncasecmp(endptr, "f32", left) == 0)) {
    if (!IS_VALID_FLOAT(result)) {
453 454
      smlBuildInvalidDataMsg(msg, "float out of range[-3.402823466e+38,3.402823466e+38]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
455
    }
456 457
    kvVal->type = TSDB_DATA_TYPE_FLOAT;
    kvVal->f = (float)result;
X
Xiaoyu Wang 已提交
458 459
  } else if ((left == 1 && *endptr == 'i') || (left == 3 && strncasecmp(endptr, "i64", left) == 0)) {
    if (smlDoubleToInt64OverFlow(result)) {
wmmhello's avatar
wmmhello 已提交
460 461
      errno = 0;
      int64_t tmp = taosStr2Int64(pVal, &endptr, 10);
X
Xiaoyu Wang 已提交
462
      if (errno == ERANGE) {
wmmhello's avatar
wmmhello 已提交
463 464 465 466 467 468
        smlBuildInvalidDataMsg(msg, "big int out of range[-9223372036854775808,9223372036854775807]", pVal);
        return false;
      }
      kvVal->type = TSDB_DATA_TYPE_BIGINT;
      kvVal->i = tmp;
      return true;
wmmhello's avatar
wmmhello 已提交
469
    }
470
    kvVal->type = TSDB_DATA_TYPE_BIGINT;
wmmhello's avatar
wmmhello 已提交
471
    kvVal->i = (int64_t)result;
wmmhello's avatar
wmmhello 已提交
472
  } else if ((left == 1 && *endptr == 'u') || (left == 3 && strncasecmp(endptr, "u64", left) == 0)) {
X
Xiaoyu Wang 已提交
473
    if (result >= (double)UINT64_MAX || result < 0) {
wmmhello's avatar
wmmhello 已提交
474 475
      errno = 0;
      uint64_t tmp = taosStr2UInt64(pVal, &endptr, 10);
X
Xiaoyu Wang 已提交
476
      if (errno == ERANGE || result < 0) {
wmmhello's avatar
wmmhello 已提交
477 478 479 480 481 482
        smlBuildInvalidDataMsg(msg, "unsigned big int out of range[0,18446744073709551615]", pVal);
        return false;
      }
      kvVal->type = TSDB_DATA_TYPE_UBIGINT;
      kvVal->u = tmp;
      return true;
483
    }
484
    kvVal->type = TSDB_DATA_TYPE_UBIGINT;
wmmhello's avatar
wmmhello 已提交
485
    kvVal->u = result;
X
Xiaoyu Wang 已提交
486 487
  } else if (left == 3 && strncasecmp(endptr, "i32", left) == 0) {
    if (!IS_VALID_INT(result)) {
488 489
      smlBuildInvalidDataMsg(msg, "int out of range[-2147483648,2147483647]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
490
    }
491 492
    kvVal->type = TSDB_DATA_TYPE_INT;
    kvVal->i = result;
X
Xiaoyu Wang 已提交
493 494
  } else if (left == 3 && strncasecmp(endptr, "u32", left) == 0) {
    if (!IS_VALID_UINT(result)) {
495 496
      smlBuildInvalidDataMsg(msg, "unsigned int out of range[0,4294967295]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
497
    }
498 499
    kvVal->type = TSDB_DATA_TYPE_UINT;
    kvVal->u = result;
X
Xiaoyu Wang 已提交
500 501
  } else if (left == 3 && strncasecmp(endptr, "i16", left) == 0) {
    if (!IS_VALID_SMALLINT(result)) {
502 503
      smlBuildInvalidDataMsg(msg, "small int our of range[-32768,32767]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
504
    }
505 506
    kvVal->type = TSDB_DATA_TYPE_SMALLINT;
    kvVal->i = result;
X
Xiaoyu Wang 已提交
507 508
  } else if (left == 3 && strncasecmp(endptr, "u16", left) == 0) {
    if (!IS_VALID_USMALLINT(result)) {
509 510
      smlBuildInvalidDataMsg(msg, "unsigned small int out of rang[0,65535]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
511
    }
512 513
    kvVal->type = TSDB_DATA_TYPE_USMALLINT;
    kvVal->u = result;
X
Xiaoyu Wang 已提交
514 515
  } else if (left == 2 && strncasecmp(endptr, "i8", left) == 0) {
    if (!IS_VALID_TINYINT(result)) {
516 517
      smlBuildInvalidDataMsg(msg, "tiny int out of range[-128,127]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
518
    }
519 520
    kvVal->type = TSDB_DATA_TYPE_TINYINT;
    kvVal->i = result;
X
Xiaoyu Wang 已提交
521 522
  } else if (left == 2 && strncasecmp(endptr, "u8", left) == 0) {
    if (!IS_VALID_UTINYINT(result)) {
523 524
      smlBuildInvalidDataMsg(msg, "unsigned tiny int out of range[0,255]", pVal);
      return false;
wmmhello's avatar
wmmhello 已提交
525
    }
526 527
    kvVal->type = TSDB_DATA_TYPE_UTINYINT;
    kvVal->u = result;
X
Xiaoyu Wang 已提交
528
  } else {
529
    smlBuildInvalidDataMsg(msg, "invalid data", pVal);
wmmhello's avatar
wmmhello 已提交
530 531
    return false;
  }
532
  return true;
wmmhello's avatar
wmmhello 已提交
533 534
}

X
Xiaoyu Wang 已提交
535
STableMeta *smlGetMeta(SSmlHandle *info, const void *measure, int32_t measureLen) {
536
  STableMeta *pTableMeta = NULL;
wmmhello's avatar
wmmhello 已提交
537

538 539
  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));
wmmhello's avatar
wmmhello 已提交
540

541 542 543 544 545 546 547
  SRequestConnInfo conn = {0};
  conn.pTrans = info->taos->pAppInfo->pTransporter;
  conn.requestId = info->pRequest->requestId;
  conn.requestObjRefId = info->pRequest->self;
  conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
  memset(pName.tname, 0, TSDB_TABLE_NAME_LEN);
  memcpy(pName.tname, measure, measureLen);
wmmhello's avatar
wmmhello 已提交
548

549
  int32_t code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
X
Xiaoyu Wang 已提交
550
  if (code != TSDB_CODE_SUCCESS) {
551 552
    return NULL;
  }
553
  return pTableMeta;
wmmhello's avatar
wmmhello 已提交
554
}
wmmhello's avatar
wmmhello 已提交
555

556 557
static int64_t smlGenId() {
  static volatile int64_t linesSmlHandleId = 0;
wmmhello's avatar
wmmhello 已提交
558

559 560 561 562
  int64_t id = 0;
  do {
    id = atomic_add_fetch_64(&linesSmlHandleId, 1);
  } while (id == 0);
563

564
  return id;
565 566
}

567 568 569 570 571
static int32_t smlGenerateSchemaAction(SSchema *colField, SHashObj *colHash, SSmlKv *kv, bool isTag,
                                       ESchemaAction *action, SSmlHandle *info) {
  uint16_t *index = colHash ? (uint16_t *)taosHashGet(colHash, kv->key, kv->keyLen) : NULL;
  if (index) {
    if (colField[*index].type != kv->type) {
572 573
      uError("SML:0x%" PRIx64 " point type and db type mismatch. db type: %d, point type: %d, key: %s", info->id,
             colField[*index].type, kv->type, kv->key);
574
      return TSDB_CODE_SML_INVALID_DATA;
575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592
    }

    if ((colField[*index].type == TSDB_DATA_TYPE_VARCHAR &&
         (colField[*index].bytes - VARSTR_HEADER_SIZE) < kv->length) ||
        (colField[*index].type == TSDB_DATA_TYPE_NCHAR &&
         ((colField[*index].bytes - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE < kv->length))) {
      if (isTag) {
        *action = SCHEMA_ACTION_CHANGE_TAG_SIZE;
      } else {
        *action = SCHEMA_ACTION_CHANGE_COLUMN_SIZE;
      }
    }
  } else {
    if (isTag) {
      *action = SCHEMA_ACTION_ADD_TAG;
    } else {
      *action = SCHEMA_ACTION_ADD_COLUMN;
    }
wmmhello's avatar
wmmhello 已提交
593
  }
594
  return 0;
595 596
}

597
#define BOUNDARY 1024
598 599
static int32_t smlFindNearestPowerOf2(int32_t length, uint8_t type) {
  int32_t result = 1;
600
  if (length >= BOUNDARY) {
601
    result = length;
602
  } else {
603
    while (result <= length) {
K
kailixu 已提交
604
      result <<= 1;
605
    }
606
  }
K
kailixu 已提交
607

608 609
  if (type == TSDB_DATA_TYPE_BINARY && result > TSDB_MAX_BINARY_LEN - VARSTR_HEADER_SIZE) {
    result = TSDB_MAX_BINARY_LEN - VARSTR_HEADER_SIZE;
K
kailixu 已提交
610 611
  } else if (type == TSDB_DATA_TYPE_NCHAR && result > (TSDB_MAX_NCHAR_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
    result = (TSDB_MAX_NCHAR_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE;
612
  }
613 614 615 616 617

  if (type == TSDB_DATA_TYPE_NCHAR) {
    result = result * TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE;
  } else if (type == TSDB_DATA_TYPE_BINARY) {
    result = result + VARSTR_HEADER_SIZE;
618
  }
619
  return result;
620 621
}

622 623 624 625 626 627 628 629 630 631
static int32_t smlProcessSchemaAction(SSmlHandle *info, SSchema *schemaField, SHashObj *schemaHash, SArray *cols,
                                      ESchemaAction *action, bool isTag) {
  int32_t code = TSDB_CODE_SUCCESS;
  for (int j = 0; j < taosArrayGetSize(cols); ++j) {
    if (j == 0 && !isTag) continue;
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, j);
    code = smlGenerateSchemaAction(schemaField, schemaHash, kv, isTag, action, info);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
632
  }
633
  return TSDB_CODE_SUCCESS;
634 635
}

636 637 638 639 640
static int32_t smlCheckMeta(SSchema *schema, int32_t length, SArray *cols, bool isTag) {
  SHashObj *hashTmp = taosHashInit(length, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
  int32_t   i = 0;
  for (; i < length; i++) {
    taosHashPut(hashTmp, schema[i].name, strlen(schema[i].name), &i, SHORT_BYTES);
641 642
  }

643 644 645 646
  if (isTag) {
    i = 0;
  } else {
    i = 1;
647
  }
648 649 650 651 652 653
  for (; i < taosArrayGetSize(cols); i++) {
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
    if (taosHashGet(hashTmp, kv->key, kv->keyLen) == NULL) {
      taosHashCleanup(hashTmp);
      return -1;
    }
654
  }
655 656 657
  taosHashCleanup(hashTmp);
  return 0;
}
658

659 660 661 662 663 664
static int32_t getBytes(uint8_t type, int32_t length) {
  if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
    return smlFindNearestPowerOf2(length, type);
  } else {
    return tDataTypes[type].bytes;
  }
665 666
}

667 668 669 670 671
static int32_t smlBuildFieldsList(SSmlHandle *info, SSchema *schemaField, SHashObj *schemaHash, SArray *cols,
                                  SArray *results, int32_t numOfCols, bool isTag) {
  for (int j = 0; j < taosArrayGetSize(cols); ++j) {
    SSmlKv       *kv = (SSmlKv *)taosArrayGet(cols, j);
    ESchemaAction action = SCHEMA_ACTION_NULL;
X
Xiaoyu Wang 已提交
672 673
    int           code = smlGenerateSchemaAction(schemaField, schemaHash, kv, isTag, &action, info);
    if (code != 0) {
D
dapan1121 已提交
674 675
      return code;
    }
676 677 678 679 680 681 682 683
    if (action == SCHEMA_ACTION_ADD_COLUMN || action == SCHEMA_ACTION_ADD_TAG) {
      SField field = {0};
      field.type = kv->type;
      field.bytes = getBytes(kv->type, kv->length);
      memcpy(field.name, kv->key, kv->keyLen);
      taosArrayPush(results, &field);
    } else if (action == SCHEMA_ACTION_CHANGE_COLUMN_SIZE || action == SCHEMA_ACTION_CHANGE_TAG_SIZE) {
      uint16_t *index = (uint16_t *)taosHashGet(schemaHash, kv->key, kv->keyLen);
X
Xiaoyu Wang 已提交
684
      if (index == NULL) {
D
dapan1121 已提交
685 686 687
        uError("smlBuildFieldsList get error, key:%s", kv->key);
        return TSDB_CODE_SML_INVALID_DATA;
      }
X
Xiaoyu Wang 已提交
688
      uint16_t newIndex = *index;
689 690 691
      if (isTag) newIndex -= numOfCols;
      SField *field = (SField *)taosArrayGet(results, newIndex);
      field->bytes = getBytes(kv->type, kv->length);
692 693
    }
  }
694 695 696 697 698 699 700 701

  int32_t maxLen = isTag ? TSDB_MAX_TAGS_LEN : TSDB_MAX_BYTES_PER_ROW;
  int32_t len = 0;
  for (int j = 0; j < taosArrayGetSize(results); ++j) {
    SField *field = taosArrayGet(results, j);
    len += field->bytes;
  }
  if(len > maxLen){
702
    return isTag ? TSDB_CODE_PAR_INVALID_TAGS_LENGTH : TSDB_CODE_PAR_INVALID_ROW_LENGTH;
703 704
  }

705 706 707
  return TSDB_CODE_SUCCESS;
}

708 709 710 711 712 713 714 715
// static int32_t smlSendMetaMsg(SSmlHandle *info, SName *pName, SSmlSTableMeta *sTableData,
//                               int32_t colVer, int32_t tagVer, int8_t source, uint64_t suid){
static int32_t smlSendMetaMsg(SSmlHandle *info, SName *pName, SArray *pColumns, SArray *pTags, STableMeta *pTableMeta,
                              ESchemaAction action) {
  SRequestObj   *pRequest = NULL;
  SMCreateStbReq pReq = {0};
  int32_t        code = TSDB_CODE_SUCCESS;
  SCmdMsgInfo    pCmdMsg = {0};
716
  char          *pSql = NULL;
717 718 719 720 721 722 723 724 725 726 727 728

  // put front for free
  pReq.numOfColumns = taosArrayGetSize(pColumns);
  pReq.pColumns = pColumns;
  pReq.numOfTags = taosArrayGetSize(pTags);
  pReq.pTags = pTags;

  if (action == SCHEMA_ACTION_CREATE_STABLE) {
    pReq.colVer = 1;
    pReq.tagVer = 1;
    pReq.suid = 0;
    pReq.source = TD_REQ_FROM_APP;
729
    pSql = "sml_create_stable";
730 731 732 733 734
  } else if (action == SCHEMA_ACTION_ADD_TAG || action == SCHEMA_ACTION_CHANGE_TAG_SIZE) {
    pReq.colVer = pTableMeta->sversion;
    pReq.tagVer = pTableMeta->tversion + 1;
    pReq.suid = pTableMeta->uid;
    pReq.source = TD_REQ_FROM_TAOX;
735
    pSql = (action == SCHEMA_ACTION_ADD_TAG) ? "sml_add_tag" : "sml_modify_tag_size";
736 737 738 739 740
  } else if (action == SCHEMA_ACTION_ADD_COLUMN || action == SCHEMA_ACTION_CHANGE_COLUMN_SIZE) {
    pReq.colVer = pTableMeta->sversion + 1;
    pReq.tagVer = pTableMeta->tversion;
    pReq.suid = pTableMeta->uid;
    pReq.source = TD_REQ_FROM_TAOX;
741 742 743 744 745 746 747 748 749 750 751 752
    pSql = (action == SCHEMA_ACTION_ADD_COLUMN) ? "sml_add_column" : "sml_modify_column_size";
  }

  code = buildRequest(info->taos->id, pSql, strlen(pSql), NULL, false, &pRequest, 0);
  if (code != TSDB_CODE_SUCCESS) {
    goto end;
  }

  pRequest->syncQuery = true;
  if (!pRequest->pDb) {
    code = TSDB_CODE_PAR_DB_NOT_SPECIFIED;
    goto end;
753
  }
754

755 756 757 758
  if (pReq.numOfTags == 0) {
    pReq.numOfTags = 1;
    SField field = {0};
    field.type = TSDB_DATA_TYPE_NCHAR;
759
    field.bytes = TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE;
760 761
    strcpy(field.name, tsSmlTagName);
    taosArrayPush(pReq.pTags, &field);
762
  }
wmmhello's avatar
wmmhello 已提交
763

764 765 766
  pReq.commentLen = -1;
  pReq.igExists = true;
  tNameExtractFullName(pName, pReq.name);
767

768 769 770 771 772 773 774 775 776
  pCmdMsg.epSet = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
  pCmdMsg.msgType = TDMT_MND_CREATE_STB;
  pCmdMsg.msgLen = tSerializeSMCreateStbReq(NULL, 0, &pReq);
  pCmdMsg.pMsg = taosMemoryMalloc(pCmdMsg.msgLen);
  if (NULL == pCmdMsg.pMsg) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto end;
  }
  tSerializeSMCreateStbReq(pCmdMsg.pMsg, pCmdMsg.msgLen, &pReq);
777

778 779 780 781 782 783
  SQuery pQuery;
  memset(&pQuery, 0, sizeof(pQuery));
  pQuery.execMode = QUERY_EXEC_MODE_RPC;
  pQuery.pCmdMsg = &pCmdMsg;
  pQuery.msgType = pQuery.pCmdMsg->msgType;
  pQuery.stableQuery = true;
784

785
  launchQueryImpl(pRequest, &pQuery, true, NULL);
786

787 788
  if (pRequest->code == TSDB_CODE_SUCCESS) {
    catalogRemoveTableMeta(info->pCatalog, pName);
789
  }
790 791
  code = pRequest->code;
  taosMemoryFree(pCmdMsg.pMsg);
792

X
Xiaoyu Wang 已提交
793
end:
794 795 796 797
  destroyRequest(pRequest);
  tFreeSMCreateStbReq(&pReq);
  return code;
}
798

799
static int32_t smlModifyDBSchemas(SSmlHandle *info) {
X
Xiaoyu Wang 已提交
800 801
  uDebug("SML:0x%" PRIx64 " smlModifyDBSchemas start, format:%d, needModifySchema:%d", info->id, info->dataFormat,
         info->needModifySchema);
X
Xiaoyu Wang 已提交
802
  if (info->dataFormat && !info->needModifySchema) {
803
    return TSDB_CODE_SUCCESS;
804
  }
805 806 807
  int32_t     code = 0;
  SHashObj   *hashTmp = NULL;
  STableMeta *pTableMeta = NULL;
808 809 810 811 812 813 814 815 816

  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));

  SRequestConnInfo conn = {0};
  conn.pTrans = info->taos->pAppInfo->pTransporter;
  conn.requestId = info->pRequest->requestId;
  conn.requestObjRefId = info->pRequest->self;
  conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
817

818
  SSmlSTableMeta **tmp = (SSmlSTableMeta **)taosHashIterate(info->superTables, NULL);
819
  while (tmp) {
820
    SSmlSTableMeta *sTableData = *tmp;
821
    bool            needCheckMeta = false;  // for multi thread
822

823 824
    size_t superTableLen = 0;
    void  *superTable = taosHashGetKey(tmp, &superTableLen);
825
    char  *measure = taosMemoryMalloc(superTableLen);
wmmhello's avatar
wmmhello 已提交
826 827
    memcpy(measure, superTable, superTableLen);
    PROCESS_SLASH_IN_MEASUREMENT(measure, superTableLen);
828
    memset(pName.tname, 0, TSDB_TABLE_NAME_LEN);
wmmhello's avatar
wmmhello 已提交
829 830
    memcpy(pName.tname, measure, superTableLen);
    taosMemoryFree(measure);
831

832
    code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
833

834
    if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST || code == TSDB_CODE_MND_STB_NOT_EXIST) {
wmmhello's avatar
wmmhello 已提交
835 836 837 838
      code = smlCheckAuth(info, &conn, NULL, AUTH_TYPE_WRITE);
      if(code != TSDB_CODE_SUCCESS){
        goto end;
      }
D
dapan1121 已提交
839
      uDebug("SML:0x%" PRIx64 " smlModifyDBSchemas create table:%s", info->id, pName.tname);
840 841
      SArray *pColumns = taosArrayInit(taosArrayGetSize(sTableData->cols), sizeof(SField));
      SArray *pTags = taosArrayInit(taosArrayGetSize(sTableData->tags), sizeof(SField));
D
dapan1121 已提交
842 843 844
      code = smlBuildFieldsList(info, NULL, NULL, sTableData->tags, pTags, 0, true);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " smlBuildFieldsList tag1 failed. %s", info->id, pName.tname);
845 846
        taosArrayDestroy(pColumns);
        taosArrayDestroy(pTags);
D
dapan1121 已提交
847 848 849 850 851
        goto end;
      }
      code = smlBuildFieldsList(info, NULL, NULL, sTableData->cols, pColumns, 0, false);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " smlBuildFieldsList col1 failed. %s", info->id, pName.tname);
852 853
        taosArrayDestroy(pColumns);
        taosArrayDestroy(pTags);
D
dapan1121 已提交
854 855
        goto end;
      }
856 857 858 859
      code = smlSendMetaMsg(info, &pName, pColumns, pTags, NULL, SCHEMA_ACTION_CREATE_STABLE);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " smlSendMetaMsg failed. can not create %s", info->id, pName.tname);
        goto end;
860
      }
861 862
      info->cost.numOfCreateSTables++;
      taosMemoryFreeClear(pTableMeta);
863

864 865 866 867
      code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " catalogGetSTableMeta failed. super table name %s", info->id, pName.tname);
        goto end;
868
      }
869 870 871 872 873 874
    } else if (code == TSDB_CODE_SUCCESS) {
      hashTmp = taosHashInit(pTableMeta->tableInfo.numOfTags, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true,
                             HASH_NO_LOCK);
      for (uint16_t i = pTableMeta->tableInfo.numOfColumns;
           i < pTableMeta->tableInfo.numOfColumns + pTableMeta->tableInfo.numOfTags; i++) {
        taosHashPut(hashTmp, pTableMeta->schema[i].name, strlen(pTableMeta->schema[i].name), &i, SHORT_BYTES);
875
      }
876

877 878 879 880
      ESchemaAction action = SCHEMA_ACTION_NULL;
      code = smlProcessSchemaAction(info, pTableMeta->schema, hashTmp, sTableData->tags, &action, true);
      if (code != TSDB_CODE_SUCCESS) {
        goto end;
881
      }
882
      if (action != SCHEMA_ACTION_NULL) {
wmmhello's avatar
wmmhello 已提交
883 884 885 886
        code = smlCheckAuth(info, &conn, pName.tname, AUTH_TYPE_WRITE);
        if(code != TSDB_CODE_SUCCESS){
          goto end;
        }
X
Xiaoyu Wang 已提交
887 888
        uDebug("SML:0x%" PRIx64 " smlModifyDBSchemas change table tag, table:%s, action:%d", info->id, pName.tname,
               action);
889 890 891 892
        SArray *pColumns =
            taosArrayInit(taosArrayGetSize(sTableData->cols) + pTableMeta->tableInfo.numOfColumns, sizeof(SField));
        SArray *pTags =
            taosArrayInit(taosArrayGetSize(sTableData->tags) + pTableMeta->tableInfo.numOfTags, sizeof(SField));
893

894 895 896 897 898 899 900 901 902
        for (uint16_t i = 0; i < pTableMeta->tableInfo.numOfColumns + pTableMeta->tableInfo.numOfTags; i++) {
          SField field = {0};
          field.type = pTableMeta->schema[i].type;
          field.bytes = pTableMeta->schema[i].bytes;
          strcpy(field.name, pTableMeta->schema[i].name);
          if (i < pTableMeta->tableInfo.numOfColumns) {
            taosArrayPush(pColumns, &field);
          } else {
            taosArrayPush(pTags, &field);
903 904
          }
        }
D
dapan1121 已提交
905
        code = smlBuildFieldsList(info, pTableMeta->schema, hashTmp, sTableData->tags, pTags,
X
Xiaoyu Wang 已提交
906
                                  pTableMeta->tableInfo.numOfColumns, true);
D
dapan1121 已提交
907 908
        if (code != TSDB_CODE_SUCCESS) {
          uError("SML:0x%" PRIx64 " smlBuildFieldsList tag2 failed. %s", info->id, pName.tname);
909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
          taosArrayDestroy(pColumns);
          taosArrayDestroy(pTags);
          goto end;
        }

        if (taosArrayGetSize(pTags) + pTableMeta->tableInfo.numOfColumns > TSDB_MAX_COLUMNS) {
          uError("SML:0x%" PRIx64 " too many columns than 4096", info->id);
          code = TSDB_CODE_PAR_TOO_MANY_COLUMNS;
          taosArrayDestroy(pColumns);
          taosArrayDestroy(pTags);
          goto end;
        }
        if (taosArrayGetSize(pTags) > TSDB_MAX_TAGS) {
          uError("SML:0x%" PRIx64 " too many tags than 128", info->id);
          code = TSDB_CODE_PAR_INVALID_TAGS_NUM;
          taosArrayDestroy(pColumns);
          taosArrayDestroy(pTags);
D
dapan1121 已提交
926 927
          goto end;
        }
928

929 930 931 932
        code = smlSendMetaMsg(info, &pName, pColumns, pTags, pTableMeta, action);
        if (code != TSDB_CODE_SUCCESS) {
          uError("SML:0x%" PRIx64 " smlSendMetaMsg failed. can not create %s", info->id, pName.tname);
          goto end;
933
        }
934

935 936 937 938 939 940 941 942 943
        info->cost.numOfAlterTagSTables++;
        taosMemoryFreeClear(pTableMeta);
        code = catalogRefreshTableMeta(info->pCatalog, &conn, &pName, -1);
        if (code != TSDB_CODE_SUCCESS) {
          goto end;
        }
        code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
        if (code != TSDB_CODE_SUCCESS) {
          goto end;
wmmhello's avatar
wmmhello 已提交
944
        }
945 946
      }

947 948 949
      taosHashClear(hashTmp);
      for (uint16_t i = 0; i < pTableMeta->tableInfo.numOfColumns; i++) {
        taosHashPut(hashTmp, pTableMeta->schema[i].name, strlen(pTableMeta->schema[i].name), &i, SHORT_BYTES);
950
      }
951 952 953 954
      action = SCHEMA_ACTION_NULL;
      code = smlProcessSchemaAction(info, pTableMeta->schema, hashTmp, sTableData->cols, &action, false);
      if (code != TSDB_CODE_SUCCESS) {
        goto end;
955
      }
956
      if (action != SCHEMA_ACTION_NULL) {
wmmhello's avatar
wmmhello 已提交
957 958 959 960
        code = smlCheckAuth(info, &conn, pName.tname, AUTH_TYPE_WRITE);
        if(code != TSDB_CODE_SUCCESS){
          goto end;
        }
X
Xiaoyu Wang 已提交
961 962
        uDebug("SML:0x%" PRIx64 " smlModifyDBSchemas change table col, table:%s, action:%d", info->id, pName.tname,
               action);
963 964 965 966
        SArray *pColumns =
            taosArrayInit(taosArrayGetSize(sTableData->cols) + pTableMeta->tableInfo.numOfColumns, sizeof(SField));
        SArray *pTags =
            taosArrayInit(taosArrayGetSize(sTableData->tags) + pTableMeta->tableInfo.numOfTags, sizeof(SField));
967

968 969 970 971 972 973 974 975 976 977
        for (uint16_t i = 0; i < pTableMeta->tableInfo.numOfColumns + pTableMeta->tableInfo.numOfTags; i++) {
          SField field = {0};
          field.type = pTableMeta->schema[i].type;
          field.bytes = pTableMeta->schema[i].bytes;
          strcpy(field.name, pTableMeta->schema[i].name);
          if (i < pTableMeta->tableInfo.numOfColumns) {
            taosArrayPush(pColumns, &field);
          } else {
            taosArrayPush(pTags, &field);
          }
978 979
        }

D
dapan1121 已提交
980
        code = smlBuildFieldsList(info, pTableMeta->schema, hashTmp, sTableData->cols, pColumns,
X
Xiaoyu Wang 已提交
981
                                  pTableMeta->tableInfo.numOfColumns, false);
D
dapan1121 已提交
982 983
        if (code != TSDB_CODE_SUCCESS) {
          uError("SML:0x%" PRIx64 " smlBuildFieldsList col2 failed. %s", info->id, pName.tname);
984 985 986 987 988 989 990 991 992 993
          taosArrayDestroy(pColumns);
          taosArrayDestroy(pTags);
          goto end;
        }

        if (taosArrayGetSize(pColumns) + pTableMeta->tableInfo.numOfTags > TSDB_MAX_COLUMNS) {
          uError("SML:0x%" PRIx64 " too many columns than 4096", info->id);
          code = TSDB_CODE_PAR_TOO_MANY_COLUMNS;
          taosArrayDestroy(pColumns);
          taosArrayDestroy(pTags);
D
dapan1121 已提交
994 995
          goto end;
        }
996

997 998 999 1000
        code = smlSendMetaMsg(info, &pName, pColumns, pTags, pTableMeta, action);
        if (code != TSDB_CODE_SUCCESS) {
          uError("SML:0x%" PRIx64 " smlSendMetaMsg failed. can not create %s", info->id, pName.tname);
          goto end;
1001
        }
1002

1003 1004 1005 1006 1007 1008 1009 1010 1011 1012
        info->cost.numOfAlterColSTables++;
        taosMemoryFreeClear(pTableMeta);
        code = catalogRefreshTableMeta(info->pCatalog, &conn, &pName, -1);
        if (code != TSDB_CODE_SUCCESS) {
          goto end;
        }
        code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
        if (code != TSDB_CODE_SUCCESS) {
          uError("SML:0x%" PRIx64 " catalogGetSTableMeta failed. super table name %s", info->id, pName.tname);
          goto end;
1013 1014
        }
      }
wmmhello's avatar
wmmhello 已提交
1015

1016 1017 1018
      needCheckMeta = true;
      taosHashCleanup(hashTmp);
      hashTmp = NULL;
X
Xiaoyu Wang 已提交
1019
    } else {
1020 1021
      uError("SML:0x%" PRIx64 " load table meta error: %s", info->id, tstrerror(code));
      goto end;
wmmhello's avatar
wmmhello 已提交
1022
    }
1023

1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034
    if (needCheckMeta) {
      code = smlCheckMeta(&(pTableMeta->schema[pTableMeta->tableInfo.numOfColumns]), pTableMeta->tableInfo.numOfTags,
                          sTableData->tags, true);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " check tag failed. super table name %s", info->id, pName.tname);
        goto end;
      }
      code = smlCheckMeta(&(pTableMeta->schema[0]), pTableMeta->tableInfo.numOfColumns, sTableData->cols, false);
      if (code != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " check cols failed. super table name %s", info->id, pName.tname);
        goto end;
1035 1036
      }
    }
1037 1038

    sTableData->tableMeta = pTableMeta;
X
Xiaoyu Wang 已提交
1039 1040
    uDebug("SML:0x%" PRIx64 "modify schema uid:%" PRIu64 ", sversion:%d, tversion:%d", info->id, pTableMeta->uid,
           pTableMeta->sversion, pTableMeta->tversion) tmp = (SSmlSTableMeta **)taosHashIterate(info->superTables, tmp);
wmmhello's avatar
wmmhello 已提交
1041
  }
X
Xiaoyu Wang 已提交
1042 1043
  uDebug("SML:0x%" PRIx64 " smlModifyDBSchemas end success, format:%d, needModifySchema:%d", info->id, info->dataFormat,
         info->needModifySchema);
D
dapan1121 已提交
1044

1045
  return 0;
1046

X
Xiaoyu Wang 已提交
1047
end:
1048 1049
  taosHashCleanup(hashTmp);
  taosMemoryFreeClear(pTableMeta);
D
dapan1121 已提交
1050
  catalogRefreshTableMeta(info->pCatalog, &conn, &pName, 1);
X
Xiaoyu Wang 已提交
1051 1052
  uError("SML:0x%" PRIx64 " smlModifyDBSchemas end failed:%d:%s, format:%d, needModifySchema:%d", info->id, code,
         tstrerror(code), info->dataFormat, info->needModifySchema);
D
dapan1121 已提交
1053

1054
  return code;
wmmhello's avatar
wmmhello 已提交
1055 1056
}

1057 1058 1059 1060 1061 1062 1063
/*
static int32_t smlCheckDupUnit(SHashObj *dumplicateKey, SArray *tags, SSmlMsgBuf *msg){
  for(int i = 0; i < taosArrayGetSize(tags); i++) {
    SSmlKv *tag = taosArrayGet(tags, i);
    if (smlCheckDuplicateKey(tag->key, tag->keyLen, dumplicateKey)) {
      smlBuildInvalidDataMsg(msg, "dumplicate key", tag->key);
      return TSDB_CODE_TSC_DUP_NAMES;
1064
    }
wmmhello's avatar
wmmhello 已提交
1065
  }
1066 1067 1068
  return TSDB_CODE_SUCCESS;
}

1069 1070 1071 1072 1073
static int32_t smlJudgeDupColName(SArray *cols, SArray *tags, SSmlMsgBuf *msg) {
  SHashObj *dumplicateKey = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  int ret = smlCheckDupUnit(dumplicateKey, cols, msg);
  if(ret != TSDB_CODE_SUCCESS){
    goto end;
1074
  }
1075 1076 1077
  ret = smlCheckDupUnit(dumplicateKey, tags, msg);
  if(ret != TSDB_CODE_SUCCESS){
    goto end;
1078
  }
1079

1080 1081 1082
  end:
  taosHashCleanup(dumplicateKey);
  return ret;
wmmhello's avatar
wmmhello 已提交
1083
}
1084
*/
wmmhello's avatar
wmmhello 已提交
1085

X
Xiaoyu Wang 已提交
1086
static void smlInsertMeta(SHashObj *metaHash, SArray *metaArray, SArray *cols) {
1087 1088
  for (int16_t i = 0; i < taosArrayGetSize(cols); ++i) {
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
X
Xiaoyu Wang 已提交
1089 1090
    int     ret = taosHashPut(metaHash, kv->key, kv->keyLen, &i, SHORT_BYTES);
    if (ret == 0) {
1091 1092
      taosArrayPush(metaArray, kv);
    }
1093
  }
1094
}
wmmhello's avatar
wmmhello 已提交
1095

1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111
static void smlDestroySTableMeta(SSmlSTableMeta *meta) {
  taosHashCleanup(meta->tagHash);
  taosHashCleanup(meta->colHash);
  taosArrayDestroy(meta->tags);
  taosArrayDestroy(meta->cols);
  taosMemoryFree(meta->tableMeta);
  taosMemoryFree(meta);
}

static int32_t smlUpdateMeta(SHashObj *metaHash, SArray *metaArray, SArray *cols, bool isTag, SSmlMsgBuf *msg) {
  for (int i = 0; i < taosArrayGetSize(cols); ++i) {
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);

    int16_t *index = (int16_t *)taosHashGet(metaHash, kv->key, kv->keyLen);
    if (index) {
      SSmlKv *value = (SSmlKv *)taosArrayGet(metaArray, *index);
X
Xiaoyu Wang 已提交
1112
      if (isTag) {
1113 1114
        if (kv->length > value->length) {
          value->length = kv->length;
1115
        }
1116 1117 1118 1119 1120
        continue;
      }
      if (kv->type != value->type) {
        smlBuildInvalidDataMsg(msg, "the type is not the same like before", kv->key);
        return TSDB_CODE_SML_NOT_SAME_TYPE;
1121 1122
      }

1123 1124 1125 1126 1127
      if (IS_VAR_DATA_TYPE(kv->type) && (kv->length > value->length)) {  // update string len, if bigger
        value->length = kv->length;
      }
    } else {
      size_t tmp = taosArrayGetSize(metaArray);
X
Xiaoyu Wang 已提交
1128
      if (tmp > INT16_MAX) {
D
dapan1121 已提交
1129
        smlBuildInvalidDataMsg(msg, "too many cols or tags", kv->key);
1130
        uError("too many cols or tags");
D
dapan1121 已提交
1131
        return TSDB_CODE_SML_INVALID_DATA;
1132
      }
1133
      int16_t size = tmp;
X
Xiaoyu Wang 已提交
1134 1135
      int     ret = taosHashPut(metaHash, kv->key, kv->keyLen, &size, SHORT_BYTES);
      if (ret == 0) {
1136
        taosArrayPush(metaArray, kv);
1137 1138 1139 1140
      }
    }
  }

1141 1142
  return TSDB_CODE_SUCCESS;
}
1143

1144
void smlDestroyTableInfo(SSmlHandle *info, SSmlTableInfo *tag) {
1145 1146 1147 1148
  for (size_t i = 0; i < taosArrayGetSize(tag->cols); i++) {
    SHashObj *kvHash = (SHashObj *)taosArrayGetP(tag->cols, i);
    taosHashCleanup(kvHash);
  }
1149

X
Xiaoyu Wang 已提交
1150 1151 1152 1153 1154
  //  if (info->parseJsonByLib) {
  //    SSmlLineInfo *key = (SSmlLineInfo *)(tag->key);
  //    if (key != NULL) taosMemoryFree(key->tags);
  //  }
  //  taosMemoryFree(tag->key);
1155
  taosArrayDestroy(tag->cols);
wmmhello's avatar
wmmhello 已提交
1156
  taosArrayDestroyEx(tag->tags, freeSSmlKv);
1157 1158
  taosMemoryFree(tag);
}
1159

X
Xiaoyu Wang 已提交
1160
void clearColValArray(SArray *pCols) {
wmmhello's avatar
wmmhello 已提交
1161 1162
  int32_t num = taosArrayGetSize(pCols);
  for (int32_t i = 0; i < num; ++i) {
X
Xiaoyu Wang 已提交
1163
    SColVal *pCol = taosArrayGet(pCols, i);
wmmhello's avatar
wmmhello 已提交
1164 1165 1166 1167 1168 1169
    if (TSDB_DATA_TYPE_NCHAR == pCol->type) {
      taosMemoryFreeClear(pCol->value.pData);
    }
  }
}

1170 1171 1172 1173
void freeSSmlKv(void *data) {
  SSmlKv *kv = (SSmlKv *)data;
  if (kv->keyEscaped) taosMemoryFree((void *)(kv->key));
  if (kv->valueEscaped) taosMemoryFree((void *)(kv->value));
wmmhello's avatar
wmmhello 已提交
1174 1175
}

wmmhello's avatar
wmmhello 已提交
1176
void smlDestroyInfo(SSmlHandle *info) {
1177 1178
  if (!info) return;
  qDestroyQuery(info->pQuery);
1179

1180
  // destroy info->childTables
1181 1182 1183 1184
  SSmlTableInfo **oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, NULL);
  while (oneTable) {
    smlDestroyTableInfo(info, *oneTable);
    oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, oneTable);
1185 1186
  }

1187
  // destroy info->superTables
1188 1189 1190 1191
  SSmlSTableMeta **oneSTable = (SSmlSTableMeta **)taosHashIterate(info->superTables, NULL);
  while (oneSTable) {
    smlDestroySTableMeta(*oneSTable);
    oneSTable = (SSmlSTableMeta **)taosHashIterate(info->superTables, oneSTable);
1192
  }
1193

1194 1195
  // destroy info->pVgHash
  taosHashCleanup(info->pVgHash);
1196 1197
  taosHashCleanup(info->childTables);
  taosHashCleanup(info->superTables);
1198
  taosHashCleanup(info->tableUids);
wmmhello's avatar
wmmhello 已提交
1199

X
Xiaoyu Wang 已提交
1200
  for (int i = 0; i < taosArrayGetSize(info->tagJsonArray); i++) {
wmmhello's avatar
wmmhello 已提交
1201 1202 1203 1204 1205
    cJSON *tags = (cJSON *)taosArrayGetP(info->tagJsonArray, i);
    cJSON_Delete(tags);
  }
  taosArrayDestroy(info->tagJsonArray);

1206 1207 1208 1209 1210 1211
  for (int i = 0; i < taosArrayGetSize(info->valueJsonArray); i++) {
    cJSON *value = (cJSON *)taosArrayGetP(info->valueJsonArray, i);
    cJSON_Delete(value);
  }
  taosArrayDestroy(info->valueJsonArray);

wmmhello's avatar
wmmhello 已提交
1212
  taosArrayDestroyEx(info->preLineTagKV, freeSSmlKv);
wmmhello's avatar
wmmhello 已提交
1213

X
Xiaoyu Wang 已提交
1214 1215
  if (!info->dataFormat) {
    for (int i = 0; i < info->lineNum; i++) {
wmmhello's avatar
wmmhello 已提交
1216
      taosArrayDestroyEx(info->lines[i].colArray, freeSSmlKv);
X
Xiaoyu Wang 已提交
1217
      if (info->parseJsonByLib) {
wmmhello's avatar
wmmhello 已提交
1218 1219
        taosMemoryFree(info->lines[i].tags);
      }
X
Xiaoyu Wang 已提交
1220
      if (info->lines[i].measureTagsLen != 0) taosMemoryFree(info->lines[i].measureTag);
1221 1222
    }
    taosMemoryFree(info->lines);
wmmhello's avatar
wmmhello 已提交
1223
  }
1224

wmmhello's avatar
wmmhello 已提交
1225
  cJSON_Delete(info->root);
1226 1227
  taosMemoryFreeClear(info);
}
wmmhello's avatar
wmmhello 已提交
1228

wmmhello's avatar
wmmhello 已提交
1229
SSmlHandle *smlBuildSmlInfo(TAOS *taos) {
1230 1231 1232 1233
  int32_t     code = TSDB_CODE_SUCCESS;
  SSmlHandle *info = (SSmlHandle *)taosMemoryCalloc(1, sizeof(SSmlHandle));
  if (NULL == info) {
    return NULL;
1234
  }
X
Xiaoyu Wang 已提交
1235
  if (taos != NULL) {
1236
    info->taos = acquireTscObj(*(int64_t *)taos);
X
Xiaoyu Wang 已提交
1237
    if (info->taos == NULL) {
D
dapan1121 已提交
1238 1239
      goto cleanup;
    }
1240 1241 1242 1243 1244
    code = catalogGetHandle(info->taos->pAppInfo->clusterId, &info->pCatalog);
    if (code != TSDB_CODE_SUCCESS) {
      uError("SML:0x%" PRIx64 " get catalog error %d", info->id, code);
      goto cleanup;
    }
1245
  }
wmmhello's avatar
wmmhello 已提交
1246

1247
  info->pVgHash = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_NO_LOCK);
1248
  info->childTables = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
1249
  info->tableUids = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
1250 1251
  info->superTables = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);

1252 1253 1254
  info->id = smlGenId();
  info->pQuery = smlInitHandle();
  info->dataFormat = true;
1255

wmmhello's avatar
wmmhello 已提交
1256
  info->tagJsonArray = taosArrayInit(8, POINTER_BYTES);
1257
  info->valueJsonArray = taosArrayInit(8, POINTER_BYTES);
1258 1259
  info->preLineTagKV = taosArrayInit(8, sizeof(SSmlKv));

1260
  if (NULL == info->pVgHash || NULL == info->childTables || NULL == info->superTables || NULL == info->tableUids) {
1261 1262
    uError("create SSmlHandle failed");
    goto cleanup;
1263 1264
  }

1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279
  return info;

cleanup:
  smlDestroyInfo(info);
  return NULL;
}

static int32_t smlPushCols(SArray *colsArray, SArray *cols) {
  SHashObj *kvHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  if (!kvHash) {
    uError("SML:smlDealCols failed to allocate memory");
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  for (size_t i = 0; i < taosArrayGetSize(cols); i++) {
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
1280
    terrno = 0;
1281
    taosHashPut(kvHash, kv->key, kv->keyLen, &kv, POINTER_BYTES);
X
Xiaoyu Wang 已提交
1282
    if (terrno == TSDB_CODE_DUP_KEY) {
1283
      taosHashCleanup(kvHash);
X
Xiaoyu Wang 已提交
1284 1285
      return terrno;
    }
1286 1287
  }

1288
  taosArrayPush(colsArray, &kvHash);
1289
  return TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
1290
}
1291

1292
static int32_t smlParseLineBottom(SSmlHandle *info) {
X
Xiaoyu Wang 已提交
1293 1294
  uDebug("SML:0x%" PRIx64 " smlParseLineBottom start, format:%d, linenum:%d", info->id, info->dataFormat,
         info->lineNum);
X
Xiaoyu Wang 已提交
1295
  if (info->dataFormat) return TSDB_CODE_SUCCESS;
1296

X
Xiaoyu Wang 已提交
1297 1298
  for (int32_t i = 0; i < info->lineNum; i++) {
    SSmlLineInfo  *elements = info->lines + i;
1299
    SSmlTableInfo *tinfo = NULL;
X
Xiaoyu Wang 已提交
1300
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
X
Xiaoyu Wang 已提交
1301 1302 1303
      SSmlTableInfo **tmp =
          (SSmlTableInfo **)taosHashGet(info->childTables, elements->measure, elements->measureTagsLen);
      if (tmp) tinfo = *tmp;
X
Xiaoyu Wang 已提交
1304
    } else if (info->protocol == TSDB_SML_TELNET_PROTOCOL) {
X
Xiaoyu Wang 已提交
1305
      SSmlTableInfo **tmp = (SSmlTableInfo **)taosHashGet(info->childTables, elements->measureTag,
wmmhello's avatar
wmmhello 已提交
1306
                                                          elements->measureLen + elements->tagsLen);
X
Xiaoyu Wang 已提交
1307
      if (tmp) tinfo = *tmp;
X
Xiaoyu Wang 已提交
1308
    } else {
X
Xiaoyu Wang 已提交
1309
      SSmlTableInfo **tmp = (SSmlTableInfo **)taosHashGet(info->childTables, elements->measureTag,
wmmhello's avatar
wmmhello 已提交
1310
                                                          elements->measureLen + elements->tagsLen);
X
Xiaoyu Wang 已提交
1311
      if (tmp) tinfo = *tmp;
1312 1313
    }

X
Xiaoyu Wang 已提交
1314
    if (tinfo == NULL) {
1315 1316 1317
      uError("SML:0x%" PRIx64 "get oneTable failed, line num:%d", info->id, i);
      smlBuildInvalidDataMsg(&info->msgBuf, "get oneTable failed", elements->measure);
      return TSDB_CODE_SML_INVALID_DATA;
1318
    }
wmmhello's avatar
wmmhello 已提交
1319

1320 1321 1322 1323 1324
    if (taosArrayGetSize(tinfo->tags) > TSDB_MAX_TAGS) {
      smlBuildInvalidDataMsg(&info->msgBuf, "too many tags than 128", NULL);
      return TSDB_CODE_PAR_INVALID_TAGS_NUM;
    }

1325 1326 1327
    if (taosArrayGetSize(elements->colArray) + taosArrayGetSize(tinfo->tags) > TSDB_MAX_COLUMNS) {
      smlBuildInvalidDataMsg(&info->msgBuf, "too many columns than 4096", NULL);
      return TSDB_CODE_PAR_TOO_MANY_COLUMNS;
wmmhello's avatar
wmmhello 已提交
1328
    }
wmmhello's avatar
wmmhello 已提交
1329

1330
    int ret = smlPushCols(tinfo->cols, elements->colArray);
X
Xiaoyu Wang 已提交
1331
    if (ret != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1332 1333 1334
      return ret;
    }

1335 1336
    SSmlSTableMeta **tableMeta =
        (SSmlSTableMeta **)taosHashGet(info->superTables, elements->measure, elements->measureLen);
1337
    if (tableMeta) {  // update meta
X
Xiaoyu Wang 已提交
1338 1339
      uDebug("SML:0x%" PRIx64 " smlParseLineBottom update meta, format:%d, linenum:%d", info->id, info->dataFormat,
             info->lineNum);
1340
      ret = smlUpdateMeta((*tableMeta)->colHash, (*tableMeta)->cols, elements->colArray, false, &info->msgBuf);
1341
      if (ret == TSDB_CODE_SUCCESS) {
1342
        ret = smlUpdateMeta((*tableMeta)->tagHash, (*tableMeta)->tags, tinfo->tags, true, &info->msgBuf);
1343 1344 1345 1346 1347
      }
      if (ret != TSDB_CODE_SUCCESS) {
        uError("SML:0x%" PRIx64 " smlUpdateMeta failed", info->id);
        return ret;
      }
X
Xiaoyu Wang 已提交
1348
    } else {
X
Xiaoyu Wang 已提交
1349 1350 1351 1352 1353
      //      ret = smlJudgeDupColName(elements->colArray, tinfo->tags, &info->msgBuf);
      //      if (ret != TSDB_CODE_SUCCESS) {
      //        uError("SML:0x%" PRIx64 " smlUpdateMeta failed", info->id);
      //        return ret;
      //      }
X
Xiaoyu Wang 已提交
1354 1355
      uDebug("SML:0x%" PRIx64 " smlParseLineBottom add meta, format:%d, linenum:%d", info->id, info->dataFormat,
             info->lineNum);
1356
      SSmlSTableMeta *meta = smlBuildSTableMeta(info->dataFormat);
1357
      taosHashPut(info->superTables, elements->measure, elements->measureLen, &meta, POINTER_BYTES);
1358
      terrno = 0;
1359
      smlInsertMeta(meta->tagHash, meta->tags, tinfo->tags);
X
Xiaoyu Wang 已提交
1360 1361 1362
      if (terrno == TSDB_CODE_DUP_KEY) {
        return terrno;
      }
1363
      smlInsertMeta(meta->colHash, meta->cols, elements->colArray);
wmmhello's avatar
wmmhello 已提交
1364
    }
wmmhello's avatar
wmmhello 已提交
1365
  }
D
dapan1121 已提交
1366
  uDebug("SML:0x%" PRIx64 " smlParseLineBottom end, format:%d, linenum:%d", info->id, info->dataFormat, info->lineNum);
1367

wmmhello's avatar
wmmhello 已提交
1368 1369 1370
  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1371
static int32_t smlInsertData(SSmlHandle *info) {
wmmhello's avatar
wmmhello 已提交
1372
  int32_t code = TSDB_CODE_SUCCESS;
D
dapan1121 已提交
1373
  uDebug("SML:0x%" PRIx64 " smlInsertData start, format:%d", info->id, info->dataFormat);
wmmhello's avatar
wmmhello 已提交
1374

X
Xiaoyu Wang 已提交
1375
  if (info->pRequest->dbList == NULL) {
wmmhello's avatar
wmmhello 已提交
1376 1377
    info->pRequest->dbList = taosArrayInit(1, TSDB_DB_FNAME_LEN);
  }
1378
  char *data = (char *)taosArrayReserve(info->pRequest->dbList, 1);
1379 1380 1381
  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));
  tNameGetFullDbName(&pName, data);
wmmhello's avatar
wmmhello 已提交
1382

1383 1384 1385
  SSmlTableInfo **oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, NULL);
  while (oneTable) {
    SSmlTableInfo *tableData = *oneTable;
1386
    tstrncpy(pName.tname, tableData->sTableName, tableData->sTableNameLen + 1);
D
dapan1121 已提交
1387

X
Xiaoyu Wang 已提交
1388
    if (info->pRequest->tableList == NULL) {
wmmhello's avatar
wmmhello 已提交
1389 1390 1391 1392
      info->pRequest->tableList = taosArrayInit(1, sizeof(SName));
    }
    taosArrayPush(info->pRequest->tableList, &pName);

wmmhello's avatar
wmmhello 已提交
1393
    strcpy(pName.tname, tableData->childTableName);
1394

D
dapan1121 已提交
1395 1396 1397 1398 1399
    SRequestConnInfo conn = {0};
    conn.pTrans = info->taos->pAppInfo->pTransporter;
    conn.requestId = info->pRequest->requestId;
    conn.requestObjRefId = info->pRequest->self;
    conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
X
Xiaoyu Wang 已提交
1400

wmmhello's avatar
wmmhello 已提交
1401 1402 1403 1404 1405
    code = smlCheckAuth(info, &conn, pName.tname, AUTH_TYPE_WRITE);
    if(code != TSDB_CODE_SUCCESS){
      return code;
    }

wmmhello's avatar
wmmhello 已提交
1406
    SVgroupInfo vg;
D
dapan1121 已提交
1407
    code = catalogGetTableHashVgroup(info->pCatalog, &conn, &pName, &vg);
1408
    if (code != TSDB_CODE_SUCCESS) {
X
Xiaoyu Wang 已提交
1409
      uError("SML:0x%" PRIx64 " catalogGetTableHashVgroup failed. table name: %s", info->id, tableData->childTableName);
wmmhello's avatar
wmmhello 已提交
1410 1411
      return code;
    }
X
Xiaoyu Wang 已提交
1412
    taosHashPut(info->pVgHash, (const char *)&vg.vgId, sizeof(vg.vgId), (char *)&vg, sizeof(vg));
wmmhello's avatar
wmmhello 已提交
1413

1414 1415 1416
    SSmlSTableMeta **pMeta =
        (SSmlSTableMeta **)taosHashGet(info->superTables, tableData->sTableName, tableData->sTableNameLen);
    if (unlikely(NULL == pMeta || NULL == (*pMeta)->tableMeta)) {
1417 1418 1419
      uError("SML:0x%" PRIx64 " NULL == pMeta. table name: %s", info->id, tableData->childTableName);
      return TSDB_CODE_SML_INTERNAL_ERROR;
    }
wmmhello's avatar
wmmhello 已提交
1420

1421
    // use tablemeta of stable to save vgid and uid of child table
1422 1423
    (*pMeta)->tableMeta->vgId = vg.vgId;
    (*pMeta)->tableMeta->uid = tableData->uid;  // one table merge data block together according uid
X
Xiaoyu Wang 已提交
1424 1425
    uDebug("SML:0x%" PRIx64 " smlInsertData table:%s, uid:%" PRIu64 ", format:%d", info->id, pName.tname,
           tableData->uid, info->dataFormat);
wmmhello's avatar
wmmhello 已提交
1426

1427
    int   measureLen = tableData->sTableNameLen;
1428
    char *measure = (char *)taosMemoryMalloc(tableData->sTableNameLen);
1429 1430 1431
    memcpy(measure, tableData->sTableName, tableData->sTableNameLen);
    PROCESS_SLASH_IN_MEASUREMENT(measure, measureLen);

X
Xiaoyu Wang 已提交
1432
    code = smlBindData(info->pQuery, info->dataFormat, tableData->tags, (*pMeta)->cols, tableData->cols,
1433 1434
                       (*pMeta)->tableMeta, tableData->childTableName, measure, measureLen, info->ttl, info->msgBuf.buf,
                       info->msgBuf.len);
1435
    taosMemoryFree(measure);
X
Xiaoyu Wang 已提交
1436 1437
    if (code != TSDB_CODE_SUCCESS) {
      uError("SML:0x%" PRIx64 " smlBindData failed", info->id);
wmmhello's avatar
wmmhello 已提交
1438 1439
      return code;
    }
1440
    oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, oneTable);
wmmhello's avatar
wmmhello 已提交
1441
  }
wmmhello's avatar
wmmhello 已提交
1442

wmmhello's avatar
wmmhello 已提交
1443
  code = smlBuildOutput(info->pQuery, info->pVgHash);
1444
  if (code != TSDB_CODE_SUCCESS) {
X
Xiaoyu Wang 已提交
1445
    uError("SML:0x%" PRIx64 " smlBuildOutput failed", info->id);
1446 1447
    return code;
  }
1448 1449
  info->cost.insertRpcTime = taosGetTimestampUs();

1450 1451 1452
  SAppClusterSummary *pActivity = &info->taos->pAppInfo->summary;
  atomic_add_fetch_64((int64_t *)&pActivity->numOfInsertsReq, 1);

wmmhello's avatar
wmmhello 已提交
1453
  launchQueryImpl(info->pRequest, info->pQuery, true, NULL);
X
Xiaoyu Wang 已提交
1454 1455
  uDebug("SML:0x%" PRIx64 " smlInsertData end, format:%d, code:%d,%s", info->id, info->dataFormat, info->pRequest->code,
         tstrerror(info->pRequest->code));
D
dapan1121 已提交
1456

wmmhello's avatar
wmmhello 已提交
1457
  return info->pRequest->code;
wmmhello's avatar
wmmhello 已提交
1458 1459
}

X
Xiaoyu Wang 已提交
1460
static void smlPrintStatisticInfo(SSmlHandle *info) {
1461
  uDebug(
X
Xiaoyu Wang 已提交
1462
      "SML:0x%" PRIx64
D
dapan1121 已提交
1463
      " smlInsertLines result, code:%d, msg:%s, lineNum:%d,stable num:%d,ctable num:%d,create stable num:%d,alter stable tag num:%d,alter stable col num:%d \
X
Xiaoyu Wang 已提交
1464
        parse cost:%" PRId64 ",schema cost:%" PRId64 ",bind cost:%" PRId64 ",rpc cost:%" PRId64 ",total cost:%" PRId64
X
Xiaoyu Wang 已提交
1465
      "",
X
Xiaoyu Wang 已提交
1466 1467 1468 1469 1470
      info->id, info->cost.code, tstrerror(info->cost.code), info->cost.lineNum, info->cost.numOfSTables,
      info->cost.numOfCTables, info->cost.numOfCreateSTables, info->cost.numOfAlterTagSTables,
      info->cost.numOfAlterColSTables, info->cost.schemaTime - info->cost.parseTime,
      info->cost.insertBindTime - info->cost.schemaTime, info->cost.insertRpcTime - info->cost.insertBindTime,
      info->cost.endTime - info->cost.insertRpcTime, info->cost.endTime - info->cost.parseTime);
1471 1472
}

X
Xiaoyu Wang 已提交
1473
int32_t smlClearForRerun(SSmlHandle *info) {
1474 1475
  info->reRun = false;
  // clear info->childTables
1476 1477 1478 1479
  SSmlTableInfo **oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, NULL);
  while (oneTable) {
    smlDestroyTableInfo(info, *oneTable);
    oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, oneTable);
1480 1481 1482
  }

  // clear info->superTables
1483 1484 1485 1486
  SSmlSTableMeta **oneSTable = (SSmlSTableMeta **)taosHashIterate(info->superTables, NULL);
  while (oneSTable) {
    smlDestroySTableMeta(*oneSTable);
    oneSTable = (SSmlSTableMeta **)taosHashIterate(info->superTables, oneSTable);
1487 1488
  }

wmmhello's avatar
wmmhello 已提交
1489 1490
  taosHashClear(info->childTables);
  taosHashClear(info->superTables);
1491
  taosHashClear(info->tableUids);
wmmhello's avatar
wmmhello 已提交
1492

X
Xiaoyu Wang 已提交
1493
  if (!info->dataFormat) {
wmmhello's avatar
wmmhello 已提交
1494 1495 1496 1497 1498
    if (unlikely(info->lines != NULL)) {
      uError("SML:0x%" PRIx64 " info->lines != NULL", info->id);
      return TSDB_CODE_SML_INVALID_DATA;
    }
    info->lines = (SSmlLineInfo *)taosMemoryCalloc(info->lineNum, sizeof(SSmlLineInfo));
1499 1500 1501 1502 1503 1504
  }

  memset(&info->preLine, 0, sizeof(SSmlLineInfo));
  info->currSTableMeta = NULL;
  info->currTableDataCtx = NULL;

X
Xiaoyu Wang 已提交
1505
  SVnodeModifyOpStmt *stmt = (SVnodeModifyOpStmt *)(info->pQuery->pRoot);
1506 1507 1508 1509 1510
  stmt->freeHashFunc(stmt->pTableBlockHashObj);
  stmt->pTableBlockHashObj = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
1511
static int32_t smlParseLine(SSmlHandle *info, char *lines[], char *rawLine, char *rawLineEnd, int numLines) {
D
dapan1121 已提交
1512
  uDebug("SML:0x%" PRIx64 " smlParseLine start", info->id);
wmmhello's avatar
wmmhello 已提交
1513
  int32_t code = TSDB_CODE_SUCCESS;
1514
  if (info->protocol == TSDB_SML_JSON_PROTOCOL) {
dengyihao's avatar
dengyihao 已提交
1515
    if (lines) {
wmmhello's avatar
wmmhello 已提交
1516
      code = smlParseJSON(info, *lines);
dengyihao's avatar
dengyihao 已提交
1517
    } else if (rawLine) {
wmmhello's avatar
wmmhello 已提交
1518 1519
      code = smlParseJSON(info, rawLine);
    }
1520
    if (code != TSDB_CODE_SUCCESS) {
dengyihao's avatar
dengyihao 已提交
1521
      uError("SML:0x%" PRIx64 " smlParseJSON failed:%s", info->id, lines ? *lines : rawLine);
1522 1523
      return code;
    }
wmmhello's avatar
wmmhello 已提交
1524
    return code;
wmmhello's avatar
wmmhello 已提交
1525
  }
wmmhello's avatar
wmmhello 已提交
1526

X
Xiaoyu Wang 已提交
1527
  char   *oldRaw = rawLine;
1528 1529
  int32_t i = 0;
  while (i < numLines) {
wmmhello's avatar
wmmhello 已提交
1530
    char *tmp = NULL;
dengyihao's avatar
dengyihao 已提交
1531 1532
    int   len = 0;
    if (lines) {
wmmhello's avatar
wmmhello 已提交
1533 1534
      tmp = lines[i];
      len = strlen(tmp);
dengyihao's avatar
dengyihao 已提交
1535
    } else if (rawLine) {
wmmhello's avatar
wmmhello 已提交
1536
      tmp = rawLine;
dengyihao's avatar
dengyihao 已提交
1537 1538
      while (rawLine < rawLineEnd) {
        if (*(rawLine++) == '\n') {
wmmhello's avatar
wmmhello 已提交
1539 1540 1541 1542
          break;
        }
        len++;
      }
dengyihao's avatar
dengyihao 已提交
1543
      if (info->protocol == TSDB_SML_LINE_PROTOCOL && tmp[0] == '#') {  // this line is comment
wmmhello's avatar
wmmhello 已提交
1544 1545
        continue;
      }
wmmhello's avatar
wmmhello 已提交
1546 1547
    }

X
Xiaoyu Wang 已提交
1548 1549
    char cTmp = 0;  // for print tmp if is raw
    if (info->isRawLine) {
wmmhello's avatar
wmmhello 已提交
1550 1551
      cTmp = tmp[len];
      tmp[len] = '\0';
D
dapan1121 已提交
1552 1553
    }

X
Xiaoyu Wang 已提交
1554 1555 1556
    uDebug("SML:0x%" PRIx64 " smlParseLine israw:%d, numLines:%d, protocol:%d, len:%d, sql:%s", info->id,
           info->isRawLine, numLines, info->protocol, len, tmp);
    if (info->isRawLine) {
wmmhello's avatar
wmmhello 已提交
1557
      tmp[len] = cTmp;
D
dapan1121 已提交
1558
    }
1559

X
Xiaoyu Wang 已提交
1560
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
X
Xiaoyu Wang 已提交
1561
      if (info->dataFormat) {
wmmhello's avatar
wmmhello 已提交
1562 1563
        SSmlLineInfo element = {0};
        code = smlParseInfluxString(info, tmp, tmp + len, &element);
X
Xiaoyu Wang 已提交
1564
      } else {
wmmhello's avatar
wmmhello 已提交
1565 1566
        code = smlParseInfluxString(info, tmp, tmp + len, info->lines + i);
      }
X
Xiaoyu Wang 已提交
1567
    } else if (info->protocol == TSDB_SML_TELNET_PROTOCOL) {
X
Xiaoyu Wang 已提交
1568
      if (info->dataFormat) {
1569 1570
        SSmlLineInfo element = {0};
        code = smlParseTelnetString(info, (char *)tmp, (char *)tmp + len, &element);
X
Xiaoyu Wang 已提交
1571
        if (element.measureTagsLen != 0) taosMemoryFree(element.measureTag);
X
Xiaoyu Wang 已提交
1572
      } else {
1573 1574
        code = smlParseTelnetString(info, (char *)tmp, (char *)tmp + len, info->lines + i);
      }
X
Xiaoyu Wang 已提交
1575
    } else {
1576
      code = TSDB_CODE_SML_INVALID_PROTOCOL_TYPE;
1577
    }
wmmhello's avatar
wmmhello 已提交
1578
    if (code != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
1579
      tmp[len] = '\0';
wmmhello's avatar
wmmhello 已提交
1580
      uError("SML:0x%" PRIx64 " smlParseLine failed. line %d : %s", info->id, i, tmp);
1581
      return code;
wmmhello's avatar
wmmhello 已提交
1582
    }
X
Xiaoyu Wang 已提交
1583
    if (info->reRun) {
D
dapan1121 已提交
1584
      uDebug("SML:0x%" PRIx64 " smlParseLine re run", info->id);
1585
      i = 0;
1586
      rawLine = oldRaw;
1587
      code = smlClearForRerun(info);
X
Xiaoyu Wang 已提交
1588
      if (code != TSDB_CODE_SUCCESS) {
1589
        return code;
1590
      }
1591
      continue;
1592
    }
1593
    i++;
wmmhello's avatar
wmmhello 已提交
1594
  }
D
dapan1121 已提交
1595
  uDebug("SML:0x%" PRIx64 " smlParseLine end", info->id);
1596

1597 1598 1599
  return code;
}

dengyihao's avatar
dengyihao 已提交
1600
static int smlProcess(SSmlHandle *info, char *lines[], char *rawLine, char *rawLineEnd, int numLines) {
1601
  int32_t code = TSDB_CODE_SUCCESS;
1602 1603
  int32_t retryNum = 0;

1604 1605
  info->cost.parseTime = taosGetTimestampUs();

wmmhello's avatar
wmmhello 已提交
1606
  code = smlParseLine(info, lines, rawLine, rawLineEnd, numLines);
1607
  if (code != 0) {
X
Xiaoyu Wang 已提交
1608
    uError("SML:0x%" PRIx64 " smlParseLine error : %s", info->id, tstrerror(code));
wmmhello's avatar
wmmhello 已提交
1609
    return code;
1610
  }
1611 1612 1613 1614 1615 1616
  code = smlParseLineBottom(info);
  if (code != 0) {
    uError("SML:0x%" PRIx64 " smlParseLineBottom error : %s", info->id, tstrerror(code));
    return code;
  }

1617
  info->cost.lineNum = info->lineNum;
1618 1619
  info->cost.numOfSTables = taosHashGetSize(info->superTables);
  info->cost.numOfCTables = taosHashGetSize(info->childTables);
1620 1621

  info->cost.schemaTime = taosGetTimestampUs();
1622

X
Xiaoyu Wang 已提交
1623
  do {
1624
    code = smlModifyDBSchemas(info);
wmmhello's avatar
wmmhello 已提交
1625
    if (code != TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER && code != TSDB_CODE_SDB_OBJ_CREATING && code != TSDB_CODE_MND_TRANS_CONFLICT) {
1626 1627
      break;
    }
1628
    taosMsleep(100);
D
dapan1121 已提交
1629
    uInfo("SML:0x%" PRIx64 " smlModifyDBSchemas retry code:%s, times:%d", info->id, tstrerror(code), retryNum);
1630
  } while (retryNum++ < taosHashGetSize(info->superTables) * MAX_RETRY_TIMES);
1631

wmmhello's avatar
wmmhello 已提交
1632
  if (code != 0) {
X
Xiaoyu Wang 已提交
1633
    uError("SML:0x%" PRIx64 " smlModifyDBSchemas error : %s", info->id, tstrerror(code));
wmmhello's avatar
wmmhello 已提交
1634
    return code;
wmmhello's avatar
wmmhello 已提交
1635
  }
wmmhello's avatar
wmmhello 已提交
1636

1637
  info->cost.insertBindTime = taosGetTimestampUs();
wmmhello's avatar
wmmhello 已提交
1638 1639
  code = smlInsertData(info);
  if (code != 0) {
X
Xiaoyu Wang 已提交
1640
    uError("SML:0x%" PRIx64 " smlInsertData error : %s", info->id, tstrerror(code));
wmmhello's avatar
wmmhello 已提交
1641
    return code;
wmmhello's avatar
wmmhello 已提交
1642 1643 1644 1645 1646
  }

  return code;
}

1647 1648 1649 1650 1651
void smlSetReqSQL(SRequestObj *request, char *lines[], char *rawLine, char *rawLineEnd) {
  if (tsSlowLogScope & SLOW_LOG_TYPE_INSERT) {
    int32_t len = 0;
    int32_t rlen = 0;
    char* p = NULL;
1652

1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667
    if (lines && lines[0]) {
      len = strlen(lines[0]);
      p = lines[0];
    } else if (rawLine) {
      if (rawLineEnd) {
        len = rawLineEnd - rawLine;
      } else {
        len = strlen(rawLine);
      }
      p = rawLine;
    }

    if (NULL == p) {
      return;
    }
1668

1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684
    rlen = TMIN(len, TSDB_MAX_ALLOWED_SQL_LEN);
    rlen = TMAX(rlen, 0);

    char *sql = taosMemoryMalloc(rlen + 1);
    if (NULL == sql) {
      uError("malloc %d for sml sql failed", rlen + 1);
      return;
    }
    memcpy(sql, p, rlen);
    sql[rlen] = 0;

    request->sqlstr = sql;
    request->sqlLen = rlen;
  }
}

X
Xiaoyu Wang 已提交
1685 1686
TAOS_RES *taos_schemaless_insert_inner(TAOS *taos, char *lines[], char *rawLine, char *rawLineEnd, int numLines,
                                       int protocol, int precision, int32_t ttl, int64_t reqid) {
1687
  int32_t code = TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
1688 1689 1690
  if (NULL == taos) {
    terrno = TSDB_CODE_TSC_DISCONNECTED;
    return NULL;
1691
  }
wmmhello's avatar
wmmhello 已提交
1692
  SRequestObj *request = NULL;
X
Xiaoyu Wang 已提交
1693 1694 1695
  SSmlHandle  *info = NULL;
  int          cnt = 0;
  while (1) {
wmmhello's avatar
wmmhello 已提交
1696 1697 1698 1699 1700
    request = (SRequestObj *)createRequest(*(int64_t *)taos, TSDB_SQL_INSERT, reqid);
    if (request == NULL) {
      uError("SML:taos_schemaless_insert error request is null");
      return NULL;
    }
1701

wmmhello's avatar
wmmhello 已提交
1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
    info = smlBuildSmlInfo(taos);
    if (info == NULL) {
      request->code = TSDB_CODE_OUT_OF_MEMORY;
      uError("SML:taos_schemaless_insert error SSmlHandle is null");
      return (TAOS_RES *)request;
    }
    info->pRequest = request;
    info->isRawLine = rawLine != NULL;
    info->ttl = ttl;
    info->precision = precision;
    info->protocol = (TSDB_SML_PROTOCOL_TYPE)protocol;
    info->msgBuf.buf = info->pRequest->msgBuf;
    info->msgBuf.len = ERROR_MSG_BUF_DEFAULT_SIZE;
    info->lineNum = numLines;

1717 1718
    smlSetReqSQL(request, lines, rawLine, rawLineEnd);

wmmhello's avatar
wmmhello 已提交
1719 1720 1721 1722 1723 1724
    SSmlMsgBuf msg = {ERROR_MSG_BUF_DEFAULT_SIZE, request->msgBuf};
    if (request->pDb == NULL) {
      request->code = TSDB_CODE_PAR_DB_NOT_SPECIFIED;
      smlBuildInvalidDataMsg(&msg, "Database not specified", NULL);
      goto end;
    }
1725

wmmhello's avatar
wmmhello 已提交
1726 1727 1728 1729 1730
    if (protocol < TSDB_SML_LINE_PROTOCOL || protocol > TSDB_SML_JSON_PROTOCOL) {
      request->code = TSDB_CODE_SML_INVALID_PROTOCOL_TYPE;
      smlBuildInvalidDataMsg(&msg, "protocol invalidate", NULL);
      goto end;
    }
wmmhello's avatar
wmmhello 已提交
1731

wmmhello's avatar
wmmhello 已提交
1732 1733 1734 1735 1736 1737
    if (protocol == TSDB_SML_LINE_PROTOCOL &&
        (precision < TSDB_SML_TIMESTAMP_NOT_CONFIGURED || precision > TSDB_SML_TIMESTAMP_NANO_SECONDS)) {
      request->code = TSDB_CODE_SML_INVALID_PRECISION_TYPE;
      smlBuildInvalidDataMsg(&msg, "precision invalidate for line protocol", NULL);
      goto end;
    }
wmmhello's avatar
wmmhello 已提交
1738

wmmhello's avatar
wmmhello 已提交
1739 1740 1741 1742 1743 1744 1745
    if (protocol == TSDB_SML_JSON_PROTOCOL) {
      numLines = 1;
    } else if (numLines <= 0) {
      request->code = TSDB_CODE_SML_INVALID_DATA;
      smlBuildInvalidDataMsg(&msg, "line num is invalid", NULL);
      goto end;
    }
1746

wmmhello's avatar
wmmhello 已提交
1747 1748 1749 1750
    code = smlProcess(info, lines, rawLine, rawLineEnd, numLines);
    request->code = code;
    info->cost.endTime = taosGetTimestampUs();
    info->cost.code = code;
X
Xiaoyu Wang 已提交
1751
    if (code == TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER || code == TSDB_CODE_SDB_OBJ_CREATING ||
1752 1753
        code == TSDB_CODE_PAR_VALUE_TOO_LONG || code == TSDB_CODE_MND_TRANS_CONFLICT ||
        code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
X
Xiaoyu Wang 已提交
1754 1755
      if (cnt++ >= 10) {
        uInfo("SML:%" PRIx64 " retry:%d/10 end code:%d, msg:%s", info->id, cnt, code, tstrerror(code));
D
dapan1121 已提交
1756 1757 1758
        break;
      }
      taosMsleep(100);
wmmhello's avatar
wmmhello 已提交
1759
      refreshMeta(request->pTscObj, request);
X
Xiaoyu Wang 已提交
1760 1761
      uInfo("SML:%" PRIx64 " retry:%d/10,ver is old retry or object is creating code:%d, msg:%s", info->id, cnt, code,
            tstrerror(code));
wmmhello's avatar
wmmhello 已提交
1762 1763 1764 1765 1766 1767
      smlDestroyInfo(info);
      info = NULL;
      taos_free_result(request);
      request = NULL;
      continue;
    }
D
dapan1121 已提交
1768
    smlPrintStatisticInfo(info);
wmmhello's avatar
wmmhello 已提交
1769
    break;
wmmhello's avatar
wmmhello 已提交
1770 1771
  }

wmmhello's avatar
wmmhello 已提交
1772
end:
wmmhello's avatar
wmmhello 已提交
1773
  smlDestroyInfo(info);
1774
  return (TAOS_RES *)request;
wmmhello's avatar
wmmhello 已提交
1775
}
wmmhello's avatar
wmmhello 已提交
1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792

/**
 * taos_schemaless_insert() parse and insert data points into database according to
 * different protocol.
 *
 * @param $lines input array may contain multiple lines, each line indicates a data point.
 *               If protocol=2 is used input array should contain single JSON
 *               string(e.g. char *lines[] = {"$JSON_string"}). If need to insert
 *               multiple data points in JSON format, should include them in $JSON_string
 *               as a JSON array.
 * @param $numLines indicates how many data points in $lines.
 *                  If protocol = 2 is used this param will be ignored as $lines should
 *                  contain single JSON string.
 * @param $protocol indicates which protocol to use for parsing:
 *                  0 - influxDB line protocol
 *                  1 - OpenTSDB telnet line protocol
 *                  2 - OpenTSDB JSON format protocol
dengyihao's avatar
dengyihao 已提交
1793
 * @return TAOS_RES
wmmhello's avatar
wmmhello 已提交
1794 1795
 */

1796 1797
TAOS_RES *taos_schemaless_insert_ttl_with_reqid(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
                                                int32_t ttl, int64_t reqid) {
wmmhello's avatar
wmmhello 已提交
1798
  return taos_schemaless_insert_inner(taos, lines, NULL, NULL, numLines, protocol, precision, ttl, reqid);
dengyihao's avatar
dengyihao 已提交
1799 1800
}

1801 1802 1803
TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision) {
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, TSDB_DEFAULT_TABLE_TTL, 0);
}
wmmhello's avatar
wmmhello 已提交
1804

X
Xiaoyu Wang 已提交
1805 1806
TAOS_RES *taos_schemaless_insert_ttl(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
                                     int32_t ttl) {
1807 1808
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, ttl, 0);
}
wmmhello's avatar
wmmhello 已提交
1809

X
Xiaoyu Wang 已提交
1810 1811 1812 1813
TAOS_RES *taos_schemaless_insert_with_reqid(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
                                            int64_t reqid) {
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, TSDB_DEFAULT_TABLE_TTL,
                                               reqid);
wmmhello's avatar
wmmhello 已提交
1814 1815
}

1816
TAOS_RES *taos_schemaless_insert_raw_ttl_with_reqid(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
1817
                                                    int precision, int32_t ttl, int64_t reqid) {
dengyihao's avatar
dengyihao 已提交
1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829
  int numLines = 0;
  *totalRows = 0;
  char *tmp = lines;
  for (int i = 0; i < len; i++) {
    if (lines[i] == '\n' || i == len - 1) {
      numLines++;
      if (tmp[0] != '#' || protocol != TSDB_SML_LINE_PROTOCOL) {  // ignore comment
        (*totalRows)++;
      }
      tmp = lines + i + 1;
    }
  }
wmmhello's avatar
wmmhello 已提交
1830
  return taos_schemaless_insert_inner(taos, NULL, lines, lines + len, *totalRows, protocol, precision, ttl, reqid);
dengyihao's avatar
dengyihao 已提交
1831 1832
}

X
Xiaoyu Wang 已提交
1833 1834 1835 1836
TAOS_RES *taos_schemaless_insert_raw_with_reqid(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
                                                int precision, int64_t reqid) {
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision,
                                                   TSDB_DEFAULT_TABLE_TTL, reqid);
1837
}
X
Xiaoyu Wang 已提交
1838 1839
TAOS_RES *taos_schemaless_insert_raw_ttl(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
                                         int precision, int32_t ttl) {
1840 1841
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision, ttl, 0);
}
wmmhello's avatar
wmmhello 已提交
1842

X
Xiaoyu Wang 已提交
1843 1844 1845 1846
TAOS_RES *taos_schemaless_insert_raw(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
                                     int precision) {
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision,
                                                   TSDB_DEFAULT_TABLE_TTL, 0);
wmmhello's avatar
wmmhello 已提交
1847
}