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

taosdata / TDengine / #3720

26 Mar 2025 06:20AM UTC coverage: 30.242% (-31.7%) from 61.936%
#3720

push

travis-ci

web-flow
feat(taosBenchmark): supports decimal data type (#30456)

* feat: taosBenchmark supports decimal data type

* build: decimal script not use pytest.sh

* fix: fix typo for decimal script

* test: insertBasic.py debug

71234 of 313946 branches covered (22.69%)

Branch coverage included in aggregate %.

38 of 423 new or added lines in 8 files covered. (8.98%)

120240 existing lines in 447 files now uncovered.

118188 of 312400 relevant lines covered (37.83%)

1450220.33 hits per line

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

29.45
/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) {
198,325,522✔
66
  int32_t n = 0;
198,325,522✔
67
  n += tPutI8(p ? p + n : p, index->type);
198,325,522✔
68
  n += tPutU32v(p ? p + n : p, index->offset);
198,325,522✔
69
  return n;
198,325,522✔
70
}
71

72
static int32_t tGetPrimaryKeyIndex(uint8_t *p, SPrimaryKeyIndex *index) {
187,549,452✔
73
  int32_t n = 0;
187,549,452✔
74
  n += tGetI8(p + n, &index->type);
187,549,452!
75
  n += tGetU32v(p + n, &index->offset);
187,549,452!
76
  return n;
187,549,452✔
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++;
113,746✔
108
  return 0;
113,746✔
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++;
5,082,600✔
114
  sinfo->kvMaxOffset = sinfo->kvPayloadSize;
5,082,600✔
115
  sinfo->kvPayloadSize += tPutI16v(NULL, -pTColumn->colId);
5,082,600✔
116
  return 0;
5,082,600✔
117
}
118

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

122
  if (isPK) {
191,177,258✔
123
    sinfo->tupleIndices[sinfo->numOfPKs].type = colVal->value.type;
67,899,492✔
124
    sinfo->tupleIndices[sinfo->numOfPKs].offset =
67,899,492✔
125
        IS_VAR_DATA_TYPE(pTColumn->type) ? sinfo->tupleVarSize + sinfo->tupleFixedSize : pTColumn->offset;
67,899,492!
126
    sinfo->kvIndices[sinfo->numOfPKs].type = colVal->value.type;
67,899,492✔
127
    sinfo->kvIndices[sinfo->numOfPKs].offset = sinfo->kvPayloadSize;
67,899,492✔
128
    sinfo->numOfPKs++;
67,899,492✔
129
  }
130

131
  sinfo->kvMaxOffset = sinfo->kvPayloadSize;
191,177,258✔
132
  if (IS_VAR_DATA_TYPE(colVal->value.type)) {
191,177,258!
UNCOV
133
    sinfo->tupleVarSize += tPutU32v(NULL, colVal->value.nData)  // size
×
UNCOV
134
                           + colVal->value.nData;               // value
×
135

UNCOV
136
    sinfo->kvPayloadSize += tPutI16v(NULL, colVal->cid)            // colId
×
UNCOV
137
                            + tPutU32v(NULL, colVal->value.nData)  // size
×
UNCOV
138
                            + colVal->value.nData;                 // value
×
139
  } else {
140
    sinfo->kvPayloadSize += tPutI16v(NULL, colVal->cid)              // colId
193,029,022✔
141
                            + tDataTypes[colVal->value.type].bytes;  // value
193,029,022✔
142
  }
143
  sinfo->numOfValue++;
191,177,258✔
144
}
191,177,258✔
145

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

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

162
  *sinfo = (SRowBuildScanInfo){
64,376,500✔
163
      .tupleFixedSize = schema->flen,
64,376,500✔
164
  };
165

166
  // loop scan
167
  for (int32_t i = 1; i < schema->numOfCols; i++) {
259,933,403✔
168
    for (;;) {
169
      if (colValIndex >= numOfColVals) {
194,498,592!
UNCOV
170
        if ((code = tRowBuildScanAddNone(sinfo, schema->columns + i))) goto _exit;
×
UNCOV
171
        break;
×
172
      }
173

174
      if (colValArray[colValIndex].cid == schema->columns[i].colId) {
194,498,592!
175
        if (!(colValArray[colValIndex].value.type == schema->columns[i].type)) {
195,556,903!
176
          code = TSDB_CODE_INVALID_PARA;
×
177
          goto _exit;
×
178
        }
179

180
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {
195,556,903✔
181
          tRowBuildScanAddValue(sinfo, &colValArray[colValIndex], schema->columns + i);
191,177,258✔
182
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {
4,379,645!
183
          if ((code = tRowBuildScanAddNull(sinfo, schema->columns + i))) goto _exit;
10,165,200!
UNCOV
184
        } else if (COL_VAL_IS_NONE(&colValArray[colValIndex])) {
×
185
          if ((code = tRowBuildScanAddNone(sinfo, schema->columns + i))) goto _exit;
227,492!
186
        }
187

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

199
  if (sinfo->numOfNone) {
64,376,500✔
200
    sinfo->flag |= HAS_NONE;
56,875✔
201
  }
202
  if (sinfo->numOfNull) {
64,376,500✔
203
    sinfo->flag |= HAS_NULL;
4,873,954✔
204
  }
205
  if (sinfo->numOfValue) {
64,376,500!
206
    sinfo->flag |= HAS_VALUE;
66,769,460✔
207
  }
208

209
  // Tuple
210
  sinfo->tupleFlag = sinfo->flag;
64,376,500✔
211
  switch (sinfo->flag) {
64,376,500!
212
    case HAS_NONE:
10,001✔
213
    case HAS_NULL:
214
      sinfo->tupleBitmapSize = 0;
10,001✔
215
      sinfo->tupleFixedSize = 0;
10,001✔
216
      break;
10,001✔
217
    case HAS_VALUE:
62,680,373✔
218
      sinfo->tupleBitmapSize = 0;
62,680,373✔
219
      sinfo->tupleFixedSize = schema->flen;
62,680,373✔
220
      break;
62,680,373✔
UNCOV
221
    case (HAS_NONE | HAS_NULL):
×
UNCOV
222
      sinfo->tupleBitmapSize = BIT1_SIZE(schema->numOfCols - 1);
×
UNCOV
223
      sinfo->tupleFixedSize = 0;
×
UNCOV
224
      break;
×
225
    case (HAS_NONE | HAS_VALUE):
4,923,095✔
226
    case (HAS_NULL | HAS_VALUE):
227
      sinfo->tupleBitmapSize = BIT1_SIZE(schema->numOfCols - 1);
4,923,095✔
228
      sinfo->tupleFixedSize = schema->flen;
4,923,095✔
229
      break;
4,923,095✔
UNCOV
230
    case (HAS_NONE | HAS_NULL | HAS_VALUE):
×
UNCOV
231
      sinfo->tupleBitmapSize = BIT2_SIZE(schema->numOfCols - 1);
×
UNCOV
232
      sinfo->tupleFixedSize = schema->flen;
×
UNCOV
233
      break;
×
234
  }
235
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
131,579,594✔
236
    sinfo->tupleIndices[i].offset += sinfo->tupleBitmapSize;
67,261,144✔
237
    sinfo->tuplePKSize += tPutPrimaryKeyIndex(NULL, sinfo->tupleIndices + i);
67,261,144✔
238
  }
239
  sinfo->tupleRowSize = sizeof(SRow)              // SRow
64,318,450✔
240
                        + sinfo->tuplePKSize      // primary keys
64,318,450✔
241
                        + sinfo->tupleBitmapSize  // bitmap
64,318,450✔
242
                        + sinfo->tupleFixedSize   // fixed part
64,318,450✔
243
                        + sinfo->tupleVarSize;    // var part
64,318,450✔
244

245
  // Key-Value
246
  if (sinfo->kvMaxOffset <= UINT8_MAX) {
64,318,450!
247
    sinfo->kvFlag = (KV_FLG_LIT | sinfo->flag);
67,367,555✔
248
    sinfo->kvIndexSize = sizeof(SKVIdx) + (sinfo->numOfNull + sinfo->numOfValue) * sizeof(uint8_t);
67,367,555✔
UNCOV
249
  } else if (sinfo->kvMaxOffset <= UINT16_MAX) {
×
UNCOV
250
    sinfo->kvFlag = (KV_FLG_MID | sinfo->flag);
×
UNCOV
251
    sinfo->kvIndexSize = sizeof(SKVIdx) + (sinfo->numOfNull + sinfo->numOfValue) * sizeof(uint16_t);
×
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++) {
131,672,592✔
257
    sinfo->kvIndices[i].offset += sinfo->kvIndexSize;
67,448,787✔
258
    sinfo->kvPKSize += tPutPrimaryKeyIndex(NULL, sinfo->kvIndices + i);
67,448,787✔
259
  }
260
  sinfo->kvRowSize = sizeof(SRow)             // SRow
64,223,805✔
261
                     + sinfo->kvPKSize        // primary keys
64,223,805✔
262
                     + sinfo->kvIndexSize     // index array
64,223,805✔
263
                     + sinfo->kvPayloadSize;  // payload
64,223,805✔
264

265
_exit:
64,223,805✔
266
  return code;
64,223,805✔
267
}
268

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

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

283
  if (sinfo->tupleFlag == HAS_NONE || sinfo->tupleFlag == HAS_NULL) {
68,393,089!
UNCOV
284
    return 0;
×
285
  }
286

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

292
  // primary keys
293
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
136,820,637✔
294
    primaryKeys += tPutPrimaryKeyIndex(primaryKeys, sinfo->tupleIndices + i);
69,034,845✔
295
  }
296

297
  // bitmap + fixed + varlen
298
  int32_t numOfColVals = TARRAY_SIZE(aColVal);
67,785,792✔
299
  int32_t colValIndex = 1;
67,785,792✔
300
  for (int32_t i = 1; i < schema->numOfCols; i++) {
264,608,497✔
301
    for (;;) {
302
      if (colValIndex >= numOfColVals) {  // NONE
196,385,421!
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) {
196,385,421!
308
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {  // value
196,822,705✔
309
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_VALUE);
192,626,088!
310

311
          if (IS_VAR_DATA_TYPE(schema->columns[i].type)) {
192,626,088!
UNCOV
312
            *(int32_t *)(fixed + schema->columns[i].offset) = varlen - fixed - sinfo->tupleFixedSize;
×
UNCOV
313
            varlen += tPutU32v(varlen, colValArray[colValIndex].value.nData);
×
UNCOV
314
            if (colValArray[colValIndex].value.nData) {
×
315
              (void)memcpy(varlen, colValArray[colValIndex].value.pData, colValArray[colValIndex].value.nData);
36,329✔
316
              varlen += colValArray[colValIndex].value.nData;
36,329✔
317
            }
318
          } else {
319
            (void)memcpy(fixed + schema->columns[i].offset,
192,672,488✔
320
                         VALUE_GET_DATUM(&colValArray[colValIndex].value, schema->columns[i].type),
192,672,488!
321
                         tDataTypes[schema->columns[i].type].bytes);
192,672,488✔
322
          }
323
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {  // NULL
4,196,617!
324
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NULL);
4,740,063!
UNCOV
325
        } else if (COL_VAL_IS_NONE(&colValArray[colValIndex])) {  // NONE
×
UNCOV
326
          ROW_SET_BITMAP(bitmap, sinfo->tupleFlag, i - 1, BIT_FLG_NONE);
×
327
        }
328

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

340
  return 0;
67,785,792✔
341
}
342

343
static FORCE_INLINE void tRowBuildKVRowSetIndex(uint8_t flag, SKVIdx *indices, uint32_t offset) {
344
  if (flag & KV_FLG_LIT) {
432,411✔
345
    ((uint8_t *)indices->idx)[indices->nCol] = (uint8_t)offset;
432,416✔
UNCOV
346
  } else if (flag & KV_FLG_MID) {
×
UNCOV
347
    ((uint16_t *)indices->idx)[indices->nCol] = (uint16_t)offset;
×
348
  } else {
349
    ((uint32_t *)indices->idx)[indices->nCol] = (uint32_t)offset;
×
350
  }
351
  indices->nCol++;
432,411✔
352
}
432,411✔
353

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

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

367
  if (!(sinfo->flag != HAS_NONE && sinfo->flag != HAS_NULL)) {
182,064!
368
    return TSDB_CODE_INVALID_PARA;
×
369
  }
370

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

376
  // primary keys
377
  for (int32_t i = 0; i < sinfo->numOfPKs; i++) {
364,133✔
378
    primaryKeys += tPutPrimaryKeyIndex(primaryKeys, sinfo->kvIndices + i);
182,068✔
379
  }
380

381
  int32_t numOfColVals = TARRAY_SIZE(aColVal);
182,065✔
382
  int32_t colValIndex = 1;
182,065✔
383
  for (int32_t i = 1; i < schema->numOfCols; i++) {
728,194✔
384
    for (;;) {
385
      if (colValIndex >= numOfColVals) {  // NONE
546,123!
UNCOV
386
        break;
×
387
      }
388

389
      if (colValArray[colValIndex].cid == schema->columns[i].colId) {
546,123!
390
        if (COL_VAL_IS_VALUE(&colValArray[colValIndex])) {  // value
546,129✔
391
          tRowBuildKVRowSetIndex(sinfo->kvFlag, indices, payloadSize);
182,069✔
392
          if (IS_VAR_DATA_TYPE(schema->columns[i].type)) {
182,069!
393
            payloadSize += tPutI16v(payload + payloadSize, colValArray[colValIndex].cid);
2✔
394
            payloadSize += tPutU32v(payload + payloadSize, colValArray[colValIndex].value.nData);
2✔
395
            if (colValArray[colValIndex].value.nData > 0) {
2!
UNCOV
396
              (void)memcpy(payload + payloadSize, colValArray[colValIndex].value.pData,
×
UNCOV
397
                           colValArray[colValIndex].value.nData);
×
398
            }
399
            payloadSize += colValArray[colValIndex].value.nData;
2✔
400
          } else {
401
            payloadSize += tPutI16v(payload + payloadSize, colValArray[colValIndex].cid);
182,067✔
402
            (void)memcpy(payload + payloadSize, VALUE_GET_DATUM(&colValArray[colValIndex].value, schema->columns[i].type),
182,067!
403
                         tDataTypes[schema->columns[i].type].bytes);
182,067✔
404
            payloadSize += tDataTypes[schema->columns[i].type].bytes;
182,067✔
405
          }
406
        } else if (COL_VAL_IS_NULL(&colValArray[colValIndex])) {  // NULL
364,060✔
407
          tRowBuildKVRowSetIndex(sinfo->kvFlag, indices, payloadSize);
250,342!
408
          payloadSize += tPutI16v(payload + payloadSize, -schema->columns[i].colId);
500,684✔
409
        }
410

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

421
  return 0;
182,065✔
422
}
423

424
int32_t tRowBuild(SArray *aColVal, const STSchema *pTSchema, SRow **ppRow) {
64,540,825✔
425
  int32_t           code;
426
  SRowBuildScanInfo sinfo;
427

428
  code = tRowBuildScan(aColVal, pTSchema, &sinfo);
64,540,825✔
429
  if (code) return code;
67,139,410!
430

431
  if (sinfo.tupleRowSize <= sinfo.kvRowSize) {
67,139,410✔
432
    code = tRowBuildTupleRow(aColVal, &sinfo, pTSchema, ppRow);
67,055,689✔
433
  } else {
434
    code = tRowBuildKVRow(aColVal, &sinfo, pTSchema, ppRow);
83,721✔
435
  }
436
  return code;
67,263,253✔
437
}
438

UNCOV
439
static int32_t tBindInfoCompare(const void *p1, const void *p2, const void *param) {
×
UNCOV
440
  if (((SBindInfo *)p1)->columnId < ((SBindInfo *)p2)->columnId) {
×
UNCOV
441
    return -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
 */
UNCOV
457
int32_t tRowBuildFromBind(SBindInfo *infos, int32_t numOfInfos, bool infoSorted, const STSchema *pTSchema,
×
458
                          SArray *rowArray, bool *pOrdered, bool *pDupTs) {
UNCOV
459
  if (infos == NULL || numOfInfos <= 0 || numOfInfos > pTSchema->numOfCols || pTSchema == NULL || rowArray == NULL) {
×
460
    return TSDB_CODE_INVALID_PARA;
×
461
  }
462

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

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

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

476
  SRowKey rowKey, lastRowKey;
UNCOV
477
  for (int32_t iRow = 0; iRow < numOfRows; iRow++) {
×
UNCOV
478
    taosArrayClear(colValArray);
×
479

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

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

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

UNCOV
517
    if (pOrdered && pDupTs) {
×
UNCOV
518
      tRowGetKey(row, &rowKey);
×
UNCOV
519
      if (iRow == 0) {
×
UNCOV
520
        *pOrdered = true;
×
UNCOV
521
        *pDupTs = false;
×
522
      } else {
UNCOV
523
        if (*pOrdered) {
×
UNCOV
524
          int32_t res = tRowKeyCompare(&rowKey, &lastRowKey);
×
UNCOV
525
          *pOrdered = (res >= 0);
×
UNCOV
526
          if (!*pDupTs) {
×
UNCOV
527
            *pDupTs = (res == 0);
×
528
          }
529
        }
530
      }
UNCOV
531
      lastRowKey = rowKey;
×
532
    }
533
  }
534

UNCOV
535
_exit:
×
UNCOV
536
  taosArrayDestroy(colValArray);
×
UNCOV
537
  return code;
×
538
}
539

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

544
  STColumn *pTColumn = pTSchema->columns + iCol;
981,802✔
545

546
  if (iCol == 0) {
981,802✔
547
    pColVal->cid = pTColumn->colId;
360✔
548
    pColVal->value.type = pTColumn->type;
360✔
549
    pColVal->flag = CV_FLAG_VALUE;
360✔
550
    VALUE_SET_TRIVIAL_DATUM(&pColVal->value, pRow->ts);
360✔
551
    return 0;
360✔
552
  }
553

554
  if (pRow->flag == HAS_NONE) {
981,442!
UNCOV
555
    *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
×
UNCOV
556
    return 0;
×
557
  }
558

559
  if (pRow->flag == HAS_NULL) {
981,442✔
560
    *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
16✔
561
    return 0;
16✔
562
  }
563

564
  SPrimaryKeyIndex index;
565
  uint8_t         *data = pRow->data;
981,426✔
566
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
1,796,196✔
567
    data += tGetPrimaryKeyIndex(data, &index);
814,783✔
568
  }
569

570
  if (pRow->flag >> 4) {  // KV Row
981,413✔
571
    SKVIdx  *pIdx = (SKVIdx *)data;
58,216✔
572
    uint8_t *pv = NULL;
58,216✔
573

574
    if (pRow->flag & KV_FLG_LIT) {
58,216!
575
      pv = pIdx->idx + pIdx->nCol;
58,216✔
UNCOV
576
    } else if (pRow->flag & KV_FLG_MID) {
×
UNCOV
577
      pv = pIdx->idx + (pIdx->nCol << 1);
×
578
    } else {
UNCOV
579
      pv = pIdx->idx + (pIdx->nCol << 2);
×
580
    }
581

582
    int16_t lidx = 0;
58,216✔
583
    int16_t ridx = pIdx->nCol - 1;
58,216✔
584
    while (lidx <= ridx) {
59,252!
585
      int16_t  mid = (lidx + ridx) >> 1;
59,252✔
586
      uint8_t *pData = NULL;
59,252✔
587
      if (pRow->flag & KV_FLG_LIT) {
59,252!
588
        pData = pv + ((uint8_t *)pIdx->idx)[mid];
59,252✔
UNCOV
589
      } else if (pRow->flag & KV_FLG_MID) {
×
UNCOV
590
        pData = pv + ((uint16_t *)pIdx->idx)[mid];
×
591
      } else {
UNCOV
592
        pData = pv + ((uint32_t *)pIdx->idx)[mid];
×
593
      }
594

595
      int16_t cid;
596
      pData += tGetI16v(pData, &cid);
59,252✔
597

598
      if (TABS(cid) == pTColumn->colId) {
59,252✔
599
        if (cid < 0) {
58,216✔
600
          *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
616✔
601
        } else {
602
          pColVal->cid = pTColumn->colId;
57,600✔
603
          pColVal->value.type = pTColumn->type;
57,600✔
604
          pColVal->flag = CV_FLAG_VALUE;
57,600✔
605

606
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
57,600!
UNCOV
607
            pData += tGetU32v(pData, &pColVal->value.nData);
×
UNCOV
608
            if (pColVal->value.nData > 0) {
×
UNCOV
609
              pColVal->value.pData = pData;
×
610
            } else {
UNCOV
611
              pColVal->value.pData = NULL;
×
612
            }
613
          } else {
614
            valueSetDatum(&pColVal->value, pTColumn->type, pData, pTColumn->bytes);
57,600✔
615
          }
616
        }
617
        return 0;
58,218✔
618
      } else if (TABS(cid) < pTColumn->colId) {
1,036✔
619
        lidx = mid + 1;
308✔
620
      } else {
621
        ridx = mid - 1;
728✔
622
      }
623
    }
624

UNCOV
625
    *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
×
626
  } else {  // Tuple Row
627
    uint8_t *bitmap = data;
923,197✔
628
    uint8_t *fixed;
629
    uint8_t *varlen;
630
    uint8_t  bit;
631

632
    if (pRow->flag == HAS_VALUE) {
923,197✔
633
      fixed = bitmap;
876,197✔
634
      bit = BIT_FLG_VALUE;
876,197✔
635
    } else if (pRow->flag == (HAS_NONE | HAS_NULL | HAS_VALUE)) {
47,000!
636
      fixed = BIT2_SIZE(pTSchema->numOfCols - 1) + bitmap;
×
UNCOV
637
      bit = GET_BIT2(bitmap, iCol - 1);
×
638
    } else {
639
      fixed = BIT1_SIZE(pTSchema->numOfCols - 1) + bitmap;
47,000✔
640
      bit = GET_BIT1(bitmap, iCol - 1);
47,000✔
641

642
      if (pRow->flag == (HAS_NONE | HAS_VALUE)) {
47,000!
UNCOV
643
        if (bit) bit++;
×
644
      } else if (pRow->flag == (HAS_NULL | HAS_VALUE)) {
47,000!
645
        bit++;
47,007✔
646
      }
647
    }
648
    varlen = fixed + pTSchema->flen;
923,197✔
649

650
    if (bit == BIT_FLG_NONE) {
923,197!
UNCOV
651
      *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
×
UNCOV
652
      return 0;
×
653
    } else if (bit == BIT_FLG_NULL) {
923,197✔
654
      *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
10,372✔
655
      return 0;
10,372✔
656
    }
657

658
    pColVal->cid = pTColumn->colId;
912,825✔
659
    pColVal->value.type = pTColumn->type;
912,825✔
660
    pColVal->flag = CV_FLAG_VALUE;
912,825✔
661
    if (IS_VAR_DATA_TYPE(pTColumn->type)) {
912,825!
662
      pColVal->value.pData = varlen + *(int32_t *)(fixed + pTColumn->offset);
23,301✔
663
      pColVal->value.pData += tGetU32v(pColVal->value.pData, &pColVal->value.nData);
46,602!
664
    } else {
665
      valueSetDatum(&pColVal->value, pTColumn->type, fixed + pTColumn->offset, TYPE_BYTES[pTColumn->type]);
889,524✔
666
    }
667
  }
668

669
  return 0;
912,541✔
670
}
671

672
void tRowDestroy(SRow *pRow) {
69,825,898✔
673
  if (pRow) taosMemoryFree(pRow);
69,825,898!
674
}
69,857,188✔
675

UNCOV
676
static int32_t tRowPCmprFn(const void *p1, const void *p2) {
×
677
  SRowKey key1, key2;
UNCOV
678
  tRowGetKey(*(SRow **)p1, &key1);
×
UNCOV
679
  tRowGetKey(*(SRow **)p2, &key2);
×
UNCOV
680
  return tRowKeyCompare(&key1, &key2);
×
681
}
UNCOV
682
static void    tRowPDestroy(SRow **ppRow) { tRowDestroy(*ppRow); }
×
UNCOV
683
static int32_t tRowMergeImpl(SArray *aRowP, STSchema *pTSchema, int32_t iStart, int32_t iEnd, int8_t flag) {
×
UNCOV
684
  int32_t code = 0;
×
685

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

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

UNCOV
697
  for (int32_t i = 0; i < nRow; i++) {
×
UNCOV
698
    SRow *pRowT = taosArrayGetP(aRowP, iStart + i);
×
699

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

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

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

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

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

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

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

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

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

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

UNCOV
772
  int32_t iStart = 0;
×
UNCOV
773
  while (iStart < aRowP->size) {
×
774
    SRowKey key1;
UNCOV
775
    SRow   *row1 = (SRow *)taosArrayGetP(aRowP, iStart);
×
776

UNCOV
777
    tRowGetKey(row1, &key1);
×
778

UNCOV
779
    int32_t iEnd = iStart + 1;
×
UNCOV
780
    while (iEnd < aRowP->size) {
×
781
      SRowKey key2;
UNCOV
782
      SRow   *row2 = (SRow *)taosArrayGetP(aRowP, iEnd);
×
UNCOV
783
      tRowGetKey(row2, &key2);
×
784

UNCOV
785
      if (tRowKeyCompare(&key1, &key2) != 0) break;
×
786

UNCOV
787
      iEnd++;
×
788
    }
789

UNCOV
790
    if (iEnd - iStart > 1) {
×
UNCOV
791
      code = tRowMergeImpl(aRowP, pTSchema, iStart, iEnd, flag);
×
UNCOV
792
      if (code) return code;
×
793
    }
794

795
    // the array is also changing, so the iStart just ++ instead of iEnd
UNCOV
796
    iStart++;
×
797
  }
798

UNCOV
799
  return code;
×
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

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

UNCOV
825
  int32_t code = 0;
×
826

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

UNCOV
833
  pIter->pRow = pRow;
×
UNCOV
834
  pIter->pTSchema = pTSchema;
×
UNCOV
835
  pIter->iTColumn = 0;
×
836

UNCOV
837
  if (pRow->flag == HAS_NONE || pRow->flag == HAS_NULL) goto _exit;
×
838

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

UNCOV
845
  if (pRow->flag >> 4) {
×
UNCOV
846
    pIter->iCol = 0;
×
UNCOV
847
    pIter->pIdx = (SKVIdx *)data;
×
UNCOV
848
    if (pRow->flag & KV_FLG_LIT) {
×
UNCOV
849
      pIter->pv = pIter->pIdx->idx + pIter->pIdx->nCol;
×
UNCOV
850
    } else if (pRow->flag & KV_FLG_MID) {
×
UNCOV
851
      pIter->pv = pIter->pIdx->idx + (pIter->pIdx->nCol << 1);  // * sizeof(uint16_t)
×
852
    } else {
UNCOV
853
      pIter->pv = pIter->pIdx->idx + (pIter->pIdx->nCol << 2);  // * sizeof(uint32_t)
×
854
    }
855
  } else {
UNCOV
856
    switch (pRow->flag) {
×
UNCOV
857
      case (HAS_NULL | HAS_NONE):
×
UNCOV
858
        pIter->pb = data;
×
UNCOV
859
        break;
×
UNCOV
860
      case HAS_VALUE:
×
UNCOV
861
        pIter->pf = data;
×
UNCOV
862
        pIter->pv = pIter->pf + pTSchema->flen;
×
UNCOV
863
        break;
×
UNCOV
864
      case (HAS_VALUE | HAS_NONE):
×
865
      case (HAS_VALUE | HAS_NULL):
UNCOV
866
        pIter->pb = data;
×
UNCOV
867
        pIter->pf = data + BIT1_SIZE(pTSchema->numOfCols - 1);
×
UNCOV
868
        pIter->pv = pIter->pf + pTSchema->flen;
×
869
        break;
×
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:
×
UNCOV
876
        break;
×
877
    }
878
  }
879

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

UNCOV
889
void tRowIterClose(SRowIter **ppIter) {
×
UNCOV
890
  SRowIter *pIter = *ppIter;
×
UNCOV
891
  if (pIter) {
×
UNCOV
892
    taosMemoryFree(pIter);
×
893
  }
UNCOV
894
  *ppIter = NULL;
×
UNCOV
895
}
×
896

UNCOV
897
SColVal *tRowIterNext(SRowIter *pIter) {
×
UNCOV
898
  if (pIter->iTColumn >= pIter->pTSchema->numOfCols) {
×
UNCOV
899
    return NULL;
×
900
  }
901

UNCOV
902
  STColumn *pTColumn = pIter->pTSchema->columns + pIter->iTColumn;
×
903

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

UNCOV
913
  if (pIter->pRow->flag == HAS_NONE) {
×
UNCOV
914
    pIter->cv = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
×
UNCOV
915
    goto _exit;
×
916
  }
917

UNCOV
918
  if (pIter->pRow->flag == HAS_NULL) {
×
UNCOV
919
    pIter->cv = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
×
UNCOV
920
    goto _exit;
×
921
  }
922

UNCOV
923
  if (pIter->pRow->flag >> 4) {  // KV
×
UNCOV
924
    if (pIter->iCol < pIter->pIdx->nCol) {
×
925
      uint8_t *pData;
926

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

935
      int16_t cid;
UNCOV
936
      pData += tGetI16v(pData, &cid);
×
937

UNCOV
938
      if (TABS(cid) == pTColumn->colId) {
×
UNCOV
939
        if (cid < 0) {
×
UNCOV
940
          pIter->cv = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
×
941
        } else {
UNCOV
942
          pIter->cv.cid = pTColumn->colId;
×
UNCOV
943
          pIter->cv.value.type = pTColumn->type;
×
UNCOV
944
          pIter->cv.flag = CV_FLAG_VALUE;
×
945

UNCOV
946
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
×
UNCOV
947
            pData += tGetU32v(pData, &pIter->cv.value.nData);
×
UNCOV
948
            if (pIter->cv.value.nData > 0) {
×
UNCOV
949
              pIter->cv.value.pData = pData;
×
950
            } else {
UNCOV
951
              pIter->cv.value.pData = NULL;
×
952
            }
953
          } else {
UNCOV
954
            valueSetDatum(&pIter->cv.value, pTColumn->type, pData, pTColumn->bytes);
×
955
          }
956
        }
957

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

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

1001
    pIter->cv.cid = pTColumn->colId;
×
1002
    pIter->cv.value.type = pTColumn->type;
×
1003
    pIter->cv.flag = CV_FLAG_VALUE;
×
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);
×
UNCOV
1007
      if (pIter->cv.value.nData > 0) {
×
UNCOV
1008
        pIter->cv.value.pData = pData;
×
1009
      } else {
UNCOV
1010
        pIter->cv.value.pData = NULL;
×
1011
      }
1012
    } else {
UNCOV
1013
      valueSetDatum(&pIter->cv.value, pTColumn->type, pIter->pf + pTColumn->offset, TYPE_BYTES[pTColumn->type]);
×
1014
    }
UNCOV
1015
    goto _exit;
×
1016
  }
1017

UNCOV
1018
_exit:
×
UNCOV
1019
  pIter->iTColumn++;
×
UNCOV
1020
  return &pIter->cv;
×
1021
}
1022

UNCOV
1023
static int32_t tRowNoneUpsertColData(SColData *aColData, int32_t nColData, int32_t flag) {
×
UNCOV
1024
  int32_t code = 0;
×
1025

UNCOV
1026
  if (flag) return code;
×
1027

UNCOV
1028
  for (int32_t iColData = 0; iColData < nColData; iColData++) {
×
UNCOV
1029
    code = tColDataAppendValueImpl[aColData[iColData].flag][CV_FLAG_NONE](&aColData[iColData], NULL, 0);
×
UNCOV
1030
    if (code) return code;
×
1031
  }
1032

UNCOV
1033
  return code;
×
1034
}
1035
static int32_t tRowNullUpsertColData(SColData *aColData, int32_t nColData, STSchema *pSchema, int32_t flag) {
10,001✔
1036
  int32_t code = 0;
10,001✔
1037

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

1043
  while (pColData) {
100,003✔
1044
    if (pTColumn) {
90,002!
1045
      if (pTColumn->colId == pColData->cid) {  // NULL
90,002!
1046
        if (flag == 0) {
90,002!
1047
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
90,002✔
1048
        } else {
UNCOV
1049
          code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
×
1050
        }
1051
        if (code) goto _exit;
90,002!
1052

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

1067
_exit:
10,001✔
1068
  return code;
10,001✔
1069
}
1070
static int32_t tRowTupleUpsertColData(SRow *pRow, STSchema *pTSchema, SColData *aColData, int32_t nColData,
17,940,642✔
1071
                                      int32_t flag) {
1072
  int32_t code = 0;
17,940,642✔
1073

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

1079
  uint8_t         *pb = NULL, *pf = NULL, *pv = NULL;
17,940,642✔
1080
  SPrimaryKeyIndex index;
1081
  uint8_t         *data = pRow->data;
17,940,642✔
1082
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
35,863,383✔
1083
    data += tGetPrimaryKeyIndex(data, &index);
17,922,741✔
1084
  }
1085

1086
  switch (pRow->flag) {
17,940,642!
1087
    case HAS_VALUE:
17,044,455✔
1088
      pf = data;  // TODO: fix here
17,044,455✔
1089
      pv = pf + pTSchema->flen;
17,044,455✔
1090
      break;
17,044,455✔
UNCOV
1091
    case (HAS_NULL | HAS_NONE):
×
UNCOV
1092
      pb = data;
×
UNCOV
1093
      break;
×
1094
    case (HAS_VALUE | HAS_NONE):
896,192✔
1095
    case (HAS_VALUE | HAS_NULL):
1096
      pb = data;
896,192✔
1097
      pf = pb + BIT1_SIZE(pTSchema->numOfCols - 1);
896,192✔
1098
      pv = pf + pTSchema->flen;
896,192✔
1099
      break;
896,192✔
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:
×
UNCOV
1106
      return TSDB_CODE_INVALID_DATA_FMT;
×
1107
  }
1108

1109
  while (pColData) {
71,853,891✔
1110
    if (pTColumn) {
53,911,377!
1111
      if (pTColumn->colId == pColData->cid) {
53,911,377!
1112
        if (!(pTColumn->type == pColData->type)) {
53,911,922!
UNCOV
1113
          return TSDB_CODE_INVALID_PARA;
×
1114
        }
1115
        if (pb) {
53,911,922✔
1116
          uint8_t bv;
1117
          switch (pRow->flag) {
2,688,576!
UNCOV
1118
            case (HAS_NULL | HAS_NONE):
×
UNCOV
1119
              bv = GET_BIT1(pb, iTColumn - 1);
×
UNCOV
1120
              break;
×
UNCOV
1121
            case (HAS_VALUE | HAS_NONE):
×
UNCOV
1122
              bv = GET_BIT1(pb, iTColumn - 1);
×
UNCOV
1123
              if (bv) bv++;
×
UNCOV
1124
              break;
×
1125
            case (HAS_VALUE | HAS_NULL):
2,688,576✔
1126
              bv = GET_BIT1(pb, iTColumn - 1) + 1;
2,688,576✔
1127
              break;
2,688,576✔
1128
            case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
1129
              bv = GET_BIT2(pb, iTColumn - 1);
×
1130
              break;
×
1131
            default:
×
UNCOV
1132
              return TSDB_CODE_INVALID_DATA_FMT;
×
1133
          }
1134

1135
          if (bv == BIT_FLG_NONE) {
2,688,576!
1136
            if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0)))
×
UNCOV
1137
              goto _exit;
×
UNCOV
1138
            goto _continue;
×
1139
          } else if (bv == BIT_FLG_NULL) {
2,688,576✔
1140
            if (flag == 0) {
896,192!
1141
              code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
896,192✔
1142
            } else {
UNCOV
1143
              code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
×
1144
            }
1145
            if (code) goto _exit;
896,192!
1146
            goto _continue;
896,192✔
1147
          }
1148
        }
1149

1150
        if (IS_VAR_DATA_TYPE(pColData->type)) {
53,050,291!
1151
          uint8_t *pData = pv + *(int32_t *)(pf + pTColumn->offset);
32,244!
1152
          uint32_t nData;
1153
          pData += tGetU32v(pData, &nData);
32,244✔
1154
          if (flag == 0) {
32,244!
1155
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData);
34,456✔
1156
          } else {
UNCOV
1157
            code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData, flag > 0);
×
1158
          }
1159
          if (code) goto _exit;
34,561!
1160
        } else {
1161
          if (flag == 0) {
52,983,486!
1162
            code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pf + pTColumn->offset,
52,983,491✔
1163
                                                                          TYPE_BYTES[pColData->type]);
52,983,491✔
1164
          } else {
UNCOV
1165
            code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pf + pTColumn->offset,
×
UNCOV
1166
                                                                          TYPE_BYTES[pColData->type], flag > 0);
×
1167
          }
1168
          if (code) goto _exit;
52,983,036!
1169
        }
1170

1171
      _continue:
52,983,036✔
1172
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
53,913,789✔
1173
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
53,913,789✔
1174
      } else if (pTColumn->colId > pColData->cid) {  // NONE
×
1175
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
UNCOV
1176
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1177
      } else {
UNCOV
1178
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
×
1179
      }
1180
    } else {
UNCOV
1181
      if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
UNCOV
1182
      pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1183
    }
1184
  }
1185

1186
_exit:
17,942,514✔
1187
  return code;
17,942,514✔
1188
}
1189
static int32_t tRowKVUpsertColData(SRow *pRow, STSchema *pTSchema, SColData *aColData, int32_t nColData, int32_t flag) {
23,543✔
1190
  int32_t code = 0;
23,543✔
1191

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

1199
  // primary keys
1200
  uint8_t         *data = pRow->data;
23,543✔
1201
  SPrimaryKeyIndex index;
1202
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
47,086✔
1203
    data += tGetPrimaryKeyIndex(data, &index);
23,543✔
1204
  }
1205

1206
  SKVIdx *pKVIdx = (SKVIdx *)data;
23,543✔
1207
  if (pRow->flag & KV_FLG_LIT) {
23,543!
1208
    pv = pKVIdx->idx + pKVIdx->nCol;
23,543✔
UNCOV
1209
  } else if (pRow->flag & KV_FLG_MID) {
×
1210
    pv = pKVIdx->idx + (pKVIdx->nCol << 1);
×
1211
  } else if (pRow->flag & KV_FLG_BIG) {
×
UNCOV
1212
    pv = pKVIdx->idx + (pKVIdx->nCol << 2);
×
1213
  } else {
UNCOV
1214
    return TSDB_CODE_INVALID_PARA;
×
1215
  }
1216

1217
  while (pColData) {
94,172✔
1218
    if (pTColumn) {
70,629!
1219
      if (pTColumn->colId == pColData->cid) {
70,629!
1220
        while (iCol < pKVIdx->nCol) {
70,629!
1221
          uint8_t *pData;
1222
          if (pRow->flag & KV_FLG_LIT) {
70,629!
1223
            pData = pv + ((uint8_t *)pKVIdx->idx)[iCol];
70,629✔
UNCOV
1224
          } else if (pRow->flag & KV_FLG_MID) {
×
1225
            pData = pv + ((uint16_t *)pKVIdx->idx)[iCol];
×
1226
          } else if (pRow->flag & KV_FLG_BIG) {
×
UNCOV
1227
            pData = pv + ((uint32_t *)pKVIdx->idx)[iCol];
×
1228
          } else {
UNCOV
1229
            return TSDB_CODE_INVALID_DATA_FMT;
×
1230
          }
1231

1232
          int16_t cid;
1233
          pData += tGetI16v(pData, &cid);
70,629✔
1234

1235
          if (TABS(cid) == pTColumn->colId) {
70,629!
1236
            if (cid < 0) {
70,629✔
1237
              if (flag == 0) {
47,086!
1238
                code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
47,086✔
1239
              } else {
UNCOV
1240
                code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0, flag > 0);
×
1241
              }
1242
              if (code) goto _exit;
47,086!
1243
            } else {
1244
              uint32_t nData;
1245
              if (IS_VAR_DATA_TYPE(pTColumn->type)) {
23,543!
UNCOV
1246
                pData += tGetU32v(pData, &nData);
×
1247
              } else {
1248
                nData = 0;
23,543✔
1249
              }
1250
              if (flag == 0) {
23,543!
1251
                code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData);
23,543✔
1252
              } else {
UNCOV
1253
                code = tColDataUpdateValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, pData, nData, flag > 0);
×
1254
              }
1255
              if (code) goto _exit;
23,543!
1256
            }
1257
            iCol++;
70,629✔
1258
            goto _continue;
70,629✔
UNCOV
1259
          } else if (TABS(cid) > pTColumn->colId) {  // NONE
×
UNCOV
1260
            break;
×
1261
          } else {
UNCOV
1262
            iCol++;
×
1263
          }
1264
        }
1265

UNCOV
1266
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
1267

UNCOV
1268
      _continue:
×
1269
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
70,629✔
1270
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
70,629✔
1271
      } else if (pTColumn->colId > pColData->cid) {
×
1272
        if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
UNCOV
1273
        pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1274
      } else {
UNCOV
1275
        pTColumn = (++iTColumn < pTSchema->numOfCols) ? &pTSchema->columns[iTColumn] : NULL;
×
1276
      }
1277
    } else {
UNCOV
1278
      if (flag == 0 && (code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0))) goto _exit;
×
UNCOV
1279
      pColData = (++iColData < nColData) ? &aColData[iColData] : NULL;
×
1280
    }
1281
  }
1282

1283
_exit:
23,543✔
1284
  return code;
23,543✔
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) {
17,973,724✔
1291
  if (!(pRow->sver == pTSchema->version)) return TSDB_CODE_INVALID_PARA;
17,973,724!
1292
  if (!(nColData > 0)) return TSDB_CODE_INVALID_PARA;
17,973,724!
1293

1294
  if (pRow->flag == HAS_NONE) {
17,973,724!
UNCOV
1295
    return tRowNoneUpsertColData(aColData, nColData, flag);
×
1296
  } else if (pRow->flag == HAS_NULL) {
17,973,724✔
1297
    return tRowNullUpsertColData(aColData, nColData, pTSchema, flag);
10,001✔
1298
  } else if (pRow->flag >> 4) {  // KV row
17,963,723✔
1299
    return tRowKVUpsertColData(pRow, pTSchema, aColData, nColData, flag);
23,543✔
1300
  } else {  // TUPLE row
1301
    return tRowTupleUpsertColData(pRow, pTSchema, aColData, nColData, flag);
17,940,180✔
1302
  }
1303
}
1304

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

1308
  if (key->numOfPKs == 0) {
168,833,768!
UNCOV
1309
    return;
×
1310
  }
1311

1312
  SPrimaryKeyIndex indices[TD_MAX_PK_COLS];
1313

1314
  uint8_t *data = row->data;
168,833,768✔
1315

1316
  for (int32_t i = 0; i < row->numOfPKs; i++) {
338,164,612✔
1317
    data += tGetPrimaryKeyIndex(data, &indices[i]);
169,141,780✔
1318
  }
1319

1320
  // primary keys
1321
  for (int32_t i = 0; i < row->numOfPKs; i++) {
338,883,361✔
1322
    key->pks[i].type = indices[i].type;
169,938,972✔
1323

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

1329
    if (IS_VAR_DATA_TYPE(indices[i].type)) {
169,938,972!
UNCOV
1330
      key->pks[i].pData = tdata;
×
UNCOV
1331
      key->pks[i].pData += tGetU32v(key->pks[i].pData, &key->pks[i].nData);
×
1332
    } else {
1333
      valueSetDatum(key->pks + i, indices[i].type, tdata, tDataTypes[indices[i].type].bytes);
169,938,972✔
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) {
16,836,486✔
1350
  switch (tv1->type) {
16,836,486!
UNCOV
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:
×
UNCOV
1355
      T_COMPARE_SCALAR_VALUE(int16_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
UNCOV
1356
    case TSDB_DATA_TYPE_INT:
×
UNCOV
1357
      T_COMPARE_SCALAR_VALUE(int32_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
1358
    case TSDB_DATA_TYPE_BIGINT:
16,836,490✔
1359
    case TSDB_DATA_TYPE_TIMESTAMP:
1360
      T_COMPARE_SCALAR_VALUE(int64_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
16,836,490✔
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:
×
UNCOV
1372
      T_COMPARE_SCALAR_VALUE(uint64_t, &VALUE_GET_TRIVIAL_DATUM(tv1), &VALUE_GET_TRIVIAL_DATUM(tv2));
×
UNCOV
1373
    case TSDB_DATA_TYPE_GEOMETRY:
×
1374
    case TSDB_DATA_TYPE_BINARY: {
UNCOV
1375
      int32_t ret = strncmp((const char *)tv1->pData, (const char *)tv2->pData, TMIN(tv1->nData, tv2->nData));
×
UNCOV
1376
      return ret ? ret : (tv1->nData < tv2->nData ? -1 : (tv1->nData > tv2->nData ? 1 : 0));
×
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);
×
UNCOV
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);
×
UNCOV
1385
      return ret ? ret : (tv1->nData < tv2->nData ? -1 : (tv1->nData > tv2->nData ? 1 : 0));
×
1386
    }
1387
    default:
×
UNCOV
1388
      break;
×
1389
  }
1390

UNCOV
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) {
269,020,835✔
1398
  if (key1->ts < key2->ts) {
269,023,013!
1399
    return -1;
65,999,400✔
1400
  } else if (key1->ts > key2->ts) {
203,023,613!
1401
    return 1;
187,017,248✔
1402
  }
1403

1404
  if (key1->numOfPKs == key2->numOfPKs) {
16,006,365!
1405
    for (uint8_t iKey = 0; iKey < key1->numOfPKs; iKey++) {
25,024,628!
1406
      int32_t ret = tValueCompare(&key1->pks[iKey], &key2->pks[iKey]);
16,836,494✔
1407
      if (ret) return ret;
16,836,358!
1408
    }
1409
  } else if (key1->numOfPKs < key2->numOfPKs) {
×
UNCOV
1410
    return -1;
×
1411
  } else {
UNCOV
1412
    return 1;
×
1413
  }
1414

1415
  return 0;
8,188,134✔
1416
}
1417

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

1422
  if (pSrc->numOfPKs > 0) {
245,991✔
1423
    for (int32_t i = 0; i < pSrc->numOfPKs; ++i) {
334,072✔
1424
      SValue *pVal = &pDst->pks[i];
167,034✔
1425
      pVal->type = pSrc->pks[i].type;
167,034✔
1426

1427
      valueCloneDatum(pVal, pSrc->pks + i, pVal->type);
167,034✔
1428
    }
1429
  }
1430
}
245,997✔
1431

1432
// STag ========================================
1433
static int tTagValCmprFn(const void *p1, const void *p2) {
6,035✔
1434
  if (((STagVal *)p1)->cid < ((STagVal *)p2)->cid) {
6,035✔
1435
    return -1;
132✔
1436
  } else if (((STagVal *)p1)->cid > ((STagVal *)p2)->cid) {
5,903!
UNCOV
1437
    return 1;
×
1438
  }
1439

1440
  return 0;
5,903✔
1441
}
1442
static int tTagValJsonCmprFn(const void *p1, const void *p2) {
752✔
1443
  return strcmp(((STagVal *)p1)[0].pKey, ((STagVal *)p2)[0].pKey);
752✔
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) {
5,366✔
1540
  int32_t n = 0;
5,366✔
1541

1542
  // key
1543
  if (isJson) {
5,366✔
1544
    n += tPutCStr(p ? p + n : p, pTagVal->pKey);
64✔
1545
  } else {
1546
    n += tPutI16v(p ? p + n : p, pTagVal->cid);
10,668✔
1547
  }
1548

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

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

1561
  return n;
5,366✔
1562
}
1563
static int32_t tGetTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson) {
6,715✔
1564
  int32_t n = 0;
6,715✔
1565

1566
  // key
1567
  if (isJson) {
6,715✔
1568
    n += tGetCStr(p + n, &pTagVal->pKey);
1,504!
1569
  } else {
1570
    n += tGetI16v(p + n, &pTagVal->cid);
11,926!
1571
  }
1572

1573
  // type
1574
  n += tGetI8(p + n, &pTagVal->type);
6,715!
1575

1576
  // value
1577
  if (IS_VAR_DATA_TYPE(pTagVal->type)) {
6,715!
1578
    n += tGetBinary(p + n, &pTagVal->pData, &pTagVal->nData);
1,388!
1579
  } else {
1580
    (void)memcpy(&(pTagVal->i64), p + n, tDataTypes[pTagVal->type].bytes);
6,021✔
1581
    n += tDataTypes[pTagVal->type].bytes;
6,021✔
1582
  }
1583

1584
  return n;
6,715✔
1585
}
1586

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

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

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

1604
  // sort
1605
  if (isJson) {
2,609✔
1606
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValJsonCmprFn);
8✔
1607
  } else {
1608
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValCmprFn);
2,601✔
1609
  }
1610

1611
  // get size
1612
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
5,292✔
1613
    szTag += tPutTagVal(NULL, (STagVal *)taosArrayGet(pArray, iTag), isJson);
2,686✔
1614
  }
1615
  if (szTag <= INT8_MAX) {
2,606!
1616
    szTag = szTag + sizeof(STag) + sizeof(int8_t) * nTag;
2,606✔
1617
  } else {
UNCOV
1618
    szTag = szTag + sizeof(STag) + sizeof(int16_t) * nTag;
×
UNCOV
1619
    isLarge = 1;
×
1620
  }
1621

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

1639
  if (isLarge) {
2,610!
UNCOV
1640
    p = (uint8_t *)&((int16_t *)(*ppTag)->idx)[nTag];
×
1641
  } else {
1642
    p = (uint8_t *)&(*ppTag)->idx[nTag];
2,610✔
1643
  }
1644
  n = 0;
2,610✔
1645
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
5,294✔
1646
    if (isLarge) {
2,686!
UNCOV
1647
      ((int16_t *)(*ppTag)->idx)[iTag] = n;
×
1648
    } else {
1649
      (*ppTag)->idx[iTag] = n;
2,686✔
1650
    }
1651
    n += tPutTagVal(p + n, (STagVal *)taosArrayGet(pArray, iTag), isJson);
2,686✔
1652
  }
1653
#ifdef TD_DEBUG_PRINT_TAG
1654
  debugPrintSTag(*ppTag, __func__, __LINE__);
1655
#endif
1656

1657
  return code;
2,608✔
1658

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

1663
void tTagFree(STag *pTag) {
34✔
1664
  if (pTag) taosMemoryFree(pTag);
34!
1665
}
34✔
1666

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

1672
  char  *data = NULL;
3,520✔
1673
  int8_t typeBytes = 0;
3,520✔
1674
  if (isJson) {
3,520✔
1675
    typeBytes = CHAR_BYTES;
584✔
1676
  }
1677

1678
  if (IS_VAR_DATA_TYPE(value->type)) {
3,520!
1679
    data = taosMemoryCalloc(1, typeBytes + VARSTR_HEADER_SIZE + value->nData);
430!
1680
    if (data == NULL) {
430!
UNCOV
1681
      return NULL;
×
1682
    }
1683

1684
    if (isJson) {
430✔
1685
      *data = value->type;
424✔
1686
    }
1687

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

1694
  return data;
3,520✔
1695
}
1696

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

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

1712
  if (isLarge) {
6,487!
UNCOV
1713
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
×
1714
  } else {
1715
    p = (uint8_t *)&pTag->idx[pTag->nTag];
6,487✔
1716
  }
1717

1718
  pTagVal->type = TSDB_DATA_TYPE_NULL;
6,487✔
1719
  pTagVal->pData = NULL;
6,487✔
1720
  pTagVal->nData = 0;
6,487✔
1721
  while (lidx <= ridx) {
6,708!
1722
    midx = (lidx + ridx) / 2;
6,708✔
1723
    if (isLarge) {
6,708!
UNCOV
1724
      offset = ((int16_t *)pTag->idx)[midx];
×
1725
    } else {
1726
      offset = pTag->idx[midx];
6,708✔
1727
    }
1728

1729
    int32_t nt = tGetTagVal(p + offset, &tv, isJson);
6,708✔
1730
    if (isJson) {
6,708✔
1731
      c = tTagValJsonCmprFn(pTagVal, &tv);
744✔
1732
    } else {
1733
      c = tTagValCmprFn(pTagVal, &tv);
5,964✔
1734
    }
1735

1736
    if (c < 0) {
6,708✔
1737
      ridx = midx - 1;
61✔
1738
    } else if (c > 0) {
6,647✔
1739
      lidx = midx + 1;
160✔
1740
    } else {
1741
      (void)memcpy(pTagVal, &tv, sizeof(tv));
6,487✔
1742
      return true;
6,487✔
1743
    }
1744
  }
UNCOV
1745
  return false;
×
1746
}
1747

1748
int32_t tEncodeTag(SEncoder *pEncoder, const STag *pTag) {
8,661✔
1749
  return tEncodeBinary(pEncoder, (const uint8_t *)pTag, pTag->len);
17,322✔
1750
}
1751

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

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

1761
  if (isLarge) {
4!
UNCOV
1762
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
×
1763
  } else {
1764
    p = (uint8_t *)&pTag->idx[pTag->nTag];
4✔
1765
  }
1766

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

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

1786
  return code;
4✔
1787

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

1792
// STSchema ========================================
1793
STSchema *tBuildTSchema(SSchema *aSchema, int32_t numOfCols, int32_t version) {
7,522,452✔
1794
  STSchema *pTSchema = taosMemoryCalloc(1, sizeof(STSchema) + sizeof(STColumn) * numOfCols);
7,522,452!
1795
  if (pTSchema == NULL) {
7,548,673!
1796
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
1797
    return NULL;
×
1798
  }
1799

1800
  pTSchema->numOfCols = numOfCols;
7,548,673✔
1801
  pTSchema->version = version;
7,548,673✔
1802

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

1818
  // other columns
1819
  for (int32_t iCol = 1; iCol < numOfCols; iCol++) {
30,334,424✔
1820
    SSchema  *pSchema = &aSchema[iCol];
22,785,751✔
1821
    STColumn *pTColumn = &pTSchema->columns[iCol];
22,785,751✔
1822

1823
    pTColumn->colId = pSchema->colId;
22,785,751✔
1824
    pTColumn->type = pSchema->type;
22,785,751✔
1825
    pTColumn->flags = pSchema->flags;
22,785,751✔
1826
    pTColumn->offset = pTSchema->flen;
22,785,751✔
1827

1828
    if (IS_VAR_DATA_TYPE(pSchema->type)) {
22,785,751!
1829
      pTColumn->bytes = pSchema->bytes;
56,883✔
1830
      pTSchema->tlen += (TYPE_BYTES[pSchema->type] + pSchema->bytes);  // todo: remove
56,883✔
1831
    } else {
1832
      pTColumn->bytes = TYPE_BYTES[pSchema->type];
22,728,868✔
1833
      pTSchema->tlen += TYPE_BYTES[pSchema->type];  // todo: remove
22,728,868✔
1834
    }
1835

1836
    pTSchema->flen += TYPE_BYTES[pTColumn->type];
22,785,751✔
1837
  }
1838

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

1843
  return pTSchema;
7,548,673✔
1844
}
1845

UNCOV
1846
static int32_t tTColumnCompare(const void *p1, const void *p2) {
×
UNCOV
1847
  if (((STColumn *)p1)->colId < ((STColumn *)p2)->colId) {
×
UNCOV
1848
    return -1;
×
UNCOV
1849
  } else if (((STColumn *)p1)->colId > ((STColumn *)p2)->colId) {
×
UNCOV
1850
    return 1;
×
1851
  }
1852

UNCOV
1853
  return 0;
×
1854
}
1855

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

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

1864
// SColData ========================================
1865
void tColDataDestroy(void *ph) {
54,413✔
1866
  if (ph) {
54,413!
1867
    SColData *pColData = (SColData *)ph;
54,413✔
1868

1869
    tFree(pColData->pBitMap);
54,413!
1870
    tFree(pColData->aOffset);
54,413!
1871
    tFree(pColData->pData);
54,413!
1872
  }
1873
}
54,414✔
1874

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

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

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

1896
  tColDataClear(pColData);
200✔
1897
}
200✔
1898

1899
static FORCE_INLINE int32_t tColDataPutValue(SColData *pColData, uint8_t *pData, uint32_t nData) {
1900
  int32_t code = 0;
174,114,688✔
1901

1902
  if (IS_VAR_DATA_TYPE(pColData->type)) {
174,101,430!
1903
    code = tRealloc((uint8_t **)(&pColData->aOffset), ((int64_t)(pColData->nVal + 1)) << 2);
29,783!
1904
    if (code) goto _exit;
34,533!
1905
    pColData->aOffset[pColData->nVal] = pColData->nData;
34,533✔
1906

1907
    if (nData) {
34,533!
1908
      code = tRealloc(&pColData->pData, pColData->nData + nData);
34,419!
1909
      if (code) goto _exit;
34,740!
1910
      (void)memcpy(pColData->pData + pColData->nData, pData, nData);
34,740✔
1911
      pColData->nData += nData;
34,740✔
1912
    }
1913
  } else {
1914
    if (!(pColData->nData == tDataTypes[pColData->type].bytes * pColData->nVal)) {
174,084,905!
UNCOV
1915
      return TSDB_CODE_INVALID_PARA;
×
1916
    }
1917
    code = tRealloc(&pColData->pData, pColData->nData + tDataTypes[pColData->type].bytes);
174,084,905!
1918
    if (code) goto _exit;
174,080,941!
1919
    if (pData) {
174,080,941!
1920
      (void)memcpy(pColData->pData + pColData->nData, pData, TYPE_BYTES[pColData->type]);
170,521,665✔
1921
    } else {
1922
      memset(pColData->pData + pColData->nData, 0, TYPE_BYTES[pColData->type]);
3,559,276✔
1923
    }
1924
    pColData->nData += tDataTypes[pColData->type].bytes;
174,080,941✔
1925
  }
1926
  pColData->nVal++;
174,115,795✔
1927

1928
_exit:
174,115,795✔
1929
  return code;
174,115,795✔
1930
}
1931
static FORCE_INLINE int32_t tColDataAppendValue00(SColData *pColData, uint8_t *pData, uint32_t nData) {
53,104✔
1932
  pColData->flag = HAS_VALUE;
53,334✔
1933
  pColData->numOfValue++;
53,104✔
1934
  return tColDataPutValue(pColData, pData, nData);
53,332✔
1935
}
UNCOV
1936
static FORCE_INLINE int32_t tColDataAppendValue01(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
1937
  pColData->flag = HAS_NONE;
×
UNCOV
1938
  pColData->numOfNone++;
×
UNCOV
1939
  pColData->nVal++;
×
UNCOV
1940
  return 0;
×
1941
}
1942
static FORCE_INLINE int32_t tColDataAppendValue02(SColData *pColData, uint8_t *pData, uint32_t nData) {
1,123✔
1943
  pColData->flag = HAS_NULL;
1,310✔
1944
  pColData->numOfNull++;
1,310✔
1945
  pColData->nVal++;
1,310✔
1946
  return 0;
1,123✔
1947
}
UNCOV
1948
static FORCE_INLINE int32_t tColDataAppendValue10(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
1949
  int32_t code = 0;
×
1950

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

UNCOV
1955
  memset(pColData->pBitMap, 0, nBit);
×
UNCOV
1956
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
×
1957

UNCOV
1958
  pColData->flag |= HAS_VALUE;
×
UNCOV
1959
  pColData->numOfValue++;
×
1960

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

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

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

UNCOV
1989
  memset(pColData->pBitMap, 0, nBit);
×
UNCOV
1990
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
×
1991

UNCOV
1992
  pColData->flag |= HAS_NULL;
×
UNCOV
1993
  pColData->numOfNull++;
×
UNCOV
1994
  pColData->nVal++;
×
1995

UNCOV
1996
  return code;
×
1997
}
1998
static FORCE_INLINE int32_t tColDataAppendValue20(SColData *pColData, uint8_t *pData, uint32_t nData) {
1,047✔
1999
  int32_t code = 0;
1,060✔
2000

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

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

2008
  pColData->flag |= HAS_VALUE;
1,060✔
2009
  pColData->numOfValue++;
1,060✔
2010

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

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

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

UNCOV
2034
  memset(pColData->pBitMap, 255, nBit);
×
UNCOV
2035
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
×
2036

UNCOV
2037
  pColData->flag |= HAS_NONE;
×
UNCOV
2038
  pColData->numOfNone++;
×
UNCOV
2039
  pColData->nVal++;
×
2040

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

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

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

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

2063
  tFree(pColData->pBitMap);
×
UNCOV
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;
×
UNCOV
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;
×
UNCOV
2076
      memset(pColData->pData, 0, pColData->nData);
×
2077
    }
2078
  }
2079

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

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

UNCOV
2088
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
×
UNCOV
2089
  pColData->numOfNone++;
×
UNCOV
2090
  pColData->nVal++;
×
2091

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

UNCOV
2097
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
×
UNCOV
2098
  if (code) return code;
×
2099

UNCOV
2100
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
×
UNCOV
2101
  pColData->numOfNull++;
×
UNCOV
2102
  pColData->nVal++;
×
2103

UNCOV
2104
  return code;
×
2105
}
2106
static FORCE_INLINE int32_t tColDataAppendValue40(SColData *pColData, uint8_t *pData, uint32_t nData) {
99,083,792✔
2107
  pColData->numOfValue++;
99,083,792✔
2108
  return tColDataPutValue(pColData, pData, nData);
99,087,464✔
2109
}
UNCOV
2110
static FORCE_INLINE int32_t tColDataAppendValue41(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
2111
  int32_t code = 0;
×
2112

UNCOV
2113
  pColData->flag |= HAS_NONE;
×
UNCOV
2114
  pColData->numOfNone++;
×
2115

UNCOV
2116
  int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
×
UNCOV
2117
  code = tRealloc(&pColData->pBitMap, nBit);
×
UNCOV
2118
  if (code) return code;
×
2119

UNCOV
2120
  memset(pColData->pBitMap, 255, nBit);
×
UNCOV
2121
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
×
2122

UNCOV
2123
  return tColDataPutValue(pColData, NULL, 0);
×
2124
}
2125
static FORCE_INLINE int32_t tColDataAppendValue42(SColData *pColData, uint8_t *pData, uint32_t nData) {
19,883✔
2126
  int32_t code = 0;
22,454✔
2127

2128
  pColData->flag |= HAS_NULL;
22,454✔
2129
  pColData->numOfNull++;
22,454✔
2130

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

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

2138
  return tColDataPutValue(pColData, NULL, 0);
22,454✔
2139
}
UNCOV
2140
static FORCE_INLINE int32_t tColDataAppendValue50(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
2141
  int32_t code = 0;
×
2142

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

UNCOV
2146
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
×
UNCOV
2147
  pColData->numOfValue++;
×
2148

UNCOV
2149
  return tColDataPutValue(pColData, pData, nData);
×
2150
}
UNCOV
2151
static FORCE_INLINE int32_t tColDataAppendValue51(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
2152
  int32_t code = 0;
×
2153

UNCOV
2154
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
×
UNCOV
2155
  if (code) return code;
×
2156

UNCOV
2157
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
×
UNCOV
2158
  pColData->numOfNone++;
×
2159

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

UNCOV
2165
  pColData->flag |= HAS_NULL;
×
UNCOV
2166
  pColData->numOfNull++;
×
2167

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

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

UNCOV
2177
  tFree(pColData->pBitMap);
×
UNCOV
2178
  pColData->pBitMap = pBitMap;
×
2179

UNCOV
2180
  return tColDataPutValue(pColData, NULL, 0);
×
2181
}
2182
static FORCE_INLINE int32_t tColDataAppendValue60(SColData *pColData, uint8_t *pData, uint32_t nData) {
59,990,856✔
2183
  int32_t code = 0;
60,363,250✔
2184

2185
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
60,363,250!
2186
  if (code) return code;
60,363,250!
2187
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 1);
60,363,250!
2188
  pColData->numOfValue++;
60,363,250!
2189

2190
  return tColDataPutValue(pColData, pData, nData);
60,363,250✔
2191
}
UNCOV
2192
static FORCE_INLINE int32_t tColDataAppendValue61(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
2193
  int32_t code = 0;
×
2194

UNCOV
2195
  pColData->flag |= HAS_NONE;
×
UNCOV
2196
  pColData->numOfNone++;
×
2197

UNCOV
2198
  uint8_t *pBitMap = NULL;
×
UNCOV
2199
  code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
×
UNCOV
2200
  if (code) return code;
×
2201

UNCOV
2202
  for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
2203
    SET_BIT2_EX(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal) ? 2 : 1);
×
2204
  }
UNCOV
2205
  SET_BIT2_EX(pBitMap, pColData->nVal, 0);
×
2206

UNCOV
2207
  tFree(pColData->pBitMap);
×
UNCOV
2208
  pColData->pBitMap = pBitMap;
×
2209

UNCOV
2210
  return tColDataPutValue(pColData, NULL, 0);
×
2211
}
2212
static FORCE_INLINE int32_t tColDataAppendValue62(SColData *pColData, uint8_t *pData, uint32_t nData) {
3,161,222✔
2213
  int32_t code = 0;
3,533,478✔
2214

2215
  code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
3,533,478!
2216
  if (code) return code;
3,533,478!
2217
  SET_BIT1_EX(pColData->pBitMap, pColData->nVal, 0);
3,533,478!
2218
  pColData->numOfNull++;
3,533,478!
2219

2220
  return tColDataPutValue(pColData, NULL, 0);
3,533,478✔
2221
}
2222
static FORCE_INLINE int32_t tColDataAppendValue70(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
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);
×
UNCOV
2228
  pColData->numOfValue++;
×
2229

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

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

UNCOV
2240
  return tColDataPutValue(pColData, NULL, 0);
×
2241
}
2242
static FORCE_INLINE int32_t tColDataAppendValue72(SColData *pColData, uint8_t *pData, uint32_t nData) {
×
UNCOV
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);
×
UNCOV
2248
  pColData->numOfNull++;
×
2249

UNCOV
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) {
108,174,092✔
2265
  if (!(pColData->cid == pColVal->cid && pColData->type == pColVal->value.type)) {
108,174,092!
UNCOV
2266
    return TSDB_CODE_INVALID_PARA;
×
2267
  }
2268
  return tColDataAppendValueImpl[pColData->flag][pColVal->flag](
108,179,880✔
2269
      pColData, VALUE_GET_DATUM(&pColVal->value, pColData->type), pColVal->value.nData);
108,179,880!
2270
}
2271

UNCOV
2272
static FORCE_INLINE int32_t tColDataUpdateValue10(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2273
  pColData->numOfNone--;
×
UNCOV
2274
  pColData->nVal--;
×
2275
  if (pColData->numOfNone) {
×
UNCOV
2276
    return tColDataAppendValue10(pColData, pData, nData);
×
2277
  } else {
UNCOV
2278
    pColData->flag = 0;
×
UNCOV
2279
    return tColDataAppendValue00(pColData, pData, nData);
×
2280
  }
2281
}
UNCOV
2282
static FORCE_INLINE int32_t tColDataUpdateValue12(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2283
  pColData->numOfNone--;
×
UNCOV
2284
  pColData->nVal--;
×
2285
  if (pColData->numOfNone) {
×
UNCOV
2286
    return tColDataAppendValue12(pColData, pData, nData);
×
2287
  } else {
UNCOV
2288
    pColData->flag = 0;
×
UNCOV
2289
    return tColDataAppendValue02(pColData, pData, nData);
×
2290
  }
2291
  return 0;
2292
}
2293
static FORCE_INLINE int32_t tColDataUpdateValue20(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
236✔
2294
  if (forward) {
236!
2295
    pColData->numOfNull--;
236✔
2296
    pColData->nVal--;
236✔
2297
    if (pColData->numOfNull) {
236✔
2298
      return tColDataAppendValue20(pColData, pData, nData);
6✔
2299
    } else {
2300
      pColData->flag = 0;
230!
2301
      return tColDataAppendValue00(pColData, pData, nData);
230✔
2302
    }
2303
  }
UNCOV
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) {
×
UNCOV
2311
      return tColDataAppendValue30(pColData, pData, nData);
×
2312
    } else {
2313
      pColData->flag = HAS_NULL;
×
UNCOV
2314
      return tColDataAppendValue20(pColData, pData, nData);
×
2315
    }
2316
  } else if (forward) {  // NULL ==> VALUE
×
2317
    pColData->numOfNull--;
×
2318
    pColData->nVal--;
×
2319
    if (pColData->numOfNull) {
×
UNCOV
2320
      return tColDataAppendValue30(pColData, pData, nData);
×
2321
    } else {
2322
      pColData->flag = HAS_NONE;
×
UNCOV
2323
      return tColDataAppendValue10(pColData, pData, nData);
×
2324
    }
2325
  }
UNCOV
2326
  return 0;
×
2327
}
UNCOV
2328
static FORCE_INLINE int32_t tColDataUpdateValue32(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2329
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> NULL
×
UNCOV
2330
    pColData->numOfNone--;
×
UNCOV
2331
    pColData->numOfNull++;
×
2332
    if (pColData->numOfNone) {
×
UNCOV
2333
      SET_BIT1(pColData->pBitMap, pColData->nVal - 1, 1);
×
2334
    } else {
UNCOV
2335
      pColData->flag = HAS_NULL;
×
2336
    }
2337
  }
UNCOV
2338
  return 0;
×
2339
}
2340
static FORCE_INLINE int32_t tColDataUpdateValue40(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
3,994,792✔
2341
  if (forward) {  // VALUE ==> VALUE
3,994,792!
2342
    pColData->nVal--;
3,994,792✔
2343
    if (IS_VAR_DATA_TYPE(pColData->type)) {
3,994,792!
UNCOV
2344
      pColData->nData = pColData->aOffset[pColData->nVal];
×
2345
    } else {
2346
      pColData->nData -= TYPE_BYTES[pColData->type];
3,994,792✔
2347
    }
2348
    return tColDataPutValue(pColData, pData, nData);
3,994,792✔
2349
  }
UNCOV
2350
  return 0;
×
2351
}
2352
static FORCE_INLINE int32_t tColDataUpdateValue42(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
2,758✔
2353
  if (forward) {  // VALUE ==> NULL
2,758!
2354
    pColData->numOfValue--;
2,758✔
2355
    pColData->nVal--;
2,758✔
2356
    if (pColData->numOfValue) {
2,758✔
2357
      if (IS_VAR_DATA_TYPE(pColData->type)) {
2,571!
UNCOV
2358
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2359
      } else {
2360
        pColData->nData -= TYPE_BYTES[pColData->type];
2,571✔
2361
      }
2362
      return tColDataAppendValue42(pColData, pData, nData);
2,571✔
2363
    } else {
2364
      pColData->flag = 0;
187✔
2365
      pColData->nData = 0;
187✔
2366
      return tColDataAppendValue02(pColData, pData, nData);
187✔
2367
    }
2368
  }
UNCOV
2369
  return 0;
×
2370
}
UNCOV
2371
static FORCE_INLINE int32_t tColDataUpdateValue50(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2372
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> VALUE
×
UNCOV
2373
    pColData->numOfNone--;
×
UNCOV
2374
    pColData->nVal--;
×
UNCOV
2375
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2376
      pColData->nData -= TYPE_BYTES[pColData->type];
×
2377
    }
UNCOV
2378
    if (pColData->numOfNone) {
×
UNCOV
2379
      return tColDataAppendValue50(pColData, pData, nData);
×
2380
    } else {
UNCOV
2381
      pColData->flag = HAS_VALUE;
×
UNCOV
2382
      return tColDataAppendValue40(pColData, pData, nData);
×
2383
    }
UNCOV
2384
  } else if (forward) {  // VALUE ==> VALUE
×
UNCOV
2385
    pColData->nVal--;
×
UNCOV
2386
    if (IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2387
      pColData->nData = pColData->aOffset[pColData->nVal];
×
2388
    } else {
UNCOV
2389
      pColData->nData -= TYPE_BYTES[pColData->type];
×
2390
    }
UNCOV
2391
    return tColDataPutValue(pColData, pData, nData);
×
2392
  }
UNCOV
2393
  return 0;
×
2394
}
UNCOV
2395
static FORCE_INLINE int32_t tColDataUpdateValue52(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2396
  if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NONE ==> NULL
×
UNCOV
2397
    pColData->numOfNone--;
×
UNCOV
2398
    pColData->nVal--;
×
UNCOV
2399
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2400
      pColData->nData -= TYPE_BYTES[pColData->type];
×
2401
    }
2402
    if (pColData->numOfNone) {
×
UNCOV
2403
      return tColDataAppendValue52(pColData, pData, nData);
×
2404
    } else {
UNCOV
2405
      pColData->flag = HAS_VALUE;
×
UNCOV
2406
      return tColDataAppendValue42(pColData, pData, nData);
×
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)) {
×
UNCOV
2413
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2414
      } else {
UNCOV
2415
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2416
      }
UNCOV
2417
      return tColDataAppendValue52(pColData, pData, nData);
×
2418
    } else {
2419
      pColData->flag = HAS_NONE;
×
2420
      pColData->nData = 0;
×
UNCOV
2421
      return tColDataAppendValue12(pColData, pData, nData);
×
2422
    }
2423
  }
UNCOV
2424
  return 0;
×
2425
}
2426
static FORCE_INLINE int32_t tColDataUpdateValue60(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
7,434,922✔
2427
  if (forward) {
7,434,922!
2428
    if (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 0) {  // NULL ==> VALUE
7,434,922✔
2429
      pColData->numOfNull--;
374,957✔
2430
      pColData->nVal--;
374,957✔
2431
      if (!IS_VAR_DATA_TYPE(pColData->type)) {
374,957!
2432
        pColData->nData -= TYPE_BYTES[pColData->type];
374,957✔
2433
      }
2434
      if (pColData->numOfNull) {
374,957✔
2435
        return tColDataAppendValue60(pColData, pData, nData);
372,394✔
2436
      } else {
2437
        pColData->flag = HAS_VALUE;
2,563!
2438
        return tColDataAppendValue40(pColData, pData, nData);
2,563✔
2439
      }
2440
    } else {  // VALUE ==> VALUE
2441
      pColData->nVal--;
7,059,965✔
2442
      if (IS_VAR_DATA_TYPE(pColData->type)) {
7,059,965!
UNCOV
2443
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2444
      } else {
2445
        pColData->nData -= TYPE_BYTES[pColData->type];
7,059,965✔
2446
      }
2447
      return tColDataPutValue(pColData, pData, nData);
7,059,965✔
2448
    }
2449
  }
UNCOV
2450
  return 0;
×
2451
}
2452
static FORCE_INLINE int32_t tColDataUpdateValue62(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
392,141✔
2453
  if (forward && (GET_BIT1(pColData->pBitMap, pColData->nVal - 1) == 1)) {  // VALUE ==> NULL
392,141!
2454
    pColData->numOfValue--;
372,263✔
2455
    pColData->nVal--;
372,263✔
2456
    if (pColData->numOfValue) {
372,263✔
2457
      if (IS_VAR_DATA_TYPE(pColData->type)) {
372,256!
UNCOV
2458
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2459
      } else {
2460
        pColData->nData -= TYPE_BYTES[pColData->type];
372,256✔
2461
      }
2462
      return tColDataAppendValue62(pColData, pData, nData);
372,256✔
2463
    } else {
2464
      pColData->flag = HAS_NULL;
7✔
2465
      pColData->nData = 0;
7!
2466
      return tColDataAppendValue20(pColData, pData, nData);
7✔
2467
    }
2468
  }
2469
  return 0;
19,878✔
2470
}
UNCOV
2471
static FORCE_INLINE int32_t tColDataUpdateValue70(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2472
  int32_t code = 0;
×
2473

UNCOV
2474
  uint8_t bv = GET_BIT2(pColData->pBitMap, pColData->nVal - 1);
×
UNCOV
2475
  if (bv == 0) {  // NONE ==> VALUE
×
UNCOV
2476
    pColData->numOfNone--;
×
UNCOV
2477
    pColData->nVal--;
×
UNCOV
2478
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2479
      pColData->nData -= TYPE_BYTES[pColData->type];
×
2480
    }
2481
    if (pColData->numOfNone) {
×
UNCOV
2482
      return tColDataAppendValue70(pColData, pData, nData);
×
2483
    } else {
UNCOV
2484
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
2485
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) - 1);
×
2486
      }
UNCOV
2487
      pColData->flag = (HAS_VALUE | HAS_NULL);
×
UNCOV
2488
      return tColDataAppendValue60(pColData, pData, nData);
×
2489
    }
UNCOV
2490
  } else if (bv == 1) {  // NULL ==> VALUE
×
UNCOV
2491
    if (forward) {
×
UNCOV
2492
      pColData->numOfNull--;
×
UNCOV
2493
      pColData->nVal--;
×
UNCOV
2494
      if (!IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2495
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2496
      }
2497
      if (pColData->numOfNull) {
×
UNCOV
2498
        return tColDataAppendValue70(pColData, pData, nData);
×
2499
      } else {
UNCOV
2500
        for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
2501
          SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) ? 1 : 0);
×
2502
        }
UNCOV
2503
        pColData->flag = (HAS_VALUE | HAS_NONE);
×
UNCOV
2504
        return tColDataAppendValue50(pColData, pData, nData);
×
2505
      }
2506
    }
2507
  } else if (bv == 2) {  // VALUE ==> VALUE
×
2508
    if (forward) {
×
2509
      pColData->nVal--;
×
2510
      if (IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2511
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2512
      } else {
UNCOV
2513
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2514
      }
UNCOV
2515
      return tColDataPutValue(pColData, pData, nData);
×
2516
    }
2517
  } else {
UNCOV
2518
    return TSDB_CODE_INVALID_PARA;
×
2519
  }
UNCOV
2520
  return 0;
×
2521
}
UNCOV
2522
static int32_t tColDataUpdateValue72(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
×
UNCOV
2523
  uint8_t bv = GET_BIT2(pColData->pBitMap, pColData->nVal - 1);
×
UNCOV
2524
  if (bv == 0) {  // NONE ==> NULL
×
UNCOV
2525
    pColData->numOfNone--;
×
UNCOV
2526
    pColData->nVal--;
×
UNCOV
2527
    if (!IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2528
      pColData->nData -= TYPE_BYTES[pColData->type];
×
2529
    }
2530
    if (pColData->numOfNone) {
×
UNCOV
2531
      return tColDataAppendValue72(pColData, pData, nData);
×
2532
    } else {
UNCOV
2533
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
2534
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal) - 1);
×
2535
      }
UNCOV
2536
      pColData->flag = (HAS_VALUE | HAS_NULL);
×
UNCOV
2537
      return tColDataAppendValue62(pColData, pData, nData);
×
2538
    }
2539
  } else if (bv == 2 && forward) {  // VALUE ==> NULL
×
2540
    pColData->numOfValue--;
×
2541
    pColData->nVal--;
×
2542
    if (pColData->numOfValue) {
×
2543
      if (IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
2544
        pColData->nData = pColData->aOffset[pColData->nVal];
×
2545
      } else {
UNCOV
2546
        pColData->nData -= TYPE_BYTES[pColData->type];
×
2547
      }
UNCOV
2548
      return tColDataAppendValue72(pColData, pData, nData);
×
2549
    } else {
2550
      for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
2551
        SET_BIT1(pColData->pBitMap, iVal, GET_BIT2(pColData->pBitMap, iVal));
×
2552
      }
2553
      pColData->flag = (HAS_NULL | HAS_NONE);
×
2554
      pColData->nData = 0;
×
UNCOV
2555
      return tColDataAppendValue32(pColData, pData, nData);
×
2556
    }
2557
  }
UNCOV
2558
  return 0;
×
2559
}
2560
static FORCE_INLINE int32_t tColDataUpdateNothing(SColData *pColData, uint8_t *pData, uint32_t nData, bool forward) {
8✔
2561
  return 0;
8✔
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) {
11,824,857✔
2576
  if (!(pColData->cid == pColVal->cid && pColData->type == pColVal->value.type)) return TSDB_CODE_INVALID_PARA;
11,824,857!
2577
  if (!(pColData->nVal > 0)) return TSDB_CODE_INVALID_PARA;
11,824,857!
2578

2579
  if (tColDataUpdateValueImpl[pColData->flag][pColVal->flag] == NULL) return 0;
11,824,857!
2580

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

UNCOV
2585
static FORCE_INLINE void tColDataGetValue1(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NONE
×
UNCOV
2586
  *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
×
UNCOV
2587
}
×
2588
static FORCE_INLINE void tColDataGetValue2(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NULL
450,000✔
2589
  *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
450,000✔
2590
}
450,000✔
UNCOV
2591
static FORCE_INLINE void tColDataGetValue3(SColData *pColData, int32_t iVal,
×
2592
                                           SColVal *pColVal) {  // HAS_NULL|HAS_NONE
UNCOV
2593
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
×
UNCOV
2594
    case 0:
×
UNCOV
2595
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
×
UNCOV
2596
      break;
×
UNCOV
2597
    case 1:
×
UNCOV
2598
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
×
2599
      break;
×
2600
    default:
×
UNCOV
2601
      break;
×
2602
  }
UNCOV
2603
}
×
2604
static FORCE_INLINE void tColDataGetValue4(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_VALUE
227,943,668✔
2605
  SValue value = {.type = pColData->type};
277,925,875✔
2606
  if (IS_VAR_DATA_TYPE(pColData->type)) {
277,925,875!
UNCOV
2607
    if (iVal + 1 < pColData->nVal) {
×
2608
      value.nData = pColData->aOffset[iVal + 1] - pColData->aOffset[iVal];
19,614✔
2609
    } else {
UNCOV
2610
      value.nData = pColData->nData - pColData->aOffset[iVal];
×
2611
    }
UNCOV
2612
    value.pData = pColData->pData + pColData->aOffset[iVal];
×
2613
  } else {
2614
    valueSetDatum(&value, pColData->type, pColData->pData + tDataTypes[pColData->type].bytes * iVal,
277,955,824✔
2615
                  tDataTypes[pColData->type].bytes);
277,955,824✔
2616
  }
2617
  *pColVal = COL_VAL_VALUE(pColData->cid, value);
277,895,264✔
2618
}
49,982,207✔
UNCOV
2619
static FORCE_INLINE void tColDataGetValue5(SColData *pColData, int32_t iVal,
×
2620
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NONE
UNCOV
2621
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
×
UNCOV
2622
    case 0:
×
UNCOV
2623
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
×
UNCOV
2624
      break;
×
UNCOV
2625
    case 1:
×
2626
      tColDataGetValue4(pColData, iVal, pColVal);
2627
      break;
×
2628
    default:
×
UNCOV
2629
      break;
×
2630
  }
UNCOV
2631
}
×
2632
static FORCE_INLINE void tColDataGetValue6(SColData *pColData, int32_t iVal,
52,616,096✔
2633
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL
2634
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
52,616,096!
2635
    case 0:
2,633,889✔
2636
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
2,633,889✔
2637
      break;
2,633,889✔
2638
    case 1:
49,982,207!
2639
      tColDataGetValue4(pColData, iVal, pColVal);
2640
      break;
49,982,207✔
2641
    default:
×
UNCOV
2642
      break;
×
2643
  }
2644
}
52,616,096✔
UNCOV
2645
static FORCE_INLINE void tColDataGetValue7(SColData *pColData, int32_t iVal,
×
2646
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL|HAS_NONE
UNCOV
2647
  switch (GET_BIT2(pColData->pBitMap, iVal)) {
×
UNCOV
2648
    case 0:
×
UNCOV
2649
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
×
UNCOV
2650
      break;
×
UNCOV
2651
    case 1:
×
UNCOV
2652
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
×
UNCOV
2653
      break;
×
UNCOV
2654
    case 2:
×
2655
      tColDataGetValue4(pColData, iVal, pColVal);
2656
      break;
×
2657
    default:
×
UNCOV
2658
      break;
×
2659
  }
UNCOV
2660
}
×
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) {
280,759,691✔
2672
  if (iVal < 0 || iVal >= pColData->nVal ||
280,759,691!
2673
      (pColData->flag <= 0 || pColData->flag >= sizeof(tColDataGetValueImpl) / POINTER_BYTES)) {
280,822,972!
UNCOV
2674
    return TSDB_CODE_INVALID_PARA;
×
2675
  }
2676
  tColDataGetValueImpl[pColData->flag](pColData, iVal, pColVal);
280,761,425✔
2677
  return TSDB_CODE_SUCCESS;
280,736,894✔
2678
}
2679

2680
uint8_t tColDataGetBitValue(const SColData *pColData, int32_t iVal) {
59,668,954✔
2681
  switch (pColData->flag) {
59,668,954!
2682
    case HAS_NONE:
×
2683
      return 0;
×
2684
    case HAS_NULL:
×
UNCOV
2685
      return 1;
×
UNCOV
2686
    case (HAS_NULL | HAS_NONE):
×
2687
      return GET_BIT1(pColData->pBitMap, iVal);
×
2688
    case HAS_VALUE:
×
UNCOV
2689
      return 2;
×
UNCOV
2690
    case (HAS_VALUE | HAS_NONE):
×
UNCOV
2691
      return (GET_BIT1(pColData->pBitMap, iVal)) ? 2 : 0;
×
2692
    case (HAS_VALUE | HAS_NULL):
59,668,954✔
2693
      return GET_BIT1(pColData->pBitMap, iVal) + 1;
59,668,954✔
UNCOV
2694
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
2695
      return GET_BIT2(pColData->pBitMap, iVal);
×
2696
    default:
×
UNCOV
2697
      return 0;
×
2698
  }
2699
}
2700

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

UNCOV
2704
  *pColData = *pColDataFrom;
×
2705

2706
  // bitmap
UNCOV
2707
  switch (pColData->flag) {
×
UNCOV
2708
    case (HAS_NULL | HAS_NONE):
×
2709
    case (HAS_VALUE | HAS_NONE):
2710
    case (HAS_VALUE | HAS_NULL):
UNCOV
2711
      pColData->pBitMap = xMalloc(arg, BIT1_SIZE(pColData->nVal));
×
2712
      if (pColData->pBitMap == NULL) {
×
2713
        code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
2714
        goto _exit;
×
2715
      }
UNCOV
2716
      (void)memcpy(pColData->pBitMap, pColDataFrom->pBitMap, BIT1_SIZE(pColData->nVal));
×
2717
      break;
×
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;
×
UNCOV
2722
        goto _exit;
×
2723
      }
2724
      (void)memcpy(pColData->pBitMap, pColDataFrom->pBitMap, BIT2_SIZE(pColData->nVal));
×
UNCOV
2725
      break;
×
UNCOV
2726
    default:
×
UNCOV
2727
      pColData->pBitMap = NULL;
×
UNCOV
2728
      break;
×
2729
  }
2730

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

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

UNCOV
2751
    (void)memcpy(pColData->pData, pColDataFrom->pData, pColData->nData);
×
2752
  } else {
UNCOV
2753
    pColData->pData = NULL;
×
2754
  }
2755

UNCOV
2756
_exit:
×
UNCOV
2757
  return code;
×
2758
}
2759

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

2764
  if (!(colData->nVal > 0)) {
35,816!
UNCOV
2765
    return TSDB_CODE_INVALID_PARA;
×
2766
  }
2767

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

2777
  if (colData->flag == HAS_NONE || colData->flag == HAS_NULL) {
35,816!
2778
    return 0;
27✔
2779
  }
2780

2781
  tBufferInit(&local);
2782
  if (assist == NULL) {
35,789!
UNCOV
2783
    assist = &local;
×
2784
  }
2785

2786
  // bitmap
2787
  if (colData->flag != HAS_VALUE) {
35,789✔
2788
    if (colData->flag == (HAS_NONE | HAS_NULL | HAS_VALUE)) {
13,408!
UNCOV
2789
      info->bitmapOriginalSize = BIT2_SIZE(colData->nVal);
×
2790
    } else {
2791
      info->bitmapOriginalSize = BIT1_SIZE(colData->nVal);
13,408✔
2792
    }
2793

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

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

2806
    info->bitmapCompressedSize = cinfo.compressedSize;
13,408✔
2807
  }
2808

2809
  if (colData->flag == (HAS_NONE | HAS_NULL)) {
35,789!
2810
    tBufferDestroy(&local);
UNCOV
2811
    return 0;
×
2812
  }
2813

2814
  // offset
2815
  if (IS_VAR_DATA_TYPE(colData->type)) {
35,789!
2816
    info->offsetOriginalSize = sizeof(int32_t) * info->numOfData;
41✔
2817

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

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

2830
    info->offsetCompressedSize = cinfo.compressedSize;
42✔
2831
  }
2832

2833
  // data
2834
  if (colData->nData > 0) {
35,790!
2835
    info->dataOriginalSize = colData->nData;
35,790✔
2836

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

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

2849
    info->dataCompressedSize = cinfo.compressedSize;
35,790✔
2850
  }
2851

2852
  tBufferDestroy(&local);
2853
  return 0;
35,790✔
2854
}
2855

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

2861
  tBufferInit(&local);
2862
  if (assist == NULL) {
30,916!
UNCOV
2863
    assist = &local;
×
2864
  }
2865

2866
  tColDataClear(colData);
30,916✔
2867
  colData->cid = info->columnId;
30,918✔
2868
  colData->type = info->dataType;
30,918✔
2869
  colData->cflag = info->columnFlag;
30,918✔
2870
  colData->nVal = info->numOfData;
30,918✔
2871
  colData->flag = info->flag;
30,918✔
2872

2873
  if (info->flag == HAS_NONE || info->flag == HAS_NULL) {
30,918!
2874
    goto _exit;
135✔
2875
  }
2876

2877
  // bitmap
2878
  if (info->bitmapOriginalSize > 0) {
30,783✔
2879
    SCompressInfo cinfo = {
9,638✔
2880
        .dataType = TSDB_DATA_TYPE_TINYINT,
2881
        .cmprAlg = info->cmprAlg,
9,638✔
2882
        .originalSize = info->bitmapOriginalSize,
9,638✔
2883
        .compressedSize = info->bitmapCompressedSize,
9,638✔
2884
    };
2885

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

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

2898
    data += cinfo.compressedSize;
9,638✔
2899
  }
2900

2901
  if (info->flag == (HAS_NONE | HAS_NULL)) {
30,783!
UNCOV
2902
    goto _exit;
×
2903
  }
2904

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

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

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

2926
    data += cinfo.compressedSize;
11✔
2927
  }
2928

2929
  // data
2930
  if (info->dataOriginalSize > 0) {
30,783✔
2931
    colData->nData = info->dataOriginalSize;
30,782✔
2932

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

2940
    code = tRealloc((uint8_t **)&colData->pData, cinfo.originalSize);
30,782!
2941
    if (code) {
30,782!
2942
      tBufferDestroy(&local);
UNCOV
2943
      return code;
×
2944
    }
2945

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

2952
    data += cinfo.compressedSize;
30,781✔
2953
  }
2954

2955
_exit:
1✔
2956
  switch (colData->flag) {
30,917!
UNCOV
2957
    case HAS_NONE:
×
UNCOV
2958
      colData->numOfNone = colData->nVal;
×
UNCOV
2959
      break;
×
2960
    case HAS_NULL:
135✔
2961
      colData->numOfNull = colData->nVal;
135✔
2962
      break;
135✔
2963
    case HAS_VALUE:
21,143✔
2964
      colData->numOfValue = colData->nVal;
21,143✔
2965
      break;
21,143✔
2966
    default:
9,639✔
2967
      for (int32_t i = 0; i < colData->nVal; i++) {
33,785,735✔
2968
        uint8_t bitValue = tColDataGetBitValue(colData, i);
33,776,096✔
2969
        if (bitValue == 0) {
33,776,096!
UNCOV
2970
          colData->numOfNone++;
×
2971
        } else if (bitValue == 1) {
33,776,096✔
2972
          colData->numOfNull++;
1,690,611✔
2973
        } else {
2974
          colData->numOfValue++;
32,085,485✔
2975
        }
2976
      }
2977
  }
2978
  tBufferDestroy(&local);
2979
  return 0;
30,917✔
2980
}
2981

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

UNCOV
2996
  if (IS_VAR_DATA_TYPE(type)) {  // var-length data type
×
UNCOV
2997
    for (int32_t i = 0; i < nRows; ++i) {
×
UNCOV
2998
      int32_t offset = *((int32_t *)lengthOrbitmap + i);
×
2999
      if (offset == -1) {
×
3000
        if (pColData->cflag & COL_IS_KEY) {
×
3001
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3002
          goto _exit;
×
3003
        }
3004
        if ((code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0))) {
×
UNCOV
3005
          goto _exit;
×
3006
        }
3007
      } else {
3008
        if (varDataTLen(data + offset) > bytes) {
×
UNCOV
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;
×
UNCOV
3012
          goto _exit;
×
3013
        }
UNCOV
3014
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, (uint8_t *)varDataVal(data + offset),
×
UNCOV
3015
                                                                      varDataLen(data + offset));
×
3016
      }
3017
    }
3018
  } else {  // fixed-length data type
UNCOV
3019
    bool allValue = true;
×
UNCOV
3020
    bool allNull = true;
×
UNCOV
3021
    for (int32_t i = 0; i < nRows; ++i) {
×
UNCOV
3022
      if (!colDataIsNull_f(lengthOrbitmap, i)) {
×
UNCOV
3023
        allNull = false;
×
3024
      } else {
UNCOV
3025
        allValue = false;
×
3026
      }
3027
    }
3028
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
×
3029
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3030
      goto _exit;
×
3031
    }
3032

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

UNCOV
3056
_exit:
×
UNCOV
3057
  return code;
×
3058
}
3059

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

3064
  if (!(pBind->num == 1 && pBind->is_null && *pBind->is_null)) {
924!
3065
    if (!(pColData->type == pBind->buffer_type)) {
924!
UNCOV
3066
      return TSDB_CODE_INVALID_PARA;
×
3067
    }
3068
  }
3069

3070
  if (IS_VAR_DATA_TYPE(pColData->type)) {  // var-length data type
924!
3071
    if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
106!
UNCOV
3072
      code = igeos();
×
3073
      if (code) {
×
UNCOV
3074
        return code;
×
3075
      }
3076
    }
3077
    for (int32_t i = 0; i < pBind->num; ++i) {
312✔
3078
      if (pBind->is_null && pBind->is_null[i]) {
206!
3079
        if (pColData->cflag & COL_IS_KEY) {
×
3080
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3081
          goto _exit;
×
3082
        }
UNCOV
3083
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
UNCOV
3084
        if (code) goto _exit;
×
3085
      } else if (pBind->length[i] > buffMaxLen) {
206!
UNCOV
3086
        return TSDB_CODE_PAR_VALUE_TOO_LONG;
×
3087
      } else {
3088
        if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
206!
UNCOV
3089
          code = cgeos((char *)pBind->buffer + pBind->buffer_length * i, (size_t)pBind->length[i]);
×
UNCOV
3090
          if (code) goto _exit;
×
3091
        }
3092
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
206✔
3093
            pColData, (uint8_t *)pBind->buffer + pBind->buffer_length * i, pBind->length[i]);
206✔
3094
      }
3095
    }
3096
  } else {  // fixed-length data type
3097
    bool allValue;
3098
    bool allNull;
3099
    if (pBind->is_null) {
818!
UNCOV
3100
      bool same = (memcmp(pBind->is_null, pBind->is_null + 1, pBind->num - 1) == 0);
×
UNCOV
3101
      allNull = (same && pBind->is_null[0] != 0);
×
UNCOV
3102
      allValue = (same && pBind->is_null[0] == 0);
×
3103
    } else {
3104
      allNull = false;
818✔
3105
      allValue = true;
818✔
3106
    }
3107

3108
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
818!
3109
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3110
      goto _exit;
×
3111
    }
3112

3113
    if (allValue) {
818!
3114
      // optimize (todo)
3115
      for (int32_t i = 0; i < pBind->num; ++i) {
4,054✔
3116
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
3,236✔
3117
            pColData, (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i, pBind->buffer_length);
3,236✔
3118
      }
UNCOV
3119
    } else if (allNull) {
×
3120
      // optimize (todo)
UNCOV
3121
      for (int32_t i = 0; i < pBind->num; ++i) {
×
UNCOV
3122
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
UNCOV
3123
        if (code) goto _exit;
×
3124
      }
3125
    } else {
UNCOV
3126
      for (int32_t i = 0; i < pBind->num; ++i) {
×
UNCOV
3127
        if (pBind->is_null[i]) {
×
UNCOV
3128
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
UNCOV
3129
          if (code) goto _exit;
×
3130
        } else {
UNCOV
3131
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](
×
UNCOV
3132
              pColData, (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i, pBind->buffer_length);
×
3133
        }
3134
      }
3135
    }
3136
  }
3137

UNCOV
3138
_exit:
×
3139
  return code;
924✔
3140
}
3141

UNCOV
3142
int32_t tColDataAddValueByBind2(SColData *pColData, TAOS_STMT2_BIND *pBind, int32_t buffMaxLen, initGeosFn igeos,
×
3143
                                checkWKBGeometryFn cgeos) {
UNCOV
3144
  int32_t code = 0;
×
3145

UNCOV
3146
  if (!(pBind->num == 1 && pBind->is_null && *pBind->is_null)) {
×
3147
    if (!(pColData->type == pBind->buffer_type)) {
×
UNCOV
3148
      return TSDB_CODE_INVALID_PARA;
×
3149
    }
3150
  }
3151

UNCOV
3152
  if (IS_VAR_DATA_TYPE(pColData->type)) {  // var-length data type
×
UNCOV
3153
    if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
×
UNCOV
3154
      code = igeos();
×
3155
      if (code) {
×
UNCOV
3156
        return code;
×
3157
      }
3158
    }
3159

UNCOV
3160
    uint8_t *buf = pBind->buffer;
×
UNCOV
3161
    for (int32_t i = 0; i < pBind->num; ++i) {
×
3162
      if (pBind->is_null && pBind->is_null[i]) {
×
3163
        if (pColData->cflag & COL_IS_KEY) {
×
3164
          code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3165
          goto _exit;
×
3166
        }
3167
        if (pBind->is_null[i] == 1) {
×
3168
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NULL](pColData, NULL, 0);
×
UNCOV
3169
          if (code) goto _exit;
×
3170
        } else {
3171
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_NONE](pColData, NULL, 0);
×
UNCOV
3172
          if (code) goto _exit;
×
3173
        }
3174
      } else if (pBind->length[i] > buffMaxLen) {
×
UNCOV
3175
        return TSDB_CODE_PAR_VALUE_TOO_LONG;
×
3176
      } else {
UNCOV
3177
        if (pColData->type == TSDB_DATA_TYPE_GEOMETRY) {
×
UNCOV
3178
          code = cgeos(buf, pBind->length[i]);
×
UNCOV
3179
          if (code) goto _exit;
×
3180
        }
UNCOV
3181
        code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, buf, pBind->length[i]);
×
UNCOV
3182
        buf += pBind->length[i];
×
3183
      }
3184
    }
3185
  } else {  // fixed-length data type
3186
    bool allValue;
3187
    bool allNull;
3188
    bool allNone;
3189
    if (pBind->is_null) {
×
3190
      bool same = (memcmp(pBind->is_null, pBind->is_null + 1, pBind->num - 1) == 0);
×
3191
      allNull = (same && pBind->is_null[0] == 1);
×
3192
      allNone = (same && pBind->is_null[0] > 1);
×
UNCOV
3193
      allValue = (same && pBind->is_null[0] == 0);
×
3194
    } else {
UNCOV
3195
      allNull = false;
×
UNCOV
3196
      allNone = false;
×
UNCOV
3197
      allValue = true;
×
3198
    }
3199

3200
    if ((pColData->cflag & COL_IS_KEY) && !allValue) {
×
3201
      code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3202
      goto _exit;
×
3203
    }
3204

UNCOV
3205
    if (allValue) {
×
3206
      // optimize (todo)
UNCOV
3207
      for (int32_t i = 0; i < pBind->num; ++i) {
×
UNCOV
3208
        uint8_t *val = (uint8_t *)pBind->buffer + TYPE_BYTES[pColData->type] * i;
×
UNCOV
3209
        if (TSDB_DATA_TYPE_BOOL == pColData->type && *val > 1) {
×
UNCOV
3210
          *val = 1;
×
3211
        }
3212

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

UNCOV
3243
          code = tColDataAppendValueImpl[pColData->flag][CV_FLAG_VALUE](pColData, val, TYPE_BYTES[pColData->type]);
×
3244
        }
3245
      }
3246
    }
3247
  }
3248

UNCOV
3249
_exit:
×
UNCOV
3250
  return code;
×
3251
}
3252

3253
/* build rows to `rowArray` from bind
3254
 * `infos` is the bind information array
3255
 * `numOfInfos` is the number of bind information
3256
 * `infoSorted` is whether the bind information is sorted by column id
3257
 * `pTSchema` is the schema of the table
3258
 * `rowArray` is the array to store the rows
3259
 * `pOrdered` is the pointer to store ordered
3260
 * `pDupTs` is the pointer to store duplicateTs
3261
 */
UNCOV
3262
int32_t tRowBuildFromBind2(SBindInfo2 *infos, int32_t numOfInfos, bool infoSorted, const STSchema *pTSchema,
×
3263
                           SArray *rowArray, bool *pOrdered, bool *pDupTs) {
3264
  if (infos == NULL || numOfInfos <= 0 || numOfInfos > pTSchema->numOfCols || pTSchema == NULL || rowArray == NULL) {
×
UNCOV
3265
    return TSDB_CODE_INVALID_PARA;
×
3266
  }
3267

UNCOV
3268
  if (!infoSorted) {
×
UNCOV
3269
    taosqsort_r(infos, numOfInfos, sizeof(SBindInfo2), NULL, tBindInfoCompare);
×
3270
  }
3271

UNCOV
3272
  int32_t code = 0;
×
UNCOV
3273
  int32_t numOfRows = infos[0].bind->num;
×
3274
  SArray *colValArray, *bufArray;
3275
  SColVal colVal;
3276

3277
  if ((colValArray = taosArrayInit(numOfInfos, sizeof(SColVal))) == NULL) {
×
UNCOV
3278
    return terrno;
×
3279
  }
3280
  if ((bufArray = taosArrayInit(numOfInfos, sizeof(uint8_t *))) == NULL) {
×
3281
    taosArrayDestroy(colValArray);
×
UNCOV
3282
    return terrno;
×
3283
  }
UNCOV
3284
  for (int i = 0; i < numOfInfos; ++i) {
×
3285
    if (!taosArrayPush(bufArray, &infos[i].bind->buffer)) {
×
3286
      taosArrayDestroy(colValArray);
×
3287
      taosArrayDestroy(bufArray);
×
UNCOV
3288
      return terrno;
×
3289
    }
3290
  }
3291

3292
  SRowKey rowKey, lastRowKey;
UNCOV
3293
  for (int32_t iRow = 0; iRow < numOfRows; iRow++) {
×
UNCOV
3294
    taosArrayClear(colValArray);
×
3295

UNCOV
3296
    for (int32_t iInfo = 0; iInfo < numOfInfos; iInfo++) {
×
UNCOV
3297
      if (infos[iInfo].bind->is_null && infos[iInfo].bind->is_null[iRow]) {
×
UNCOV
3298
        if (infos[iInfo].bind->is_null[iRow] == 1) {
×
UNCOV
3299
          if(iInfo == 0) {
×
UNCOV
3300
            code = TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
UNCOV
3301
            goto _exit;
×
3302
          }
UNCOV
3303
          colVal = COL_VAL_NULL(infos[iInfo].columnId, infos[iInfo].type);
×
3304
        } else {
UNCOV
3305
          colVal = COL_VAL_NONE(infos[iInfo].columnId, infos[iInfo].type);
×
3306
        }
3307
      } else {
UNCOV
3308
        SValue value = {
×
UNCOV
3309
            .type = infos[iInfo].type,
×
3310
        };
UNCOV
3311
        if (IS_VAR_DATA_TYPE(infos[iInfo].type)) {
×
UNCOV
3312
          int32_t   length = infos[iInfo].bind->length[iRow];
×
UNCOV
3313
          uint8_t **data = &((uint8_t **)TARRAY_DATA(bufArray))[iInfo];
×
UNCOV
3314
          value.nData = length;
×
3315
          if (value.nData > pTSchema->columns[iInfo].bytes - VARSTR_HEADER_SIZE) {
×
3316
            code = TSDB_CODE_INVALID_PARA;
×
UNCOV
3317
            goto _exit;
×
3318
          }
UNCOV
3319
          value.pData = *data;
×
UNCOV
3320
          *data += length;
×
3321
          // value.pData = (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bind->buffer_length * iRow;
3322
        } else {
UNCOV
3323
          uint8_t *val = (uint8_t *)infos[iInfo].bind->buffer + infos[iInfo].bytes * iRow;
×
UNCOV
3324
          if (TSDB_DATA_TYPE_BOOL == value.type && *val > 1) {
×
UNCOV
3325
            *val = 1;
×
3326
          }
UNCOV
3327
          valueSetDatum(&value, infos[iInfo].type, val, infos[iInfo].bytes);
×
3328
        }
UNCOV
3329
        colVal = COL_VAL_VALUE(infos[iInfo].columnId, value);
×
3330
      }
3331
      if (taosArrayPush(colValArray, &colVal) == NULL) {
×
3332
        code = terrno;
×
UNCOV
3333
        goto _exit;
×
3334
      }
3335
    }
3336

3337
    SRow *row;
3338
    if ((code = tRowBuild(colValArray, pTSchema, &row))) {
×
UNCOV
3339
      goto _exit;
×
3340
    }
3341

3342
    if ((taosArrayPush(rowArray, &row)) == NULL) {
×
3343
      code = terrno;
×
UNCOV
3344
      goto _exit;
×
3345
    }
3346

UNCOV
3347
    if (pOrdered && pDupTs) {
×
UNCOV
3348
      tRowGetKey(row, &rowKey);
×
UNCOV
3349
      if (iRow == 0) {
×
UNCOV
3350
        *pOrdered = true;
×
UNCOV
3351
        *pDupTs = false;
×
3352
      } else {
UNCOV
3353
        if (*pOrdered) {
×
UNCOV
3354
          int32_t res = tRowKeyCompare(&rowKey, &lastRowKey);
×
UNCOV
3355
          *pOrdered = (res >= 0);
×
UNCOV
3356
          if (!*pDupTs) {
×
UNCOV
3357
            *pDupTs = (res == 0);
×
3358
          }
3359
        }
UNCOV
3360
        lastRowKey = rowKey;
×
3361
      }
3362
    }
3363
  }
3364
_exit:
×
UNCOV
3365
  taosArrayDestroy(colValArray);
×
UNCOV
3366
  taosArrayDestroy(bufArray);
×
UNCOV
3367
  return code;
×
3368
}
3369

UNCOV
3370
static int32_t tColDataCopyRowCell(SColData *pFromColData, int32_t iFromRow, SColData *pToColData, int32_t iToRow) {
×
UNCOV
3371
  int32_t code = TSDB_CODE_SUCCESS;
×
3372

UNCOV
3373
  if (IS_VAR_DATA_TYPE(pToColData->type)) {
×
UNCOV
3374
    int32_t nData = (iFromRow < pFromColData->nVal - 1)
×
UNCOV
3375
                        ? pFromColData->aOffset[iFromRow + 1] - pFromColData->aOffset[iFromRow]
×
UNCOV
3376
                        : pFromColData->nData - pFromColData->aOffset[iFromRow];
×
UNCOV
3377
    if (iToRow == 0) {
×
UNCOV
3378
      pToColData->aOffset[iToRow] = 0;
×
3379
    }
3380

UNCOV
3381
    if (iToRow < pToColData->nVal - 1) {
×
UNCOV
3382
      pToColData->aOffset[iToRow + 1] = pToColData->aOffset[iToRow] + nData;
×
3383
    }
3384

UNCOV
3385
    (void)memcpy(pToColData->pData + pToColData->aOffset[iToRow], pFromColData->pData + pFromColData->aOffset[iFromRow],
×
3386
                 nData);
3387
  } else {
UNCOV
3388
    (void)memcpy(&pToColData->pData[TYPE_BYTES[pToColData->type] * iToRow],
×
UNCOV
3389
                 &pFromColData->pData[TYPE_BYTES[pToColData->type] * iFromRow], TYPE_BYTES[pToColData->type]);
×
3390
  }
UNCOV
3391
  return code;
×
3392
}
3393

UNCOV
3394
static int32_t tColDataCopyRowSingleCol(SColData *pFromColData, int32_t iFromRow, SColData *pToColData,
×
3395
                                        int32_t iToRow) {
UNCOV
3396
  int32_t code = TSDB_CODE_SUCCESS;
×
3397

UNCOV
3398
  switch (pFromColData->flag) {
×
UNCOV
3399
    case HAS_NONE:
×
3400
    case HAS_NULL:
3401
      break;
×
3402
    case (HAS_NULL | HAS_NONE): {
×
3403
      SET_BIT1(pToColData->pBitMap, iToRow, GET_BIT1(pFromColData->pBitMap, iFromRow));
×
UNCOV
3404
    } break;
×
UNCOV
3405
    case HAS_VALUE: {
×
UNCOV
3406
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
×
UNCOV
3407
    } break;
×
UNCOV
3408
    case (HAS_VALUE | HAS_NONE):
×
3409
    case (HAS_VALUE | HAS_NULL): {
UNCOV
3410
      SET_BIT1(pToColData->pBitMap, iToRow, GET_BIT1(pFromColData->pBitMap, iFromRow));
×
UNCOV
3411
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
×
3412
    } break;
×
3413
    case (HAS_VALUE | HAS_NULL | HAS_NONE): {
×
3414
      SET_BIT2(pToColData->pBitMap, iToRow, GET_BIT2(pFromColData->pBitMap, iFromRow));
×
3415
      TAOS_CHECK_RETURN(tColDataCopyRowCell(pFromColData, iFromRow, pToColData, iToRow));
×
3416
    } break;
×
3417
    default:
×
UNCOV
3418
      return -1;
×
3419
  }
3420

UNCOV
3421
  return code;
×
3422
}
3423

UNCOV
3424
static int32_t tColDataCopyRow(SColData *aFromColData, int32_t iFromRow, SColData *aToColData, int32_t iToRow,
×
3425
                               int32_t nColData) {
UNCOV
3426
  int32_t code = TSDB_CODE_SUCCESS;
×
3427

UNCOV
3428
  for (int32_t i = 0; i < nColData; i++) {
×
UNCOV
3429
    code = tColDataCopyRowSingleCol(&aFromColData[i], iFromRow, &aToColData[i], iToRow);
×
3430
    if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3431
      return code;
×
3432
    }
3433
  }
3434

UNCOV
3435
  return code;
×
3436
}
3437

UNCOV
3438
static int32_t tColDataCopyRowAppend(SColData *aFromColData, int32_t iFromRow, SColData *aToColData, int32_t nColData) {
×
UNCOV
3439
  int32_t code = TSDB_CODE_SUCCESS;
×
3440

UNCOV
3441
  for (int32_t i = 0; i < nColData; i++) {
×
UNCOV
3442
    SColVal cv = {0};
×
UNCOV
3443
    code = tColDataGetValue(&aFromColData[i], iFromRow, &cv);
×
3444
    if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3445
      return code;
×
3446
    }
UNCOV
3447
    code = tColDataAppendValue(&aToColData[i], &cv);
×
3448
    if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3449
      return code;
×
3450
    }
3451
  }
3452

UNCOV
3453
  return code;
×
3454
}
3455

3456
void tColDataArrGetRowKey(SColData *aColData, int32_t nColData, int32_t iRow, SRowKey *key) {
2,220✔
3457
  SColVal cv;
3458

3459
  key->ts = ((TSKEY *)aColData[0].pData)[iRow];
2,220✔
3460
  key->numOfPKs = 0;
2,220✔
3461

3462
  for (int i = 1; i < nColData; i++) {
2,220!
3463
    if (aColData[i].cflag & COL_IS_KEY) {
2,220!
UNCOV
3464
      tColDataGetValue4(&aColData[i], iRow, &cv);
×
UNCOV
3465
      key->pks[key->numOfPKs++] = cv.value;
×
3466
    } else {
3467
      break;
2,220✔
3468
    }
3469
  }
3470
}
2,220✔
3471

UNCOV
3472
static int32_t tColDataMergeSortMerge(SColData *aColData, int32_t start, int32_t mid, int32_t end, int32_t nColData) {
×
UNCOV
3473
  SColData *aDstColData = NULL;
×
UNCOV
3474
  int32_t   i = start, j = mid + 1, k = 0;
×
3475
  SRowKey   keyi, keyj;
3476

UNCOV
3477
  if (end > start) {
×
UNCOV
3478
    aDstColData = taosMemoryCalloc(1, sizeof(SColData) * nColData);
×
3479
    if (aDstColData == NULL) {
×
UNCOV
3480
      return terrno;
×
3481
    }
UNCOV
3482
    for (int c = 0; c < nColData; ++c) {
×
UNCOV
3483
      tColDataInit(&aDstColData[c], aColData[c].cid, aColData[c].type, aColData[c].cflag);
×
3484
    }
3485
  }
3486

UNCOV
3487
  tColDataArrGetRowKey(aColData, nColData, i, &keyi);
×
UNCOV
3488
  tColDataArrGetRowKey(aColData, nColData, j, &keyj);
×
UNCOV
3489
  while (i <= mid && j <= end) {
×
UNCOV
3490
    if (tRowKeyCompare(&keyi, &keyj) <= 0) {
×
UNCOV
3491
      TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, i++, aDstColData, nColData));
×
UNCOV
3492
      tColDataArrGetRowKey(aColData, nColData, i, &keyi);
×
3493
    } else {
UNCOV
3494
      TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, j++, aDstColData, nColData));
×
UNCOV
3495
      tColDataArrGetRowKey(aColData, nColData, j, &keyj);
×
3496
    }
3497
  }
3498

UNCOV
3499
  while (i <= mid) {
×
UNCOV
3500
    TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, i++, aDstColData, nColData));
×
3501
  }
3502

UNCOV
3503
  while (j <= end) {
×
UNCOV
3504
    TAOS_CHECK_RETURN(tColDataCopyRowAppend(aColData, j++, aDstColData, nColData));
×
3505
  }
3506

UNCOV
3507
  for (i = start, k = 0; i <= end; ++i, ++k) {
×
UNCOV
3508
    TAOS_CHECK_RETURN(tColDataCopyRow(aDstColData, k, aColData, i, nColData));
×
3509
  }
3510

UNCOV
3511
  if (aDstColData) {
×
UNCOV
3512
    for (int32_t i = 0; i < nColData; i++) {
×
UNCOV
3513
      tColDataDestroy(&aDstColData[i]);
×
3514
    }
UNCOV
3515
    taosMemoryFree(aDstColData);
×
3516
  }
3517

UNCOV
3518
  return TSDB_CODE_SUCCESS;
×
3519
}
3520

UNCOV
3521
static int32_t tColDataMergeSort(SColData *aColData, int32_t start, int32_t end, int32_t nColData) {
×
UNCOV
3522
  int32_t ret = TSDB_CODE_SUCCESS;
×
3523
  int32_t mid;
3524

UNCOV
3525
  if (start >= end) {
×
UNCOV
3526
    return TSDB_CODE_SUCCESS;
×
3527
  }
3528

UNCOV
3529
  mid = (start + end) / 2;
×
3530

UNCOV
3531
  ret = tColDataMergeSort(aColData, start, mid, nColData);
×
3532
  if (ret != TSDB_CODE_SUCCESS) {
×
UNCOV
3533
    return ret;
×
3534
  }
3535

UNCOV
3536
  ret = tColDataMergeSort(aColData, mid + 1, end, nColData);
×
3537
  if (ret != TSDB_CODE_SUCCESS) {
×
UNCOV
3538
    return ret;
×
3539
  }
3540

UNCOV
3541
  return tColDataMergeSortMerge(aColData, start, mid, end, nColData);
×
3542
}
3543

UNCOV
3544
static int32_t tColDataSort(SColData *aColData, int32_t nColData) {
×
UNCOV
3545
  int32_t nVal = aColData[0].nVal;
×
3546

UNCOV
3547
  if (nVal < 2) return TSDB_CODE_SUCCESS;
×
3548

UNCOV
3549
  return tColDataMergeSort(aColData, 0, nVal - 1, nColData);
×
3550
}
3551

UNCOV
3552
static int32_t tColDataMerge(SArray **colArr) {
×
UNCOV
3553
  int32_t code = 0;
×
UNCOV
3554
  SArray *src = *colArr;
×
UNCOV
3555
  SArray *dst = NULL;
×
3556

UNCOV
3557
  dst = taosArrayInit(taosArrayGetSize(src), sizeof(SColData));
×
3558
  if (dst == NULL) {
×
UNCOV
3559
    return terrno;
×
3560
  }
3561

UNCOV
3562
  for (int32_t i = 0; i < taosArrayGetSize(src); i++) {
×
UNCOV
3563
    SColData *srcCol = taosArrayGet(src, i);
×
3564

UNCOV
3565
    SColData *dstCol = taosArrayReserve(dst, 1);
×
3566
    if (dstCol == NULL) {
×
3567
      code = terrno;
×
UNCOV
3568
      goto _exit;
×
3569
    }
UNCOV
3570
    tColDataInit(dstCol, srcCol->cid, srcCol->type, srcCol->cflag);
×
3571
  }
3572

UNCOV
3573
  int32_t numRows = ((SColData *)TARRAY_DATA(src))->nVal;
×
3574
  SRowKey lastKey;
UNCOV
3575
  for (int32_t i = 0; i < numRows; i++) {
×
3576
    SRowKey key;
UNCOV
3577
    tColDataArrGetRowKey((SColData *)TARRAY_DATA(src), taosArrayGetSize(src), i, &key);
×
3578

UNCOV
3579
    if (i == 0 || tRowKeyCompare(&key, &lastKey) != 0) {  // append new row
×
UNCOV
3580
      for (int32_t j = 0; j < taosArrayGetSize(src); j++) {
×
UNCOV
3581
        SColData *srcCol = taosArrayGet(src, j);
×
UNCOV
3582
        SColData *dstCol = taosArrayGet(dst, j);
×
3583

3584
        SColVal cv;
UNCOV
3585
        code = tColDataGetValue(srcCol, i, &cv);
×
3586
        if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3587
          goto _exit;
×
3588
        }
UNCOV
3589
        code = tColDataAppendValue(dstCol, &cv);
×
3590
        if (code) {
×
UNCOV
3591
          goto _exit;
×
3592
        }
3593
      }
UNCOV
3594
      lastKey = key;
×
3595
    } else {  // update existing row
UNCOV
3596
      for (int32_t j = 0; j < taosArrayGetSize(src); j++) {
×
UNCOV
3597
        SColData *srcCol = taosArrayGet(src, j);
×
UNCOV
3598
        SColData *dstCol = taosArrayGet(dst, j);
×
3599

3600
        SColVal cv;
UNCOV
3601
        code = tColDataGetValue(srcCol, i, &cv);
×
3602
        if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
3603
          goto _exit;
×
3604
        }
UNCOV
3605
        code = tColDataUpdateValue(dstCol, &cv, true);
×
3606
        if (code) {
×
UNCOV
3607
          goto _exit;
×
3608
        }
3609
      }
3610
    }
3611
  }
3612

UNCOV
3613
_exit:
×
3614
  if (code) {
×
UNCOV
3615
    taosArrayDestroyEx(dst, tColDataDestroy);
×
3616
  } else {
UNCOV
3617
    taosArrayDestroyEx(src, tColDataDestroy);
×
UNCOV
3618
    *colArr = dst;
×
3619
  }
UNCOV
3620
  return code;
×
3621
}
3622

3623
int32_t tColDataSortMerge(SArray **arr) {
23✔
3624
  SArray   *colDataArr = *arr;
23✔
3625
  int32_t   nColData = TARRAY_SIZE(colDataArr);
23✔
3626
  SColData *aColData = (SColData *)TARRAY_DATA(colDataArr);
23✔
3627

3628
  if (!(aColData[0].type == TSDB_DATA_TYPE_TIMESTAMP)) {
23!
UNCOV
3629
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;
×
3630
  }
3631
  if (!(aColData[0].cid == PRIMARYKEY_TIMESTAMP_COL_ID)) {
23!
UNCOV
3632
    return TSDB_CODE_PAR_INVALID_FIRST_COLUMN;
×
3633
  }
3634
  if (!(aColData[0].flag == HAS_VALUE)) {
23!
UNCOV
3635
    return TSDB_CODE_PAR_PRIMARY_KEY_IS_NULL;
×
3636
  }
3637

3638
  if (aColData[0].nVal <= 1) goto _exit;
23✔
3639

3640
  int8_t doSort = 0;
21✔
3641
  int8_t doMerge = 0;
21✔
3642
  // scan -------
3643
  SRowKey lastKey;
3644
  tColDataArrGetRowKey(aColData, nColData, 0, &lastKey);
21✔
3645
  for (int32_t iVal = 1; iVal < aColData[0].nVal; ++iVal) {
1,110✔
3646
    SRowKey key;
3647
    tColDataArrGetRowKey(aColData, nColData, iVal, &key);
1,089✔
3648

3649
    int32_t c = tRowKeyCompare(&lastKey, &key);
1,089✔
3650
    if (c < 0) {
1,089!
3651
      lastKey = key;
1,089✔
3652
      continue;
1,089✔
UNCOV
3653
    } else if (c > 0) {
×
UNCOV
3654
      doSort = 1;
×
UNCOV
3655
      break;
×
3656
    } else {
UNCOV
3657
      doMerge = 1;
×
3658
    }
3659
  }
3660

3661
  // sort -------
3662
  if (doSort) {
21!
UNCOV
3663
    TAOS_CHECK_RETURN(tColDataSort(aColData, nColData));
×
3664
  }
3665

3666
  if (doMerge != 1) {
21!
3667
    tColDataArrGetRowKey(aColData, nColData, 0, &lastKey);
21✔
3668
    for (int32_t iVal = 1; iVal < aColData[0].nVal; ++iVal) {
1,110✔
3669
      SRowKey key;
3670
      tColDataArrGetRowKey(aColData, nColData, iVal, &key);
1,089✔
3671

3672
      int32_t c = tRowKeyCompare(&lastKey, &key);
1,089✔
3673
      if (c == 0) {
1,089!
3674
        doMerge = 1;
×
UNCOV
3675
        break;
×
3676
      }
3677
      lastKey = key;
1,089✔
3678
    }
3679
  }
3680

3681
  // merge -------
3682
  if (doMerge) {
21!
UNCOV
3683
    int32_t code = tColDataMerge(arr);
×
UNCOV
3684
    if (code) return code;
×
3685
  }
3686

3687
_exit:
21✔
3688
  return 0;
23✔
3689
}
3690

3691
static int32_t tEncodeColDataVersion0(SEncoder *pEncoder, SColData *pColData) {
368✔
3692
  int32_t code = 0;
368✔
3693

3694
  if ((code = tEncodeI16v(pEncoder, pColData->cid))) return code;
736!
3695
  if ((code = tEncodeI8(pEncoder, pColData->type))) return code;
736!
3696
  if ((code = tEncodeI32v(pEncoder, pColData->nVal))) return code;
736!
3697
  if ((code = tEncodeI8(pEncoder, pColData->flag))) return code;
736!
3698

3699
  // bitmap
3700
  switch (pColData->flag) {
368!
UNCOV
3701
    case (HAS_NULL | HAS_NONE):
×
3702
    case (HAS_VALUE | HAS_NONE):
3703
    case (HAS_VALUE | HAS_NULL):
UNCOV
3704
      code = tEncodeFixed(pEncoder, pColData->pBitMap, BIT1_SIZE(pColData->nVal));
×
UNCOV
3705
      if (code) return code;
×
3706
      break;
×
3707
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
3708
      code = tEncodeFixed(pEncoder, pColData->pBitMap, BIT2_SIZE(pColData->nVal));
×
3709
      if (code) return code;
×
UNCOV
3710
      break;
×
3711
    default:
368✔
3712
      break;
368✔
3713
  }
3714

3715
  // value
3716
  if (pColData->flag & HAS_VALUE) {
368!
3717
    if (IS_VAR_DATA_TYPE(pColData->type)) {
368!
3718
      code = tEncodeFixed(pEncoder, pColData->aOffset, pColData->nVal << 2);
52✔
3719
      if (code) return code;
52!
3720

3721
      code = tEncodeI32v(pEncoder, pColData->nData);
52✔
3722
      if (code) return code;
52!
3723

3724
      code = tEncodeFixed(pEncoder, pColData->pData, pColData->nData);
52✔
3725
      if (code) return code;
52!
3726
    } else {
3727
      code = tEncodeFixed(pEncoder, pColData->pData, pColData->nData);
316✔
3728
      if (code) return code;
316!
3729
    }
3730
  }
3731

3732
  return code;
368✔
3733
}
3734

UNCOV
3735
static int32_t tDecodeColDataVersion0(SDecoder *pDecoder, SColData *pColData) {
×
UNCOV
3736
  int32_t code = 0;
×
3737

UNCOV
3738
  if ((code = tDecodeI16v(pDecoder, &pColData->cid))) return code;
×
UNCOV
3739
  if ((code = tDecodeI8(pDecoder, &pColData->type))) return code;
×
UNCOV
3740
  if ((code = tDecodeI32v(pDecoder, &pColData->nVal))) return code;
×
UNCOV
3741
  if ((code = tDecodeI8(pDecoder, &pColData->flag))) return code;
×
3742

3743
  if (pColData->type <= 0 || pColData->type >= TSDB_DATA_TYPE_MAX || pColData->flag <= 0 || pColData->flag >= 8) {
×
UNCOV
3744
    return TSDB_CODE_INVALID_PARA;
×
3745
  }
3746

3747
  // bitmap
UNCOV
3748
  switch (pColData->flag) {
×
UNCOV
3749
    case (HAS_NULL | HAS_NONE):
×
3750
    case (HAS_VALUE | HAS_NONE):
3751
    case (HAS_VALUE | HAS_NULL):
UNCOV
3752
      code = tDecodeBinaryWithSize(pDecoder, BIT1_SIZE(pColData->nVal), &pColData->pBitMap);
×
UNCOV
3753
      if (code) return code;
×
3754
      break;
×
3755
    case (HAS_VALUE | HAS_NULL | HAS_NONE):
×
3756
      code = tDecodeBinaryWithSize(pDecoder, BIT2_SIZE(pColData->nVal), &pColData->pBitMap);
×
3757
      if (code) return code;
×
UNCOV
3758
      break;
×
UNCOV
3759
    default:
×
UNCOV
3760
      break;
×
3761
  }
3762

3763
  // value
UNCOV
3764
  if (pColData->flag & HAS_VALUE) {
×
UNCOV
3765
    if (IS_VAR_DATA_TYPE(pColData->type)) {
×
UNCOV
3766
      code = tDecodeBinaryWithSize(pDecoder, pColData->nVal << 2, (uint8_t **)&pColData->aOffset);
×
UNCOV
3767
      if (code) return code;
×
3768

UNCOV
3769
      code = tDecodeI32v(pDecoder, &pColData->nData);
×
UNCOV
3770
      if (code) return code;
×
3771

UNCOV
3772
      code = tDecodeBinaryWithSize(pDecoder, pColData->nData, &pColData->pData);
×
UNCOV
3773
      if (code) return code;
×
3774
    } else {
UNCOV
3775
      pColData->nData = TYPE_BYTES[pColData->type] * pColData->nVal;
×
UNCOV
3776
      code = tDecodeBinaryWithSize(pDecoder, pColData->nData, &pColData->pData);
×
UNCOV
3777
      if (code) return code;
×
3778
    }
3779
  }
UNCOV
3780
  pColData->cflag = 0;
×
3781

UNCOV
3782
  return code;
×
3783
}
3784

3785
static int32_t tEncodeColDataVersion1(SEncoder *pEncoder, SColData *pColData) {
368✔
3786
  int32_t code = tEncodeColDataVersion0(pEncoder, pColData);
368✔
3787
  if (code) return code;
368!
3788
  return tEncodeI8(pEncoder, pColData->cflag);
736✔
3789
}
3790

UNCOV
3791
static int32_t tDecodeColDataVersion1(SDecoder *pDecoder, SColData *pColData) {
×
UNCOV
3792
  int32_t code = tDecodeColDataVersion0(pDecoder, pColData);
×
UNCOV
3793
  if (code) return code;
×
3794

UNCOV
3795
  code = tDecodeI8(pDecoder, &pColData->cflag);
×
UNCOV
3796
  return code;
×
3797
}
3798

3799
int32_t tEncodeColData(uint8_t version, SEncoder *pEncoder, SColData *pColData) {
368✔
3800
  if (version == 0) {
368!
UNCOV
3801
    return tEncodeColDataVersion0(pEncoder, pColData);
×
3802
  } else if (version == 1) {
368!
3803
    return tEncodeColDataVersion1(pEncoder, pColData);
368✔
3804
  } else {
UNCOV
3805
    return TSDB_CODE_INVALID_PARA;
×
3806
  }
3807
}
3808

UNCOV
3809
int32_t tDecodeColData(uint8_t version, SDecoder *pDecoder, SColData *pColData) {
×
3810
  if (version == 0) {
×
UNCOV
3811
    return tDecodeColDataVersion0(pDecoder, pColData);
×
UNCOV
3812
  } else if (version == 1) {
×
UNCOV
3813
    return tDecodeColDataVersion1(pDecoder, pColData);
×
3814
  } else {
UNCOV
3815
    return TSDB_CODE_INVALID_PARA;
×
3816
  }
3817
}
3818

3819
int32_t tEncodeRow(SEncoder *pEncoder, SRow *pRow) { return tEncodeFixed(pEncoder, pRow, pRow->len); }
278,124,758✔
3820

3821
int32_t tDecodeRow(SDecoder *pDecoder, SRow **ppRow) {
17,976,383✔
3822
  if (ppRow == NULL) {
17,976,383!
UNCOV
3823
    return TSDB_CODE_INVALID_PARA;
×
3824
  }
3825

3826
  if (pDecoder->pos + sizeof(SRow) > pDecoder->size) {
17,976,383!
UNCOV
3827
    return TSDB_CODE_OUT_OF_RANGE;
×
3828
  }
3829

3830
  SRow *pRow = (SRow *)(pDecoder->data + pDecoder->pos);
17,976,383✔
3831
  return tDecodeBinaryWithSize(pDecoder, pRow->len, (uint8_t **)ppRow);
35,952,766!
3832
}
3833

3834
#define CALC_SUM_MAX_MIN(SUM, MAX, MIN, VAL) \
3835
  do {                                       \
3836
    (SUM) += (VAL);                          \
3837
    if ((MAX) < (VAL)) (MAX) = (VAL);        \
3838
    if ((MIN) > (VAL)) (MIN) = (VAL);        \
3839
  } while (0)
3840

UNCOV
3841
static FORCE_INLINE void tColDataCalcSMABool(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
3842
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
3843
  int16_t*numOfNull = &pAggs->numOfNull;
×
UNCOV
3844
  *sum = 0;
×
UNCOV
3845
  *max = 0;
×
UNCOV
3846
  *min = 1;
×
UNCOV
3847
  *numOfNull = 0;
×
3848

3849
  int8_t val;
UNCOV
3850
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
3851
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
3852
      val = ((int8_t *)pColData->pData)[iVal] ? 1 : 0;
×
UNCOV
3853
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3854
    }
3855
  } else {
3856
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3857
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
3858
        case 0:
×
3859
        case 1:
3860
          (*numOfNull)++;
×
3861
          break;
×
3862
        case 2:
×
3863
          val = ((int8_t *)pColData->pData)[iVal] ? 1 : 0;
×
3864
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3865
          break;
×
3866
        default:
×
UNCOV
3867
          break;
×
3868
      }
3869
    }
3870
  }
UNCOV
3871
}
×
3872

UNCOV
3873
static FORCE_INLINE void tColDataCalcSMATinyInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
3874
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
3875
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
3876
  *sum = 0;
×
UNCOV
3877
  *max = INT8_MIN;
×
UNCOV
3878
  *min = INT8_MAX;
×
UNCOV
3879
  *numOfNull = 0;
×
3880

3881
  int8_t val;
UNCOV
3882
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
3883
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
3884
      val = ((int8_t *)pColData->pData)[iVal];
×
UNCOV
3885
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3886
    }
3887
  } else {
3888
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3889
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
3890
        case 0:
×
3891
        case 1:
3892
          (*numOfNull)++;
×
3893
          break;
×
3894
        case 2:
×
3895
          val = ((int8_t *)pColData->pData)[iVal];
×
3896
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3897
          break;
×
3898
        default:
×
UNCOV
3899
          break;
×
3900
      }
3901
    }
3902
  }
UNCOV
3903
}
×
3904

UNCOV
3905
static FORCE_INLINE void tColDataCalcSMATinySmallInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
3906
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
3907
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
3908
  *sum = 0;
×
UNCOV
3909
  *max = INT16_MIN;
×
UNCOV
3910
  *min = INT16_MAX;
×
UNCOV
3911
  *numOfNull = 0;
×
3912

3913
  int16_t val;
UNCOV
3914
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
3915
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
3916
      val = ((int16_t *)pColData->pData)[iVal];
×
UNCOV
3917
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3918
    }
3919
  } else {
3920
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
3921
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
3922
        case 0:
×
3923
        case 1:
3924
          (*numOfNull)++;
×
3925
          break;
×
3926
        case 2:
×
3927
          val = ((int16_t *)pColData->pData)[iVal];
×
3928
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3929
          break;
×
3930
        default:
×
UNCOV
3931
          break;
×
3932
      }
3933
    }
3934
  }
UNCOV
3935
}
×
3936

3937
static FORCE_INLINE void tColDataCalcSMAInt(SColData *pColData, SColumnDataAgg* pAggs) {
14,938✔
3938
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
14,938✔
3939
  int16_t *numOfNull = &pAggs->numOfNull;
14,938✔
3940
  *sum = 0;
14,938✔
3941
  *max = INT32_MIN;
14,938✔
3942
  *min = INT32_MAX;
14,938✔
3943
  *numOfNull = 0;
14,938✔
3944

3945
  int32_t val;
3946
  if (HAS_VALUE == pColData->flag) {
14,938✔
3947
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
16,016,450✔
3948
      val = ((int32_t *)pColData->pData)[iVal];
16,010,304✔
3949
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
16,010,304✔
3950
    }
3951
  } else {
3952
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
25,901,650✔
3953
      switch (tColDataGetBitValue(pColData, iVal)) {
25,892,858!
3954
        case 0:
1,295,525✔
3955
        case 1:
3956
          (*numOfNull)++;
1,295,525✔
3957
          break;
1,295,525✔
3958
        case 2:
24,597,333✔
3959
          val = ((int32_t *)pColData->pData)[iVal];
24,597,333✔
3960
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
24,597,333✔
3961
          break;
24,597,333✔
3962
        default:
×
UNCOV
3963
          break;
×
3964
      }
3965
    }
3966
  }
3967
}
14,938✔
3968

3969
static FORCE_INLINE void tColDataCalcSMABigInt(SColData *pColData, SColumnDataAgg* pAggs) {
7,469✔
3970
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
7,469✔
3971
  int16_t *numOfNull = &pAggs->numOfNull;
7,469✔
3972
  *sum = 0;
7,469✔
3973
  *max = INT64_MIN;
7,469✔
3974
  *min = INT64_MAX;
7,469✔
3975
  *numOfNull = 0;
7,469✔
3976

3977
  int64_t val;
3978
  if (HAS_VALUE == pColData->flag) {
7,469!
3979
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
20,956,926✔
3980
      val = ((int64_t *)pColData->pData)[iVal];
20,949,457✔
3981
      CALC_SUM_MAX_MIN(*sum, *max, *min, val);
20,949,457✔
3982
    }
3983
  } else {
UNCOV
3984
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
3985
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
3986
        case 0:
×
3987
        case 1:
UNCOV
3988
          (*numOfNull)++;
×
UNCOV
3989
          break;
×
UNCOV
3990
        case 2:
×
UNCOV
3991
          val = ((int64_t *)pColData->pData)[iVal];
×
UNCOV
3992
          CALC_SUM_MAX_MIN(*sum, *max, *min, val);
×
3993
          break;
×
3994
        default:
×
UNCOV
3995
          break;
×
3996
      }
3997
    }
3998
  }
3999
}
7,469✔
4000

UNCOV
4001
static FORCE_INLINE void tColDataCalcSMAFloat(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4002
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4003
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4004
  *(double *)sum = 0;
×
UNCOV
4005
  *(double *)max = -FLT_MAX;
×
UNCOV
4006
  *(double *)min = FLT_MAX;
×
UNCOV
4007
  *numOfNull = 0;
×
4008

4009
  float val;
UNCOV
4010
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4011
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4012
      val = ((float *)pColData->pData)[iVal];
×
UNCOV
4013
      CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
×
4014
    }
4015
  } else {
UNCOV
4016
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4017
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4018
        case 0:
×
4019
        case 1:
UNCOV
4020
          (*numOfNull)++;
×
UNCOV
4021
          break;
×
UNCOV
4022
        case 2:
×
UNCOV
4023
          val = ((float *)pColData->pData)[iVal];
×
UNCOV
4024
          CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
×
4025
          break;
×
4026
        default:
×
UNCOV
4027
          break;
×
4028
      }
4029
    }
4030
  }
UNCOV
4031
}
×
4032

UNCOV
4033
static FORCE_INLINE void tColDataCalcSMADouble(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4034
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4035
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4036
  *(double *)sum = 0;
×
UNCOV
4037
  *(double *)max = -DBL_MAX;
×
UNCOV
4038
  *(double *)min = DBL_MAX;
×
UNCOV
4039
  *numOfNull = 0;
×
4040

4041
  double val;
UNCOV
4042
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4043
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4044
      val = ((double *)pColData->pData)[iVal];
×
UNCOV
4045
      CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
×
4046
    }
4047
  } else {
4048
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4049
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4050
        case 0:
×
4051
        case 1:
4052
          (*numOfNull)++;
×
4053
          break;
×
4054
        case 2:
×
4055
          val = ((double *)pColData->pData)[iVal];
×
4056
          CALC_SUM_MAX_MIN(*(double *)sum, *(double *)max, *(double *)min, val);
×
4057
          break;
×
4058
        default:
×
UNCOV
4059
          break;
×
4060
      }
4061
    }
4062
  }
UNCOV
4063
}
×
4064

UNCOV
4065
static FORCE_INLINE void tColDataCalcSMAUTinyInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4066
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4067
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4068
  *(uint64_t *)sum = 0;
×
UNCOV
4069
  *(uint64_t *)max = 0;
×
UNCOV
4070
  *(uint64_t *)min = UINT8_MAX;
×
UNCOV
4071
  *numOfNull = 0;
×
4072

4073
  uint8_t val;
UNCOV
4074
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4075
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4076
      val = ((uint8_t *)pColData->pData)[iVal];
×
UNCOV
4077
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4078
    }
4079
  } else {
4080
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4081
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4082
        case 0:
×
4083
        case 1:
4084
          (*numOfNull)++;
×
4085
          break;
×
4086
        case 2:
×
4087
          val = ((uint8_t *)pColData->pData)[iVal];
×
4088
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4089
          break;
×
4090
        default:
×
UNCOV
4091
          break;
×
4092
      }
4093
    }
4094
  }
UNCOV
4095
}
×
4096

UNCOV
4097
static FORCE_INLINE void tColDataCalcSMATinyUSmallInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4098
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4099
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4100
  *(uint64_t *)sum = 0;
×
UNCOV
4101
  *(uint64_t *)max = 0;
×
UNCOV
4102
  *(uint64_t *)min = UINT16_MAX;
×
UNCOV
4103
  *numOfNull = 0;
×
4104

4105
  uint16_t val;
UNCOV
4106
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4107
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4108
      val = ((uint16_t *)pColData->pData)[iVal];
×
UNCOV
4109
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4110
    }
4111
  } else {
4112
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4113
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4114
        case 0:
×
4115
        case 1:
4116
          (*numOfNull)++;
×
4117
          break;
×
4118
        case 2:
×
4119
          val = ((uint16_t *)pColData->pData)[iVal];
×
4120
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4121
          break;
×
4122
        default:
×
UNCOV
4123
          break;
×
4124
      }
4125
    }
4126
  }
UNCOV
4127
}
×
4128

UNCOV
4129
static FORCE_INLINE void tColDataCalcSMAUInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4130
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4131
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4132
  *(uint64_t *)sum = 0;
×
UNCOV
4133
  *(uint64_t *)max = 0;
×
UNCOV
4134
  *(uint64_t *)min = UINT32_MAX;
×
UNCOV
4135
  *numOfNull = 0;
×
4136

4137
  uint32_t val;
UNCOV
4138
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4139
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4140
      val = ((uint32_t *)pColData->pData)[iVal];
×
UNCOV
4141
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4142
    }
4143
  } else {
4144
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4145
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4146
        case 0:
×
4147
        case 1:
4148
          (*numOfNull)++;
×
4149
          break;
×
4150
        case 2:
×
4151
          val = ((uint32_t *)pColData->pData)[iVal];
×
4152
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4153
          break;
×
4154
        default:
×
UNCOV
4155
          break;
×
4156
      }
4157
    }
4158
  }
UNCOV
4159
}
×
4160

UNCOV
4161
static FORCE_INLINE void tColDataCalcSMAUBigInt(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4162
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4163
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4164
  *(uint64_t *)sum = 0;
×
UNCOV
4165
  *(uint64_t *)max = 0;
×
UNCOV
4166
  *(uint64_t *)min = UINT64_MAX;
×
UNCOV
4167
  *numOfNull = 0;
×
4168

4169
  uint64_t val;
UNCOV
4170
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4171
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
UNCOV
4172
      val = ((uint64_t *)pColData->pData)[iVal];
×
UNCOV
4173
      CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4174
    }
4175
  } else {
4176
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4177
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4178
        case 0:
×
4179
        case 1:
4180
          (*numOfNull)++;
×
4181
          break;
×
4182
        case 2:
×
4183
          val = ((uint64_t *)pColData->pData)[iVal];
×
4184
          CALC_SUM_MAX_MIN(*(uint64_t *)sum, *(uint64_t *)max, *(uint64_t *)min, val);
×
4185
          break;
×
4186
        default:
×
UNCOV
4187
          break;
×
4188
      }
4189
    }
4190
  }
UNCOV
4191
}
×
4192

UNCOV
4193
static FORCE_INLINE void tColDataCalcSMAVarType(SColData *pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4194
  int64_t *sum = &pAggs->sum, *max = &pAggs->max, *min = &pAggs->min;
×
UNCOV
4195
  int16_t *numOfNull = &pAggs->numOfNull;
×
UNCOV
4196
  *(uint64_t *)sum = 0;
×
UNCOV
4197
  *(uint64_t *)max = 0;
×
UNCOV
4198
  *(uint64_t *)min = 0;
×
UNCOV
4199
  *numOfNull = 0;
×
4200

4201
  switch (pColData->flag) {
×
UNCOV
4202
    case HAS_NONE:
×
4203
    case HAS_NULL:
4204
    case (HAS_NONE | HAS_NULL):
4205
      *numOfNull = pColData->nVal;
×
UNCOV
4206
      break;
×
UNCOV
4207
    case HAS_VALUE:
×
UNCOV
4208
      *numOfNull = 0;
×
4209
      break;
×
UNCOV
4210
    case (HAS_VALUE | HAS_NULL):
×
4211
    case (HAS_VALUE | HAS_NONE):
4212
      for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4213
        if (GET_BIT1(pColData->pBitMap, iVal) == 0) {
×
UNCOV
4214
          (*numOfNull)++;
×
4215
        }
4216
      }
4217
      break;
×
4218
    case (HAS_VALUE | HAS_NONE | HAS_NULL):
×
4219
      for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
×
4220
        if (GET_BIT2(pColData->pBitMap, iVal) != 2) {
×
UNCOV
4221
          (*numOfNull)++;
×
4222
        }
4223
      }
4224
      break;
×
4225
    default:
×
UNCOV
4226
      break;
×
4227
  }
UNCOV
4228
}
×
4229

4230
#define CALC_DECIMAL_SUM_MAX_MIN(TYPE, pSumOp, pCompOp, pColData, pSum, pMax, pMin)           \
4231
  do {                                                                                        \
4232
    if (decimal128AddCheckOverflow((Decimal *)pSum, pVal, DECIMAL_WORD_NUM(TYPE))) *pOverflow = true; \
4233
    pSumOp->add(pSum, pVal, DECIMAL_WORD_NUM(TYPE));                                                  \
4234
    if (pCompOp->gt(pVal, pMax, DECIMAL_WORD_NUM(TYPE))) {                                            \
4235
      *(pMax) = *pVal;                                                                        \
4236
    }                                                                                         \
4237
    if (pCompOp->lt(pVal, pMin, DECIMAL_WORD_NUM(TYPE))) {                                            \
4238
      *(pMin) = *pVal;                                                                        \
4239
    }                                                                                         \
4240
  } while (0)
4241

UNCOV
4242
static FORCE_INLINE void tColDataCalcSMADecimal64Type(SColData* pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4243
  Decimal128 *pSum = (Decimal128 *)pAggs->decimal128Sum;
×
UNCOV
4244
  Decimal64  *pMax = (Decimal64 *)pAggs->decimal128Max, *pMin = (Decimal64 *)pAggs->decimal128Min;
×
UNCOV
4245
  uint8_t *pOverflow = &pAggs->overflow;
×
UNCOV
4246
  *pSum = DECIMAL128_ZERO;
×
UNCOV
4247
  *pMax = DECIMAL64_MIN;
×
UNCOV
4248
  *pMin = DECIMAL64_MAX;
×
UNCOV
4249
  pAggs->numOfNull = 0;
×
UNCOV
4250
  pAggs->colId |= DECIMAL_AGG_FLAG;
×
4251

UNCOV
4252
  Decimal64   *pVal = NULL;
×
UNCOV
4253
  const SDecimalOps *pSumOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL);
×
UNCOV
4254
  const SDecimalOps *pCompOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL64);
×
UNCOV
4255
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4256
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
4257
      pVal = ((Decimal64*)pColData->pData) + iVal;
×
UNCOV
4258
      CALC_DECIMAL_SUM_MAX_MIN(Decimal64, pSumOps, pCompOps, pColData, pSum, pMax, pMin);
×
4259
    }
4260
  } else {
UNCOV
4261
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
4262
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4263
        case 0:
×
4264
        case 1:
UNCOV
4265
          pAggs->numOfNull++;
×
UNCOV
4266
          break;
×
UNCOV
4267
        case 2:
×
UNCOV
4268
          pVal = ((Decimal64 *)pColData->pData) + iVal;
×
UNCOV
4269
          CALC_DECIMAL_SUM_MAX_MIN(Decimal64, pSumOps, pCompOps, pColData, pSum, pMax, pMin);
×
4270
          break;
×
4271
        default:
×
UNCOV
4272
          break;
×
4273
      }
4274
    }
4275
  }
UNCOV
4276
}
×
4277

UNCOV
4278
static FORCE_INLINE void tColDataCalcSMADecimal128Type(SColData* pColData, SColumnDataAgg* pAggs) {
×
UNCOV
4279
  Decimal128 *pSum = (Decimal128 *)pAggs->decimal128Sum, *pMax = (Decimal128 *)pAggs->decimal128Max,
×
UNCOV
4280
             *pMin = (Decimal128 *)pAggs->decimal128Min;
×
UNCOV
4281
  uint8_t *pOverflow = &pAggs->overflow;
×
UNCOV
4282
  *pSum = DECIMAL128_ZERO;
×
UNCOV
4283
  *pMax = DECIMAL128_MIN;
×
UNCOV
4284
  *pMin = DECIMAL128_MAX;
×
UNCOV
4285
  pAggs->numOfNull = 0;
×
UNCOV
4286
  pAggs->colId |= DECIMAL_AGG_FLAG;
×
4287

UNCOV
4288
  Decimal128        *pVal = NULL;
×
UNCOV
4289
  const SDecimalOps *pOps = getDecimalOps(TSDB_DATA_TYPE_DECIMAL);
×
UNCOV
4290
  if (HAS_VALUE == pColData->flag) {
×
UNCOV
4291
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
4292
      pVal = ((Decimal128*)pColData->pData) + iVal;
×
UNCOV
4293
      CALC_DECIMAL_SUM_MAX_MIN(Decimal128, pOps, pOps, pColData, pSum, pMax, pMin);
×
4294
    }
4295
  } else {
UNCOV
4296
    for (int32_t iVal = 0; iVal < pColData->nVal; ++iVal) {
×
UNCOV
4297
      switch (tColDataGetBitValue(pColData, iVal)) {
×
UNCOV
4298
        case 0:
×
4299
        case 1:
UNCOV
4300
          pAggs->numOfNull++;
×
UNCOV
4301
          break;
×
UNCOV
4302
        case 2:
×
UNCOV
4303
          pVal = ((Decimal128*)pColData->pData) + iVal;
×
UNCOV
4304
          CALC_DECIMAL_SUM_MAX_MIN(Decimal128, pOps, pOps, pColData, pSum, pMax, pMin);
×
4305
          break;
×
4306
        default:
×
UNCOV
4307
          break;
×
4308
      }
4309
    }
4310
  }
UNCOV
4311
}
×
4312

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

4338
// SValueColumn ================================
4339
int32_t tValueColumnInit(SValueColumn *valCol) {
34,729✔
4340
  valCol->type = TSDB_DATA_TYPE_NULL;
34,729✔
4341
  valCol->numOfValues = 0;
34,729✔
4342
  tBufferInit(&valCol->data);
34,729✔
4343
  tBufferInit(&valCol->offsets);
34,729✔
4344
  return 0;
34,729✔
4345
}
4346

4347
void tValueColumnDestroy(SValueColumn *valCol) {
74,896✔
4348
  valCol->type = TSDB_DATA_TYPE_NULL;
74,896✔
4349
  valCol->numOfValues = 0;
74,896✔
4350
  tBufferDestroy(&valCol->data);
74,896✔
4351
  tBufferDestroy(&valCol->offsets);
74,896!
4352
  return;
74,896✔
4353
}
4354

4355
void tValueColumnClear(SValueColumn *valCol) {
85,312✔
4356
  valCol->type = TSDB_DATA_TYPE_NULL;
85,312✔
4357
  valCol->numOfValues = 0;
85,312✔
4358
  tBufferClear(&valCol->data);
85,312✔
4359
  tBufferClear(&valCol->offsets);
85,312✔
4360
  return;
85,312✔
4361
}
4362

4363
int32_t tValueColumnAppend(SValueColumn *valCol, const SValue *value) {
36,830✔
4364
  int32_t code;
4365

4366
  if (valCol->numOfValues == 0) {
36,830✔
4367
    valCol->type = value->type;
172✔
4368
  }
4369

4370
  if (!(value->type == valCol->type)) {
36,830!
UNCOV
4371
    return TSDB_CODE_INVALID_PARA;
×
4372
  }
4373

4374
  if (IS_VAR_DATA_TYPE(value->type)) {
36,830!
4375
    if ((code = tBufferPutI32(&valCol->offsets, tBufferGetSize(&valCol->data)))) {
×
UNCOV
4376
      return code;
×
4377
    }
4378
    if ((code = tBufferPut(&valCol->data, value->pData, value->nData))) {
×
UNCOV
4379
      return code;
×
4380
    }
4381
  } else {
4382
    code = tBufferPut(&valCol->data, VALUE_GET_DATUM(value, value->type), tDataTypes[value->type].bytes);
36,830!
4383
    if (code) return code;
36,830!
4384
  }
4385
  valCol->numOfValues++;
36,830✔
4386

4387
  return 0;
36,830✔
4388
}
4389

4390
int32_t tValueColumnUpdate(SValueColumn *valCol, int32_t idx, const SValue *value) {
17,942,341✔
4391
  int32_t code;
4392

4393
  if (idx < 0 || idx >= valCol->numOfValues) {
17,942,341!
UNCOV
4394
    return TSDB_CODE_OUT_OF_RANGE;
×
4395
  }
4396

4397
  if (IS_VAR_DATA_TYPE(valCol->type)) {
17,942,341!
4398
    int32_t *offsets = (int32_t *)tBufferGetData(&valCol->offsets);
×
4399
    int32_t  nextOffset = (idx == valCol->numOfValues - 1) ? tBufferGetSize(&valCol->data) : offsets[idx + 1];
×
4400
    int32_t  oldDataSize = nextOffset - offsets[idx];
×
UNCOV
4401
    int32_t  bytesAdded = value->nData - oldDataSize;
×
4402

4403
    if (bytesAdded != 0) {
×
4404
      if ((code = tBufferEnsureCapacity(&valCol->data, tBufferGetSize(&valCol->data) + bytesAdded))) return code;
×
4405
      memmove(tBufferGetDataAt(&valCol->data, nextOffset + bytesAdded), tBufferGetDataAt(&valCol->data, nextOffset),
×
4406
              tBufferGetSize(&valCol->data) - nextOffset);
×
UNCOV
4407
      valCol->data.size += bytesAdded;
×
4408

4409
      for (int32_t i = idx + 1; i < valCol->numOfValues; i++) {
×
UNCOV
4410
        offsets[i] += bytesAdded;
×
4411
      }
4412
    }
UNCOV
4413
    return tBufferPutAt(&valCol->data, offsets[idx], value->pData, value->nData);
×
4414
  } else {
4415
    return tBufferPutAt(&valCol->data, idx * tDataTypes[valCol->type].bytes, VALUE_GET_DATUM(value, valCol->type),
17,942,341!
4416
                        tDataTypes[valCol->type].bytes);
17,942,341!
4417
  }
4418
  return 0;
4419
}
4420

4421
int32_t tValueColumnGet(SValueColumn *valCol, int32_t idx, SValue *value) {
38,431,964✔
4422
  if (idx < 0 || idx >= valCol->numOfValues) {
38,431,964!
UNCOV
4423
    return TSDB_CODE_OUT_OF_RANGE;
×
4424
  }
4425

4426
  value->type = valCol->type;
38,432,445✔
4427
  if (IS_VAR_DATA_TYPE(value->type)) {
38,432,445!
4428
    int32_t       offset, nextOffset;
4429
    SBufferReader reader = BUFFER_READER_INITIALIZER(idx * sizeof(offset), &valCol->offsets);
182✔
4430

4431
    TAOS_CHECK_RETURN(tBufferGetI32(&reader, &offset));
182!
UNCOV
4432
    if (idx == valCol->numOfValues - 1) {
×
UNCOV
4433
      nextOffset = tBufferGetSize(&valCol->data);
×
4434
    } else {
UNCOV
4435
      TAOS_CHECK_RETURN(tBufferGetI32(&reader, &nextOffset));
×
4436
    }
UNCOV
4437
    value->nData = nextOffset - offset;
×
UNCOV
4438
    value->pData = (uint8_t *)tBufferGetDataAt(&valCol->data, offset);
×
4439
  } else {
4440
    SBufferReader reader = BUFFER_READER_INITIALIZER(idx * tDataTypes[value->type].bytes, &valCol->data);
38,432,263✔
4441
    TAOS_CHECK_RETURN(tBufferGet(&reader, tDataTypes[value->type].bytes, VALUE_GET_DATUM(value, value->type)));
76,864,526!
4442
  }
4443
  return 0;
38,432,263✔
4444
}
4445

4446
int32_t tValueColumnCompress(SValueColumn *valCol, SValueColumnCompressInfo *info, SBuffer *output, SBuffer *assist) {
172✔
4447
  int32_t code;
4448

4449
  if (!(valCol->numOfValues > 0)) {
172!
UNCOV
4450
    return TSDB_CODE_INVALID_PARA;
×
4451
  }
4452

4453
  (*info) = (SValueColumnCompressInfo){
172✔
4454
      .cmprAlg = info->cmprAlg,
172✔
4455
      .type = valCol->type,
172✔
4456
  };
4457

4458
  // offset
4459
  if (IS_VAR_DATA_TYPE(valCol->type)) {
172!
UNCOV
4460
    SCompressInfo cinfo = {
×
UNCOV
4461
        .cmprAlg = info->cmprAlg,
×
4462
        .dataType = TSDB_DATA_TYPE_INT,
UNCOV
4463
        .originalSize = valCol->offsets.size,
×
4464
    };
4465

UNCOV
4466
    code = tCompressDataToBuffer(valCol->offsets.data, &cinfo, output, assist);
×
UNCOV
4467
    if (code) return code;
×
4468

UNCOV
4469
    info->offsetOriginalSize = cinfo.originalSize;
×
UNCOV
4470
    info->offsetCompressedSize = cinfo.compressedSize;
×
4471
  }
4472

4473
  // data
4474
  SCompressInfo cinfo = {
172✔
4475
      .cmprAlg = info->cmprAlg,
172✔
4476
      .dataType = valCol->type,
172✔
4477
      .originalSize = valCol->data.size,
172✔
4478
  };
4479

4480
  code = tCompressDataToBuffer(valCol->data.data, &cinfo, output, assist);
172✔
4481
  if (code) return code;
172!
4482

4483
  info->dataOriginalSize = cinfo.originalSize;
172✔
4484
  info->dataCompressedSize = cinfo.compressedSize;
172✔
4485

4486
  return 0;
172✔
4487
}
4488

4489
int32_t tValueColumnDecompress(void *input, const SValueColumnCompressInfo *info, SValueColumn *valCol,
16,102✔
4490
                               SBuffer *assist) {
4491
  int32_t code;
4492

4493
  tValueColumnClear(valCol);
16,102✔
4494
  valCol->type = info->type;
16,102✔
4495
  // offset
4496
  if (IS_VAR_DATA_TYPE(valCol->type)) {
16,102!
UNCOV
4497
    valCol->numOfValues = info->offsetOriginalSize / tDataTypes[TSDB_DATA_TYPE_INT].bytes;
×
4498

UNCOV
4499
    SCompressInfo cinfo = {
×
4500
        .dataType = TSDB_DATA_TYPE_INT,
4501
        .cmprAlg = info->cmprAlg,
×
4502
        .originalSize = info->offsetOriginalSize,
×
UNCOV
4503
        .compressedSize = info->offsetCompressedSize,
×
4504
    };
4505

4506
    code = tDecompressDataToBuffer(input, &cinfo, &valCol->offsets, assist);
×
4507
    if (code) {
×
UNCOV
4508
      return code;
×
4509
    }
4510
  } else {
4511
    valCol->numOfValues = info->dataOriginalSize / tDataTypes[valCol->type].bytes;
16,102✔
4512
  }
4513

4514
  // data
4515
  SCompressInfo cinfo = {
16,102✔
4516
      .dataType = valCol->type,
16,102✔
4517
      .cmprAlg = info->cmprAlg,
16,102✔
4518
      .originalSize = info->dataOriginalSize,
16,102✔
4519
      .compressedSize = info->dataCompressedSize,
16,102✔
4520
  };
4521

4522
  code = tDecompressDataToBuffer((char *)input + info->offsetCompressedSize, &cinfo, &valCol->data, assist);
16,102✔
4523
  if (code) {
16,102!
UNCOV
4524
    return code;
×
4525
  }
4526

4527
  return 0;
16,102✔
4528
}
4529

4530
int32_t tValueColumnCompressInfoEncode(const SValueColumnCompressInfo *info, SBuffer *buffer) {
172✔
4531
  int32_t code;
4532
  uint8_t fmtVer = 0;
172✔
4533

4534
  if ((code = tBufferPutU8(buffer, fmtVer))) return code;
344!
4535
  if ((code = tBufferPutI8(buffer, info->cmprAlg))) return code;
344!
4536
  if ((code = tBufferPutI8(buffer, info->type))) return code;
344!
4537
  if (IS_VAR_DATA_TYPE(info->type)) {
172!
UNCOV
4538
    if ((code = tBufferPutI32v(buffer, info->offsetOriginalSize))) return code;
×
UNCOV
4539
    if ((code = tBufferPutI32v(buffer, info->offsetCompressedSize))) return code;
×
4540
  }
4541
  if ((code = tBufferPutI32v(buffer, info->dataOriginalSize))) return code;
344!
4542
  if ((code = tBufferPutI32v(buffer, info->dataCompressedSize))) return code;
344!
4543

4544
  return 0;
172✔
4545
}
4546

4547
int32_t tValueColumnCompressInfoDecode(SBufferReader *reader, SValueColumnCompressInfo *info) {
16,102✔
4548
  int32_t code;
4549
  uint8_t fmtVer;
4550

4551
  if ((code = tBufferGetU8(reader, &fmtVer))) return code;
16,102!
4552
  if (fmtVer == 0) {
16,102!
4553
    if ((code = tBufferGetI8(reader, &info->cmprAlg))) return code;
16,102!
4554
    if ((code = tBufferGetI8(reader, &info->type))) return code;
16,102!
4555
    if (IS_VAR_DATA_TYPE(info->type)) {
16,102!
4556
      if ((code = tBufferGetI32v(reader, &info->offsetOriginalSize))) return code;
×
UNCOV
4557
      if ((code = tBufferGetI32v(reader, &info->offsetCompressedSize))) return code;
×
4558
    } else {
4559
      info->offsetOriginalSize = 0;
16,102✔
4560
      info->offsetCompressedSize = 0;
16,102✔
4561
    }
4562
    if ((code = tBufferGetI32v(reader, &info->dataOriginalSize))) return code;
16,102!
4563
    if ((code = tBufferGetI32v(reader, &info->dataCompressedSize))) return code;
16,101!
4564
  } else {
UNCOV
4565
    return TSDB_CODE_INVALID_PARA;
×
4566
  }
4567

4568
  return 0;
16,102✔
4569
}
4570

4571
int32_t tCompressData(void          *input,       // input
78,942✔
4572
                      SCompressInfo *info,        // compress info
4573
                      void          *output,      // output
4574
                      int32_t        outputSize,  // output size
4575
                      SBuffer       *buffer       // assistant buffer provided by caller, can be NULL
4576
) {
4577
  int32_t extraSizeNeeded;
4578
  int32_t code;
4579

4580
  extraSizeNeeded = (info->cmprAlg == NO_COMPRESSION) ? info->originalSize : info->originalSize + COMP_OVERFLOW_BYTES;
78,942!
4581
  if (!(outputSize >= extraSizeNeeded)) {
78,942!
UNCOV
4582
    return TSDB_CODE_INVALID_PARA;
×
4583
  }
4584

4585
  if (info->cmprAlg == NO_COMPRESSION) {
78,942!
4586
    (void)memcpy(output, input, info->originalSize);
×
UNCOV
4587
    info->compressedSize = info->originalSize;
×
4588
  } else if (info->cmprAlg == ONE_STAGE_COMP || info->cmprAlg == TWO_STAGE_COMP) {
80,374✔
4589
    SBuffer local;
4590

4591
    tBufferInit(&local);
4592
    if (buffer == NULL) {
1,433!
UNCOV
4593
      buffer = &local;
×
4594
    }
4595

4596
    if (info->cmprAlg == TWO_STAGE_COMP) {
1,433✔
4597
      code = tBufferEnsureCapacity(buffer, extraSizeNeeded);
1,432✔
4598
      if (code) {
1,432!
4599
        tBufferDestroy(&local);
UNCOV
4600
        return code;
×
4601
      }
4602
    }
4603

4604
    info->compressedSize = tDataTypes[info->dataType].compFunc(  //
2,865✔
4605
        input,                                                   // input
4606
        info->originalSize,                                      // input size
4607
        info->originalSize / tDataTypes[info->dataType].bytes,   // number of elements
1,433✔
4608
        output,                                                  // output
4609
        outputSize,                                              // output size
4610
        info->cmprAlg,                                           // compression algorithm
1,433✔
4611
        buffer->data,                                            // buffer
4612
        buffer->capacity                                         // buffer size
1,433✔
4613
    );
4614
    if (info->compressedSize < 0) {
1,432!
4615
      tBufferDestroy(&local);
UNCOV
4616
      return TSDB_CODE_COMPRESS_ERROR;
×
4617
    }
4618

4619
    tBufferDestroy(&local);
4620
  } else {
4621
    DEFINE_VAR(info->cmprAlg)
77,509✔
4622
    if ((l1 == L1_UNKNOWN && l2 == L2_UNKNOWN) || (l1 == L1_DISABLED && l2 == L2_DISABLED)) {
77,509!
4623
      (void)memcpy(output, input, info->originalSize);
×
4624
      info->compressedSize = info->originalSize;
×
UNCOV
4625
      return 0;
×
4626
    }
4627
    SBuffer local;
4628

4629
    tBufferInit(&local);
4630
    if (buffer == NULL) {
77,509!
UNCOV
4631
      buffer = &local;
×
4632
    }
4633
    code = tBufferEnsureCapacity(buffer, extraSizeNeeded);
77,509✔
4634

4635
    info->compressedSize = tDataCompress[info->dataType].compFunc(  //
155,018✔
4636
        input,                                                      // input
4637
        info->originalSize,                                         // input size
4638
        info->originalSize / tDataTypes[info->dataType].bytes,      // number of elements
77,509✔
4639
        output,                                                     // output
4640
        outputSize,                                                 // output size
4641
        info->cmprAlg,                                              // compression algorithm
4642
        buffer->data,                                               // buffer
4643
        buffer->capacity                                            // buffer size
77,509✔
4644
    );
4645
    if (info->compressedSize < 0) {
77,509!
4646
      tBufferDestroy(&local);
UNCOV
4647
      return TSDB_CODE_COMPRESS_ERROR;
×
4648
    }
4649

4650
    tBufferDestroy(&local);
4651
    // new col compress
4652
  }
4653

4654
  return 0;
78,941✔
4655
}
4656

4657
int32_t tDecompressData(void                *input,       // input
260,899✔
4658
                        const SCompressInfo *info,        // compress info
4659
                        void                *output,      // output
4660
                        int32_t              outputSize,  // output size
4661
                        SBuffer             *buffer       // assistant buffer provided by caller, can be NULL
4662
) {
4663
  int32_t code;
4664

4665
  if (!(outputSize >= info->originalSize)) {
260,899!
UNCOV
4666
    return TSDB_CODE_INVALID_PARA;
×
4667
  }
4668

4669
  if (info->cmprAlg == NO_COMPRESSION) {
260,899!
4670
    if (!(info->compressedSize == info->originalSize)) {
×
UNCOV
4671
      return TSDB_CODE_INVALID_PARA;
×
4672
    }
UNCOV
4673
    (void)memcpy(output, input, info->compressedSize);
×
4674
  } else if (info->cmprAlg == ONE_STAGE_COMP || info->cmprAlg == TWO_STAGE_COMP) {
429,517!
4675
    SBuffer local;
4676

4677
    tBufferInit(&local);
4678
    if (buffer == NULL) {
168,596!
UNCOV
4679
      buffer = &local;
×
4680
    }
4681

4682
    if (info->cmprAlg == TWO_STAGE_COMP) {
168,596!
4683
      code = tBufferEnsureCapacity(buffer, info->originalSize + COMP_OVERFLOW_BYTES);
168,629✔
4684
      if (code) {
168,626!
4685
        tBufferDestroy(&local);
UNCOV
4686
        return code;
×
4687
      }
4688
    }
4689

4690
    int32_t decompressedSize = tDataTypes[info->dataType].decompFunc(
168,593✔
4691
        input,                                                  // input
4692
        info->compressedSize,                                   // inputSize
168,593✔
4693
        info->originalSize / tDataTypes[info->dataType].bytes,  // number of elements
168,593✔
4694
        output,                                                 // output
4695
        outputSize,                                             // output size
4696
        info->cmprAlg,                                          // compression algorithm
168,593✔
4697
        buffer->data,                                           // helper buffer
4698
        buffer->capacity                                        // extra buffer size
168,593✔
4699
    );
4700
    if (decompressedSize < 0) {
168,618!
4701
      tBufferDestroy(&local);
UNCOV
4702
      return TSDB_CODE_COMPRESS_ERROR;
×
4703
    }
4704

4705
    if (!(decompressedSize == info->originalSize)) {
168,618!
UNCOV
4706
      return TSDB_CODE_COMPRESS_ERROR;
×
4707
    }
4708
    tBufferDestroy(&local);
4709
  } else {
4710
    DEFINE_VAR(info->cmprAlg);
92,303✔
4711
    if (l1 == L1_DISABLED && l2 == L2_DISABLED) {
92,303!
4712
      (void)memcpy(output, input, info->compressedSize);
×
UNCOV
4713
      return 0;
×
4714
    }
4715
    SBuffer local;
4716

4717
    tBufferInit(&local);
4718
    if (buffer == NULL) {
92,303!
UNCOV
4719
      buffer = &local;
×
4720
    }
4721
    code = tBufferEnsureCapacity(buffer, info->originalSize + COMP_OVERFLOW_BYTES);
92,303✔
4722
    if (code) {
92,304!
UNCOV
4723
      return code;
×
4724
    }
4725

4726
    int32_t decompressedSize = tDataCompress[info->dataType].decompFunc(
92,304✔
4727
        input,                                                  // input
4728
        info->compressedSize,                                   // inputSize
92,304✔
4729
        info->originalSize / tDataTypes[info->dataType].bytes,  // number of elements
92,304✔
4730
        output,                                                 // output
4731
        outputSize,                                             // output size
4732
        info->cmprAlg,                                          // compression algorithm
92,304✔
4733
        buffer->data,                                           // helper buffer
4734
        buffer->capacity                                        // extra buffer size
92,304✔
4735
    );
4736
    if (decompressedSize < 0) {
92,299!
4737
      tBufferDestroy(&local);
UNCOV
4738
      return TSDB_CODE_COMPRESS_ERROR;
×
4739
    }
4740

4741
    if (!(decompressedSize == info->originalSize)) {
92,299!
UNCOV
4742
      return TSDB_CODE_COMPRESS_ERROR;
×
4743
    }
4744
    tBufferDestroy(&local);
4745
  }
4746

4747
  return 0;
260,917✔
4748
}
4749

4750
int32_t tCompressDataToBuffer(void *input, SCompressInfo *info, SBuffer *output, SBuffer *assist) {
78,940✔
4751
  int32_t code;
4752

4753
  code = tBufferEnsureCapacity(output, output->size + info->originalSize + COMP_OVERFLOW_BYTES);
78,940✔
4754
  if (code) return code;
78,940!
4755

4756
  code = tCompressData(input, info, tBufferGetDataEnd(output), output->capacity - output->size, assist);
78,940✔
4757
  if (code) return code;
78,941!
4758

4759
  output->size += info->compressedSize;
78,941✔
4760
  return 0;
78,941✔
4761
}
4762

4763
int32_t tDecompressDataToBuffer(void *input, SCompressInfo *info, SBuffer *output, SBuffer *assist) {
168,619✔
4764
  int32_t code;
4765

4766
  code = tBufferEnsureCapacity(output, output->size + info->originalSize);
168,619✔
4767
  if (code) return code;
168,623!
4768

4769
  code = tDecompressData(input, info, tBufferGetDataEnd(output), output->capacity - output->size, assist);
168,623✔
4770
  if (code) return code;
168,621!
4771

4772
  output->size += info->originalSize;
168,621✔
4773
  return 0;
168,621✔
4774
}
4775

4776
// handle all types, including var data
4777
void valueSetDatum(SValue *pVal, int8_t type, void *pDatum, uint32_t len) {
446,743,967✔
4778
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
446,743,967!
4779
    pVal->pData = pDatum;
×
UNCOV
4780
    pVal->nData = len;
×
4781
  } else {
4782
    switch (len) {
448,058,352!
4783
      case sizeof(uint8_t):
43,793✔
4784
        pVal->val = *(uint8_t *)pDatum;
43,793✔
4785
        break;
43,793✔
4786
      case sizeof(uint16_t):
21,888✔
4787
        pVal->val = *(uint16_t *)pDatum;
21,888✔
4788
        break;
21,888✔
4789
      case sizeof(uint32_t):
77,720,754✔
4790
        pVal->val = *(uint32_t *)pDatum;
77,720,754✔
4791
        break;
77,720,754✔
4792
      case sizeof(uint64_t):
370,822,749✔
4793
        pVal->val = *(uint64_t *)pDatum;
370,822,749✔
4794
        break;
370,822,749✔
4795
      default:
×
UNCOV
4796
        break;
×
4797
    }
4798
  }
4799
}
446,743,967✔
4800

4801
void valueCloneDatum(SValue *pDst, const SValue *pSrc, int8_t type) {
167,040✔
4802
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
167,040!
4803
    memcpy(pDst->pData, pSrc->pData, pSrc->nData);
×
UNCOV
4804
    pDst->nData = pSrc->nData;
×
4805
  } else {
4806
    pDst->val = pSrc->val;
167,044✔
4807
  }
4808
}
167,040✔
4809
void valueClearDatum(SValue *pVal, int8_t type) {
×
4810
  if (IS_VAR_DATA_TYPE(type) || type == TSDB_DATA_TYPE_DECIMAL) {
×
4811
    taosMemoryFreeClear(pVal->pData);
×
UNCOV
4812
    pVal->nData = 0;
×
4813
  } else {
UNCOV
4814
    pVal->val = 0;
×
4815
  }
UNCOV
4816
}
×
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