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

#define _DEFAULT_SOURCE
S
slguan 已提交
17
#include "tdataformat.h"
H
refact  
Hongze Cheng 已提交
18
#include "tRealloc.h"
S
Shengliang Guan 已提交
19
#include "tcoding.h"
L
Liu Jicong 已提交
20
#include "tdatablock.h"
S
log  
Shengliang Guan 已提交
21
#include "tlog.h"
H
more  
hzcheng 已提交
22

C
Cary Xu 已提交
23 24
static int32_t tGetTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson);

H
Hongze Cheng 已提交
25 26 27
#pragma pack(push, 1)
typedef struct {
  int16_t nCols;
H
more  
Hongze Cheng 已提交
28
  uint8_t idx[];
H
Hongze Cheng 已提交
29 30 31
} STSKVRow;
#pragma pack(pop)

H
Hongze Cheng 已提交
32 33
#define TSROW_IS_KV_ROW(r) ((r)->flags & TSROW_KV_ROW)

H
Hongze Cheng 已提交
34
// SValue
H
Hongze Cheng 已提交
35
int32_t tPutValue(uint8_t *p, SValue *pValue, int8_t type) {
H
Hongze Cheng 已提交
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
  int32_t n = 0;

  if (IS_VAR_DATA_TYPE(type)) {
    n += tPutBinary(p ? p + n : p, pValue->pData, pValue->nData);
  } else {
    switch (type) {
      case TSDB_DATA_TYPE_BOOL:
        n += tPutI8(p ? p + n : p, pValue->i8 ? 1 : 0);
        break;
      case TSDB_DATA_TYPE_TINYINT:
        n += tPutI8(p ? p + n : p, pValue->i8);
        break;
      case TSDB_DATA_TYPE_SMALLINT:
        n += tPutI16(p ? p + n : p, pValue->i16);
        break;
      case TSDB_DATA_TYPE_INT:
        n += tPutI32(p ? p + n : p, pValue->i32);
        break;
      case TSDB_DATA_TYPE_BIGINT:
        n += tPutI64(p ? p + n : p, pValue->i64);
        break;
      case TSDB_DATA_TYPE_FLOAT:
        n += tPutFloat(p ? p + n : p, pValue->f);
        break;
      case TSDB_DATA_TYPE_DOUBLE:
        n += tPutDouble(p ? p + n : p, pValue->d);
        break;
      case TSDB_DATA_TYPE_TIMESTAMP:
        n += tPutI64(p ? p + n : p, pValue->ts);
        break;
      case TSDB_DATA_TYPE_UTINYINT:
        n += tPutU8(p ? p + n : p, pValue->u8);
        break;
      case TSDB_DATA_TYPE_USMALLINT:
        n += tPutU16(p ? p + n : p, pValue->u16);
        break;
      case TSDB_DATA_TYPE_UINT:
        n += tPutU32(p ? p + n : p, pValue->u32);
        break;
      case TSDB_DATA_TYPE_UBIGINT:
        n += tPutU64(p ? p + n : p, pValue->u64);
        break;
      default:
        ASSERT(0);
    }
  }

  return n;
}

H
Hongze Cheng 已提交
86
int32_t tGetValue(uint8_t *p, SValue *pValue, int8_t type) {
H
Hongze Cheng 已提交
87 88 89
  int32_t n = 0;

  if (IS_VAR_DATA_TYPE(type)) {
H
Hongze Cheng 已提交
90
    n += tGetBinary(p, &pValue->pData, pValue ? &pValue->nData : NULL);
H
Hongze Cheng 已提交
91 92 93
  } else {
    switch (type) {
      case TSDB_DATA_TYPE_BOOL:
H
Hongze Cheng 已提交
94
        n += tGetI8(p, &pValue->i8);
H
Hongze Cheng 已提交
95 96
        break;
      case TSDB_DATA_TYPE_TINYINT:
H
Hongze Cheng 已提交
97
        n += tGetI8(p, &pValue->i8);
H
Hongze Cheng 已提交
98 99
        break;
      case TSDB_DATA_TYPE_SMALLINT:
H
Hongze Cheng 已提交
100
        n += tGetI16(p, &pValue->i16);
H
Hongze Cheng 已提交
101 102
        break;
      case TSDB_DATA_TYPE_INT:
H
Hongze Cheng 已提交
103
        n += tGetI32(p, &pValue->i32);
H
Hongze Cheng 已提交
104 105
        break;
      case TSDB_DATA_TYPE_BIGINT:
H
Hongze Cheng 已提交
106
        n += tGetI64(p, &pValue->i64);
H
Hongze Cheng 已提交
107 108
        break;
      case TSDB_DATA_TYPE_FLOAT:
H
Hongze Cheng 已提交
109
        n += tGetFloat(p, &pValue->f);
H
Hongze Cheng 已提交
110 111
        break;
      case TSDB_DATA_TYPE_DOUBLE:
H
Hongze Cheng 已提交
112
        n += tGetDouble(p, &pValue->d);
H
Hongze Cheng 已提交
113 114
        break;
      case TSDB_DATA_TYPE_TIMESTAMP:
H
Hongze Cheng 已提交
115
        n += tGetI64(p, &pValue->ts);
H
Hongze Cheng 已提交
116 117
        break;
      case TSDB_DATA_TYPE_UTINYINT:
H
Hongze Cheng 已提交
118
        n += tGetU8(p, &pValue->u8);
H
Hongze Cheng 已提交
119 120
        break;
      case TSDB_DATA_TYPE_USMALLINT:
H
Hongze Cheng 已提交
121
        n += tGetU16(p, &pValue->u16);
H
Hongze Cheng 已提交
122 123
        break;
      case TSDB_DATA_TYPE_UINT:
H
Hongze Cheng 已提交
124
        n += tGetU32(p, &pValue->u32);
H
Hongze Cheng 已提交
125 126
        break;
      case TSDB_DATA_TYPE_UBIGINT:
H
Hongze Cheng 已提交
127
        n += tGetU64(p, &pValue->u64);
H
Hongze Cheng 已提交
128 129 130 131 132 133 134 135 136
        break;
      default:
        ASSERT(0);
    }
  }

  return n;
}

H
more  
Hongze Cheng 已提交
137 138 139 140 141
int tValueCmprFn(const SValue *pValue1, const SValue *pValue2, int8_t type) {
  // TODO
  return 0;
}

H
Hongze Cheng 已提交
142
// STSRow2 ========================================================================
H
Hongze Cheng 已提交
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
static void setBitMap(uint8_t *pb, uint8_t v, int32_t idx, uint8_t flags) {
  if (pb) {
    switch (flags & 0xf) {
      case TSROW_HAS_NULL | TSROW_HAS_NONE:
      case TSROW_HAS_VAL | TSROW_HAS_NONE:
        if (v) {
          SET_BIT1(pb, idx, (uint8_t)1);
        } else {
          SET_BIT1(pb, idx, (uint8_t)0);
        }
        break;
      case TSROW_HAS_VAL | TSROW_HAS_NULL:
        v = v - 1;
        SET_BIT1(pb, idx, v);
        break;
      case TSROW_HAS_VAL | TSROW_HAS_NULL | TSROW_HAS_NONE:
        SET_BIT2(pb, idx, v);
        break;
H
more  
Hongze Cheng 已提交
161

H
Hongze Cheng 已提交
162 163
      default:
        ASSERT(0);
H
more  
Hongze Cheng 已提交
164 165
    }
  }
H
Hongze Cheng 已提交
166 167 168 169 170 171 172 173 174 175 176
}
#define SET_IDX(p, i, n, f)        \
  do {                             \
    if ((f)&TSROW_KV_SMALL) {      \
      ((uint8_t *)(p))[i] = (n);   \
    } else if ((f)&TSROW_KV_MID) { \
      ((uint16_t *)(p))[i] = (n);  \
    } else {                       \
      ((uint32_t *)(p))[i] = (n);  \
    }                              \
  } while (0)
H
more  
Hongze Cheng 已提交
177

H
Hongze Cheng 已提交
178
int32_t tTSRowNew(STSRowBuilder *pBuilder, SArray *pArray, STSchema *pTSchema, STSRow2 **ppRow) {
H
Hongze Cheng 已提交
179 180
  int32_t code = 0;
#if 0
H
Hongze Cheng 已提交
181 182 183 184
  STColumn *pTColumn;
  SColVal  *pColVal;
  int32_t   nColVal = taosArrayGetSize(pArray);
  int32_t   iColVal;
H
more  
Hongze Cheng 已提交
185

H
Hongze Cheng 已提交
186
  ASSERT(nColVal > 0);
H
more  
Hongze Cheng 已提交
187

H
Hongze Cheng 已提交
188 189 190 191 192 193
  // try
  uint8_t  flags = 0;
  uint32_t ntv = 0;
  uint32_t nkv = 0;
  int16_t  nTag = 0;
  uint32_t maxIdx = 0;
H
more  
Hongze Cheng 已提交
194

H
Hongze Cheng 已提交
195 196
  iColVal = 0;
  for (int32_t iColumn = 0; iColumn < pTSchema->numOfCols; iColumn++) {
H
more  
Hongze Cheng 已提交
197 198 199 200 201 202 203
    pTColumn = &pTSchema->columns[iColumn];
    if (iColVal < nColVal) {
      pColVal = (SColVal *)taosArrayGet(pArray, iColVal);
    } else {
      pColVal = NULL;
    }

H
Hongze Cheng 已提交
204 205 206
    if (iColumn == 0) {
      ASSERT(pColVal->cid == pTColumn->colId);
      ASSERT(pTColumn->type == TSDB_DATA_TYPE_TIMESTAMP);
C
Cary Xu 已提交
207
      ASSERT(pTColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID);
H
Hongze Cheng 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233

      iColVal++;
    } else {
      if (pColVal) {
        if (pColVal->cid == pTColumn->colId) {
          iColVal++;

          if (pColVal->isNone) {
            flags |= TSROW_HAS_NONE;
          } else if (pColVal->isNull) {
            flags |= TSROW_HAS_NULL;
            maxIdx = nkv;
            nTag++;
            nkv += tPutI16v(NULL, -pTColumn->colId);
          } else {
            flags |= TSROW_HAS_VAL;
            maxIdx = nkv;
            nTag++;
            nkv += tPutI16v(NULL, pTColumn->colId);
            nkv += tPutValue(NULL, &pColVal->value, pTColumn->type);
            if (IS_VAR_DATA_TYPE(pTColumn->type)) {
              ntv += tPutValue(NULL, &pColVal->value, pTColumn->type);
            }
          }
        } else if (pColVal->cid > pTColumn->colId) {
          flags |= TSROW_HAS_NONE;
H
more  
Hongze Cheng 已提交
234
        } else {
H
Hongze Cheng 已提交
235
          ASSERT(0);
H
more  
Hongze Cheng 已提交
236 237
        }
      } else {
H
Hongze Cheng 已提交
238
        flags |= TSROW_HAS_NONE;
H
more  
Hongze Cheng 已提交
239 240
      }
    }
H
Hongze Cheng 已提交
241
  }
H
more  
Hongze Cheng 已提交
242

C
Cary Xu 已提交
243
  ASSERT(flags);
H
more  
Hongze Cheng 已提交
244

H
Hongze Cheng 已提交
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
  // decide
  uint32_t nData = 0;
  uint32_t nDataT = 0;
  uint32_t nDataK = 0;
  if (flags == TSROW_HAS_NONE || flags == TSROW_HAS_NULL) {
    nData = 0;
  } else {
    switch (flags) {
      case TSROW_HAS_VAL:
        nDataT = pTSchema->flen + ntv;
        break;
      case TSROW_HAS_NULL | TSROW_HAS_NONE:
        nDataT = BIT1_SIZE(pTSchema->numOfCols - 1);
        break;
      case TSROW_HAS_VAL | TSROW_HAS_NONE:
      case TSROW_HAS_VAL | TSROW_HAS_NULL:
        nDataT = BIT1_SIZE(pTSchema->numOfCols - 1) + pTSchema->flen + ntv;
        break;
      case TSROW_HAS_VAL | TSROW_HAS_NULL | TSROW_HAS_NONE:
        nDataT = BIT2_SIZE(pTSchema->numOfCols - 1) + pTSchema->flen + ntv;
        break;
      default:
C
Cary Xu 已提交
267 268
        break;
        ASSERT(0);
H
more  
Hongze Cheng 已提交
269 270
    }

H
Hongze Cheng 已提交
271 272 273 274 275 276 277 278 279 280
    uint8_t tflags = 0;
    if (maxIdx <= UINT8_MAX) {
      nDataK = sizeof(STSKVRow) + sizeof(uint8_t) * nTag + nkv;
      tflags |= TSROW_KV_SMALL;
    } else if (maxIdx <= UINT16_MAX) {
      nDataK = sizeof(STSKVRow) + sizeof(uint16_t) * nTag + nkv;
      tflags |= TSROW_KV_MID;
    } else {
      nDataK = sizeof(STSKVRow) + sizeof(uint32_t) * nTag + nkv;
      tflags |= TSROW_KV_BIG;
H
more  
Hongze Cheng 已提交
281 282
    }

C
Cary Xu 已提交
283
    if (nDataT <= nDataK) {
H
Hongze Cheng 已提交
284
      nData = nDataT;
H
more  
Hongze Cheng 已提交
285
    } else {
H
Hongze Cheng 已提交
286 287
      nData = nDataK;
      flags |= tflags;
H
more  
Hongze Cheng 已提交
288 289
    }
  }
H
more  
Hongze Cheng 已提交
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307

  // alloc
  if (pBuilder) {
    // create from a builder
    if (nData == 0) {
      pBuilder->tsRow.nData = 0;
      pBuilder->tsRow.pData = NULL;
    } else {
      if (pBuilder->szBuf < nData) {
        uint8_t *p = taosMemoryRealloc(pBuilder->pBuf, nData);
        if (p == NULL) {
          code = TSDB_CODE_OUT_OF_MEMORY;
          goto _exit;
        }
        pBuilder->pBuf = p;
        pBuilder->szBuf = nData;
      }

H
Hongze Cheng 已提交
308
      pBuilder->tsRow.nData = nData;
H
more  
Hongze Cheng 已提交
309 310 311 312 313 314 315 316
      pBuilder->tsRow.pData = pBuilder->pBuf;
    }

    *ppRow = &pBuilder->tsRow;
  } else {
    // create a new one
    *ppRow = (STSRow2 *)taosMemoryMalloc(sizeof(STSRow2));
    if (*ppRow == NULL) {
H
more  
Hongze Cheng 已提交
317 318 319
      code = TSDB_CODE_OUT_OF_MEMORY;
      goto _exit;
    }
H
Hongze Cheng 已提交
320 321 322 323 324
    if (nData == 0) {
      (*ppRow)->nData = 0;
      (*ppRow)->pData = NULL;
    } else {
      (*ppRow)->nData = nData;
H
more  
Hongze Cheng 已提交
325 326
      (*ppRow)->pData = taosMemoryMalloc(nData);
      if ((*ppRow)->pData == NULL) {
H
Hongze Cheng 已提交
327
        taosMemoryFree(*ppRow);
H
more  
Hongze Cheng 已提交
328 329 330 331
        code = TSDB_CODE_OUT_OF_MEMORY;
        goto _exit;
      }
    }
H
more  
Hongze Cheng 已提交
332
  }
H
more  
Hongze Cheng 已提交
333

H
Hongze Cheng 已提交
334 335 336
  // build
  (*ppRow)->flags = flags;
  (*ppRow)->sver = pTSchema->version;
H
more  
Hongze Cheng 已提交
337

H
Hongze Cheng 已提交
338 339 340 341 342 343 344 345 346 347 348 349 350 351
  pColVal = (SColVal *)taosArrayGet(pArray, 0);
  (*ppRow)->ts = pColVal->value.ts;

  if ((*ppRow)->pData) {
    STSKVRow *pTSKVRow = NULL;
    uint8_t  *pidx = NULL;
    uint8_t  *pkv = NULL;
    uint8_t  *pb = NULL;
    uint8_t  *pf = NULL;
    uint8_t  *ptv = NULL;
    nkv = 0;
    ntv = 0;
    iColVal = 1;

C
Cary Xu 已提交
352
    if ((flags & 0xf0) == 0) {
H
Hongze Cheng 已提交
353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372
      switch (flags & 0xf) {
        case TSROW_HAS_VAL:
          pf = (*ppRow)->pData;
          ptv = pf + pTSchema->flen;
          break;
        case TSROW_HAS_NULL | TSROW_HAS_NONE:
          pb = (*ppRow)->pData;
          break;
        case TSROW_HAS_VAL | TSROW_HAS_NONE:
        case TSROW_HAS_VAL | TSROW_HAS_NULL:
          pb = (*ppRow)->pData;
          pf = pb + BIT1_SIZE(pTSchema->numOfCols - 1);
          ptv = pf + pTSchema->flen;
          break;
        case TSROW_HAS_VAL | TSROW_HAS_NULL | TSROW_HAS_NONE:
          pb = (*ppRow)->pData;
          pf = pb + BIT2_SIZE(pTSchema->numOfCols - 1);
          ptv = pf + pTSchema->flen;
          break;
        default:
C
Cary Xu 已提交
373
          ASSERT(0);
C
Cary Xu 已提交
374
          break;
H
Hongze Cheng 已提交
375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
      }
    } else {
      pTSKVRow = (STSKVRow *)(*ppRow)->pData;
      pTSKVRow->nCols = 0;
      pidx = pTSKVRow->idx;
      if (flags & TSROW_KV_SMALL) {
        pkv = pidx + sizeof(uint8_t) * nTag;
      } else if (flags & TSROW_KV_MID) {
        pkv = pidx + sizeof(uint16_t) * nTag;
      } else {
        pkv = pidx + sizeof(uint32_t) * nTag;
      }
    }

    for (int32_t iColumn = 1; iColumn < pTSchema->numOfCols; iColumn++) {
      pTColumn = &pTSchema->columns[iColumn];
      if (iColVal < nColVal) {
        pColVal = (SColVal *)taosArrayGet(pArray, iColVal);
      } else {
        pColVal = NULL;
      }

      if (pColVal) {
        if (pColVal->cid == pTColumn->colId) {
          iColVal++;

          if (pColVal->isNone) {
            goto _set_none;
          } else if (pColVal->isNull) {
            goto _set_null;
          } else {
            goto _set_value;
          }
        } else if (pColVal->cid > pTColumn->colId) {
          goto _set_none;
        } else {
          ASSERT(0);
        }
      } else {
        goto _set_none;
      }

    _set_none:
C
Cary Xu 已提交
418
      if ((flags & 0xf0) == 0) {
H
Hongze Cheng 已提交
419
        setBitMap(pb, 0, iColumn - 1, flags);
H
Hongze Cheng 已提交
420
        if (flags & TSROW_HAS_VAL) {  // set 0
C
Cary Xu 已提交
421 422 423 424 425 426
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
            *(VarDataOffsetT *)(pf + pTColumn->offset) = 0;
          } else {
            tPutValue(pf + pTColumn->offset, &((SValue){0}), pTColumn->type);
          }
        }
H
Hongze Cheng 已提交
427 428 429 430
      }
      continue;

    _set_null:
C
Cary Xu 已提交
431
      if ((flags & 0xf0) == 0) {
H
Hongze Cheng 已提交
432
        setBitMap(pb, 1, iColumn - 1, flags);
H
Hongze Cheng 已提交
433
        if (flags & TSROW_HAS_VAL) {  // set 0
C
Cary Xu 已提交
434 435 436 437 438 439
          if (IS_VAR_DATA_TYPE(pTColumn->type)) {
            *(VarDataOffsetT *)(pf + pTColumn->offset) = 0;
          } else {
            tPutValue(pf + pTColumn->offset, &((SValue){0}), pTColumn->type);
          }
        }
H
Hongze Cheng 已提交
440 441 442 443 444 445
      } else {
        SET_IDX(pidx, pTSKVRow->nCols, nkv, flags);
        pTSKVRow->nCols++;
        nkv += tPutI16v(pkv + nkv, -pTColumn->colId);
      }
      continue;
H
more  
Hongze Cheng 已提交
446

H
Hongze Cheng 已提交
447
    _set_value:
C
Cary Xu 已提交
448
      if ((flags & 0xf0) == 0) {
H
Hongze Cheng 已提交
449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464
        setBitMap(pb, 2, iColumn - 1, flags);

        if (IS_VAR_DATA_TYPE(pTColumn->type)) {
          *(VarDataOffsetT *)(pf + pTColumn->offset) = ntv;
          ntv += tPutValue(ptv + ntv, &pColVal->value, pTColumn->type);
        } else {
          tPutValue(pf + pTColumn->offset, &pColVal->value, pTColumn->type);
        }
      } else {
        SET_IDX(pidx, pTSKVRow->nCols, nkv, flags);
        pTSKVRow->nCols++;
        nkv += tPutI16v(pkv + nkv, pColVal->cid);
        nkv += tPutValue(pkv + nkv, &pColVal->value, pTColumn->type);
      }
      continue;
    }
H
more  
Hongze Cheng 已提交
465 466
  }

H
Hongze Cheng 已提交
467
#endif
H
more  
Hongze Cheng 已提交
468
_exit:
H
Hongze Cheng 已提交
469 470 471
  return code;
}

H
Hongze Cheng 已提交
472 473
int32_t tTSRowClone(const STSRow2 *pRow, STSRow2 **ppRow) {
  int32_t code = 0;
H
Hongze Cheng 已提交
474
  int32_t rLen;
H
Hongze Cheng 已提交
475

H
Hongze Cheng 已提交
476 477
  TSROW_LEN(pRow, rLen);
  (*ppRow) = (STSRow2 *)taosMemoryMalloc(rLen);
H
Hongze Cheng 已提交
478
  if (*ppRow == NULL) {
H
Hongze Cheng 已提交
479 480
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
H
Hongze Cheng 已提交
481
  }
H
Hongze Cheng 已提交
482
  memcpy(*ppRow, pRow, rLen);
H
Hongze Cheng 已提交
483

H
Hongze Cheng 已提交
484 485
_exit:
  return code;
H
Hongze Cheng 已提交
486 487 488
}

void tTSRowFree(STSRow2 *pRow) {
H
Hongze Cheng 已提交
489 490 491
  if (pRow) {
    taosMemoryFree(pRow);
  }
H
Hongze Cheng 已提交
492 493
}

H
Hongze Cheng 已提交
494
void tTSRowGet(STSRow2 *pRow, STSchema *pTSchema, int32_t iCol, SColVal *pColVal) {
H
Hongze Cheng 已提交
495
#if 0
C
Cary Xu 已提交
496
  uint8_t   isTuple = ((pRow->flags & 0xf0) == 0) ? 1 : 0;
H
Hongze Cheng 已提交
497
  STColumn *pTColumn = &pTSchema->columns[iCol];
H
Hongze Cheng 已提交
498 499
  uint8_t   flags = pRow->flags & (uint8_t)0xf;
  SValue    value;
H
Hongze Cheng 已提交
500

H
Hongze Cheng 已提交
501
  ASSERT(iCol < pTSchema->numOfCols);
C
Cary Xu 已提交
502
  ASSERT(flags);
H
Hongze Cheng 已提交
503
  ASSERT(pRow->sver == pTSchema->version);
H
Hongze Cheng 已提交
504

H
Hongze Cheng 已提交
505 506 507
  if (iCol == 0) {
    value.ts = pRow->ts;
    goto _return_value;
H
Hongze Cheng 已提交
508 509
  }

H
Hongze Cheng 已提交
510
  if (flags == TSROW_HAS_NONE) {
H
more  
Hongze Cheng 已提交
511
    goto _return_none;
C
Cary Xu 已提交
512
  } else if (flags == TSROW_HAS_NULL) {
H
more  
Hongze Cheng 已提交
513
    goto _return_null;
H
Hongze Cheng 已提交
514
  }
H
Hongze Cheng 已提交
515

H
Hongze Cheng 已提交
516 517 518 519 520 521 522 523 524 525 526 527 528 529
  ASSERT(pRow->nData && pRow->pData);

  if (isTuple) {
    uint8_t *pb = pRow->pData;
    uint8_t *pf = NULL;
    uint8_t *pv = NULL;
    uint8_t *p;
    uint8_t  b;

    // bit
    switch (flags) {
      case TSROW_HAS_VAL:
        pf = pb;
        break;
H
Hongze Cheng 已提交
530
      case TSROW_HAS_NULL | TSROW_HAS_NONE:
H
Hongze Cheng 已提交
531 532 533
        b = GET_BIT1(pb, iCol - 1);
        if (b == 0) {
          goto _return_none;
H
Hongze Cheng 已提交
534
        } else {
H
Hongze Cheng 已提交
535
          goto _return_null;
H
Hongze Cheng 已提交
536
        }
H
Hongze Cheng 已提交
537
      case TSROW_HAS_VAL | TSROW_HAS_NONE:
H
Hongze Cheng 已提交
538 539 540
        b = GET_BIT1(pb, iCol - 1);
        if (b == 0) {
          goto _return_none;
H
Hongze Cheng 已提交
541
        } else {
H
Hongze Cheng 已提交
542 543
          pf = pb + BIT1_SIZE(pTSchema->numOfCols - 1);
          break;
H
Hongze Cheng 已提交
544
        }
H
Hongze Cheng 已提交
545
      case TSROW_HAS_VAL | TSROW_HAS_NULL:
H
Hongze Cheng 已提交
546 547 548
        b = GET_BIT1(pb, iCol - 1);
        if (b == 0) {
          goto _return_null;
H
Hongze Cheng 已提交
549
        } else {
H
Hongze Cheng 已提交
550 551
          pf = pb + BIT1_SIZE(pTSchema->numOfCols - 1);
          break;
H
Hongze Cheng 已提交
552
        }
H
Hongze Cheng 已提交
553
      case TSROW_HAS_VAL | TSROW_HAS_NULL | TSROW_HAS_NONE:
H
Hongze Cheng 已提交
554 555 556 557 558
        b = GET_BIT2(pb, iCol - 1);
        if (b == 0) {
          goto _return_none;
        } else if (b == 1) {
          goto _return_null;
H
Hongze Cheng 已提交
559
        } else {
H
Hongze Cheng 已提交
560 561
          pf = pb + BIT2_SIZE(pTSchema->numOfCols - 1);
          break;
H
Hongze Cheng 已提交
562
        }
H
Hongze Cheng 已提交
563
      default:
H
Hongze Cheng 已提交
564
        ASSERT(0);
H
Hongze Cheng 已提交
565
    }
H
Hongze Cheng 已提交
566

H
Hongze Cheng 已提交
567 568 569
    ASSERT(pf);

    p = pf + pTColumn->offset;
H
Hongze Cheng 已提交
570
    if (IS_VAR_DATA_TYPE(pTColumn->type)) {
H
Hongze Cheng 已提交
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592
      pv = pf + pTSchema->flen;
      p = pv + *(VarDataOffsetT *)p;
    }
    tGetValue(p, &value, pTColumn->type);
    goto _return_value;
  } else {
    STSKVRow *pRowK = (STSKVRow *)pRow->pData;
    int16_t   lidx = 0;
    int16_t   ridx = pRowK->nCols - 1;
    uint8_t  *p;
    int16_t   midx;
    uint32_t  n;
    int16_t   cid;

    ASSERT(pRowK->nCols > 0);

    if (pRow->flags & TSROW_KV_SMALL) {
      p = pRow->pData + sizeof(STSKVRow) + sizeof(uint8_t) * pRowK->nCols;
    } else if (pRow->flags & TSROW_KV_MID) {
      p = pRow->pData + sizeof(STSKVRow) + sizeof(uint16_t) * pRowK->nCols;
    } else if (pRow->flags & TSROW_KV_BIG) {
      p = pRow->pData + sizeof(STSKVRow) + sizeof(uint32_t) * pRowK->nCols;
H
Hongze Cheng 已提交
593
    } else {
H
Hongze Cheng 已提交
594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622
      ASSERT(0);
    }
    while (lidx <= ridx) {
      midx = (lidx + ridx) / 2;

      if (pRow->flags & TSROW_KV_SMALL) {
        n = ((uint8_t *)pRowK->idx)[midx];
      } else if (pRow->flags & TSROW_KV_MID) {
        n = ((uint16_t *)pRowK->idx)[midx];
      } else {
        n = ((uint32_t *)pRowK->idx)[midx];
      }

      n += tGetI16v(p + n, &cid);

      if (TABS(cid) == pTColumn->colId) {
        if (cid < 0) {
          goto _return_null;
        } else {
          n += tGetValue(p + n, &value, pTColumn->type);
          goto _return_value;
        }

        return;
      } else if (TABS(cid) > pTColumn->colId) {
        ridx = midx - 1;
      } else {
        lidx = midx + 1;
      }
H
Hongze Cheng 已提交
623
    }
H
Hongze Cheng 已提交
624 625 626

    // not found, return NONE
    goto _return_none;
H
Hongze Cheng 已提交
627
  }
H
Hongze Cheng 已提交
628

H
Hongze Cheng 已提交
629
_return_none:
H
Hongze Cheng 已提交
630
  *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
H
Hongze Cheng 已提交
631 632 633
  return;

_return_null:
H
Hongze Cheng 已提交
634
  *pColVal = COL_VAL_NULL(pTColumn->colId, pTColumn->type);
H
Hongze Cheng 已提交
635 636 637
  return;

_return_value:
H
Hongze Cheng 已提交
638
  *pColVal = COL_VAL_VALUE(pTColumn->colId, pTColumn->type, value);
H
Hongze Cheng 已提交
639
  return;
H
Hongze Cheng 已提交
640
#endif
H
Hongze Cheng 已提交
641 642 643 644
}

int32_t tTSRowToArray(STSRow2 *pRow, STSchema *pTSchema, SArray **ppArray) {
  int32_t code = 0;
H
Hongze Cheng 已提交
645
#if 0
H
Hongze Cheng 已提交
646 647 648 649 650 651 652 653 654 655 656 657 658
  SColVal cv;

  (*ppArray) = taosArrayInit(pTSchema->numOfCols, sizeof(SColVal));
  if (*ppArray == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

  for (int32_t iColumn = 0; iColumn < pTSchema->numOfCols; iColumn++) {
    tTSRowGet(pRow, pTSchema, iColumn, &cv);
    taosArrayPush(*ppArray, &cv);
  }

H
Hongze Cheng 已提交
659
#endif
H
Hongze Cheng 已提交
660 661 662 663 664
_exit:
  return code;
}

int32_t tPutTSRow(uint8_t *p, STSRow2 *pRow) {
H
Hongze Cheng 已提交
665
  int32_t n;
H
Hongze Cheng 已提交
666

H
Hongze Cheng 已提交
667 668 669
  TSROW_LEN(pRow, n);
  if (p) {
    memcpy(p, pRow, n);
H
Hongze Cheng 已提交
670 671 672 673 674
  }

  return n;
}

H
Hongze Cheng 已提交
675 676
int32_t tGetTSRow(uint8_t *p, STSRow2 **ppRow) {
  int32_t n;
H
Hongze Cheng 已提交
677

H
Hongze Cheng 已提交
678 679
  *ppRow = (STSRow2 *)p;
  TSROW_LEN(*ppRow, n);
H
Hongze Cheng 已提交
680 681

  return n;
H
Hongze Cheng 已提交
682 683
}

H
Hongze Cheng 已提交
684
// STSchema ========================================
H
Hongze Cheng 已提交
685 686 687 688 689 690 691
int32_t tTSchemaCreate(int32_t sver, SSchema *pSchema, int32_t ncols, STSchema **ppTSchema) {
  *ppTSchema = (STSchema *)taosMemoryMalloc(sizeof(STSchema) + sizeof(STColumn) * ncols);
  if (*ppTSchema == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

H
Hongze Cheng 已提交
692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716
  (*ppTSchema)->numOfCols = ncols;
  (*ppTSchema)->version = sver;
  (*ppTSchema)->flen = 0;
  (*ppTSchema)->vlen = 0;
  (*ppTSchema)->tlen = 0;

  for (int32_t iCol = 0; iCol < ncols; iCol++) {
    SSchema  *pColumn = &pSchema[iCol];
    STColumn *pTColumn = &((*ppTSchema)->columns[iCol]);

    pTColumn->colId = pColumn->colId;
    pTColumn->type = pColumn->type;
    pTColumn->flags = pColumn->flags;
    pTColumn->bytes = pColumn->bytes;
    pTColumn->offset = (*ppTSchema)->flen;

    // skip first column
    if (iCol) {
      (*ppTSchema)->flen += TYPE_BYTES[pColumn->type];
      if (IS_VAR_DATA_TYPE(pColumn->type)) {
        (*ppTSchema)->vlen += (pColumn->bytes + 5);
      }
    }
  }

H
Hongze Cheng 已提交
717 718 719
  return 0;
}

H
Hongze Cheng 已提交
720 721 722
void tTSchemaDestroy(STSchema *pTSchema) {
  if (pTSchema) taosMemoryFree(pTSchema);
}
H
Hongze Cheng 已提交
723

H
Hongze Cheng 已提交
724
// STag ========================================
H
Hongze Cheng 已提交
725 726
static int tTagValCmprFn(const void *p1, const void *p2) {
  if (((STagVal *)p1)->cid < ((STagVal *)p2)->cid) {
H
Hongze Cheng 已提交
727
    return -1;
H
Hongze Cheng 已提交
728
  } else if (((STagVal *)p1)->cid > ((STagVal *)p2)->cid) {
H
Hongze Cheng 已提交
729 730
    return 1;
  }
H
Hongze Cheng 已提交
731

H
Hongze Cheng 已提交
732 733
  return 0;
}
H
Hongze Cheng 已提交
734 735 736
static int tTagValJsonCmprFn(const void *p1, const void *p2) {
  return strcmp(((STagVal *)p1)[0].pKey, ((STagVal *)p2)[0].pKey);
}
C
Cary Xu 已提交
737 738 739 740 741 742 743

static void debugPrintTagVal(int8_t type, const void *val, int32_t vlen, const char *tag, int32_t ln) {
  switch (type) {
    case TSDB_DATA_TYPE_JSON:
    case TSDB_DATA_TYPE_VARCHAR:
    case TSDB_DATA_TYPE_NCHAR: {
      char tmpVal[32] = {0};
744
      strncpy(tmpVal, val, vlen > 31 ? 31 : vlen);
C
Cary Xu 已提交
745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
      printf("%s:%d type:%d vlen:%d, val:\"%s\"\n", tag, ln, (int32_t)type, vlen, tmpVal);
    } break;
    case TSDB_DATA_TYPE_FLOAT:
      printf("%s:%d type:%d vlen:%d, val:%f\n", tag, ln, (int32_t)type, vlen, *(float *)val);
      break;
    case TSDB_DATA_TYPE_DOUBLE:
      printf("%s:%d type:%d vlen:%d, val:%lf\n", tag, ln, (int32_t)type, vlen, *(double *)val);
      break;
    case TSDB_DATA_TYPE_BOOL:
      printf("%s:%d type:%d vlen:%d, val:%" PRIu8 "\n", tag, ln, (int32_t)type, vlen, *(uint8_t *)val);
      break;
    case TSDB_DATA_TYPE_TINYINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi8 "\n", tag, ln, (int32_t)type, vlen, *(int8_t *)val);
      break;
    case TSDB_DATA_TYPE_SMALLINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi16 "\n", tag, ln, (int32_t)type, vlen, *(int16_t *)val);
      break;
    case TSDB_DATA_TYPE_INT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi32 "\n", tag, ln, (int32_t)type, vlen, *(int32_t *)val);
      break;
    case TSDB_DATA_TYPE_BIGINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi64 "\n", tag, ln, (int32_t)type, vlen, *(int64_t *)val);
      break;
    case TSDB_DATA_TYPE_TIMESTAMP:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi64 "\n", tag, ln, (int32_t)type, vlen, *(int64_t *)val);
      break;
    case TSDB_DATA_TYPE_UTINYINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIu8 "\n", tag, ln, (int32_t)type, vlen, *(uint8_t *)val);
      break;
    case TSDB_DATA_TYPE_USMALLINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIu16 "\n", tag, ln, (int32_t)type, vlen, *(uint16_t *)val);
      break;
    case TSDB_DATA_TYPE_UINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIu32 "\n", tag, ln, (int32_t)type, vlen, *(uint32_t *)val);
      break;
    case TSDB_DATA_TYPE_UBIGINT:
      printf("%s:%d type:%d vlen:%d, val:%" PRIu64 "\n", tag, ln, (int32_t)type, vlen, *(uint64_t *)val);
      break;
wmmhello's avatar
wmmhello 已提交
783 784 785
    case TSDB_DATA_TYPE_NULL:
      printf("%s:%d type:%d vlen:%d, val:%" PRIi8 "\n", tag, ln, (int32_t)type, vlen, *(int8_t *)val);
      break;
C
Cary Xu 已提交
786 787 788 789 790 791 792
    default:
      ASSERT(0);
      break;
  }
}

void debugPrintSTag(STag *pTag, const char *tag, int32_t ln) {
H
Hongze Cheng 已提交
793 794
  int8_t   isJson = pTag->flags & TD_TAG_JSON;
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
795 796 797 798 799 800 801 802 803 804
  uint8_t *p = NULL;
  int16_t  offset = 0;

  if (isLarge) {
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
  } else {
    p = (uint8_t *)&pTag->idx[pTag->nTag];
  }
  printf("%s:%d >>> STAG === %s:%s, len: %d, nTag: %d, sver:%d\n", tag, ln, isJson ? "json" : "normal",
         isLarge ? "large" : "small", (int32_t)pTag->len, (int32_t)pTag->nTag, pTag->ver);
C
Cary Xu 已提交
805
  for (uint16_t n = 0; n < pTag->nTag; ++n) {
806 807 808 809 810 811 812 813
    if (isLarge) {
      offset = ((int16_t *)pTag->idx)[n];
    } else {
      offset = pTag->idx[n];
    }
    STagVal tagVal = {0};
    if (isJson) {
      tagVal.pKey = (char *)POINTER_SHIFT(p, offset);
C
Cary Xu 已提交
814
    } else {
815
      tagVal.cid = *(int16_t *)POINTER_SHIFT(p, offset);
C
Cary Xu 已提交
816
    }
817
    printf("%s:%d loop[%d-%d] offset=%d\n", __func__, __LINE__, (int32_t)pTag->nTag, (int32_t)n, (int32_t)offset);
C
Cary Xu 已提交
818
    tGetTagVal(p + offset, &tagVal, isJson);
819
    if (IS_VAR_DATA_TYPE(tagVal.type)) {
wmmhello's avatar
wmmhello 已提交
820
      debugPrintTagVal(tagVal.type, tagVal.pData, tagVal.nData, __func__, __LINE__);
821
    } else {
wmmhello's avatar
wmmhello 已提交
822 823
      debugPrintTagVal(tagVal.type, &tagVal.i64, tDataTypes[tagVal.type].bytes, __func__, __LINE__);
    }
C
Cary Xu 已提交
824 825 826 827
  }
  printf("\n");
}

H
Hongze Cheng 已提交
828 829
static int32_t tPutTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson) {
  int32_t n = 0;
H
Hongze Cheng 已提交
830

H
Hongze Cheng 已提交
831 832 833 834 835 836
  // key
  if (isJson) {
    n += tPutCStr(p ? p + n : p, pTagVal->pKey);
  } else {
    n += tPutI16v(p ? p + n : p, pTagVal->cid);
  }
H
Hongze Cheng 已提交
837

H
Hongze Cheng 已提交
838 839 840 841 842 843 844
  // type
  n += tPutI8(p ? p + n : p, pTagVal->type);

  // value
  if (IS_VAR_DATA_TYPE(pTagVal->type)) {
    n += tPutBinary(p ? p + n : p, pTagVal->pData, pTagVal->nData);
  } else {
H
Hongze Cheng 已提交
845
    p = p ? p + n : p;
wmmhello's avatar
wmmhello 已提交
846
    n += tDataTypes[pTagVal->type].bytes;
847
    if (p) memcpy(p, &(pTagVal->i64), tDataTypes[pTagVal->type].bytes);
H
Hongze Cheng 已提交
848 849
  }

H
Hongze Cheng 已提交
850 851 852 853 854 855 856 857 858 859
  return n;
}
static int32_t tGetTagVal(uint8_t *p, STagVal *pTagVal, int8_t isJson) {
  int32_t n = 0;

  // key
  if (isJson) {
    n += tGetCStr(p + n, &pTagVal->pKey);
  } else {
    n += tGetI16v(p + n, &pTagVal->cid);
H
Hongze Cheng 已提交
860 861
  }

H
Hongze Cheng 已提交
862 863
  // type
  n += tGetI8(p + n, &pTagVal->type);
H
Hongze Cheng 已提交
864

H
Hongze Cheng 已提交
865 866 867 868
  // value
  if (IS_VAR_DATA_TYPE(pTagVal->type)) {
    n += tGetBinary(p + n, &pTagVal->pData, &pTagVal->nData);
  } else {
wmmhello's avatar
wmmhello 已提交
869
    memcpy(&(pTagVal->i64), p + n, tDataTypes[pTagVal->type].bytes);
wmmhello's avatar
wmmhello 已提交
870
    n += tDataTypes[pTagVal->type].bytes;
H
Hongze Cheng 已提交
871 872 873 874
  }

  return n;
}
wmmhello's avatar
wmmhello 已提交
875

H
Hongze Cheng 已提交
876
bool tTagIsJson(const void *pTag) { return (((const STag *)pTag)->flags & TD_TAG_JSON); }
wmmhello's avatar
wmmhello 已提交
877

H
Hongze Cheng 已提交
878 879 880 881 882
bool tTagIsJsonNull(void *data) {
  STag  *pTag = (STag *)data;
  int8_t isJson = tTagIsJson(pTag);
  if (!isJson) return false;
  return ((STag *)data)->nTag == 0;
wmmhello's avatar
wmmhello 已提交
883 884
}

H
Hongze Cheng 已提交
885
int32_t tTagNew(SArray *pArray, int32_t version, int8_t isJson, STag **ppTag) {
H
Hongze Cheng 已提交
886 887 888
  int32_t  code = 0;
  uint8_t *p = NULL;
  int16_t  n = 0;
H
Hongze Cheng 已提交
889
  int16_t  nTag = taosArrayGetSize(pArray);
H
Hongze Cheng 已提交
890 891
  int32_t  szTag = 0;
  int8_t   isLarge = 0;
H
Hongze Cheng 已提交
892 893 894

  // sort
  if (isJson) {
wafwerar's avatar
wafwerar 已提交
895
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValJsonCmprFn);
H
Hongze Cheng 已提交
896
  } else {
wafwerar's avatar
wafwerar 已提交
897
    taosSort(pArray->pData, nTag, sizeof(STagVal), tTagValCmprFn);
H
Hongze Cheng 已提交
898 899 900
  }

  // get size
H
Hongze Cheng 已提交
901
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
H
Hongze Cheng 已提交
902
    szTag += tPutTagVal(NULL, (STagVal *)taosArrayGet(pArray, iTag), isJson);
H
Hongze Cheng 已提交
903
  }
H
Hongze Cheng 已提交
904 905 906 907 908 909
  if (szTag <= INT8_MAX) {
    szTag = szTag + sizeof(STag) + sizeof(int8_t) * nTag;
  } else {
    szTag = szTag + sizeof(STag) + sizeof(int16_t) * nTag;
    isLarge = 1;
  }
H
Hongze Cheng 已提交
910

H
Hongze Cheng 已提交
911
  ASSERT(szTag <= INT16_MAX);
H
Hongze Cheng 已提交
912 913

  // build tag
C
Cary Xu 已提交
914
  (*ppTag) = (STag *)taosMemoryCalloc(szTag, 1);
H
Hongze Cheng 已提交
915 916 917 918
  if ((*ppTag) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }
H
Hongze Cheng 已提交
919 920 921 922 923 924 925
  (*ppTag)->flags = 0;
  if (isJson) {
    (*ppTag)->flags |= TD_TAG_JSON;
  }
  if (isLarge) {
    (*ppTag)->flags |= TD_TAG_LARGE;
  }
H
Hongze Cheng 已提交
926 927 928
  (*ppTag)->len = szTag;
  (*ppTag)->nTag = nTag;
  (*ppTag)->ver = version;
H
Hongze Cheng 已提交
929

H
Hongze Cheng 已提交
930 931 932 933 934
  if (isLarge) {
    p = (uint8_t *)&((int16_t *)(*ppTag)->idx)[nTag];
  } else {
    p = (uint8_t *)&(*ppTag)->idx[nTag];
  }
H
Hongze Cheng 已提交
935 936
  n = 0;
  for (int16_t iTag = 0; iTag < nTag; iTag++) {
H
Hongze Cheng 已提交
937 938 939 940 941
    if (isLarge) {
      ((int16_t *)(*ppTag)->idx)[iTag] = n;
    } else {
      (*ppTag)->idx[iTag] = n;
    }
H
Hongze Cheng 已提交
942
    n += tPutTagVal(p + n, (STagVal *)taosArrayGet(pArray, iTag), isJson);
H
Hongze Cheng 已提交
943
  }
944
#ifdef TD_DEBUG_PRINT_TAG
C
Cary Xu 已提交
945
  debugPrintSTag(*ppTag, __func__, __LINE__);
946
#endif
C
Cary Xu 已提交
947

H
Hongze Cheng 已提交
948 949 950 951
  return code;

_err:
  return code;
H
Hongze Cheng 已提交
952 953 954 955 956 957
}

void tTagFree(STag *pTag) {
  if (pTag) taosMemoryFree(pTag);
}

958 959 960
char *tTagValToData(const STagVal *value, bool isJson) {
  if (!value) return NULL;
  char  *data = NULL;
wmmhello's avatar
wmmhello 已提交
961 962 963 964
  int8_t typeBytes = 0;
  if (isJson) {
    typeBytes = CHAR_BYTES;
  }
965
  if (IS_VAR_DATA_TYPE(value->type)) {
wmmhello's avatar
wmmhello 已提交
966
    data = taosMemoryCalloc(1, typeBytes + VARSTR_HEADER_SIZE + value->nData);
967 968
    if (data == NULL) return NULL;
    if (isJson) *data = value->type;
wmmhello's avatar
wmmhello 已提交
969 970
    varDataLen(data + typeBytes) = value->nData;
    memcpy(varDataVal(data + typeBytes), value->pData, value->nData);
971 972
  } else {
    data = ((char *)&(value->i64)) - typeBytes;  // json with type
wmmhello's avatar
wmmhello 已提交
973 974 975 976 977
  }

  return data;
}

C
Cary Xu 已提交
978
bool tTagGet(const STag *pTag, STagVal *pTagVal) {
H
Hongze Cheng 已提交
979 980 981
  int16_t  lidx = 0;
  int16_t  ridx = pTag->nTag - 1;
  int16_t  midx;
H
Hongze Cheng 已提交
982 983 984 985
  uint8_t *p;
  int8_t   isJson = pTag->flags & TD_TAG_JSON;
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
  int16_t  offset;
H
Hongze Cheng 已提交
986 987 988
  STagVal  tv;
  int      c;

H
Hongze Cheng 已提交
989 990 991 992 993 994
  if (isLarge) {
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
  } else {
    p = (uint8_t *)&pTag->idx[pTag->nTag];
  }

H
Hongze Cheng 已提交
995 996 997 998 999
  pTagVal->type = TSDB_DATA_TYPE_NULL;
  pTagVal->pData = NULL;
  pTagVal->nData = 0;
  while (lidx <= ridx) {
    midx = (lidx + ridx) / 2;
H
Hongze Cheng 已提交
1000 1001 1002 1003 1004
    if (isLarge) {
      offset = ((int16_t *)pTag->idx)[midx];
    } else {
      offset = pTag->idx[midx];
    }
H
Hongze Cheng 已提交
1005

H
Hongze Cheng 已提交
1006 1007
    tGetTagVal(p + offset, &tv, isJson);
    if (isJson) {
H
Hongze Cheng 已提交
1008
      c = tTagValJsonCmprFn(pTagVal, &tv);
H
Hongze Cheng 已提交
1009
    } else {
H
Hongze Cheng 已提交
1010
      c = tTagValCmprFn(pTagVal, &tv);
H
Hongze Cheng 已提交
1011 1012
    }

H
Hongze Cheng 已提交
1013 1014 1015 1016
    if (c < 0) {
      ridx = midx - 1;
    } else if (c > 0) {
      lidx = midx + 1;
H
Hongze Cheng 已提交
1017
    } else {
H
Hongze Cheng 已提交
1018
      memcpy(pTagVal, &tv, sizeof(tv));
C
Cary Xu 已提交
1019
      return true;
H
Hongze Cheng 已提交
1020 1021
    }
  }
C
Cary Xu 已提交
1022
  return false;
H
Hongze Cheng 已提交
1023 1024
}

H
more  
Hongze Cheng 已提交
1025 1026
int32_t tEncodeTag(SEncoder *pEncoder, const STag *pTag) {
  return tEncodeBinary(pEncoder, (const uint8_t *)pTag, pTag->len);
H
Hongze Cheng 已提交
1027 1028
}

H
more  
Hongze Cheng 已提交
1029
int32_t tDecodeTag(SDecoder *pDecoder, STag **ppTag) { return tDecodeBinary(pDecoder, (uint8_t **)ppTag, NULL); }
H
Hongze Cheng 已提交
1030

C
Cary Xu 已提交
1031
int32_t tTagToValArray(const STag *pTag, SArray **ppArray) {
H
Hongze Cheng 已提交
1032
  int32_t  code = 0;
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
  uint8_t *p = NULL;
  STagVal  tv = {0};
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
  int16_t  offset = 0;

  if (isLarge) {
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
  } else {
    p = (uint8_t *)&pTag->idx[pTag->nTag];
  }
H
Hongze Cheng 已提交
1043

H
more  
Hongze Cheng 已提交
1044
  (*ppArray) = taosArrayInit(pTag->nTag + 1, sizeof(STagVal));
H
Hongze Cheng 已提交
1045
  if (*ppArray == NULL) {
H
Hongze Cheng 已提交
1046 1047 1048 1049 1050
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
  }

  for (int16_t iTag = 0; iTag < pTag->nTag; iTag++) {
1051 1052 1053 1054 1055 1056
    if (isLarge) {
      offset = ((int16_t *)pTag->idx)[iTag];
    } else {
      offset = pTag->idx[iTag];
    }
    tGetTagVal(p + offset, &tv, pTag->flags & TD_TAG_JSON);
H
Hongze Cheng 已提交
1057
    taosArrayPush(*ppArray, &tv);
H
Hongze Cheng 已提交
1058 1059 1060 1061 1062 1063 1064 1065
  }

  return code;

_err:
  return code;
}

wmmhello's avatar
wmmhello 已提交
1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085
void tTagSetCid(const STag *pTag, int16_t iTag, int16_t cid) {
  uint8_t *p = NULL;
  int8_t   isLarge = pTag->flags & TD_TAG_LARGE;
  int16_t  offset = 0;

  if (isLarge) {
    p = (uint8_t *)&((int16_t *)pTag->idx)[pTag->nTag];
  } else {
    p = (uint8_t *)&pTag->idx[pTag->nTag];
  }

  if (isLarge) {
    offset = ((int16_t *)pTag->idx)[iTag];
  } else {
    offset = pTag->idx[iTag];
  }

  tPutI16v(p + offset, cid);
}

H
Hongze Cheng 已提交
1086
#if 1  // ===================================================================================================================
C
Cary Xu 已提交
1087
int tdInitTSchemaBuilder(STSchemaBuilder *pBuilder, schema_ver_t version) {
H
Hongze Cheng 已提交
1088 1089 1090
  if (pBuilder == NULL) return -1;

  pBuilder->tCols = 256;
wafwerar's avatar
wafwerar 已提交
1091
  pBuilder->columns = (STColumn *)taosMemoryMalloc(sizeof(STColumn) * pBuilder->tCols);
H
Hongze Cheng 已提交
1092 1093 1094 1095 1096 1097 1098 1099
  if (pBuilder->columns == NULL) return -1;

  tdResetTSchemaBuilder(pBuilder, version);
  return 0;
}

void tdDestroyTSchemaBuilder(STSchemaBuilder *pBuilder) {
  if (pBuilder) {
wafwerar's avatar
wafwerar 已提交
1100
    taosMemoryFreeClear(pBuilder->columns);
H
Hongze Cheng 已提交
1101 1102 1103
  }
}

C
Cary Xu 已提交
1104
void tdResetTSchemaBuilder(STSchemaBuilder *pBuilder, schema_ver_t version) {
H
Hongze Cheng 已提交
1105 1106 1107
  pBuilder->nCols = 0;
  pBuilder->tlen = 0;
  pBuilder->flen = 0;
T
Tao Liu 已提交
1108
  pBuilder->vlen = 0;
H
Hongze Cheng 已提交
1109 1110 1111
  pBuilder->version = version;
}

C
Cary Xu 已提交
1112
int32_t tdAddColToSchema(STSchemaBuilder *pBuilder, int8_t type, int8_t flags, col_id_t colId, col_bytes_t bytes) {
1113
  if (!isValidDataType(type)) return -1;
H
Hongze Cheng 已提交
1114 1115 1116

  if (pBuilder->nCols >= pBuilder->tCols) {
    pBuilder->tCols *= 2;
wafwerar's avatar
wafwerar 已提交
1117
    STColumn *columns = (STColumn *)taosMemoryRealloc(pBuilder->columns, sizeof(STColumn) * pBuilder->tCols);
T
tickduan 已提交
1118 1119
    if (columns == NULL) return -1;
    pBuilder->columns = columns;
H
Hongze Cheng 已提交
1120 1121 1122
  }

  STColumn *pCol = &(pBuilder->columns[pBuilder->nCols]);
H
Hongze Cheng 已提交
1123 1124 1125
  pCol->type = type;
  pCol->colId = colId;
  pCol->flags = flags;
H
Hongze Cheng 已提交
1126
  if (pBuilder->nCols == 0) {
H
Hongze Cheng 已提交
1127
    pCol->offset = 0;
H
Hongze Cheng 已提交
1128
  } else {
S
Shengliang Guan 已提交
1129
    STColumn *pTCol = &(pBuilder->columns[pBuilder->nCols - 1]);
H
Hongze Cheng 已提交
1130
    pCol->offset = pTCol->offset + TYPE_BYTES[pTCol->type];
H
Hongze Cheng 已提交
1131 1132 1133
  }

  if (IS_VAR_DATA_TYPE(type)) {
H
Hongze Cheng 已提交
1134
    pCol->bytes = bytes;
T
Tao Liu 已提交
1135 1136
    pBuilder->tlen += (TYPE_BYTES[type] + bytes);
    pBuilder->vlen += bytes - sizeof(VarDataLenT);
H
Hongze Cheng 已提交
1137
  } else {
H
Hongze Cheng 已提交
1138
    pCol->bytes = TYPE_BYTES[type];
H
Hongze Cheng 已提交
1139
    pBuilder->tlen += TYPE_BYTES[type];
T
Tao Liu 已提交
1140
    pBuilder->vlen += TYPE_BYTES[type];
H
Hongze Cheng 已提交
1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155
  }

  pBuilder->nCols++;
  pBuilder->flen += TYPE_BYTES[type];

  ASSERT(pCol->offset < pBuilder->flen);

  return 0;
}

STSchema *tdGetSchemaFromBuilder(STSchemaBuilder *pBuilder) {
  if (pBuilder->nCols <= 0) return NULL;

  int tlen = sizeof(STSchema) + sizeof(STColumn) * pBuilder->nCols;

wafwerar's avatar
wafwerar 已提交
1156
  STSchema *pSchema = (STSchema *)taosMemoryMalloc(tlen);
H
Hongze Cheng 已提交
1157 1158
  if (pSchema == NULL) return NULL;

H
Hongze Cheng 已提交
1159 1160 1161 1162 1163
  pSchema->version = pBuilder->version;
  pSchema->numOfCols = pBuilder->nCols;
  pSchema->tlen = pBuilder->tlen;
  pSchema->flen = pBuilder->flen;
  pSchema->vlen = pBuilder->vlen;
H
Hongze Cheng 已提交
1164

C
Cary Xu 已提交
1165
#ifdef TD_SUPPORT_BITMAP
H
Hongze Cheng 已提交
1166
  pSchema->tlen += (int)TD_BITMAP_BYTES(pSchema->numOfCols);
C
Cary Xu 已提交
1167 1168
#endif

H
Hongze Cheng 已提交
1169
  memcpy(&pSchema->columns[0], pBuilder->columns, sizeof(STColumn) * pBuilder->nCols);
H
Hongze Cheng 已提交
1170

H
TD-27  
hzcheng 已提交
1171 1172 1173
  return pSchema;
}

H
refact  
Hongze Cheng 已提交
1174 1175 1176
#endif

// SColData ========================================
H
Hongze Cheng 已提交
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188
void tColDataDestroy(void *ph) {
  SColData *pColData = (SColData *)ph;

  tFree(pColData->pBitMap);
  tFree((uint8_t *)pColData->aOffset);
  tFree(pColData->pData);
}

void tColDataInit(SColData *pColData, int16_t cid, int8_t type, int8_t smaOn) {
  pColData->cid = cid;
  pColData->type = type;
  pColData->smaOn = smaOn;
H
Hongze Cheng 已提交
1189
  tColDataClear(pColData);
H
Hongze Cheng 已提交
1190 1191
}

H
Hongze Cheng 已提交
1192
void tColDataClear(SColData *pColData) {
H
Hongze Cheng 已提交
1193 1194 1195 1196 1197
  pColData->nVal = 0;
  pColData->flag = 0;
  pColData->nData = 0;
}

H
Hongze Cheng 已提交
1198 1199 1200 1201 1202 1203 1204 1205
static FORCE_INLINE int32_t tColDataPutValue(SColData *pColData, SColVal *pColVal) {
  int32_t code = 0;

  if (IS_VAR_DATA_TYPE(pColData->type)) {
    code = tRealloc((uint8_t **)(&pColData->aOffset), sizeof(int32_t) * (pColData->nVal + 1));
    if (code) goto _exit;
    pColData->aOffset[pColData->nVal] = pColData->nData;

H
Hongze Cheng 已提交
1206
    if (pColVal->value.nData) {
H
Hongze Cheng 已提交
1207 1208 1209 1210 1211 1212
      code = tRealloc(&pColData->pData, pColData->nData + pColVal->value.nData);
      if (code) goto _exit;
      memcpy(pColData->pData + pColData->nData, pColVal->value.pData, pColVal->value.nData);
      pColData->nData += pColVal->value.nData;
    }
  } else {
H
Hongze Cheng 已提交
1213
    ASSERT(pColData->nData == tDataTypes[pColData->type].bytes * pColData->nVal);
H
Hongze Cheng 已提交
1214 1215 1216 1217 1218 1219 1220 1221
    code = tRealloc(&pColData->pData, pColData->nData + tDataTypes[pColData->type].bytes);
    if (code) goto _exit;
    pColData->nData += tPutValue(pColData->pData + pColData->nData, &pColVal->value, pColVal->type);
  }

_exit:
  return code;
}
H
Hongze Cheng 已提交
1222
static FORCE_INLINE int32_t tColDataAppendValue0(SColData *pColData, SColVal *pColVal) {  // 0
H
Hongze Cheng 已提交
1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241
  int32_t code = 0;

  if (pColVal->isNone) {
    pColData->flag = HAS_NONE;
  } else if (pColVal->isNull) {
    pColData->flag = HAS_NULL;
  } else {
    pColData->flag = HAS_VALUE;
    code = tColDataPutValue(pColData, pColVal);
    if (code) goto _exit;
  }
  pColData->nVal++;

_exit:
  return code;
}
static FORCE_INLINE int32_t tColDataAppendValue1(SColData *pColData, SColVal *pColVal) {  // HAS_NONE
  int32_t code = 0;

H
Hongze Cheng 已提交
1242
  if (!pColVal->isNone) {
H
Hongze Cheng 已提交
1243 1244 1245
    int32_t nBit = BIT1_SIZE(pColData->nVal + 1);

    code = tRealloc(&pColData->pBitMap, nBit);
H
Hongze Cheng 已提交
1246 1247
    if (code) goto _exit;

H
Hongze Cheng 已提交
1248
    memset(pColData->pBitMap, 0, nBit);
H
Hongze Cheng 已提交
1249 1250 1251 1252 1253 1254
    SET_BIT1(pColData->pBitMap, pColData->nVal, 1);

    if (pColVal->isNull) {
      pColData->flag |= HAS_NULL;
    } else {
      pColData->flag |= HAS_VALUE;
H
Hongze Cheng 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269

      if (pColData->nVal) {
        if (IS_VAR_DATA_TYPE(pColData->type)) {
          int32_t nOffset = sizeof(int32_t) * pColData->nVal;
          code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
          if (code) goto _exit;
          memset(pColData->aOffset, 0, nOffset);
        } else {
          pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
          code = tRealloc(&pColData->pData, pColData->nData);
          if (code) goto _exit;
          memset(pColData->pData, 0, pColData->nData);
        }
      }

H
Hongze Cheng 已提交
1270 1271 1272
      code = tColDataPutValue(pColData, pColVal);
      if (code) goto _exit;
    }
H
Hongze Cheng 已提交
1273 1274 1275
  }
  pColData->nVal++;

H
Hongze Cheng 已提交
1276
_exit:
H
Hongze Cheng 已提交
1277 1278 1279 1280 1281
  return code;
}
static FORCE_INLINE int32_t tColDataAppendValue2(SColData *pColData, SColVal *pColVal) {  // HAS_NULL
  int32_t code = 0;

H
Hongze Cheng 已提交
1282
  if (!pColVal->isNull) {
H
Hongze Cheng 已提交
1283 1284
    int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
    code = tRealloc(&pColData->pBitMap, nBit);
H
Hongze Cheng 已提交
1285 1286 1287 1288 1289
    if (code) goto _exit;

    if (pColVal->isNone) {
      pColData->flag |= HAS_NONE;

H
Hongze Cheng 已提交
1290
      memset(pColData->pBitMap, 255, nBit);
H
Hongze Cheng 已提交
1291 1292 1293 1294
      SET_BIT1(pColData->pBitMap, pColData->nVal, 0);
    } else {
      pColData->flag |= HAS_VALUE;

H
Hongze Cheng 已提交
1295
      memset(pColData->pBitMap, 0, nBit);
H
Hongze Cheng 已提交
1296 1297
      SET_BIT1(pColData->pBitMap, pColData->nVal, 1);

H
Hongze Cheng 已提交
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
      if (pColData->nVal) {
        if (IS_VAR_DATA_TYPE(pColData->type)) {
          int32_t nOffset = sizeof(int32_t) * pColData->nVal;
          code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
          if (code) goto _exit;
          memset(pColData->aOffset, 0, nOffset);
        } else {
          pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
          code = tRealloc(&pColData->pData, pColData->nData);
          if (code) goto _exit;
          memset(pColData->pData, 0, pColData->nData);
        }
      }

H
Hongze Cheng 已提交
1312 1313 1314
      code = tColDataPutValue(pColData, pColVal);
      if (code) goto _exit;
    }
H
Hongze Cheng 已提交
1315 1316 1317
  }
  pColData->nVal++;

H
Hongze Cheng 已提交
1318
_exit:
H
Hongze Cheng 已提交
1319 1320
  return code;
}
H
Hongze Cheng 已提交
1321
static FORCE_INLINE int32_t tColDataAppendValue3(SColData *pColData, SColVal *pColVal) {  // HAS_NULL|HAS_NONE
H
Hongze Cheng 已提交
1322 1323 1324
  int32_t code = 0;

  if (pColVal->isNone) {
H
Hongze Cheng 已提交
1325 1326 1327 1328
    code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

    SET_BIT1(pColData->pBitMap, pColData->nVal, 0);
H
Hongze Cheng 已提交
1329
  } else if (pColVal->isNull) {
H
Hongze Cheng 已提交
1330 1331 1332 1333
    code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

    SET_BIT1(pColData->pBitMap, pColData->nVal, 1);
H
Hongze Cheng 已提交
1334
  } else {
H
Hongze Cheng 已提交
1335 1336
    pColData->flag |= HAS_VALUE;

H
Hongze Cheng 已提交
1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348
    uint8_t *pBitMap = NULL;
    code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
      SET_BIT2(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal));
    }
    SET_BIT2(pBitMap, pColData->nVal, 2);

    tFree(pColData->pBitMap);
    pColData->pBitMap = pBitMap;

H
Hongze Cheng 已提交
1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362
    if (pColData->nVal) {
      if (IS_VAR_DATA_TYPE(pColData->type)) {
        int32_t nOffset = sizeof(int32_t) * pColData->nVal;
        code = tRealloc((uint8_t **)(&pColData->aOffset), nOffset);
        if (code) goto _exit;
        memset(pColData->aOffset, 0, nOffset);
      } else {
        pColData->nData = tDataTypes[pColData->type].bytes * pColData->nVal;
        code = tRealloc(&pColData->pData, pColData->nData);
        if (code) goto _exit;
        memset(pColData->pData, 0, pColData->nData);
      }
    }

H
Hongze Cheng 已提交
1363 1364
    code = tColDataPutValue(pColData, pColVal);
    if (code) goto _exit;
H
Hongze Cheng 已提交
1365 1366 1367
  }
  pColData->nVal++;

H
Hongze Cheng 已提交
1368
_exit:
H
Hongze Cheng 已提交
1369 1370 1371 1372 1373
  return code;
}
static FORCE_INLINE int32_t tColDataAppendValue4(SColData *pColData, SColVal *pColVal) {  // HAS_VALUE
  int32_t code = 0;

H
Hongze Cheng 已提交
1374 1375 1376 1377 1378
  if (pColVal->isNone || pColVal->isNull) {
    if (pColVal->isNone) {
      pColData->flag |= HAS_NONE;
    } else {
      pColData->flag |= HAS_NULL;
H
Hongze Cheng 已提交
1379
    }
H
Hongze Cheng 已提交
1380

H
Hongze Cheng 已提交
1381 1382
    int32_t nBit = BIT1_SIZE(pColData->nVal + 1);
    code = tRealloc(&pColData->pBitMap, nBit);
H
Hongze Cheng 已提交
1383 1384
    if (code) goto _exit;

H
Hongze Cheng 已提交
1385
    memset(pColData->pBitMap, 255, nBit);
H
Hongze Cheng 已提交
1386
    SET_BIT1(pColData->pBitMap, pColData->nVal, 0);
H
Hongze Cheng 已提交
1387

H
Hongze Cheng 已提交
1388
    code = tColDataPutValue(pColData, pColVal);
H
Hongze Cheng 已提交
1389
    if (code) goto _exit;
H
Hongze Cheng 已提交
1390
  } else {
H
Hongze Cheng 已提交
1391 1392
    code = tColDataPutValue(pColData, pColVal);
    if (code) goto _exit;
H
Hongze Cheng 已提交
1393 1394 1395
  }
  pColData->nVal++;

H
Hongze Cheng 已提交
1396
_exit:
H
Hongze Cheng 已提交
1397 1398
  return code;
}
H
Hongze Cheng 已提交
1399
static FORCE_INLINE int32_t tColDataAppendValue5(SColData *pColData, SColVal *pColVal) {  // HAS_VALUE|HAS_NONE
H
Hongze Cheng 已提交
1400 1401
  int32_t code = 0;

H
Hongze Cheng 已提交
1402 1403 1404 1405 1406 1407 1408
  if (pColVal->isNull) {
    pColData->flag |= HAS_NULL;

    uint8_t *pBitMap = NULL;
    code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

H
Hongze Cheng 已提交
1409
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
H
Hongze Cheng 已提交
1410 1411 1412 1413 1414 1415
      SET_BIT2(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal) ? 2 : 0);
    }
    SET_BIT2(pBitMap, pColData->nVal, 1);

    tFree(pColData->pBitMap);
    pColData->pBitMap = pBitMap;
H
Hongze Cheng 已提交
1416
  } else {
H
Hongze Cheng 已提交
1417 1418 1419 1420 1421 1422 1423 1424
    code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

    if (pColVal->isNone) {
      SET_BIT1(pColData->pBitMap, pColData->nVal, 0);
    } else {
      SET_BIT1(pColData->pBitMap, pColData->nVal, 1);
    }
H
Hongze Cheng 已提交
1425
  }
H
Hongze Cheng 已提交
1426 1427
  code = tColDataPutValue(pColData, pColVal);
  if (code) goto _exit;
H
Hongze Cheng 已提交
1428

H
Hongze Cheng 已提交
1429 1430
  pColData->nVal++;

H
Hongze Cheng 已提交
1431
_exit:
H
Hongze Cheng 已提交
1432 1433
  return code;
}
H
Hongze Cheng 已提交
1434
static FORCE_INLINE int32_t tColDataAppendValue6(SColData *pColData, SColVal *pColVal) {  // HAS_VALUE|HAS_NULL
H
Hongze Cheng 已提交
1435 1436 1437
  int32_t code = 0;

  if (pColVal->isNone) {
H
Hongze Cheng 已提交
1438 1439 1440 1441 1442 1443
    pColData->flag |= HAS_NONE;

    uint8_t *pBitMap = NULL;
    code = tRealloc(&pBitMap, BIT2_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

H
Hongze Cheng 已提交
1444
    for (int32_t iVal = 0; iVal < pColData->nVal; iVal++) {
H
Hongze Cheng 已提交
1445 1446 1447 1448 1449 1450
      SET_BIT2(pBitMap, iVal, GET_BIT1(pColData->pBitMap, iVal) ? 2 : 1);
    }
    SET_BIT2(pBitMap, pColData->nVal, 0);

    tFree(pColData->pBitMap);
    pColData->pBitMap = pBitMap;
H
Hongze Cheng 已提交
1451
  } else {
H
Hongze Cheng 已提交
1452 1453 1454 1455 1456 1457 1458 1459
    code = tRealloc(&pColData->pBitMap, BIT1_SIZE(pColData->nVal + 1));
    if (code) goto _exit;

    if (pColVal->isNull) {
      SET_BIT1(pColData->pBitMap, pColData->nVal, 0);
    } else {
      SET_BIT1(pColData->pBitMap, pColData->nVal, 1);
    }
H
Hongze Cheng 已提交
1460
  }
H
Hongze Cheng 已提交
1461 1462 1463
  code = tColDataPutValue(pColData, pColVal);
  if (code) goto _exit;

H
Hongze Cheng 已提交
1464 1465
  pColData->nVal++;

H
Hongze Cheng 已提交
1466
_exit:
H
Hongze Cheng 已提交
1467 1468 1469
  return code;
}
static FORCE_INLINE int32_t tColDataAppendValue7(SColData *pColData,
H
Hongze Cheng 已提交
1470
                                                 SColVal  *pColVal) {  // HAS_VALUE|HAS_NULL|HAS_NONE
H
Hongze Cheng 已提交
1471 1472
  int32_t code = 0;

H
Hongze Cheng 已提交
1473 1474 1475
  code = tRealloc(&pColData->pBitMap, BIT2_SIZE(pColData->nVal + 1));
  if (code) goto _exit;

H
Hongze Cheng 已提交
1476
  if (pColVal->isNone) {
H
Hongze Cheng 已提交
1477
    SET_BIT2(pColData->pBitMap, pColData->nVal, 0);
H
Hongze Cheng 已提交
1478
  } else if (pColVal->isNull) {
H
Hongze Cheng 已提交
1479
    SET_BIT2(pColData->pBitMap, pColData->nVal, 1);
H
Hongze Cheng 已提交
1480
  } else {
H
Hongze Cheng 已提交
1481
    SET_BIT2(pColData->pBitMap, pColData->nVal, 2);
H
Hongze Cheng 已提交
1482
  }
H
Hongze Cheng 已提交
1483 1484 1485
  code = tColDataPutValue(pColData, pColVal);
  if (code) goto _exit;

H
Hongze Cheng 已提交
1486 1487
  pColData->nVal++;

H
Hongze Cheng 已提交
1488
_exit:
H
Hongze Cheng 已提交
1489 1490
  return code;
}
H
Hongze Cheng 已提交
1491 1492 1493 1494
static int32_t (*tColDataAppendValueImpl[])(SColData *pColData, SColVal *pColVal) = {
    tColDataAppendValue0,  // 0
    tColDataAppendValue1,  // HAS_NONE
    tColDataAppendValue2,  // HAS_NULL
H
refact  
Hongze Cheng 已提交
1495
    tColDataAppendValue3,  // HAS_NULL|HAS_NONE
H
Hongze Cheng 已提交
1496
    tColDataAppendValue4,  // HAS_VALUE
H
refact  
Hongze Cheng 已提交
1497 1498 1499
    tColDataAppendValue5,  // HAS_VALUE|HAS_NONE
    tColDataAppendValue6,  // HAS_VALUE|HAS_NULL
    tColDataAppendValue7   // HAS_VALUE|HAS_NULL|HAS_NONE
H
Hongze Cheng 已提交
1500
};
H
refact  
Hongze Cheng 已提交
1501
int32_t tColDataAppendValue(SColData *pColData, SColVal *pColVal) {
H
Hongze Cheng 已提交
1502 1503
  ASSERT(pColData->cid == pColVal->cid && pColData->type == pColVal->type);
  return tColDataAppendValueImpl[pColData->flag](pColData, pColVal);
H
refact  
Hongze Cheng 已提交
1504 1505
}

H
Hongze Cheng 已提交
1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590
static FORCE_INLINE void tColDataGetValue1(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NONE
  *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
}
static FORCE_INLINE void tColDataGetValue2(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NULL
  *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
}
static FORCE_INLINE void tColDataGetValue3(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_NULL|HAS_NONE
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
    case 0:
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
      break;
    case 1:
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
      break;
    default:
      ASSERT(0);
  }
}
static FORCE_INLINE void tColDataGetValue4(SColData *pColData, int32_t iVal, SColVal *pColVal) {  // HAS_VALUE
  SValue value;
  if (IS_VAR_DATA_TYPE(pColData->type)) {
    if (iVal + 1 < pColData->nVal) {
      value.nData = pColData->aOffset[iVal + 1] - pColData->aOffset[iVal];
    } else {
      value.nData = pColData->nData - pColData->aOffset[iVal];
    }
    value.pData = pColData->pData + pColData->aOffset[iVal];
  } else {
    tGetValue(pColData->pData + tDataTypes[pColData->type].bytes * iVal, &value, pColData->type);
  }
  *pColVal = COL_VAL_VALUE(pColData->cid, pColData->type, value);
}
static FORCE_INLINE void tColDataGetValue5(SColData *pColData, int32_t iVal,
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NONE
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
    case 0:
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
      break;
    case 1:
      tColDataGetValue4(pColData, iVal, pColVal);
      break;
    default:
      ASSERT(0);
  }
}
static FORCE_INLINE void tColDataGetValue6(SColData *pColData, int32_t iVal,
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL
  switch (GET_BIT1(pColData->pBitMap, iVal)) {
    case 0:
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
      break;
    case 1:
      tColDataGetValue4(pColData, iVal, pColVal);
      break;
    default:
      ASSERT(0);
  }
}
static FORCE_INLINE void tColDataGetValue7(SColData *pColData, int32_t iVal,
                                           SColVal *pColVal) {  // HAS_VALUE|HAS_NULL|HAS_NONE
  switch (GET_BIT2(pColData->pBitMap, iVal)) {
    case 0:
      *pColVal = COL_VAL_NONE(pColData->cid, pColData->type);
      break;
    case 1:
      *pColVal = COL_VAL_NULL(pColData->cid, pColData->type);
      break;
    case 2:
      tColDataGetValue4(pColData, iVal, pColVal);
      break;
    default:
      ASSERT(0);
  }
}
static void (*tColDataGetValueImpl[])(SColData *pColData, int32_t iVal, SColVal *pColVal) = {
    NULL,               // 0
    tColDataGetValue1,  // HAS_NONE
    tColDataGetValue2,  // HAS_NULL
    tColDataGetValue3,  // HAS_NULL | HAS_NONE
    tColDataGetValue4,  // HAS_VALUE
    tColDataGetValue5,  // HAS_VALUE | HAS_NONE
    tColDataGetValue6,  // HAS_VALUE | HAS_NULL
    tColDataGetValue7   // HAS_VALUE | HAS_NULL | HAS_NONE
};
void tColDataGetValue(SColData *pColData, int32_t iVal, SColVal *pColVal) {
H
Hongze Cheng 已提交
1591
  ASSERT(iVal >= 0 && iVal < pColData->nVal && pColData->flag);
H
Hongze Cheng 已提交
1592 1593 1594
  tColDataGetValueImpl[pColData->flag](pColData, iVal, pColVal);
}

H
refact  
Hongze Cheng 已提交
1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
int32_t tColDataCopy(SColData *pColDataSrc, SColData *pColDataDest) {
  int32_t code = 0;
  int32_t size;

  ASSERT(pColDataSrc->nVal > 0);
  ASSERT(pColDataDest->cid = pColDataSrc->cid);
  ASSERT(pColDataDest->type = pColDataSrc->type);

  pColDataDest->smaOn = pColDataSrc->smaOn;
  pColDataDest->nVal = pColDataSrc->nVal;
  pColDataDest->flag = pColDataSrc->flag;

  // bitmap
  if (pColDataSrc->flag != HAS_NONE && pColDataSrc->flag != HAS_NULL && pColDataSrc->flag != HAS_VALUE) {
    size = BIT2_SIZE(pColDataSrc->nVal);
    code = tRealloc(&pColDataDest->pBitMap, size);
    if (code) goto _exit;
    memcpy(pColDataDest->pBitMap, pColDataSrc->pBitMap, size);
  }

  // offset
  if (IS_VAR_DATA_TYPE(pColDataDest->type)) {
    size = sizeof(int32_t) * pColDataSrc->nVal;

    code = tRealloc((uint8_t **)&pColDataDest->aOffset, size);
    if (code) goto _exit;

    memcpy(pColDataDest->aOffset, pColDataSrc->aOffset, size);
  }

  // value
  pColDataDest->nData = pColDataSrc->nData;
  code = tRealloc(&pColDataDest->pData, pColDataSrc->nData);
  if (code) goto _exit;
  memcpy(pColDataDest->pData, pColDataSrc->pData, pColDataDest->nData);

_exit:
  return code;
H
Hongze Cheng 已提交
1633
}