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

taosdata / TDengine / #3837

03 Apr 2025 01:31PM UTC coverage: 61.991% (+26.1%) from 35.847%
#3837

push

travis-ci

web-flow
Merge pull request #30644 from taosdata/merge/mainto3.0

merge: from main to 3.0 branch

152850 of 315065 branches covered (48.51%)

Branch coverage included in aggregate %.

402 of 459 new or added lines in 2 files covered. (87.58%)

284 existing lines in 50 files now uncovered.

237727 of 314991 relevant lines covered (75.47%)

5799871.48 hits per line

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

61.32
/source/common/src/tdataformat.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
#define _DEFAULT_SOURCE
17
#include "tdataformat.h"
18
#include "tRealloc.h"
19
#include "tdatablock.h"
20
#include "tlog.h"
21
#include "decimal.h"
22

23
static int32_t (*tColDataAppendValueImpl[8][3])(SColData *pColData, uint8_t *pData, uint32_t nData);
24
static int32_t (*tColDataUpdateValueImpl[8][3])(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward);
25

26
// ================================
27
static int32_t tGetTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson);
28

29
// SRow ========================================================================
30
#define KV_FLG_LIT ((uint8_t)0x10)
31
#define KV_FLG_MID ((uint8_t)0x20)
32
#define KV_FLG_BIG ((uint8_t)0x40)
33

34
#define BIT_FLG_NONE  ((uint8_t)0x0)
35
#define BIT_FLG_NULL  ((uint8_t)0x1)
36
#define BIT_FLG_VALUE ((uint8_t)0x2)
37

38
#pragma pack(push, 1)
39
typedef struct {
40
  int16_t nCol;
41
  uint8_t idx[];  // uint8_t * | uint16_t * | uint32_t *
42
} SKVIdx;
43
#pragma pack(pop)
44

45
#define ROW_SET_BITMAP(PB, FLAG, IDX, VAL)      \
46
  do {                                          \
47
    switch (FLAG) {                             \
48
      case (HAS_NULL | HAS_NONE):               \
49
        SET_BIT1(PB, IDX, VAL);                 \
50
        break;                                  \
51
      case (HAS_VALUE | HAS_NONE):              \
52
        SET_BIT1(PB, IDX, (VAL) ? (VAL)-1 : 0); \
53
        break;                                  \
54
      case (HAS_VALUE | HAS_NULL):              \
55
        SET_BIT1(PB, IDX, (VAL)-1);             \
56
        break;                                  \
57
      case (HAS_VALUE | HAS_NULL | HAS_NONE):   \
58
        SET_BIT2(PB, IDX, VAL);                 \
59
        break;                                  \
60
      default:                                  \
61
        break;                                  \
62
    }                                           \
63
  } while (0)
64

65
static int32_t tPutPrimaryKeyIndex(uint8_t *p, const SPrimaryKeyIndex *index) {
31,130✔
66
  int32_t n = 0;
31,130✔
67
  n += tPutI8(p ? p + n : p, index->type);
31,130✔
68
  n += tPutU32v(p ? p + n : p, index->offset);
31,130✔
69
  return n;
31,130✔
70
}
71

72
static int32_t tGetPrimaryKeyIndex(uint8_t *p, SPrimaryKeyIndex *index) {
19,437✔
73
  int32_t n = 0;
19,437✔
74
  n += tGetI8(p + n, &index->type);
19,437!
75
  n += tGetU32v(p + n, &index->offset);
19,437!
76
  return n;
19,437✔
77
}
78

79
typedef struct {
80
  int32_t numOfNone;
81
  int32_t numOfNull;
82
  int32_t numOfValue;
83
  int32_t numOfPKs;
84
  int8_t  flag;
85

86
  // tuple
87
  int8_t           tupleFlag;
88
  SPrimaryKeyIndex tupleIndices[TD_MAX_PK_COLS];
89
  int32_t          tuplePKSize;      // primary key size
90
  int32_t          tupleBitmapSize;  // bitmap size
91
  int32_t          tupleFixedSize;   // fixed part size
92
  int32_t          tupleVarSize;     // var part size
93
  int32_t          tupleRowSize;
94

95
  // key-value
96
  int8_t           kvFlag;
97
  SPrimaryKeyIndex kvIndices[TD_MAX_PK_COLS];
98
  int32_t          kvMaxOffset;
99
  int32_t          kvPKSize;       // primary key size
100
  int32_t          kvIndexSize;    // offset array size
101
  int32_t          kvPayloadSize;  // payload size
102
  int32_t          kvRowSize;
103
} SRowBuildScanInfo;
104

105
static FORCE_INLINE int32_t tRowBuildScanAddNone(SRowBuildScanInfo *sinfo, const STColumn *pTColumn) {
106
  if ((pTColumn->flags & COL_IS_KEY)) return TSDB_CODE_PAR_PRIMARY_KEY_IS_NONE;
×
107
  sinfo->numOfNone++;
224,068,600✔
108
  return 0;
224,068,600✔
109
}
110

111
static FORCE_INLINE int32_t tRowBuildScanAddNull(SRowBuildScanInfo *sinfo, const STColumn *pTColumn) {
112
  if ((pTColumn->flags & COL_IS_KEY)) return TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
113
  sinfo->numOfNull++;
15,325,421✔
114
  sinfo->kvMaxOffset = sinfo->kvPayloadSize;
15,325,421✔
115
  sinfo->kvPayloadSize += tPutI16v(NULL, -pTColumn->colId);
15,325,421✔
116
  return 0;
15,325,421✔
117
}
118

119
static FORCE_INLINE void tRowBuildScanAddValue(SRowBuildScanInfo *sinfo, SColVal *colVal, const STColumn *pTColumn) {
120
  bool isPK = ((pTColumn->flags & COL_IS_KEY) != 0);
1,073,908,680✔
121

122
  if (isPK) {
1,073,908,680✔
123
    sinfo->tupleIndices[sinfo->numOfPKs].type = colVal->value.type;
10,442✔
124
    sinfo->tupleIndices[sinfo->numOfPKs].offset =
10,442✔
125
        IS_VAR_DATA_TYPE(pTColumn->type) ? sinfo->tupleVarSize + sinfo->tupleFixedSize : pTColumn->offset;
10,442!
126
    sinfo->kvIndices[sinfo->numOfPKs].type = colVal->value.type;
10,442✔
127
    sinfo->kvIndices[sinfo->numOfPKs].offset = sinfo->kvPayloadSize;
10,442✔
128
    sinfo->numOfPKs++;
10,442✔
129
  }
130

131
  sinfo->kvMaxOffset = sinfo->kvPayloadSize;
1,073,908,680✔
132
  if (IS_VAR_DATA_TYPE(colVal->value.type)) {
1,073,908,680!
133
    sinfo->tupleVarSize += tPutU32v(NULL, colVal->value.nData)  // size
178,601,378✔
134
                           + colVal->value.nData;               // value
178,601,378✔
135

136
    sinfo->kvPayloadSize += tPutI16v(NULL, colVal->cid)            // colId
178,601,378✔
137
                            + tPutU32v(NULL, colVal->value.nData)  // size
178,601,378✔
138
                            + colVal->value.nData;                 // value
178,601,378✔
139
  } else {
140
    sinfo->kvPayloadSize += tPutI16v(NULL, colVal->cid)              // colId
895,307,302✔
141
                            + tDataTypes[colVal->value.type].bytes;  // value
895,307,302✔
142
  }
143
  sinfo->numOfValue++;
1,073,908,680✔
144
}
1,073,908,680✔
145

146
static int32_t tRowBuildScan(SArray *colVals, const STSchema *schema, SRowBuildScanInfo *sinfo) {
196,288,018✔
147
  int32_t  code = 0;
196,288,018✔
148
  int32_t  colValIndex = 1;
196,288,018✔
149
  int32_t  numOfColVals = TARRAY_SIZE(colVals);
196,288,018✔
150
  SColVal *colValArray = (SColVal *)TARRAY_DATA(colVals);
196,288,018✔
151

152
  if (!(numOfColVals > 0)) {
196,288,018!
153
    return TSDB_CODE_INVALID_PARA;
×
154
  }
155
  if (!(colValArray[0].cid == PRIMARYKEY_TIMESTAMP_COL_ID)) {
196,288,018!
156
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;
×
157
  }
158
  if (!(colValArray[0].value.type == TSDB_DATA_TYPE_TIMESTAMP)) {
196,288,018!
159
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;;
×
160
  }
161

162
  *sinfo = (SRowBuildScanInfo){
196,288,018✔
163
      .tupleFixedSize = schema->flen,
196,288,018✔
164
  };
165

166
  // loop scan
167
  for (int32_t i = 1; i < schema->numOfCols; i++) {
1,508,595,242✔
168
    for (;;) {
169
      if (colValIndex >= numOfColVals) {
1,311,599,049✔
170
        if ((code = tRowBuildScanAddNone(sinfo, schema->columns + i))) goto _exit;
496!
171
        break;
248✔
172
      }
173

174
      if (colValArray[colValIndex].cid == schema->columns[i].colId) {
1,311,598,801!
175
        if (!(colValArray[colValIndex].value.type == schema->columns[i].type)) {
1,312,306,945!
176
          code = TSDB_CODE_INVALID_PARA;
×
177
          goto _exit;
×
178
        }
179

180
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {
1,312,306,945✔
181
          tRowBuildScanAddValue(sinfo, &colValArray[colValIndex], schema->columns + i);
1,073,908,680✔
182
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {
238,398,265✔
183
          if ((code = tRowBuildScanAddNull(sinfo, schema->columns + i))) goto _exit;
30,650,842!
184
        } else if (COL_VAL_IS_NONE(&colValArray[colValIndex])) {
223,072,844!
185
          if ((code = tRowBuildScanAddNone(sinfo, schema->columns + i))) goto _exit;
448,136,642!
186
        }
187

188
        colValIndex++;
1,312,306,945✔
189
        break;
1,312,306,945✔
190
      } else if (colValArray[colValIndex].cid > schema->columns[i].colId) {
×
191
        if ((code = tRowBuildScanAddNone(sinfo, schema->columns + i))) goto _exit;
62!
192
        break;
31✔
193
      } else {  // skip useless value
194
        colValIndex++;
×
195
      }
196
    }
197
  }
198

199
  if (sinfo->numOfNone) {
196,288,018✔
200
    sinfo->flag |= HAS_NONE;
55,769,986✔
201
  }
202
  if (sinfo->numOfNull) {
196,288,018✔
203
    sinfo->flag |= HAS_NULL;
8,376,103✔
204
  }
205
  if (sinfo->numOfValue) {
196,288,018✔
206
    sinfo->flag |= HAS_VALUE;
157,058,824✔
207
  }
208

209
  // Tuple
210
  sinfo->tupleFlag = sinfo->flag;
196,288,018✔
211
  switch (sinfo->flag) {
196,288,018!
212
    case HAS_NONE:
40,903,305✔
213
    case HAS_NULL:
214
      sinfo->tupleBitmapSize = 0;
40,903,305✔
215
      sinfo->tupleFixedSize = 0;
40,903,305✔
216
      break;
40,903,305✔
217
    case HAS_VALUE:
134,661,346✔
218
      sinfo->tupleBitmapSize = 0;
134,661,346✔
219
      sinfo->tupleFixedSize = schema->flen;
134,661,346✔
220
      break;
134,661,346✔
221
    case (HAS_NONE | HAS_NULL):
24,398✔
222
      sinfo->tupleBitmapSize = BIT1_SIZE(schema->numOfCols - 1);
24,398✔
223
      sinfo->tupleFixedSize = 0;
24,398✔
224
      break;
24,398✔
225
    case (HAS_NONE | HAS_VALUE):
22,321,662✔
226
    case (HAS_NULL | HAS_VALUE):
227
      sinfo->tupleBitmapSize = BIT1_SIZE(schema->numOfCols - 1);
22,321,662✔
228
      sinfo->tupleFixedSize = schema->flen;
22,321,662✔
229
      break;
22,321,662✔
230
    case (HAS_NONE | HAS_NULL | HAS_VALUE):
451,628✔
231
      sinfo->tupleBitmapSize = BIT2_SIZE(schema->numOfCols - 1);
451,628✔
232
      sinfo->tupleFixedSize = schema->flen;
451,628✔
233
      break;
451,628✔
234
  }
235
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
196,298,491✔
236
    sinfo->tupleIndices[i].offset += sinfo->tupleBitmapSize;
10,485✔
237
    sinfo->tuplePKSize += tPutPrimaryKeyIndex(NULL, sinfo->tupleIndices + i);
10,485✔
238
  }
239
  sinfo->tupleRowSize = sizeof(SRow)              // SRow
196,288,006✔
240
                        + sinfo->tuplePKSize      // primary keys
196,288,006✔
241
                        + sinfo->tupleBitmapSize  // bitmap
196,288,006✔
242
                        + sinfo->tupleFixedSize   // fixed part
196,288,006✔
243
                        + sinfo->tupleVarSize;    // var part
196,288,006✔
244

245
  // Key-Value
246
  if (sinfo->kvMaxOffset <= UINT8_MAX) {
196,288,006✔
247
    sinfo->kvFlag = (KV_FLG_LIT | sinfo->flag);
193,019,470✔
248
    sinfo->kvIndexSize = sizeof(SKVIdx) + (sinfo->numOfNull + sinfo->numOfValue) * sizeof(uint8_t);
193,019,470✔
249
  } else if (sinfo->kvMaxOffset <= UINT16_MAX) {
3,268,536!
250
    sinfo->kvFlag = (KV_FLG_MID | sinfo->flag);
5,315,057✔
251
    sinfo->kvIndexSize = sizeof(SKVIdx) + (sinfo->numOfNull + sinfo->numOfValue) * sizeof(uint16_t);
5,315,057✔
252
  } else {
253
    sinfo->kvFlag = (KV_FLG_BIG | sinfo->flag);
×
254
    sinfo->kvIndexSize = sizeof(SKVIdx) + (sinfo->numOfNull + sinfo->numOfValue) * sizeof(uint32_t);
×
255
  }
256
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
196,298,448✔
257
    sinfo->kvIndices[i].offset += sinfo->kvIndexSize;
10,480✔
258
    sinfo->kvPKSize += tPutPrimaryKeyIndex(NULL, sinfo->kvIndices + i);
10,480✔
259
  }
260
  sinfo->kvRowSize = sizeof(SRow)             // SRow
196,287,968✔
261
                     + sinfo->kvPKSize        // primary keys
196,287,968✔
262
                     + sinfo->kvIndexSize     // index array
196,287,968✔
263
                     + sinfo->kvPayloadSize;  // payload
196,287,968✔
264

265
_exit:
196,287,968✔
266
  return code;
196,287,968✔
267
}
268

269
static int32_t tRowBuildTupleRow(SArray *aColVal, const SRowBuildScanInfo *sinfo, const STSchema *schema,
177,250,006✔
270
                                 SRow **ppRow) {
271
  SColVal *colValArray = (SColVal *)TARRAY_DATA(aColVal);
177,250,006✔
272

273
  *ppRow = (SRow *)taosMemoryCalloc(1, sinfo->tupleRowSize);
177,250,006!
274
  if (*ppRow == NULL) {
178,942,761!
275
    return terrno;
×
276
  }
277
  (*ppRow)->flag = sinfo->tupleFlag;
178,942,761✔
278
  (*ppRow)->numOfPKs = sinfo->numOfPKs;
178,942,761✔
279
  (*ppRow)->sver = schema->version;
178,942,761✔
280
  (*ppRow)->len = sinfo->tupleRowSize;
178,942,761✔
281
  (*ppRow)->ts = VALUE_GET_TRIVIAL_DATUM(&colValArray[0].value);
178,942,761✔
282

283
  if (sinfo->tupleFlag == HAS_NONE || sinfo->tupleFlag == HAS_NULL) {
178,942,761✔
284
    return 0;
41,286,686✔
285
  }
286

287
  uint8_t *primaryKeys = (*ppRow)->data;
137,656,075✔
288
  uint8_t *bitmap = primaryKeys + sinfo->tuplePKSize;
137,656,075✔
289
  uint8_t *fixed = bitmap + sinfo->tupleBitmapSize;
137,656,075✔
290
  uint8_t *varlen = fixed + sinfo->tupleFixedSize;
137,656,075✔
291

292
  // primary keys
293
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
137,666,423✔
294
    primaryKeys += tPutPrimaryKeyIndex(primaryKeys, sinfo->tupleIndices + i);
10,378✔
295
  }
296

297
  // bitmap + fixed + varlen
298
  int32_t numOfColVals = TARRAY_SIZE(aColVal);
137,656,045✔
299
  int32_t colValIndex = 1;
137,656,045✔
300
  for (int32_t i = 1; i < schema->numOfCols; i++) {
1,147,430,215✔
301
    for (;;) {
302
      if (colValIndex >= numOfColVals) {  // NONE
1,008,822,721!
303
        ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NONE);
×
304
        break;
×
305
      }
306

307
      if (colValArray[colValIndex].cid == schema->columns[i].colId) {
1,008,822,721!
308
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {  // value
1,009,774,161✔
309
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_VALUE);
1,006,370,447!
310

311
          if (IS_VAR_DATA_TYPE(schema->columns[i].type)) {
1,006,370,447!
312
            *(int32_t *)(fixed + schema->columns[i].offset) = varlen - fixed - sinfo->tupleFixedSize;
156,372,344✔
313
            varlen += tPutU32v(varlen, colValArray[colValIndex].value.nData);
156,372,344✔
314
            if (colValArray[colValIndex].value.nData) {
156,372,344!
315
              (void)memcpy(varlen, colValArray[colValIndex].value.pData, colValArray[colValIndex].value.nData);
160,713,141✔
316
              varlen += colValArray[colValIndex].value.nData;
160,713,141✔
317
            }
318
          } else {
319
            (void)memcpy(fixed + schema->columns[i].offset,
849,998,103✔
320
                         VALUE_GET_DATUM(&colValArray[colValIndex].value, schema->columns[i].type),
849,998,103!
321
                         tDataTypes[schema->columns[i].type].bytes);
849,998,103✔
322
          }
323
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {  // NULL
3,403,714✔
324
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NULL);
2,806,076!
325
        } else if (COL_VAL_IS_NONE(&colValArray[colValIndex])) {  // NONE
597,638✔
326
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NONE);
469,209!
327
        }
328

329
        colValIndex++;
1,009,774,161✔
330
        break;
1,009,774,161✔
331
      } else if (colValArray[colValIndex].cid > schema->columns[i].colId) {  // NONE
×
332
        ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NONE);
9!
333
        break;
9✔
334
      } else {
335
        colValIndex++;
×
336
      }
337
    }
338
  }
339

340
  return 0;
137,656,045✔
341
}
342

343
static FORCE_INLINE void tRowBuildKVRowSetIndex(uint8_t flag, SKVIdx *indices, uint32_t offset) {
344
  if (flag & KV_FLG_LIT) {
83,377,117✔
345
    ((uint8_t *)indices->idx)[indices->nCol] = (uint8_t)offset;
75,938,203✔
346
  } else if (flag & KV_FLG_MID) {
7,438,914!
347
    ((uint16_t *)indices->idx)[indices->nCol] = (uint16_t)offset;
7,438,735✔
348
  } else {
349
    ((uint32_t *)indices->idx)[indices->nCol] = (uint32_t)offset;
179✔
350
  }
351
  indices->nCol++;
83,377,117✔
352
}
83,377,117✔
353

354
static int32_t tRowBuildKVRow(SArray *aColVal, const SRowBuildScanInfo *sinfo, const STSchema *schema, SRow **ppRow) {
20,641,240✔
355
  SColVal *colValArray = (SColVal *)TARRAY_DATA(aColVal);
20,641,240✔
356

357
  *ppRow = (SRow *)taosMemoryCalloc(1, sinfo->kvRowSize);
20,641,240!
358
  if (*ppRow == NULL) {
20,701,601!
359
    return terrno;
×
360
  }
361
  (*ppRow)->flag = sinfo->kvFlag;
20,701,601✔
362
  (*ppRow)->numOfPKs = sinfo->numOfPKs;
20,701,601✔
363
  (*ppRow)->sver = schema->version;
20,701,601✔
364
  (*ppRow)->len = sinfo->kvRowSize;
20,701,601✔
365
  (*ppRow)->ts = VALUE_GET_TRIVIAL_DATUM(&colValArray[0].value);
20,701,601✔
366

367
  if (!(sinfo->flag != HAS_NONE && sinfo->flag != HAS_NULL)) {
20,701,601!
368
    return TSDB_CODE_INVALID_PARA;
2,817✔
369
  }
370

371
  uint8_t *primaryKeys = (*ppRow)->data;
20,698,784✔
372
  SKVIdx  *indices = (SKVIdx *)(primaryKeys + sinfo->kvPKSize);
20,698,784✔
373
  uint8_t *payload = primaryKeys + sinfo->kvPKSize + sinfo->kvIndexSize;
20,698,784✔
374
  uint32_t payloadSize = 0;
20,698,784✔
375

376
  // primary keys
377
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
20,698,920✔
378
    primaryKeys += tPutPrimaryKeyIndex(primaryKeys, sinfo->kvIndices + i);
136✔
379
  }
380

381
  int32_t numOfColVals = TARRAY_SIZE(aColVal);
20,698,784✔
382
  int32_t colValIndex = 1;
20,698,784✔
383
  for (int32_t i = 1; i < schema->numOfCols; i++) {
163,576,477✔
384
    for (;;) {
385
      if (colValIndex >= numOfColVals) {  // NONE
142,856,620✔
386
        break;
44✔
387
      }
388

389
      if (colValArray[colValIndex].cid == schema->columns[i].colId) {
142,856,576!
390
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {  // value
142,877,627✔
391
          tRowBuildKVRowSetIndex(sinfo->kvFlag, indices, payloadSize);
71,095,558✔
392
          if (IS_VAR_DATA_TYPE(schema->columns[i].type)) {
71,095,558!
393
            payloadSize += tPutI16v(payload + payloadSize, colValArray[colValIndex].cid);
22,025,460✔
394
            payloadSize += tPutU32v(payload + payloadSize, colValArray[colValIndex].value.nData);
22,025,460✔
395
            if (colValArray[colValIndex].value.nData > 0) {
22,025,460✔
396
              (void)memcpy(payload + payloadSize, colValArray[colValIndex].value.pData,
21,998,618✔
397
                           colValArray[colValIndex].value.nData);
21,998,618✔
398
            }
399
            payloadSize += colValArray[colValIndex].value.nData;
22,025,460✔
400
          } else {
401
            payloadSize += tPutI16v(payload + payloadSize, colValArray[colValIndex].cid);
49,070,098✔
402
            (void)memcpy(payload + payloadSize, VALUE_GET_DATUM(&colValArray[colValIndex].value, schema->columns[i].type),
49,070,098!
403
                         tDataTypes[schema->columns[i].type].bytes);
49,070,098✔
404
            payloadSize += tDataTypes[schema->columns[i].type].bytes;
49,070,098✔
405
          }
406
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {  // NULL
71,782,069✔
407
          tRowBuildKVRowSetIndex(sinfo->kvFlag, indices, payloadSize);
12,281,559✔
408
          payloadSize += tPutI16v(payload + payloadSize, -schema->columns[i].colId);
24,563,118✔
409
        }
410

411
        colValIndex++;
142,877,627✔
412
        break;
142,877,627✔
413
      } else if (colValArray[colValIndex].cid > schema->columns[i].colId) {  // NONE
×
414
        break;
22✔
415
      } else {
416
        colValIndex++;
×
417
      }
418
    }
419
  }
420

421
  return 0;
20,698,784✔
422
}
423

424
int32_t tRowBuild(SArray *aColVal, const STSchema *pTSchema, SRow **ppRow) {
196,164,005✔
425
  int32_t           code;
426
  SRowBuildScanInfo sinfo;
427

428
  code = tRowBuildScan(aColVal, pTSchema, &sinfo);
196,164,005✔
429
  if (code) return code;
198,356,905!
430

431
  if (sinfo.tupleRowSize <= sinfo.kvRowSize) {
198,356,905✔
432
    code = tRowBuildTupleRow(aColVal, &sinfo, pTSchema, ppRow);
177,798,687✔
433
  } else {
434
    code = tRowBuildKVRow(aColVal, &sinfo, pTSchema, ppRow);
20,558,218✔
435
  }
436
  return code;
199,260,023✔
437
}
438

439
static int32_t tBindInfoCompare(const void *p1, const void *p2, const void *param) {
1✔
440
  if (((SBindInfo *)p1)->columnId < ((SBindInfo *)p2)->columnId) {
1!
441
    return -1;
1✔
442
  } else if (((SBindInfo *)p1)->columnId > ((SBindInfo *)p2)->columnId) {
×
443
    return 1;
×
444
  }
445
  return 0;
×
446
}
447

448
/* build rows to `rowArray` from bind
449
 * `infos` is the bind information array
450
 * `numOfInfos` is the number of bind information
451
 * `infoSorted` is whether the bind information is sorted by column id
452
 * `pTSchema` is the schema of the table
453
 * `rowArray` is the array to store the rows
454
 * `pOrdered` is the pointer to store ordered
455
 * `pDupTs` is the pointer to store duplicateTs
456
 */
457
int32_t tRowBuildFromBind(SBindInfo *infos, int32_t numOfInfos, bool infoSorted, const STSchema *pTSchema,
39,234✔
458
                          SArray *rowArray, bool *pOrdered, bool *pDupTs) {
459
  if (infos == NULL || numOfInfos <= 0 || numOfInfos > pTSchema->numOfCols || pTSchema == NULL || rowArray == NULL) {
39,234!
460
    return TSDB_CODE_INVALID_PARA;
×
461
  }
462

463
  if (!infoSorted) {
39,293!
464
    taosqsort_r(infos, numOfInfos, sizeof(SBindInfo), NULL, tBindInfoCompare);
×
465
  }
466

467
  int32_t code = 0;
39,293✔
468
  int32_t numOfRows = infos[0].bind->num;
39,293✔
469
  SArray *colValArray;
470
  SColVal colVal;
471

472
  if ((colValArray = taosArrayInit(numOfInfos, sizeof(SColVal))) == NULL) {
39,293!
473
    return terrno;
×
474
  }
475

476
  SRowKey rowKey, lastRowKey;
477
  for (int32_t iRow = 0; iRow < numOfRows; iRow++) {
148,282✔
478
    taosArrayClear(colValArray);
111,842✔
479

480
    for (int32_t iInfo = 0; iInfo < numOfInfos; iInfo++) {
846,289✔
481
      if (infos[iInfo].bind->is_null && infos[iInfo].bind->is_null[iRow]) {
753,484✔
482
        colVal = COL_VAL_NULL(infos[iInfo].columnId, infos[iInfo].type);
32✔
483
      } else {
484
        SValue value = {
753,452✔
485
            .type = infos[iInfo].type,
753,452✔
486
        };
487
        if (IS_VAR_DATA_TYPE(infos[iInfo].type)) {
753,452!
488
          value.nData = infos[iInfo].bind->length[iRow];
174,304✔
489
          if (value.nData > pTSchema->columns[iInfo].bytes - VARSTR_HEADER_SIZE) {
174,304!
490
            code = TSDB_CODE_INVALID_PARA;
×
491
            goto _exit;
×
492
          }
493
          value.pData = (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bind->buffer_length * iRow;
174,304✔
494
        } else {
495
          valueSetDatum(&value, infos[iInfo].type,
579,148✔
496
                        (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bind->buffer_length * iRow,
579,148✔
497
                        infos[iInfo].bind->buffer_length);
579,148✔
498
        }
499
        colVal = COL_VAL_VALUE(infos[iInfo].columnId, value);
771,818✔
500
      }
501
      if (taosArrayPush(colValArray, &colVal) == NULL) {
735,164!
502
        code = terrno;
×
503
        goto _exit;
×
504
      }
505
    }
506

507
    SRow *row;
508
    if ((code = tRowBuild(colValArray, pTSchema, &row))) {
92,805!
509
      goto _exit;
×
510
    }
511

512
    if ((taosArrayPush(rowArray, &row)) == NULL) {
108,926!
513
      code = terrno;
×
514
      goto _exit;
×
515
    }
516

517
    if (pOrdered && pDupTs) {
108,926!
518
      tRowGetKey(row, &rowKey);
217,968!
519
      if (iRow == 0) {
109,063✔
520
        *pOrdered = true;
38,821✔
521
        *pDupTs = false;
38,821✔
522
      } else {
523
        if (*pOrdered) {
70,242!
524
          int32_t res = tRowKeyCompare(&rowKey, &lastRowKey);
70,247✔
525
          *pOrdered = (res >= 0);
70,247✔
526
          if (!*pDupTs) {
70,247✔
527
            *pDupTs = (res == 0);
69,983✔
528
          }
529
        }
530
      }
531
      lastRowKey = rowKey;
109,063✔
532
    }
533
  }
534

535
_exit:
36,440✔
536
  taosArrayDestroy(colValArray);
36,440✔
537
  return code;
39,340✔
538
}
539

540
int32_t tRowGet(SRow *pRow, STSchema *pTSchema, int32_t iCol, SColVal *pColVal) {
911,417,579✔
541
  if (!(iCol < pTSchema->numOfCols)) return TSDB_CODE_INVALID_PARA;
911,417,579!
542
  if (!(pRow->sver == pTSchema->version)) return TSDB_CODE_INVALID_PARA;
911,417,579✔
543

544
  STColumn *pTColumn = pTSchema->columns + iCol;
911,417,574✔
545

546
  if (iCol == 0) {
911,417,574✔
547
    pColVal->cid = pTColumn->colId;
35,738,501✔
548
    pColVal->value.type = pTColumn->type;
35,738,501✔
549
    pColVal->flag = CV_FLAG_VALUE;
35,738,501✔
550
    VALUE_SET_TRIVIAL_DATUM(&pColVal->value, pRow->ts);
35,738,501✔
551
    return 0;
35,738,501✔
552
  }
553

554
  if (pRow->flag == HAS_NONE) {
875,679,073✔
555
    *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
39,025✔
556
    return 0;
39,025✔
557
  }
558

559
  if (pRow->flag == HAS_NULL) {
875,640,048✔
560
    *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
847,456✔
561
    return 0;
847,456✔
562
  }
563

564
  SPrimaryKeyIndex index;
565
  uint8_t         *data = pRow->data;
874,792,592✔
566
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
874,795,464✔
567
    data += tGetPrimaryKeyIndex(data, &index);
2,872✔
568
  }
569

570
  if (pRow->flag >> 4) {  // KV Row
874,792,592✔
571
    SKVIdx  *pIdx = (SKVIdx *)data;
293,484,816✔
572
    uint8_t *pv = NULL;
293,484,816✔
573

574
    if (pRow->flag & KV_FLG_LIT) {
293,484,816✔
575
      pv = pIdx->idx + pIdx->nCol;
225,065,007✔
576
    } else if (pRow->flag & KV_FLG_MID) {
68,419,809✔
577
      pv = pIdx->idx + (pIdx->nCol << 1);
68,225,062✔
578
    } else {
579
      pv = pIdx->idx + (pIdx->nCol << 2);
194,747✔
580
    }
581

582
    int16_t lidx = 0;
293,484,816✔
583
    int16_t ridx = pIdx->nCol - 1;
293,484,816✔
584
    while (lidx <= ridx) {
864,343,113✔
585
      int16_t  mid = (lidx + ridx) >> 1;
836,709,624✔
586
      uint8_t *pData = NULL;
836,709,624✔
587
      if (pRow->flag & KV_FLG_LIT) {
836,709,624✔
588
        pData = pv + ((uint8_t *)pIdx->idx)[mid];
552,309,994✔
589
      } else if (pRow->flag & KV_FLG_MID) {
284,399,630!
590
        pData = pv + ((uint16_t *)pIdx->idx)[mid];
284,454,683✔
591
      } else {
592
        pData = pv + ((uint32_t *)pIdx->idx)[mid];
×
593
      }
594

595
      int16_t cid;
596
      pData += tGetI16v(pData, &cid);
836,709,624✔
597

598
      if (TABS(cid) == pTColumn->colId) {
836,709,624✔
599
        if (cid < 0) {
265,851,327✔
600
          *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
87,156,792✔
601
        } else {
602
          pColVal->cid = pTColumn->colId;
178,694,535✔
603
          pColVal->value.type = pTColumn->type;
178,694,535✔
604
          pColVal->flag = CV_FLAG_VALUE;
178,694,535✔
605

606
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
178,694,535!
607
            pData += tGetU32v(pData, &pColVal->value.nData);
28,517,619!
608
            if (pColVal->value.nData > 0) {
28,517,619!
609
              pColVal->value.pData = pData;
30,988,433✔
610
            } else {
611
              pColVal->value.pData = NULL;
×
612
            }
613
          } else {
614
            valueSetDatum(&pColVal->value, pTColumn->type, pData, pTColumn->bytes);
150,176,916✔
615
          }
616
        }
617
        return 0;
267,312,513✔
618
      } else if (TABS(cid) < pTColumn->colId) {
570,858,297✔
619
        lidx = mid + 1;
241,792,606✔
620
      } else {
621
        ridx = mid - 1;
329,065,691✔
622
      }
623
    }
624

625
    *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
27,633,489✔
626
  } else {  // Tuple Row
627
    uint8_t *bitmap = data;
581,307,776✔
628
    uint8_t *fixed;
629
    uint8_t *varlen;
630
    uint8_t  bit;
631

632
    if (pRow->flag == HAS_VALUE) {
581,307,776!
633
      fixed = bitmap;
605,802,106✔
634
      bit = BIT_FLG_VALUE;
605,802,106✔
635
    } else if (pRow->flag == (HAS_NONE | HAS_NULL | HAS_VALUE)) {
×
636
      fixed = BIT2_SIZE(pTSchema->numOfCols - 1) + bitmap;
×
637
      bit = GET_BIT2(bitmap, iCol - 1);
×
638
    } else {
639
      fixed = BIT1_SIZE(pTSchema->numOfCols - 1) + bitmap;
×
640
      bit = GET_BIT1(bitmap, iCol - 1);
×
641

642
      if (pRow->flag == (HAS_NONE | HAS_VALUE)) {
×
643
        if (bit) bit++;
20,213✔
644
      } else if (pRow->flag == (HAS_NULL | HAS_VALUE)) {
×
645
        bit++;
1,157,820✔
646
      }
647
    }
648
    varlen = fixed + pTSchema->flen;
581,307,776✔
649

650
    if (bit == BIT_FLG_NONE) {
581,307,776✔
651
      *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
7,218✔
652
      return 0;
7,218✔
653
    } else if (bit == BIT_FLG_NULL) {
581,300,558✔
654
      *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
290,008✔
655
      return 0;
290,008✔
656
    }
657

658
    pColVal->cid = pTColumn->colId;
581,010,550✔
659
    pColVal->value.type = pTColumn->type;
581,010,550✔
660
    pColVal->flag = CV_FLAG_VALUE;
581,010,550✔
661
    if (IS_VAR_DATA_TYPE(pTColumn->type)) {
581,010,550!
662
      pColVal->value.pData = varlen + *(int32_t *)(fixed + pTColumn->offset);
5,318,458✔
663
      pColVal->value.pData += tGetU32v(pColVal->value.pData, &pColVal->value.nData);
10,636,916!
664
    } else {
665
      valueSetDatum(&pColVal->value, pTColumn->type, fixed + pTColumn->offset, TYPE_BYTES[pTColumn->type]);
575,692,092✔
666
    }
667
  }
668

669
  return 0;
641,018,911✔
670
}
671

672
void tRowDestroy(SRow *pRow) {
131,631,857✔
673
  if (pRow) taosMemoryFree(pRow);
131,631,857!
674
}
131,648,017✔
675

676
static int32_t tRowPCmprFn(const void *p1, const void *p2) {
36,336,722✔
677
  SRowKey key1, key2;
678
  tRowGetKey(*(SRow **)p1, &key1);
72,673,444✔
679
  tRowGetKey(*(SRow **)p2, &key2);
73,221,294✔
680
  return tRowKeyCompare(&key1, &key2);
37,437,910✔
681
}
682
static void    tRowPDestroy(SRow **ppRow) { tRowDestroy(*ppRow); }
996✔
683
static int32_t tRowMergeImpl(SArray *aRowP, STSchema *pTSchema, int32_t iStart, int32_t iEnd, int8_t flag) {
240✔
684
  int32_t code = 0;
240✔
685

686
  int32_t    nRow = iEnd - iStart;
240✔
687
  SRowIter **aIter = NULL;
240✔
688
  SArray    *aColVal = NULL;
240✔
689
  SRow      *pRow = NULL;
240✔
690

691
  aIter = taosMemoryCalloc(nRow, sizeof(SRowIter *));
240!
692
  if (aIter == NULL) {
240!
693
    code = terrno;
×
694
    goto _exit;
×
695
  }
696

697
  for (int32_t i = 0; i < nRow; i++) {
1,236✔
698
    SRow *pRowT = taosArrayGetP(aRowP, iStart + i);
996✔
699

700
    code = tRowIterOpen(pRowT, pTSchema, &aIter[i]);
996✔
701
    if (code) goto _exit;
996!
702
  }
703

704
  // merge
705
  aColVal = taosArrayInit(pTSchema->numOfCols, sizeof(SColVal));
240✔
706
  if (aColVal == NULL) {
240!
707
    code = terrno;
×
708
    goto _exit;
×
709
  }
710

711
  for (int32_t iCol = 0; iCol < pTSchema->numOfCols; iCol++) {
920✔
712
    SColVal *pColVal = NULL;
680✔
713
    for (int32_t iRow = nRow - 1; iRow >= 0; --iRow) {
810✔
714
      SColVal *pColValT = tRowIterNext(aIter[iRow]);
766✔
715
      while (pColValT->cid < pTSchema->columns[iCol].colId) {
858✔
716
        pColValT = tRowIterNext(aIter[iRow]);
92✔
717
      }
718

719
      // todo: take strategy according to the flag
720
      if (COL_VAL_IS_VALUE(pColValT)) {
766✔
721
        pColVal = pColValT;
636✔
722
        break;
636✔
723
      } else if (COL_VAL_IS_NULL(pColValT)) {
130✔
724
        if (pColVal == NULL) {
2✔
725
          pColVal = pColValT;
1✔
726
        }
727
      }
728
    }
729

730
    if (pColVal) {
680✔
731
      if (taosArrayPush(aColVal, pColVal) == NULL) {
637!
732
        code = terrno;
×
733
        goto _exit;
×
734
      }
735
    }
736
  }
737

738
  // build
739
  code = tRowBuild(aColVal, pTSchema, &pRow);
240✔
740
  if (code) goto _exit;
240!
741

742
  taosArrayRemoveBatch(aRowP, iStart, nRow, (FDelete)tRowPDestroy);
240✔
743
  if (taosArrayInsert(aRowP, iStart, &pRow) == NULL) {
240!
744
    code = terrno;
×
745
    goto _exit;
×
746
  }
747

748
_exit:
240✔
749
  if (aIter) {
240!
750
    for (int32_t i = 0; i < nRow; i++) {
1,236✔
751
      tRowIterClose(&aIter[i]);
996✔
752
    }
753
    taosMemoryFree(aIter);
240!
754
  }
755
  if (aColVal) taosArrayDestroy(aColVal);
240!
756
  if (code) tRowDestroy(pRow);
240!
757
  return code;
240✔
758
}
759

760
int32_t tRowSort(SArray *aRowP) {
42,311✔
761
  if (TARRAY_SIZE(aRowP) <= 1) return 0;
42,311✔
762
  int32_t code = taosArrayMSort(aRowP, tRowPCmprFn);
10,173✔
763
  if (code != TSDB_CODE_SUCCESS) {
10,188!
764
    uError("taosArrayMSort failed caused by %d", code);
×
765
  }
766
  return code;
10,188✔
767
}
768

769
int32_t tRowMerge(SArray *aRowP, STSchema *pTSchema, int8_t flag) {
42,362✔
770
  int32_t code = 0;
42,362✔
771

772
  int32_t iStart = 0;
42,362✔
773
  while (iStart < aRowP->size) {
6,660,527✔
774
    SRowKey key1;
775
    SRow   *row1 = (SRow *)taosArrayGetP(aRowP, iStart);
6,614,326✔
776

777
    tRowGetKey(row1, &key1);
13,214,428✔
778

779
    int32_t iEnd = iStart + 1;
6,610,147✔
780
    while (iEnd < aRowP->size) {
6,614,677✔
781
      SRowKey key2;
782
      SRow   *row2 = (SRow *)taosArrayGetP(aRowP, iEnd);
6,565,166✔
783
      tRowGetKey(row2, &key2);
13,111,840✔
784

785
      if (tRowKeyCompare(&key1, &key2) != 0) break;
6,573,184✔
786

787
      iEnd++;
4,530✔
788
    }
789

790
    if (iEnd - iStart > 1) {
6,618,165✔
791
      code = tRowMergeImpl(aRowP, pTSchema, iStart, iEnd, flag);
240✔
792
      if (code) return code;
240!
793
    }
794

795
    // the array is also changing, so the iStart just ++ instead of iEnd
796
    iStart++;
6,618,165✔
797
  }
798

799
  return code;
46,201✔
800
}
801

802
// SRowIter ========================================
803
struct SRowIter {
804
  SRow     *pRow;
805
  STSchema *pTSchema;
806

807
  int32_t iTColumn;
808
  union {
809
    struct {  // kv
810
      int32_t iCol;
811
      SKVIdx *pIdx;
812
    };
813
    struct {  // tuple
814
      uint8_t *pb;
815
      uint8_t *pf;
816
    };
817
  };
818
  uint8_t *pv;
819
  SColVal  cv;
820
};
821

822
int32_t tRowIterOpen(SRow *pRow, STSchema *pTSchema, SRowIter **ppIter) {
897,626✔
823
  if (!(pRow->sver == pTSchema->version)) return TSDB_CODE_INVALID_PARA;
897,626!
824

825
  int32_t code = 0;
897,626✔
826

827
  SRowIter *pIter = taosMemoryCalloc(1, sizeof(*pIter));
897,626!
828
  if (pIter == NULL) {
937,659!
829
    code = terrno;
×
830
    goto _exit;
×
831
  }
832

833
  pIter->pRow = pRow;
937,659✔
834
  pIter->pTSchema = pTSchema;
937,659✔
835
  pIter->iTColumn = 0;
937,659✔
836

837
  if (pRow->flag == HAS_NONE || pRow->flag == HAS_NULL) goto _exit;
937,659✔
838

839
  uint8_t         *data = pRow->data;
936,330✔
840
  SPrimaryKeyIndex index;
841
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
936,371✔
842
    data += tGetPrimaryKeyIndex(data, &index);
41✔
843
  }
844

845
  if (pRow->flag >> 4) {
936,330✔
846
    pIter->iCol = 0;
932,665✔
847
    pIter->pIdx = (SKVIdx *)data;
932,665✔
848
    if (pRow->flag & KV_FLG_LIT) {
932,665!
849
      pIter->pv = pIter->pIdx->idx + pIter->pIdx->nCol;
932,697✔
850
    } else if (pRow->flag & KV_FLG_MID) {
×
851
      pIter->pv = pIter->pIdx->idx + (pIter->pIdx->nCol << 1);  // * sizeof(uint16_t)
×
852
    } else {
853
      pIter->pv = pIter->pIdx->idx + (pIter->pIdx->nCol << 2);  // * sizeof(uint32_t)
×
854
    }
855
  } else {
856
    switch (pRow->flag) {
3,665!
857
      case (HAS_NULL | HAS_NONE):
1✔
858
        pIter->pb = data;
1✔
859
        break;
1✔
860
      case HAS_VALUE:
2,839✔
861
        pIter->pf = data;
2,839✔
862
        pIter->pv = pIter->pf + pTSchema->flen;
2,839✔
863
        break;
2,839✔
864
      case (HAS_VALUE | HAS_NONE):
741✔
865
      case (HAS_VALUE | HAS_NULL):
866
        pIter->pb = data;
741✔
867
        pIter->pf = data + BIT1_SIZE(pTSchema->numOfCols - 1);
741✔
868
        pIter->pv = pIter->pf + pTSchema->flen;
741✔
869
        break;
741✔
870
      case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
871
        pIter->pb = data;
×
872
        pIter->pf = data + BIT2_SIZE(pTSchema->numOfCols - 1);
×
873
        pIter->pv = pIter->pf + pTSchema->flen;
×
874
        break;
×
875
      default:
84✔
876
        break;
84✔
877
    }
878
  }
879

880
_exit:
937,659✔
881
  if (code) {
937,659!
882
    *ppIter = NULL;
×
883
  } else {
884
    *ppIter = pIter;
937,659✔
885
  }
886
  return code;
937,659✔
887
}
888

889
void tRowIterClose(SRowIter **ppIter) {
875,876✔
890
  SRowIter *pIter = *ppIter;
875,876✔
891
  if (pIter) {
875,876!
892
    taosMemoryFree(pIter);
876,292!
893
  }
894
  *ppIter = NULL;
947,228✔
895
}
947,228✔
896

897
SColVal *tRowIterNext(SRowIter *pIter) {
7,573,254✔
898
  if (pIter->iTColumn >= pIter->pTSchema->numOfCols) {
7,573,254✔
899
    return NULL;
881,024✔
900
  }
901

902
  STColumn *pTColumn = pIter->pTSchema->columns + pIter->iTColumn;
6,692,230✔
903

904
  // timestamp
905
  if (0 == pIter->iTColumn) {
6,692,230✔
906
    pIter->cv.cid = pTColumn->colId;
909,071✔
907
    pIter->cv.value.type = pTColumn->type;
909,071✔
908
    pIter->cv.flag = CV_FLAG_VALUE;
909,071✔
909
    VALUE_SET_TRIVIAL_DATUM(&pIter->cv.value, pIter->pRow->ts);
909,071✔
910
    goto _exit;
909,071✔
911
  }
912

913
  if (pIter->pRow->flag == HAS_NONE) {
5,783,159✔
914
    pIter->cv = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
1✔
915
    goto _exit;
1✔
916
  }
917

918
  if (pIter->pRow->flag == HAS_NULL) {
5,783,158✔
919
    pIter->cv = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
1,841✔
920
    goto _exit;
1,841✔
921
  }
922

923
  if (pIter->pRow->flag >> 4) {  // KV
5,781,317!
924
    if (pIter->iCol < pIter->pIdx->nCol) {
6,521,653!
925
      uint8_t *pData;
926

927
      if (pIter->pRow->flag & KV_FLG_LIT) {
6,540,638!
928
        pData = pIter->pv + ((uint8_t *)pIter->pIdx->idx)[pIter->iCol];
6,548,429✔
929
      } else if (pIter->pRow->flag & KV_FLG_MID) {
×
930
        pData = pIter->pv + ((uint16_t *)pIter->pIdx->idx)[pIter->iCol];
×
931
      } else {
932
        pData = pIter->pv + ((uint32_t *)pIter->pIdx->idx)[pIter->iCol];
×
933
      }
934

935
      int16_t cid;
936
      pData += tGetI16v(pData, &cid);
6,540,638✔
937

938
      if (TABS(cid) == pTColumn->colId) {
6,540,638!
939
        if (cid < 0) {
6,706,813✔
940
          pIter->cv = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
1,694,151✔
941
        } else {
942
          pIter->cv.cid = pTColumn->colId;
5,012,662✔
943
          pIter->cv.value.type = pTColumn->type;
5,012,662✔
944
          pIter->cv.flag = CV_FLAG_VALUE;
5,012,662✔
945

946
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
5,012,662!
947
            pData += tGetU32v(pData, &pIter->cv.value.nData);
341,372!
948
            if (pIter->cv.value.nData > 0) {
341,372!
949
              pIter->cv.value.pData = pData;
1,749,241✔
950
            } else {
951
              pIter->cv.value.pData = NULL;
×
952
            }
953
          } else {
954
            valueSetDatum(&pIter->cv.value, pTColumn->type, pData, pTColumn->bytes);
4,671,290✔
955
          }
956
        }
957

958
        pIter->iCol++;
7,477,612✔
959
        goto _exit;
7,477,612✔
960
      } else if (TABS(cid) > pTColumn->colId) {
×
961
        pIter->cv = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
82✔
962
        goto _exit;
82✔
963
      } else {
964
        uError("unexpected column id %d, %d", cid, pTColumn->colId);
×
965
        goto _exit;
×
966
      }
967
    } else {
968
      pIter->cv = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
×
969
      goto _exit;
×
970
    }
971
  } else {  // Tuple
UNCOV
972
    uint8_t bv = BIT_FLG_VALUE;
×
UNCOV
973
    if (pIter->pb) {
×
974
      switch (pIter->pRow->flag) {
3,196!
975
        case (HAS_NULL | HAS_NONE):
3✔
976
          bv = GET_BIT1(pIter->pb, pIter->iTColumn - 1);
3✔
977
          break;
3✔
978
        case (HAS_VALUE | HAS_NONE):
31✔
979
          bv = GET_BIT1(pIter->pb, pIter->iTColumn - 1);
31✔
980
          if (bv) bv++;
31✔
981
          break;
31✔
982
        case (HAS_VALUE | HAS_NULL):
3,162✔
983
          bv = GET_BIT1(pIter->pb, pIter->iTColumn - 1) + 1;
3,162✔
984
          break;
3,162✔
985
        case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
986
          bv = GET_BIT2(pIter->pb, pIter->iTColumn - 1);
×
987
          break;
×
988
        default:
×
989
          break;
×
990
      }
991

992
      if (bv == BIT_FLG_NONE) {
3,196✔
993
        pIter->cv = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
13✔
994
        goto _exit;
13✔
995
      } else if (bv == BIT_FLG_NULL) {
3,183✔
996
        pIter->cv = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
343✔
997
        goto _exit;
343✔
998
      }
999
    }
1000

UNCOV
1001
    pIter->cv.cid = pTColumn->colId;
×
UNCOV
1002
    pIter->cv.value.type = pTColumn->type;
×
UNCOV
1003
    pIter->cv.flag = CV_FLAG_VALUE;
×
UNCOV
1004
    if (IS_VAR_DATA_TYPE(pTColumn->type)) {
×
1005
      uint8_t *pData = pIter->pv + *(int32_t *)(pIter->pf + pTColumn->offset);
×
1006
      pData += tGetU32v(pData, &pIter->cv.value.nData);
×
1007
      if (pIter->cv.value.nData > 0) {
×
1008
        pIter->cv.value.pData = pData;
2,681✔
1009
      } else {
1010
        pIter->cv.value.pData = NULL;
×
1011
      }
1012
    } else {
1013
      valueSetDatum(&pIter->cv.value, pTColumn->type, pIter->pf + pTColumn->offset, TYPE_BYTES[pTColumn->type]);
30,864✔
1014
    }
1015
    goto _exit;
33,529✔
1016
  }
1017

1018
_exit:
8,403,507✔
1019
  pIter->iTColumn++;
8,403,507✔
1020
  return &pIter->cv;
8,403,507✔
1021
}
1022

1023
static int32_t tRowNoneUpsertColData(SColData *aColData, int32_t nColData, int32_t flag) {
1,062✔
1024
  int32_t code = 0;
1,062✔
1025

1026
  if (flag) return code;
1,062!
1027

1028
  for (int32_t iColData = 0; iColData < nColData; iColData++) {
7,149✔
1029
    code = tColDataAppendValueImpl[aColData[iColData].flag][CV_FLAG_NONE](&aColData[iColData], NULL, 0);
6,068✔
1030
    if (code) return code;
6,087!
1031
  }
1032

1033
  return code;
1,081✔
1034
}
1035
static int32_t tRowNullUpsertColData(SColData *aColData, int32_t nColData, STSchema *pSchema, int32_t flag) {
35,019✔
1036
  int32_t code = 0;
35,019✔
1037

1038
  int32_t   iColData = 0;
35,019✔
1039
  SColData *pColData = &aColData[iColData];
35,019✔
1040
  int32_t   iTColumn = 1;
35,019✔
1041
  STColumn *pTColumn = &pSchema->columns[iTColumn];
35,019✔
1042

1043
  while (pColData) {
244,307✔
1044
    if (pTColumn) {
209,232!
1045
      if (pTColumn->colId == pColData->cid) {  // NULL
209,232✔
1046
        if (flag == 0) {
209,227✔
1047
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
209,054✔
1048
        } else {
1049
          code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
173✔
1050
        }
1051
        if (code) goto _exit;
209,283!
1052

1053
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
209,283✔
1054
        pTColumn = (++iTColumn < pSchema->numOfCols) ? &pSchema->columns[iTColumn] : NULL;
209,283✔
1055
      } else if (pTColumn->colId > pColData->cid) {  // NONE
5!
1056
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
1057
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1058
      } else {
1059
        pTColumn = (++iTColumn < pSchema->numOfCols) ? &pSchema->columns[iTColumn] : NULL;
5!
1060
      }
1061
    } else {  // NONE
1062
      if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
1063
      pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1064
    }
1065
  }
1066

1067
_exit:
35,075✔
1068
  return code;
35,075✔
1069
}
1070
static int32_t tRowTupleUpsertColData(SRow *pRow, STSchema *pTSchema, SColData *aColData, int32_t nColData,
73,835,728✔
1071
                                      int32_t flag) {
1072
  int32_t code = 0;
73,835,728✔
1073

1074
  int32_t   iColData = 0;
73,835,728✔
1075
  SColData *pColData = &aColData[iColData];
73,835,728✔
1076
  int32_t   iTColumn = 1;
73,835,728✔
1077
  STColumn *pTColumn = &pTSchema->columns[iTColumn];
73,835,728✔
1078

1079
  uint8_t         *pb = NULL, *pf = NULL, *pv = NULL;
73,835,728✔
1080
  SPrimaryKeyIndex index;
1081
  uint8_t         *data = pRow->data;
73,835,728✔
1082
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
73,836,073✔
1083
    data += tGetPrimaryKeyIndex(data, &index);
345✔
1084
  }
1085

1086
  switch (pRow->flag) {
73,835,728!
1087
    case HAS_VALUE:
73,781,079✔
1088
      pf = data;  // TODO: fix here
73,781,079✔
1089
      pv = pf + pTSchema->flen;
73,781,079✔
1090
      break;
73,781,079✔
1091
    case (HAS_NULL | HAS_NONE):
1,297✔
1092
      pb = data;
1,297✔
1093
      break;
1,297✔
1094
    case (HAS_VALUE | HAS_NONE):
147,216✔
1095
    case (HAS_VALUE | HAS_NULL):
1096
      pb = data;
147,216✔
1097
      pf = pb + BIT1_SIZE(pTSchema->numOfCols - 1);
147,216✔
1098
      pv = pf + pTSchema->flen;
147,216✔
1099
      break;
147,216✔
1100
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
1101
      pb = data;
×
1102
      pf = pb + BIT2_SIZE(pTSchema->numOfCols - 1);
×
1103
      pv = pf + pTSchema->flen;
×
1104
      break;
×
1105
    default:
×
1106
      return TSDB_CODE_INVALID_DATA_FMT;
×
1107
  }
1108

1109
  while (pColData) {
440,708,205✔
1110
    if (pTColumn) {
364,931,012✔
1111
      if (pTColumn->colId == pColData->cid) {
364,510,884!
1112
        if (!(pTColumn->type == pColData->type)) {
365,682,563!
1113
          return TSDB_CODE_INVALID_PARA;
×
1114
        }
1115
        if (pb) {
365,682,563✔
1116
          uint8_t bv;
1117
          switch (pRow->flag) {
1,481,989!
1118
            case (HAS_NULL | HAS_NONE):
25,933✔
1119
              bv = GET_BIT1(pb, iTColumn - 1);
25,933✔
1120
              break;
25,933✔
1121
            case (HAS_VALUE | HAS_NONE):
199,018✔
1122
              bv = GET_BIT1(pb, iTColumn - 1);
199,018✔
1123
              if (bv) bv++;
199,018✔
1124
              break;
199,018✔
1125
            case (HAS_VALUE | HAS_NULL):
1,258,034✔
1126
              bv = GET_BIT1(pb, iTColumn - 1) + 1;
1,258,034✔
1127
              break;
1,258,034✔
1128
            case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
1129
              bv = GET_BIT2(pb, iTColumn - 1);
×
1130
              break;
×
1131
            default:
×
1132
              return TSDB_CODE_INVALID_DATA_FMT;
×
1133
          }
1134

1135
          if (bv == BIT_FLG_NONE) {
1,482,985✔
1136
            if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0)))
43,367!
1137
              goto _exit;
×
1138
            goto _continue;
43,364✔
1139
          } else if (bv == BIT_FLG_NULL) {
1,439,618✔
1140
            if (flag == 0) {
130,094✔
1141
              code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
127,919✔
1142
            } else {
1143
              code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
2,175✔
1144
            }
1145
            if (code) goto _exit;
130,107!
1146
            goto _continue;
130,107✔
1147
          }
1148
        }
1149

1150
        if (IS_VAR_DATA_TYPE(pColData->type)) {
414,438,100!
1151
          uint8_t *pData = pv + *(int32_t *)(pf + pTColumn->offset);
44,937,736!
1152
          uint32_t nData;
1153
          pData += tGetU32v(pData, &nData);
44,937,736✔
1154
          if (flag == 0) {
44,937,736!
1155
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData);
48,163,387✔
1156
          } else {
1157
            code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData, flag > 0);
×
1158
          }
1159
          if (code) goto _exit;
48,928,002!
1160
        } else {
1161
          if (flag == 0) {
320,572,362✔
1162
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pf + pTColumn->offset,
319,711,204✔
1163
                                                                          TYPE_BYTES[pColData->type]);
319,711,204✔
1164
          } else {
1165
            code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pf + pTColumn->offset,
861,158✔
1166
                                                                          TYPE_BYTES[pColData->type], flag > 0);
861,158✔
1167
          }
1168
          if (code) goto _exit;
318,428,702!
1169
        }
1170

1171
      _continue:
318,428,702✔
1172
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
367,530,175✔
1173
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
367,530,175✔
UNCOV
1174
      } else if (pTColumn->colId > pColData->cid) {  // NONE
×
1175
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
1176
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1177
      } else {
UNCOV
1178
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
×
1179
      }
1180
    } else {
1181
      if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
420,128!
1182
      pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
420,117✔
1183
    }
1184
  }
1185

1186
_exit:
75,777,193✔
1187
  return code;
75,777,193✔
1188
}
1189
static int32_t tRowKVUpsertColData(SRow *pRow, STSchema *pTSchema, SColData *aColData, int32_t nColData, int32_t flag) {
3,098,068✔
1190
  int32_t code = 0;
3,098,068✔
1191

1192
  uint8_t  *pv = NULL;
3,098,068✔
1193
  int32_t   iColData = 0;
3,098,068✔
1194
  SColData *pColData = &aColData[iColData];
3,098,068✔
1195
  int32_t   iTColumn = 1;
3,098,068✔
1196
  STColumn *pTColumn = &pTSchema->columns[iTColumn];
3,098,068✔
1197
  int32_t   iCol = 0;
3,098,068✔
1198

1199
  // primary keys
1200
  uint8_t         *data = pRow->data;
3,098,068✔
1201
  SPrimaryKeyIndex index;
1202
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
3,098,068!
1203
    data += tGetPrimaryKeyIndex(data, &index);
×
1204
  }
1205

1206
  SKVIdx *pKVIdx = (SKVIdx *)data;
3,098,068✔
1207
  if (pRow->flag & KV_FLG_LIT) {
3,098,068✔
1208
    pv = pKVIdx->idx + pKVIdx->nCol;
3,045,420✔
1209
  } else if (pRow->flag & KV_FLG_MID) {
52,648!
1210
    pv = pKVIdx->idx + (pKVIdx->nCol << 1);
57,502✔
1211
  } else if (pRow->flag & KV_FLG_BIG) {
×
1212
    pv = pKVIdx->idx + (pKVIdx->nCol << 2);
×
1213
  } else {
1214
    return TSDB_CODE_INVALID_PARA;
×
1215
  }
1216

1217
  while (pColData) {
32,746,512✔
1218
    if (pTColumn) {
29,609,396✔
1219
      if (pTColumn->colId == pColData->cid) {
29,210,172✔
1220
        while (iCol < pKVIdx->nCol) {
29,164,118✔
1221
          uint8_t *pData;
1222
          if (pRow->flag & KV_FLG_LIT) {
27,795,968✔
1223
            pData = pv + ((uint8_t *)pKVIdx->idx)[iCol];
26,245,745✔
1224
          } else if (pRow->flag & KV_FLG_MID) {
1,550,223!
1225
            pData = pv + ((uint16_t *)pKVIdx->idx)[iCol];
1,550,265✔
1226
          } else if (pRow->flag & KV_FLG_BIG) {
×
1227
            pData = pv + ((uint32_t *)pKVIdx->idx)[iCol];
×
1228
          } else {
1229
            return TSDB_CODE_INVALID_DATA_FMT;
×
1230
          }
1231

1232
          int16_t cid;
1233
          pData += tGetI16v(pData, &cid);
27,796,010✔
1234

1235
          if (TABS(cid) == pTColumn->colId) {
27,796,010✔
1236
            if (cid < 0) {
27,370,837✔
1237
              if (flag == 0) {
6,084,039✔
1238
                code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
6,081,322✔
1239
              } else {
1240
                code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
2,717✔
1241
              }
1242
              if (code) goto _exit;
5,560,093!
1243
            } else {
1244
              uint32_t nData;
1245
              if (IS_VAR_DATA_TYPE(pTColumn->type)) {
21,286,798!
1246
                pData += tGetU32v(pData, &nData);
5,913,094✔
1247
              } else {
1248
                nData = 0;
15,373,704✔
1249
              }
1250
              if (flag == 0) {
21,286,798!
1251
                code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData);
21,548,713✔
1252
              } else {
1253
                code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData, flag > 0);
×
1254
              }
1255
              if (code) goto _exit;
21,845,132!
1256
            }
1257
            iCol++;
27,405,225✔
1258
            goto _continue;
27,405,225✔
1259
          } else if (TABS(cid) > pTColumn->colId) {  // NONE
425,173✔
1260
            break;
39,333✔
1261
          } else {
1262
            iCol++;
385,840✔
1263
          }
1264
        }
1265

1266
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
1,407,483!
1267

1268
      _continue:
1,407,266✔
1269
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
28,812,491✔
1270
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
28,812,491✔
1271
      } else if (pTColumn->colId > pColData->cid) {
431,894!
1272
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
1273
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1274
      } else {
1275
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
431,894✔
1276
      }
1277
    } else {
1278
      if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
399,224!
1279
      pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
399,205✔
1280
    }
1281
  }
1282

1283
_exit:
3,137,116✔
1284
  return code;
3,137,116✔
1285
}
1286
/* flag > 0: forward update
1287
 * flag == 0: append
1288
 * flag < 0: backward update
1289
 */
1290
int32_t tRowUpsertColData(SRow *pRow, STSchema *pTSchema, SColData *aColData, int32_t nColData, int32_t flag) {
77,006,516✔
1291
  if (!(pRow->sver == pTSchema->version)) return TSDB_CODE_INVALID_PARA;
77,006,516!
1292
  if (!(nColData > 0)) return TSDB_CODE_INVALID_PARA;
77,006,516!
1293

1294
  if (pRow->flag == HAS_NONE) {
77,006,516✔
1295
    return tRowNoneUpsertColData(aColData, nColData, flag);
1,062✔
1296
  } else if (pRow->flag == HAS_NULL) {
77,005,454✔
1297
    return tRowNullUpsertColData(aColData, nColData, pTSchema, flag);
35,032✔
1298
  } else if (pRow->flag >> 4) {  // KV row
76,970,422✔
1299
    return tRowKVUpsertColData(pRow, pTSchema, aColData, nColData, flag);
3,097,897✔
1300
  } else {  // TUPLE row
1301
    return tRowTupleUpsertColData(pRow, pTSchema, aColData, nColData, flag);
73,872,525✔
1302
  }
1303
}
1304

1305
void tRowGetPrimaryKey(SRow *row, SRowKey *key) {
16,177✔
1306
  key->numOfPKs = row->numOfPKs;
16,177✔
1307

1308
  if (key->numOfPKs == 0) {
16,177!
1309
    return;
×
1310
  }
1311

1312
  SPrimaryKeyIndex indices[TD_MAX_PK_COLS];
1313

1314
  uint8_t *data = row->data;
16,177✔
1315

1316
  for (int32_t i = 0; i < row->numOfPKs; i++) {
32,357✔
1317
    data += tGetPrimaryKeyIndex(data, &indices[i]);
16,178✔
1318
  }
1319

1320
  // primary keys
1321
  for (int32_t i = 0; i < row->numOfPKs; i++) {
32,400✔
1322
    key->pks[i].type = indices[i].type;
16,205✔
1323

1324
    uint8_t *tdata = data + indices[i].offset;
16,205✔
1325
    if (row->flag >> 4) {
16,205✔
1326
      tdata += tGetI16v(tdata, NULL);
658✔
1327
    }
1328

1329
    if (IS_VAR_DATA_TYPE(indices[i].type)) {
16,205!
1330
      key->pks[i].pData = tdata;
780✔
1331
      key->pks[i].pData += tGetU32v(key->pks[i].pData, &key->pks[i].nData);
1,560!
1332
    } else {
1333
      valueSetDatum(key->pks + i, indices[i].type, tdata, tDataTypes[indices[i].type].bytes);
15,425✔
1334
    }
1335
  }
1336
}
1337

1338
#define T_COMPARE_SCALAR_VALUE(TYPE, V1, V2)    \
1339
  do {                                          \
1340
    if (*(TYPE *)(V1) < *(TYPE *)(V2)) {        \
1341
      return -1;                                \
1342
    } else if (*(TYPE *)(V1) > *(TYPE *)(V2)) { \
1343
      return 1;                                 \
1344
    } else {                                    \
1345
      return 0;                                 \
1346
    }                                           \
1347
  } while (0)
1348

1349
int32_t tValueCompare(const SValue *tv1, const SValue *tv2) {
32,557✔
1350
  switch (tv1->type) {
32,557!
1351
    case TSDB_DATA_TYPE_BOOL:
×
1352
    case TSDB_DATA_TYPE_TINYINT:
1353
      T_COMPARE_SCALAR_VALUE(int8_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1354
    case TSDB_DATA_TYPE_SMALLINT:
×
1355
      T_COMPARE_SCALAR_VALUE(int16_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1356
    case TSDB_DATA_TYPE_INT:
32,288✔
1357
      T_COMPARE_SCALAR_VALUE(int32_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
32,288✔
1358
    case TSDB_DATA_TYPE_BIGINT:
69✔
1359
    case TSDB_DATA_TYPE_TIMESTAMP:
1360
      T_COMPARE_SCALAR_VALUE(int64_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
69!
1361
    case TSDB_DATA_TYPE_FLOAT:
×
1362
      T_COMPARE_SCALAR_VALUE(float, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1363
    case TSDB_DATA_TYPE_DOUBLE:
×
1364
      T_COMPARE_SCALAR_VALUE(double, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1365
    case TSDB_DATA_TYPE_UTINYINT:
×
1366
      T_COMPARE_SCALAR_VALUE(uint8_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1367
    case TSDB_DATA_TYPE_USMALLINT:
×
1368
      T_COMPARE_SCALAR_VALUE(uint16_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1369
    case TSDB_DATA_TYPE_UINT:
×
1370
      T_COMPARE_SCALAR_VALUE(uint32_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1371
    case TSDB_DATA_TYPE_UBIGINT:
×
1372
      T_COMPARE_SCALAR_VALUE(uint64_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1373
    case TSDB_DATA_TYPE_GEOMETRY:
219✔
1374
    case TSDB_DATA_TYPE_BINARY: {
1375
      int32_t ret = strncmp((const char *)tv1->pData, (const char *)tv2->pData, TMIN(tv1->nData, tv2->nData));
219✔
1376
      return ret ? ret : (tv1->nData < tv2->nData ? -1 : (tv1->nData > tv2->nData ? 1 : 0));
219✔
1377
    }
1378
    case TSDB_DATA_TYPE_NCHAR: {
×
1379
      int32_t ret = taosUcs4Compare((TdUcs4 *)tv1->pData, (TdUcs4 *)tv2->pData,
×
1380
                                    tv1->nData < tv2->nData ? tv1->nData : tv2->nData);
×
1381
      return ret ? ret : (tv1->nData < tv2->nData ? -1 : (tv1->nData > tv2->nData ? 1 : 0));
×
1382
    }
1383
    case TSDB_DATA_TYPE_VARBINARY: {
×
1384
      int32_t ret = memcmp(tv1->pData, tv2->pData, tv1->nData < tv2->nData ? tv1->nData : tv2->nData);
×
1385
      return ret ? ret : (tv1->nData < tv2->nData ? -1 : (tv1->nData > tv2->nData ? 1 : 0));
×
1386
    }
1387
    default:
×
1388
      break;
×
1389
  }
1390

1391
  return 0;
×
1392
}
1393

1394
// NOTE:
1395
// set key->numOfPKs to 0 as the smallest key with ts
1396
// set key->numOfPKs to (TD_MAX_PK_COLS + 1) as the largest key with ts
1397
FORCE_INLINE int32_t tRowKeyCompare(const SRowKey *key1, const SRowKey *key2) {
1,411,527,527✔
1398
  if (key1->ts < key2->ts) {
1,669,915,105!
1399
    return -1;
1,152,183,235✔
1400
  } else if (key1->ts > key2->ts) {
517,731,870!
1401
    return 1;
503,147,011✔
1402
  }
1403

1404
  if (key1->numOfPKs == key2->numOfPKs) {
14,584,859!
1405
    for (uint8_t iKey = 0; iKey < key1->numOfPKs; iKey++) {
21,303,001!
1406
      int32_t ret = tValueCompare(&key1->pks[iKey], &key2->pks[iKey]);
33,488✔
1407
      if (ret) return ret;
32,846!
1408
    }
1409
  } else if (key1->numOfPKs < key2->numOfPKs) {
×
1410
    return -1;
×
1411
  } else {
1412
    return 1;
×
1413
  }
1414

1415
  return 0;
21,269,513✔
1416
}
1417

1418
void tRowKeyAssign(SRowKey *pDst, SRowKey *pSrc) {
336,691,557✔
1419
  pDst->ts = pSrc->ts;
336,691,557✔
1420
  pDst->numOfPKs = pSrc->numOfPKs;
336,691,557✔
1421

1422
  if (pSrc->numOfPKs > 0) {
336,691,557✔
1423
    for (int32_t i = 0; i < pSrc->numOfPKs; ++i) {
48✔
1424
      SValue *pVal = &pDst->pks[i];
24✔
1425
      pVal->type = pSrc->pks[i].type;
24✔
1426

1427
      valueCloneDatum(pVal, pSrc->pks + i, pVal->type);
24✔
1428
    }
1429
  }
1430
}
336,691,557✔
1431

1432
// STag ========================================
1433
static int tTagValCmprFn(const void *p1, const void *p2) {
26,025,349✔
1434
  if (((STagVal *)p1)->cid < ((STagVal *)p2)->cid) {
26,025,349✔
1435
    return -1;
7,751,351✔
1436
  } else if (((STagVal *)p1)->cid > ((STagVal *)p2)->cid) {
18,273,998✔
1437
    return 1;
10,359,643✔
1438
  }
1439

1440
  return 0;
7,914,355✔
1441
}
1442
static int tTagValJsonCmprFn(const void *p1, const void *p2) {
13,698✔
1443
  return strcmp(((STagVal *)p1)[0].pKey, ((STagVal *)p2)[0].pKey);
13,698✔
1444
}
1445

1446
#ifdef TD_DEBUG_PRINT_TAG
1447
static void debugPrintTagVal(int8_t type, const void *val, int32_t vlen, const char *tag, int32_t ln) {
1448
  switch (type) {
1449
    case TSDB_DATA_TYPE_VARBINARY:
1450
    case TSDB_DATA_TYPE_JSON:
1451
    case TSDB_DATA_TYPE_VARCHAR:
1452
    case TSDB_DATA_TYPE_NCHAR:
1453
    case TSDB_DATA_TYPE_GEOMETRY: {
1454
      char tmpVal[32] = {0};
1455
      tstrncpy(tmpVal, val, vlen > 31 ? 31 : vlen);
1456
      printf("%s:%d type:%d vlen:%d, val:\"%s\"\n", tag, ln, (int32_t)type, vlen, tmpVal);
1457
    } break;
1458
    case TSDB_DATA_TYPE_FLOAT:
1459
      printf("%s:%d type:%d vlen:%d, val:%f\n", tag, ln, (int32_t)type, vlen, *(float *)val);
1460
      break;
1461
    case TSDB_DATA_TYPE_DOUBLE:
1462
      printf("%s:%d type:%d vlen:%d, val:%lf\n", tag, ln, (int32_t)type, vlen, *(double *)val);
1463
      break;
1464
    case TSDB_DATA_TYPE_BOOL:
1465
      printf("%s:%d type:%d vlen:%d, val:%" PRIu8 "\n", tag, ln, (int32_t)type, vlen, *(uint8_t *)val);
1466
      break;
1467
    case TSDB_DATA_TYPE_TINYINT:
1468
      printf("%s:%d type:%d vlen:%d, val:%" PRIi8 "\n", tag, ln, (int32_t)type, vlen, *(int8_t *)val);
1469
      break;
1470
    case TSDB_DATA_TYPE_SMALLINT:
1471
      printf("%s:%d type:%d vlen:%d, val:%" PRIi16 "\n", tag, ln, (int32_t)type, vlen, *(int16_t *)val);
1472
      break;
1473
    case TSDB_DATA_TYPE_INT:
1474
      printf("%s:%d type:%d vlen:%d, val:%" PRIi32 "\n", tag, ln, (int32_t)type, vlen, *(int32_t *)val);
1475
      break;
1476
    case TSDB_DATA_TYPE_BIGINT:
1477
      printf("%s:%d type:%d vlen:%d, val:%" PRIi64 "\n", tag, ln, (int32_t)type, vlen, *(int64_t *)val);
1478
      break;
1479
    case TSDB_DATA_TYPE_TIMESTAMP:
1480
      printf("%s:%d type:%d vlen:%d, val:%" PRIi64 "\n", tag, ln, (int32_t)type, vlen, *(int64_t *)val);
1481
      break;
1482
    case TSDB_DATA_TYPE_UTINYINT:
1483
      printf("%s:%d type:%d vlen:%d, val:%" PRIu8 "\n", tag, ln, (int32_t)type, vlen, *(uint8_t *)val);
1484
      break;
1485
    case TSDB_DATA_TYPE_USMALLINT:
1486
      printf("%s:%d type:%d vlen:%d, val:%" PRIu16 "\n", tag, ln, (int32_t)type, vlen, *(uint16_t *)val);
1487
      break;
1488
    case TSDB_DATA_TYPE_UINT:
1489
      printf("%s:%d type:%d vlen:%d, val:%" PRIu32 "\n", tag, ln, (int32_t)type, vlen, *(uint32_t *)val);
1490
      break;
1491
    case TSDB_DATA_TYPE_UBIGINT:
1492
      printf("%s:%d type:%d vlen:%d, val:%" PRIu64 "\n", tag, ln, (int32_t)type, vlen, *(uint64_t *)val);
1493
      break;
1494
    case TSDB_DATA_TYPE_NULL:
1495
      printf("%s:%d type:%d vlen:%d, val:%" PRIi8 "\n", tag, ln, (int32_t)type, vlen, *(int8_t *)val);
1496
      break;
1497
    default:
1498
      break;
1499
  }
1500
}
1501

1502
void debugPrintSTag(STag *pTag, const char *tag, int32_t ln) {
1503
  int8_t   isJson = pTag->flags & TD_TAG_JSON;
1504
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
1505
  uint8_t *p = NULL;
1506
  int16_t  offset = 0;
1507

1508
  if (isLarge) {
1509
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
1510
  } else {
1511
    p = (uint8_t *)&pTag->idx[pTag->nTag];
1512
  }
1513
  printf("%s:%d >>> STAG === %s:%s, len: %d, nTag: %d, sver:%d\n", tag, ln, isJson ? "json" : "normal",
1514
         isLarge ? "large" : "small", (int32_t)pTag->len, (int32_t)pTag->nTag, pTag->ver);
1515
  for (uint16_t n = 0; n < pTag->nTag; ++n) {
1516
    if (isLarge) {
1517
      offset = ((int16_t *)pTag->idx)[n];
1518
    } else {
1519
      offset = pTag->idx[n];
1520
    }
1521
    STagVal tagVal = {0};
1522
    if (isJson) {
1523
      tagVal.pKey = (char *)POINTER_SHIFT(p, offset);
1524
    } else {
1525
      tagVal.cid = *(int16_t *)POINTER_SHIFT(p, offset);
1526
    }
1527
    printf("%s:%d loop[%d-%d] offset=%d\n", __func__, __LINE__, (int32_t)pTag->nTag, (int32_t)n, (int32_t)offset);
1528
    tGetTagVal(p + offset, &tagVal, isJson);
1529
    if (IS_VAR_DATA_TYPE(tagVal.type)) {
1530
      debugPrintTagVal(tagVal.type, tagVal.pData, tagVal.nData, __func__, __LINE__);
1531
    } else {
1532
      debugPrintTagVal(tagVal.type, &tagVal.i64, tDataTypes[tagVal.type].bytes, __func__, __LINE__);
1533
    }
1534
  }
1535
  printf("\n");
1536
}
1537
#endif
1538

1539
static int32_t tPutTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson) {
858,501✔
1540
  int32_t n = 0;
858,501✔
1541

1542
  // key
1543
  if (isJson) {
858,501✔
1544
    n += tPutCStr(p ? p + n : p, pTagVal->pKey);
1,620✔
1545
  } else {
1546
    n += tPutI16v(p ? p + n : p, pTagVal->cid);
1,715,382✔
1547
  }
1548

1549
  // type
1550
  n += tPutI8(p ? p + n : p, pTagVal->type);
858,501✔
1551

1552
  // value
1553
  if (IS_VAR_DATA_TYPE(pTagVal->type)) {
858,501!
1554
    n += tPutBinary(p ? p + n : p, pTagVal->pData, pTagVal->nData);
654,478✔
1555
  } else {
1556
    p = p ? p + n : p;
531,262✔
1557
    n += tDataTypes[pTagVal->type].bytes;
531,262✔
1558
    if (p) (void)memcpy(p, &(pTagVal->i64), tDataTypes[pTagVal->type].bytes);
531,262✔
1559
  }
1560

1561
  return n;
858,501✔
1562
}
1563
static int32_t tGetTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson) {
25,611,158✔
1564
  int32_t n = 0;
25,611,158✔
1565

1566
  // key
1567
  if (isJson) {
25,611,158✔
1568
    n += tGetCStr(p + n, &pTagVal->pKey);
32,542!
1569
  } else {
1570
    n += tGetI16v(p + n, &pTagVal->cid);
51,189,774✔
1571
  }
1572

1573
  // type
1574
  n += tGetI8(p + n, &pTagVal->type);
25,611,158!
1575

1576
  // value
1577
  if (IS_VAR_DATA_TYPE(pTagVal->type)) {
25,611,158!
1578
    n += tGetBinary(p + n, &pTagVal->pData, &pTagVal->nData);
4,650,284!
1579
  } else {
1580
    (void)memcpy(&(pTagVal->i64), p + n, tDataTypes[pTagVal->type].bytes);
23,286,016✔
1581
    n += tDataTypes[pTagVal->type].bytes;
23,286,016✔
1582
  }
1583

1584
  return n;
25,611,158✔
1585
}
1586

1587
bool tTagIsJson(const void *pTag) { return (((const STag *)pTag)->flags & TD_TAG_JSON); }
47,829✔
1588

1589
bool tTagIsJsonNull(void *data) {
8,427✔
1590
  STag  *pTag = (STag *)data;
8,427✔
1591
  int8_t isJson = tTagIsJson(pTag);
8,427✔
1592
  if (!isJson) return false;
8,427✔
1593
  return ((STag *)data)->nTag == 0;
3,584✔
1594
}
1595

1596
int32_t tTagNew(SArray *pArray, int32_t version, int8_t isJson, STag **ppTag) {
151,502✔
1597
  int32_t  code = 0;
151,502✔
1598
  uint8_t *p = NULL;
151,502✔
1599
  int16_t  n = 0;
151,502✔
1600
  int16_t  nTag = taosArrayGetSize(pArray);
151,502✔
1601
  int32_t  szTag = 0;
151,526✔
1602
  int8_t   isLarge = 0;
151,526✔
1603

1604
  // sort
1605
  if (isJson) {
151,526✔
1606
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValJsonCmprFn);
324✔
1607
  } else {
1608
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValCmprFn);
151,202✔
1609
  }
1610

1611
  // get size
1612
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
580,830✔
1613
    szTag += tPutTagVal(NULL, (STagVal *)taosArrayGet(pArray, iTag), isJson);
429,289✔
1614
  }
1615
  if (szTag <= INT8_MAX) {
151,541✔
1616
    szTag = szTag + sizeof(STag) + sizeof(int8_t) * nTag;
135,156✔
1617
  } else {
1618
    szTag = szTag + sizeof(STag) + sizeof(int16_t) * nTag;
16,385✔
1619
    isLarge = 1;
16,385✔
1620
  }
1621

1622
  // build tag
1623
  (*ppTag) = (STag *)taosMemoryCalloc(szTag, 1);
151,541!
1624
  if ((*ppTag) == NULL) {
151,562!
1625
    code = terrno;
×
1626
    goto _err;
×
1627
  }
1628
  (*ppTag)->flags = 0;
151,562✔
1629
  if (isJson) {
151,562✔
1630
    (*ppTag)->flags |= TD_TAG_JSON;
324✔
1631
  }
1632
  if (isLarge) {
151,562✔
1633
    (*ppTag)->flags |= TD_TAG_LARGE;
16,411✔
1634
  }
1635
  (*ppTag)->len = szTag;
151,562✔
1636
  (*ppTag)->nTag = nTag;
151,562✔
1637
  (*ppTag)->ver = version;
151,562✔
1638

1639
  if (isLarge) {
151,562✔
1640
    p = (uint8_t *)&((int16_t *)(*ppTag)->idx)[nTag];
16,411✔
1641
  } else {
1642
    p = (uint8_t *)&(*ppTag)->idx[nTag];
135,151✔
1643
  }
1644
  n = 0;
151,562✔
1645
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
580,884✔
1646
    if (isLarge) {
429,322✔
1647
      ((int16_t *)(*ppTag)->idx)[iTag] = n;
114,882✔
1648
    } else {
1649
      (*ppTag)->idx[iTag] = n;
314,440✔
1650
    }
1651
    n += tPutTagVal(p + n, (STagVal *)taosArrayGet(pArray, iTag), isJson);
429,322✔
1652
  }
1653
#ifdef TD_DEBUG_PRINT_TAG
1654
  debugPrintSTag(*ppTag, __func__, __LINE__);
1655
#endif
1656

1657
  return code;
151,562✔
1658

1659
_err:
×
1660
  return code;
×
1661
}
1662

1663
void tTagFree(STag *pTag) {
78,598✔
1664
  if (pTag) taosMemoryFree(pTag);
78,598!
1665
}
78,601✔
1666

1667
char *tTagValToData(const STagVal *value, bool isJson) {
2,888,287✔
1668
  if (!value) {
2,888,287!
1669
    return NULL;
×
1670
  }
1671

1672
  char  *data = NULL;
2,888,287✔
1673
  int8_t typeBytes = 0;
2,888,287✔
1674
  if (isJson) {
2,888,287✔
1675
    typeBytes = CHAR_BYTES;
7,436✔
1676
  }
1677

1678
  if (IS_VAR_DATA_TYPE(value->type)) {
2,888,287!
1679
    data = taosMemoryCalloc(1, typeBytes + VARSTR_HEADER_SIZE + value->nData);
1,085,293!
1680
    if (data == NULL) {
1,085,875!
1681
      return NULL;
×
1682
    }
1683

1684
    if (isJson) {
1,085,875✔
1685
      *data = value->type;
2,943✔
1686
    }
1687

1688
    varDataLen(data + typeBytes) = value->nData;
1,085,875✔
1689
    (void)memcpy(varDataVal(data + typeBytes), value->pData, value->nData);
1,085,875✔
1690
  } else {
1691
    data = ((char *)&(value->i64)) - typeBytes;  // json with type
1,802,994✔
1692
  }
1693

1694
  return data;
2,888,869✔
1695
}
1696

1697
bool tTagGet(const STag *pTag, STagVal *pTagVal) {
8,733,977✔
1698
  if (!pTag || !pTagVal) {
8,733,977!
1699
    return false;
×
1700
  }
1701

1702
  int16_t  lidx = 0;
8,735,058✔
1703
  int16_t  ridx = pTag->nTag - 1;
8,735,058✔
1704
  int16_t  midx;
1705
  uint8_t *p;
1706
  int8_t   isJson = pTag->flags & TD_TAG_JSON;
8,735,058✔
1707
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
8,735,058✔
1708
  int16_t  offset;
1709
  STagVal  tv;
1710
  int      c;
1711

1712
  if (isLarge) {
8,735,058✔
1713
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
1,117,864✔
1714
  } else {
1715
    p = (uint8_t *)&pTag->idx[pTag->nTag];
7,617,194✔
1716
  }
1717

1718
  pTagVal->type = TSDB_DATA_TYPE_NULL;
8,735,058✔
1719
  pTagVal->pData = NULL;
8,735,058✔
1720
  pTagVal->nData = 0;
8,735,058✔
1721
  while (lidx <= ridx) {
26,405,815✔
1722
    midx = (lidx + ridx) / 2;
25,632,461✔
1723
    if (isLarge) {
25,632,461✔
1724
      offset = ((int16_t *)pTag->idx)[midx];
3,217,717✔
1725
    } else {
1726
      offset = pTag->idx[midx];
22,414,744✔
1727
    }
1728

1729
    int32_t nt = tGetTagVal(p + offset, &tv, isJson);
25,632,461✔
1730
    if (isJson) {
25,582,426✔
1731
      c = tTagValJsonCmprFn(pTagVal, &tv);
13,486✔
1732
    } else {
1733
      c = tTagValCmprFn(pTagVal, &tv);
25,568,940✔
1734
    }
1735

1736
    if (c < 0) {
26,002,898✔
1737
      ridx = midx - 1;
7,314,270✔
1738
    } else if (c > 0) {
18,688,628✔
1739
      lidx = midx + 1;
10,356,487✔
1740
    } else {
1741
      (void)memcpy(pTagVal, &tv, sizeof(tv));
8,332,141✔
1742
      return true;
8,332,141✔
1743
    }
1744
  }
1745
  return false;
773,354✔
1746
}
1747

1748
int32_t tEncodeTag(SEncoder *pEncoder, const STag *pTag) {
539,776✔
1749
  return tEncodeBinary(pEncoder, (const uint8_t *)pTag, pTag->len);
1,079,552✔
1750
}
1751

1752
int32_t tDecodeTag(SDecoder *pDecoder, STag **ppTag) { return tDecodeBinary(pDecoder, (uint8_t **)ppTag, NULL); }
4,941,336✔
1753

1754
int32_t tTagToValArray(const STag *pTag, SArray **ppArray) {
1,862✔
1755
  int32_t  code = 0;
1,862✔
1756
  uint8_t *p = NULL;
1,862✔
1757
  STagVal  tv = {0};
1,862✔
1758
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
1,862✔
1759
  int16_t  offset = 0;
1,862✔
1760

1761
  if (isLarge) {
1,862✔
1762
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
16✔
1763
  } else {
1764
    p = (uint8_t *)&pTag->idx[pTag->nTag];
1,846✔
1765
  }
1766

1767
  (*ppArray) = taosArrayInit(pTag->nTag + 1, sizeof(STagVal));
1,862✔
1768
  if (*ppArray == NULL) {
1,862!
1769
    code = terrno;
×
1770
    goto _err;
×
1771
  }
1772

1773
  for (int16_t iTag = 0; iTag < pTag->nTag; iTag++) {
5,452✔
1774
    if (isLarge) {
3,590✔
1775
      offset = ((int16_t *)pTag->idx)[iTag];
16✔
1776
    } else {
1777
      offset = pTag->idx[iTag];
3,574✔
1778
    }
1779
    int32_t nt = tGetTagVal(p + offset, &tv, pTag->flags & TD_TAG_JSON);
3,590✔
1780
    if (taosArrayPush(*ppArray, &tv) == NULL) {
7,180!
1781
      code = terrno;
×
1782
      goto _err;
×
1783
    }
1784
  }
1785

1786
  return code;
1,862✔
1787

1788
_err:
×
1789
  return code;
×
1790
}
1791

1792
// STSchema ========================================
1793
STSchema *tBuildTSchema(SSchema *aSchema, int32_t numOfCols, int32_t version) {
4,860,397✔
1794
  STSchema *pTSchema = taosMemoryCalloc(1, sizeof(STSchema) + sizeof(STColumn) * numOfCols);
4,860,397!
1795
  if (pTSchema == NULL) {
4,862,783!
1796
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1797
    return NULL;
×
1798
  }
1799

1800
  pTSchema->numOfCols = numOfCols;
4,862,783✔
1801
  pTSchema->version = version;
4,862,783✔
1802

1803
  // timestamp column
1804
  if (!(aSchema[0].type == TSDB_DATA_TYPE_TIMESTAMP)) {
4,862,783!
1805
    terrno = TSDB_CODE_INVALID_PARA;
×
1806
    return NULL;
×
1807
  }
1808
  if (!(aSchema[0].colId == PRIMARYKEY_TIMESTAMP_COL_ID)) {
4,862,783!
1809
    terrno = TSDB_CODE_INVALID_PARA;
×
1810
    return NULL;
×
1811
  }
1812
  pTSchema->columns[0].colId = aSchema[0].colId;
4,862,783✔
1813
  pTSchema->columns[0].type = aSchema[0].type;
4,862,783✔
1814
  pTSchema->columns[0].flags = aSchema[0].flags;
4,862,783✔
1815
  pTSchema->columns[0].bytes = TYPE_BYTES[aSchema[0].type];
4,862,783✔
1816
  pTSchema->columns[0].offset = -1;
4,862,783✔
1817

1818
  // other columns
1819
  for (int32_t iCol = 1; iCol < numOfCols; iCol++) {
73,664,064✔
1820
    SSchema  *pSchema = &aSchema[iCol];
68,801,281✔
1821
    STColumn *pTColumn = &pTSchema->columns[iCol];
68,801,281✔
1822

1823
    pTColumn->colId = pSchema->colId;
68,801,281✔
1824
    pTColumn->type = pSchema->type;
68,801,281✔
1825
    pTColumn->flags = pSchema->flags;
68,801,281✔
1826
    pTColumn->offset = pTSchema->flen;
68,801,281✔
1827

1828
    if (IS_VAR_DATA_TYPE(pSchema->type)) {
68,801,281!
1829
      pTColumn->bytes = pSchema->bytes;
20,695,188✔
1830
      pTSchema->tlen += (TYPE_BYTES[pSchema->type] + pSchema->bytes);  // todo: remove
20,695,188✔
1831
    } else {
1832
      pTColumn->bytes = TYPE_BYTES[pSchema->type];
48,106,093✔
1833
      pTSchema->tlen += TYPE_BYTES[pSchema->type];  // todo: remove
48,106,093✔
1834
    }
1835

1836
    pTSchema->flen += TYPE_BYTES[pTColumn->type];
68,801,281✔
1837
  }
1838

1839
#if 1  // todo : remove this
1840
  pTSchema->tlen += (int32_t)TD_BITMAP_BYTES(numOfCols);
4,862,783✔
1841
#endif
1842

1843
  return pTSchema;
4,862,783✔
1844
}
1845

1846
static int32_t tTColumnCompare(const void *p1, const void *p2) {
97,429✔
1847
  if (((STColumn *)p1)->colId < ((STColumn *)p2)->colId) {
97,429✔
1848
    return -1;
16,973✔
1849
  } else if (((STColumn *)p1)->colId > ((STColumn *)p2)->colId) {
80,456✔
1850
    return 1;
55,069✔
1851
  }
1852

1853
  return 0;
25,387✔
1854
}
1855

1856
const STColumn *tTSchemaSearchColumn(const STSchema *pTSchema, int16_t cid) {
25,459✔
1857
  STColumn tcol = {
25,459✔
1858
      .colId = cid,
1859
  };
1860

1861
  return taosbsearch(&tcol, pTSchema->columns, pTSchema->numOfCols, sizeof(STColumn), tTColumnCompare, TD_EQ);
25,459✔
1862
}
1863

1864
// SColData ========================================
1865
void tColDataDestroy(void *ph) {
6,830,699✔
1866
  if (ph) {
6,830,699!
1867
    SColData *pColData = (SColData *)ph;
6,830,823✔
1868

1869
    tFree(pColData->pBitMap);
6,830,823!
1870
    tFree(pColData->aOffset);
6,830,836✔
1871
    tFree(pColData->pData);
6,830,853!
1872
  }
1873
}
6,831,303✔
1874

1875
void tColDataInit(SColData *pColData, int16_t cid, int8_t type, int8_t cflag) {
8,622,867✔
1876
  pColData->cid = cid;
8,622,867✔
1877
  pColData->type = type;
8,622,867✔
1878
  pColData->cflag = cflag;
8,622,867✔
1879
  tColDataClear(pColData);
8,622,867✔
1880
}
8,623,967✔
1881

1882
void tColDataClear(SColData *pColData) {
17,749,840✔
1883
  pColData->numOfNone = 0;
17,749,840✔
1884
  pColData->numOfNull = 0;
17,749,840✔
1885
  pColData->numOfValue = 0;
17,749,840✔
1886
  pColData->nVal = 0;
17,749,840✔
1887
  pColData->flag = 0;
17,749,840✔
1888
  pColData->nData = 0;
17,749,840✔
1889
}
17,749,840✔
1890

1891
void tColDataDeepClear(SColData *pColData) {
185,210✔
1892
  pColData->pBitMap = NULL;
185,210✔
1893
  pColData->aOffset = NULL;
185,210✔
1894
  pColData->pData = NULL;
185,210✔
1895

1896
  tColDataClear(pColData);
185,210✔
1897
}
185,222✔
1898

1899
static FORCE_INLINE int32_t tColDataPutValue(SColData *pColData, uint8_t *pData, uint32_t nData) {
1900
  int32_t code = 0;
895,096,192✔
1901

1902
  if (IS_VAR_DATA_TYPE(pColData->type)) {
853,490,022!
1903
    code = tRealloc((uint8_t **)(&pColData->aOffset), ((int64_t)(pColData->nVal + 1)) << 2);
81,324,642!
1904
    if (code) goto _exit;
92,287,261!
1905
    pColData->aOffset[pColData->nVal] = pColData->nData;
92,287,261✔
1906

1907
    if (nData) {
92,287,261!
1908
      code = tRealloc(&pColData->pData, pColData->nData + nData);
91,051,983!
1909
      if (code) goto _exit;
91,633,055!
1910
      (void)memcpy(pColData->pData + pColData->nData, pData, nData);
91,633,055✔
1911
      pColData->nData += nData;
91,633,055✔
1912
    }
1913
  } else {
1914
    if (!(pColData->nData == tDataTypes[pColData->type].bytes * pColData->nVal)) {
813,771,550!
1915
      return TSDB_CODE_INVALID_PARA;
×
1916
    }
1917
    code = tRealloc(&pColData->pData, pColData->nData + tDataTypes[pColData->type].bytes);
813,771,550!
1918
    if (code) goto _exit;
814,884,622!
1919
    if (pData) {
814,884,622!
1920
      (void)memcpy(pColData->pData + pColData->nData, pData, TYPE_BYTES[pColData->type]);
809,103,517✔
1921
    } else {
1922
      memset(pColData->pData + pColData->nData, 0, TYPE_BYTES[pColData->type]);
5,781,105✔
1923
    }
1924
    pColData->nData += tDataTypes[pColData->type].bytes;
814,884,622✔
1925
  }
1926
  pColData->nVal++;
907,752,955✔
1927

1928
_exit:
907,752,955✔
1929
  return code;
907,752,955✔
1930
}
1931
static FORCE_INLINE int32_t tColDataAppendValue00(SColData *pColData, uint8_t *pData, uint32_t nData) {
5,001,621✔
1932
  pColData->flag = HAS_VALUE;
5,001,719✔
1933
  pColData->numOfValue++;
5,001,621✔
1934
  return tColDataPutValue(pColData, pData, nData);
5,002,303✔
1935
}
1936
static FORCE_INLINE int32_t tColDataAppendValue01(SColData *pColData, uint8_t *pData, uint32_t nData) {
80,435✔
1937
  pColData->flag = HAS_NONE;
80,435✔
1938
  pColData->numOfNone++;
80,435✔
1939
  pColData->nVal++;
80,435✔
1940
  return 0;
80,435✔
1941
}
1942
static FORCE_INLINE int32_t tColDataAppendValue02(SColData *pColData, uint8_t *pData, uint32_t nData) {
333,405✔
1943
  pColData->flag = HAS_NULL;
333,464✔
1944
  pColData->numOfNull++;
333,464✔
1945
  pColData->nVal++;
333,464✔
1946
  return 0;
333,405✔
1947
}
1948
static FORCE_INLINE int32_t tColDataAppendValue10(SColData *pColData, uint8_t *pData, uint32_t nData) {
1,302✔
1949
  int32_t code = 0;
1,302✔
1950

1951
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
1,302✔
1952
  code = tRealloc(&pColData->pBitMap, nBit);
1,302!
1953
  if (code) return code;
1,302!
1954

1955
  memset(pColData->pBitMap, 0, nBit);
1,302✔
1956
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
1,302!
1957

1958
  pColData->flag |= HAS_VALUE;
1,302✔
1959
  pColData->numOfValue++;
1,302✔
1960

1961
  if (pColData->nVal) {
1,302!
1962
    if (IS_VAR_DATA_TYPE(pColData->type)) {
1,301!
1963
      int32_t nOffset = sizeof(int32_t) * pColData->nVal;
225✔
1964
      code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
225!
1965
      if (code) return code;
230!
1966
      memset(pColData->aOffset, 0, nOffset);
230✔
1967
    } else {
1968
      pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
1,076✔
1969
      code = tRealloc(&pColData->pData, pColData->nData);
1,076!
1970
      if (code) return code;
1,075!
1971
      memset(pColData->pData, 0, pColData->nData);
1,075✔
1972
    }
1973
  }
1974

1975
  return tColDataPutValue(pColData, pData, nData);
1,305✔
1976
}
1977
static FORCE_INLINE int32_t tColDataAppendValue11(SColData *pColData, uint8_t *pData, uint32_t nData) {
2,869,187✔
1978
  pColData->nVal++;
2,869,187✔
1979
  pColData->numOfNone++;
2,869,187✔
1980
  return 0;
2,869,187✔
1981
}
1982
static FORCE_INLINE int32_t tColDataAppendValue12(SColData *pColData, uint8_t *pData, uint32_t nData) {
600✔
1983
  int32_t code = 0;
600✔
1984

1985
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
600✔
1986
  code = tRealloc(&pColData->pBitMap, nBit);
600!
1987
  if (code) return code;
600!
1988

1989
  memset(pColData->pBitMap, 0, nBit);
600✔
1990
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
600!
1991

1992
  pColData->flag |= HAS_NULL;
600✔
1993
  pColData->numOfNull++;
600✔
1994
  pColData->nVal++;
600✔
1995

1996
  return code;
600✔
1997
}
1998
static FORCE_INLINE int32_t tColDataAppendValue20(SColData *pColData, uint8_t *pData, uint32_t nData) {
91,011✔
1999
  int32_t code = 0;
91,013✔
2000

2001
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
91,013✔
2002
  code = tRealloc(&pColData->pBitMap, nBit);
91,011!
2003
  if (code) return code;
91,013!
2004

2005
  memset(pColData->pBitMap, 0, nBit);
91,013✔
2006
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
91,013!
2007

2008
  pColData->flag |= HAS_VALUE;
91,013✔
2009
  pColData->numOfValue++;
91,013✔
2010

2011
  if (pColData->nVal) {
91,013!
2012
    if (IS_VAR_DATA_TYPE(pColData->type)) {
91,013!
2013
      int32_t nOffset = sizeof(int32_t) * pColData->nVal;
7,393✔
2014
      code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
7,393!
2015
      if (code) return code;
7,393!
2016
      memset(pColData->aOffset, 0, nOffset);
7,393✔
2017
    } else {
2018
      pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
83,620✔
2019
      code = tRealloc(&pColData->pData, pColData->nData);
83,620!
2020
      if (code) return code;
83,620!
2021
      memset(pColData->pData, 0, pColData->nData);
83,620✔
2022
    }
2023
  }
2024

2025
  return tColDataPutValue(pColData, pData, nData);
91,013✔
2026
}
2027
static FORCE_INLINE int32_t tColDataAppendValue21(SColData *pColData, uint8_t *pData, uint32_t nData) {
80✔
2028
  int32_t code = 0;
80✔
2029

2030
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
80✔
2031
  code = tRealloc(&pColData->pBitMap, nBit);
80✔
2032
  if (code) return code;
80!
2033

2034
  memset(pColData->pBitMap, 255, nBit);
80✔
2035
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
80!
2036

2037
  pColData->flag |= HAS_NONE;
80✔
2038
  pColData->numOfNone++;
80✔
2039
  pColData->nVal++;
80✔
2040

2041
  return code;
80✔
2042
}
2043
static FORCE_INLINE int32_t tColDataAppendValue22(SColData *pColData, uint8_t *pData, uint32_t nData) {
4,506,099✔
2044
  pColData->nVal++;
4,506,099✔
2045
  pColData->numOfNull++;
4,506,099✔
2046
  return 0;
4,506,099✔
2047
}
2048
static FORCE_INLINE int32_t tColDataAppendValue30(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
2049
  int32_t code = 0;
×
2050

2051
  pColData->flag |= HAS_VALUE;
×
2052
  pColData->numOfValue++;
×
2053

2054
  uint8_t *pBitMap = NULL;
×
2055
  code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
×
2056
  if (code) return code;
×
2057

2058
  for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
2059
    SET_BIT2_EX(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal));
×
2060
  }
2061
  SET_BIT2_EX(pBitMap, pColData->nVal, 2);
×
2062

2063
  tFree(pColData->pBitMap);
×
2064
  pColData->pBitMap = pBitMap;
×
2065

2066
  if (pColData->nVal) {
×
2067
    if (IS_VAR_DATA_TYPE(pColData->type)) {
×
2068
      int32_t nOffset = sizeof(int32_t) * pColData->nVal;
×
2069
      code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
×
2070
      if (code) return code;
×
2071
      memset(pColData->aOffset, 0, nOffset);
×
2072
    } else {
2073
      pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
×
2074
      code = tRealloc(&pColData->pData, pColData->nData);
×
2075
      if (code) return code;
×
2076
      memset(pColData->pData, 0, pColData->nData);
×
2077
    }
2078
  }
2079

2080
  return tColDataPutValue(pColData, pData, nData);
×
2081
}
2082
static FORCE_INLINE int32_t tColDataAppendValue31(SColData *pColData, uint8_t *pData, uint32_t nData) {
400✔
2083
  int32_t code = 0;
400✔
2084

2085
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
400!
2086
  if (code) return code;
400!
2087

2088
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
400!
2089
  pColData->numOfNone++;
400✔
2090
  pColData->nVal++;
400✔
2091

2092
  return code;
400✔
2093
}
2094
static FORCE_INLINE int32_t tColDataAppendValue32(SColData *pColData, uint8_t *pData, uint32_t nData) {
9,385✔
2095
  int32_t code = 0;
9,385✔
2096

2097
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
9,385!
2098
  if (code) return code;
9,384!
2099

2100
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
9,384!
2101
  pColData->numOfNull++;
9,384✔
2102
  pColData->nVal++;
9,384✔
2103

2104
  return code;
9,384✔
2105
}
2106
static FORCE_INLINE int32_t tColDataAppendValue40(SColData *pColData, uint8_t *pData, uint32_t nData) {
876,498,777✔
2107
  pColData->numOfValue++;
876,498,777✔
2108
  return tColDataPutValue(pColData, pData, nData);
886,656,535✔
2109
}
2110
static FORCE_INLINE int32_t tColDataAppendValue41(SColData *pColData, uint8_t *pData, uint32_t nData) {
6,083✔
2111
  int32_t code = 0;
6,083✔
2112

2113
  pColData->flag |= HAS_NONE;
6,083✔
2114
  pColData->numOfNone++;
6,083✔
2115

2116
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
6,083✔
2117
  code = tRealloc(&pColData->pBitMap, nBit);
6,083✔
2118
  if (code) return code;
6,084!
2119

2120
  memset(pColData->pBitMap, 255, nBit);
6,084✔
2121
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
6,084✔
2122

2123
  return tColDataPutValue(pColData, NULL, 0);
6,084✔
2124
}
2125
static FORCE_INLINE int32_t tColDataAppendValue42(SColData *pColData, uint8_t *pData, uint32_t nData) {
571,092✔
2126
  int32_t code = 0;
576,480✔
2127

2128
  pColData->flag |= HAS_NULL;
576,480✔
2129
  pColData->numOfNull++;
576,480✔
2130

2131
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
576,480✔
2132
  code = tRealloc(&pColData->pBitMap, nBit);
576,176✔
2133
  if (code) return code;
576,494!
2134

2135
  memset(pColData->pBitMap, 255, nBit);
576,494✔
2136
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
576,494!
2137

2138
  return tColDataPutValue(pColData, NULL, 0);
576,506✔
2139
}
2140
static FORCE_INLINE int32_t tColDataAppendValue50(SColData *pColData, uint8_t *pData, uint32_t nData) {
53,517✔
2141
  int32_t code = 0;
55,517✔
2142

2143
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
55,516!
2144
  if (code) return code;
55,516!
2145

2146
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
55,516!
2147
  pColData->numOfValue++;
55,516!
2148

2149
  return tColDataPutValue(pColData, pData, nData);
56,598✔
2150
}
2151
static FORCE_INLINE int32_t tColDataAppendValue51(SColData *pColData, uint8_t *pData, uint32_t nData) {
101,720✔
2152
  int32_t code = 0;
101,720✔
2153

2154
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
101,720!
2155
  if (code) return code;
101,749!
2156

2157
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
101,749✔
2158
  pColData->numOfNone++;
101,749✔
2159

2160
  return tColDataPutValue(pColData, NULL, 0);
101,844✔
2161
}
2162
static FORCE_INLINE int32_t tColDataAppendValue52(SColData *pColData, uint8_t *pData, uint32_t nData) {
1✔
2163
  int32_t code = 0;
1✔
2164

2165
  pColData->flag |= HAS_NULL;
1✔
2166
  pColData->numOfNull++;
1✔
2167

2168
  uint8_t *pBitMap = NULL;
1✔
2169
  code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
1!
2170
  if (code) return code;
1!
2171

2172
  for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
3!
2173
    SET_BIT2_EX(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal) ? 2 : 0);
2!
2174
  }
2175
  SET_BIT2_EX(pBitMap, pColData->nVal, 1);
1!
2176

2177
  tFree(pColData->pBitMap);
1!
2178
  pColData->pBitMap = pBitMap;
1!
2179

2180
  return tColDataPutValue(pColData, NULL, 0);
1✔
2181
}
2182
static FORCE_INLINE int32_t tColDataAppendValue60(SColData *pColData, uint8_t *pData, uint32_t nData) {
7,044,230✔
2183
  int32_t code = 0;
7,047,523✔
2184

2185
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
7,047,123!
2186
  if (code) return code;
7,659,357!
2187
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
7,659,357!
2188
  pColData->numOfValue++;
7,659,357!
2189

2190
  return tColDataPutValue(pColData, pData, nData);
10,139,776✔
2191
}
2192
static FORCE_INLINE int32_t tColDataAppendValue61(SColData *pColData, uint8_t *pData, uint32_t nData) {
3,991✔
2193
  int32_t code = 0;
3,991✔
2194

2195
  pColData->flag |= HAS_NONE;
3,991✔
2196
  pColData->numOfNone++;
3,991✔
2197

2198
  uint8_t *pBitMap = NULL;
3,991✔
2199
  code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
3,991!
2200
  if (code) return code;
3,993!
2201

2202
  for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
135,416✔
2203
    SET_BIT2_EX(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal) ? 2 : 1);
131,423✔
2204
  }
2205
  SET_BIT2_EX(pBitMap, pColData->nVal, 0);
3,993✔
2206

2207
  tFree(pColData->pBitMap);
3,993!
2208
  pColData->pBitMap = pBitMap;
3,992✔
2209

2210
  return tColDataPutValue(pColData, NULL, 0);
3,995✔
2211
}
2212
static FORCE_INLINE int32_t tColDataAppendValue62(SColData *pColData, uint8_t *pData, uint32_t nData) {
2,489,811✔
2213
  int32_t code = 0;
2,503,743✔
2214

2215
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
2,500,151!
2216
  if (code) return code;
2,519,593!
2217
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
2,519,593✔
2218
  pColData->numOfNull++;
2,519,593✔
2219

2220
  return tColDataPutValue(pColData, NULL, 0);
2,534,587✔
2221
}
2222
static FORCE_INLINE int32_t tColDataAppendValue70(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
2223
  int32_t code = 0;
×
2224

2225
  code = tRealloc(&pColData->pBitMap, BIT2_SIZE(pColData->nVal + 1));
×
2226
  if (code) return code;
×
2227
  SET_BIT2_EX(pColData->pBitMap, pColData->nVal, 2);
×
2228
  pColData->numOfValue++;
×
2229

2230
  return tColDataPutValue(pColData, pData, nData);
×
2231
}
2232
static FORCE_INLINE int32_t tColDataAppendValue71(SColData *pColData, uint8_t *pData, uint32_t nData) {
4✔
2233
  int32_t code = 0;
4✔
2234

2235
  code = tRealloc(&pColData->pBitMap, BIT2_SIZE(pColData->nVal + 1));
4!
2236
  if (code) return code;
4!
2237
  SET_BIT2_EX(pColData->pBitMap, pColData->nVal, 0);
4✔
2238
  pColData->numOfNone++;
4!
2239

2240
  return tColDataPutValue(pColData, NULL, 0);
4✔
2241
}
2242
static FORCE_INLINE int32_t tColDataAppendValue72(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
2243
  int32_t code = 0;
×
2244

2245
  code = tRealloc(&pColData->pBitMap, BIT2_SIZE(pColData->nVal + 1));
×
2246
  if (code) return code;
×
2247
  SET_BIT2_EX(pColData->pBitMap, pColData->nVal, 1);
×
2248
  pColData->numOfNull++;
×
2249

2250
  return tColDataPutValue(pColData, NULL, 0);
×
2251
}
2252
static int32_t (*tColDataAppendValueImpl[8][3])(SColData *pColData, uint8_t *pData, uint32_t nData) = {
2253
    {tColDataAppendValue00, tColDataAppendValue01, tColDataAppendValue02},  // 0
2254
    {tColDataAppendValue10, tColDataAppendValue11, tColDataAppendValue12},  // HAS_NONE
2255
    {tColDataAppendValue20, tColDataAppendValue21, tColDataAppendValue22},  // HAS_NULL
2256
    {tColDataAppendValue30, tColDataAppendValue31, tColDataAppendValue32},  // HAS_NULL|HAS_NONE
2257
    {tColDataAppendValue40, tColDataAppendValue41, tColDataAppendValue42},  // HAS_VALUE
2258
    {tColDataAppendValue50, tColDataAppendValue51, tColDataAppendValue52},  // HAS_VALUE|HAS_NONE
2259
    {tColDataAppendValue60, tColDataAppendValue61, tColDataAppendValue62},  // HAS_VALUE|HAS_NULL
2260
    {tColDataAppendValue70, tColDataAppendValue71, tColDataAppendValue72},  // HAS_VALUE|HAS_NULL|HAS_NONE
2261

2262
    //       VALUE                  NONE                     NULL
2263
};
2264
int32_t tColDataAppendValue(SColData *pColData, SColVal *pColVal) {
253,539,309✔
2265
  if (!(pColData->cid == pColVal->cid && pColData->type == pColVal->value.type)) {
253,539,309!
2266
    return TSDB_CODE_INVALID_PARA;
×
2267
  }
2268
  return tColDataAppendValueImpl[pColData->flag][pColVal->flag](
253,872,112✔
2269
      pColData, VALUE_GET_DATUM(&pColVal->value, pColData->type), pColVal->value.nData);
253,872,112!
2270
}
2271

2272
static FORCE_INLINE int32_t tColDataUpdateValue10(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
9✔
2273
  pColData->numOfNone--;
9✔
2274
  pColData->nVal--;
9✔
2275
  if (pColData->numOfNone) {
9!
2276
    return tColDataAppendValue10(pColData, pData, nData);
×
2277
  } else {
2278
    pColData->flag = 0;
9✔
2279
    return tColDataAppendValue00(pColData, pData, nData);
9✔
2280
  }
2281
}
2282
static FORCE_INLINE int32_t tColDataUpdateValue12(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
24✔
2283
  pColData->numOfNone--;
24✔
2284
  pColData->nVal--;
24✔
2285
  if (pColData->numOfNone) {
24!
2286
    return tColDataAppendValue12(pColData, pData, nData);
×
2287
  } else {
2288
    pColData->flag = 0;
24✔
2289
    return tColDataAppendValue02(pColData, pData, nData);
24✔
2290
  }
2291
  return 0;
2292
}
2293
static FORCE_INLINE int32_t tColDataUpdateValue20(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
89✔
2294
  if (forward) {
89!
2295
    pColData->numOfNull--;
89✔
2296
    pColData->nVal--;
89✔
2297
    if (pColData->numOfNull) {
89!
2298
      return tColDataAppendValue20(pColData, pData, nData);
×
2299
    } else {
2300
      pColData->flag = 0;
89✔
2301
      return tColDataAppendValue00(pColData, pData, nData);
89✔
2302
    }
2303
  }
2304
  return 0;
×
2305
}
2306
static FORCE_INLINE int32_t tColDataUpdateValue30(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
2307
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> VALUE
×
2308
    pColData->numOfNone--;
×
2309
    pColData->nVal--;
×
2310
    if (pColData->numOfNone) {
×
2311
      return tColDataAppendValue30(pColData, pData, nData);
×
2312
    } else {
2313
      pColData->flag = HAS_NULL;
×
2314
      return tColDataAppendValue20(pColData, pData, nData);
×
2315
    }
2316
  } else if (forward) {  // NULL ==> VALUE
×
2317
    pColData->numOfNull--;
×
2318
    pColData->nVal--;
×
2319
    if (pColData->numOfNull) {
×
2320
      return tColDataAppendValue30(pColData, pData, nData);
×
2321
    } else {
2322
      pColData->flag = HAS_NONE;
×
2323
      return tColDataAppendValue10(pColData, pData, nData);
×
2324
    }
2325
  }
2326
  return 0;
×
2327
}
2328
static FORCE_INLINE int32_t tColDataUpdateValue32(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
80✔
2329
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> NULL
80!
2330
    pColData->numOfNone--;
80✔
2331
    pColData->numOfNull++;
80✔
2332
    if (pColData->numOfNone) {
80!
2333
      SET_BIT1(pColData->pBitMap, pColData->nVal - 1, 1);
×
2334
    } else {
2335
      pColData->flag = HAS_NULL;
80✔
2336
    }
2337
  }
2338
  return 0;
80✔
2339
}
2340
static FORCE_INLINE int32_t tColDataUpdateValue40(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
2,547,625✔
2341
  if (forward) {  // VALUE ==> VALUE
2,547,625!
2342
    pColData->nVal--;
2,547,766✔
2343
    if (IS_VAR_DATA_TYPE(pColData->type)) {
2,547,766!
2344
      pColData->nData = pColData->aOffset[pColData->nVal];
751,327✔
2345
    } else {
2346
      pColData->nData -= TYPE_BYTES[pColData->type];
1,796,439✔
2347
    }
2348
    return tColDataPutValue(pColData, pData, nData);
2,572,120✔
2349
  }
2350
  return 0;
×
2351
}
2352
static FORCE_INLINE int32_t tColDataUpdateValue42(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
5,119✔
2353
  if (forward) {  // VALUE ==> NULL
5,119!
2354
    pColData->numOfValue--;
5,119✔
2355
    pColData->nVal--;
5,119✔
2356
    if (pColData->numOfValue) {
5,119✔
2357
      if (IS_VAR_DATA_TYPE(pColData->type)) {
5,084!
2358
        pColData->nData = pColData->aOffset[pColData->nVal];
1,040✔
2359
      } else {
2360
        pColData->nData -= TYPE_BYTES[pColData->type];
4,044✔
2361
      }
2362
      return tColDataAppendValue42(pColData, pData, nData);
5,086✔
2363
    } else {
2364
      pColData->flag = 0;
35✔
2365
      pColData->nData = 0;
35✔
2366
      return tColDataAppendValue02(pColData, pData, nData);
35✔
2367
    }
2368
  }
2369
  return 0;
×
2370
}
2371
static FORCE_INLINE int32_t tColDataUpdateValue50(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
5,709✔
2372
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> VALUE
5,709✔
2373
    pColData->numOfNone--;
5,592✔
2374
    pColData->nVal--;
5,592✔
2375
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
5,592!
2376
      pColData->nData -= TYPE_BYTES[pColData->type];
4,473✔
2377
    }
2378
    if (pColData->numOfNone) {
5,592✔
2379
      return tColDataAppendValue50(pColData, pData, nData);
1,999✔
2380
    } else {
2381
      pColData->flag = HAS_VALUE;
3,593✔
2382
      return tColDataAppendValue40(pColData, pData, nData);
3,597✔
2383
    }
2384
  } else if (forward) {  // VALUE ==> VALUE
117!
2385
    pColData->nVal--;
117✔
2386
    if (IS_VAR_DATA_TYPE(pColData->type)) {
117!
2387
      pColData->nData = pColData->aOffset[pColData->nVal];
26✔
2388
    } else {
2389
      pColData->nData -= TYPE_BYTES[pColData->type];
91✔
2390
    }
2391
    return tColDataPutValue(pColData, pData, nData);
117✔
2392
  }
2393
  return 0;
×
2394
}
2395
static FORCE_INLINE int32_t tColDataUpdateValue52(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
304✔
2396
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> NULL
304!
2397
    pColData->numOfNone--;
304✔
2398
    pColData->nVal--;
304✔
2399
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
304!
2400
      pColData->nData -= TYPE_BYTES[pColData->type];
244✔
2401
    }
2402
    if (pColData->numOfNone) {
304!
2403
      return tColDataAppendValue52(pColData, pData, nData);
×
2404
    } else {
2405
      pColData->flag = HAS_VALUE;
304!
2406
      return tColDataAppendValue42(pColData, pData, nData);
304✔
2407
    }
2408
  } else if (forward) {  // VALUE ==> NULL
×
2409
    pColData->numOfValue--;
×
2410
    pColData->nVal--;
×
2411
    if (pColData->numOfValue) {
×
2412
      if (IS_VAR_DATA_TYPE(pColData->type)) {
×
2413
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2414
      } else {
2415
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2416
      }
2417
      return tColDataAppendValue52(pColData, pData, nData);
×
2418
    } else {
2419
      pColData->flag = HAS_NONE;
×
2420
      pColData->nData = 0;
×
2421
      return tColDataAppendValue12(pColData, pData, nData);
×
2422
    }
2423
  }
2424
  return 0;
×
2425
}
2426
static FORCE_INLINE int32_t tColDataUpdateValue60(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
32,003✔
2427
  if (forward) {
32,003!
2428
    if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NULL ==> VALUE
32,004✔
2429
      pColData->numOfNull--;
21,838✔
2430
      pColData->nVal--;
21,838✔
2431
      if (!IS_VAR_DATA_TYPE(pColData->type)) {
21,838!
2432
        pColData->nData -= TYPE_BYTES[pColData->type];
16,994✔
2433
      }
2434
      if (pColData->numOfNull) {
21,838✔
2435
        return tColDataAppendValue60(pColData, pData, nData);
2,893✔
2436
      } else {
2437
        pColData->flag = HAS_VALUE;
18,945✔
2438
        return tColDataAppendValue40(pColData, pData, nData);
18,947✔
2439
      }
2440
    } else {  // VALUE ==> VALUE
2441
      pColData->nVal--;
10,166✔
2442
      if (IS_VAR_DATA_TYPE(pColData->type)) {
10,166!
2443
        pColData->nData = pColData->aOffset[pColData->nVal];
2,664✔
2444
      } else {
2445
        pColData->nData -= TYPE_BYTES[pColData->type];
7,502✔
2446
      }
2447
      return tColDataPutValue(pColData, pData, nData);
10,167✔
2448
    }
2449
  }
2450
  return 0;
×
2451
}
2452
static FORCE_INLINE int32_t tColDataUpdateValue62(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
24,856✔
2453
  if (forward && (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 1)) {  // VALUE ==> NULL
24,856!
2454
    pColData->numOfValue--;
10,342✔
2455
    pColData->nVal--;
10,342✔
2456
    if (pColData->numOfValue) {
10,342✔
2457
      if (IS_VAR_DATA_TYPE(pColData->type)) {
10,340!
2458
        pColData->nData = pColData->aOffset[pColData->nVal];
2,510✔
2459
      } else {
2460
        pColData->nData -= TYPE_BYTES[pColData->type];
7,830✔
2461
      }
2462
      return tColDataAppendValue62(pColData, pData, nData);
10,342✔
2463
    } else {
2464
      pColData->flag = HAS_NULL;
2✔
2465
      pColData->nData = 0;
2!
2466
      return tColDataAppendValue20(pColData, pData, nData);
2✔
2467
    }
2468
  }
2469
  return 0;
14,514✔
2470
}
2471
static FORCE_INLINE int32_t tColDataUpdateValue70(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
401✔
2472
  int32_t code = 0;
401✔
2473

2474
  uint8_t bv = GET_BIT2(pColData->pBitMap, pColData->nVal - 1);
401✔
2475
  if (bv == 0) {  // NONE ==> VALUE
401✔
2476
    pColData->numOfNone--;
400✔
2477
    pColData->nVal--;
400✔
2478
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
400!
2479
      pColData->nData -= TYPE_BYTES[pColData->type];
320✔
2480
    }
2481
    if (pColData->numOfNone) {
400!
2482
      return tColDataAppendValue70(pColData, pData, nData);
×
2483
    } else {
2484
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
21,200✔
2485
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) - 1);
20,800✔
2486
      }
2487
      pColData->flag = (HAS_VALUE | HAS_NULL);
400!
2488
      return tColDataAppendValue60(pColData, pData, nData);
400✔
2489
    }
2490
  } else if (bv == 1) {  // NULL ==> VALUE
1!
2491
    if (forward) {
1!
2492
      pColData->numOfNull--;
1✔
2493
      pColData->nVal--;
1✔
2494
      if (!IS_VAR_DATA_TYPE(pColData->type)) {
1!
2495
        pColData->nData -= TYPE_BYTES[pColData->type];
1✔
2496
      }
2497
      if (pColData->numOfNull) {
1!
2498
        return tColDataAppendValue70(pColData, pData, nData);
×
2499
      } else {
2500
        for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
3✔
2501
          SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) ? 1 : 0);
2✔
2502
        }
2503
        pColData->flag = (HAS_VALUE | HAS_NONE);
1!
2504
        return tColDataAppendValue50(pColData, pData, nData);
1✔
2505
      }
2506
    }
2507
  } else if (bv == 2) {  // VALUE ==> VALUE
×
2508
    if (forward) {
×
2509
      pColData->nVal--;
×
2510
      if (IS_VAR_DATA_TYPE(pColData->type)) {
×
2511
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2512
      } else {
2513
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2514
      }
2515
      return tColDataPutValue(pColData, pData, nData);
×
2516
    }
2517
  } else {
2518
    return TSDB_CODE_INVALID_PARA;
×
2519
  }
2520
  return 0;
×
2521
}
2522
static int32_t tColDataUpdateValue72(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
3,593✔
2523
  uint8_t bv = GET_BIT2(pColData->pBitMap, pColData->nVal - 1);
3,593✔
2524
  if (bv == 0) {  // NONE ==> NULL
3,593✔
2525
    pColData->numOfNone--;
3,592✔
2526
    pColData->nVal--;
3,592✔
2527
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
3,592!
2528
      pColData->nData -= TYPE_BYTES[pColData->type];
2,541✔
2529
    }
2530
    if (pColData->numOfNone) {
3,592!
2531
      return tColDataAppendValue72(pColData, pData, nData);
×
2532
    } else {
2533
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
114,192✔
2534
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) - 1);
110,600✔
2535
      }
2536
      pColData->flag = (HAS_VALUE | HAS_NULL);
3,592!
2537
      return tColDataAppendValue62(pColData, pData, nData);
3,595✔
2538
    }
2539
  } else if (bv == 2 && forward) {  // VALUE ==> NULL
1!
2540
    pColData->numOfValue--;
×
2541
    pColData->nVal--;
×
2542
    if (pColData->numOfValue) {
×
2543
      if (IS_VAR_DATA_TYPE(pColData->type)) {
×
2544
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2545
      } else {
2546
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2547
      }
2548
      return tColDataAppendValue72(pColData, pData, nData);
×
2549
    } else {
2550
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
2551
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal));
×
2552
      }
2553
      pColData->flag = (HAS_NULL | HAS_NONE);
×
2554
      pColData->nData = 0;
×
2555
      return tColDataAppendValue32(pColData, pData, nData);
×
2556
    }
2557
  }
2558
  return 0;
1✔
2559
}
2560
static FORCE_INLINE int32_t tColDataUpdateNothing(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
12,519✔
2561
  return 0;
12,519✔
2562
}
2563
static int32_t (*tColDataUpdateValueImpl[8][3])(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) = {
2564
    {NULL, NULL, NULL},                                                     // 0
2565
    {tColDataUpdateValue10, tColDataUpdateNothing, tColDataUpdateValue12},  // HAS_NONE
2566
    {tColDataUpdateValue20, tColDataUpdateNothing, tColDataUpdateNothing},  // HAS_NULL
2567
    {tColDataUpdateValue30, tColDataUpdateNothing, tColDataUpdateValue32},  // HAS_NULL|HAS_NONE
2568
    {tColDataUpdateValue40, tColDataUpdateNothing, tColDataUpdateValue42},  // HAS_VALUE
2569
    {tColDataUpdateValue50, tColDataUpdateNothing, tColDataUpdateValue52},  // HAS_VALUE|HAS_NONE
2570
    {tColDataUpdateValue60, tColDataUpdateNothing, tColDataUpdateValue62},  // HAS_VALUE|HAS_NULL
2571
    {tColDataUpdateValue70, tColDataUpdateNothing, tColDataUpdateValue72},  // HAS_VALUE|HAS_NULL|HAS_NONE
2572

2573
    //    VALUE             NONE        NULL
2574
};
2575
int32_t tColDataUpdateValue(SColData *pColData, SColVal *pColVal, bool forward) {
1,501,275✔
2576
  if (!(pColData->cid == pColVal->cid && pColData->type == pColVal->value.type)) return TSDB_CODE_INVALID_PARA;
1,501,275!
2577
  if (!(pColData->nVal > 0)) return TSDB_CODE_INVALID_PARA;
1,501,518!
2578

2579
  if (tColDataUpdateValueImpl[pColData->flag][pColVal->flag] == NULL) return 0;
1,501,518!
2580

2581
  return tColDataUpdateValueImpl[pColData->flag][pColVal->flag](
1,501,518✔
2582
      pColData, VALUE_GET_DATUM(&pColVal->value, pColData->type), pColVal->value.nData, forward);
1,501,518!
2583
}
2584

2585
static FORCE_INLINE void tColDataGetValue1(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NONE
2,809,613✔
2586
  *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
2,809,613✔
2587
}
2,809,613✔
2588
static FORCE_INLINE void tColDataGetValue2(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NULL
1,547,102✔
2589
  *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
1,547,102✔
2590
}
1,547,102✔
2591
static FORCE_INLINE void tColDataGetValue3(SColData *pColData, int32_t iVal,
11,708✔
2592
                                           SColVal *pColVal) {  // HAS_NULL|HAS_NONE
2593
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
11,708!
2594
    case 0:
1,065✔
2595
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
1,065✔
2596
      break;
1,065✔
2597
    case 1:
10,645✔
2598
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
10,645✔
2599
      break;
10,645✔
2600
    default:
×
2601
      break;
×
2602
  }
2603
}
11,708✔
2604
static FORCE_INLINE void tColDataGetValue4(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_VALUE
560,334,112✔
2605
  SValue value = {.type = pColData->type};
566,422,946✔
2606
  if (IS_VAR_DATA_TYPE(pColData->type)) {
566,168,220!
2607
    if (iVal + 1 < pColData->nVal) {
112,529,358!
2608
      value.nData = pColData->aOffset[iVal + 1] - pColData->aOffset[iVal];
114,044,280✔
2609
    } else {
2610
      value.nData = pColData->nData - pColData->aOffset[iVal];
×
2611
    }
2612
    value.pData = pColData->pData + pColData->aOffset[iVal];
112,529,358✔
2613
  } else {
2614
    valueSetDatum(&value, pColData->type, pColData->pData + tDataTypes[pColData->type].bytes * iVal,
453,893,588✔
2615
                  tDataTypes[pColData->type].bytes);
453,893,588✔
2616
  }
2617
  *pColVal = COL_VAL_VALUE(pColData->cid, value);
566,173,406✔
2618
}
6,096,200✔
2619
static FORCE_INLINE void tColDataGetValue5(SColData *pColData, int32_t iVal,
225,315✔
2620
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NONE
2621
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
225,315!
2622
    case 0:
102,673✔
2623
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
102,673✔
2624
      break;
102,673✔
2625
    case 1:
122,950✔
2626
      tColDataGetValue4(pColData, iVal, pColVal);
2627
      break;
123,098✔
2628
    default:
×
2629
      break;
×
2630
  }
2631
}
225,463✔
2632
static FORCE_INLINE void tColDataGetValue6(SColData *pColData, int32_t iVal,
7,030,884✔
2633
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL
2634
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
7,030,884!
2635
    case 0:
1,087,980✔
2636
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
1,087,980✔
2637
      break;
1,087,980✔
2638
    case 1:
5,952,182✔
2639
      tColDataGetValue4(pColData, iVal, pColVal);
2640
      break;
5,956,005✔
2641
    default:
×
2642
      break;
×
2643
  }
2644
}
7,034,707✔
2645
static FORCE_INLINE void tColDataGetValue7(SColData *pColData, int32_t iVal,
11✔
2646
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL|HAS_NONE
2647
  switch (GET_BIT2(pColData->pBitMap, iVal)) {
11!
2648
    case 0:
5✔
2649
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
5✔
2650
      break;
5✔
2651
    case 1:
5✔
2652
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
5✔
2653
      break;
5✔
2654
    case 2:
1!
2655
      tColDataGetValue4(pColData, iVal, pColVal);
2656
      break;
1✔
2657
    default:
×
2658
      break;
×
2659
  }
2660
}
11✔
2661
static void (*tColDataGetValueImpl[])(SColData *pColData, int32_t iVal, SColVal *pColVal) = {
2662
    NULL,               // 0
2663
    tColDataGetValue1,  // HAS_NONE
2664
    tColDataGetValue2,  // HAS_NULL
2665
    tColDataGetValue3,  // HAS_NULL | HAS_NONE
2666
    tColDataGetValue4,  // HAS_VALUE
2667
    tColDataGetValue5,  // HAS_VALUE | HAS_NONE
2668
    tColDataGetValue6,  // HAS_VALUE | HAS_NULL
2669
    tColDataGetValue7   // HAS_VALUE | HAS_NULL | HAS_NONE
2670
};
2671
int32_t tColDataGetValue(SColData *pColData, int32_t iVal, SColVal *pColVal) {
570,797,815✔
2672
  if (iVal < 0 || iVal >= pColData->nVal ||
570,797,815!
2673
      (pColData->flag <= 0 || pColData->flag >= sizeof(tColDataGetValueImpl) / POINTER_BYTES)) {
571,626,967!
2674
    return TSDB_CODE_INVALID_PARA;
×
2675
  }
2676
  tColDataGetValueImpl[pColData->flag](pColData, iVal, pColVal);
571,813,359✔
2677
  return TSDB_CODE_SUCCESS;
576,754,578✔
2678
}
2679

2680
uint8_t tColDataGetBitValue(const SColData *pColData, int32_t iVal) {
40,554,780✔
2681
  switch (pColData->flag) {
40,554,780!
2682
    case HAS_NONE:
×
2683
      return 0;
×
2684
    case HAS_NULL:
×
2685
      return 1;
×
2686
    case (HAS_NULL | HAS_NONE):
11,715✔
2687
      return GET_BIT1(pColData->pBitMap, iVal);
11,715✔
2688
    case HAS_VALUE:
×
2689
      return 2;
×
2690
    case (HAS_VALUE | HAS_NONE):
607,401✔
2691
      return (GET_BIT1(pColData->pBitMap, iVal)) ? 2 : 0;
607,401✔
2692
    case (HAS_VALUE | HAS_NULL):
39,936,514✔
2693
      return GET_BIT1(pColData->pBitMap, iVal) + 1;
39,936,514✔
2694
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
22✔
2695
      return GET_BIT2(pColData->pBitMap, iVal);
22✔
2696
    default:
×
2697
      return 0;
×
2698
  }
2699
}
2700

2701
int32_t tColDataCopy(SColData *pColDataFrom, SColData *pColData, xMallocFn xMalloc, void *arg) {
27,994✔
2702
  int32_t code = 0;
27,994✔
2703

2704
  *pColData = *pColDataFrom;
27,994✔
2705

2706
  // bitmap
2707
  switch (pColData->flag) {
27,994!
2708
    case (HAS_NULL | HAS_NONE):
106✔
2709
    case (HAS_VALUE | HAS_NONE):
2710
    case (HAS_VALUE | HAS_NULL):
2711
      pColData->pBitMap = xMalloc(arg, BIT1_SIZE(pColData->nVal));
106✔
2712
      if (pColData->pBitMap == NULL) {
106!
2713
        code = TSDB_CODE_OUT_OF_MEMORY;
×
2714
        goto _exit;
×
2715
      }
2716
      (void)memcpy(pColData->pBitMap, pColDataFrom->pBitMap, BIT1_SIZE(pColData->nVal));
106✔
2717
      break;
106✔
2718
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
2719
      pColData->pBitMap = xMalloc(arg, BIT2_SIZE(pColData->nVal));
×
2720
      if (pColData->pBitMap == NULL) {
×
2721
        code = TSDB_CODE_OUT_OF_MEMORY;
×
2722
        goto _exit;
×
2723
      }
2724
      (void)memcpy(pColData->pBitMap, pColDataFrom->pBitMap, BIT2_SIZE(pColData->nVal));
×
2725
      break;
×
2726
    default:
27,888✔
2727
      pColData->pBitMap = NULL;
27,888✔
2728
      break;
27,888✔
2729
  }
2730

2731
  // offset
2732
  if (IS_VAR_DATA_TYPE(pColData->type) && (pColData->flag & HAS_VALUE)) {
27,994!
2733
    pColData->aOffset = xMalloc(arg, pColData->nVal << 2);
257✔
2734
    if (pColData->aOffset == NULL) {
257!
2735
      code = TSDB_CODE_OUT_OF_MEMORY;
×
2736
      goto _exit;
×
2737
    }
2738
    (void)memcpy(pColData->aOffset, pColDataFrom->aOffset, pColData->nVal << 2);
257✔
2739
  } else {
2740
    pColData->aOffset = NULL;
27,737✔
2741
  }
2742

2743
  // value
2744
  if (pColData->nData) {
27,994✔
2745
    pColData->pData = xMalloc(arg, pColData->nData);
27,904✔
2746
    if (pColData->pData == NULL) {
27,902!
2747
      code = TSDB_CODE_OUT_OF_MEMORY;
×
2748
      goto _exit;
×
2749
    }
2750

2751
    (void)memcpy(pColData->pData, pColDataFrom->pData, pColData->nData);
27,902✔
2752
  } else {
2753
    pColData->pData = NULL;
90✔
2754
  }
2755

2756
_exit:
27,992✔
2757
  return code;
27,992✔
2758
}
2759

2760
int32_t tColDataCompress(SColData *colData, SColDataCompressInfo *info, SBuffer *output, SBuffer *assist) {
5,129,685✔
2761
  int32_t code;
2762
  SBuffer local;
2763

2764
  if (!(colData->nVal > 0)) {
5,129,685!
2765
    return TSDB_CODE_INVALID_PARA;
×
2766
  }
2767

2768
  (*info) = (SColDataCompressInfo){
5,129,685✔
2769
      .cmprAlg = info->cmprAlg,
5,129,685✔
2770
      .columnFlag = colData->cflag,
5,129,685✔
2771
      .flag = colData->flag,
5,129,685✔
2772
      .dataType = colData->type,
5,129,685✔
2773
      .columnId = colData->cid,
5,129,685✔
2774
      .numOfData = colData->nVal,
5,129,685✔
2775
  };
2776

2777
  if (colData->flag == HAS_NONE || colData->flag == HAS_NULL) {
5,129,685!
2778
    return 0;
237,752✔
2779
  }
2780

2781
  tBufferInit(&local);
2782
  if (assist == NULL) {
4,891,933!
2783
    assist = &local;
×
2784
  }
2785

2786
  // bitmap
2787
  if (colData->flag != HAS_VALUE) {
4,891,933✔
2788
    if (colData->flag == (HAS_NONE | HAS_NULL | HAS_VALUE)) {
642,053✔
2789
      info->bitmapOriginalSize = BIT2_SIZE(colData->nVal);
1✔
2790
    } else {
2791
      info->bitmapOriginalSize = BIT1_SIZE(colData->nVal);
642,052✔
2792
    }
2793

2794
    SCompressInfo cinfo = {
642,053✔
2795
        .dataType = TSDB_DATA_TYPE_TINYINT,
2796
        .cmprAlg = info->cmprAlg,
642,053✔
2797
        .originalSize = info->bitmapOriginalSize,
642,053✔
2798
    };
2799

2800
    code = tCompressDataToBuffer(colData->pBitMap, &cinfo, output, assist);
642,053✔
2801
    if (code) {
642,071!
2802
      tBufferDestroy(&local);
2803
      return code;
×
2804
    }
2805

2806
    info->bitmapCompressedSize = cinfo.compressedSize;
642,071✔
2807
  }
2808

2809
  if (colData->flag == (HAS_NONE | HAS_NULL)) {
4,891,951✔
2810
    tBufferDestroy(&local);
2811
    return 0;
601✔
2812
  }
2813

2814
  // offset
2815
  if (IS_VAR_DATA_TYPE(colData->type)) {
4,891,350!
2816
    info->offsetOriginalSize = sizeof(int32_t) * info->numOfData;
953,779✔
2817

2818
    SCompressInfo cinfo = {
953,779✔
2819
        .dataType = TSDB_DATA_TYPE_INT,
2820
        .cmprAlg = info->cmprAlg,
953,779✔
2821
        .originalSize = info->offsetOriginalSize,
953,779✔
2822
    };
2823

2824
    code = tCompressDataToBuffer(colData->aOffset, &cinfo, output, assist);
953,779✔
2825
    if (code) {
953,873!
2826
      tBufferDestroy(&local);
2827
      return code;
×
2828
    }
2829

2830
    info->offsetCompressedSize = cinfo.compressedSize;
953,873✔
2831
  }
2832

2833
  // data
2834
  if (colData->nData > 0) {
4,891,444!
2835
    info->dataOriginalSize = colData->nData;
4,891,559✔
2836

2837
    SCompressInfo cinfo = {
4,891,559✔
2838
        .dataType = colData->type,
4,891,559✔
2839
        .cmprAlg = info->cmprAlg,
4,891,559✔
2840
        .originalSize = info->dataOriginalSize,
4,891,559✔
2841
    };
2842

2843
    code = tCompressDataToBuffer(colData->pData, &cinfo, output, assist);
4,891,559✔
2844
    if (code) {
4,891,999!
2845
      tBufferDestroy(&local);
2846
      return code;
×
2847
    }
2848

2849
    info->dataCompressedSize = cinfo.compressedSize;
4,891,999✔
2850
  }
2851

2852
  tBufferDestroy(&local);
2853
  return 0;
4,891,884✔
2854
}
2855

2856
int32_t tColDataDecompress(void *input, SColDataCompressInfo *info, SColData *colData, SBuffer *assist) {
3,233,425✔
2857
  int32_t  code;
2858
  SBuffer  local;
2859
  uint8_t *data = (uint8_t *)input;
3,233,425✔
2860

2861
  tBufferInit(&local);
2862
  if (assist == NULL) {
3,233,425!
2863
    assist = &local;
×
2864
  }
2865

2866
  tColDataClear(colData);
3,233,425✔
2867
  colData->cid = info->columnId;
3,233,549✔
2868
  colData->type = info->dataType;
3,233,549✔
2869
  colData->cflag = info->columnFlag;
3,233,549✔
2870
  colData->nVal = info->numOfData;
3,233,549✔
2871
  colData->flag = info->flag;
3,233,549✔
2872

2873
  if (info->flag == HAS_NONE || info->flag == HAS_NULL) {
3,233,549✔
2874
    goto _exit;
278,576✔
2875
  }
2876

2877
  // bitmap
2878
  if (info->bitmapOriginalSize > 0) {
2,954,973✔
2879
    SCompressInfo cinfo = {
423,619✔
2880
        .dataType = TSDB_DATA_TYPE_TINYINT,
2881
        .cmprAlg = info->cmprAlg,
423,619✔
2882
        .originalSize = info->bitmapOriginalSize,
423,619✔
2883
        .compressedSize = info->bitmapCompressedSize,
423,619✔
2884
    };
2885

2886
    code = tRealloc(&colData->pBitMap, cinfo.originalSize);
423,619!
2887
    if (code) {
423,625!
2888
      tBufferDestroy(&local);
2889
      return code;
×
2890
    }
2891

2892
    code = tDecompressData(data, &cinfo, colData->pBitMap, cinfo.originalSize, assist);
423,625✔
2893
    if (code) {
423,620!
2894
      tBufferDestroy(&local);
2895
      return code;
×
2896
    }
2897

2898
    data += cinfo.compressedSize;
423,620✔
2899
  }
2900

2901
  if (info->flag == (HAS_NONE | HAS_NULL)) {
2,954,974✔
2902
    goto _exit;
613✔
2903
  }
2904

2905
  // offset
2906
  if (info->offsetOriginalSize > 0) {
2,954,361✔
2907
    SCompressInfo cinfo = {
705,739✔
2908
        .cmprAlg = info->cmprAlg,
705,739✔
2909
        .dataType = TSDB_DATA_TYPE_INT,
2910
        .originalSize = info->offsetOriginalSize,
705,739✔
2911
        .compressedSize = info->offsetCompressedSize,
705,739✔
2912
    };
2913

2914
    code = tRealloc((uint8_t **)&colData->aOffset, cinfo.originalSize);
705,739!
2915
    if (code) {
705,930!
2916
      tBufferDestroy(&local);
2917
      return code;
×
2918
    }
2919

2920
    code = tDecompressData(data, &cinfo, colData->aOffset, cinfo.originalSize, assist);
705,930✔
2921
    if (code) {
705,867!
2922
      tBufferDestroy(&local);
2923
      return code;
×
2924
    }
2925

2926
    data += cinfo.compressedSize;
705,867✔
2927
  }
2928

2929
  // data
2930
  if (info->dataOriginalSize > 0) {
2,954,489✔
2931
    colData->nData = info->dataOriginalSize;
2,953,272✔
2932

2933
    SCompressInfo cinfo = {
2,953,272✔
2934
        .cmprAlg = info->cmprAlg,
2,953,272✔
2935
        .dataType = colData->type,
2,953,272✔
2936
        .originalSize = info->dataOriginalSize,
2,953,272✔
2937
        .compressedSize = info->dataCompressedSize,
2,953,272✔
2938
    };
2939

2940
    code = tRealloc((uint8_t **)&colData->pData, cinfo.originalSize);
2,953,272!
2941
    if (code) {
2,954,391!
2942
      tBufferDestroy(&local);
2943
      return code;
×
2944
    }
2945

2946
    code = tDecompressData(data, &cinfo, colData->pData, cinfo.originalSize, assist);
2,954,391✔
2947
    if (code) {
2,954,688!
2948
      tBufferDestroy(&local);
2949
      return code;
×
2950
    }
2951

2952
    data += cinfo.compressedSize;
2,954,688✔
2953
  }
2954

2955
_exit:
1,217✔
2956
  switch (colData->flag) {
3,235,094✔
2957
    case HAS_NONE:
25,461✔
2958
      colData->numOfNone = colData->nVal;
25,461✔
2959
      break;
25,461✔
2960
    case HAS_NULL:
254,079✔
2961
      colData->numOfNull = colData->nVal;
254,079✔
2962
      break;
254,079✔
2963
    case HAS_VALUE:
2,531,263✔
2964
      colData->numOfValue = colData->nVal;
2,531,263✔
2965
      break;
2,531,263✔
2966
    default:
424,291✔
2967
      for (int32_t i = 0; i < colData->nVal; i++) {
24,752,361✔
2968
        uint8_t bitValue = tColDataGetBitValue(colData, i);
24,327,933✔
2969
        if (bitValue == 0) {
24,328,070✔
2970
          colData->numOfNone++;
195,366✔
2971
        } else if (bitValue == 1) {
24,132,704✔
2972
          colData->numOfNull++;
1,699,347✔
2973
        } else {
2974
          colData->numOfValue++;
22,433,357✔
2975
        }
2976
      }
2977
  }
2978
  tBufferDestroy(&local);
2979
  return 0;
3,235,231✔
2980
}
2981

2982
int32_t tColDataAddValueByDataBlock(SColData *pColData, int8_t type, int32_t bytes, int32_t nRows, char *lengthOrbitmap,
579✔
2983
                                    char *data) {
2984
  int32_t code = 0;
579✔
2985
  if (data == NULL) {
579✔
2986
    if (pColData->cflag & COL_IS_KEY) {
13!
2987
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
2988
    } else {
2989
      for (int32_t i = 0; i < nRows; ++i) {
28✔
2990
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0);
15✔
2991
      }
2992
    }
2993
    goto _exit;
13✔
2994
  }
2995

2996
  if (IS_VAR_DATA_TYPE(type)) {  // var-length data type
566!
2997
    for (int32_t i = 0; i < nRows; ++i) {
283✔
2998
      int32_t offset = *((int32_t *)lengthOrbitmap + i);
182✔
2999
      if (offset == -1) {
182!
3000
        if (pColData->cflag & COL_IS_KEY) {
×
3001
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3002
          goto _exit;
×
3003
        }
3004
        if ((code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0))) {
×
3005
          goto _exit;
×
3006
        }
3007
      } else {
3008
        if (varDataTLen(data + offset) > bytes) {
182!
3009
          uError("var data length invalid, varDataTLen(data + offset):%d > bytes:%d", (int)varDataTLen(data + offset),
×
3010
                 bytes);
3011
          code = TSDB_CODE_PAR_VALUE_TOO_LONG;
×
3012
          goto _exit;
×
3013
        }
3014
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, (uint8_t *)varDataVal(data + offset),
182✔
3015
                                                                      varDataLen(data + offset));
182✔
3016
      }
3017
    }
3018
  } else {  // fixed-length data type
3019
    bool allValue = true;
465✔
3020
    bool allNull = true;
465✔
3021
    for (int32_t i = 0; i < nRows; ++i) {
1,304✔
3022
      if (!colDataIsNull_f(lengthOrbitmap, i)) {
839✔
3023
        allNull = false;
626✔
3024
      } else {
3025
        allValue = false;
213✔
3026
      }
3027
    }
3028
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
465!
3029
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3030
      goto _exit;
×
3031
    }
3032

3033
    if (allValue) {
465✔
3034
      // optimize (todo)
3035
      for (int32_t i = 0; i < nRows; ++i) {
983✔
3036
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, (uint8_t *)data + bytes * i, bytes);
612✔
3037
      }
3038
    } else if (allNull) {
94✔
3039
      // optimize (todo)
3040
      for (int32_t i = 0; i < nRows; ++i) {
249✔
3041
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
165✔
3042
        if (code) goto _exit;
165!
3043
      }
3044
    } else {
3045
      for (int32_t i = 0; i < nRows; ++i) {
72✔
3046
        if (colDataIsNull_f(lengthOrbitmap, i)) {
62✔
3047
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
48✔
3048
          if (code) goto _exit;
48!
3049
        } else {
3050
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, (uint8_t *)data + bytes * i, bytes);
14✔
3051
        }
3052
      }
3053
    }
3054
  }
3055

3056
_exit:
10✔
3057
  return code;
579✔
3058
}
3059

3060
int32_t tColDataAddValueByBind(SColData *pColData, TAOS_MULTI_BIND *pBind, int32_t buffMaxLen, initGeosFn igeos,
6,101,285✔
3061
                               checkWKBGeometryFn cgeos) {
3062
  int32_t code = 0;
6,101,285✔
3063

3064
  if (!(pBind->num == 1 && pBind->is_null && *pBind->is_null)) {
6,101,285✔
3065
    if (!(pColData->type == pBind->buffer_type)) {
5,534,796!
3066
      return TSDB_CODE_INVALID_PARA;
×
3067
    }
3068
  }
3069

3070
  if (IS_VAR_DATA_TYPE(pColData->type)) {  // var-length data type
6,101,285!
3071
    if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
508,421✔
3072
      code = igeos();
204✔
3073
      if (code) {
204!
3074
        return code;
×
3075
      }
3076
    }
3077
    for (int32_t i = 0; i < pBind->num; ++i) {
2,455,369✔
3078
      if (pBind->is_null && pBind->is_null[i]) {
1,800,583✔
3079
        if (pColData->cflag & COL_IS_KEY) {
104,115!
3080
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3081
          goto _exit;
×
3082
        }
3083
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
104,115✔
3084
        if (code) goto _exit;
104,718!
3085
      } else if (pBind->length[i] > buffMaxLen) {
1,696,468!
3086
        return TSDB_CODE_PAR_VALUE_TOO_LONG;
×
3087
      } else {
3088
        if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
1,696,468✔
3089
          code = cgeos((char *)pBind->buffer + pBind->buffer_length * i, (size_t)pBind->length[i]);
207✔
3090
          if (code) {
207✔
3091
            uError("stmt col[%d] bind geometry wrong format", i);
1!
3092
            goto _exit;
1✔
3093
          }
3094
        }
3095
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
1,696,467✔
3096
            pColData, (uint8_t *)pBind->buffer + pBind->buffer_length * i, pBind->length[i]);
1,696,467✔
3097
      }
3098
    }
3099
  } else {  // fixed-length data type
3100
    bool allValue;
3101
    bool allNull;
3102
    if (pBind->is_null) {
5,592,864✔
3103
      bool same = (memcmp(pBind->is_null, pBind->is_null + 1, pBind->num - 1) == 0);
555,748✔
3104
      allNull = (same && pBind->is_null[0] != 0);
555,748✔
3105
      allValue = (same && pBind->is_null[0] == 0);
555,748✔
3106
    } else {
3107
      allNull = false;
5,037,116✔
3108
      allValue = true;
5,037,116✔
3109
    }
3110

3111
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
5,592,864!
3112
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3113
      goto _exit;
×
3114
    }
3115

3116
    if (allValue) {
5,592,864✔
3117
      // optimize (todo)
3118
      for (int32_t i = 0; i < pBind->num; ++i) {
260,969,966✔
3119
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
253,004,285✔
3120
            pColData, (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i, pBind->buffer_length);
253,004,285✔
3121
      }
3122
    } else if (allNull) {
261,886!
3123
      // optimize (todo)
3124
      for (int32_t i = 0; i < pBind->num; ++i) {
972,553✔
3125
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
478,902✔
3126
        if (code) goto _exit;
493,592!
3127
      }
3128
    } else {
UNCOV
3129
      for (int32_t i = 0; i < pBind->num; ++i) {
×
3130
        if (pBind->is_null[i]) {
50,012✔
3131
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
4,245✔
3132
          if (code) goto _exit;
4,245!
3133
        } else {
3134
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
45,767✔
3135
              pColData, (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i, pBind->buffer_length);
45,767✔
3136
        }
3137
      }
3138
    }
3139
  }
3140

UNCOV
3141
_exit:
×
3142
  return code;
8,897,044✔
3143
}
3144

3145
int32_t tColDataAddValueByBind2(SColData *pColData, TAOS_STMT2_BIND *pBind, int32_t buffMaxLen, initGeosFn igeos,
213✔
3146
                                checkWKBGeometryFn cgeos) {
3147
  int32_t code = 0;
213✔
3148

3149
  if (!(pBind->num == 1 && pBind->is_null && *pBind->is_null)) {
213!
3150
    if (!(pColData->type == pBind->buffer_type)) {
213!
3151
      return TSDB_CODE_INVALID_PARA;
×
3152
    }
3153
  }
3154

3155
  if (IS_VAR_DATA_TYPE(pColData->type)) {  // var-length data type
311!
3156
    if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
98✔
3157
      code = igeos();
3✔
3158
      if (code) {
3!
3159
        return code;
×
3160
      }
3161
    }
3162

3163
    uint8_t *buf = pBind->buffer;
98✔
3164
    for (int32_t i = 0; i < pBind->num; ++i) {
388✔
3165
      if (pBind->is_null && pBind->is_null[i]) {
290✔
3166
        if (pColData->cflag & COL_IS_KEY) {
2!
3167
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3168
          goto _exit;
×
3169
        }
3170
        if (pBind->is_null[i] == 1) {
2!
3171
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
2✔
3172
          if (code) goto _exit;
2!
3173
        } else {
3174
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0);
×
3175
          if (code) goto _exit;
×
3176
        }
3177
      } else if (pBind->length[i] > buffMaxLen) {
288!
3178
        return TSDB_CODE_PAR_VALUE_TOO_LONG;
×
3179
      } else {
3180
        if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
288✔
3181
          code = cgeos(buf, pBind->length[i]);
7✔
3182
          if (code) {
7!
3183
            uError("stmt2 col[%d] bind geometry wrong format", i);
×
3184
            goto _exit;
×
3185
          }
3186
        }
3187
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, buf, pBind->length[i]);
288✔
3188
        buf += pBind->length[i];
288✔
3189
      }
3190
    }
3191
  } else {  // fixed-length data type
3192
    bool allValue;
3193
    bool allNull;
3194
    bool allNone;
3195
    if (pBind->is_null) {
115!
3196
      bool same = (memcmp(pBind->is_null, pBind->is_null + 1, pBind->num - 1) == 0);
×
3197
      allNull = (same && pBind->is_null[0] == 1);
×
3198
      allNone = (same && pBind->is_null[0] > 1);
×
3199
      allValue = (same && pBind->is_null[0] == 0);
×
3200
    } else {
3201
      allNull = false;
115✔
3202
      allNone = false;
115✔
3203
      allValue = true;
115✔
3204
    }
3205

3206
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
115!
3207
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3208
      goto _exit;
×
3209
    }
3210

3211
    if (allValue) {
115!
3212
      // optimize (todo)
3213
      for (int32_t i = 0; i < pBind->num; ++i) {
434✔
3214
        uint8_t *val = (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i;
319✔
3215
        if (TSDB_DATA_TYPE_BOOL == pColData->type && *val > 1) {
319!
3216
          *val = 1;
1✔
3217
        }
3218

3219
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, val, TYPE_BYTES[pColData->type]);
319✔
3220
      }
3221
    } else if (allNull) {
×
3222
      // optimize (todo)
3223
      for (int32_t i = 0; i < pBind->num; ++i) {
×
3224
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
3225
        if (code) goto _exit;
×
3226
      }
3227
    } else if (allNone) {
×
3228
      // optimize (todo)
3229
      for (int32_t i = 0; i < pBind->num; ++i) {
×
3230
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0);
×
3231
        if (code) goto _exit;
×
3232
      }
3233
    } else {
3234
      for (int32_t i = 0; i < pBind->num; ++i) {
×
3235
        if (pBind->is_null[i]) {
×
3236
          if (pBind->is_null[i] == 1) {
×
3237
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
3238
            if (code) goto _exit;
×
3239
          } else {
3240
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0);
×
3241
            if (code) goto _exit;
×
3242
          }
3243
        } else {
3244
          uint8_t *val = (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i;
×
3245
          if (TSDB_DATA_TYPE_BOOL == pColData->type && *val > 1) {
×
3246
            *val = 1;
×
3247
          }
3248

3249
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, val, TYPE_BYTES[pColData->type]);
×
3250
        }
3251
      }
3252
    }
3253
  }
3254

3255
_exit:
×
3256
  return code;
213✔
3257
}
3258

3259
/* build rows to `rowArray` from bind
3260
 * `infos` is the bind information array
3261
 * `numOfInfos` is the number of bind information
3262
 * `infoSorted` is whether the bind information is sorted by column id
3263
 * `pTSchema` is the schema of the table
3264
 * `rowArray` is the array to store the rows
3265
 * `pOrdered` is the pointer to store ordered
3266
 * `pDupTs` is the pointer to store duplicateTs
3267
 */
3268
int32_t tRowBuildFromBind2(SBindInfo2 *infos, int32_t numOfInfos, bool infoSorted, const STSchema *pTSchema,
5,719✔
3269
                           SArray *rowArray, bool *pOrdered, bool *pDupTs) {
3270
  if (infos == NULL || numOfInfos <= 0 || numOfInfos > pTSchema->numOfCols || pTSchema == NULL || rowArray == NULL) {
5,719!
3271
    return TSDB_CODE_INVALID_PARA;
×
3272
  }
3273

3274
  if (!infoSorted) {
5,735✔
3275
    taosqsort_r(infos, numOfInfos, sizeof(SBindInfo2), NULL, tBindInfoCompare);
1✔
3276
  }
3277

3278
  int32_t code = 0;
5,735✔
3279
  int32_t numOfRows = infos[0].bind->num;
5,735✔
3280
  SArray *colValArray, *bufArray;
3281
  SColVal colVal;
3282

3283
  if ((colValArray = taosArrayInit(numOfInfos, sizeof(SColVal))) == NULL) {
5,735!
3284
    return terrno;
×
3285
  }
3286
  if ((bufArray = taosArrayInit(numOfInfos, sizeof(uint8_t *))) == NULL) {
5,729✔
3287
    taosArrayDestroy(colValArray);
1✔
3288
    return terrno;
×
3289
  }
3290
  for (int i = 0; i < numOfInfos; ++i) {
28,448✔
3291
    if (!taosArrayPush(bufArray, &infos[i].bind->buffer)) {
45,453!
3292
      taosArrayDestroy(colValArray);
×
3293
      taosArrayDestroy(bufArray);
×
3294
      return terrno;
×
3295
    }
3296
  }
3297

3298
  SRowKey rowKey, lastRowKey;
3299
  for (int32_t iRow = 0; iRow < numOfRows; iRow++) {
236,172!
3300
    taosArrayClear(colValArray);
246,342✔
3301

3302
    for (int32_t iInfo = 0; iInfo < numOfInfos; iInfo++) {
1,052,791✔
3303
      if (infos[iInfo].bind->is_null && infos[iInfo].bind->is_null[iRow]) {
781,147✔
3304
        if (infos[iInfo].bind->is_null[iRow] == 1) {
2!
3305
          if(iInfo == 0) {
2✔
3306
            code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
1✔
3307
            goto _exit;
1✔
3308
          }
3309
          colVal = COL_VAL_NULL(infos[iInfo].columnId, infos[iInfo].type);
1✔
3310
        } else {
3311
          colVal = COL_VAL_NONE(infos[iInfo].columnId, infos[iInfo].type);
×
3312
        }
3313
      } else {
3314
        SValue value = {
781,145✔
3315
            .type = infos[iInfo].type,
781,145✔
3316
        };
3317
        if (IS_VAR_DATA_TYPE(infos[iInfo].type)) {
781,145!
3318
          int32_t   length = infos[iInfo].bind->length[iRow];
6,629✔
3319
          uint8_t **data = &((uint8_t **)TARRAY_DATA(bufArray))[iInfo];
6,629✔
3320
          value.nData = length;
6,629✔
3321
          if (value.nData > pTSchema->columns[iInfo].bytes - VARSTR_HEADER_SIZE) {
6,629!
3322
            code = TSDB_CODE_INVALID_PARA;
×
3323
            goto _exit;
×
3324
          }
3325
          value.pData = *data;
6,629✔
3326
          *data += length;
6,629✔
3327
          // value.pData = (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bind->buffer_length * iRow;
3328
        } else {
3329
          uint8_t *val = (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bytes * iRow;
774,516✔
3330
          if (TSDB_DATA_TYPE_BOOL == value.type && *val > 1) {
774,516✔
3331
            *val = 1;
1✔
3332
          }
3333
          valueSetDatum(&value, infos[iInfo].type, val, infos[iInfo].bytes);
774,516✔
3334
        }
3335
        colVal = COL_VAL_VALUE(infos[iInfo].columnId, value);
855,558✔
3336
      }
3337
      if (taosArrayPush(colValArray, &colVal) == NULL) {
807,256!
3338
        code = terrno;
×
3339
        goto _exit;
×
3340
      }
3341
    }
3342

3343
    SRow *row;
3344
    if ((code = tRowBuild(colValArray, pTSchema, &row))) {
271,644!
3345
      goto _exit;
×
3346
    }
3347

3348
    if ((taosArrayPush(rowArray, &row)) == NULL) {
230,796!
3349
      code = terrno;
×
3350
      goto _exit;
×
3351
    }
3352

3353
    if (pOrdered && pDupTs) {
230,796!
3354
      tRowGetKey(row, &rowKey);
462,674✔
3355
      if (iRow == 0) {
231,002✔
3356
        *pOrdered = true;
5,720✔
3357
        *pDupTs = false;
5,720✔
3358
      } else {
3359
        if (*pOrdered) {
225,282✔
3360
          int32_t res = tRowKeyCompare(&rowKey, &lastRowKey);
225,183✔
3361
          *pOrdered = (res >= 0);
225,183✔
3362
          if (!*pDupTs) {
225,183!
3363
            *pDupTs = (res == 0);
225,279✔
3364
          }
3365
        }
3366
        lastRowKey = rowKey;
225,282✔
3367
      }
3368
    }
3369
  }
UNCOV
3370
_exit:
×
UNCOV
3371
  taosArrayDestroy(colValArray);
×
3372
  taosArrayDestroy(bufArray);
5,737✔
3373
  return code;
5,750✔
3374
}
3375

3376
static int32_t tColDataCopyRowCell(SColData *pFromColData, int32_t iFromRow, SColData *pToColData, int32_t iToRow) {
346✔
3377
  int32_t code = TSDB_CODE_SUCCESS;
346✔
3378

3379
  if (IS_VAR_DATA_TYPE(pToColData->type)) {
346!
3380
    int32_t nData = (iFromRow < pFromColData->nVal - 1)
146✔
3381
                        ? pFromColData->aOffset[iFromRow + 1] - pFromColData->aOffset[iFromRow]
51✔
3382
                        : pFromColData->nData - pFromColData->aOffset[iFromRow];
73✔
3383
    if (iToRow == 0) {
73✔
3384
      pToColData->aOffset[iToRow] = 0;
14✔
3385
    }
3386

3387
    if (iToRow < pToColData->nVal - 1) {
73✔
3388
      pToColData->aOffset[iToRow + 1] = pToColData->aOffset[iToRow] + nData;
64✔
3389
    }
3390

3391
    (void)memcpy(pToColData->pData + pToColData->aOffset[iToRow], pFromColData->pData + pFromColData->aOffset[iFromRow],
73✔
3392
                 nData);
3393
  } else {
3394
    (void)memcpy(&pToColData->pData[TYPE_BYTES[pToColData->type] * iToRow],
273✔
3395
                 &pFromColData->pData[TYPE_BYTES[pToColData->type] * iFromRow], TYPE_BYTES[pToColData->type]);
273✔
3396
  }
3397
  return code;
346✔
3398
}
3399

3400
static int32_t tColDataCopyRowSingleCol(SColData *pFromColData, int32_t iFromRow, SColData *pToColData,
354✔
3401
                                        int32_t iToRow) {
3402
  int32_t code = TSDB_CODE_SUCCESS;
354✔
3403

3404
  switch (pFromColData->flag) {
354!
3405
    case HAS_NONE:
8✔
3406
    case HAS_NULL:
3407
      break;
8✔
3408
    case (HAS_NULL | HAS_NONE): {
×
3409
      SET_BIT1(pToColData->pBitMap, iToRow, GET_BIT1(pFromColData->pBitMap, iFromRow));
×
3410
    } break;
×
3411
    case HAS_VALUE: {
298✔
3412
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
298!
3413
    } break;
298✔
3414
    case (HAS_VALUE | HAS_NONE):
48✔
3415
    case (HAS_VALUE | HAS_NULL): {
3416
      SET_BIT1(pToColData->pBitMap, iToRow, GET_BIT1(pFromColData->pBitMap, iFromRow));
48✔
3417
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
48!
3418
    } break;
48✔
3419
    case (HAS_VALUE | HAS_NULL | HAS_NONE): {
×
3420
      SET_BIT2(pToColData->pBitMap, iToRow, GET_BIT2(pFromColData->pBitMap, iFromRow));
×
3421
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
×
3422
    } break;
×
3423
    default:
×
3424
      return -1;
×
3425
  }
3426

3427
  return code;
354✔
3428
}
3429

3430
static int32_t tColDataCopyRow(SColData *aFromColData, int32_t iFromRow, SColData *aToColData, int32_t iToRow,
75✔
3431
                               int32_t nColData) {
3432
  int32_t code = TSDB_CODE_SUCCESS;
75✔
3433

3434
  for (int32_t i = 0; i < nColData; i++) {
429✔
3435
    code = tColDataCopyRowSingleCol(&aFromColData[i], iFromRow, &aToColData[i], iToRow);
354✔
3436
    if (code != TSDB_CODE_SUCCESS) {
354!
3437
      return code;
×
3438
    }
3439
  }
3440

3441
  return code;
75✔
3442
}
3443

3444
static int32_t tColDataCopyRowAppend(SColData *aFromColData, int32_t iFromRow, SColData *aToColData, int32_t nColData) {
75✔
3445
  int32_t code = TSDB_CODE_SUCCESS;
75✔
3446

3447
  for (int32_t i = 0; i < nColData; i++) {
429✔
3448
    SColVal cv = {0};
354✔
3449
    code = tColDataGetValue(&aFromColData[i], iFromRow, &cv);
354✔
3450
    if (code != TSDB_CODE_SUCCESS) {
354!
3451
      return code;
×
3452
    }
3453
    code = tColDataAppendValue(&aToColData[i], &cv);
354✔
3454
    if (code != TSDB_CODE_SUCCESS) {
354!
3455
      return code;
×
3456
    }
3457
  }
3458

3459
  return code;
75✔
3460
}
3461

3462
void tColDataArrGetRowKey(SColData *aColData, int32_t nColData, int32_t iRow, SRowKey *key) {
230,813,753✔
3463
  SColVal cv;
3464

3465
  key->ts = ((TSKEY *)aColData[0].pData)[iRow];
230,813,753✔
3466
  key->numOfPKs = 0;
230,813,753✔
3467

3468
  for (int i = 1; i < nColData; i++) {
230,830,849!
3469
    if (aColData[i].cflag & COL_IS_KEY) {
231,427,380✔
3470
      tColDataGetValue4(&aColData[i], iRow, &cv);
13,701!
3471
      key->pks[key->numOfPKs++] = cv.value;
17,096✔
3472
    } else {
3473
      break;
231,413,679✔
3474
    }
3475
  }
3476
}
230,817,148✔
3477

3478
static int32_t tColDataMergeSortMerge(SColData *aColData, int32_t start, int32_t mid, int32_t end, int32_t nColData) {
23✔
3479
  SColData *aDstColData = NULL;
23✔
3480
  int32_t   i = start, j = mid + 1, k = 0;
23✔
3481
  SRowKey   keyi, keyj;
3482

3483
  if (end > start) {
23!
3484
    aDstColData = taosMemoryCalloc(1, sizeof(SColData) * nColData);
23!
3485
    if (aDstColData == NULL) {
23!
3486
      return terrno;
×
3487
    }
3488
    for (int c = 0; c < nColData; ++c) {
129✔
3489
      tColDataInit(&aDstColData[c], aColData[c].cid, aColData[c].type, aColData[c].cflag);
106✔
3490
    }
3491
  }
3492

3493
  tColDataArrGetRowKey(aColData, nColData, i, &keyi);
23✔
3494
  tColDataArrGetRowKey(aColData, nColData, j, &keyj);
23✔
3495
  while (i <= mid && j <= end) {
63✔
3496
    if (tRowKeyCompare(&keyi, &keyj) <= 0) {
40✔
3497
      TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, i++, aDstColData, nColData));
30!
3498
      tColDataArrGetRowKey(aColData, nColData, i, &keyi);
30✔
3499
    } else {
3500
      TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, j++, aDstColData, nColData));
10!
3501
      tColDataArrGetRowKey(aColData, nColData, j, &keyj);
10✔
3502
    }
3503
  }
3504

3505
  while (i <= mid) {
35✔
3506
    TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, i++, aDstColData, nColData));
12!
3507
  }
3508

3509
  while (j <= end) {
46✔
3510
    TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, j++, aDstColData, nColData));
23!
3511
  }
3512

3513
  for (i = start, k = 0; i <= end; ++i, ++k) {
98✔
3514
    TAOS_CHECK_RETURN(tColDataCopyRow(aDstColData, k, aColData, i, nColData));
75!
3515
  }
3516

3517
  if (aDstColData) {
23!
3518
    for (int32_t i = 0; i < nColData; i++) {
129✔
3519
      tColDataDestroy(&aDstColData[i]);
106✔
3520
    }
3521
    taosMemoryFree(aDstColData);
23!
3522
  }
3523

3524
  return TSDB_CODE_SUCCESS;
23✔
3525
}
3526

3527
static int32_t tColDataMergeSort(SColData *aColData, int32_t start, int32_t end, int32_t nColData) {
52✔
3528
  int32_t ret = TSDB_CODE_SUCCESS;
52✔
3529
  int32_t mid;
3530

3531
  if (start >= end) {
52✔
3532
    return TSDB_CODE_SUCCESS;
29✔
3533
  }
3534

3535
  mid = (start + end) / 2;
23✔
3536

3537
  ret = tColDataMergeSort(aColData, start, mid, nColData);
23✔
3538
  if (ret != TSDB_CODE_SUCCESS) {
23!
3539
    return ret;
×
3540
  }
3541

3542
  ret = tColDataMergeSort(aColData, mid + 1, end, nColData);
23✔
3543
  if (ret != TSDB_CODE_SUCCESS) {
23!
3544
    return ret;
×
3545
  }
3546

3547
  return tColDataMergeSortMerge(aColData, start, mid, end, nColData);
23✔
3548
}
3549

3550
static int32_t tColDataSort(SColData *aColData, int32_t nColData) {
6✔
3551
  int32_t nVal = aColData[0].nVal;
6✔
3552

3553
  if (nVal < 2) return TSDB_CODE_SUCCESS;
6!
3554

3555
  return tColDataMergeSort(aColData, 0, nVal - 1, nColData);
6✔
3556
}
3557

3558
static int32_t tColDataMerge(SArray **colArr) {
3✔
3559
  int32_t code = 0;
3✔
3560
  SArray *src = *colArr;
3✔
3561
  SArray *dst = NULL;
3✔
3562

3563
  dst = taosArrayInit(taosArrayGetSize(src), sizeof(SColData));
3✔
3564
  if (dst == NULL) {
3!
3565
    return terrno;
×
3566
  }
3567

3568
  for (int32_t i = 0; i < taosArrayGetSize(src); i++) {
12✔
3569
    SColData *srcCol = taosArrayGet(src, i);
9✔
3570

3571
    SColData *dstCol = taosArrayReserve(dst, 1);
9✔
3572
    if (dstCol == NULL) {
9!
3573
      code = terrno;
×
3574
      goto _exit;
×
3575
    }
3576
    tColDataInit(dstCol, srcCol->cid, srcCol->type, srcCol->cflag);
9✔
3577
  }
3578

3579
  int32_t numRows = ((SColData *)TARRAY_DATA(src))->nVal;
3✔
3580
  SRowKey lastKey;
3581
  for (int32_t i = 0; i < numRows; i++) {
9✔
3582
    SRowKey key;
3583
    tColDataArrGetRowKey((SColData *)TARRAY_DATA(src), taosArrayGetSize(src), i, &key);
6✔
3584

3585
    if (i == 0 || tRowKeyCompare(&key, &lastKey) != 0) {  // append new row
9!
3586
      for (int32_t j = 0; j < taosArrayGetSize(src); j++) {
12✔
3587
        SColData *srcCol = taosArrayGet(src, j);
9✔
3588
        SColData *dstCol = taosArrayGet(dst, j);
9✔
3589

3590
        SColVal cv;
3591
        code = tColDataGetValue(srcCol, i, &cv);
9✔
3592
        if (code != TSDB_CODE_SUCCESS) {
9!
3593
          goto _exit;
×
3594
        }
3595
        code = tColDataAppendValue(dstCol, &cv);
9✔
3596
        if (code) {
9!
3597
          goto _exit;
×
3598
        }
3599
      }
3600
      lastKey = key;
3✔
3601
    } else {  // update existing row
3602
      for (int32_t j = 0; j < taosArrayGetSize(src); j++) {
12✔
3603
        SColData *srcCol = taosArrayGet(src, j);
9✔
3604
        SColData *dstCol = taosArrayGet(dst, j);
9✔
3605

3606
        SColVal cv;
3607
        code = tColDataGetValue(srcCol, i, &cv);
9✔
3608
        if (code != TSDB_CODE_SUCCESS) {
9!
3609
          goto _exit;
×
3610
        }
3611
        code = tColDataUpdateValue(dstCol, &cv, true);
9✔
3612
        if (code) {
9!
3613
          goto _exit;
×
3614
        }
3615
      }
3616
    }
3617
  }
3618

3619
_exit:
3✔
3620
  if (code) {
3!
3621
    taosArrayDestroyEx(dst, tColDataDestroy);
×
3622
  } else {
3623
    taosArrayDestroyEx(src, tColDataDestroy);
3✔
3624
    *colArr = dst;
3✔
3625
  }
3626
  return code;
3✔
3627
}
3628

3629
int32_t tColDataSortMerge(SArray **arr) {
40,786✔
3630
  SArray   *colDataArr = *arr;
40,786✔
3631
  int32_t   nColData = TARRAY_SIZE(colDataArr);
40,786✔
3632
  SColData *aColData = (SColData *)TARRAY_DATA(colDataArr);
40,786✔
3633

3634
  if (!(aColData[0].type == TSDB_DATA_TYPE_TIMESTAMP)) {
40,786!
3635
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;
×
3636
  }
3637
  if (!(aColData[0].cid == PRIMARYKEY_TIMESTAMP_COL_ID)) {
40,786!
3638
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;
×
3639
  }
3640
  if (!(aColData[0].flag == HAS_VALUE)) {
40,786!
3641
    return TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3642
  }
3643

3644
  if (aColData[0].nVal <= 1) goto _exit;
40,786✔
3645

3646
  int8_t doSort = 0;
40,581✔
3647
  int8_t doMerge = 0;
40,581✔
3648
  // scan -------
3649
  SRowKey lastKey;
3650
  tColDataArrGetRowKey(aColData, nColData, 0, &lastKey);
40,581✔
3651
  for (int32_t iVal = 1; iVal < aColData[0].nVal; ++iVal) {
107,117,824!
3652
    SRowKey key;
3653
    tColDataArrGetRowKey(aColData, nColData, iVal, &key);
107,880,586✔
3654

3655
    int32_t c = tRowKeyCompare(&lastKey, &key);
107,077,241✔
3656
    if (c < 0) {
107,077,241✔
3657
      lastKey = key;
107,037,040✔
3658
      continue;
107,037,040✔
3659
    } else if (c > 0) {
40,201✔
3660
      doSort = 1;
6✔
3661
      break;
6✔
3662
    } else {
3663
      doMerge = 1;
40,195✔
3664
    }
3665
  }
3666

3667
  // sort -------
UNCOV
3668
  if (doSort) {
×
3669
    TAOS_CHECK_RETURN(tColDataSort(aColData, nColData));
6!
3670
  }
3671

3672
  if (doMerge != 1) {
1,431,827✔
3673
    tColDataArrGetRowKey(aColData, nColData, 0, &lastKey);
40,607✔
3674
    for (int32_t iVal = 1; iVal < aColData[0].nVal; ++iVal) {
107,043,734!
3675
      SRowKey key;
3676
      tColDataArrGetRowKey(aColData, nColData, iVal, &key);
108,394,336✔
3677

3678
      int32_t c = tRowKeyCompare(&lastKey, &key);
107,003,127✔
3679
      if (c == 0) {
107,003,127!
3680
        doMerge = 1;
×
3681
        break;
×
3682
      }
3683
      lastKey = key;
107,003,127✔
3684
    }
3685
  }
3686

3687
  // merge -------
3688
  if (doMerge) {
40,618✔
3689
    int32_t code = tColDataMerge(arr);
3✔
3690
    if (code) return code;
3!
3691
  }
3692

3693
_exit:
40,618✔
3694
  return 0;
40,823✔
3695
}
3696

3697
static int32_t tEncodeColDataVersion0(SEncoder *pEncoder, SColData *pColData) {
249,737✔
3698
  int32_t code = 0;
249,737✔
3699

3700
  if ((code = tEncodeI16v(pEncoder, pColData->cid))) return code;
499,474!
3701
  if ((code = tEncodeI8(pEncoder, pColData->type))) return code;
499,474!
3702
  if ((code = tEncodeI32v(pEncoder, pColData->nVal))) return code;
499,474!
3703
  if ((code = tEncodeI8(pEncoder, pColData->flag))) return code;
499,474!
3704

3705
  // bitmap
3706
  switch (pColData->flag) {
249,737!
3707
    case (HAS_NULL | HAS_NONE):
2,197✔
3708
    case (HAS_VALUE | HAS_NONE):
3709
    case (HAS_VALUE | HAS_NULL):
3710
      code = tEncodeFixed(pEncoder, pColData->pBitMap, BIT1_SIZE(pColData->nVal));
2,197✔
3711
      if (code) return code;
2,197!
3712
      break;
2,197✔
3713
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
3714
      code = tEncodeFixed(pEncoder, pColData->pBitMap, BIT2_SIZE(pColData->nVal));
×
3715
      if (code) return code;
×
3716
      break;
×
3717
    default:
247,540✔
3718
      break;
247,540✔
3719
  }
3720

3721
  // value
3722
  if (pColData->flag & HAS_VALUE) {
249,737✔
3723
    if (IS_VAR_DATA_TYPE(pColData->type)) {
249,392!
3724
      code = tEncodeFixed(pEncoder, pColData->aOffset, pColData->nVal << 2);
1,522✔
3725
      if (code) return code;
1,522!
3726

3727
      code = tEncodeI32v(pEncoder, pColData->nData);
1,522✔
3728
      if (code) return code;
1,522!
3729

3730
      code = tEncodeFixed(pEncoder, pColData->pData, pColData->nData);
1,522✔
3731
      if (code) return code;
1,522!
3732
    } else {
3733
      code = tEncodeFixed(pEncoder, pColData->pData, pColData->nData);
247,870✔
3734
      if (code) return code;
247,870!
3735
    }
3736
  }
3737

3738
  return code;
249,737✔
3739
}
3740

3741
static int32_t tDecodeColDataVersion0(SDecoder *pDecoder, SColData *pColData) {
74,343✔
3742
  int32_t code = 0;
74,343✔
3743

3744
  if ((code = tDecodeI16v(pDecoder, &pColData->cid))) return code;
148,686!
3745
  if ((code = tDecodeI8(pDecoder, &pColData->type))) return code;
148,675!
3746
  if ((code = tDecodeI32v(pDecoder, &pColData->nVal))) return code;
148,657!
3747
  if ((code = tDecodeI8(pDecoder, &pColData->flag))) return code;
148,649!
3748

3749
  if (pColData->type <= 0 || pColData->type >= TSDB_DATA_TYPE_MAX || pColData->flag <= 0 || pColData->flag >= 8) {
74,324!
3750
    return TSDB_CODE_INVALID_PARA;
×
3751
  }
3752

3753
  // bitmap
3754
  switch (pColData->flag) {
74,327!
3755
    case (HAS_NULL | HAS_NONE):
236✔
3756
    case (HAS_VALUE | HAS_NONE):
3757
    case (HAS_VALUE | HAS_NULL):
3758
      code = tDecodeBinaryWithSize(pDecoder, BIT1_SIZE(pColData->nVal), &pColData->pBitMap);
236!
3759
      if (code) return code;
236!
3760
      break;
236✔
3761
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
3762
      code = tDecodeBinaryWithSize(pDecoder, BIT2_SIZE(pColData->nVal), &pColData->pBitMap);
×
3763
      if (code) return code;
×
3764
      break;
×
3765
    default:
74,091✔
3766
      break;
74,091✔
3767
  }
3768

3769
  // value
3770
  if (pColData->flag & HAS_VALUE) {
74,327✔
3771
    if (IS_VAR_DATA_TYPE(pColData->type)) {
74,133!
3772
      code = tDecodeBinaryWithSize(pDecoder, pColData->nVal << 2, (uint8_t **)&pColData->aOffset);
687!
3773
      if (code) return code;
687!
3774

3775
      code = tDecodeI32v(pDecoder, &pColData->nData);
687✔
3776
      if (code) return code;
688!
3777

3778
      code = tDecodeBinaryWithSize(pDecoder, pColData->nData, &pColData->pData);
688!
3779
      if (code) return code;
688!
3780
    } else {
3781
      pColData->nData = TYPE_BYTES[pColData->type] * pColData->nVal;
73,446✔
3782
      code = tDecodeBinaryWithSize(pDecoder, pColData->nData, &pColData->pData);
73,446!
3783
      if (code) return code;
73,446!
3784
    }
3785
  }
3786
  pColData->cflag = 0;
74,328✔
3787

3788
  return code;
74,328✔
3789
}
3790

3791
static int32_t tEncodeColDataVersion1(SEncoder *pEncoder, SColData *pColData) {
249,748✔
3792
  int32_t code = tEncodeColDataVersion0(pEncoder, pColData);
249,748✔
3793
  if (code) return code;
249,791!
3794
  return tEncodeI8(pEncoder, pColData->cflag);
499,582✔
3795
}
3796

3797
static int32_t tDecodeColDataVersion1(SDecoder *pDecoder, SColData *pColData) {
74,343✔
3798
  int32_t code = tDecodeColDataVersion0(pDecoder, pColData);
74,343✔
3799
  if (code) return code;
74,324!
3800

3801
  code = tDecodeI8(pDecoder, &pColData->cflag);
74,324✔
3802
  return code;
74,322✔
3803
}
3804

3805
int32_t tEncodeColData(uint8_t version, SEncoder *pEncoder, SColData *pColData) {
249,751✔
3806
  if (version == 0) {
249,751!
3807
    return tEncodeColDataVersion0(pEncoder, pColData);
×
3808
  } else if (version == 1) {
249,751!
3809
    return tEncodeColDataVersion1(pEncoder, pColData);
249,775✔
3810
  } else {
3811
    return TSDB_CODE_INVALID_PARA;
×
3812
  }
3813
}
3814

3815
int32_t tDecodeColData(uint8_t version, SDecoder *pDecoder, SColData *pColData) {
74,342✔
3816
  if (version == 0) {
74,342!
3817
    return tDecodeColDataVersion0(pDecoder, pColData);
×
3818
  } else if (version == 1) {
74,342!
3819
    return tDecodeColDataVersion1(pDecoder, pColData);
74,342✔
3820
  } else {
3821
    return TSDB_CODE_INVALID_PARA;
×
3822
  }
3823
}
3824

3825
int32_t tEncodeRow(SEncoder *pEncoder, SRow *pRow) { return tEncodeFixed(pEncoder, pRow, pRow->len); }
523,715,716✔
3826

3827
int32_t tDecodeRow(SDecoder *pDecoder, SRow **ppRow) {
154,274,838✔
3828
  if (ppRow == NULL) {
154,274,838!
3829
    return TSDB_CODE_INVALID_PARA;
×
3830
  }
3831

3832
  if (pDecoder->pos + sizeof(SRow) > pDecoder->size) {
154,274,838!
3833
    return TSDB_CODE_OUT_OF_RANGE;
×
3834
  }
3835

3836
  SRow *pRow = (SRow *)(pDecoder->data + pDecoder->pos);
154,274,838✔
3837
  return tDecodeBinaryWithSize(pDecoder, pRow->len, (uint8_t **)ppRow);
308,549,676!
3838
}
3839

3840
#define CALC_SUM_MAX_MIN(SUM, MAX, MIN, VAL) \
3841
  do {                                       \
3842
    (SUM) += (VAL);                          \
3843
    if ((MAX) < (VAL)) (MAX) = (VAL);        \
3844
    if ((MIN) > (VAL)) (MIN) = (VAL);        \
3845
  } while (0)
3846

3847
static FORCE_INLINE void tColDataCalcSMABool(SColData *pColData, SColumnDataAgg* pAggs) {
3,786✔
3848
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
3,786✔
3849
  int16_t*numOfNull = &pAggs->numOfNull;
3,786✔
3850
  *sum = 0;
3,786✔
3851
  *max = 0;
3,786✔
3852
  *min = 1;
3,786✔
3853
  *numOfNull = 0;
3,786✔
3854

3855
  int8_t val;
3856
  if (HAS_VALUE == pColData->flag) {
3,786!
3857
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
3,549,266✔
3858
      val = ((int8_t *)pColData->pData)[iVal] ? 1 : 0;
3,545,480✔
3859
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
3,545,480✔
3860
    }
3861
  } else {
3862
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3863
      switch (tColDataGetBitValue(pColData, iVal)) {
×
3864
        case 0:
×
3865
        case 1:
3866
          (*numOfNull)++;
×
3867
          break;
×
3868
        case 2:
×
3869
          val = ((int8_t *)pColData->pData)[iVal] ? 1 : 0;
×
3870
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3871
          break;
×
3872
        default:
×
3873
          break;
×
3874
      }
3875
    }
3876
  }
3877
}
3,786✔
3878

3879
static FORCE_INLINE void tColDataCalcSMATinyInt(SColData *pColData, SColumnDataAgg* pAggs) {
3,958✔
3880
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
3,958✔
3881
  int16_t *numOfNull = &pAggs->numOfNull;
3,958✔
3882
  *sum = 0;
3,958✔
3883
  *max = INT8_MIN;
3,958✔
3884
  *min = INT8_MAX;
3,958✔
3885
  *numOfNull = 0;
3,958✔
3886

3887
  int8_t val;
3888
  if (HAS_VALUE == pColData->flag) {
3,958!
3889
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
3,840,674✔
3890
      val = ((int8_t *)pColData->pData)[iVal];
3,836,715✔
3891
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
3,836,715✔
3892
    }
3893
  } else {
3894
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3895
      switch (tColDataGetBitValue(pColData, iVal)) {
×
3896
        case 0:
×
3897
        case 1:
3898
          (*numOfNull)++;
×
3899
          break;
×
3900
        case 2:
×
3901
          val = ((int8_t *)pColData->pData)[iVal];
×
3902
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3903
          break;
×
3904
        default:
×
3905
          break;
×
3906
      }
3907
    }
3908
  }
3909
}
3,958✔
3910

3911
static FORCE_INLINE void tColDataCalcSMATinySmallInt(SColData *pColData, SColumnDataAgg* pAggs) {
3,822✔
3912
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
3,822✔
3913
  int16_t *numOfNull = &pAggs->numOfNull;
3,822✔
3914
  *sum = 0;
3,822✔
3915
  *max = INT16_MIN;
3,822✔
3916
  *min = INT16_MAX;
3,822✔
3917
  *numOfNull = 0;
3,822✔
3918

3919
  int16_t val;
3920
  if (HAS_VALUE == pColData->flag) {
3,822!
3921
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
3,668,393✔
3922
      val = ((int16_t *)pColData->pData)[iVal];
3,664,571✔
3923
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
3,664,571✔
3924
    }
3925
  } else {
3926
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3927
      switch (tColDataGetBitValue(pColData, iVal)) {
×
3928
        case 0:
×
3929
        case 1:
3930
          (*numOfNull)++;
×
3931
          break;
×
3932
        case 2:
×
3933
          val = ((int16_t *)pColData->pData)[iVal];
×
3934
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3935
          break;
×
3936
        default:
×
3937
          break;
×
3938
      }
3939
    }
3940
  }
3941
}
3,822✔
3942

3943
static FORCE_INLINE void tColDataCalcSMAInt(SColData *pColData, SColumnDataAgg* pAggs) {
31,455✔
3944
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
31,455✔
3945
  int16_t *numOfNull = &pAggs->numOfNull;
31,455✔
3946
  *sum = 0;
31,455✔
3947
  *max = INT32_MIN;
31,455✔
3948
  *min = INT32_MAX;
31,455✔
3949
  *numOfNull = 0;
31,455✔
3950

3951
  int32_t val;
3952
  if (HAS_VALUE == pColData->flag) {
31,455!
3953
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
70,545,712✔
3954
      val = ((int32_t *)pColData->pData)[iVal];
70,514,257✔
3955
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
70,514,257✔
3956
    }
3957
  } else {
3958
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
105!
3959
      switch (tColDataGetBitValue(pColData, iVal)) {
105!
3960
        case 0:
95✔
3961
        case 1:
3962
          (*numOfNull)++;
95✔
3963
          break;
95✔
3964
        case 2:
10✔
3965
          val = ((int32_t *)pColData->pData)[iVal];
10✔
3966
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
10!
3967
          break;
10✔
3968
        default:
×
3969
          break;
×
3970
      }
3971
    }
3972
  }
3973
}
31,455✔
3974

3975
static FORCE_INLINE void tColDataCalcSMABigInt(SColData *pColData, SColumnDataAgg* pAggs) {
14,266✔
3976
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
14,266✔
3977
  int16_t *numOfNull = &pAggs->numOfNull;
14,266✔
3978
  *sum = 0;
14,266✔
3979
  *max = INT64_MIN;
14,266✔
3980
  *min = INT64_MAX;
14,266✔
3981
  *numOfNull = 0;
14,266✔
3982

3983
  int64_t val;
3984
  if (HAS_VALUE == pColData->flag) {
14,266✔
3985
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
42,978,668✔
3986
      val = ((int64_t *)pColData->pData)[iVal];
42,964,864✔
3987
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
42,964,864✔
3988
    }
3989
  } else {
3990
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
90,144✔
3991
      switch (tColDataGetBitValue(pColData, iVal)) {
89,682!
3992
        case 0:
75,200✔
3993
        case 1:
3994
          (*numOfNull)++;
75,200✔
3995
          break;
75,200✔
3996
        case 2:
14,482✔
3997
          val = ((int64_t *)pColData->pData)[iVal];
14,482✔
3998
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
14,482✔
3999
          break;
14,482✔
4000
        default:
×
4001
          break;
×
4002
      }
4003
    }
4004
  }
4005
}
14,266✔
4006

4007
static FORCE_INLINE void tColDataCalcSMAFloat(SColData *pColData, SColumnDataAgg* pAggs) {
7,183✔
4008
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
7,183✔
4009
  int16_t *numOfNull = &pAggs->numOfNull;
7,183✔
4010
  *(double *)sum = 0;
7,183✔
4011
  *(double *)max = -FLT_MAX;
7,183✔
4012
  *(double *)min = FLT_MAX;
7,183✔
4013
  *numOfNull = 0;
7,183✔
4014

4015
  float val;
4016
  if (HAS_VALUE == pColData->flag) {
7,183✔
4017
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
15,936,607✔
4018
      val = ((float *)pColData->pData)[iVal];
15,929,425✔
4019
      CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
15,929,425✔
4020
    }
4021
  } else {
4022
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
106✔
4023
      switch (tColDataGetBitValue(pColData, iVal)) {
105!
4024
        case 0:
95✔
4025
        case 1:
4026
          (*numOfNull)++;
95✔
4027
          break;
95✔
4028
        case 2:
10✔
4029
          val = ((float *)pColData->pData)[iVal];
10✔
4030
          CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
10!
4031
          break;
10✔
4032
        default:
×
4033
          break;
×
4034
      }
4035
    }
4036
  }
4037
}
7,183✔
4038

4039
static FORCE_INLINE void tColDataCalcSMADouble(SColData *pColData, SColumnDataAgg* pAggs) {
19,055✔
4040
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
19,055✔
4041
  int16_t *numOfNull = &pAggs->numOfNull;
19,055✔
4042
  *(double *)sum = 0;
19,055✔
4043
  *(double *)max = -DBL_MAX;
19,055✔
4044
  *(double *)min = DBL_MAX;
19,055✔
4045
  *numOfNull = 0;
19,055✔
4046

4047
  double val;
4048
  if (HAS_VALUE == pColData->flag) {
19,055!
4049
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
33,687,662✔
4050
      val = ((double *)pColData->pData)[iVal];
33,668,607✔
4051
      CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
33,668,607✔
4052
    }
4053
  } else {
4054
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4055
      switch (tColDataGetBitValue(pColData, iVal)) {
×
4056
        case 0:
×
4057
        case 1:
4058
          (*numOfNull)++;
×
4059
          break;
×
4060
        case 2:
×
4061
          val = ((double *)pColData->pData)[iVal];
×
4062
          CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
×
4063
          break;
×
4064
        default:
×
4065
          break;
×
4066
      }
4067
    }
4068
  }
4069
}
19,055✔
4070

4071
static FORCE_INLINE void tColDataCalcSMAUTinyInt(SColData *pColData, SColumnDataAgg* pAggs) {
907✔
4072
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
907✔
4073
  int16_t *numOfNull = &pAggs->numOfNull;
907✔
4074
  *(uint64_t *)sum = 0;
907✔
4075
  *(uint64_t *)max = 0;
907✔
4076
  *(uint64_t *)min = UINT8_MAX;
907✔
4077
  *numOfNull = 0;
907✔
4078

4079
  uint8_t val;
4080
  if (HAS_VALUE == pColData->flag) {
907!
4081
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
2,785,009✔
4082
      val = ((uint8_t *)pColData->pData)[iVal];
2,784,102✔
4083
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
2,784,102✔
4084
    }
4085
  } else {
4086
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4087
      switch (tColDataGetBitValue(pColData, iVal)) {
×
4088
        case 0:
×
4089
        case 1:
4090
          (*numOfNull)++;
×
4091
          break;
×
4092
        case 2:
×
4093
          val = ((uint8_t *)pColData->pData)[iVal];
×
4094
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4095
          break;
×
4096
        default:
×
4097
          break;
×
4098
      }
4099
    }
4100
  }
4101
}
907✔
4102

4103
static FORCE_INLINE void tColDataCalcSMATinyUSmallInt(SColData *pColData, SColumnDataAgg* pAggs) {
907✔
4104
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
907✔
4105
  int16_t *numOfNull = &pAggs->numOfNull;
907✔
4106
  *(uint64_t *)sum = 0;
907✔
4107
  *(uint64_t *)max = 0;
907✔
4108
  *(uint64_t *)min = UINT16_MAX;
907✔
4109
  *numOfNull = 0;
907✔
4110

4111
  uint16_t val;
4112
  if (HAS_VALUE == pColData->flag) {
907!
4113
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
2,785,009✔
4114
      val = ((uint16_t *)pColData->pData)[iVal];
2,784,102✔
4115
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
2,784,102✔
4116
    }
4117
  } else {
4118
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4119
      switch (tColDataGetBitValue(pColData, iVal)) {
×
4120
        case 0:
×
4121
        case 1:
4122
          (*numOfNull)++;
×
4123
          break;
×
4124
        case 2:
×
4125
          val = ((uint16_t *)pColData->pData)[iVal];
×
4126
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4127
          break;
×
4128
        default:
×
4129
          break;
×
4130
      }
4131
    }
4132
  }
4133
}
907✔
4134

4135
static FORCE_INLINE void tColDataCalcSMAUInt(SColData *pColData, SColumnDataAgg* pAggs) {
907✔
4136
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
907✔
4137
  int16_t *numOfNull = &pAggs->numOfNull;
907✔
4138
  *(uint64_t *)sum = 0;
907✔
4139
  *(uint64_t *)max = 0;
907✔
4140
  *(uint64_t *)min = UINT32_MAX;
907✔
4141
  *numOfNull = 0;
907✔
4142

4143
  uint32_t val;
4144
  if (HAS_VALUE == pColData->flag) {
907!
4145
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
2,785,009✔
4146
      val = ((uint32_t *)pColData->pData)[iVal];
2,784,102✔
4147
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
2,784,102✔
4148
    }
4149
  } else {
4150
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4151
      switch (tColDataGetBitValue(pColData, iVal)) {
×
4152
        case 0:
×
4153
        case 1:
4154
          (*numOfNull)++;
×
4155
          break;
×
4156
        case 2:
×
4157
          val = ((uint32_t *)pColData->pData)[iVal];
×
4158
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4159
          break;
×
4160
        default:
×
4161
          break;
×
4162
      }
4163
    }
4164
  }
4165
}
907✔
4166

4167
static FORCE_INLINE void tColDataCalcSMAUBigInt(SColData *pColData, SColumnDataAgg* pAggs) {
907✔
4168
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
907✔
4169
  int16_t *numOfNull = &pAggs->numOfNull;
907✔
4170
  *(uint64_t *)sum = 0;
907✔
4171
  *(uint64_t *)max = 0;
907✔
4172
  *(uint64_t *)min = UINT64_MAX;
907✔
4173
  *numOfNull = 0;
907✔
4174

4175
  uint64_t val;
4176
  if (HAS_VALUE == pColData->flag) {
907!
4177
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
2,785,009✔
4178
      val = ((uint64_t *)pColData->pData)[iVal];
2,784,102✔
4179
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
2,784,102✔
4180
    }
4181
  } else {
4182
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4183
      switch (tColDataGetBitValue(pColData, iVal)) {
×
4184
        case 0:
×
4185
        case 1:
4186
          (*numOfNull)++;
×
4187
          break;
×
4188
        case 2:
×
4189
          val = ((uint64_t *)pColData->pData)[iVal];
×
4190
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4191
          break;
×
4192
        default:
×
4193
          break;
×
4194
      }
4195
    }
4196
  }
4197
}
907✔
4198

4199
static FORCE_INLINE void tColDataCalcSMAVarType(SColData *pColData, SColumnDataAgg* pAggs) {
14,550✔
4200
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
14,550✔
4201
  int16_t *numOfNull = &pAggs->numOfNull;
14,550✔
4202
  *(uint64_t *)sum = 0;
14,550✔
4203
  *(uint64_t *)max = 0;
14,550✔
4204
  *(uint64_t *)min = 0;
14,550✔
4205
  *numOfNull = 0;
14,550✔
4206

4207
  switch (pColData->flag) {
14,550!
4208
    case HAS_NONE:
×
4209
    case HAS_NULL:
4210
    case (HAS_NONE | HAS_NULL):
4211
      *numOfNull = pColData->nVal;
×
4212
      break;
×
4213
    case HAS_VALUE:
14,550✔
4214
      *numOfNull = 0;
14,550✔
4215
      break;
14,550✔
4216
    case (HAS_VALUE | HAS_NULL):
×
4217
    case (HAS_VALUE | HAS_NONE):
4218
      for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4219
        if (GET_BIT1(pColData->pBitMap, iVal) == 0) {
×
4220
          (*numOfNull)++;
×
4221
        }
4222
      }
4223
      break;
×
4224
    case (HAS_VALUE | HAS_NONE | HAS_NULL):
×
4225
      for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4226
        if (GET_BIT2(pColData->pBitMap, iVal) != 2) {
×
4227
          (*numOfNull)++;
×
4228
        }
4229
      }
4230
      break;
×
4231
    default:
×
4232
      break;
×
4233
  }
4234
}
14,550✔
4235

4236
#define CALC_DECIMAL_SUM_MAX_MIN(TYPE, pSumOp, pCompOp, pColData, pSum, pMax, pMin)           \
4237
  do {                                                                                        \
4238
    if (decimal128AddCheckOverflow((Decimal *)pSum, pVal, DECIMAL_WORD_NUM(TYPE))) *pOverflow = true; \
4239
    pSumOp->add(pSum, pVal, DECIMAL_WORD_NUM(TYPE));                                                  \
4240
    if (pCompOp->gt(pVal, pMax, DECIMAL_WORD_NUM(TYPE))) {                                            \
4241
      *(pMax) = *pVal;                                                                        \
4242
    }                                                                                         \
4243
    if (pCompOp->lt(pVal, pMin, DECIMAL_WORD_NUM(TYPE))) {                                            \
4244
      *(pMin) = *pVal;                                                                        \
4245
    }                                                                                         \
4246
  } while (0)
4247

4248
static FORCE_INLINE void tColDataCalcSMADecimal64Type(SColData* pColData, SColumnDataAgg* pAggs) {
163✔
4249
  Decimal128 *pSum = (Decimal128 *)pAggs->decimal128Sum;
163✔
4250
  Decimal64  *pMax = (Decimal64 *)pAggs->decimal128Max, *pMin = (Decimal64 *)pAggs->decimal128Min;
163✔
4251
  uint8_t *pOverflow = &pAggs->overflow;
163✔
4252
  *pSum = DECIMAL128_ZERO;
163✔
4253
  *pMax = DECIMAL64_MIN;
163✔
4254
  *pMin = DECIMAL64_MAX;
163✔
4255
  pAggs->numOfNull = 0;
163✔
4256
  pAggs->colId |= DECIMAL_AGG_FLAG;
163✔
4257

4258
  Decimal64   *pVal = NULL;
163✔
4259
  const SDecimalOps *pSumOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL);
163✔
4260
  const SDecimalOps *pCompOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL64);
163✔
4261
  if (HAS_VALUE == pColData->flag) {
163✔
4262
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
808✔
4263
      pVal = ((Decimal64*)pColData->pData) + iVal;
800✔
4264
      CALC_DECIMAL_SUM_MAX_MIN(Decimal64, pSumOps, pCompOps, pColData, pSum, pMax, pMin);
800!
4265
    }
4266
  } else {
4267
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
125,455✔
4268
      switch (tColDataGetBitValue(pColData, iVal)) {
125,300!
4269
        case 0:
2,034✔
4270
        case 1:
4271
          pAggs->numOfNull++;
2,034✔
4272
          break;
2,034✔
4273
        case 2:
123,266✔
4274
          pVal = ((Decimal64 *)pColData->pData) + iVal;
123,266✔
4275
          CALC_DECIMAL_SUM_MAX_MIN(Decimal64, pSumOps, pCompOps, pColData, pSum, pMax, pMin);
123,266!
4276
          break;
123,266✔
4277
        default:
×
4278
          break;
×
4279
      }
4280
    }
4281
  }
4282
}
163✔
4283

4284
static FORCE_INLINE void tColDataCalcSMADecimal128Type(SColData* pColData, SColumnDataAgg* pAggs) {
435✔
4285
  Decimal128 *pSum = (Decimal128 *)pAggs->decimal128Sum, *pMax = (Decimal128 *)pAggs->decimal128Max,
435✔
4286
             *pMin = (Decimal128 *)pAggs->decimal128Min;
435✔
4287
  uint8_t *pOverflow = &pAggs->overflow;
435✔
4288
  *pSum = DECIMAL128_ZERO;
435✔
4289
  *pMax = DECIMAL128_MIN;
435✔
4290
  *pMin = DECIMAL128_MAX;
435✔
4291
  pAggs->numOfNull = 0;
435✔
4292
  pAggs->colId |= DECIMAL_AGG_FLAG;
435✔
4293

4294
  Decimal128        *pVal = NULL;
435✔
4295
  const SDecimalOps *pOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL);
435✔
4296
  if (HAS_VALUE == pColData->flag) {
435✔
4297
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
1,010✔
4298
      pVal = ((Decimal128*)pColData->pData) + iVal;
1,000✔
4299
      CALC_DECIMAL_SUM_MAX_MIN(Decimal128, pOps, pOps, pColData, pSum, pMax, pMin);
1,000✔
4300
    }
4301
  } else {
4302
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
397,525✔
4303
      switch (tColDataGetBitValue(pColData, iVal)) {
397,100!
4304
        case 0:
6,459✔
4305
        case 1:
4306
          pAggs->numOfNull++;
6,459✔
4307
          break;
6,459✔
4308
        case 2:
390,641✔
4309
          pVal = ((Decimal128*)pColData->pData) + iVal;
390,641✔
4310
          CALC_DECIMAL_SUM_MAX_MIN(Decimal128, pOps, pOps, pColData, pSum, pMax, pMin);
390,641✔
4311
          break;
390,641✔
4312
        default:
×
4313
          break;
×
4314
      }
4315
    }
4316
  }
4317
}
435✔
4318

4319
void (*tColDataCalcSMA[])(SColData *pColData, SColumnDataAgg* pAggs) = {
4320
    NULL,
4321
    tColDataCalcSMABool,           // TSDB_DATA_TYPE_BOOL
4322
    tColDataCalcSMATinyInt,        // TSDB_DATA_TYPE_TINYINT
4323
    tColDataCalcSMATinySmallInt,   // TSDB_DATA_TYPE_SMALLINT
4324
    tColDataCalcSMAInt,            // TSDB_DATA_TYPE_INT
4325
    tColDataCalcSMABigInt,         // TSDB_DATA_TYPE_BIGINT
4326
    tColDataCalcSMAFloat,          // TSDB_DATA_TYPE_FLOAT
4327
    tColDataCalcSMADouble,         // TSDB_DATA_TYPE_DOUBLE
4328
    tColDataCalcSMAVarType,        // TSDB_DATA_TYPE_VARCHAR
4329
    tColDataCalcSMABigInt,         // TSDB_DATA_TYPE_TIMESTAMP
4330
    tColDataCalcSMAVarType,        // TSDB_DATA_TYPE_NCHAR
4331
    tColDataCalcSMAUTinyInt,       // TSDB_DATA_TYPE_UTINYINT
4332
    tColDataCalcSMATinyUSmallInt,  // TSDB_DATA_TYPE_USMALLINT
4333
    tColDataCalcSMAUInt,           // TSDB_DATA_TYPE_UINT
4334
    tColDataCalcSMAUBigInt,        // TSDB_DATA_TYPE_UBIGINT
4335
    tColDataCalcSMAVarType,        // TSDB_DATA_TYPE_JSON
4336
    tColDataCalcSMAVarType,        // TSDB_DATA_TYPE_VARBINARY
4337
    tColDataCalcSMADecimal128Type, // TSDB_DATA_TYPE_DECIMAL
4338
    tColDataCalcSMAVarType,        // TSDB_DATA_TYPE_BLOB
4339
    NULL,                          // TSDB_DATA_TYPE_MEDIUMBLOB
4340
    tColDataCalcSMAVarType,         // TSDB_DATA_TYPE_GEOMETRY
4341
    tColDataCalcSMADecimal64Type,  // TSDB_DATA_TYPE_DECIMAL64
4342
};
4343

4344
// SValueColumn ================================
4345
int32_t tValueColumnInit(SValueColumn *valCol) {
4,677,714✔
4346
  valCol->type = TSDB_DATA_TYPE_NULL;
4,677,714✔
4347
  valCol->numOfValues = 0;
4,677,714✔
4348
  tBufferInit(&valCol->data);
4,677,714✔
4349
  tBufferInit(&valCol->offsets);
4,677,714✔
4350
  return 0;
4,677,714✔
4351
}
4352

4353
void tValueColumnDestroy(SValueColumn *valCol) {
12,452,793✔
4354
  valCol->type = TSDB_DATA_TYPE_NULL;
12,452,793✔
4355
  valCol->numOfValues = 0;
12,452,793✔
4356
  tBufferDestroy(&valCol->data);
12,452,793✔
4357
  tBufferDestroy(&valCol->offsets);
12,452,793✔
4358
  return;
12,452,793✔
4359
}
4360

4361
void tValueColumnClear(SValueColumn *valCol) {
7,047,600✔
4362
  valCol->type = TSDB_DATA_TYPE_NULL;
7,047,600✔
4363
  valCol->numOfValues = 0;
7,047,600✔
4364
  tBufferClear(&valCol->data);
7,047,600✔
4365
  tBufferClear(&valCol->offsets);
7,047,600✔
4366
  return;
7,047,600✔
4367
}
4368

4369
int32_t tValueColumnAppend(SValueColumn *valCol, const SValue *value) {
144✔
4370
  int32_t code;
4371

4372
  if (valCol->numOfValues == 0) {
144✔
4373
    valCol->type = value->type;
88✔
4374
  }
4375

4376
  if (!(value->type == valCol->type)) {
144!
4377
    return TSDB_CODE_INVALID_PARA;
×
4378
  }
4379

4380
  if (IS_VAR_DATA_TYPE(value->type)) {
144!
4381
    if ((code = tBufferPutI32(&valCol->offsets, tBufferGetSize(&valCol->data)))) {
20!
4382
      return code;
×
4383
    }
4384
    if ((code = tBufferPut(&valCol->data, value->pData, value->nData))) {
20!
4385
      return code;
×
4386
    }
4387
  } else {
4388
    code = tBufferPut(&valCol->data, VALUE_GET_DATUM(value, value->type), tDataTypes[value->type].bytes);
134!
4389
    if (code) return code;
134!
4390
  }
4391
  valCol->numOfValues++;
144✔
4392

4393
  return 0;
144✔
4394
}
4395

4396
int32_t tValueColumnUpdate(SValueColumn *valCol, int32_t idx, const SValue *value) {
264✔
4397
  int32_t code;
4398

4399
  if (idx < 0 || idx >= valCol->numOfValues) {
264!
4400
    return TSDB_CODE_OUT_OF_RANGE;
×
4401
  }
4402

4403
  if (IS_VAR_DATA_TYPE(valCol->type)) {
264!
4404
    int32_t *offsets = (int32_t *)tBufferGetData(&valCol->offsets);
20✔
4405
    int32_t  nextOffset = (idx == valCol->numOfValues - 1) ? tBufferGetSize(&valCol->data) : offsets[idx + 1];
20!
4406
    int32_t  oldDataSize = nextOffset - offsets[idx];
20✔
4407
    int32_t  bytesAdded = value->nData - oldDataSize;
20✔
4408

4409
    if (bytesAdded != 0) {
20!
4410
      if ((code = tBufferEnsureCapacity(&valCol->data, tBufferGetSize(&valCol->data) + bytesAdded))) return code;
×
4411
      memmove(tBufferGetDataAt(&valCol->data, nextOffset + bytesAdded), tBufferGetDataAt(&valCol->data, nextOffset),
×
4412
              tBufferGetSize(&valCol->data) - nextOffset);
×
4413
      valCol->data.size += bytesAdded;
×
4414

4415
      for (int32_t i = idx + 1; i < valCol->numOfValues; i++) {
×
4416
        offsets[i] += bytesAdded;
×
4417
      }
4418
    }
4419
    return tBufferPutAt(&valCol->data, offsets[idx], value->pData, value->nData);
20✔
4420
  } else {
4421
    return tBufferPutAt(&valCol->data, idx * tDataTypes[valCol->type].bytes, VALUE_GET_DATUM(value, valCol->type),
244!
4422
                        tDataTypes[valCol->type].bytes);
244!
4423
  }
4424
  return 0;
4425
}
4426

4427
int32_t tValueColumnGet(SValueColumn *valCol, int32_t idx, SValue *value) {
728✔
4428
  if (idx < 0 || idx >= valCol->numOfValues) {
728!
4429
    return TSDB_CODE_OUT_OF_RANGE;
×
4430
  }
4431

4432
  value->type = valCol->type;
728✔
4433
  if (IS_VAR_DATA_TYPE(value->type)) {
728!
4434
    int32_t       offset, nextOffset;
4435
    SBufferReader reader = BUFFER_READER_INITIALIZER(idx * sizeof(offset), &valCol->offsets);
52✔
4436

4437
    TAOS_CHECK_RETURN(tBufferGetI32(&reader, &offset));
52!
4438
    if (idx == valCol->numOfValues - 1) {
52✔
4439
      nextOffset = tBufferGetSize(&valCol->data);
48✔
4440
    } else {
4441
      TAOS_CHECK_RETURN(tBufferGetI32(&reader, &nextOffset));
4!
4442
    }
4443
    value->nData = nextOffset - offset;
52✔
4444
    value->pData = (uint8_t *)tBufferGetDataAt(&valCol->data, offset);
52✔
4445
  } else {
4446
    SBufferReader reader = BUFFER_READER_INITIALIZER(idx * tDataTypes[value->type].bytes, &valCol->data);
676✔
4447
    TAOS_CHECK_RETURN(tBufferGet(&reader, tDataTypes[value->type].bytes, VALUE_GET_DATUM(value, value->type)));
1,352!
4448
  }
4449
  return 0;
728✔
4450
}
4451

4452
int32_t tValueColumnCompress(SValueColumn *valCol, SValueColumnCompressInfo *info, SBuffer *output, SBuffer *assist) {
87✔
4453
  int32_t code;
4454

4455
  if (!(valCol->numOfValues > 0)) {
87!
4456
    return TSDB_CODE_INVALID_PARA;
×
4457
  }
4458

4459
  (*info) = (SValueColumnCompressInfo){
87✔
4460
      .cmprAlg = info->cmprAlg,
87✔
4461
      .type = valCol->type,
87✔
4462
  };
4463

4464
  // offset
4465
  if (IS_VAR_DATA_TYPE(valCol->type)) {
87!
4466
    SCompressInfo cinfo = {
6✔
4467
        .cmprAlg = info->cmprAlg,
6✔
4468
        .dataType = TSDB_DATA_TYPE_INT,
4469
        .originalSize = valCol->offsets.size,
6✔
4470
    };
4471

4472
    code = tCompressDataToBuffer(valCol->offsets.data, &cinfo, output, assist);
6✔
4473
    if (code) return code;
6!
4474

4475
    info->offsetOriginalSize = cinfo.originalSize;
6✔
4476
    info->offsetCompressedSize = cinfo.compressedSize;
6✔
4477
  }
4478

4479
  // data
4480
  SCompressInfo cinfo = {
87✔
4481
      .cmprAlg = info->cmprAlg,
87✔
4482
      .dataType = valCol->type,
87✔
4483
      .originalSize = valCol->data.size,
87✔
4484
  };
4485

4486
  code = tCompressDataToBuffer(valCol->data.data, &cinfo, output, assist);
87✔
4487
  if (code) return code;
88!
4488

4489
  info->dataOriginalSize = cinfo.originalSize;
88✔
4490
  info->dataCompressedSize = cinfo.compressedSize;
88✔
4491

4492
  return 0;
88✔
4493
}
4494

4495
int32_t tValueColumnDecompress(void *input, const SValueColumnCompressInfo *info, SValueColumn *valCol,
×
4496
                               SBuffer *assist) {
4497
  int32_t code;
4498

4499
  tValueColumnClear(valCol);
×
4500
  valCol->type = info->type;
×
4501
  // offset
4502
  if (IS_VAR_DATA_TYPE(valCol->type)) {
×
4503
    valCol->numOfValues = info->offsetOriginalSize / tDataTypes[TSDB_DATA_TYPE_INT].bytes;
×
4504

4505
    SCompressInfo cinfo = {
×
4506
        .dataType = TSDB_DATA_TYPE_INT,
4507
        .cmprAlg = info->cmprAlg,
×
4508
        .originalSize = info->offsetOriginalSize,
×
4509
        .compressedSize = info->offsetCompressedSize,
×
4510
    };
4511

4512
    code = tDecompressDataToBuffer(input, &cinfo, &valCol->offsets, assist);
×
4513
    if (code) {
×
4514
      return code;
×
4515
    }
4516
  } else {
4517
    valCol->numOfValues = info->dataOriginalSize / tDataTypes[valCol->type].bytes;
×
4518
  }
4519

4520
  // data
4521
  SCompressInfo cinfo = {
×
4522
      .dataType = valCol->type,
×
4523
      .cmprAlg = info->cmprAlg,
×
4524
      .originalSize = info->dataOriginalSize,
×
4525
      .compressedSize = info->dataCompressedSize,
×
4526
  };
4527

4528
  code = tDecompressDataToBuffer((char *)input + info->offsetCompressedSize, &cinfo, &valCol->data, assist);
×
4529
  if (code) {
×
4530
    return code;
×
4531
  }
4532

4533
  return 0;
×
4534
}
4535

4536
int32_t tValueColumnCompressInfoEncode(const SValueColumnCompressInfo *info, SBuffer *buffer) {
88✔
4537
  int32_t code;
4538
  uint8_t fmtVer = 0;
88✔
4539

4540
  if ((code = tBufferPutU8(buffer, fmtVer))) return code;
176!
4541
  if ((code = tBufferPutI8(buffer, info->cmprAlg))) return code;
176!
4542
  if ((code = tBufferPutI8(buffer, info->type))) return code;
176!
4543
  if (IS_VAR_DATA_TYPE(info->type)) {
88!
4544
    if ((code = tBufferPutI32v(buffer, info->offsetOriginalSize))) return code;
12!
4545
    if ((code = tBufferPutI32v(buffer, info->offsetCompressedSize))) return code;
12!
4546
  }
4547
  if ((code = tBufferPutI32v(buffer, info->dataOriginalSize))) return code;
176!
4548
  if ((code = tBufferPutI32v(buffer, info->dataCompressedSize))) return code;
176!
4549

4550
  return 0;
88✔
4551
}
4552

4553
int32_t tValueColumnCompressInfoDecode(SBufferReader *reader, SValueColumnCompressInfo *info) {
×
4554
  int32_t code;
4555
  uint8_t fmtVer;
4556

4557
  if ((code = tBufferGetU8(reader, &fmtVer))) return code;
×
4558
  if (fmtVer == 0) {
×
4559
    if ((code = tBufferGetI8(reader, &info->cmprAlg))) return code;
×
4560
    if ((code = tBufferGetI8(reader, &info->type))) return code;
×
4561
    if (IS_VAR_DATA_TYPE(info->type)) {
×
4562
      if ((code = tBufferGetI32v(reader, &info->offsetOriginalSize))) return code;
×
4563
      if ((code = tBufferGetI32v(reader, &info->offsetCompressedSize))) return code;
×
4564
    } else {
4565
      info->offsetOriginalSize = 0;
×
4566
      info->offsetCompressedSize = 0;
×
4567
    }
4568
    if ((code = tBufferGetI32v(reader, &info->dataOriginalSize))) return code;
×
4569
    if ((code = tBufferGetI32v(reader, &info->dataCompressedSize))) return code;
×
4570
  } else {
4571
    return TSDB_CODE_INVALID_PARA;
×
4572
  }
4573

4574
  return 0;
×
4575
}
4576

4577
int32_t tCompressData(void          *input,       // input
10,040,884✔
4578
                      SCompressInfo *info,        // compress info
4579
                      void          *output,      // output
4580
                      int32_t        outputSize,  // output size
4581
                      SBuffer       *buffer       // assistant buffer provided by caller, can be NULL
4582
) {
4583
  int32_t extraSizeNeeded;
4584
  int32_t code;
4585

4586
  extraSizeNeeded = (info->cmprAlg == NO_COMPRESSION) ? info->originalSize : info->originalSize + COMP_OVERFLOW_BYTES;
10,040,884!
4587
  if (!(outputSize >= extraSizeNeeded)) {
10,040,884!
4588
    return TSDB_CODE_INVALID_PARA;
×
4589
  }
4590

4591
  if (info->cmprAlg == NO_COMPRESSION) {
10,040,884!
4592
    (void)memcpy(output, input, info->originalSize);
×
4593
    info->compressedSize = info->originalSize;
×
4594
  } else if (info->cmprAlg == ONE_STAGE_COMP || info->cmprAlg == TWO_STAGE_COMP) {
11,764,751✔
4595
    SBuffer local;
4596

4597
    tBufferInit(&local);
4598
    if (buffer == NULL) {
1,721,598!
4599
      buffer = &local;
×
4600
    }
4601

4602
    if (info->cmprAlg == TWO_STAGE_COMP) {
1,721,598!
4603
      code = tBufferEnsureCapacity(buffer, extraSizeNeeded);
1,723,650✔
4604
      if (code) {
1,723,650!
4605
        tBufferDestroy(&local);
4606
        return code;
×
4607
      }
4608
    }
4609

4610
    info->compressedSize = tDataTypes[info->dataType].compFunc(  //
3,445,465✔
4611
        input,                                                   // input
4612
        info->originalSize,                                      // input size
4613
        info->originalSize / tDataTypes[info->dataType].bytes,   // number of elements
1,721,598✔
4614
        output,                                                  // output
4615
        outputSize,                                              // output size
4616
        info->cmprAlg,                                           // compression algorithm
1,721,598✔
4617
        buffer->data,                                            // buffer
4618
        buffer->capacity                                         // buffer size
1,721,598✔
4619
    );
4620
    if (info->compressedSize < 0) {
1,723,867!
4621
      tBufferDestroy(&local);
4622
      return TSDB_CODE_COMPRESS_ERROR;
×
4623
    }
4624

4625
    tBufferDestroy(&local);
4626
  } else {
4627
    DEFINE_VAR(info->cmprAlg)
8,319,286✔
4628
    if ((l1 == L1_UNKNOWN && l2 == L2_UNKNOWN) || (l1 == L1_DISABLED && l2 == L2_DISABLED)) {
8,319,286!
4629
      (void)memcpy(output, input, info->originalSize);
×
4630
      info->compressedSize = info->originalSize;
×
4631
      return 0;
×
4632
    }
4633
    SBuffer local;
4634

4635
    tBufferInit(&local);
4636
    if (buffer == NULL) {
8,319,286!
4637
      buffer = &local;
×
4638
    }
4639
    code = tBufferEnsureCapacity(buffer, extraSizeNeeded);
8,319,286✔
4640

4641
    info->compressedSize = tDataCompress[info->dataType].compFunc(  //
16,639,773✔
4642
        input,                                                      // input
4643
        info->originalSize,                                         // input size
4644
        info->originalSize / tDataTypes[info->dataType].bytes,      // number of elements
8,319,309✔
4645
        output,                                                     // output
4646
        outputSize,                                                 // output size
4647
        info->cmprAlg,                                              // compression algorithm
4648
        buffer->data,                                               // buffer
4649
        buffer->capacity                                            // buffer size
8,319,309✔
4650
    );
4651
    if (info->compressedSize < 0) {
8,320,464!
4652
      tBufferDestroy(&local);
4653
      return TSDB_CODE_COMPRESS_ERROR;
×
4654
    }
4655

4656
    tBufferDestroy(&local);
4657
    // new col compress
4658
  }
4659

4660
  return 0;
10,044,331✔
4661
}
4662

4663
int32_t tDecompressData(void                *input,       // input
15,703,977✔
4664
                        const SCompressInfo *info,        // compress info
4665
                        void                *output,      // output
4666
                        int32_t              outputSize,  // output size
4667
                        SBuffer             *buffer       // assistant buffer provided by caller, can be NULL
4668
) {
4669
  int32_t code;
4670

4671
  if (!(outputSize >= info->originalSize)) {
15,703,977!
4672
    return TSDB_CODE_INVALID_PARA;
×
4673
  }
4674

4675
  if (info->cmprAlg == NO_COMPRESSION) {
15,703,977!
4676
    if (!(info->compressedSize == info->originalSize)) {
×
4677
      return TSDB_CODE_INVALID_PARA;
×
4678
    }
4679
    (void)memcpy(output, input, info->compressedSize);
×
4680
  } else if (info->cmprAlg == ONE_STAGE_COMP || info->cmprAlg == TWO_STAGE_COMP) {
23,608,299!
4681
    SBuffer local;
4682

4683
    tBufferInit(&local);
4684
    if (buffer == NULL) {
7,888,086!
4685
      buffer = &local;
×
4686
    }
4687

4688
    if (info->cmprAlg == TWO_STAGE_COMP) {
7,888,086!
4689
      code = tBufferEnsureCapacity(buffer, info->originalSize + COMP_OVERFLOW_BYTES);
7,903,364✔
4690
      if (code) {
7,903,851!
4691
        tBufferDestroy(&local);
4692
        return code;
×
4693
      }
4694
    }
4695

4696
    int32_t decompressedSize = tDataTypes[info->dataType].decompFunc(
7,888,573✔
4697
        input,                                                  // input
4698
        info->compressedSize,                                   // inputSize
7,888,573✔
4699
        info->originalSize / tDataTypes[info->dataType].bytes,  // number of elements
7,888,573✔
4700
        output,                                                 // output
4701
        outputSize,                                             // output size
4702
        info->cmprAlg,                                          // compression algorithm
7,888,573✔
4703
        buffer->data,                                           // helper buffer
4704
        buffer->capacity                                        // extra buffer size
7,888,573✔
4705
    );
4706
    if (decompressedSize < 0) {
7,904,322!
4707
      tBufferDestroy(&local);
4708
      return TSDB_CODE_COMPRESS_ERROR;
×
4709
    }
4710

4711
    if (!(decompressedSize == info->originalSize)) {
7,904,322!
4712
      return TSDB_CODE_COMPRESS_ERROR;
×
4713
    }
4714
    tBufferDestroy(&local);
4715
  } else {
4716
    DEFINE_VAR(info->cmprAlg);
7,815,891✔
4717
    if (l1 == L1_DISABLED && l2 == L2_DISABLED) {
7,815,891!
4718
      (void)memcpy(output, input, info->compressedSize);
×
4719
      return 0;
×
4720
    }
4721
    SBuffer local;
4722

4723
    tBufferInit(&local);
4724
    if (buffer == NULL) {
7,815,891!
4725
      buffer = &local;
×
4726
    }
4727
    code = tBufferEnsureCapacity(buffer, info->originalSize + COMP_OVERFLOW_BYTES);
7,815,891✔
4728
    if (code) {
7,816,507!
4729
      return code;
×
4730
    }
4731

4732
    int32_t decompressedSize = tDataCompress[info->dataType].decompFunc(
7,816,507✔
4733
        input,                                                  // input
4734
        info->compressedSize,                                   // inputSize
7,816,507✔
4735
        info->originalSize / tDataTypes[info->dataType].bytes,  // number of elements
7,816,507✔
4736
        output,                                                 // output
4737
        outputSize,                                             // output size
4738
        info->cmprAlg,                                          // compression algorithm
7,816,507✔
4739
        buffer->data,                                           // helper buffer
4740
        buffer->capacity                                        // extra buffer size
7,816,507✔
4741
    );
4742
    if (decompressedSize < 0) {
7,817,719!
4743
      tBufferDestroy(&local);
4744
      return TSDB_CODE_COMPRESS_ERROR;
×
4745
    }
4746

4747
    if (!(decompressedSize == info->originalSize)) {
7,817,719!
4748
      return TSDB_CODE_COMPRESS_ERROR;
×
4749
    }
4750
    tBufferDestroy(&local);
4751
  }
4752

4753
  return 0;
15,722,041✔
4754
}
4755

4756
int32_t tCompressDataToBuffer(void *input, SCompressInfo *info, SBuffer *output, SBuffer *assist) {
10,041,360✔
4757
  int32_t code;
4758

4759
  code = tBufferEnsureCapacity(output, output->size + info->originalSize + COMP_OVERFLOW_BYTES);
10,041,360✔
4760
  if (code) return code;
10,041,345!
4761

4762
  code = tCompressData(input, info, tBufferGetDataEnd(output), output->capacity - output->size, assist);
10,041,345✔
4763
  if (code) return code;
10,042,307!
4764

4765
  output->size += info->compressedSize;
10,042,307✔
4766
  return 0;
10,042,307✔
4767
}
4768

4769
int32_t tDecompressDataToBuffer(void *input, SCompressInfo *info, SBuffer *output, SBuffer *assist) {
7,903,078✔
4770
  int32_t code;
4771

4772
  code = tBufferEnsureCapacity(output, output->size + info->originalSize);
7,903,078✔
4773
  if (code) return code;
7,902,287!
4774

4775
  code = tDecompressData(input, info, tBufferGetDataEnd(output), output->capacity - output->size, assist);
7,902,287✔
4776
  if (code) return code;
7,903,666!
4777

4778
  output->size += info->originalSize;
7,903,666✔
4779
  return 0;
7,903,666✔
4780
}
4781

4782
// handle all types, including var data
4783
void valueSetDatum(SValue *pVal, int8_t type, void *pDatum, uint32_t len) {
1,299,531,140✔
4784
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
1,299,531,140!
4785
    pVal->pData = pDatum;
×
4786
    pVal->nData = len;
×
4787
  } else {
4788
    switch (len) {
1,300,989,350!
4789
      case sizeof(uint8_t):
35,430,819✔
4790
        pVal->val = *(uint8_t *)pDatum;
35,430,819✔
4791
        break;
35,430,819✔
4792
      case sizeof(uint16_t):
21,056,585✔
4793
        pVal->val = *(uint16_t *)pDatum;
21,056,585✔
4794
        break;
21,056,585✔
4795
      case sizeof(uint32_t):
665,215,579✔
4796
        pVal->val = *(uint32_t *)pDatum;
665,215,579✔
4797
        break;
665,215,579✔
4798
      case sizeof(uint64_t):
592,900,007✔
4799
        pVal->val = *(uint64_t *)pDatum;
592,900,007✔
4800
        break;
592,900,007✔
4801
      default:
×
4802
        break;
×
4803
    }
4804
  }
4805
}
1,299,531,140✔
4806

4807
void valueCloneDatum(SValue *pDst, const SValue *pSrc, int8_t type) {
24✔
4808
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
24!
4809
    memcpy(pDst->pData, pSrc->pData, pSrc->nData);
×
4810
    pDst->nData = pSrc->nData;
×
4811
  } else {
4812
    pDst->val = pSrc->val;
24✔
4813
  }
4814
}
24✔
4815
void valueClearDatum(SValue *pVal, int8_t type) {
×
4816
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
×
4817
    taosMemoryFreeClear(pVal->pData);
×
4818
    pVal->nData = 0;
×
4819
  } else {
4820
    pVal->val = 0;
×
4821
  }
4822
}
×
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