• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #4869

26 Nov 2025 05:46AM UTC coverage: 64.539% (-0.09%) from 64.629%
#4869

push

travis-ci

guanshengliang
Merge branch '3.0' into cover/3.0

771 of 945 new or added lines in 33 files covered. (81.59%)

3214 existing lines in 124 files now uncovered.

158203 of 245129 relevant lines covered (64.54%)

113224023.06 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

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

16
#include <ctype.h>
17
#include <stdio.h>
18
#include <stdlib.h>
19
#include <string.h>
20

21
#include "clientSml.h"
22

23
#define RETURN_FALSE                                 \
24
  smlBuildInvalidDataMsg(msg, "invalid data", pVal); \
25
  return false;
26

27
#define SET_DOUBLE                     \
28
  kvVal->type = TSDB_DATA_TYPE_DOUBLE; \
29
  kvVal->d = result;
30

31
#define SET_FLOAT                                                                              \
32
  if (!IS_VALID_FLOAT(result)) {                                                               \
33
    smlBuildInvalidDataMsg(msg, "float out of range[-3.402823466e+38,3.402823466e+38]", pVal); \
34
    return false;                                                                              \
35
  }                                                                                            \
36
  kvVal->type = TSDB_DATA_TYPE_FLOAT;                                                          \
37
  kvVal->f = (float)result;
38

39
#define SET_BIGINT                                                                                       \
40
  SET_ERRNO(0);                                                                                          \
41
  int64_t tmp = taosStr2Int64(pVal, &endptr, 10);                                                        \
42
  if (ERRNO == ERANGE) {                                                                                 \
43
    smlBuildInvalidDataMsg(msg, "big int out of range[-9223372036854775808,9223372036854775807]", pVal); \
44
    return false;                                                                                        \
45
  }                                                                                                      \
46
  kvVal->type = TSDB_DATA_TYPE_BIGINT;                                                                   \
47
  kvVal->i = tmp;
48

49
#define SET_INT                                                                    \
50
  if (!IS_VALID_INT(result)) {                                                     \
51
    smlBuildInvalidDataMsg(msg, "int out of range[-2147483648,2147483647]", pVal); \
52
    return false;                                                                  \
53
  }                                                                                \
54
  kvVal->type = TSDB_DATA_TYPE_INT;                                                \
55
  kvVal->i = result;
56

57
#define SET_SMALL_INT                                                          \
58
  if (!IS_VALID_SMALLINT(result)) {                                            \
59
    smlBuildInvalidDataMsg(msg, "small int our of range[-32768,32767]", pVal); \
60
    return false;                                                              \
61
  }                                                                            \
62
  kvVal->type = TSDB_DATA_TYPE_SMALLINT;                                       \
63
  kvVal->i = result;
64

65
#define SET_UBIGINT                                                                             \
66
  SET_ERRNO(0);                                                                                 \
67
  uint64_t tmp = taosStr2UInt64(pVal, &endptr, 10);                                             \
68
  if (ERRNO == ERANGE || result < 0) {                                                          \
69
    smlBuildInvalidDataMsg(msg, "unsigned big int out of range[0,18446744073709551615]", pVal); \
70
    return false;                                                                               \
71
  }                                                                                             \
72
  kvVal->type = TSDB_DATA_TYPE_UBIGINT;                                                         \
73
  kvVal->u = tmp;
74

75
#define SET_UINT                                                                  \
76
  if (!IS_VALID_UINT(result)) {                                                   \
77
    smlBuildInvalidDataMsg(msg, "unsigned int out of range[0,4294967295]", pVal); \
78
    return false;                                                                 \
79
  }                                                                               \
80
  kvVal->type = TSDB_DATA_TYPE_UINT;                                              \
81
  kvVal->u = result;
82

83
#define SET_USMALL_INT                                                            \
84
  if (!IS_VALID_USMALLINT(result)) {                                              \
85
    smlBuildInvalidDataMsg(msg, "unsigned small int out of rang[0,65535]", pVal); \
86
    return false;                                                                 \
87
  }                                                                               \
88
  kvVal->type = TSDB_DATA_TYPE_USMALLINT;                                         \
89
  kvVal->u = result;
90

91
#define SET_TINYINT                                                       \
92
  if (!IS_VALID_TINYINT(result)) {                                        \
93
    smlBuildInvalidDataMsg(msg, "tiny int out of range[-128,127]", pVal); \
94
    return false;                                                         \
95
  }                                                                       \
96
  kvVal->type = TSDB_DATA_TYPE_TINYINT;                                   \
97
  kvVal->i = result;
98

99
#define SET_UTINYINT                                                            \
100
  if (!IS_VALID_UTINYINT(result)) {                                             \
101
    smlBuildInvalidDataMsg(msg, "unsigned tiny int out of range[0,255]", pVal); \
102
    return false;                                                               \
103
  }                                                                             \
104
  kvVal->type = TSDB_DATA_TYPE_UTINYINT;                                        \
105
  kvVal->u = result;
106

107
#define IS_COMMENT(protocol, data) (protocol == TSDB_SML_LINE_PROTOCOL && data == '#')
108

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

113
static int32_t smlCheckAuth(SSmlHandle *info, SRequestConnInfo *conn, const char *pTabName, AUTH_TYPE type) {
1,655,329✔
114
  SUserAuthInfo pAuth = {0};
1,655,329✔
115
  (void)snprintf(pAuth.user, sizeof(pAuth.user), "%s", info->taos->user);
1,655,329✔
116
  if (NULL == pTabName) {
1,655,057✔
117
    if (tNameSetDbName(&pAuth.tbName, info->taos->acctId, info->pRequest->pDb, strlen(info->pRequest->pDb)) != 0) {
372,983✔
118
      return TSDB_CODE_SML_INVALID_DATA;
×
119
    }
120
  } else {
121
    toName(info->taos->acctId, info->pRequest->pDb, pTabName, &pAuth.tbName);
1,282,074✔
122
  }
123
  pAuth.type = type;
1,655,011✔
124

125
  int32_t      code = TSDB_CODE_SUCCESS;
1,655,011✔
126
  SUserAuthRes authRes = {0};
1,655,011✔
127

128
  code = catalogChkAuth(info->pCatalog, conn, &pAuth, &authRes);
1,655,011✔
129
  nodesDestroyNode(authRes.pCond[AUTH_RES_BASIC]);
1,655,057✔
130

131
  return (code == TSDB_CODE_SUCCESS)
132
             ? (authRes.pass[AUTH_RES_BASIC] ? TSDB_CODE_SUCCESS : TSDB_CODE_PAR_PERMISSION_DENIED)
1,655,329✔
133
             : code;
3,013,013✔
134
}
135

136
void smlBuildInvalidDataMsg(SSmlMsgBuf *pBuf, const char *msg1, const char *msg2) {
70,737✔
137
  if (pBuf->buf == NULL) {
70,737✔
138
    return;
×
139
  }
140
  pBuf->buf[0] = 0;
70,737✔
141
  if (msg1) {
70,737✔
142
    (void)strncat(pBuf->buf, msg1, pBuf->len - 1);
70,737✔
143
  }
144
  int32_t left = pBuf->len - strlen(pBuf->buf);
70,737✔
145
  if (left > 2 && msg2) {
70,737✔
146
    (void)strncat(pBuf->buf, ":", left - 1);
63,829✔
147
    (void)strncat(pBuf->buf, msg2, left - 2);
63,829✔
148
  }
149
}
150

151
int64_t smlGetTimeValue(const char *value, int32_t len, uint8_t fromPrecision, uint8_t toPrecision) {
50,587,049✔
152
  char   *endPtr = NULL;
50,587,049✔
153
  int64_t tsInt64 = taosStr2Int64(value, &endPtr, 10);
50,809,155✔
154
  if (unlikely(value + len != endPtr)) {
51,034,902✔
155
    return -1;
2,512✔
156
  }
157

158
  if (unlikely(fromPrecision >= TSDB_TIME_PRECISION_HOURS)) {
51,055,793✔
159
    int64_t unit = smlToMilli[fromPrecision - TSDB_TIME_PRECISION_HOURS];
8,913✔
160
    if (tsInt64 != 0 && unit > INT64_MAX / tsInt64) {
8,913✔
161
      return -1;
×
162
    }
163
    tsInt64 *= unit;
8,913✔
164
    fromPrecision = TSDB_TIME_PRECISION_MILLI;
8,913✔
165
  }
166

167
  return convertTimePrecision(tsInt64, fromPrecision, toPrecision);
51,055,793✔
168
}
169

170
int32_t smlBuildTableInfo(int numRows, const char *measure, int32_t measureLen, SSmlTableInfo **tInfo) {
1,426,609✔
171
  int32_t        code = 0;
1,426,609✔
172
  int32_t        lino = 0;
1,426,609✔
173
  SSmlTableInfo *tag = (SSmlTableInfo *)taosMemoryCalloc(sizeof(SSmlTableInfo), 1);
1,426,609✔
174
  SML_CHECK_NULL(tag)
1,427,040✔
175

176
  tag->sTableName = measure;
1,427,040✔
177
  tag->sTableNameLen = measureLen;
1,427,040✔
178

179
  tag->cols = taosArrayInit(numRows, POINTER_BYTES);
1,427,040✔
180
  SML_CHECK_NULL(tag->cols)
1,427,312✔
181
  *tInfo = tag;
1,427,040✔
182
  return code;
1,427,040✔
183

184
END:
×
185
  taosMemoryFree(tag);
×
186
  uError("%s failed code:%d line:%d", __FUNCTION__, code, lino);
×
187
  return code;
×
188
}
189

190
void smlBuildTsKv(SSmlKv *kv, int64_t ts) {
51,429,643✔
191
  kv->key = tsSmlTsDefaultName;
51,429,643✔
192
  kv->keyLen = strlen(tsSmlTsDefaultName);
51,732,806✔
193
  kv->type = TSDB_DATA_TYPE_TIMESTAMP;
52,098,111✔
194
  kv->i = ts;
52,214,690✔
195
  kv->length = (size_t)tDataTypes[TSDB_DATA_TYPE_TIMESTAMP].bytes;
51,641,921✔
196
}
52,267,718✔
197

198
static void smlDestroySTableMeta(void *para) {
1,233,810✔
199
  if (para == NULL) {
1,233,810✔
200
    return;
×
201
  }
202
  SSmlSTableMeta *meta = *(SSmlSTableMeta **)para;
1,233,810✔
203
  if (meta == NULL) {
1,233,538✔
204
    return;
422,533✔
205
  }
206
  taosHashCleanup(meta->tagHash);
811,005✔
207
  taosHashCleanup(meta->colHash);
811,005✔
208
  taosArrayDestroy(meta->tags);
811,277✔
209
  taosArrayDestroy(meta->cols);
811,277✔
210
  taosMemoryFreeClear(meta->tableMeta);
810,736✔
211
  taosMemoryFree(meta);
811,005✔
212
}
213

214
int32_t smlBuildSuperTableInfo(SSmlHandle *info, SSmlLineInfo *currElement, SSmlSTableMeta **sMeta) {
714,202✔
215
  int32_t     code = TSDB_CODE_SUCCESS;
714,202✔
216
  int32_t     lino = 0;
714,202✔
217
  STableMeta *pTableMeta = NULL;
714,202✔
218

219
  int   measureLen = currElement->measureLen;
714,202✔
220
  char *measure = (char *)taosMemoryMalloc(measureLen);
714,202✔
221
  SML_CHECK_NULL(measure);
714,406✔
222
  (void)memcpy(measure, currElement->measure, measureLen);
714,406✔
223
  if (currElement->measureEscaped) {
714,406✔
224
    PROCESS_SLASH_IN_MEASUREMENT(measure, measureLen);
6,174✔
225
  }
226
  smlStrReplace(measure, measureLen);
714,678✔
227
  code = smlGetMeta(info, measure, measureLen, &pTableMeta);
714,661✔
228
  taosMemoryFree(measure);
714,922✔
229
  if (code != TSDB_CODE_SUCCESS) {
715,353✔
230
    info->dataFormat = false;
422,533✔
231
    info->reRun = true;
422,533✔
232
    goto END;
422,533✔
233
  }
234
  SML_CHECK_CODE(smlBuildSTableMeta(info->dataFormat, sMeta));
292,820✔
235
  for (int i = 0; i < pTableMeta->tableInfo.numOfTags + pTableMeta->tableInfo.numOfColumns; i++) {
5,675,908✔
236
    SSchema *col = pTableMeta->schema + i;
5,382,935✔
237
    SSmlKv   kv = {.key = col->name, .keyLen = strlen(col->name), .type = col->type};
5,382,935✔
238
    if (col->type == TSDB_DATA_TYPE_NCHAR) {
5,383,204✔
239
      kv.length = (col->bytes - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE;
912,287✔
240
    } else if (col->type == TSDB_DATA_TYPE_BINARY || col->type == TSDB_DATA_TYPE_GEOMETRY ||
4,470,917✔
241
               col->type == TSDB_DATA_TYPE_VARBINARY) {
2,775,879✔
242
      kv.length = col->bytes - VARSTR_HEADER_SIZE;
1,696,468✔
243
    } else {
244
      kv.length = col->bytes;
2,774,449✔
245
    }
246

247
    if (i < pTableMeta->tableInfo.numOfColumns) {
5,383,204✔
248
      SML_CHECK_NULL(taosArrayPush((*sMeta)->cols, &kv));
7,139,547✔
249
    } else {
250
      SML_CHECK_NULL(taosArrayPush((*sMeta)->tags, &kv));
3,626,904✔
251
    }
252
  }
253
  SML_CHECK_CODE(taosHashPut(info->superTables, currElement->measure, currElement->measureLen, sMeta, POINTER_BYTES));
292,704✔
254
  (*sMeta)->tableMeta = pTableMeta;
292,820✔
255
  return code;
292,820✔
256

257
END:
422,533✔
258
  smlDestroySTableMeta(sMeta);
422,533✔
259
  taosMemoryFreeClear(pTableMeta);
422,533✔
260
  RETURN
422,533✔
261
}
262

263
bool isSmlColAligned(SSmlHandle *info, int cnt, SSmlKv *kv) {
122,589✔
264
  // cnt begin 0, add ts so + 2
265
  if (unlikely(cnt + 2 > info->currSTableMeta->tableInfo.numOfColumns)) {
122,589✔
266
    goto END;
×
267
  }
268
  // bind data
269
  int32_t ret =
270
      smlBuildCol(info->currTableDataCtx, info->currSTableMeta->schema, kv, cnt + 1, info->taos->optionInfo.charsetCxt);
122,589✔
271
  if (unlikely(ret != TSDB_CODE_SUCCESS)) {
122,861✔
272
    uDebug("smlBuildCol error, retry");
1,942✔
273
    goto END;
1,942✔
274
  }
275
  if (cnt >= taosArrayGetSize(info->maxColKVs)) {
120,919✔
276
    goto END;
×
277
  }
278
  SSmlKv *maxKV = (SSmlKv *)taosArrayGet(info->maxColKVs, cnt);
120,647✔
279
  if (maxKV == NULL) {
120,919✔
280
    goto END;
×
281
  }
282
  if (unlikely(!IS_SAME_KEY)) {
120,919✔
283
    goto END;
120,647✔
284
  }
285

286
  if (unlikely(IS_VAR_DATA_TYPE(kv->type) && kv->length > maxKV->length)) {
×
287
    maxKV->length = kv->length;
×
288
    info->needModifySchema = true;
×
289
  }
290
  return true;
×
291

292
END:
122,589✔
293
  info->dataFormat = false;
122,589✔
294
  info->reRun = true;
122,589✔
295
  return false;
122,861✔
296
}
297

298
bool isSmlTagAligned(SSmlHandle *info, int cnt, SSmlKv *kv) {
1,944,818✔
299
  if (unlikely(cnt + 1 > info->currSTableMeta->tableInfo.numOfTags)) {
1,944,818✔
300
    goto END;
4,354✔
301
  }
302

303
  if (unlikely(cnt >= taosArrayGetSize(info->maxTagKVs))) {
1,940,464✔
304
    goto END;
×
305
  }
306
  SSmlKv *maxKV = (SSmlKv *)taosArrayGet(info->maxTagKVs, cnt);
1,940,837✔
307
  if (maxKV == NULL) {
1,940,623✔
308
    goto END;
×
309
  }
310
  if (unlikely(!IS_SAME_KEY)) {
1,940,623✔
311
    goto END;
4,747✔
312
  }
313

314
  if (unlikely(kv->length > maxKV->length)) {
1,935,876✔
315
    maxKV->length = kv->length;
7,582✔
316
    info->needModifySchema = true;
7,582✔
317
  }
318
  return true;
1,935,876✔
319

320
END:
9,101✔
321
  info->dataFormat = false;
9,101✔
322
  info->reRun = true;
9,101✔
323
  return false;
9,101✔
324
}
325

326
int32_t smlJoinMeasureTag(SSmlLineInfo *elements) {
473,966✔
327
  elements->measureTag = (char *)taosMemoryMalloc(elements->measureLen + elements->tagsLen);
473,966✔
328
  if (elements->measureTag == NULL) {
474,125✔
329
    return terrno;
×
330
  }
331
  (void)memcpy(elements->measureTag, elements->measure, elements->measureLen);
474,125✔
332
  (void)memcpy(elements->measureTag + elements->measureLen, elements->tags, elements->tagsLen);
474,125✔
333
  elements->measureTagsLen = elements->measureLen + elements->tagsLen;
474,125✔
334
  return TSDB_CODE_SUCCESS;
474,125✔
335
}
336

337
static bool smlIsPKTable(STableMeta *pTableMeta) {
466,830✔
338
  for (int i = 0; i < pTableMeta->tableInfo.numOfColumns; i++) {
7,515,084✔
339
    if (pTableMeta->schema[i].flags & COL_IS_KEY) {
7,051,606✔
340
      return true;
3,352✔
341
    }
342
  }
343

344
  return false;
463,478✔
345
}
346

347
int32_t smlProcessSuperTable(SSmlHandle *info, SSmlLineInfo *elements) {
735,641✔
348
  bool isSameMeasure = IS_SAME_SUPER_TABLE;
735,641✔
349
  if (isSameMeasure) {
735,641✔
350
    return TSDB_CODE_SUCCESS;
19,016✔
351
  }
352
  SSmlSTableMeta **tmp = (SSmlSTableMeta **)taosHashGet(info->superTables, elements->measure, elements->measureLen);
716,625✔
353

354
  SSmlSTableMeta *sMeta = NULL;
717,341✔
355
  if (unlikely(tmp == NULL)) {
717,341✔
356
    int32_t code = smlBuildSuperTableInfo(info, elements, &sMeta);
714,641✔
357
    if (code != 0) return code;
715,194✔
358
  } else {
359
    sMeta = *tmp;
2,700✔
360
  }
361
  if (sMeta == NULL) {
295,361✔
362
    uError("smlProcessSuperTable failed to get super table meta");
×
363
    return TSDB_CODE_SML_INTERNAL_ERROR;
×
364
  }
365
  info->currSTableMeta = sMeta->tableMeta;
295,361✔
366
  info->maxTagKVs = sMeta->tags;
295,361✔
367
  info->maxColKVs = sMeta->cols;
295,361✔
368

369
  if (smlIsPKTable(sMeta->tableMeta)) {
295,361✔
370
    return TSDB_CODE_SML_NOT_SUPPORT_PK;
3,352✔
371
  }
372
  return 0;
292,168✔
373
}
374

375
int32_t smlProcessChildTable(SSmlHandle *info, SSmlLineInfo *elements) {
1,431,251✔
376
  int32_t         code = TSDB_CODE_SUCCESS;
1,431,251✔
377
  int32_t         lino = 0;
1,431,251✔
378
  SSmlTableInfo **oneTable =
379
      (SSmlTableInfo **)taosHashGet(info->childTables, elements->measureTag, elements->measureTagsLen);
1,431,251✔
380
  SSmlTableInfo *tinfo = NULL;
1,431,954✔
381
  if (unlikely(oneTable == NULL)) {
1,431,682✔
382
    SML_CHECK_CODE(smlBuildTableInfo(1, elements->measure, elements->measureLen, &tinfo));
1,426,768✔
383
    SML_CHECK_CODE(
1,427,312✔
384
        taosHashPut(info->childTables, elements->measureTag, elements->measureTagsLen, &tinfo, POINTER_BYTES));
385

386
    tinfo->tags = taosArrayDup(info->preLineTagKV, NULL);
1,427,040✔
387
    SML_CHECK_NULL(tinfo->tags);
1,427,040✔
388
    for (size_t i = 0; i < taosArrayGetSize(info->preLineTagKV); i++) {
9,770,074✔
389
      SSmlKv *kv = (SSmlKv *)taosArrayGet(info->preLineTagKV, i);
8,343,677✔
390
      SML_CHECK_NULL(kv);
8,342,974✔
391
      if (kv->keyEscaped) kv->key = NULL;
8,343,034✔
392
      if (kv->valueEscaped) kv->value = NULL;
8,343,034✔
393
    }
394

395
    SML_CHECK_CODE(smlSetCTableName(tinfo, info->tbnameKey));
1,426,910✔
396
    SML_CHECK_CODE(getTableUid(info, elements, tinfo));
1,426,224✔
397
    if (info->dataFormat) {
1,426,224✔
398
      info->currSTableMeta->uid = tinfo->uid;
298,127✔
399
      SML_CHECK_CODE(smlInitTableDataCtx(info->pQuery, info->currSTableMeta, &tinfo->tableDataCtx));
298,127✔
400
    }
401
  } else {
402
    tinfo = *oneTable;
4,914✔
403
  }
404
  if (info->dataFormat) info->currTableDataCtx = tinfo->tableDataCtx;
1,431,523✔
405
  return TSDB_CODE_SUCCESS;
1,430,979✔
406

407
END:
×
408
  smlDestroyTableInfo(&tinfo);
×
409
  RETURN
×
410
}
411

412
int32_t smlParseEndTelnetJsonFormat(SSmlHandle *info, SSmlLineInfo *elements, SSmlKv *kvTs, SSmlKv *kv) {
1,380,667✔
413
  int32_t code = 0;
1,380,667✔
414
  int32_t lino = 0;
1,380,667✔
415
  uDebug("SML:0x%" PRIx64 ", %s format true, ts:%" PRId64, info->id, __FUNCTION__, kvTs->i);
1,380,667✔
416
  SML_CHECK_CODE(
1,380,667✔
417
      smlBuildCol(info->currTableDataCtx, info->currSTableMeta->schema, kvTs, 0, info->taos->optionInfo.charsetCxt));
418
  SML_CHECK_CODE(
1,380,566✔
419
      smlBuildCol(info->currTableDataCtx, info->currSTableMeta->schema, kv, 1, info->taos->optionInfo.charsetCxt));
420
  SML_CHECK_CODE(smlBuildRow(info->currTableDataCtx));
1,380,768✔
421

422
END:
1,380,669✔
423
  clearColValArraySml(info->currTableDataCtx->pValues);
1,380,669✔
424
  RETURN
1,380,471✔
425
}
426

427
int32_t smlParseEndTelnetJsonUnFormat(SSmlHandle *info, SSmlLineInfo *elements, SSmlKv *kvTs, SSmlKv *kv) {
659,550✔
428
  int32_t code = 0;
659,550✔
429
  int32_t lino = 0;
659,550✔
430
  uDebug("SML:0x%" PRIx64 ", %s format false, ts:%" PRId64, info->id, __FUNCTION__, kvTs->i);
659,550✔
431
  if (elements->colArray == NULL) {
659,550✔
432
    elements->colArray = taosArrayInit(16, sizeof(SSmlKv));
659,550✔
433
    SML_CHECK_NULL(elements->colArray);
659,651✔
434
  }
435
  SML_CHECK_NULL(taosArrayPush(elements->colArray, kvTs));
1,319,203✔
436
  SML_CHECK_NULL(taosArrayPush(elements->colArray, kv));
1,319,104✔
437

438
END:
659,552✔
439
  RETURN
659,552✔
440
}
441

442
int32_t smlParseEndLine(SSmlHandle *info, SSmlLineInfo *elements, SSmlKv *kvTs) {
49,748,141✔
443
  if (info->dataFormat) {
49,748,141✔
444
    uDebug("SML:0x%" PRIx64 ", %s format true, ts:%" PRId64, info->id, __FUNCTION__, kvTs->i);
×
445
    int32_t ret =
446
        smlBuildCol(info->currTableDataCtx, info->currSTableMeta->schema, kvTs, 0, info->taos->optionInfo.charsetCxt);
×
447
    if (ret == TSDB_CODE_SUCCESS) {
×
448
      ret = smlBuildRow(info->currTableDataCtx);
×
449
    }
450

451
    clearColValArraySml(info->currTableDataCtx->pValues);
×
452
    taosArrayClearP(info->escapedStringList, NULL);
×
453
    if (unlikely(ret != TSDB_CODE_SUCCESS)) {
×
454
      uError("SML:0x%" PRIx64 ", %s smlBuildCol error:%d", info->id, __FUNCTION__, ret);
×
455
      return ret;
×
456
    }
457
  } else {
458
    uDebug("SML:0x%" PRIx64 ", %s format false, ts:%" PRId64, info->id, __FUNCTION__, kvTs->i);
49,875,016✔
459
    taosArraySet(elements->colArray, 0, kvTs);
49,875,016✔
460
  }
461
  info->preLine = *elements;
49,988,853✔
462

463
  return TSDB_CODE_SUCCESS;
49,504,560✔
464
}
465

466
static int32_t smlParseTableName(SArray *tags, char *childTableName, char *tbnameKey) {
1,426,337✔
467
  int32_t code = 0;
1,426,337✔
468
  int32_t lino = 0;
1,426,337✔
469
  bool    autoChildName = false;
1,426,337✔
470
  size_t  delimiter = strlen(tsSmlAutoChildTableNameDelimiter);
1,426,337✔
471
  if (delimiter > 0 && tbnameKey == NULL) {
1,426,337✔
472
    size_t totalNameLen = delimiter * (taosArrayGetSize(tags) - 1);
1,056✔
473
    for (int i = 0; i < taosArrayGetSize(tags); i++) {
3,344✔
474
      SSmlKv *tag = (SSmlKv *)taosArrayGet(tags, i);
2,112✔
475
      SML_CHECK_NULL(tag);
2,288✔
476
      totalNameLen += tag->length;
2,288✔
477
    }
478
    if (totalNameLen < TSDB_TABLE_NAME_LEN) {
1,328✔
479
      autoChildName = true;
704✔
480
    }
481
  }
482
  if (autoChildName) {
1,426,609✔
483
    childTableName[0] = '\0';
704✔
484
    for (int i = 0; i < taosArrayGetSize(tags); i++) {
2,464✔
485
      SSmlKv *tag = (SSmlKv *)taosArrayGet(tags, i);
1,760✔
486
      SML_CHECK_NULL(tag);
1,760✔
487
      (void)strncat(childTableName, tag->value, TMIN(tag->length, TSDB_TABLE_NAME_LEN - 1 - strlen(childTableName)));
1,760✔
488
      if (i != taosArrayGetSize(tags) - 1) {
1,760✔
489
        (void)strncat(childTableName, tsSmlAutoChildTableNameDelimiter,
1,056✔
490
                      TSDB_TABLE_NAME_LEN - 1 - strlen(childTableName));
1,056✔
491
      }
492
    }
493
    if (tsSmlDot2Underline) {
704✔
494
      smlStrReplace(childTableName, strlen(childTableName));
704✔
495
    }
496
  } else {
497
    if (tbnameKey == NULL) {
1,425,905✔
498
      tbnameKey = tsSmlChildTableName;
1,425,905✔
499
    }
500
    size_t childTableNameLen = strlen(tbnameKey);
1,425,905✔
501
    if (childTableNameLen <= 0) return TSDB_CODE_SUCCESS;
1,425,905✔
502

503
    for (int i = 0; i < taosArrayGetSize(tags); i++) {
470,104✔
504
      SSmlKv *tag = (SSmlKv *)taosArrayGet(tags, i);
401,152✔
505
      SML_CHECK_NULL(tag);
401,152✔
506
      // handle child table name
507
      if (childTableNameLen == tag->keyLen && strncmp(tag->key, tbnameKey, tag->keyLen) == 0) {
401,152✔
508
        tstrncpy(childTableName, tag->value, TMIN(TSDB_TABLE_NAME_LEN, tag->length + 1));
3,773✔
509
        if (tsSmlDot2Underline) {
3,773✔
510
          smlStrReplace(childTableName, strlen(childTableName));
1,903✔
511
        }
512
        taosArrayRemove(tags, i);
3,773✔
513
        break;
3,773✔
514
      }
515
    }
516
  }
517

518
END:
68,952✔
519
  RETURN
73,429✔
520
}
521

522
int32_t smlSetCTableName(SSmlTableInfo *oneTable, char *tbnameKey) {
1,425,793✔
523
  int32_t code = 0;
1,425,793✔
524
  int32_t lino = 0;
1,425,793✔
525
  SArray *dst = NULL;
1,425,793✔
526
  SML_CHECK_CODE(smlParseTableName(oneTable->tags, oneTable->childTableName, tbnameKey));
1,425,793✔
527

528
  if (strlen(oneTable->childTableName) == 0) {
1,426,768✔
529
    dst = taosArrayDup(oneTable->tags, NULL);
1,422,291✔
530
    SML_CHECK_NULL(dst);
1,422,734✔
531
    if (oneTable->sTableNameLen >= TSDB_TABLE_NAME_LEN) {
1,422,577✔
532
      SML_CHECK_CODE(TSDB_CODE_SML_INTERNAL_ERROR);
×
533
    }
534
    char          superName[TSDB_TABLE_NAME_LEN] = {0};
1,422,577✔
535
    RandTableName rName = {dst, NULL, (uint8_t)oneTable->sTableNameLen, oneTable->childTableName};
1,422,577✔
536
    if (tsSmlDot2Underline) {
1,422,577✔
537
      (void)memcpy(superName, oneTable->sTableName, oneTable->sTableNameLen);
1,150,565✔
538
      smlStrReplace(superName, oneTable->sTableNameLen);
1,150,565✔
539
      rName.stbFullName = superName;
1,150,823✔
540
    } else {
541
      rName.stbFullName = oneTable->sTableName;
272,012✔
542
    }
543

544
    SML_CHECK_CODE(buildChildTableName(&rName));
1,422,835✔
545
  }
546

547
END:
4,477✔
548
  taosArrayDestroy(dst);
1,427,040✔
549
  RETURN
1,426,768✔
550
}
551

552
int32_t getTableUid(SSmlHandle *info, SSmlLineInfo *currElement, SSmlTableInfo *tinfo) {
1,425,420✔
553
  char   key[TSDB_TABLE_NAME_LEN * 2 + 1] = {0};
1,425,420✔
554
  size_t nLen = strlen(tinfo->childTableName);
1,425,692✔
555
  (void)memcpy(key, currElement->measure, currElement->measureLen);
1,426,508✔
556
  if (tsSmlDot2Underline) {
1,426,780✔
557
    smlStrReplace(key, currElement->measureLen);
1,153,170✔
558
  }
559
  (void)memcpy(key + currElement->measureLen + 1, tinfo->childTableName, nLen);
1,426,939✔
560
  void *uid =
561
      taosHashGet(info->tableUids, key,
1,427,211✔
562
                  currElement->measureLen + 1 + nLen);  // use \0 as separator for stable name and child table name
1,426,939✔
563
  if (uid == NULL) {
1,426,667✔
564
    tinfo->uid = info->uid++;
1,424,093✔
565
    return taosHashPut(info->tableUids, key, currElement->measureLen + 1 + nLen, &tinfo->uid, sizeof(uint64_t));
1,424,637✔
566
  } else {
567
    tinfo->uid = *(uint64_t *)uid;
2,574✔
568
  }
569
  return TSDB_CODE_SUCCESS;
2,574✔
570
}
571

572
int32_t smlBuildSTableMeta(bool isDataFormat, SSmlSTableMeta **sMeta) {
810,846✔
573
  int32_t         code = 0;
810,846✔
574
  int32_t         lino = 0;
810,846✔
575
  SSmlSTableMeta *meta = (SSmlSTableMeta *)taosMemoryCalloc(sizeof(SSmlSTableMeta), 1);
810,846✔
576
  SML_CHECK_NULL(meta);
810,846✔
577
  if (unlikely(!isDataFormat)) {
810,846✔
578
    meta->tagHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
518,457✔
579
    SML_CHECK_NULL(meta->tagHash);
518,457✔
580
    meta->colHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
518,457✔
581
    SML_CHECK_NULL(meta->colHash);
518,457✔
582
  }
583

584
  meta->tags = taosArrayInit(32, sizeof(SSmlKv));
810,846✔
585
  SML_CHECK_NULL(meta->tags);
811,277✔
586

587
  meta->cols = taosArrayInit(32, sizeof(SSmlKv));
811,277✔
588
  SML_CHECK_NULL(meta->cols);
811,118✔
589
  *sMeta = meta;
811,118✔
590
  return TSDB_CODE_SUCCESS;
811,118✔
591

592
END:
×
593
  smlDestroySTableMeta(&meta);
×
594
  uError("%s failed code:%d line:%d", __FUNCTION__, code, lino);
×
595
  return code;
×
596
}
597

598
int32_t smlParseNumber(SSmlKv *kvVal, SSmlMsgBuf *msg) {
169,115,336✔
599
  const char *pVal = kvVal->value;
169,115,336✔
600
  int32_t     len = kvVal->length;
170,464,214✔
601
  char       *endptr = NULL;
170,674,484✔
602
  double      result = taosStr2Double(pVal, &endptr);
171,145,265✔
603
  if (pVal == endptr) {
166,932,971✔
604
    RETURN_FALSE
730✔
605
  }
606

607
  int32_t left = len - (endptr - pVal);
166,932,241✔
608
  if (left == 0) {
166,932,241✔
609
    SET_DOUBLE
132,755✔
610
  } else if (left == 3) {
166,799,486✔
611
    if (endptr[0] == 'f' || endptr[0] == 'F') {
168,085,433✔
612
      if (endptr[1] == '6' && endptr[2] == '4') {
118,396,657✔
613
        SET_DOUBLE
306,014✔
614
      } else if (endptr[1] == '3' && endptr[2] == '2') {
119,066,847✔
615
        SET_FLOAT
119,568,538✔
616
      } else {
617
        RETURN_FALSE
×
618
      }
619
    } else if (endptr[0] == 'i' || endptr[0] == 'I') {
50,353,091✔
620
      if (endptr[1] == '6' && endptr[2] == '4') {
49,528,936✔
621
        SET_BIGINT
302,524✔
622
      } else if (endptr[1] == '3' && endptr[2] == '2') {
49,490,087✔
623
        SET_INT
49,146,218✔
624
      } else if (endptr[1] == '1' && endptr[2] == '6') {
228,407✔
625
        SET_SMALL_INT
306,901✔
626
      } else {
627
        RETURN_FALSE
×
628
      }
629
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
623,217✔
630
      if (endptr[1] == '6' && endptr[2] == '4') {
623,217✔
631
        SET_UBIGINT
286,110✔
632
      } else if (endptr[1] == '3' && endptr[2] == '2') {
337,107✔
633
        SET_UINT
167,809✔
634
      } else if (endptr[1] == '1' && endptr[2] == '6') {
169,298✔
635
        SET_USMALL_INT
169,502✔
636
      } else {
637
        RETURN_FALSE
×
638
      }
639
    } else {
640
      RETURN_FALSE
×
641
    }
642
  } else if (left == 2) {
411,619✔
643
    if (endptr[0] == 'i' || endptr[0] == 'I') {
479,736✔
644
      if (endptr[1] == '8') {
311,824✔
645
        SET_TINYINT
311,926✔
646
      } else {
UNCOV
647
        RETURN_FALSE
×
648
      }
649
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
167,912✔
650
      if (endptr[1] == '8') {
167,912✔
651
        SET_UTINYINT
167,912✔
652
      } else {
653
        RETURN_FALSE
×
654
      }
655
    } else {
656
      RETURN_FALSE
×
657
    }
658
  } else if (left == 1) {
12,290✔
659
    if (endptr[0] == 'i' || endptr[0] == 'I') {
11,521✔
660
      SET_BIGINT
11,521✔
661
    } else if (endptr[0] == 'u' || endptr[0] == 'U') {
×
662
      SET_UBIGINT
×
663
    } else {
664
      RETURN_FALSE
×
665
    }
666
  } else {
667
    RETURN_FALSE;
4,664✔
668
  }
669
  return true;
168,108,807✔
670
}
671

672
int32_t smlGetMeta(SSmlHandle *info, const void *measure, int32_t measureLen, STableMeta **pTableMeta) {
714,304✔
673
  *pTableMeta = NULL;
714,304✔
674

675
  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
714,304✔
676
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));
714,576✔
677

678
  SRequestConnInfo conn = {0};
714,304✔
679
  conn.pTrans = info->taos->pAppInfo->pTransporter;
714,304✔
680
  conn.requestId = info->pRequest->requestId;
714,304✔
681
  conn.requestObjRefId = info->pRequest->self;
714,304✔
682
  conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
714,304✔
683
  int32_t len = TMIN(measureLen, TSDB_TABLE_NAME_LEN - 1);
715,186✔
684
  (void)memcpy(pName.tname, measure, measureLen);
715,186✔
685
  pName.tname[len] = 0;
715,186✔
686

687
  return catalogGetSTableMeta(info->pCatalog, &conn, &pName, pTableMeta);
714,914✔
688
}
689

690
static int64_t smlGenId() {
768,055✔
691
  static volatile int64_t linesSmlHandleId = 0;
692

693
  int64_t id = 0;
768,055✔
694
  do {
695
    id = atomic_add_fetch_64(&linesSmlHandleId, 1);
768,055✔
696
  } while (id == 0);
768,551✔
697

698
  return id;
768,551✔
699
}
700

701
static int32_t smlGenerateSchemaAction(SSchema *colField, SHashObj *colHash, SSmlKv *kv, bool isTag,
11,712,291✔
702
                                       ESchemaAction *action, SSmlHandle *info) {
703
  uint16_t *index = colHash ? (uint16_t *)taosHashGet(colHash, kv->key, kv->keyLen) : NULL;
11,712,291✔
704
  if (index) {
11,712,290✔
705
    if (colField[*index].type != kv->type) {
4,215,550✔
706
      snprintf(info->msgBuf.buf, info->msgBuf.len,
5,826✔
707
               "SML:0x%" PRIx64 ", %s point type and db type mismatch, db type:%s, point type:%s, key:%s", info->id,
708
               __FUNCTION__, tDataTypes[colField[*index].type].name, tDataTypes[kv->type].name, kv->key);
3,884✔
709
      uError("%s", info->msgBuf.buf);
1,942✔
710
      return TSDB_CODE_SML_INVALID_DATA;
1,942✔
711
    }
712

713
    if (((colField[*index].type == TSDB_DATA_TYPE_VARCHAR || colField[*index].type == TSDB_DATA_TYPE_VARBINARY ||
4,213,608✔
714
          colField[*index].type == TSDB_DATA_TYPE_GEOMETRY) &&
2,636,345✔
715
         (colField[*index].bytes - VARSTR_HEADER_SIZE) < kv->length) ||
1,577,606✔
716
        (colField[*index].type == TSDB_DATA_TYPE_NCHAR &&
4,210,410✔
717
         ((colField[*index].bytes - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE < kv->length))) {
733,541✔
718
      if (isTag) {
27,120✔
719
        *action = SCHEMA_ACTION_CHANGE_TAG_SIZE;
23,922✔
720
      } else {
721
        *action = SCHEMA_ACTION_CHANGE_COLUMN_SIZE;
3,198✔
722
      }
723
    }
724
  } else {
725
    if (isTag) {
7,496,740✔
726
      *action = SCHEMA_ACTION_ADD_TAG;
3,320,793✔
727
    } else {
728
      *action = SCHEMA_ACTION_ADD_COLUMN;
4,175,947✔
729
    }
730
  }
731
  return TSDB_CODE_SUCCESS;
11,710,348✔
732
}
733

734
#define BOUNDARY 1024
735
static int32_t smlFindNearestPowerOf2(int32_t length, uint8_t type) {
2,311,440✔
736
  int32_t result = 1;
2,311,440✔
737
  if (length >= BOUNDARY) {
2,311,440✔
738
    result = length;
10,449✔
739
  } else {
740
    while (result <= length) {
10,981,158✔
741
      result <<= 1;
8,680,167✔
742
    }
743
  }
744

745
  if ((type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_VARBINARY || type == TSDB_DATA_TYPE_GEOMETRY) &&
2,311,440✔
746
      result > TSDB_MAX_BINARY_LEN - VARSTR_HEADER_SIZE) {
456,507✔
747
    result = TSDB_MAX_BINARY_LEN - VARSTR_HEADER_SIZE;
×
748
  } else if (type == TSDB_DATA_TYPE_NCHAR && result > (TSDB_MAX_NCHAR_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE) {
2,311,440✔
749
    result = (TSDB_MAX_NCHAR_LEN - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE;
×
750
  }
751

752
  if (type == TSDB_DATA_TYPE_NCHAR) {
2,311,440✔
753
    result = result * TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE;
1,855,035✔
754
  } else if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_VARBINARY || type == TSDB_DATA_TYPE_GEOMETRY) {
456,405✔
755
    result = result + VARSTR_HEADER_SIZE;
456,507✔
756
  }
757
  return result;
2,311,440✔
758
}
759

760
static int32_t smlProcessSchemaAction(SSmlHandle *info, SSchema *schemaField, SHashObj *schemaHash, SArray *cols,
338,502✔
761
                                      SHashObj *schemaHashCheck, ESchemaAction *action, bool isTag) {
762
  int32_t code = TSDB_CODE_SUCCESS;
338,502✔
763
  int32_t lino = 0;
338,502✔
764
  for (int j = 0; j < taosArrayGetSize(cols); ++j) {
4,612,761✔
765
    if (j == 0 && !isTag) continue;
4,277,407✔
766
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, j);
4,110,374✔
767
    SML_CHECK_NULL(kv);
4,110,272✔
768
    SML_CHECK_CODE(smlGenerateSchemaAction(schemaField, schemaHash, kv, isTag, action, info));
4,110,272✔
769
    if (taosHashGet(schemaHashCheck, kv->key, kv->keyLen) != NULL) {
4,108,329✔
770
      uError("SML:0x%" PRIx64 ", %s duplicated column %s", info->id, __FUNCTION__, kv->key);
1,206✔
771
      SML_CHECK_CODE(TSDB_CODE_PAR_DUPLICATED_COLUMN);
1,206✔
772
    }
773
  }
774

775
END:
335,354✔
776
  RETURN
338,502✔
777
}
778

779
static int32_t smlCheckMeta(SSchema *schema, int32_t length, SArray *cols) {
327,833✔
780
  int32_t   code = TSDB_CODE_SUCCESS;
327,833✔
781
  int32_t   lino = 0;
327,833✔
782
  SHashObj *hashTmp = taosHashInit(length, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
327,833✔
783
  SML_CHECK_NULL(hashTmp);
327,833✔
784
  for (int32_t i = 0; i < length; i++) {
4,585,996✔
785
    SML_CHECK_CODE(taosHashPut(hashTmp, schema[i].name, strlen(schema[i].name), &schema[i], sizeof(SSchema)));
4,258,060✔
786
  }
787
  for (int32_t i = 0; i < taosArrayGetSize(cols); i++) {
4,580,550✔
788
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
4,253,034✔
789
    SML_CHECK_NULL(kv);
4,253,034✔
790
    SSchema *sTmp = taosHashGet(hashTmp, kv->key, kv->keyLen);
4,253,034✔
791
    if (sTmp == NULL) {
4,253,137✔
792
      SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
793
    }
794
    if ((kv->type == TSDB_DATA_TYPE_VARCHAR && kv->length + VARSTR_HEADER_SIZE > sTmp->bytes) ||
4,253,137✔
795
        (kv->type == TSDB_DATA_TYPE_NCHAR && kv->length * TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE > sTmp->bytes)) {
4,253,137✔
796
      uError("column %s (type %s) bytes invalid. db bytes:%d, kv bytes:%zu", sTmp->name,
523✔
797
             tDataTypes[sTmp->type].name, sTmp->bytes, kv->length);
798
      SML_CHECK_CODE(TSDB_CODE_MND_INVALID_SCHEMA_VER);
523✔
799
    }
800
  }
801

802
END:
302,459✔
803
  taosHashCleanup(hashTmp);
302,982✔
804
  RETURN
327,833✔
805
}
806

807
static int32_t getBytes(uint8_t type, int32_t length) {
7,496,471✔
808
  if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_VARBINARY || type == TSDB_DATA_TYPE_NCHAR ||
7,496,471✔
809
      type == TSDB_DATA_TYPE_GEOMETRY) {
810
    return smlFindNearestPowerOf2(length, type);
2,311,236✔
811
  } else {
812
    return tDataTypes[type].bytes;
5,185,235✔
813
  }
814
}
815

816
static int32_t smlBuildFieldsList(SSmlHandle *info, SSchema *schemaField, SHashObj *schemaHash, SArray *cols,
763,269✔
817
                                  SArray *results, int32_t numOfCols, bool isTag) {
818
  int32_t code = TSDB_CODE_SUCCESS;
763,269✔
819
  int32_t lino = 0;
763,269✔
820
  for (int j = 0; j < taosArrayGetSize(cols); ++j) {
8,364,370✔
821
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, j);
7,602,019✔
822
    SML_CHECK_NULL(kv);
7,602,019✔
823
    ESchemaAction action = SCHEMA_ACTION_NULL;
7,602,019✔
824
    SML_CHECK_CODE(smlGenerateSchemaAction(schemaField, schemaHash, kv, isTag, &action, info));
7,602,019✔
825
    if (action == SCHEMA_ACTION_ADD_COLUMN || action == SCHEMA_ACTION_ADD_TAG) {
15,085,338✔
826
      SField field = {0};
7,482,911✔
827
      field.type = kv->type;
7,482,911✔
828
      field.bytes = getBytes(kv->type, kv->length);
7,482,911✔
829
      (void)memcpy(field.name, kv->key, TMIN(kv->keyLen, sizeof(field.name) - 1));
7,483,217✔
830
      SML_CHECK_NULL(taosArrayPush(results, &field));
7,483,319✔
831
    } else if (action == SCHEMA_ACTION_CHANGE_COLUMN_SIZE || action == SCHEMA_ACTION_CHANGE_TAG_SIZE) {
119,108✔
832
      uint16_t *index = (uint16_t *)taosHashGet(schemaHash, kv->key, kv->keyLen);
14,886✔
833
      if (index == NULL) {
13,560✔
834
        SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
835
      }
836
      uint16_t newIndex = *index;
13,560✔
837
      if (isTag) newIndex -= numOfCols;
13,560✔
838
      SField *field = (SField *)taosArrayGet(results, newIndex);
13,560✔
839
      SML_CHECK_NULL(field);
13,560✔
840
      field->bytes = getBytes(kv->type, kv->length);
13,560✔
841
    }
842
  }
843

844
  int32_t maxLen = isTag ? TSDB_MAX_TAGS_LEN : TSDB_MAX_BYTES_PER_ROW;
763,269✔
845
  int32_t len = 0;
763,269✔
846
  for (int j = 0; j < taosArrayGetSize(results); ++j) {
8,367,170✔
847
    SField *field = taosArrayGet(results, j);
7,603,901✔
848
    SML_CHECK_NULL(field);
7,603,901✔
849
    len += field->bytes;
7,603,901✔
850
  }
851
  if (len > maxLen) {
763,269✔
852
    return isTag ? TSDB_CODE_PAR_INVALID_TAGS_LENGTH : TSDB_CODE_PAR_INVALID_ROW_LENGTH;
4,111✔
853
  }
854

855
END:
759,158✔
856
  RETURN
759,158✔
857
}
858

859
static FORCE_INLINE void smlBuildCreateStbReq(SMCreateStbReq *pReq, int32_t colVer, int32_t tagVer, tb_uid_t suid,
860
                                              int8_t source) {
861
  pReq->colVer = colVer;
386,518✔
862
  pReq->tagVer = tagVer;
386,518✔
863
  pReq->suid = suid;
386,518✔
864
  pReq->source = source;
386,518✔
865
}
386,518✔
866
static int32_t smlSendMetaMsg(SSmlHandle *info, SName *pName, SArray *pColumns, SArray **pTags, STableMeta *pTableMeta,
386,518✔
867
                              ESchemaAction action) {
868
  SRequestObj   *pRequest = NULL;
386,518✔
869
  SMCreateStbReq pReq = {0};
386,518✔
870
  int32_t        code = TSDB_CODE_SUCCESS;
386,518✔
871
  int32_t        lino = 0;
386,518✔
872
  SCmdMsgInfo    pCmdMsg = {0};
386,518✔
873
  char          *pSql = NULL;
386,518✔
874

875
  // put front for free
876
  pReq.numOfColumns = taosArrayGetSize(pColumns);
386,518✔
877
  pReq.pTags = *pTags;
386,518✔
878
  pReq.numOfTags = taosArrayGetSize(*pTags);
386,518✔
879
  *pTags = NULL;
386,518✔
880

881
  pReq.pColumns = taosArrayInit(pReq.numOfColumns, sizeof(SFieldWithOptions));
386,518✔
882
  SML_CHECK_NULL(pReq.pColumns);
386,518✔
883
  for (int32_t i = 0; i < pReq.numOfColumns; ++i) {
4,618,116✔
884
    SField *pField = taosArrayGet(pColumns, i);
4,231,598✔
885
    SML_CHECK_NULL(pField);
4,231,495✔
886
    SFieldWithOptions fieldWithOption = {0};
4,231,495✔
887
    setFieldWithOptions(&fieldWithOption, pField);
4,231,495✔
888
    setDefaultOptionsForField(&fieldWithOption);
4,231,699✔
889
    SML_CHECK_NULL(taosArrayPush(pReq.pColumns, &fieldWithOption));
8,463,095✔
890
  }
891

892
  if (action == SCHEMA_ACTION_CREATE_STABLE) {
386,518✔
893
    pSql = "sml_create_stable";
372,297✔
894
    smlBuildCreateStbReq(&pReq, 1, 1, 0, TD_REQ_FROM_APP);
895
  } else if (action == SCHEMA_ACTION_ADD_TAG || action == SCHEMA_ACTION_CHANGE_TAG_SIZE) {
14,221✔
896
    pSql = (action == SCHEMA_ACTION_ADD_TAG) ? "sml_add_tag" : "sml_modify_tag_size";
10,622✔
897
    smlBuildCreateStbReq(&pReq, pTableMeta->sversion, pTableMeta->tversion + 1, pTableMeta->uid, TD_REQ_FROM_SML);
10,622✔
898
  } else if (action == SCHEMA_ACTION_ADD_COLUMN || action == SCHEMA_ACTION_CHANGE_COLUMN_SIZE) {
3,599✔
899
    pSql = (action == SCHEMA_ACTION_ADD_COLUMN) ? "sml_add_column" : "sml_modify_column_size";
3,599✔
900
    smlBuildCreateStbReq(&pReq, pTableMeta->sversion + 1, pTableMeta->tversion, pTableMeta->uid, TD_REQ_FROM_SML);
3,599✔
901
  } else {
902
    SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
903
  }
904

905
  SML_CHECK_CODE(buildRequest(info->taos->id, pSql, strlen(pSql), NULL, false, &pRequest, 0));
386,518✔
906

907
  pRequest->syncQuery = true;
386,518✔
908
  if (!pRequest->pDb) {
386,518✔
909
    SML_CHECK_CODE(TSDB_CODE_PAR_DB_NOT_SPECIFIED);
×
910
  }
911

912
  if (pReq.numOfTags == 0) {
386,518✔
913
    pReq.numOfTags = 1;
343✔
914
    SField field = {0};
343✔
915
    field.type = TSDB_DATA_TYPE_NCHAR;
343✔
916
    field.bytes = TSDB_NCHAR_SIZE + VARSTR_HEADER_SIZE;
343✔
917
    tstrncpy(field.name, tsSmlTagName, sizeof(field.name));
343✔
918
    SML_CHECK_NULL(taosArrayPush(pReq.pTags, &field));
686✔
919
  }
920

921
  pReq.commentLen = -1;
386,518✔
922
  pReq.igExists = true;
386,518✔
923
  SML_CHECK_CODE(tNameExtractFullName(pName, pReq.name));
386,518✔
924

925
  pCmdMsg.epSet = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
386,518✔
926
  pCmdMsg.msgType = TDMT_MND_CREATE_STB;
386,518✔
927
  pCmdMsg.msgLen = tSerializeSMCreateStbReq(NULL, 0, &pReq);
386,518✔
928
  if (pCmdMsg.msgLen < 0) {
386,518✔
929
    uError("failed to serialize create stable request1, code:%d, terrno:%d", pCmdMsg.msgLen, terrno);
×
930
    SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
931
  }
932
  pCmdMsg.pMsg = taosMemoryMalloc(pCmdMsg.msgLen);
386,518✔
933
  SML_CHECK_NULL(pCmdMsg.pMsg);
386,518✔
934
  code = tSerializeSMCreateStbReq(pCmdMsg.pMsg, pCmdMsg.msgLen, &pReq);
386,518✔
935
  if (code < 0) {
386,518✔
936
    taosMemoryFree(pCmdMsg.pMsg);
×
937
    uError("failed to serialize create stable request2, code:%d, terrno:%d", code, terrno);
×
938
    SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
939
  }
940

941
  SQuery pQuery = {0};
386,518✔
942
  pQuery.execMode = QUERY_EXEC_MODE_RPC;
386,518✔
943
  pQuery.pCmdMsg = &pCmdMsg;
386,518✔
944
  pQuery.msgType = pQuery.pCmdMsg->msgType;
386,518✔
945
  pQuery.stableQuery = true;
386,518✔
946

947
  launchQueryImpl(pRequest, &pQuery, true, NULL);  // no need to check return value
386,518✔
948

949
  if (pRequest->code == TSDB_CODE_SUCCESS) {
386,518✔
950
    SML_CHECK_CODE(catalogRemoveTableMeta(info->pCatalog, pName));
358,228✔
951
  }
952
  code = pRequest->code;
386,518✔
953

954
END:
386,518✔
955
  destroyRequest(pRequest);
386,518✔
956
  tFreeSMCreateStbReq(&pReq);
386,518✔
957
  RETURN
386,518✔
958
}
959

960
static int32_t smlCreateTable(SSmlHandle *info, SRequestConnInfo *conn, SSmlSTableMeta *sTableData, SName *pName,
372,983✔
961
                              STableMeta **pTableMeta) {
962
  int32_t code = 0;
372,983✔
963
  int32_t lino = 0;
372,983✔
964
  SArray *pColumns = NULL;
372,983✔
965
  SArray *pTags = NULL;
372,983✔
966
  SML_CHECK_CODE(smlCheckAuth(info, conn, NULL, AUTH_TYPE_WRITE));
372,983✔
967
  uDebug("SML:0x%" PRIx64 ", %s create table:%s", info->id, __FUNCTION__, pName->tname);
372,640✔
968
  pColumns = taosArrayInit(taosArrayGetSize(sTableData->cols), sizeof(SField));
372,640✔
969
  SML_CHECK_NULL(pColumns);
372,640✔
970
  pTags = taosArrayInit(taosArrayGetSize(sTableData->tags), sizeof(SField));
372,640✔
971
  SML_CHECK_NULL(pTags);
372,640✔
972
  SML_CHECK_CODE(smlBuildFieldsList(info, NULL, NULL, sTableData->tags, pTags, 0, true));
372,640✔
973
  SML_CHECK_CODE(smlBuildFieldsList(info, NULL, NULL, sTableData->cols, pColumns, 0, false));
372,640✔
974
  SML_CHECK_CODE(smlSendMetaMsg(info, pName, pColumns, &pTags, NULL, SCHEMA_ACTION_CREATE_STABLE));
372,297✔
975
  info->cost.numOfCreateSTables++;
344,064✔
976
  taosMemoryFreeClear(*pTableMeta);
344,064✔
977

978
  SML_CHECK_CODE(catalogGetSTableMeta(info->pCatalog, conn, pName, pTableMeta));
344,064✔
979

980
END:
347,183✔
981
  taosArrayDestroy(pColumns);
372,983✔
982
  taosArrayDestroy(pTags);
372,983✔
983
  RETURN
372,983✔
984
}
985

986
static int32_t smlBuildFields(SArray **pColumns, SArray **pTags, STableMeta *pTableMeta, SSmlSTableMeta *sTableData) {
17,989✔
987
  int32_t code = 0;
17,989✔
988
  int32_t lino = 0;
17,989✔
989
  *pColumns = taosArrayInit(taosArrayGetSize(sTableData->cols) + (pTableMeta)->tableInfo.numOfColumns, sizeof(SField));
17,989✔
990
  SML_CHECK_NULL(pColumns);
17,989✔
991
  *pTags = taosArrayInit(taosArrayGetSize(sTableData->tags) + (pTableMeta)->tableInfo.numOfTags, sizeof(SField));
17,989✔
992
  SML_CHECK_NULL(pTags);
17,989✔
993
  for (uint16_t i = 0; i < (pTableMeta)->tableInfo.numOfColumns + (pTableMeta)->tableInfo.numOfTags; i++) {
207,678✔
994
    SField field = {0};
189,689✔
995
    field.type = (pTableMeta)->schema[i].type;
189,689✔
996
    field.bytes = (pTableMeta)->schema[i].bytes;
189,689✔
997
    tstrncpy(field.name, (pTableMeta)->schema[i].name, sizeof(field.name));
189,689✔
998
    if (i < (pTableMeta)->tableInfo.numOfColumns) {
189,689✔
999
      SML_CHECK_NULL(taosArrayPush(*pColumns, &field));
150,070✔
1000
    } else {
1001
      SML_CHECK_NULL(taosArrayPush(*pTags, &field));
229,308✔
1002
    }
1003
  }
1004
END:
17,989✔
1005
  RETURN
17,989✔
1006
}
1007
static int32_t smlModifyTag(SSmlHandle *info, SHashObj *hashTmpCheck, SHashObj *hashTmp, SRequestConnInfo *conn,
171,469✔
1008
                            SSmlSTableMeta *sTableData, SName *pName, STableMeta **pTableMeta) {
1009
  ESchemaAction action = SCHEMA_ACTION_NULL;
171,469✔
1010
  SArray       *pColumns = NULL;
171,469✔
1011
  SArray       *pTags = NULL;
171,469✔
1012
  int32_t       code = 0;
171,469✔
1013
  int32_t       lino = 0;
171,469✔
1014
  SML_CHECK_CODE(
171,469✔
1015
      smlProcessSchemaAction(info, (*pTableMeta)->schema, hashTmp, sTableData->tags, hashTmpCheck, &action, true));
1016

1017
  if (action != SCHEMA_ACTION_NULL) {
169,920✔
1018
    SML_CHECK_CODE(smlCheckAuth(info, conn, pName->tname, AUTH_TYPE_WRITE));
13,134✔
1019
    uDebug("SML:0x%" PRIx64 ", %s change table tag, table:%s, action:%d", info->id, __FUNCTION__, pName->tname, action);
13,134✔
1020
    SML_CHECK_CODE(smlBuildFields(&pColumns, &pTags, *pTableMeta, sTableData));
13,134✔
1021
    SML_CHECK_CODE(smlBuildFieldsList(info, (*pTableMeta)->schema, hashTmp, sTableData->tags, pTags,
13,134✔
1022
                                      (*pTableMeta)->tableInfo.numOfColumns, true));
1023

1024
    SML_CHECK_CODE(smlSendMetaMsg(info, pName, pColumns, &pTags, (*pTableMeta), action));
10,622✔
1025

1026
    info->cost.numOfAlterTagSTables++;
10,565✔
1027
    taosMemoryFreeClear(*pTableMeta);
10,565✔
1028
    SML_CHECK_CODE(catalogRefreshTableMeta(info->pCatalog, conn, pName, -1));
10,565✔
1029
    SML_CHECK_CODE(catalogGetSTableMeta(info->pCatalog, conn, pName, pTableMeta));
10,247✔
1030
  }
1031

1032
END:
171,094✔
1033
  taosArrayDestroy(pColumns);
171,469✔
1034
  taosArrayDestroy(pTags);
171,469✔
1035
  RETURN
171,469✔
1036
}
1037

1038
static int32_t smlModifyCols(SSmlHandle *info, SHashObj *hashTmpCheck, SHashObj *hashTmp, SRequestConnInfo *conn,
167,033✔
1039
                             SSmlSTableMeta *sTableData, SName *pName, STableMeta **pTableMeta) {
1040
  ESchemaAction action = SCHEMA_ACTION_NULL;
167,033✔
1041
  SArray       *pColumns = NULL;
167,033✔
1042
  SArray       *pTags = NULL;
167,033✔
1043
  int32_t       code = 0;
167,033✔
1044
  int32_t       lino = 0;
167,033✔
1045
  SML_CHECK_CODE(
167,033✔
1046
      smlProcessSchemaAction(info, (*pTableMeta)->schema, hashTmp, sTableData->cols, hashTmpCheck, &action, false));
1047

1048
  if (action != SCHEMA_ACTION_NULL) {
165,434✔
1049
    SML_CHECK_CODE(smlCheckAuth(info, conn, pName->tname, AUTH_TYPE_WRITE));
4,855✔
1050
    uDebug("SML:0x%" PRIx64 ", %s change table col, table:%s, action:%d", info->id, __FUNCTION__, pName->tname, action);
4,855✔
1051
    SML_CHECK_CODE(smlBuildFields(&pColumns, &pTags, *pTableMeta, sTableData));
4,855✔
1052
    SML_CHECK_CODE(smlBuildFieldsList(info, (*pTableMeta)->schema, hashTmp, sTableData->cols, pColumns,
4,855✔
1053
                                      (*pTableMeta)->tableInfo.numOfColumns, false));
1054

1055
    SML_CHECK_CODE(smlSendMetaMsg(info, pName, pColumns, &pTags, (*pTableMeta), action));
3,599✔
1056

1057
    info->cost.numOfAlterColSTables++;
3,599✔
1058
    taosMemoryFreeClear(*pTableMeta);
3,599✔
1059
    SML_CHECK_CODE(catalogRefreshTableMeta(info->pCatalog, conn, pName, -1));
3,599✔
1060
    SML_CHECK_CODE(catalogGetSTableMeta(info->pCatalog, conn, pName, pTableMeta));
3,599✔
1061
  }
1062

1063
END:
167,033✔
1064
  taosArrayDestroy(pColumns);
167,033✔
1065
  taosArrayDestroy(pTags);
167,033✔
1066
  RETURN
167,033✔
1067
}
1068

1069
static int32_t smlBuildTempHash(SHashObj *hashTmp, STableMeta *pTableMeta, uint16_t start, uint16_t end) {
342,938✔
1070
  int32_t code = 0;
342,938✔
1071
  int32_t lino = 0;
342,938✔
1072
  for (uint16_t i = start; i < end; i++) {
4,625,291✔
1073
    SML_CHECK_CODE(
4,282,353✔
1074
        taosHashPut(hashTmp, pTableMeta->schema[i].name, strlen(pTableMeta->schema[i].name), &i, SHORT_BYTES));
1075
  }
1076

1077
END:
342,938✔
1078
  return code;
342,938✔
1079
}
1080

1081
static int32_t smlModifyDBSchemas(SSmlHandle *info) {
689,275✔
1082
  uDebug("SML:0x%" PRIx64 ", %s start, format:%d, needModifySchema:%d", info->id, __FUNCTION__, info->dataFormat,
689,275✔
1083
         info->needModifySchema);
1084
  if (info->dataFormat && !info->needModifySchema) {
689,275✔
1085
    return TSDB_CODE_SUCCESS;
150,407✔
1086
  }
1087
  int32_t     code = 0;
538,868✔
1088
  int32_t     lino = 0;
538,868✔
1089
  SHashObj   *colHashTmp = NULL;
538,868✔
1090
  SHashObj   *tagHashTmp = NULL;
538,868✔
1091
  STableMeta *pTableMeta = NULL;
538,868✔
1092

1093
  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
538,868✔
1094
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));
538,868✔
1095

1096
  SRequestConnInfo conn = {0};
538,868✔
1097
  conn.pTrans = info->taos->pAppInfo->pTransporter;
538,868✔
1098
  conn.requestId = info->pRequest->requestId;
538,868✔
1099
  conn.requestObjRefId = info->pRequest->self;
538,868✔
1100
  conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
538,868✔
1101

1102
  SSmlSTableMeta **tmp = (SSmlSTableMeta **)taosHashIterate(info->superTables, NULL);
538,868✔
1103
  while (tmp) {
1,046,587✔
1104
    SSmlSTableMeta *sTableData = *tmp;
544,452✔
1105
    bool            needCheckMeta = false;  // for multi thread
544,452✔
1106

1107
    size_t superTableLen = 0;
544,452✔
1108
    void  *superTable = taosHashGetKey(tmp, &superTableLen);
544,452✔
1109
    char  *measure = taosMemoryMalloc(superTableLen);
544,452✔
1110
    SML_CHECK_NULL(measure);
571,150✔
1111
    (void)memcpy(measure, superTable, superTableLen);
544,452✔
1112
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
544,452✔
1113
      PROCESS_SLASH_IN_MEASUREMENT(measure, superTableLen);
1,848,270✔
1114
    }
1115
    smlStrReplace(measure, superTableLen);
544,452✔
1116
    size_t nameLen = TMIN(superTableLen, TSDB_TABLE_NAME_LEN - 1);
544,452✔
1117
    (void)memcpy(pName.tname, measure, nameLen);
544,452✔
1118
    pName.tname[nameLen] = '\0';
544,452✔
1119
    taosMemoryFree(measure);
544,452✔
1120

1121
    pTableMeta = NULL;
544,452✔
1122
    code = catalogGetSTableMeta(info->pCatalog, &conn, &pName, &pTableMeta);
544,452✔
1123

1124
    if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST || code == TSDB_CODE_MND_STB_NOT_EXIST) {
544,452✔
1125
      SML_CHECK_CODE(smlCreateTable(info, &conn, sTableData, &pName, &pTableMeta));
372,983✔
1126
    } else if (code == TSDB_CODE_SUCCESS) {
171,469✔
1127
      if (smlIsPKTable(pTableMeta)) {
171,469✔
1128
        SML_CHECK_CODE(TSDB_CODE_SML_NOT_SUPPORT_PK);
×
1129
      }
1130

1131
      colHashTmp = taosHashInit(pTableMeta->tableInfo.numOfColumns, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY),
171,469✔
1132
                                true, HASH_NO_LOCK);
1133
      tagHashTmp = taosHashInit(pTableMeta->tableInfo.numOfTags, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY),
171,469✔
1134
                                true, HASH_NO_LOCK);
1135
      SML_CHECK_NULL(colHashTmp);
171,469✔
1136
      SML_CHECK_NULL(tagHashTmp);
171,469✔
1137
      SML_CHECK_CODE(smlBuildTempHash(tagHashTmp, pTableMeta, pTableMeta->tableInfo.numOfColumns,
171,469✔
1138
                                      pTableMeta->tableInfo.numOfColumns + pTableMeta->tableInfo.numOfTags));
1139
      SML_CHECK_CODE(smlBuildTempHash(colHashTmp, pTableMeta, 0, pTableMeta->tableInfo.numOfColumns));
171,469✔
1140

1141
      SML_CHECK_CODE(smlModifyTag(info, colHashTmp, tagHashTmp, &conn, sTableData, &pName, &pTableMeta));
171,469✔
1142
      SML_CHECK_CODE(smlModifyCols(info, tagHashTmp, colHashTmp, &conn, sTableData, &pName, &pTableMeta));
167,033✔
1143

1144
      needCheckMeta = true;
164,178✔
1145
      taosHashCleanup(colHashTmp);
164,178✔
1146
      taosHashCleanup(tagHashTmp);
164,178✔
1147
      colHashTmp = NULL;
164,178✔
1148
      tagHashTmp = NULL;
164,178✔
1149
    } else {
1150
      uError("SML:0x%" PRIx64 ", %s load table meta error:%s", info->id, __FUNCTION__, tstrerror(code));
×
1151
      goto END;
×
1152
    }
1153

1154
    if (needCheckMeta) {
508,242✔
1155
      SML_CHECK_CODE(smlCheckMeta(&(pTableMeta->schema[pTableMeta->tableInfo.numOfColumns]),
164,178✔
1156
                                  pTableMeta->tableInfo.numOfTags, sTableData->tags));
1157
      SML_CHECK_CODE(smlCheckMeta(&(pTableMeta->schema[0]), pTableMeta->tableInfo.numOfColumns, sTableData->cols));
163,655✔
1158
    }
1159

1160
    taosMemoryFreeClear(sTableData->tableMeta);
507,719✔
1161
    sTableData->tableMeta = pTableMeta;
507,719✔
1162
    uDebug("SML:0x%" PRIx64 ", %s modify schema uid:%" PRIu64 ", sversion:%d, tversion:%d", info->id, __FUNCTION__,
507,719✔
1163
           pTableMeta->uid, pTableMeta->sversion, pTableMeta->tversion);
1164
    tmp = (SSmlSTableMeta **)taosHashIterate(info->superTables, tmp);
507,719✔
1165
  }
1166
  uDebug("SML:0x%" PRIx64 ", %s end success, format:%d, needModifySchema:%d", info->id, __FUNCTION__, info->dataFormat,
502,135✔
1167
         info->needModifySchema);
1168

1169
  return TSDB_CODE_SUCCESS;
502,135✔
1170

1171
END:
36,733✔
1172
  taosHashCancelIterate(info->superTables, tmp);
36,733✔
1173
  taosHashCleanup(colHashTmp);
36,733✔
1174
  taosHashCleanup(tagHashTmp);
36,733✔
1175
  taosMemoryFreeClear(pTableMeta);
36,733✔
1176
  (void)catalogRefreshTableMeta(info->pCatalog, &conn, &pName, 1);  // ignore refresh meta code if there is an error
36,733✔
1177
  uError("SML:0x%" PRIx64 ", %s end failed:%d:%s, format:%d, needModifySchema:%d", info->id, __FUNCTION__, code,
36,733✔
1178
         tstrerror(code), info->dataFormat, info->needModifySchema);
1179

1180
  return code;
36,733✔
1181
}
1182

1183
static int32_t smlInsertMeta(SHashObj *metaHash, SArray *metaArray, SArray *cols, SHashObj *checkDuplicate) {
1,033,019✔
1184
  int32_t code = 0;
1,033,019✔
1185
  int32_t lino = 0;
1,033,019✔
1186
  terrno = 0;
1,033,019✔
1187
  for (int16_t i = 0; i < taosArrayGetSize(cols); ++i) {
12,359,879✔
1188
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
11,331,624✔
1189
    SML_CHECK_NULL(kv);
11,331,659✔
1190
    int ret = taosHashPut(metaHash, kv->key, kv->keyLen, &i, SHORT_BYTES);
11,331,659✔
1191
    if (ret == 0) {
11,333,359✔
1192
      SML_CHECK_NULL(taosArrayPush(metaArray, kv));
11,328,883✔
1193
      if (taosHashGet(checkDuplicate, kv->key, kv->keyLen) != NULL) {
11,328,883✔
1194
        uError("%s duplicated column %s", __FUNCTION__, kv->key);
1,549✔
1195
        SML_CHECK_CODE(TSDB_CODE_PAR_DUPLICATED_COLUMN);
1,549✔
1196
      }
1197
    } else if (terrno == TSDB_CODE_DUP_KEY) {
3,895✔
1198
      uError("%s duplicated column %s", __FUNCTION__, kv->key);
3,895✔
1199
      return TSDB_CODE_PAR_DUPLICATED_COLUMN;
3,895✔
1200
    }
1201
  }
1202

1203
END:
1,029,021✔
1204
  RETURN
1,029,021✔
1205
}
1206

1207
static int32_t smlUpdateMeta(SHashObj *metaHash, SArray *metaArray, SArray *cols, bool isTag, SSmlMsgBuf *msg,
100,812,818✔
1208
                             SHashObj *checkDuplicate) {
1209
  int32_t code = 0;
100,812,818✔
1210
  int32_t lino = 0;
100,812,818✔
1211
  for (int i = 0; i < taosArrayGetSize(cols); ++i) {
444,431,072✔
1212
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
345,003,179✔
1213
    SML_CHECK_NULL(kv);
344,668,014✔
1214
    int16_t *index = (int16_t *)taosHashGet(metaHash, kv->key, kv->keyLen);
344,668,014✔
1215
    if (index) {
344,871,928✔
1216
      SSmlKv *value = (SSmlKv *)taosArrayGet(metaArray, *index);
344,862,028✔
1217
      SML_CHECK_NULL(value);
345,857,100✔
1218

1219
      if (isTag) {
345,871,859✔
1220
        if (kv->length > value->length) {
106,197,488✔
1221
          value->length = kv->length;
35,766✔
1222
        }
1223
        continue;
106,228,961✔
1224
      }
1225
      if (kv->type != value->type) {
239,674,371✔
1226
        smlBuildInvalidDataMsg(msg, "the type is not the same like before", kv->key);
×
1227
        SML_CHECK_CODE(TSDB_CODE_SML_NOT_SAME_TYPE);
×
1228
      }
1229

1230
      if (IS_VAR_DATA_TYPE(kv->type) && (kv->length > value->length)) {  // update string len, if bigger
239,951,014✔
1231
        value->length = kv->length;
2,327,524✔
1232
      }
1233
    } else {
1234
      size_t tmp = taosArrayGetSize(metaArray);
10,002✔
1235
      if (tmp > INT16_MAX) {
10,002✔
1236
        smlBuildInvalidDataMsg(msg, "too many cols or tags", kv->key);
×
1237
        SML_CHECK_CODE(TSDB_CODE_SML_INVALID_DATA);
×
1238
      }
1239
      int16_t size = tmp;
10,002✔
1240
      SML_CHECK_CODE(taosHashPut(metaHash, kv->key, kv->keyLen, &size, SHORT_BYTES));
10,002✔
1241
      SML_CHECK_NULL(taosArrayPush(metaArray, kv));
10,002✔
1242
      if (taosHashGet(checkDuplicate, kv->key, kv->keyLen) != NULL) {
10,002✔
1243
        uError("%s duplicated column %s", __FUNCTION__, kv->key);
×
1244
        SML_CHECK_CODE(TSDB_CODE_PAR_DUPLICATED_COLUMN);
×
1245
      }
1246
    }
1247
  }
1248

1249
END:
99,759,868✔
1250
  RETURN
100,977,475✔
1251
}
1252

1253
void smlDestroyTableInfo(void *para) {
1,426,937✔
1254
  SSmlTableInfo *tag = *(SSmlTableInfo **)para;
1,426,937✔
1255
  for (size_t i = 0; i < taosArrayGetSize(tag->cols); i++) {
52,447,497✔
1256
    SHashObj *kvHash = (SHashObj *)taosArrayGetP(tag->cols, i);
51,080,223✔
1257
    taosHashCleanup(kvHash);
51,019,366✔
1258
  }
1259

1260
  taosArrayDestroy(tag->cols);
1,427,040✔
1261
  taosArrayDestroyEx(tag->tags, freeSSmlKv);
1,427,312✔
1262
  taosMemoryFree(tag);
1,427,040✔
1263
}
1,426,768✔
1264

1265
void freeSSmlKv(void *data) {
265,093,342✔
1266
  SSmlKv *kv = (SSmlKv *)data;
265,093,342✔
1267
  if (kv->keyEscaped) taosMemoryFreeClear(kv->key);
265,093,342✔
1268
  if (kv->valueEscaped) taosMemoryFreeClear(kv->value);
265,182,735✔
1269
#ifdef USE_GEOS
1270
  if (kv->type == TSDB_DATA_TYPE_GEOMETRY) geosFreeBuffer((void *)(kv->value));
264,732,385✔
1271
#endif
1272
  if (kv->type == TSDB_DATA_TYPE_VARBINARY) taosMemoryFreeClear(kv->value);
264,817,186✔
1273
}
263,983,246✔
1274

1275
void smlDestroyInfo(SSmlHandle *info) {
1,536,569✔
1276
  if (info == NULL) return;
1,536,569✔
1277

1278
  taosHashCleanup(info->pVgHash);
768,449✔
1279
  taosHashCleanup(info->childTables);
768,551✔
1280
  taosHashCleanup(info->superTables);
768,551✔
1281
  taosHashCleanup(info->tableUids);
768,551✔
1282

1283
  for (int i = 0; i < taosArrayGetSize(info->tagJsonArray); i++) {
768,551✔
1284
    cJSON *tags = (cJSON *)taosArrayGetP(info->tagJsonArray, i);
×
1285
    cJSON_Delete(tags);
×
1286
  }
1287
  taosArrayDestroy(info->tagJsonArray);
768,551✔
1288

1289
  for (int i = 0; i < taosArrayGetSize(info->valueJsonArray); i++) {
768,551✔
1290
    cJSON *value = (cJSON *)taosArrayGetP(info->valueJsonArray, i);
×
1291
    cJSON_Delete(value);
×
1292
  }
1293
  taosArrayDestroy(info->valueJsonArray);
768,551✔
1294

1295
  taosArrayDestroyEx(info->preLineTagKV, freeSSmlKv);
768,551✔
1296
  taosArrayDestroyP(info->escapedStringList, NULL);
768,551✔
1297

1298
  if (!info->dataFormat) {
768,551✔
1299
    for (int i = 0; i < info->lineNum; i++) {
51,693,451✔
1300
      taosArrayDestroyEx(info->lines[i].colArray, freeSSmlKv);
51,142,884✔
1301
      if (info->lines[i].measureTagsLen != 0 && info->protocol != TSDB_SML_LINE_PROTOCOL) {
51,084,280✔
1302
        taosMemoryFree(info->lines[i].measureTag);
296,159✔
1303
      }
1304
    }
1305
    taosMemoryFree(info->lines);
554,495✔
1306
  }
1307
  if (info->protocol == TSDB_SML_JSON_PROTOCOL) {
768,551✔
1308
    taosMemoryFreeClear(info->preLine.tags);
324,283✔
1309
  }
1310
  cJSON_Delete(info->root);
768,551✔
1311
  taosMemoryFreeClear(info);
768,551✔
1312
}
1313

1314
int32_t smlBuildSmlInfo(TAOS *taos, SSmlHandle **handle) {
768,327✔
1315
  int32_t     code = TSDB_CODE_SUCCESS;
768,327✔
1316
  int32_t     lino = 0;
768,327✔
1317
  SSmlHandle *info = (SSmlHandle *)taosMemoryCalloc(1, sizeof(SSmlHandle));
768,327✔
1318
  SML_CHECK_NULL(info);
768,289✔
1319
  if (taos != NULL) {
768,289✔
1320
    info->taos = acquireTscObj(*(int64_t *)taos);
768,289✔
1321
    SML_CHECK_NULL(info->taos);
768,551✔
1322
    SML_CHECK_CODE(catalogGetHandle(info->taos->pAppInfo->clusterId, &info->pCatalog));
768,279✔
1323
  }
1324

1325
  info->pVgHash = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_NO_LOCK);
768,120✔
1326
  SML_CHECK_NULL(info->pVgHash);
768,392✔
1327
  info->childTables = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
768,392✔
1328
  SML_CHECK_NULL(info->childTables);
768,551✔
1329
  info->tableUids = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
768,279✔
1330
  SML_CHECK_NULL(info->tableUids);
768,551✔
1331
  info->superTables = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
768,279✔
1332
  SML_CHECK_NULL(info->superTables);
768,551✔
1333
  taosHashSetFreeFp(info->superTables, smlDestroySTableMeta);
768,279✔
1334
  taosHashSetFreeFp(info->childTables, smlDestroyTableInfo);
768,551✔
1335

1336
  info->id = smlGenId();
768,551✔
1337
  SML_CHECK_CODE(smlInitHandle(&info->pQuery));
768,279✔
1338
  info->dataFormat = true;
768,551✔
1339
  info->tagJsonArray = taosArrayInit(8, POINTER_BYTES);
768,551✔
1340
  SML_CHECK_NULL(info->tagJsonArray);
768,551✔
1341
  info->valueJsonArray = taosArrayInit(8, POINTER_BYTES);
768,551✔
1342
  SML_CHECK_NULL(info->valueJsonArray);
768,551✔
1343
  info->preLineTagKV = taosArrayInit(8, sizeof(SSmlKv));
768,551✔
1344
  SML_CHECK_NULL(info->preLineTagKV);
768,120✔
1345
  info->escapedStringList = taosArrayInit(8, POINTER_BYTES);
768,120✔
1346
  SML_CHECK_NULL(info->escapedStringList);
768,551✔
1347

1348
  *handle = info;
768,177✔
1349
  info = NULL;
768,177✔
1350

1351
END:
768,177✔
1352
  smlDestroyInfo(info);
768,177✔
1353
  RETURN
768,120✔
1354
}
1355

1356
static int32_t smlPushCols(SArray *colsArray, SArray *cols) {
50,580,882✔
1357
  int32_t   code = TSDB_CODE_SUCCESS;
50,580,882✔
1358
  int32_t   lino = 0;
50,580,882✔
1359
  SHashObj *kvHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
50,580,882✔
1360
  SML_CHECK_NULL(kvHash);
50,994,007✔
1361
  for (size_t i = 0; i < taosArrayGetSize(cols); i++) {
298,701,927✔
1362
    SSmlKv *kv = (SSmlKv *)taosArrayGet(cols, i);
247,188,270✔
1363
    SML_CHECK_NULL(kv);
246,246,529✔
1364
    terrno = 0;
246,246,529✔
1365
    code = taosHashPut(kvHash, kv->key, kv->keyLen, &kv, POINTER_BYTES);
247,216,304✔
1366
    if (terrno == TSDB_CODE_DUP_KEY) {
247,403,446✔
1367
      uError("%s duplicated column %s", __FUNCTION__, kv->key);
1,314✔
1368
      SML_CHECK_CODE(TSDB_CODE_PAR_DUPLICATED_COLUMN);
362✔
1369
    }
1370
    SML_CHECK_CODE(code);
247,495,963✔
1371
  }
1372

1373
  SML_CHECK_NULL(taosArrayPush(colsArray, &kvHash));
50,963,873✔
1374
  return code;
50,963,873✔
1375
END:
359✔
1376
  taosHashCleanup(kvHash);
1,314✔
1377
  RETURN
1,314✔
1378
}
1379

1380
static int32_t smlParseEnd(SSmlHandle *info) {
670,724✔
1381
  uDebug("SML:0x%" PRIx64 ", %s start, format:%d, linenum:%d", info->id, __FUNCTION__, info->dataFormat, info->lineNum);
670,724✔
1382
  int32_t code = 0;
670,724✔
1383
  int32_t lino = 0;
670,724✔
1384
  if (info->dataFormat) return TSDB_CODE_SUCCESS;
670,724✔
1385

1386
  for (int32_t i = 0; i < info->lineNum; i++) {
51,268,195✔
1387
    SSmlLineInfo  *elements = info->lines + i;
51,073,114✔
1388
    SSmlTableInfo *tinfo = NULL;
51,117,925✔
1389
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
51,117,925✔
1390
      SSmlTableInfo **tmp =
1391
          (SSmlTableInfo **)taosHashGet(info->childTables, elements->measure, elements->measureTagsLen);
50,463,215✔
1392
      if (tmp) tinfo = *tmp;
50,467,827✔
1393
    } else {
1394
      SSmlTableInfo **tmp = (SSmlTableInfo **)taosHashGet(info->childTables, elements->measureTag,
659,552✔
1395
                                                          elements->measureLen + elements->tagsLen);
659,552✔
1396
      if (tmp) tinfo = *tmp;
551,070✔
1397
    }
1398

1399
    if (tinfo == NULL) {
50,769,190✔
1400
      uError("SML:0x%" PRIx64 ", get oneTable failed, line num:%d", info->id, i);
×
1401
      smlBuildInvalidDataMsg(&info->msgBuf, "get oneTable failed", elements->measure);
×
1402
      return TSDB_CODE_SML_INVALID_DATA;
×
1403
    }
1404

1405
    if (taosArrayGetSize(tinfo->tags) > TSDB_MAX_TAGS) {
50,769,190✔
1406
      smlBuildInvalidDataMsg(&info->msgBuf, "too many tags than 128", NULL);
2,512✔
1407
      return TSDB_CODE_PAR_INVALID_TAGS_NUM;
2,512✔
1408
    }
1409

1410
    if (taosArrayGetSize(elements->colArray) + taosArrayGetSize(tinfo->tags) > TSDB_MAX_COLUMNS) {
51,087,204✔
1411
      smlBuildInvalidDataMsg(&info->msgBuf, "too many columns than 4096", NULL);
628✔
1412
      return TSDB_CODE_PAR_TOO_MANY_COLUMNS;
628✔
1413
    }
1414

1415
    SML_CHECK_CODE(smlPushCols(tinfo->cols, elements->colArray));
51,112,951✔
1416

1417
    SSmlSTableMeta **tableMeta =
1418
        (SSmlSTableMeta **)taosHashGet(info->superTables, elements->measure, elements->measureLen);
50,876,014✔
1419
    if (tableMeta) {  // update meta
51,088,946✔
1420
      uDebug("SML:0x%" PRIx64 ", %s update meta, format:%d, linenum:%d", info->id, __FUNCTION__, info->dataFormat,
50,570,751✔
1421
             info->lineNum);
1422
      SML_CHECK_CODE(smlUpdateMeta((*tableMeta)->colHash, (*tableMeta)->cols, elements->colArray, false, &info->msgBuf,
50,570,751✔
1423
                                   (*tableMeta)->tagHash));
1424
      SML_CHECK_CODE(smlUpdateMeta((*tableMeta)->tagHash, (*tableMeta)->tags, tinfo->tags, true, &info->msgBuf,
50,545,663✔
1425
                                   (*tableMeta)->colHash));
1426
    } else {
1427
      uDebug("SML:0x%" PRIx64 ", %s add meta, format:%d, linenum:%d", info->id, __FUNCTION__, info->dataFormat,
518,195✔
1428
             info->lineNum);
1429
      SSmlSTableMeta *meta = NULL;
518,195✔
1430
      SML_CHECK_CODE(smlBuildSTableMeta(info->dataFormat, &meta));
518,195✔
1431
      code = taosHashPut(info->superTables, elements->measure, elements->measureLen, &meta, POINTER_BYTES);
518,457✔
1432
      if (code != TSDB_CODE_SUCCESS) {
518,457✔
1433
        smlDestroySTableMeta(&meta);
×
1434
        SML_CHECK_CODE(code);
×
1435
      }
1436
      SML_CHECK_CODE(smlInsertMeta(meta->tagHash, meta->tags, tinfo->tags, NULL));
518,457✔
1437
      SML_CHECK_CODE(smlInsertMeta(meta->colHash, meta->cols, elements->colArray, meta->tagHash));
514,562✔
1438
    }
1439
  }
1440
  uDebug("SML:0x%" PRIx64 ", %s end, format:%d, linenum:%d", info->id, __FUNCTION__, info->dataFormat, info->lineNum);
507,270✔
1441

1442
END:
160,502✔
1443
  RETURN
514,028✔
1444
}
1445

1446
static int32_t smlInsertData(SSmlHandle *info) {
652,542✔
1447
  int32_t         code = TSDB_CODE_SUCCESS;
652,542✔
1448
  int32_t         lino = 0;
652,542✔
1449
  char           *measure = NULL;
652,542✔
1450
  SSmlTableInfo **oneTable = NULL;
652,542✔
1451
  uDebug("SML:0x%" PRIx64 ", %s start, format:%d", info->id, __FUNCTION__, info->dataFormat);
652,542✔
1452

1453
  if (info->pRequest->dbList == NULL) {
652,542✔
1454
    info->pRequest->dbList = taosArrayInit(1, TSDB_DB_FNAME_LEN);
652,542✔
1455
    SML_CHECK_NULL(info->pRequest->dbList);
652,542✔
1456
  }
1457
  char *data = (char *)taosArrayReserve(info->pRequest->dbList, 1);
652,542✔
1458
  SML_CHECK_NULL(data);
652,542✔
1459
  SName pName = {TSDB_TABLE_NAME_T, info->taos->acctId, {0}, {0}};
652,542✔
1460
  tstrncpy(pName.dbname, info->pRequest->pDb, sizeof(pName.dbname));
652,542✔
1461
  (void)tNameGetFullDbName(&pName, data);  // ignore
652,542✔
1462

1463
  oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, NULL);
652,542✔
1464
  while (oneTable) {
1,915,941✔
1465
    SSmlTableInfo *tableData = *oneTable;
1,264,085✔
1466

1467
    int measureLen = tableData->sTableNameLen;
1,264,085✔
1468
    measure = (char *)taosMemoryMalloc(tableData->sTableNameLen);
1,264,357✔
1469
    SML_CHECK_NULL(measure);
1,264,357✔
1470
    (void)memcpy(measure, tableData->sTableName, tableData->sTableNameLen);
1,264,357✔
1471
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
1,264,357✔
1472
      PROCESS_SLASH_IN_MEASUREMENT(measure, measureLen);
7,659,078✔
1473
    }
1474
    smlStrReplace(measure, measureLen);
1,264,357✔
1475
    (void)memcpy(pName.tname, measure, measureLen);
1,264,357✔
1476
    pName.tname[measureLen] = '\0';
1,264,357✔
1477

1478
    if (info->pRequest->tableList == NULL) {
1,264,357✔
1479
      info->pRequest->tableList = taosArrayInit(1, sizeof(SName));
652,542✔
1480
      SML_CHECK_NULL(info->pRequest->tableList);
652,542✔
1481
    }
1482
    SML_CHECK_NULL(taosArrayPush(info->pRequest->tableList, &pName));
2,528,714✔
1483
    tstrncpy(pName.tname, tableData->childTableName, sizeof(pName.tname));
1,264,357✔
1484

1485
    SRequestConnInfo conn = {0};
1,264,357✔
1486
    conn.pTrans = info->taos->pAppInfo->pTransporter;
1,264,357✔
1487
    conn.requestId = info->pRequest->requestId;
1,264,357✔
1488
    conn.requestObjRefId = info->pRequest->self;
1,264,357✔
1489
    conn.mgmtEps = getEpSet_s(&info->taos->pAppInfo->mgmtEp);
1,264,357✔
1490

1491
    SML_CHECK_CODE(smlCheckAuth(info, &conn, pName.tname, AUTH_TYPE_WRITE));
1,264,357✔
1492

1493
    SVgroupInfo vg = {0};
1,262,039✔
1494
    SML_CHECK_CODE(catalogGetTableHashVgroup(info->pCatalog, &conn, &pName, &vg));
1,263,399✔
1495
    SML_CHECK_CODE(taosHashPut(info->pVgHash, (const char *)&vg.vgId, sizeof(vg.vgId), (char *)&vg, sizeof(vg)));
1,263,671✔
1496

1497
    SSmlSTableMeta **pMeta =
1498
        (SSmlSTableMeta **)taosHashGet(info->superTables, tableData->sTableName, tableData->sTableNameLen);
1,263,671✔
1499
    if (unlikely(NULL == pMeta || NULL == *pMeta || NULL == (*pMeta)->tableMeta)) {
1,263,671✔
1500
      uError("SML:0x%" PRIx64 ", %s NULL == pMeta. table name:%s", info->id, __FUNCTION__, tableData->childTableName);
×
1501
      SML_CHECK_CODE(TSDB_CODE_SML_INTERNAL_ERROR);
×
1502
    }
1503

1504
    // use tablemeta of stable to save vgid and uid of child table
1505
    (*pMeta)->tableMeta->vgId = vg.vgId;
1,263,671✔
1506
    (*pMeta)->tableMeta->uid = tableData->uid;  // one table merge data block together according uid
1,263,671✔
1507
    uDebug("SML:0x%" PRIx64 ", %s table:%s, uid:%" PRIu64 ", format:%d", info->id, __FUNCTION__, pName.tname,
1,263,671✔
1508
           tableData->uid, info->dataFormat);
1509

1510
    SML_CHECK_CODE(smlBindData(info->pQuery, info->dataFormat, tableData->tags, (*pMeta)->cols, tableData->cols,
1,263,671✔
1511
                               (*pMeta)->tableMeta, tableData->childTableName, measure, measureLen, info->ttl,
1512
                               info->msgBuf.buf, info->msgBuf.len, info->taos->optionInfo.charsetCxt));
1513
    taosMemoryFreeClear(measure);
1,263,298✔
1514
    oneTable = (SSmlTableInfo **)taosHashIterate(info->childTables, oneTable);
1,263,399✔
1515
  }
1516

1517
  SML_CHECK_CODE(smlBuildOutput(info->pQuery, info->pVgHash));
651,856✔
1518
  info->cost.insertRpcTime = taosGetTimestampUs();
651,856✔
1519

1520
  SAppClusterSummary *pActivity = &info->taos->pAppInfo->summary;
651,856✔
1521
  (void)atomic_add_fetch_64((int64_t *)&pActivity->numOfInsertsReq, 1);  // no need to check return code
651,587✔
1522

1523
  launchQueryImpl(info->pRequest, info->pQuery, true, NULL);  // no need to check return code
651,856✔
1524

1525
  uDebug("SML:0x%" PRIx64 ", %s end, format:%d, code:%d,%s", info->id, __FUNCTION__, info->dataFormat,
651,697✔
1526
         info->pRequest->code, tstrerror(info->pRequest->code));
1527

1528
  return info->pRequest->code;
651,856✔
1529

1530
END:
958✔
1531
  taosMemoryFree(measure);
686✔
1532
  taosHashCancelIterate(info->childTables, oneTable);
686✔
1533
  RETURN
686✔
1534
}
1535

1536
static void smlPrintStatisticInfo(SSmlHandle *info) {
767,551✔
1537
  uDebug(
767,551✔
1538
      "SML:0x%" PRIx64
1539
      " 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 \
1540
        parse cost:%" PRId64 ",schema cost:%" PRId64 ",bind cost:%" PRId64 ",rpc cost:%" PRId64 ",total cost:%" PRId64,
1541
      info->id, info->cost.code, tstrerror(info->cost.code), info->cost.lineNum, info->cost.numOfSTables,
1542
      info->cost.numOfCTables, info->cost.numOfCreateSTables, info->cost.numOfAlterTagSTables,
1543
      info->cost.numOfAlterColSTables, info->cost.schemaTime - info->cost.parseTime,
1544
      info->cost.insertBindTime - info->cost.schemaTime, info->cost.insertRpcTime - info->cost.insertBindTime,
1545
      info->cost.endTime - info->cost.insertRpcTime, info->cost.endTime - info->cost.parseTime);
1546
}
767,551✔
1547

1548
int32_t smlClearForRerun(SSmlHandle *info) {
554,223✔
1549
  int32_t code = 0;
554,223✔
1550
  int32_t lino = 0;
554,223✔
1551
  info->reRun = false;
554,223✔
1552

1553
  taosHashClear(info->childTables);
554,495✔
1554
  taosHashClear(info->superTables);
554,223✔
1555
  taosHashClear(info->tableUids);
554,495✔
1556

1557
  if (!info->dataFormat) {
554,226✔
1558
    info->lines = (SSmlLineInfo *)taosMemoryCalloc(info->lineNum, sizeof(SSmlLineInfo));
554,495✔
1559
    SML_CHECK_NULL(info->lines);
554,495✔
1560
  }
1561

1562
  taosArrayClearP(info->escapedStringList, NULL);
553,954✔
1563
  if (info->protocol == TSDB_SML_JSON_PROTOCOL) {
554,495✔
1564
    taosMemoryFreeClear(info->preLine.tags);
196,264✔
1565
  }
1566
  (void)memset(&info->preLine, 0, sizeof(SSmlLineInfo));
554,495✔
1567
  info->currSTableMeta = NULL;
554,495✔
1568
  info->currTableDataCtx = NULL;
554,495✔
1569

1570
  SVnodeModifyOpStmt *stmt = (SVnodeModifyOpStmt *)(info->pQuery->pRoot);
554,495✔
1571
  stmt->freeHashFunc(stmt->pTableBlockHashObj);
554,495✔
1572
  stmt->pTableBlockHashObj = taosHashInit(16, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
554,223✔
1573
  SML_CHECK_NULL(stmt->pTableBlockHashObj);
554,495✔
1574

1575
END:
554,495✔
1576
  RETURN
554,495✔
1577
}
1578

1579
static void printRaw(int64_t id, int lineNum, int numLines, ELogLevel level, char *data, int32_t len) {
6,174✔
1580
  char *print = taosMemoryMalloc(len + 1);
6,174✔
1581
  if (print == NULL) {
6,174✔
1582
    uError("SML:0x%" PRIx64 ", smlParseLine failed. code :%d", id, terrno);
×
1583
    return;
×
1584
  }
1585
  (void)memcpy(print, data, len);
6,174✔
1586
  print[len] = '\0';
6,174✔
1587
  if (level == DEBUG_DEBUG) {
6,174✔
1588
    uDebug("SML:0x%" PRIx64 ", smlParseLine is raw, line %d/%d :%s", id, lineNum, numLines, print);
6,174✔
1589
  } else if (level == DEBUG_ERROR) {
×
1590
    uError("SML:0x%" PRIx64 ", smlParseLine failed. line %d/%d :%s", id, lineNum, numLines, print);
×
1591
  }
1592
  taosMemoryFree(print);
6,174✔
1593
}
1594

1595
static bool getLine(SSmlHandle *info, char *lines[], char **rawLine, char *rawLineEnd, int numLines, int i, char **tmp,
49,709,498✔
1596
                    int *len) {
1597
  if (lines) {
49,709,498✔
1598
    *tmp = lines[i];
50,878,539✔
1599
    *len = strlen(*tmp);
50,686,566✔
1600
  } else if (*rawLine) {
25,572✔
1601
    *tmp = *rawLine;
32,089✔
1602
    while (*rawLine < rawLineEnd) {
57,255,454✔
1603
      if (*((*rawLine)++) == '\n') {
57,246,301✔
1604
        break;
22,936✔
1605
      }
1606
      (*len)++;
57,223,365✔
1607
    }
1608
    if (IS_COMMENT(info->protocol, (*tmp)[0])) {  // this line is comment
32,089✔
1609
      return false;
343✔
1610
    }
1611
  }
1612

1613
  if (*rawLine != NULL && (uDebugFlag & DEBUG_DEBUG)) {
51,395,618✔
1614
    printRaw(info->id, i, numLines, DEBUG_DEBUG, *tmp, *len);
6,174✔
1615
  } else {
1616
    uDebug("SML:0x%" PRIx64 ", smlParseLine is not raw, line %d/%d :%s", info->id, i, numLines, *tmp);
50,879,019✔
1617
  }
1618
  return true;
51,048,992✔
1619
}
1620

1621
static int32_t smlParseJson(SSmlHandle *info, char *lines[], char *rawLine) {
324,124✔
1622
  int32_t code = TSDB_CODE_SUCCESS;
324,124✔
1623
  if (lines) {
324,124✔
1624
    code = smlParseJSONExt(info, *lines);
321,877✔
1625
  } else if (rawLine) {
2,247✔
1626
    code = smlParseJSONExt(info, rawLine);
2,247✔
1627
  }
1628
  if (code != TSDB_CODE_SUCCESS) {
324,283✔
1629
    uError("%s failed code:%d", __FUNCTION__, code);
42,451✔
1630
  }
1631
  return code;
324,283✔
1632
}
1633

1634
static int32_t smlParseStart(SSmlHandle *info, char *lines[], char *rawLine, char *rawLineEnd, int numLines) {
767,859✔
1635
  uDebug("SML:0x%" PRIx64 ", %s start", info->id, __FUNCTION__);
767,859✔
1636
  int32_t code = TSDB_CODE_SUCCESS;
767,953✔
1637
  if (info->protocol == TSDB_SML_JSON_PROTOCOL) {
767,953✔
1638
    return smlParseJson(info, lines, rawLine);
324,218✔
1639
  }
1640

1641
  char   *oldRaw = rawLine;
443,735✔
1642
  int32_t i = 0;
443,735✔
1643
  while (i < numLines) {
51,151,654✔
1644
    char *tmp = NULL;
50,762,702✔
1645
    int   len = 0;
50,625,791✔
1646
    if (!getLine(info, lines, &rawLine, rawLineEnd, numLines, i, &tmp, &len)) {
51,155,027✔
1647
      continue;
36,258✔
1648
    }
1649
    if (info->protocol == TSDB_SML_LINE_PROTOCOL) {
51,058,026✔
1650
      if (info->dataFormat) {
50,460,920✔
1651
        SSmlLineInfo element = {0};
252,099✔
1652
        code = smlParseInfluxString(info, tmp, tmp + len, &element);
252,099✔
1653
      } else {
1654
        code = smlParseInfluxString(info, tmp, tmp + len, info->lines + i);
50,206,473✔
1655
      }
1656
    } else if (info->protocol == TSDB_SML_TELNET_PROTOCOL) {
845,409✔
1657
      if (info->dataFormat) {
845,511✔
1658
        SSmlLineInfo element = {0};
608,264✔
1659
        code = smlParseTelnetString(info, (char *)tmp, (char *)tmp + len, &element);
608,264✔
1660
        if (element.measureTagsLen != 0) taosMemoryFree(element.measureTag);
608,165✔
1661
      } else {
1662
        code = smlParseTelnetString(info, (char *)tmp, (char *)tmp + len, info->lines + i);
237,247✔
1663
      }
1664
    }
1665
    if (code != TSDB_CODE_SUCCESS) {
50,587,428✔
1666
      if (rawLine != NULL) {
55,217✔
1667
        printRaw(info->id, i, numLines, DEBUG_ERROR, tmp, len);
×
1668
      } else {
1669
        uError("SML:0x%" PRIx64 ", %s failed. line %d :%s", info->id, __FUNCTION__, i, tmp);
55,217✔
1670
      }
1671
      return code;
55,217✔
1672
    }
1673
    if (info->reRun) {
50,532,211✔
1674
      uDebug("SML:0x%" PRIx64 ", %s re run", info->id, __FUNCTION__);
351,681✔
1675
      i = 0;
351,681✔
1676
      rawLine = oldRaw;
351,681✔
1677
      code = smlClearForRerun(info);
351,681✔
1678
      if (code != TSDB_CODE_SUCCESS) {
358,231✔
1679
        return code;
×
1680
      }
1681
      continue;
358,231✔
1682
    }
1683
    i++;
50,291,689✔
1684
  }
1685
  uDebug("SML:0x%" PRIx64 ", %s end", info->id, __FUNCTION__);
388,952✔
1686

1687
  return code;
389,051✔
1688
}
1689

1690
static int smlProcess(SSmlHandle *info, char *lines[], char *rawLine, char *rawLineEnd, int numLines) {
767,953✔
1691
  int32_t code = TSDB_CODE_SUCCESS;
767,953✔
1692
  int32_t lino = 0;
767,953✔
1693
  int32_t retryNum = 0;
767,953✔
1694

1695
  info->cost.parseTime = taosGetTimestampUs();
768,120✔
1696

1697
  SML_CHECK_CODE(smlParseStart(info, lines, rawLine, rawLineEnd, numLines));
768,120✔
1698
  SML_CHECK_CODE(smlParseEnd(info));
670,883✔
1699

1700
  info->cost.lineNum = info->lineNum;
660,985✔
1701
  info->cost.numOfSTables = taosHashGetSize(info->superTables);
660,985✔
1702
  info->cost.numOfCTables = taosHashGetSize(info->childTables);
660,985✔
1703
  info->cost.schemaTime = taosGetTimestampUs();
660,985✔
1704

1705
  do {
1706
    code = smlModifyDBSchemas(info);
689,275✔
1707
    if (code != TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER && code != TSDB_CODE_SDB_OBJ_CREATING && code != TSDB_CODE_SYN_NOT_LEADER &&
689,275✔
1708
        code != TSDB_CODE_MND_TRANS_CONFLICT) {
1709
      break;
660,985✔
1710
    }
1711
    taosMsleep(100);
28,290✔
1712
    uInfo("SML:0x%" PRIx64 ", smlModifyDBSchemas retry code:%s, times:%d", info->id, tstrerror(code), retryNum);
28,290✔
1713
  } while (retryNum++ < taosHashGetSize(info->superTables) * MAX_RETRY_TIMES);
28,290✔
1714

1715
  SML_CHECK_CODE(code);
660,985✔
1716
  info->cost.insertBindTime = taosGetTimestampUs();
652,542✔
1717
  SML_CHECK_CODE(smlInsertData(info));
652,542✔
1718

1719
END:
651,697✔
1720
  RETURN
768,551✔
1721
}
1722

1723
void smlSetReqSQL(SRequestObj *request, char *lines[], char *rawLine, char *rawLineEnd) {
768,018✔
1724
  if (request->pTscObj->pAppInfo->serverCfg.monitorParas.tsSlowLogScope & SLOW_LOG_TYPE_INSERT) {
768,018✔
1725
    int32_t len = 0;
×
1726
    int32_t rlen = 0;
×
1727
    char   *p = NULL;
×
1728

1729
    if (lines && lines[0]) {
×
1730
      len = strlen(lines[0]);
×
1731
      p = lines[0];
×
1732
    } else if (rawLine) {
×
1733
      if (rawLineEnd) {
×
1734
        len = rawLineEnd - rawLine;
×
1735
      } else {
1736
        len = strlen(rawLine);
×
1737
      }
1738
      p = rawLine;
×
1739
    }
1740

1741
    if (NULL == p) {
×
1742
      return;
×
1743
    }
1744

1745
    rlen = TMIN(len, TSDB_MAX_ALLOWED_SQL_LEN);
×
1746
    rlen = TMAX(rlen, 0);
×
1747

1748
    char *sql = taosMemoryMalloc(rlen + 1);
×
1749
    if (NULL == sql) {
×
1750
      uError("malloc %d for sml sql failed", rlen + 1);
×
1751
      return;
×
1752
    }
1753
    (void)memcpy(sql, p, rlen);
×
1754
    sql[rlen] = 0;
×
1755

1756
    request->sqlstr = sql;
×
1757
    request->sqlLen = rlen;
×
1758
  }
1759
}
1760

1761
TAOS_RES *taos_schemaless_insert_inner(TAOS *taos, char *lines[], char *rawLine, char *rawLineEnd, int numLines,
767,168✔
1762
                                       int protocol, int precision, int32_t ttl, int64_t reqid, char *tbnameKey) {
1763
  int32_t      code = TSDB_CODE_SUCCESS;
767,168✔
1764
  int32_t      lino = 0;
767,168✔
1765
  SRequestObj *request = NULL;
767,168✔
1766
  SSmlHandle  *info = NULL;
767,168✔
1767
  int          cnt = 0;
767,168✔
1768
  while (1) {
1,000✔
1769
    SML_CHECK_CODE(buildRequest(*(int64_t *)taos, "", 0, NULL, false, &request, reqid));
768,168✔
1770
    SSmlMsgBuf msg = {request->msgBufLen, request->msgBuf};
768,392✔
1771
    request->code = smlBuildSmlInfo(taos, &info);
768,392✔
1772
    SML_CHECK_CODE(request->code);
768,120✔
1773

1774
    info->pRequest = request;
768,120✔
1775
    info->pRequest->pQuery = info->pQuery;
768,120✔
1776
    info->ttl = ttl;
768,120✔
1777
    info->precision = precision;
768,392✔
1778
    info->protocol = (TSDB_SML_PROTOCOL_TYPE)protocol;
768,120✔
1779
    info->msgBuf.buf = info->pRequest->msgBuf;
768,120✔
1780
    info->msgBuf.len = ERROR_MSG_BUF_DEFAULT_SIZE;
768,120✔
1781
    info->lineNum = numLines;
768,120✔
1782
    info->tbnameKey = tbnameKey;
768,120✔
1783

1784
    smlSetReqSQL(request, lines, rawLine, rawLineEnd);
768,120✔
1785

1786
    if (request->pDb == NULL) {
768,120✔
1787
      request->code = TSDB_CODE_PAR_DB_NOT_SPECIFIED;
×
1788
      smlBuildInvalidDataMsg(&msg, "Database not specified", NULL);
×
1789
      goto END;
×
1790
    }
1791

1792
    if (protocol < TSDB_SML_LINE_PROTOCOL || protocol > TSDB_SML_JSON_PROTOCOL) {
768,120✔
1793
      request->code = TSDB_CODE_SML_INVALID_PROTOCOL_TYPE;
167✔
1794
      smlBuildInvalidDataMsg(&msg, "protocol invalidate", NULL);
167✔
1795
      goto END;
×
1796
    }
1797

1798
    if (protocol == TSDB_SML_LINE_PROTOCOL &&
767,953✔
1799
        (precision < TSDB_SML_TIMESTAMP_NOT_CONFIGURED || precision > TSDB_SML_TIMESTAMP_NANO_SECONDS)) {
252,371✔
UNCOV
1800
      request->code = TSDB_CODE_SML_INVALID_PRECISION_TYPE;
×
UNCOV
1801
      smlBuildInvalidDataMsg(&msg, "precision invalidate for line protocol", NULL);
×
1802
      goto END;
×
1803
    }
1804

1805
    if (protocol == TSDB_SML_JSON_PROTOCOL) {
768,055✔
1806
      numLines = 1;
324,059✔
1807
    } else if (numLines <= 0) {
443,996✔
1808
      request->code = TSDB_CODE_SML_INVALID_DATA;
×
1809
      smlBuildInvalidDataMsg(&msg, "line num is invalid", NULL);
×
1810
      goto END;
×
1811
    }
1812

1813
    code = smlProcess(info, lines, rawLine, rawLineEnd, numLines);
768,055✔
1814
    request->code = code;
768,551✔
1815
    info->cost.endTime = taosGetTimestampUs();
1,323,585✔
1816
    info->cost.code = code;
768,551✔
1817
    if (NEED_CLIENT_HANDLE_ERROR(code) || code == TSDB_CODE_SDB_OBJ_CREATING || code == TSDB_CODE_PAR_VALUE_TOO_LONG ||
768,551✔
1818
        code == TSDB_CODE_MND_TRANS_CONFLICT || code == TSDB_CODE_SYN_NOT_LEADER) {
767,392✔
1819
      if (cnt++ >= 10) {
1,159✔
1820
        uInfo("SML:%" PRIx64 " retry:%d/10 end code:%d, msg:%s", info->id, cnt, code, tstrerror(code));
×
1821
        break;
212,517✔
1822
      }
1823
      taosMsleep(100);
1,159✔
1824
      uInfo("SML:%" PRIx64 " retry:%d/10,ver is old retry or object is creating code:%d, msg:%s", info->id, cnt, code,
1,000✔
1825
            tstrerror(code));
1826
      code = refreshMeta(request->pTscObj, request);
1,000✔
1827
      if (code != 0) {
1,000✔
1828
        uInfo("SML:%" PRIx64 " refresh meta error code:%d, msg:%s", info->id, code, tstrerror(code));
841✔
1829
      }
1830
      smlDestroyInfo(info);
1,000✔
1831
      info = NULL;
1,000✔
1832
      taos_free_result(request);
1,000✔
1833
      request = NULL;
1,000✔
1834
      continue;
1,000✔
1835
    }
1836
    smlPrintStatisticInfo(info);
767,392✔
1837
    break;
767,551✔
1838
  }
1839

1840
END:
767,551✔
1841
  smlDestroyInfo(info);
767,551✔
1842
  return (TAOS_RES *)request;
767,551✔
1843
}
1844

1845
/**
1846
 * taos_schemaless_insert() parse and insert data points into database according to
1847
 * different protocol.
1848
 *
1849
 * @param $lines input array may contain multiple lines, each line indicates a data point.
1850
 *               If protocol=2 is used input array should contain single JSON
1851
 *               string(e.g. char *lines[] = {"$JSON_string"}). If need to insert
1852
 *               multiple data points in JSON format, should include them in $JSON_string
1853
 *               as a JSON array.
1854
 * @param $numLines indicates how many data points in $lines.
1855
 *                  If protocol = 2 is used this param will be ignored as $lines should
1856
 *                  contain single JSON string.
1857
 * @param $protocol indicates which protocol to use for parsing:
1858
 *                  0 - influxDB line protocol
1859
 *                  1 - OpenTSDB telnet line protocol
1860
 *                  2 - OpenTSDB JSON format protocol
1861
 * @return TAOS_RES
1862
 */
1863

1864
TAOS_RES *taos_schemaless_insert_ttl_with_reqid_tbname_key(TAOS *taos, char *lines[], int numLines, int protocol,
757,988✔
1865
                                                           int precision, int32_t ttl, int64_t reqid, char *tbnameKey) {
1866
  if (taos == NULL || lines == NULL || numLines < 0) {
757,988✔
UNCOV
1867
    terrno = TSDB_CODE_INVALID_PARA;
×
1868
    return NULL;
×
1869
  }
1870
  for (int i = 0; i < numLines; i++) {
52,870,277✔
1871
    if (lines[i] == NULL) {
52,110,580✔
1872
      terrno = TSDB_CODE_INVALID_PARA;
×
1873
      return NULL;
×
1874
    }
1875
  }
1876
  return taos_schemaless_insert_inner(taos, lines, NULL, NULL, numLines, protocol, precision, ttl, reqid, tbnameKey);
759,697✔
1877
}
1878

1879
TAOS_RES *taos_schemaless_insert_ttl_with_reqid(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
757,996✔
1880
                                                int32_t ttl, int64_t reqid) {
1881
  return taos_schemaless_insert_ttl_with_reqid_tbname_key(taos, lines, numLines, protocol, precision, ttl, reqid, NULL);
757,996✔
1882
}
1883

1884
TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision) {
756,513✔
1885
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, TSDB_DEFAULT_TABLE_TTL, 0);
756,513✔
1886
}
1887

1888
TAOS_RES *taos_schemaless_insert_ttl(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
1,372✔
1889
                                     int32_t ttl) {
1890
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, ttl, 0);
1,372✔
1891
}
1892

1893
TAOS_RES *taos_schemaless_insert_with_reqid(TAOS *taos, char *lines[], int numLines, int protocol, int precision,
×
1894
                                            int64_t reqid) {
1895
  return taos_schemaless_insert_ttl_with_reqid(taos, lines, numLines, protocol, precision, TSDB_DEFAULT_TABLE_TTL,
×
1896
                                               reqid);
1897
}
1898

1899
static int32_t getRawLineLen(char *lines, int len, int protocol) {
9,134✔
1900
  int   numLines = 0;
9,134✔
1901
  char *tmp = lines;
9,134✔
1902
  for (int i = 0; i < len; i++) {
30,439,116✔
1903
    if (lines[i] == '\n' || i == len - 1) {
30,429,982✔
1904
      if (!IS_COMMENT(protocol, tmp[0])) {  // ignore comment
32,816✔
1905
        numLines++;
32,473✔
1906
      }
1907
      tmp = lines + i + 1;
32,816✔
1908
    }
1909
  }
1910
  return numLines;
9,134✔
1911
}
1912

1913
TAOS_RES *taos_schemaless_insert_raw_ttl_with_reqid_tbname_key(TAOS *taos, char *lines, int len, int32_t *totalRows,
9,134✔
1914
                                                               int protocol, int precision, int32_t ttl, int64_t reqid,
1915
                                                               char *tbnameKey) {
1916
  if (taos == NULL || lines == NULL || len < 0) {
9,134✔
1917
    terrno = TSDB_CODE_INVALID_PARA;
×
1918
    return NULL;
×
1919
  }
1920
  int numLines = getRawLineLen(lines, len, protocol);
9,134✔
1921
  if (totalRows != NULL) {
9,134✔
1922
    *totalRows = numLines;
9,134✔
1923
  }
1924
  return taos_schemaless_insert_inner(taos, NULL, lines, lines + len, numLines, protocol, precision, ttl, reqid,
9,134✔
1925
                                      tbnameKey);
1926
}
1927

1928
TAOS_RES *taos_schemaless_insert_raw_ttl_with_reqid(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
9,134✔
1929
                                                    int precision, int32_t ttl, int64_t reqid) {
1930
  return taos_schemaless_insert_raw_ttl_with_reqid_tbname_key(taos, lines, len, totalRows, protocol, precision, ttl,
9,134✔
1931
                                                              reqid, NULL);
1932
}
1933

1934
TAOS_RES *taos_schemaless_insert_raw_with_reqid(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
1,798✔
1935
                                                int precision, int64_t reqid) {
1936
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision,
1,798✔
1937
                                                   TSDB_DEFAULT_TABLE_TTL, reqid);
1938
}
1939
TAOS_RES *taos_schemaless_insert_raw_ttl(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
×
1940
                                         int precision, int32_t ttl) {
1941
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision, ttl, 0);
×
1942
}
1943

1944
TAOS_RES *taos_schemaless_insert_raw(TAOS *taos, char *lines, int len, int32_t *totalRows, int protocol,
3,749✔
1945
                                     int precision) {
1946
  return taos_schemaless_insert_raw_ttl_with_reqid(taos, lines, len, totalRows, protocol, precision,
3,749✔
1947
                                                   TSDB_DEFAULT_TABLE_TTL, 0);
1948
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc