taosdata / TDengine / #4998

coverage: 72.335% (+0.6%) from 71.739%
#4998

push

travis-ci

web-flow 
enh:register add secondEp (#34867)

61 of 69 new or added lines in 1 file covered. (88.41%)

8670 existing lines in 142 files now uncovered.

253516 of 350475 relevant lines covered (72.33%)

133451446.75 hits per line

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

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









16



#include "tsdbReadUtil.h"










17



#include "tsdb.h"










18



#include "tsdbDataFileRW.h"










19



#include "tsdbFS2.h"










20



#include "tsdbMerge.h"










21



#include "tsdbUtil2.h"










22



#include "tsimplehash.h"










23












24



static bool overlapWithDelSkylineWithoutVer(STableBlockScanInfo* pBlockScanInfo, const SBrinRecord* pRecord,










25



                                            int32_t order);










26












27



static int32_t initBlockScanInfoBuf(SBlockInfoBuf* pBuf, int32_t numOfTables) {




294,751,439✔





28



  int32_t              code = TSDB_CODE_SUCCESS;




294,751,439✔





29



  int32_t              lino = 0;




294,751,439✔





30



  int32_t              num = 0;




294,751,439✔





31



  int32_t              remainder = 0;




294,751,439✔





32



  STableBlockScanInfo* p = NULL;




294,751,439✔





33



  const void*          px = NULL;




295,034,861✔





34












35



  TSDB_CHECK_CONDITION(pBuf && pBuf->numPerBucket > 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




295,034,861✔





36



  TSDB_CHECK_CONDITION(numOfTables >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




294,884,018✔





37












38



  num = numOfTables / pBuf->numPerBucket;




294,884,018✔





39



  remainder = numOfTables % pBuf->numPerBucket;




294,497,472✔





40












41



  if (pBuf->pData == NULL) {




293,889,191✔





42



    pBuf->pData = taosArrayInit(num + 1, POINTER_BYTES);




294,241,489✔





43



    TSDB_CHECK_NULL(pBuf->pData, code, lino, _end, terrno);




294,606,272✔





44



  }










45












46



  for (int32_t i = 0; i < num; ++i) {




294,450,346✔





47



    p = taosMemoryCalloc(pBuf->numPerBucket, sizeof(STableBlockScanInfo));




30,082✔





48



    TSDB_CHECK_NULL(p, code, lino, _end, terrno);




30,082✔





49












50



    px = taosArrayPush(pBuf->pData, &p);




30,082✔





51



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




30,082✔





52



    p = NULL;




30,082✔





53



  }










54












55



  if (remainder > 0) {




294,420,264✔





56



    p = taosMemoryCalloc(remainder, sizeof(STableBlockScanInfo));




276,716,075✔





57



    TSDB_CHECK_NULL(p, code, lino, _end, terrno);




276,674,835✔





58












59



    px = taosArrayPush(pBuf->pData, &p);




276,674,835✔





60



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




277,215,507✔





61



    p = NULL;




277,215,507✔





62



  }










63












64



  pBuf->numOfTables = numOfTables;




294,919,696✔





65












66



_end:




295,351,848✔





67



  if (code != TSDB_CODE_SUCCESS) {




295,305,764✔





68



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







69



  }










70



  if (p) {




294,759,159✔





71



    taosMemoryFreeClear(p);




×







72



  }










73



  return code;




294,759,159✔





74



}










75












76



int32_t uidComparFunc(const void* p1, const void* p2) {




1,427,828,620✔





77



  uint64_t pu1 = *(const uint64_t*)p1;




1,427,828,620✔





78



  uint64_t pu2 = *(const uint64_t*)p2;




1,427,986,757✔





79



  if (pu1 == pu2) {




1,428,199,641✔





80



    return 0;




×







81



  } else {










82



    return (pu1 < pu2) ? -1 : 1;




1,428,199,641✔





83



  }










84



}










85












86



int32_t ensureBlockScanInfoBuf(SBlockInfoBuf* pBuf, int32_t numOfTables) {




3,258✔





87



  int32_t              code = TSDB_CODE_SUCCESS;




3,258✔





88



  int32_t              lino = 0;




3,258✔





89



  int32_t              num = 0;




3,258✔





90



  int32_t              remainder = 0;




3,258✔





91



  STableBlockScanInfo* p = NULL;




3,258✔





92



  const void*          px = NULL;




3,258✔





93












94



  TSDB_CHECK_CONDITION(pBuf && pBuf->numPerBucket > 0 && pBuf->numOfTables >= 0, code, lino, _end,




3,258✔





95



                       TSDB_CODE_INVALID_PARA);










96



  TSDB_CHECK_CONDITION(numOfTables >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,258✔





97












98



  if (numOfTables <= pBuf->numOfTables) {




3,258✔





99



    goto _end;




×







100



  }










101












102



  remainder = pBuf->numOfTables % pBuf->numPerBucket;




3,258✔





103



  if (remainder > 0) {




3,258✔





104



    TSDB_CHECK_CONDITION(taosArrayGetSize(pBuf->pData) > 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,258✔





105



    px = taosArrayPop(pBuf->pData);




3,258✔





106



    TSDB_CHECK_NULL(px, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,258✔





107



    p = *(STableBlockScanInfo**)px;




3,258✔





108



    taosMemoryFreeClear(p);




3,258✔





109



    pBuf->numOfTables -= remainder;




3,258✔





110



  }










111












112



  num = (numOfTables - pBuf->numOfTables) / pBuf->numPerBucket;




3,258✔





113



  remainder = (numOfTables - pBuf->numOfTables) % pBuf->numPerBucket;




3,258✔





114












115



  if (pBuf->pData == NULL) {




3,258✔





116



    pBuf->pData = taosArrayInit(num + 1, POINTER_BYTES);




×







117



    TSDB_CHECK_NULL(pBuf->pData, code, lino, _end, terrno);




×







118



  }










119












120



  for (int32_t i = 0; i < num; ++i) {




3,258✔





121



    p = taosMemoryCalloc(pBuf->numPerBucket, sizeof(STableBlockScanInfo));




×







122



    TSDB_CHECK_NULL(p, code, lino, _end, terrno);




×







123












124



    px = taosArrayPush(pBuf->pData, &p);




×







125



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




×







126



    p = NULL;




×







127



  }










128












129



  if (remainder > 0) {




3,258✔





130



    p = taosMemoryCalloc(remainder, sizeof(STableBlockScanInfo));




3,258✔





131



    TSDB_CHECK_NULL(p, code, lino, _end, terrno);




3,258✔





132












133



    px = taosArrayPush(pBuf->pData, &p);




3,258✔





134



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




3,258✔





135



    p = NULL;




3,258✔





136



  }










137












138



  pBuf->numOfTables = numOfTables;




3,258✔





139












140



_end:




3,258✔





141



  if (code != TSDB_CODE_SUCCESS) {




3,258✔





142



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







143



  }










144



  if (p) {




3,258✔





145



    taosMemoryFreeClear(p);




×







146



  }










147



  return code;




3,258✔





148



}










149












150



void clearBlockScanInfoBuf(SBlockInfoBuf* pBuf) {




304,141,950✔





151



  if (pBuf == NULL) return;




304,141,950✔





152



  if (pBuf->pData != NULL) {




304,141,950✔





153



    taosArrayDestroyP(pBuf->pData, NULL);




295,402,590✔





154



    pBuf->pData = NULL;




295,282,815✔





155



  }










156



}










157












158



int32_t getPosInBlockInfoBuf(SBlockInfoBuf* pBuf, int32_t index, STableBlockScanInfo** pInfo) {




690,603,325✔





159



  int32_t               code = TSDB_CODE_SUCCESS;




690,603,325✔





160



  int32_t               lino = 0;




690,603,325✔





161



  int32_t               bucketIndex = 0;




690,603,325✔





162



  STableBlockScanInfo** pBucket = NULL;




690,603,325✔





163












164



  TSDB_CHECK_CONDITION(pBuf && pBuf->numPerBucket > 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




690,603,325✔





165



  TSDB_CHECK_CONDITION(index >= 0 && index < pBuf->numOfTables, code, lino, _end, TSDB_CODE_INVALID_PARA);




690,777,225✔





166



  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




690,189,465✔





167












168



  *pInfo = NULL;




690,189,465✔





169












170



  bucketIndex = index / pBuf->numPerBucket;




690,001,075✔





171



  pBucket = taosArrayGet(pBuf->pData, bucketIndex);




690,604,733✔





172



  TSDB_CHECK_NULL(pBucket, code, lino, _end, terrno);




690,300,614✔





173












174



  *pInfo = (*pBucket) + (index % pBuf->numPerBucket);




690,300,652✔





175












176



_end:




690,722,634✔





177



  if (code != TSDB_CODE_SUCCESS) {




690,722,634✔





178



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







179



  }










180



  return code;




690,304,858✔





181



}










182












183



int32_t getTableBlockScanInfo(SSHashObj* pTableMap, uint64_t uid, STableBlockScanInfo** pInfo, const char* id) {




489,646,962✔





184



  int32_t               code = TSDB_CODE_SUCCESS;




489,646,962✔





185



  int32_t               lino = 0;




489,646,962✔





186



  STableBlockScanInfo** pVal = NULL;




489,646,962✔





187












188



  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




489,646,962✔





189



  TSDB_CHECK_NULL(id, code, lino, _end, TSDB_CODE_INVALID_PARA);




489,646,962✔





190












191



  pVal = (STableBlockScanInfo**)tSimpleHashGet(pTableMap, &uid, sizeof(uid));




489,646,962✔





192



  if (pVal == NULL) {




489,912,873✔





193



    int32_t size = tSimpleHashGetSize(pTableMap);




×







194



    tsdbError("failed to locate the uid:%" PRIu64 " in query table uid list, total tables:%d, %s", uid, size, id);




×







195



    code = TSDB_CODE_INVALID_PARA;




×







196



    TSDB_CHECK_CODE(code, lino, _end);




×







197



  }










198



  *pInfo = *pVal;




489,912,873✔





199



  TSDB_CHECK_NULL(*pInfo, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);




489,862,888✔





200












201



_end:




489,865,265✔





202



  if (code != TSDB_CODE_SUCCESS) {




489,865,265✔





203



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







204



  }










205



  return code;




489,807,900✔





206



}










207












208



int32_t initRowKey(SRowKey* pKey, int64_t ts, int32_t numOfPks, int32_t type, int32_t len, bool asc) {




2,147,483,647✔





209



  int32_t code = TSDB_CODE_SUCCESS;




2,147,483,647✔





210



  int32_t lino = 0;




2,147,483,647✔





211












212



  TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);




2,147,483,647✔





213












214



  pKey->numOfPKs = numOfPks;




2,147,483,647✔





215



  pKey->ts = ts;




2,147,483,647✔





216












217



  if (numOfPks > 0) {




2,147,483,647✔





218



    pKey->pks[0].type = type;




125,899,578✔





219












220



    if (IS_NUMERIC_TYPE(type)) {




125,890,088✔





221



      if (asc) {




88,155,533✔





222



        switch (type) {




55,667,912✔





223



          case TSDB_DATA_TYPE_BIGINT: {




13,545,209✔





224



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, INT64_MIN);




13,545,209✔





225



            break;




13,548,462✔





226



          }










227



          case TSDB_DATA_TYPE_INT: {




14,664,734✔





228



            int32_t min = INT32_MIN;




14,664,734✔





229



            valueSetDatum(pKey->pks, type, &min, tDataTypes[type].bytes);




14,665,371✔





230



            break;




14,669,913✔





231



          }










232



          case TSDB_DATA_TYPE_SMALLINT: {




×







233



            int16_t min = INT16_MIN;




×







234



            valueSetDatum(pKey->pks, type, &min, tDataTypes[type].bytes);




×







235



            break;




×







236



          }










237



          case TSDB_DATA_TYPE_TINYINT: {




×







238



            int8_t min = INT8_MIN;




×







239



            valueSetDatum(pKey->pks, type, &min, tDataTypes[type].bytes);




×







240



            break;




×







241



          }










242



          case TSDB_DATA_TYPE_UTINYINT:




27,455,290✔





243



          case TSDB_DATA_TYPE_USMALLINT:










244



          case TSDB_DATA_TYPE_UINT:










245



          case TSDB_DATA_TYPE_UBIGINT: {










246



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, 0);




27,455,290✔





247



            break;




27,465,743✔





248



          }










249



          default:




2,679✔





250



            code = TSDB_CODE_INVALID_PARA;




2,679✔





251



            TSDB_CHECK_CODE(code, lino, _end);




2,679✔





252



        }










253



      } else {










254



        switch (type) {




32,487,621✔





255



          case TSDB_DATA_TYPE_BIGINT:




7,847,427✔





256



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, INT64_MAX);




7,847,427✔





257



            break;




7,847,427✔





258



          case TSDB_DATA_TYPE_INT:




8,350,817✔





259



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, INT32_MAX);




8,350,817✔





260



            break;




8,352,728✔





261



          case TSDB_DATA_TYPE_SMALLINT:




×







262



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, INT16_MAX);




×







263



            break;




×







264



          case TSDB_DATA_TYPE_TINYINT:




×







265



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, INT8_MAX);




×







266



            break;




×







267



          case TSDB_DATA_TYPE_UBIGINT:




7,974,941✔





268



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, UINT64_MAX);




7,974,941✔





269



            break;




7,976,236✔





270



          case TSDB_DATA_TYPE_UINT:




8,328,791✔





271



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, UINT32_MAX);




8,328,791✔





272



            break;




8,330,070✔





273



          case TSDB_DATA_TYPE_USMALLINT:




×







274



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, UINT16_MAX);




×







275



            break;




×







276



          case TSDB_DATA_TYPE_UTINYINT:




×







277



            VALUE_SET_TRIVIAL_DATUM(pKey->pks, UINT8_MAX);




×







278



            break;




1,274✔





279



          default:




×







280



            code = TSDB_CODE_INVALID_PARA;




×







281



            TSDB_CHECK_CODE(code, lino, _end);




×







282



        }










283



      }










284



    } else {










285



      pKey->pks[0].nData = 0;




37,734,555✔





286












287



      if (pKey->pks[0].pData == NULL) {




37,731,938✔





288



        pKey->pks[0].pData = taosMemoryCalloc(1, len);




37,740,754✔





289



        TSDB_CHECK_NULL(pKey->pks[0].pData, code, lino, _end, terrno);




37,663,758✔





290



      }










291












292



      if (!asc) {




37,671,454✔





293



        pKey->numOfPKs = 2;




15,501,866✔





294



      }










295



    }










296



  }










297












298



_end:




2,147,483,647✔





299



  if (code != TSDB_CODE_SUCCESS) {




2,147,483,647✔





300



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







301



  }










302



  return code;




2,147,483,647✔





303



}










304












305



void clearRowKey(SRowKey* pKey) {




2,147,483,647✔





306



  if (pKey == NULL || pKey->numOfPKs == 0 || (!IS_VAR_DATA_TYPE(pKey->pks[0].type))) {




2,147,483,647✔





307



    return;




2,147,483,647✔





308



  }










309



  taosMemoryFreeClear(pKey->pks[0].pData);




4,385,153✔





310



  taosMemoryFreeClear(pKey->pks[1].pData);




43,514,288✔





311



}










312












313



static int32_t initLastProcKey(STableBlockScanInfo* pScanInfo, const STsdbReader* pReader) {




689,892,373✔





314



  int32_t  code = TSDB_CODE_SUCCESS;




689,892,373✔





315



  int32_t  lino = 0;




689,892,373✔





316



  int32_t  numOfPks = 0;




689,892,373✔





317



  int32_t  type = 0;




689,892,373✔





318



  int32_t  bytes = 0;




689,892,373✔





319



  bool     asc = false;




689,892,373✔





320



  SRowKey* pRowKey = NULL;




689,892,373✔





321












322



  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




689,892,373✔





323



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




689,892,373✔





324












325



  numOfPks = pReader->suppInfo.numOfPks;




689,892,373✔





326



  asc = ASCENDING_TRAVERSE(pReader->info.order);




690,807,693✔





327



  type = pReader->suppInfo.pk.type;




690,683,558✔





328



  bytes = pReader->suppInfo.pk.bytes;




689,980,708✔





329












330



  pRowKey = &pScanInfo->lastProcKey;




691,018,554✔





331



  if (asc) {




690,496,826✔





332



    int64_t skey = pReader->info.window.skey;




631,992,665✔





333



    int64_t ts = (skey > INT64_MIN) ? (skey - 1) : skey;




631,984,682✔





334












335



    code = initRowKey(pRowKey, ts, numOfPks, type, bytes, asc);




631,984,682✔





336



    TSDB_CHECK_CODE(code, lino, _end);




631,368,953✔





337












338



    code = initRowKey(&pScanInfo->sttKeyInfo.nextProcKey, skey, numOfPks, type, bytes, asc);




631,368,953✔





339



    TSDB_CHECK_CODE(code, lino, _end);




632,349,193✔





340



  } else {










341



    int64_t ekey = pReader->info.window.ekey;




58,504,161✔





342



    int64_t ts = (ekey < INT64_MAX) ? (ekey + 1) : ekey;




58,482,168✔





343












344



    code = initRowKey(pRowKey, ts, numOfPks, type, bytes, asc);




58,482,168✔





345



    TSDB_CHECK_CODE(code, lino, _end);




58,414,719✔





346












347



    code = initRowKey(&pScanInfo->sttKeyInfo.nextProcKey, ekey, numOfPks, type, bytes, asc);




58,414,719✔





348



    TSDB_CHECK_CODE(code, lino, _end);




58,438,707✔





349



  }










350












351



  code = initRowKey(&pScanInfo->sttRange.skey, INT64_MAX, numOfPks, type, bytes, asc);




690,787,900✔





352



  TSDB_CHECK_CODE(code, lino, _end);




690,493,068✔





353












354



  code = initRowKey(&pScanInfo->sttRange.ekey, INT64_MIN, numOfPks, type, bytes, asc);




690,493,068✔





355



  TSDB_CHECK_CODE(code, lino, _end);




690,883,800✔





356












357



_end:




690,883,800✔





358



  if (code != TSDB_CODE_SUCCESS) {




690,883,800✔





359



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







360



  }










361



  return code;




690,649,612✔





362



}










363












364



int32_t initTableBlockScanInfo(STableBlockScanInfo* pScanInfo, uint64_t uid, SSHashObj* pTableMap,




691,119,259✔





365



                               STsdbReader* pReader) {










366



  int32_t code = TSDB_CODE_SUCCESS;




691,119,259✔





367



  int32_t lino = 0;




691,119,259✔





368












369



  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




691,119,259✔





370



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




691,119,259✔





371












372



  pScanInfo->uid = uid;




691,119,259✔





373



  INIT_KEYRANGE(&pScanInfo->sttRange);




691,104,517✔





374



  INIT_TIMEWINDOW(&pScanInfo->filesetWindow);




690,142,873✔





375












376



  pScanInfo->cleanSttBlocks = false;




690,389,806✔





377



  pScanInfo->sttBlockReturned = false;




690,332,455✔





378












379



  code = initLastProcKey(pScanInfo, pReader);




690,412,886✔





380



  TSDB_CHECK_CODE(code, lino, _end);




690,479,128✔





381












382



  pScanInfo->sttKeyInfo.status = STT_FILE_READER_UNINIT;




690,479,128✔





383



  code = tSimpleHashPut(pTableMap, &pScanInfo->uid, sizeof(uint64_t), &pScanInfo, POINTER_BYTES);




690,733,536✔





384



  TSDB_CHECK_CODE(code, lino, _end);




690,803,302✔





385












386



  tsdbTrace("%p check table uid:%" PRId64 " from lastKey:%" PRId64 " %s", pReader, pScanInfo->uid,




690,803,302✔





387



            pScanInfo->lastProcKey.ts, pReader->idStr);










388












389



_end:




678,670,571✔





390



  if (code != TSDB_CODE_SUCCESS) {




690,775,874✔





391



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







392



  }










393



  return code;




690,539,388✔





394



}










395












396



// NOTE: speedup the whole processing by preparing the buffer for STableBlockScanInfo in batch model










397



int32_t createDataBlockScanInfo(STsdbReader* pTsdbReader, SBlockInfoBuf* pBuf, const STableKeyInfo* idList,




294,755,092✔





398



                                STableUidList* pUidList, int32_t numOfTables, SSHashObj** pHashObj) {










399



  int32_t    code = TSDB_CODE_SUCCESS;




294,755,092✔





400



  int32_t    lino = 0;




294,755,092✔





401



  SSHashObj* pTableMap = NULL;




294,755,092✔





402



  int64_t    st = 0;




294,755,092✔





403












404



  TSDB_CHECK_NULL(pUidList, code, lino, _end, TSDB_CODE_INVALID_PARA);




294,755,092✔





405



  TSDB_CHECK_NULL(pHashObj, code, lino, _end, TSDB_CODE_INVALID_PARA);




294,755,092✔





406












407



  *pHashObj = NULL;




294,755,092✔





408












409



  // allocate buffer in order to load data blocks from file










410



  // todo use simple hash instead, optimize the memory consumption










411



  pTableMap = tSimpleHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));




294,691,847✔





412



  TSDB_CHECK_NULL(pTableMap, code, lino, _end, terrno);




294,594,027✔





413












414



  st = taosGetTimestampUs();




295,077,549✔





415



  code = initBlockScanInfoBuf(pBuf, numOfTables);




295,077,549✔





416



  if (code != TSDB_CODE_SUCCESS) {




294,753,450✔





417



    tSimpleHashCleanup(pTableMap);




×







418



    TSDB_CHECK_CODE(code, lino, _end);




×







419



  }










420












421



  pUidList->tableUidList = taosMemoryMalloc(numOfTables * sizeof(uint64_t));




294,753,450✔





422



  if (pUidList->tableUidList == NULL) {




294,734,066✔





423



    tSimpleHashCleanup(pTableMap);




×







424



    TSDB_CHECK_NULL(pUidList->tableUidList, code, lino, _end, terrno);




×







425



  }










426












427



  pUidList->currentIndex = 0;




294,887,159✔





428












429



  for (int32_t j = 0; j < numOfTables; ++j) {




957,489,946✔





430



    pUidList->tableUidList[j] = idList[j].uid;




662,899,657✔





431












432



    STableBlockScanInfo* pScanInfo = NULL;




663,193,386✔





433



    code = getPosInBlockInfoBuf(pBuf, j, &pScanInfo);




662,830,248✔





434



    if (code != TSDB_CODE_SUCCESS) {




662,512,061✔





435



      lino = __LINE__;




×







436



      break;




×







437



    }










438












439



    code = initTableBlockScanInfo(pScanInfo, idList[j].uid, pTableMap, pTsdbReader);




662,512,061✔





440



    if (code != TSDB_CODE_SUCCESS) {




662,494,098✔





441



      lino = __LINE__;




×







442



      break;




×







443



    }










444



  }










445












446



  taosSort(pUidList->tableUidList, numOfTables, sizeof(uint64_t), uidComparFunc);




294,745,270✔





447












448



  pTsdbReader->cost.createScanInfoList = (taosGetTimestampUs() - st) / 1000.0;




295,037,006✔





449



  tsdbDebug("%p create %d tables scan-info, size:%.2f Kb, elapsed time:%.2f ms, %s", pTsdbReader, numOfTables,




294,890,820✔





450



            (sizeof(STableBlockScanInfo) * numOfTables) / 1024.0, pTsdbReader->cost.createScanInfoList,










451



            pTsdbReader->idStr);










452












453



  *pHashObj = pTableMap;




294,891,875✔





454












455



_end:




294,835,767✔





456



  if (code != TSDB_CODE_SUCCESS) {




294,835,767✔





457



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







458



  }










459



  return code;




294,500,530✔





460



}










461












462



void resetAllDataBlockScanInfo(SSHashObj* pTableMap, int64_t ts, int32_t step) {




32,887,573✔





463



  STableBlockScanInfo** p = NULL;




32,887,573✔





464



  int32_t               iter = 0;




32,887,573✔





465












466



  while ((p = tSimpleHashIterate(pTableMap, p, &iter)) != NULL) {




65,907,022✔





467



    STableBlockScanInfo* pInfo = *p;




33,074,772✔





468



    if (pInfo == NULL) {




33,074,242✔





469



      continue;




×







470



    }










471












472



    pInfo->iterInit = false;




33,074,242✔





473



    pInfo->iter.hasVal = false;




33,075,925✔





474



    pInfo->iiter.hasVal = false;




33,074,456✔





475












476



    if (pInfo->iter.iter != NULL) {




33,037,782✔





477



      pInfo->iter.iter = tsdbTbDataIterDestroy(pInfo->iter.iter);




4,567,956✔





478



    }










479












480



    if (pInfo->iiter.iter != NULL) {




33,100,758✔





481



      pInfo->iiter.iter = tsdbTbDataIterDestroy(pInfo->iiter.iter);




612✔





482



    }










483












484



    taosArrayDestroy(pInfo->delSkyline);




33,065,248✔





485



    pInfo->delSkyline = NULL;




32,933,651✔





486



    pInfo->lastProcKey.ts = ts;




32,985,961✔





487



    // todo check the nextProcKey info










488



    if ((ts == INT64_MAX && step > 0) || (ts == INT64_MIN && step < 0)) {




32,943,107✔





489



      tsdbError("%s nextProcKey ts overflow, ts:%" PRId64 ", step:%" PRId32,




×







490



                __func__, ts, step);










491



    } else {










492



      pInfo->sttKeyInfo.nextProcKey.ts = ts + step;




32,943,715✔





493



    }










494



  }










495



}




32,880,474✔





496












497



// keep the lastProcKey/sttRange/nextProcKey info










498



void clearBlockScanInfoLoadInfo(STableBlockScanInfo* p) {




692,138,828✔





499



  if (p == NULL) {




692,138,828✔





500



    return;




×







501



  }










502












503



  p->iterInit = false;




692,138,828✔





504



  p->iter.hasVal = false;




692,191,568✔





505



  p->iiter.hasVal = false;




692,111,906✔





506



  p->sttKeyInfo.status = STT_FILE_READER_UNINIT;




692,020,137✔





507












508



  if (p->iter.iter != NULL) {




692,006,711✔





509



    p->iter.iter = tsdbTbDataIterDestroy(p->iter.iter);




413,101,645✔





510



  }










511












512



  if (p->iiter.iter != NULL) {




691,997,841✔





513



    p->iiter.iter = tsdbTbDataIterDestroy(p->iiter.iter);




4,234,850✔





514



  }










515












516



  taosArrayDestroy(p->delSkyline);




692,153,424✔





517



  p->delSkyline = NULL;




691,973,204✔





518



  taosArrayDestroy(p->pBlockList);




692,099,992✔





519



  p->pBlockList = NULL;




691,990,570✔





520



  taosArrayDestroy(p->pBlockIdxList);




691,998,803✔





521



  p->pBlockIdxList = NULL;




691,794,187✔





522



  taosArrayDestroy(p->pMemDelData);




691,835,266✔





523



  p->pMemDelData = NULL;




692,112,742✔





524



  taosArrayDestroy(p->pFileDelData);




692,118,178✔





525



  p->pFileDelData = NULL;




692,119,885✔





526



}










527












528



void clearBlockScanInfo(STableBlockScanInfo* p) {




692,186,207✔





529



  if (p == NULL) {




692,186,207✔





530



    return;




×







531



  }










532












533



  clearBlockScanInfoLoadInfo(p);




692,186,207✔





534












535



  clearRowKey(&p->lastProcKey);




692,140,210✔





536



  clearRowKey(&p->sttRange.skey);




691,943,103✔





537



  clearRowKey(&p->sttRange.ekey);




691,997,086✔





538



  clearRowKey(&p->sttKeyInfo.nextProcKey);




692,054,463✔





539



}










540












541



void destroyAllBlockScanInfo(SSHashObj** pTableMap) {




295,383,877✔





542



  STableBlockScanInfo** p = NULL;




295,383,877✔





543



  int32_t               iter = 0;




295,383,877✔





544












545



  if (pTableMap == NULL || *pTableMap == NULL) {




295,402,912✔





546



    return;




×







547



  }










548












549



  while ((p = tSimpleHashIterate(*pTableMap, p, &iter)) != NULL) {




959,836,619✔





550



    clearBlockScanInfo(*p);




664,329,634✔





551



  }










552












553



  tSimpleHashCleanup(*pTableMap);




295,395,115✔





554



  *pTableMap = NULL;




295,352,459✔





555



}










556












557



static void doCleanupInfoForNextFileset(STableBlockScanInfo* pScanInfo) {




643,285,688✔





558



  if (pScanInfo == NULL) {




643,285,688✔





559



    return;




×







560



  }










561



  // reset the index in last block when handing a new file










562



  taosArrayClear(pScanInfo->pBlockList);




643,285,688✔





563



  taosArrayClear(pScanInfo->pBlockIdxList);




643,218,451✔





564



  taosArrayClear(pScanInfo->pFileDelData);  // del data from each file set




643,308,831✔





565



  pScanInfo->cleanSttBlocks = false;




643,464,403✔





566



  pScanInfo->numOfRowsInStt = 0;




643,497,522✔





567



  pScanInfo->sttBlockReturned = false;




643,286,018✔





568



  INIT_KEYRANGE(&pScanInfo->sttRange);




643,440,327✔





569



  INIT_TIMEWINDOW(&pScanInfo->filesetWindow);




643,099,515✔





570



  pScanInfo->sttKeyInfo.status = STT_FILE_READER_UNINIT;




643,652,269✔





571



}










572












573



void cleanupInfoForNextFileset(SSHashObj* pTableMap) {




457,996,482✔





574



  STableBlockScanInfo** p = NULL;




457,996,482✔





575



  int32_t               iter = 0;




457,996,482✔





576












577



  while ((p = tSimpleHashIterate(pTableMap, p, &iter)) != NULL) {




1,101,210,947✔





578



    doCleanupInfoForNextFileset(*p);




643,172,693✔





579



  }










580



}




458,542,176✔





581












582



// brin records iterator










583



void initBrinRecordIter(SBrinRecordIter* pIter, SDataFileReader* pReader, SArray* pList) {




458,312,418✔





584



  if (pIter == NULL) {




458,312,418✔





585



    return;




×







586



  }










587












588



  (void)memset(&pIter->block, 0, sizeof(SBrinBlock));




458,312,418✔





589



  (void)memset(&pIter->record, 0, sizeof(SBrinRecord));




458,563,981✔





590



  pIter->blockIndex = -1;




458,610,830✔





591



  pIter->recordIndex = -1;




458,780,860✔





592












593



  pIter->pReader = pReader;




458,071,795✔





594



  pIter->pBrinBlockList = pList;




458,126,325✔





595



}










596












597



int32_t getNextBrinRecord(SBrinRecordIter* pIter, SBrinRecord** pRecord) {




1,231,968,518✔





598



  int32_t code = TSDB_CODE_SUCCESS;




1,231,968,518✔





599



  int32_t lino = 0;




1,231,968,518✔





600












601



  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);




1,231,968,518✔





602



  TSDB_CHECK_NULL(pRecord, code, lino, _end, TSDB_CODE_INVALID_PARA);




1,231,968,518✔





603












604



  *pRecord = NULL;




1,231,968,518✔





605












606



  if (pIter->blockIndex == -1 || (pIter->recordIndex + 1) >= pIter->block.numOfRecords) {




1,232,531,149✔





607



    pIter->blockIndex += 1;




473,617,599✔





608



    if (pIter->blockIndex >= taosArrayGetSize(pIter->pBrinBlockList)) {




473,403,619✔





609



      goto _end;




451,308,480✔





610



    }










611












612



    pIter->pCurrentBlk = taosArrayGet(pIter->pBrinBlockList, pIter->blockIndex);




22,105,870✔





613



    TSDB_CHECK_NULL(pIter->pCurrentBlk, code, lino, _end, terrno);




22,111,007✔





614












615



    tBrinBlockClear(&pIter->block);




22,094,870✔





616



    TSDB_CHECK_NULL(pIter->pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




22,102,651✔





617



    code = tsdbDataFileReadBrinBlock(pIter->pReader, pIter->pCurrentBlk, &pIter->block);




22,110,208✔





618



    if (code != TSDB_CODE_SUCCESS) {




22,132,009✔





619



      tsdbError("failed to read brinBlock from file, code:%s", tstrerror(code));




×







620



      TSDB_CHECK_CODE(code, lino, _end);




×







621



    }










622












623



    pIter->recordIndex = -1;




22,132,009✔





624



  }










625












626



  pIter->recordIndex += 1;




783,054,389✔





627



  code = tBrinBlockGet(&pIter->block, pIter->recordIndex, &pIter->record);




782,860,479✔





628



  *pRecord = &pIter->record;




778,560,721✔





629



  TSDB_CHECK_CODE(code, lino, _end);




779,068,686✔





630












631



_end:




779,068,686✔





632



  if (code != TSDB_CODE_SUCCESS) {




1,230,377,166✔





633



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







634



  }










635



  return code;




1,229,647,159✔





636



}










637












638



void clearBrinBlockIter(SBrinRecordIter* pIter) {




457,782,389✔





639



  if (pIter != NULL) {




457,782,389✔





640



    tBrinBlockDestroy(&pIter->block);




458,169,826✔





641



  }










642



}




458,391,233✔





643












644



// initialize the file block access order










645



//  sort the file blocks according to the offset of each data block in the files










646



static void cleanupBlockOrderSupporter(SBlockOrderSupporter* pSup) {




21,161,018✔





647



  if (pSup == NULL) {




21,161,018✔





648



    return;




×







649



  }










650












651



  taosMemoryFreeClear(pSup->numOfBlocksPerTable);




21,161,018✔





652



  taosMemoryFreeClear(pSup->indexPerTable);




21,158,964✔





653












654



  if (pSup->pDataBlockInfo != NULL) {




21,162,280✔





655



    for (int32_t i = 0; i < pSup->numOfTables; ++i) {




64,031,560✔





656



      SBlockOrderWrapper* pBlockInfo = pSup->pDataBlockInfo[i];




42,870,457✔





657



      taosMemoryFreeClear(pBlockInfo);




42,867,717✔





658



    }










659












660



    taosMemoryFreeClear(pSup->pDataBlockInfo);




21,162,549✔





661



  }










662



}










663












664



static int32_t initBlockOrderSupporter(SBlockOrderSupporter* pSup, int32_t numOfTables) {




21,159,998✔





665



  int32_t code = TSDB_CODE_SUCCESS;




21,159,998✔





666



  int32_t lino = 0;




21,159,998✔





667












668



  TSDB_CHECK_NULL(pSup, code, lino, _end, TSDB_CODE_INVALID_PARA);




21,159,998✔





669



  TSDB_CHECK_CONDITION(numOfTables >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




21,159,998✔





670












671



  pSup->pDataBlockInfo = taosMemoryCalloc(numOfTables, POINTER_BYTES);




21,159,998✔





672



  TSDB_CHECK_NULL(pSup->pDataBlockInfo, code, lino, _end, terrno);




21,167,235✔





673



  pSup->indexPerTable = taosMemoryCalloc(numOfTables, sizeof(int32_t));




21,165,828✔





674



  TSDB_CHECK_NULL(pSup->indexPerTable, code, lino, _end, terrno);




21,165,683✔





675



  pSup->numOfBlocksPerTable = taosMemoryCalloc(numOfTables, sizeof(int32_t));




21,164,458✔





676



  TSDB_CHECK_NULL(pSup->numOfBlocksPerTable, code, lino, _end, terrno);




21,165,883✔





677



  pSup->numOfTables = 0;




21,164,322✔





678












679



_end:




21,161,264✔





680



  if (code != TSDB_CODE_SUCCESS) {




21,161,264✔





681



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







682



  }










683



  return code;




21,157,948✔





684



}










685












686



static int32_t fileDataBlockOrderCompar(const void* pLeft, const void* pRight, void* param) {




109,715,564✔





687



  int32_t                     leftIndex = 0;




109,715,564✔





688



  int32_t                     rightIndex = 0;




109,715,564✔





689



  int32_t                     leftTableBlockIndex = 0;




109,715,564✔





690



  int32_t                     rightTableBlockIndex = 0;




109,715,564✔





691



  const SBlockOrderSupporter* pSupporter = NULL;




109,715,564✔





692



  const SBlockOrderWrapper*   pLeftBlock = NULL;




109,715,564✔





693



  const SBlockOrderWrapper*   pRightBlock = NULL;




109,715,564✔





694












695



  leftIndex = *(const int32_t*)pLeft;




109,715,564✔





696



  rightIndex = *(const int32_t*)pRight;




109,716,648✔





697



  pSupporter = (const SBlockOrderSupporter*)param;




109,717,719✔





698












699



  leftTableBlockIndex = pSupporter->indexPerTable[leftIndex];




109,717,719✔





700



  rightTableBlockIndex = pSupporter->indexPerTable[rightIndex];




109,715,756✔





701












702



  if (leftTableBlockIndex >= pSupporter->numOfBlocksPerTable[leftIndex]) {




109,712,058✔





703



    /* left block is empty */










704



    return 1;




47,033,597✔





705



  } else if (rightTableBlockIndex >= pSupporter->numOfBlocksPerTable[rightIndex]) {




62,714,442✔





706



    /* right block is empty */










707



    return -1;




21,705,718✔





708



  }










709












710



  pLeftBlock = &pSupporter->pDataBlockInfo[leftIndex][leftTableBlockIndex];




41,007,461✔





711



  pRightBlock = &pSupporter->pDataBlockInfo[rightIndex][rightTableBlockIndex];




41,008,851✔





712












713



  return pLeftBlock->offset > pRightBlock->offset ? 1 : -1;




41,011,133✔





714



}










715












716



int32_t recordToBlockInfo(SFileDataBlockInfo* pBlockInfo, SBrinRecord* record) {




276,442,604✔





717



  int32_t        code = TSDB_CODE_SUCCESS;




276,442,604✔





718



  int32_t        lino = 0;




276,442,604✔





719



  const SRowKey* pFirstKey = NULL;




276,442,604✔





720



  const SRowKey* pLastKey = NULL;




276,442,604✔





721












722



  TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




276,442,604✔





723



  TSDB_CHECK_NULL(record, code, lino, _end, TSDB_CODE_INVALID_PARA);




276,442,604✔





724












725



  pBlockInfo->uid = record->uid;




276,442,604✔





726



  pBlockInfo->firstKey = record->firstKey.key.ts;




276,484,706✔





727



  pBlockInfo->lastKey = record->lastKey.key.ts;




276,521,798✔





728



  pBlockInfo->minVer = record->minVer;




276,580,015✔





729



  pBlockInfo->maxVer = record->maxVer;




276,577,382✔





730



  pBlockInfo->blockOffset = record->blockOffset;




276,559,063✔





731



  pBlockInfo->smaOffset = record->smaOffset;




276,569,242✔





732



  pBlockInfo->blockSize = record->blockSize;




276,508,737✔





733



  pBlockInfo->blockKeySize = record->blockKeySize;




276,517,174✔





734



  pBlockInfo->smaSize = record->smaSize;




276,500,632✔





735



  pBlockInfo->numRow = record->numRow;




276,533,630✔





736



  pBlockInfo->count = record->count;




276,463,258✔





737












738



  pFirstKey = &record->firstKey.key;




276,511,614✔





739



  pLastKey = &record->lastKey.key;




276,520,809✔





740



  TSDB_CHECK_CONDITION((pFirstKey->numOfPKs == pLastKey->numOfPKs), code, lino, _end, TSDB_CODE_INVALID_PARA);




276,447,504✔





741



  if (pFirstKey->numOfPKs > 0) {




276,501,618✔





742



    if (IS_NUMERIC_TYPE(pFirstKey->pks[0].type)) {




45,024✔





743



      pBlockInfo->firstPk.val = VALUE_GET_TRIVIAL_DATUM(pFirstKey->pks);




×







744



      pBlockInfo->lastPk.val = VALUE_GET_TRIVIAL_DATUM(pLastKey->pks);




×







745



    } else {










746



      int32_t keyLen = pFirstKey->pks[0].nData;




45,024✔





747



      char*   p = taosMemoryMalloc(keyLen + VARSTR_HEADER_SIZE);




45,024✔





748



      TSDB_CHECK_NULL(p, code, lino, _end, terrno);




45,024✔





749



      memcpy(varDataVal(p), pFirstKey->pks[0].pData, keyLen);




45,024✔





750



      varDataSetLen(p, keyLen);




45,024✔





751



      pBlockInfo->firstPk.pData = (uint8_t*)p;




45,024✔





752












753



      keyLen = pLastKey->pks[0].nData;




45,024✔





754



      p = taosMemoryMalloc(keyLen + VARSTR_HEADER_SIZE);




45,024✔





755



      TSDB_CHECK_NULL(p, code, lino, _end, terrno);




45,024✔





756



      memcpy(varDataVal(p), pLastKey->pks[0].pData, keyLen);




45,024✔





757



      varDataSetLen(p, keyLen);




45,024✔





758



      pBlockInfo->lastPk.pData = (uint8_t*)p;




45,024✔





759



    }










760



  }










761












762



_end:




276,388,085✔





763



  if (code != TSDB_CODE_SUCCESS) {




276,433,109✔





764



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







765



  }










766



  return code;




276,380,405✔





767



}










768












769



static void freePkItem(void* pItem) {




45,024✔





770



  SFileDataBlockInfo* p = pItem;




45,024✔





771



  if (p != NULL) {




45,024✔





772



    taosMemoryFreeClear(p->firstPk.pData);




45,024✔





773



    taosMemoryFreeClear(p->lastPk.pData);




45,024✔





774



  }










775



}




45,024✔





776












777



void clearDataBlockIterator(SDataBlockIter* pIter, bool needFree) {




502,930,193✔





778



  if (pIter == NULL) {




502,930,193✔





779



    return;




×







780



  }










781












782



  pIter->index = -1;




502,930,193✔





783



  pIter->numOfBlocks = 0;




503,818,391✔





784












785



  if (needFree) {




502,944,110✔





786



    taosArrayClearEx(pIter->blockList, freePkItem);




51,764,312✔





787



  } else {










788



    taosArrayClear(pIter->blockList);




451,179,798✔





789



  }










790



}










791












792



void cleanupDataBlockIterator(SDataBlockIter* pIter, bool needFree) {




304,284,790✔





793



  if (pIter == NULL) {




304,284,790✔





794



    return;




×







795



  }










796












797



  pIter->index = -1;




304,284,790✔





798



  pIter->numOfBlocks = 0;




304,411,846✔





799



  if (needFree) {




303,946,989✔





800



    taosArrayDestroyEx(pIter->blockList, freePkItem);




8,036,837✔





801



  } else {










802



    taosArrayDestroy(pIter->blockList);




295,910,152✔





803



  }










804



  pIter->blockList = NULL;




304,180,927✔





805



}










806












807



int32_t initBlockIterator(STsdbReader* pReader, SDataBlockIter* pBlockIter, int32_t numOfBlocks, SArray* pTableList) {




21,161,580✔





808



  int32_t                   code = TSDB_CODE_SUCCESS;




21,161,580✔





809



  int32_t                   lino = 0;




21,161,580✔





810



  bool                      asc = false;




21,161,580✔





811



  int32_t                   numOfTables = 0;




21,161,580✔





812



  int64_t                   st = 0;




21,161,580✔





813



  int64_t                   et = 0;




21,161,580✔





814



  int32_t                   cnt = 0;




21,161,580✔





815



  SBlockOrderSupporter      sup = {0};




21,161,580✔





816



  SMultiwayMergeTreeInfo*   pTree = NULL;




21,162,243✔





817



  STableBlockScanInfo*      pTableScanInfo = NULL;




21,162,996✔





818



  const SFileDataBlockInfo* pBlockInfo = NULL;




21,162,996✔





819



  const void*               px = NULL;




21,162,996✔





820












821



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




21,162,996✔





822



  TSDB_CHECK_NULL(pBlockIter, code, lino, _end, TSDB_CODE_INVALID_PARA);




21,162,996✔





823



  TSDB_CHECK_CONDITION(numOfBlocks >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);




21,162,996✔





824












825



  asc = ASCENDING_TRAVERSE(pReader->info.order);




21,162,996✔





826



  clearDataBlockIterator(pBlockIter, shouldFreePkBuf(&pReader->suppInfo));




21,162,715✔





827



  pBlockIter->numOfBlocks = numOfBlocks;




21,165,607✔





828












829



  // access data blocks according to the offset of each block in asc/desc order.










830



  numOfTables = taosArrayGetSize(pTableList);




21,165,616✔





831












832



  st = taosGetTimestampUs();




21,165,501✔





833



  code = initBlockOrderSupporter(&sup, numOfTables);




21,165,501✔





834



  TSDB_CHECK_CODE(code, lino, _end);




21,157,794✔





835












836



  for (int32_t i = 0; i < numOfTables; ++i) {




64,021,536✔





837



    pTableScanInfo = taosArrayGetP(pTableList, i);




42,864,899✔





838












839



    size_t              num = taosArrayGetSize(pTableScanInfo->pBlockList);




42,868,508✔





840



    SBlockOrderWrapper* buf = taosMemoryMalloc(sizeof(SBlockOrderWrapper) * num);




42,870,022✔





841



    TSDB_CHECK_NULL(buf, code, lino, _end, terrno);




42,861,407✔





842



    sup.numOfBlocksPerTable[sup.numOfTables] = num;




42,861,407✔





843



    sup.pDataBlockInfo[sup.numOfTables] = buf;




42,859,903✔





844



    sup.numOfTables++;




42,863,824✔





845












846



    for (int32_t k = 0; k < num; ++k) {




319,133,315✔





847



      pBlockInfo = taosArrayGet(pTableScanInfo->pBlockList, k);




276,269,573✔





848



      TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




276,191,899✔





849












850



      sup.pDataBlockInfo[i][k] =




552,457,824✔





851



          (SBlockOrderWrapper){.uid = pTableScanInfo->uid, .offset = pBlockInfo->blockOffset, .pInfo = pTableScanInfo};




276,230,183✔





852



      cnt++;




276,269,491✔





853



    }










854



  }










855












856



  TSDB_CHECK_CONDITION(!(numOfBlocks != cnt && sup.numOfTables != numOfTables), code, lino, _end,




21,156,637✔





857



                       TSDB_CODE_INVALID_PARA);










858












859



  // since there is only one table qualified, blocks are not sorted










860



  if (sup.numOfTables == 1) {




21,156,637✔





861



    pTableScanInfo = taosArrayGetP(pTableList, 0);




13,649,813✔





862



    for (int32_t i = 0; i < numOfBlocks; ++i) {




245,264,635✔





863



      STableDataBlockIdx tableDataBlockIdx = {.globalIndex = i};




231,601,916✔





864



      px = taosArrayPush(pTableScanInfo->pBlockIdxList, &tableDataBlockIdx);




231,611,499✔





865



      TSDB_CHECK_NULL(px, code, lino, _end, terrno);




231,596,458✔





866



    }










867












868



    px = taosArrayAddAll(pBlockIter->blockList, pTableScanInfo->pBlockList);




13,662,719✔





869



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




13,652,645✔





870












871



    taosArrayDestroy(pTableScanInfo->pBlockList);




13,652,645✔





872



    pTableScanInfo->pBlockList = NULL;




13,652,972✔





873












874



    et = taosGetTimestampUs();




13,652,790✔





875



    tsdbDebug("%p create blocks info struct completed for one table, %d blocks not sorted, elapsed time:%.2f ms %s",




13,652,790✔





876



              pReader, numOfBlocks, (et - st) / 1000.0, pReader->idStr);










877












878



    pBlockIter->index = asc ? 0 : (numOfBlocks - 1);




13,648,993✔





879



    goto _end;




13,650,273✔





880



  }










881












882



  tsdbDebug("%p create data blocks info struct completed, %d blocks in %d tables %s", pReader, cnt, sup.numOfTables,




7,506,824✔





883



            pReader->idStr);










884












885



  code = tMergeTreeCreate(&pTree, sup.numOfTables, &sup, fileDataBlockOrderCompar);




7,506,824✔





886



  TSDB_CHECK_CODE(code, lino, _end);




7,513,434✔





887












888



  for (int32_t i = 0; i < cnt; ++i) {




52,658,216✔





889



    int32_t pos = tMergeTreeGetChosenIndex(pTree);




45,144,391✔





890



    int32_t index = sup.indexPerTable[pos]++;




45,144,000✔





891



    pTableScanInfo = sup.pDataBlockInfo[pos][index].pInfo;




45,144,391✔





892












893



    pBlockInfo = taosArrayGet(pTableScanInfo->pBlockList, index);




45,144,391✔





894



    TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, terrno);




45,142,487✔





895












896



    px = taosArrayPush(pBlockIter->blockList, pBlockInfo);




45,142,487✔





897



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




45,122,239✔





898












899



    STableDataBlockIdx tableDataBlockIdx = {.globalIndex = i};




45,122,239✔





900



    px = taosArrayPush(pTableScanInfo->pBlockIdxList, &tableDataBlockIdx);




45,121,767✔





901



    TSDB_CHECK_NULL(px, code, lino, _end, terrno);




45,131,738✔





902












903



    // set data block index overflow, in order to disable the offset comparator










904



    if (sup.indexPerTable[pos] >= sup.numOfBlocksPerTable[pos]) {




45,131,738✔





905



      sup.indexPerTable[pos] = sup.numOfBlocksPerTable[pos] + 1;




29,217,821✔





906



    }










907












908



    code = tMergeTreeAdjust(pTree, tMergeTreeGetAdjustIndex(pTree));




45,130,527✔





909



    TSDB_CHECK_CODE(code, lino, _end);




45,144,476✔





910



  }










911












912



  for (int32_t i = 0; i < numOfTables; ++i) {




36,733,228✔





913



    pTableScanInfo = taosArrayGetP(pTableList, i);




29,219,709✔





914



    taosArrayDestroy(pTableScanInfo->pBlockList);




29,219,237✔





915



    pTableScanInfo->pBlockList = NULL;




29,219,237✔





916



  }










917












918



  et = taosGetTimestampUs();




7,513,519✔





919



  tsdbDebug("%p %d data blocks access order completed, elapsed time:%.2f ms %s", pReader, numOfBlocks,




7,513,519✔





920



            (et - st) / 1000.0, pReader->idStr);










921












922



  pBlockIter->index = asc ? 0 : (numOfBlocks - 1);




7,513,519✔





923












924



_end:




21,153,096✔





925



  if (code != TSDB_CODE_SUCCESS) {




21,156,997✔





926



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







927



  }










928



  cleanupBlockOrderSupporter(&sup);




21,156,997✔





929



  if (pTree != NULL) {




21,162,961✔





930



    tMergeTreeDestroy(&pTree);




7,513,519✔





931



  }










932



  return code;




21,162,961✔





933



}










934












935



bool blockIteratorNext(SDataBlockIter* pBlockIter) {




115,834,514✔





936



  bool asc = false;




115,834,514✔





937












938



  if (pBlockIter == NULL) {




115,834,514✔





939



    return false;




×







940



  }










941












942



  asc = ASCENDING_TRAVERSE(pBlockIter->order);




115,834,514✔





943



  if ((pBlockIter->index >= pBlockIter->numOfBlocks - 1 && asc) || (pBlockIter->index <= 0 && (!asc))) {




115,932,582✔





944



    return false;




17,825,700✔





945



  }










946



  pBlockIter->index += asc ? 1 : -1;




98,102,941✔





947



  return true;




98,102,428✔





948



}










949












950



typedef enum {










951



  BLK_CHECK_CONTINUE = 0x1,










952



  BLK_CHECK_QUIT = 0x2,










953



} ETombBlkCheckEnum;










954












955



static int32_t loadNextStatisticsBlock(SSttFileReader* pSttFileReader, STbStatisBlock* pStatisBlock,










956



                                       const TStatisBlkArray* pStatisBlkArray, int32_t numOfRows, int32_t* i,










957



                                       int32_t* j);










958



static int32_t doCheckTombBlock(STombBlock* pBlock, STsdbReader* pReader, int32_t numOfTables, int32_t* j,




7,990,697✔





959



                                ETombBlkCheckEnum* pRet) {










960



  int32_t              code = TSDB_CODE_SUCCESS;




7,990,697✔





961



  int32_t              lino = 0;




7,990,697✔





962



  STombRecord          record = {0};




7,990,697✔





963



  uint64_t             uid = 0;




7,990,697✔





964



  STableBlockScanInfo* pScanInfo = NULL;




7,990,697✔





965












966



  TSDB_CHECK_NULL(pBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);




7,990,697✔





967



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




7,990,697✔





968



  TSDB_CHECK_NULL(pRet, code, lino, _end, TSDB_CODE_INVALID_PARA);




7,990,697✔





969












970



  *pRet = BLK_CHECK_QUIT;




7,990,697✔





971



  uid = pReader->status.uidList.tableUidList[*j];




7,990,697✔





972



  code = getTableBlockScanInfo(pReader->status.pTableMap, uid, &pScanInfo, pReader->idStr);




7,990,697✔





973



  TSDB_CHECK_CODE(code, lino, _end);




7,990,697✔





974












975



  if (pScanInfo->pFileDelData == NULL) {




7,990,697✔





976



    pScanInfo->pFileDelData = taosArrayInit(4, sizeof(SDelData));




3,009,325✔





977



    TSDB_CHECK_NULL(pScanInfo->pFileDelData, code, lino, _end, terrno);




3,009,325✔





978



  }










979












980



  for (int32_t k = 0; k < pBlock->numOfRecords; ++k) {




112,197,093✔





981



    code = tTombBlockGet(pBlock, k, &record);




107,693,396✔





982



    TSDB_CHECK_CODE(code, lino, _end);




107,693,396✔





983












984



    if (record.suid < pReader->info.suid) {




107,693,396✔






UNCOV

985



      continue;




×







986



    }










987












988



    if (record.suid > pReader->info.suid) {




107,693,396✔





989



      *pRet = BLK_CHECK_QUIT;




3,398,008✔





990



      goto _end;




3,398,008✔





991



    }










992












993



    if (uid < record.uid) {




104,295,388✔





994



      while ((*j) < numOfTables && pReader->status.uidList.tableUidList[*j] < record.uid) {




5,903,804✔





995



        (*j) += 1;




2,969,102✔





996



      }










997












998



      if ((*j) >= numOfTables) {




2,934,702✔





999



        *pRet = BLK_CHECK_QUIT;




88,992✔





1000



        goto _end;




88,992✔





1001



      }










1002












1003



      uid = pReader->status.uidList.tableUidList[*j];




2,845,710✔





1004



      code = getTableBlockScanInfo(pReader->status.pTableMap, uid, &pScanInfo, pReader->idStr);




2,845,710✔





1005



      TSDB_CHECK_CODE(code, lino, _end);




2,845,710✔





1006












1007



      if (pScanInfo->pFileDelData == NULL) {




2,845,710✔





1008



        pScanInfo->pFileDelData = taosArrayInit(4, sizeof(SDelData));




1,566,774✔





1009



        TSDB_CHECK_NULL(pScanInfo->pFileDelData, code, lino, _end, terrno);




1,566,774✔





1010



      }










1011



    }










1012












1013



    if (record.uid < uid) {




104,206,396✔





1014



      continue;




46,699,637✔





1015



    }










1016












1017



    TSDB_CHECK_CONDITION((record.suid == pReader->info.suid) && (uid == record.uid), code, lino, _end,




57,506,759✔





1018



                         TSDB_CODE_INTERNAL_ERROR);










1019












1020



    if (record.version <= pReader->info.verRange.maxVer) {




57,506,759✔





1021



      SDelData    delData = {.version = record.version, .sKey = record.skey, .eKey = record.ekey};




57,506,759✔





1022



      const void* px = taosArrayPush(pScanInfo->pFileDelData, &delData);




57,506,759✔





1023



      TSDB_CHECK_NULL(px, code, lino, _end, terrno);




57,506,759✔





1024



    }










1025



  }










1026












1027



  *pRet = BLK_CHECK_CONTINUE;




4,503,697✔





1028












1029



_end:




7,990,697✔





1030



  if (code != TSDB_CODE_SUCCESS) {




7,990,697✔





1031



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1032



  }










1033



  return code;




7,990,697✔





1034



}










1035












1036



// load tomb data API










1037



static int32_t doLoadTombDataFromTombBlk(const TTombBlkArray* pTombBlkArray, STsdbReader* pReader, void* pFileReader,




437,804,493✔





1038



                                         bool isFile) {










1039



  int32_t              code = TSDB_CODE_SUCCESS;




437,804,493✔





1040



  int32_t              lino = 0;




437,804,493✔





1041



  const STableUidList* pList = NULL;




437,804,493✔





1042



  int32_t              numOfTables = 0;




437,804,493✔





1043












1044



  TSDB_CHECK_NULL(pTombBlkArray, code, lino, _end, TSDB_CODE_INVALID_PARA);




437,804,493✔





1045



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




437,804,493✔





1046












1047



  pList = &pReader->status.uidList;




437,804,493✔





1048



  numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);




437,931,937✔





1049












1050



  int32_t i = 0, j = 0;




438,644,315✔





1051



  while (i < pTombBlkArray->size && j < numOfTables) {




447,486,318✔





1052



    STombBlk* pTombBlk = &pTombBlkArray->data[i];




15,029,808✔





1053



    if (pTombBlk->maxTbid.suid < pReader->info.suid) {




15,029,150✔





1054



      i += 1;




3,178,478✔





1055



      continue;




3,178,478✔





1056



    }










1057












1058



    if (pTombBlk->minTbid.suid > pReader->info.suid) {




11,851,330✔





1059



      break;




2,561,477✔





1060



    }










1061












1062



    TSDB_CHECK_CONDITION(




9,289,853✔





1063



        (pTombBlk->minTbid.suid <= pReader->info.suid) && (pTombBlk->maxTbid.suid >= pReader->info.suid), code, lino,










1064



        _end, TSDB_CODE_INTERNAL_ERROR);










1065












1066



    if (pTombBlk->maxTbid.suid == pReader->info.suid && pTombBlk->maxTbid.uid < pList->tableUidList[0]) {




9,289,853✔





1067



      i += 1;




1,177,276✔





1068



      continue;




1,177,276✔





1069



    }










1070












1071



    if (pTombBlk->minTbid.suid == pReader->info.suid && pTombBlk->minTbid.uid > pList->tableUidList[numOfTables - 1]) {




8,112,577✔





1072



      break;




121,880✔





1073



    }










1074












1075



    STombBlock block = {0};




7,990,697✔





1076



    code = isFile ? tsdbDataFileReadTombBlock(pFileReader, &pTombBlkArray->data[i], &block)




3,539,904✔





1077



                  : tsdbSttFileReadTombBlock(pFileReader, &pTombBlkArray->data[i], &block);




7,990,697✔





1078



    TSDB_CHECK_CODE(code, lino, _end);




7,990,697✔





1079












1080



    ETombBlkCheckEnum ret = 0;




7,990,697✔





1081



    code = doCheckTombBlock(&block, pReader, numOfTables, &j, &ret);




7,990,697✔





1082












1083



    tTombBlockDestroy(&block);




7,990,697✔





1084



    TSDB_CHECK_CODE(code, lino, _end);




7,990,697✔





1085



    if (ret == BLK_CHECK_QUIT) {




7,990,697✔





1086



      break;




3,487,000✔





1087



    }










1088












1089



    i += 1;




4,503,697✔





1090



  }










1091












1092



_end:




438,302,961✔





1093



  if (code != TSDB_CODE_SUCCESS) {




438,533,940✔





1094



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1095



  }










1096



  return code;




438,533,915✔





1097



}










1098












1099



int32_t loadDataFileTombDataForAll(STsdbReader* pReader) {




525,870,066✔





1100



  int32_t              code = TSDB_CODE_SUCCESS;




525,870,066✔





1101



  int32_t              lino = 0;




525,870,066✔





1102



  const TTombBlkArray* pBlkArray = NULL;




525,870,066✔





1103












1104



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




526,193,049✔





1105












1106



  if ((pReader->status.pCurrentFileset == NULL) || (pReader->status.pCurrentFileset->farr[TSDB_FTYPE_TOMB] == NULL)) {




526,193,049✔





1107



    return TSDB_CODE_SUCCESS;




521,545,337✔





1108



  }










1109












1110



  code = tsdbDataFileReadTombBlk(pReader->pFileReader, &pBlkArray);




4,772,864✔





1111



  TSDB_CHECK_CODE(code, lino, _end);




4,772,864✔





1112












1113



  code = doLoadTombDataFromTombBlk(pBlkArray, pReader, pReader->pFileReader, true);




4,772,864✔





1114



  TSDB_CHECK_CODE(code, lino, _end);




4,772,864✔





1115












1116



_end:




4,772,864✔





1117



  if (code != TSDB_CODE_SUCCESS) {




4,772,864✔





1118



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1119



  }










1120



  return code;




4,772,864✔





1121



}










1122












1123



int32_t loadSttTombDataForAll(STsdbReader* pReader, SSttFileReader* pSttFileReader, SSttBlockLoadInfo* pLoadInfo) {




433,182,344✔





1124



  int32_t              code = TSDB_CODE_SUCCESS;




433,182,344✔





1125



  int32_t              lino = 0;




433,182,344✔





1126



  const TTombBlkArray* pBlkArray = NULL;




433,182,344✔





1127












1128



  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




433,409,123✔





1129












1130



  code = tsdbSttFileReadTombBlk(pSttFileReader, &pBlkArray);




433,409,123✔





1131



  TSDB_CHECK_CODE(code, lino, _end);




433,478,200✔





1132












1133



  code = doLoadTombDataFromTombBlk(pBlkArray, pReader, pSttFileReader, false);




433,478,200✔





1134



  TSDB_CHECK_CODE(code, lino, _end);




433,455,826✔





1135












1136



_end:




433,455,826✔





1137



  if (code != TSDB_CODE_SUCCESS) {




433,525,889✔





1138



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1139



  }










1140



  return code;




433,525,889✔





1141



}










1142












1143



int32_t loadMemTombData(SArray** ppMemDelData, STbData* pMemTbData, STbData* piMemTbData, int64_t ver) {




675,720,175✔





1144



  int32_t         code = TSDB_CODE_SUCCESS;




675,720,175✔





1145



  int32_t         lino = 0;




675,720,175✔





1146



  SArray*         pMemDelData = NULL;




675,720,175✔





1147



  const SDelData* p = NULL;




675,720,175✔





1148



  const void*     px = NULL;




675,720,175✔





1149












1150



  TSDB_CHECK_NULL(ppMemDelData, code, lino, _end, TSDB_CODE_INVALID_PARA);




675,720,175✔





1151












1152



  if (*ppMemDelData == NULL) {




675,720,175✔





1153



    *ppMemDelData = taosArrayInit(4, sizeof(SDelData));




671,355,013✔





1154



    TSDB_CHECK_NULL(*ppMemDelData, code, lino, _end, terrno);




671,139,857✔





1155



  }










1156












1157



  pMemDelData = *ppMemDelData;




676,064,855✔





1158












1159



  if (pMemTbData != NULL) {




675,864,165✔





1160



    taosRLockLatch(&pMemTbData->lock);




418,306,228✔





1161



    p = pMemTbData->pHead;




418,638,933✔





1162



    while (p) {




419,738,551✔





1163



      if (p->version <= ver) {




1,512,679✔





1164



        px = taosArrayPush(pMemDelData, p);




1,463,286✔





1165



        if (px == NULL) {




1,463,286✔





1166



          taosRUnLockLatch(&pMemTbData->lock);




×







1167



          TSDB_CHECK_NULL(px, code, lino, _end, terrno);




×







1168



        }










1169



      }










1170



      p = p->pNext;




1,513,207✔





1171



    }










1172



    taosRUnLockLatch(&pMemTbData->lock);




418,225,872✔





1173



  }










1174












1175



  if (piMemTbData != NULL) {




676,324,436✔





1176



    p = piMemTbData->pHead;




12,255,484✔





1177



    while (p) {




13,036,913✔





1178



      if (p->version <= ver) {




781,901✔





1179



        px = taosArrayPush(pMemDelData, p);




781,901✔





1180



        TSDB_CHECK_NULL(px, code, lino, _end, terrno);




781,901✔





1181



      }










1182



      p = p->pNext;




781,901✔





1183



    }










1184



  }










1185












1186



_end:




676,323,964✔





1187



  if (code != TSDB_CODE_SUCCESS) {




676,323,964✔





1188



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1189



  }










1190



  return code;




675,773,701✔





1191



}










1192












1193



int32_t getNumOfRowsInSttBlock(SSttFileReader* pSttFileReader, SSttBlockLoadInfo* pBlockLoadInfo,




3,506✔





1194



                               TStatisBlkArray* pStatisBlkArray, uint64_t suid, const uint64_t* pUidList,










1195



                               int32_t numOfTables, int32_t* pNumOfRows) {










1196



  int32_t           code = TSDB_CODE_SUCCESS;




3,506✔





1197



  int32_t           lino = 0;




3,506✔





1198



  int32_t           num = 0;




3,506✔





1199



  int64_t           st = 0;




3,506✔





1200



  const SStatisBlk* p = NULL;




3,506✔





1201



  STbStatisBlock*   pStatisBlock = NULL;




3,506✔





1202












1203



  TSDB_CHECK_NULL(pSttFileReader, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,506✔





1204



  TSDB_CHECK_NULL(pBlockLoadInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,506✔





1205



  TSDB_CHECK_NULL(pStatisBlkArray, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,506✔





1206



  TSDB_CHECK_NULL(pNumOfRows, code, lino, _end, TSDB_CODE_INVALID_PARA);




3,506✔





1207












1208



  *pNumOfRows = 0;




3,506✔





1209












1210



  if (TARRAY2_SIZE(pStatisBlkArray) <= 0) {




3,506✔





1211



    return code;




×







1212



  }










1213












1214



  int32_t i = 0;




3,506✔





1215



  while ((i < TARRAY2_SIZE(pStatisBlkArray)) && (pStatisBlkArray->data[i].maxTbid.suid < suid)) {




3,506✔





1216



    ++i;




×







1217



  }










1218












1219



  if (i >= TARRAY2_SIZE(pStatisBlkArray)) {




3,506✔





1220



    goto _end;




×







1221



  }










1222












1223



  p = &pStatisBlkArray->data[i];




3,506✔





1224



  pStatisBlock = taosMemoryCalloc(1, sizeof(STbStatisBlock));




3,506✔





1225



  TSDB_CHECK_NULL(pStatisBlock, code, lino, _end, terrno);




3,506✔





1226












1227



  code = tStatisBlockInit(pStatisBlock);




3,506✔





1228



  TSDB_CHECK_CODE(code, lino, _end);




3,506✔





1229












1230



  st = taosGetTimestampUs();




3,506✔





1231



  code = tsdbSttFileReadStatisBlock(pSttFileReader, p, pStatisBlock);




3,506✔





1232



  TSDB_CHECK_CODE(code, lino, _end);




3,506✔





1233












1234



  double el = (taosGetTimestampUs() - st) / 1000.0;




3,506✔





1235



  pBlockLoadInfo->cost.loadStatisBlocks += 1;




3,506✔





1236



  pBlockLoadInfo->cost.statisElapsedTime += el;




3,506✔





1237












1238



  int32_t index = 0;




3,506✔





1239



  while (index < pStatisBlock->numOfRecords && ((int64_t*)pStatisBlock->suids.data)[index] < suid) {




3,506✔





1240



    ++index;




×







1241



  }










1242












1243



  if (index >= pStatisBlock->numOfRecords) {




3,506✔





1244



    *pNumOfRows = num;




×







1245



    goto _end;




×







1246



  }










1247












1248



  int32_t j = index;




3,506✔





1249



  int32_t uidIndex = 0;




3,506✔





1250



  while (i < TARRAY2_SIZE(pStatisBlkArray) && uidIndex < numOfTables) {




6,453✔





1251



    p = &pStatisBlkArray->data[i];




3,539✔





1252



    if (p->minTbid.suid > suid) {




3,539✔





1253



      *pNumOfRows = num;




592✔





1254



      goto _end;




592✔





1255



    }










1256












1257



    uint64_t uid = pUidList[uidIndex];




2,947✔





1258












1259



    if (((int64_t*)pStatisBlock->uids.data)[j] == uid) {




2,947✔





1260



      num += ((int64_t*)pStatisBlock->counts.data)[j];




2,947✔





1261



      uidIndex += 1;




2,947✔





1262



      j += 1;




2,947✔





1263



      code = loadNextStatisticsBlock(pSttFileReader, pStatisBlock, pStatisBlkArray, pStatisBlock->numOfRecords, &i, &j);




2,947✔





1264



      TSDB_CHECK_CODE(code, lino, _end);




2,947✔





1265



    } else if (((int64_t*)pStatisBlock->uids.data)[j] < uid) {




×







1266



      j += 1;




×







1267



      code = loadNextStatisticsBlock(pSttFileReader, pStatisBlock, pStatisBlkArray, pStatisBlock->numOfRecords, &i, &j);




×







1268



      TSDB_CHECK_CODE(code, lino, _end);




×







1269



    } else {










1270



      uidIndex += 1;




×







1271



    }










1272



  }










1273












1274



  *pNumOfRows = num;




2,914✔





1275












1276



_end:




3,506✔





1277



  if (code != TSDB_CODE_SUCCESS) {




3,506✔





1278



    tsdbError("%s with %p failed at line %d since %s", __func__, pSttFileReader, lino, tstrerror(code));




×







1279



  }










1280



  if (pStatisBlock) {




3,506✔





1281



    tStatisBlockDestroy(pStatisBlock);




3,506✔





1282



    taosMemoryFreeClear(pStatisBlock);




3,506✔





1283



  }










1284



  return code;




3,506✔





1285



}










1286












1287



// load next stt statistics block










1288



static int32_t loadNextStatisticsBlock(SSttFileReader* pSttFileReader, STbStatisBlock* pStatisBlock,




2,947✔





1289



                                       const TStatisBlkArray* pStatisBlkArray, int32_t numOfRows, int32_t* i,










1290



                                       int32_t* j) {










1291



  int32_t code = TSDB_CODE_SUCCESS;




2,947✔





1292



  int32_t lino = 0;




2,947✔





1293












1294



  if ((*j) >= numOfRows) {




2,947✔





1295



    (*i) += 1;




2,914✔





1296



    (*j) = 0;




2,914✔





1297



    if ((*i) < TARRAY2_SIZE(pStatisBlkArray)) {




2,914✔





1298



      code = tsdbSttFileReadStatisBlock(pSttFileReader, &pStatisBlkArray->data[(*i)], pStatisBlock);




×







1299



      if (code != 0) {




×







1300



        tsdbError("%p failed to read statisBlock, code:%s", pSttFileReader, tstrerror(code));




×







1301



        TSDB_CHECK_CODE(code, lino, _end);




×







1302



      }










1303



    }










1304



  }










1305












1306



_end:




2,947✔





1307



  if (code != TSDB_CODE_SUCCESS) {




2,947✔





1308



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1309



  }










1310



  return code;




2,947✔





1311



}










1312












1313



int32_t doAdjustValidDataIters(SArray* pLDIterList, int32_t numOfFileObj) {




612,329,801✔





1314



  int32_t     code = TSDB_CODE_SUCCESS;




612,329,801✔





1315



  int32_t     lino = 0;




612,329,801✔





1316



  int32_t     size = 0;




612,329,801✔





1317



  int32_t     inc = 0;




612,329,801✔





1318



  SLDataIter* pIter = NULL;




612,329,801✔





1319












1320



  size = taosArrayGetSize(pLDIterList);




612,545,612✔





1321












1322



  if (size < numOfFileObj) {




612,253,257✔





1323



    inc = numOfFileObj - size;




433,692,417✔





1324



    for (int32_t k = 0; k < inc; ++k) {




867,944,870✔





1325



      pIter = taosMemoryCalloc(1, sizeof(SLDataIter));




433,895,011✔





1326



      TSDB_CHECK_NULL(pIter, code, lino, _end, terrno);




432,121,478✔





1327



      void* px = taosArrayPush(pLDIterList, &pIter);




434,252,453✔





1328



      if (px == NULL) {




434,252,453✔





1329



        TSDB_CHECK_NULL(px, code, lino, _end, terrno);




×







1330



        taosMemoryFreeClear(pIter);




×







1331



      }










1332



    }










1333



  } else if (size > numOfFileObj) {  // remove unused LDataIter




178,560,840✔





1334



    inc = size - numOfFileObj;




×







1335












1336



    for (int32_t i = 0; i < inc; ++i) {




×







1337



      pIter = taosArrayPop(pLDIterList);




×







1338



      destroyLDataIter(pIter);




×







1339



    }










1340



  }










1341












1342



_end:




612,609,574✔





1343



  if (code != TSDB_CODE_SUCCESS) {




611,868,073✔





1344



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1345



  }










1346



  return code;




611,868,073✔





1347



}










1348












1349



int32_t adjustSttDataIters(SArray* pSttFileBlockIterArray, STFileSet* pFileSet) {




579,821,498✔





1350



  int32_t  code = TSDB_CODE_SUCCESS;




579,821,498✔





1351



  int32_t  lino = 0;




579,821,498✔





1352



  int32_t  numOfLevels = 0;




579,821,498✔





1353



  SSttLvl* pSttLevel = NULL;




579,821,498✔





1354



  SArray*  pList = NULL;




579,821,498✔





1355












1356



  TSDB_CHECK_NULL(pFileSet, code, lino, _end, TSDB_CODE_INVALID_PARA);




580,277,395✔





1357












1358



  numOfLevels = pFileSet->lvlArr->size;




580,277,395✔





1359












1360



  // add the list/iter placeholder










1361



  while (taosArrayGetSize(pSttFileBlockIterArray) < numOfLevels) {




1,014,803,663✔





1362



    pList = taosArrayInit(4, POINTER_BYTES);




433,138,472✔





1363



    TSDB_CHECK_NULL(pList, code, lino, _end, terrno);




433,648,280✔





1364



    void* px = taosArrayPush(pSttFileBlockIterArray, &pList);




434,015,045✔





1365



    if (px == NULL) {




434,015,045✔





1366



      taosArrayDestroy(pList);




×







1367



      TSDB_CHECK_NULL(px, code, lino, _end, terrno);




×







1368



    }










1369



  }










1370












1371



  for (int32_t j = 0; j < numOfLevels; ++j) {




1,193,017,171✔





1372



    pSttLevel = pFileSet->lvlArr->data[j];




612,039,689✔





1373



    pList = taosArrayGetP(pSttFileBlockIterArray, j);




612,255,866✔





1374



    code = doAdjustValidDataIters(pList, TARRAY2_SIZE(pSttLevel->fobjArr));




612,530,724✔





1375



    TSDB_CHECK_CODE(code, lino, _end);




611,888,588✔





1376



  }










1377












1378



_end:




580,977,482✔





1379



  if (code != TSDB_CODE_SUCCESS) {




580,571,088✔





1380



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1381



  }










1382



  return code;




580,571,088✔





1383



}










1384












1385



int32_t tsdbGetRowsInSttFiles(STFileSet* pFileSet, SArray* pSttFileBlockIterArray, STsdb* pTsdb, SMergeTreeConf* pConf,




25,073✔





1386



                              const char* pstr) {










1387



  int32_t code = TSDB_CODE_SUCCESS;




25,073✔





1388



  int32_t lino = 0;




25,073✔





1389



  int32_t numOfRows = 0;




25,073✔





1390



  int32_t numOfLevels = 0;




25,073✔





1391












1392



  TSDB_CHECK_NULL(pFileSet, code, lino, _end, TSDB_CODE_INVALID_PARA);




25,073✔





1393












1394



  // no data exists, go to end










1395



  numOfLevels = pFileSet->lvlArr->size;




25,073✔





1396



  if (numOfLevels == 0) {




25,073✔





1397



    goto _end;




21,567✔





1398



  }










1399












1400



  // add the list/iter placeholder










1401



  code = adjustSttDataIters(pSttFileBlockIterArray, pFileSet);




3,506✔





1402



  TSDB_CHECK_CODE(code, lino, _end);




3,506✔





1403












1404



  for (int32_t j = 0; j < numOfLevels; ++j) {




7,012✔





1405



    SSttLvl* pSttLevel = pFileSet->lvlArr->data[j];




3,506✔





1406



    SArray*  pList = taosArrayGetP(pSttFileBlockIterArray, j);




3,506✔





1407












1408



    for (int32_t i = 0; i < taosArrayGetSize(pList); ++i) {  // open all last file




7,012✔





1409



      SLDataIter* pIter = taosArrayGetP(pList, i);




3,506✔





1410












1411



      // open stt file reader if not opened yet










1412



      // if failed to open this stt file, ignore the error and try next one










1413



      if (pIter->pReader == NULL) {




3,506✔





1414



        SSttFileReaderConfig conf = {.tsdb = pTsdb, .szPage = pTsdb->pVnode->config.tsdbPageSize};




3,506✔





1415



        conf.file[0] = *pSttLevel->fobjArr->data[i]->f;




3,506✔





1416












1417



        const char* pName = pSttLevel->fobjArr->data[i]->fname;




3,506✔





1418



        code = tsdbSttFileReaderOpen(pName, &conf, &pIter->pReader);




3,506✔





1419



        if (code != TSDB_CODE_SUCCESS) {




3,506✔





1420



          tsdbError("open stt file reader error. file:%s, code %s, %s", pName, tstrerror(code), pstr);




×







1421



          continue;




×







1422



        }










1423



      }










1424












1425



      if (pIter->pBlockLoadInfo == NULL) {




3,506✔





1426



        code = tCreateSttBlockLoadInfo(pConf->pSchema, pConf->pCols, pConf->numOfCols, &pIter->pBlockLoadInfo);




3,506✔





1427



        if (code != TSDB_CODE_SUCCESS) {




3,506✔





1428



          tsdbError("failed to create block load info, code: out of memory, %s", pstr);




×







1429



          continue;




×







1430



        }










1431



      }










1432












1433



      // load stt blocks statis for all stt-blocks, to decide if the data of queried table exists in current stt file










1434



      TStatisBlkArray* pStatisBlkArray = NULL;




3,506✔





1435



      code = tsdbSttFileReadStatisBlk(pIter->pReader, (const TStatisBlkArray**)&pStatisBlkArray);




3,506✔





1436



      if (code != TSDB_CODE_SUCCESS) {




3,506✔





1437



        tsdbError("failed to load stt block statistics, code:%s, %s", tstrerror(code), pstr);




×







1438



        continue;




×







1439



      }










1440












1441



      // extract rows from each stt file one-by-one










1442



      STsdbReader* pReader = pConf->pReader;




3,506✔





1443



      int32_t      numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);




3,506✔





1444



      uint64_t*    pUidList = pReader->status.uidList.tableUidList;




3,506✔





1445



      int32_t      n = 0;




3,506✔





1446



      code = getNumOfRowsInSttBlock(pIter->pReader, pIter->pBlockLoadInfo, pStatisBlkArray, pConf->suid, pUidList,




3,506✔





1447



                                    numOfTables, &n);










1448



      numOfRows += n;




3,506✔





1449



      if (code) {




3,506✔





1450



        tsdbError("%s failed to get rows in stt blocks, code:%s", pstr, tstrerror(code));




×







1451



      }










1452



    }










1453



  }










1454












1455



_end:




3,506✔





1456



  if (code != TSDB_CODE_SUCCESS) {




25,073✔





1457



    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







1458



  }










1459



  return numOfRows;




25,073✔





1460



}










1461












1462



static bool overlapHelper(const STimeWindow* pLeft, TSKEY minKey, TSKEY maxKey) {




4,725,319✔





1463



  return (pLeft != NULL) && (pLeft->ekey >= minKey) && (pLeft->skey <= maxKey);




4,725,319✔





1464



}










1465












1466



static bool overlapWithTimeWindow(STimeWindow* p1, STimeWindow* pQueryWindow, STableBlockScanInfo* pBlockScanInfo,




12,593,435✔





1467



                                  int32_t order) {










1468



  SIterInfo*   pMemIter = NULL;




12,593,435✔





1469



  SIterInfo*   pIMemIter = NULL;




12,593,435✔





1470



  STbData*     pTbData = NULL;




12,593,435✔





1471



  STimeWindow* pFileWin = NULL;




12,593,435✔





1472












1473



  if (p1 == NULL || pQueryWindow == NULL) {




12,593,435✔





1474



    return false;




×







1475



  }










1476












1477



  // overlap with query window










1478



  if (!(p1->skey >= pQueryWindow->skey && p1->ekey <= pQueryWindow->ekey)) {




12,614,309✔





1479



    return true;




132,608✔





1480



  }










1481












1482



  if (pBlockScanInfo == NULL) {




12,473,553✔





1483



    return false;




×







1484



  }










1485












1486



  pMemIter = &pBlockScanInfo->iter;




12,473,553✔





1487



  pIMemIter = &pBlockScanInfo->iiter;




12,493,929✔





1488












1489



  // overlap with mem data










1490



  if (pMemIter->hasVal) {




12,481,726✔





1491



    pTbData = pMemIter->iter->pTbData;




4,694,481✔





1492



    if (overlapHelper(p1, pTbData->minKey, pTbData->maxKey)) {




4,698,576✔





1493



      return true;




4,474,929✔





1494



    }










1495



  }










1496












1497



  // overlap with imem data










1498



  if (pIMemIter->hasVal) {




7,947,841✔





1499



    pTbData = pIMemIter->iter->pTbData;




68,190✔





1500



    if (overlapHelper(p1, pTbData->minKey, pTbData->maxKey)) {




68,190✔





1501



      return true;




37,902✔





1502



    }










1503



  }










1504












1505



  // overlap with data file block










1506



  pFileWin = &pBlockScanInfo->filesetWindow;




7,948,435✔





1507



  if ((taosArrayGetSize(pBlockScanInfo->pBlockIdxList) > 0) && overlapHelper(p1, pFileWin->skey, pFileWin->ekey)) {




7,956,228✔





1508



    return true;




×







1509



  }










1510












1511



  // overlap with deletion skyline










1512



  SBrinRecord record = {.firstKey = p1->skey, .lastKey = p1->ekey};




7,974,591✔





1513



  if (overlapWithDelSkylineWithoutVer(pBlockScanInfo, &record, order)) {




7,966,587✔





1514



    return true;




55,694✔





1515



  }










1516












1517



  return false;




7,889,533✔





1518



}










1519












1520



static int32_t sortUidComparFn(const void* p1, const void* p2) {




19,191,278✔





1521



  const SSttKeyRange* px1 = p1;




19,191,278✔





1522



  const SSttKeyRange* px2 = p2;




19,191,278✔





1523












1524



  int32_t ret = tRowKeyCompare(&px1->skey, &px2->skey);




19,191,278✔





1525



  return ret;




19,297,748✔





1526



}










1527












1528



bool isCleanSttBlock(SArray* pKeyRangeList, STimeWindow* pQueryWindow, STableBlockScanInfo* pScanInfo, int32_t order) {




19,772,957✔





1529



  int32_t       num = 0;




19,772,957✔





1530



  SSttKeyRange* pRange = NULL;




19,772,957✔





1531



  STimeWindow   w;




19,772,957✔





1532












1533



  num = taosArrayGetSize(pKeyRangeList);




19,800,373✔





1534



  if (num == 0) {




19,803,298✔





1535



    return false;




7,139,969✔





1536



  }










1537












1538



  // check if it overlap with del skyline










1539



  taosArraySort(pKeyRangeList, sortUidComparFn);




12,663,329✔





1540












1541



  pRange = taosArrayGet(pKeyRangeList, 0);




12,652,913✔





1542



  if (pRange == NULL) {




12,645,308✔





1543



    return false;




×







1544



  }










1545












1546



  w = (STimeWindow){.skey = pRange->skey.ts, .ekey = pRange->ekey.ts};




12,645,308✔





1547



  if (overlapWithTimeWindow(&w, pQueryWindow, pScanInfo, order)) {




12,634,022✔





1548



    return false;




4,680,658✔





1549



  }










1550












1551



  for (int32_t i = 0; i < num - 1; ++i) {




7,919,767✔





1552



    SSttKeyRange* p1 = taosArrayGet(pKeyRangeList, i);




3,550,399✔





1553



    SSttKeyRange* p2 = taosArrayGet(pKeyRangeList, i + 1);




3,565,609✔





1554



    if (p1 == NULL || p2 == NULL) {




3,571,459✔





1555



      return false;




×







1556



    }










1557












1558



    if (p1->ekey.ts >= p2->skey.ts) {




3,571,459✔





1559



      return false;




3,557,242✔





1560



    }










1561












1562



    w = (STimeWindow){.skey = p2->skey.ts, .ekey = p2->ekey.ts};




10,707✔





1563



    bool overlap = overlapWithTimeWindow(&w, pQueryWindow, pScanInfo, order);




10,707✔





1564



    if (overlap) {




10,707✔





1565



      return false;




×







1566



    }










1567



  }










1568












1569



  return true;




4,369,368✔





1570



}










1571












1572



static bool doCheckDatablockOverlap(STableBlockScanInfo* pBlockScanInfo, const SBrinRecord* pRecord,




151,120✔





1573



                                    int32_t startIndex) {










1574



  size_t num = taosArrayGetSize(pBlockScanInfo->delSkyline);




151,120✔





1575












1576



  for (int32_t i = startIndex; i < num; i += 1) {




154,424✔





1577



    TSDBKEY* p = taosArrayGet(pBlockScanInfo->delSkyline, i);




154,424✔





1578



    if (p == NULL) {




154,424✔





1579



      return false;




×







1580



    }










1581












1582



    if (p->ts >= pRecord->firstKey.key.ts && p->ts <= pRecord->lastKey.key.ts) {




154,424✔





1583



      if (p->version >= pRecord->minVer) {




145,688✔





1584



        return true;




144,744✔





1585



      }










1586



    } else if (p->ts < pRecord->firstKey.key.ts) {  // p->ts < pBlock->minKey.ts




8,736✔





1587



      if (p->version >= pRecord->minVer) {




5,432✔





1588



        if (i < num - 1) {




3,072✔





1589



          TSDBKEY* pnext = taosArrayGet(pBlockScanInfo->delSkyline, i + 1);




3,072✔





1590



          if (pnext == NULL) {




3,072✔





1591



            return false;




×







1592



          }










1593












1594



          if (pnext->ts >= pRecord->firstKey.key.ts) {




3,072✔





1595



            return true;




3,072✔





1596



          }










1597



        } else {  // it must be the last point










1598



          if (!(p->version == 0)) {




×







1599



            tsdbError("unexpected version:%" PRId64, p->version);




×







1600



          }










1601



        }










1602



      }










1603



    } else {  // (p->ts > pBlock->maxKey.ts) {










1604



      return false;




3,304✔





1605



    }










1606



  }










1607












1608



  return false;




×







1609



}










1610












1611



static bool doCheckDatablockOverlapWithoutVersion(STableBlockScanInfo* pBlockScanInfo, const SBrinRecord* pRecord,




55,694✔





1612



                                                  int32_t startIndex) {










1613



  size_t num = taosArrayGetSize(pBlockScanInfo->delSkyline);




55,694✔





1614












1615



  for (int32_t i = startIndex; i < num; i += 1) {




55,694✔





1616



    TSDBKEY* p = taosArrayGet(pBlockScanInfo->delSkyline, i);




55,694✔





1617



    if (p == NULL) {




55,694✔





1618



      return false;




×







1619



    }










1620












1621



    if (p->ts >= pRecord->firstKey.key.ts && p->ts <= pRecord->lastKey.key.ts) {




55,694✔





1622



      return true;




55,694✔





1623



    } else if (p->ts < pRecord->firstKey.key.ts) {  // p->ts < pBlock->minKey.ts




×







1624



      if (i < num - 1) {




×







1625



        TSDBKEY* pnext = taosArrayGet(pBlockScanInfo->delSkyline, i + 1);




×







1626



        if (pnext == NULL) {




×







1627



          return false;




×







1628



        }










1629












1630



        if (pnext->ts >= pRecord->firstKey.key.ts) {




×







1631



          return true;




×







1632



        }










1633



      }










1634



    } else {  // (p->ts > pBlock->maxKey.ts) {










1635



      return false;




×







1636



    }










1637



  }










1638












1639



  return false;




×







1640



}










1641












1642



bool overlapWithDelSkyline(STableBlockScanInfo* pBlockScanInfo, const SBrinRecord* pRecord, int32_t order) {




117,909,211✔





1643



  if (pBlockScanInfo->delSkyline == NULL || (taosArrayGetSize(pBlockScanInfo->delSkyline) == 0)) {




117,909,211✔





1644



    return false;




117,470,673✔





1645



  }










1646












1647



  // ts is not overlap










1648



  TSDBKEY* pFirst = taosArrayGet(pBlockScanInfo->delSkyline, 0);




489,003✔





1649



  TSDBKEY* pLast = taosArrayGetLast(pBlockScanInfo->delSkyline);




489,003✔





1650



  if (pFirst == NULL || pLast == NULL) {




489,003✔





1651



    return false;




×







1652



  }










1653












1654



  if (pRecord->firstKey.key.ts > pLast->ts || pRecord->lastKey.key.ts < pFirst->ts) {




489,003✔





1655



    return false;




337,883✔





1656



  }










1657












1658



  // version is not overlap










1659



  if (ASCENDING_TRAVERSE(order)) {




151,120✔





1660



    return doCheckDatablockOverlap(pBlockScanInfo, pRecord, pBlockScanInfo->fileDelIndex);




75,968✔





1661



  } else {










1662



    int32_t index = pBlockScanInfo->fileDelIndex;




75,152✔





1663



    while (1) {




66,552✔





1664



      TSDBKEY* p = taosArrayGet(pBlockScanInfo->delSkyline, index);




141,704✔





1665



      if (p == NULL) {




141,704✔





1666



        return false;




×







1667



      }










1668












1669



      if (p->ts > pRecord->firstKey.key.ts && index > 0) {




141,704✔





1670



        index -= 1;




66,552✔





1671



      } else {  // find the first point that is smaller than the minKey.ts of dataBlock.










1672



        if (p->ts == pRecord->firstKey.key.ts && p->version < pRecord->maxVer && index > 0) {




75,152✔





1673



          index -= 1;




8,600✔





1674



        }










1675



        break;




75,152✔





1676



      }










1677



    }










1678












1679



    return doCheckDatablockOverlap(pBlockScanInfo, pRecord, index);




75,152✔





1680



  }










1681



}










1682












1683



bool overlapWithDelSkylineWithoutVer(STableBlockScanInfo* pBlockScanInfo, const SBrinRecord* pRecord, int32_t order) {




7,969,056✔





1684



  if (pBlockScanInfo->delSkyline == NULL || (taosArrayGetSize(pBlockScanInfo->delSkyline) == 0)) {




7,969,056✔





1685



    return false;




7,913,161✔





1686



  }










1687












1688



  // ts is not overlap










1689



  TSDBKEY* pFirst = taosArrayGet(pBlockScanInfo->delSkyline, 0);




55,694✔





1690



  TSDBKEY* pLast = taosArrayGetLast(pBlockScanInfo->delSkyline);




55,694✔





1691



  if (pFirst == NULL || pLast == NULL) {




55,694✔





1692



    return false;




×







1693



  }










1694












1695



  if (pRecord->firstKey.key.ts > pLast->ts || pRecord->lastKey.key.ts < pFirst->ts) {




55,694✔





1696



    return false;




×







1697



  }










1698












1699



  // version is not overlap










1700



  if (ASCENDING_TRAVERSE(order)) {




55,694✔





1701



    return doCheckDatablockOverlapWithoutVersion(pBlockScanInfo, pRecord, pBlockScanInfo->fileDelIndex);




55,694✔





1702



  } else {










1703



    int32_t index = pBlockScanInfo->fileDelIndex;




×







1704



    while (1) {




×







1705



      TSDBKEY* p = taosArrayGet(pBlockScanInfo->delSkyline, index);




×







1706



      if (p == NULL) {




×







1707



        return false;




×







1708



      }










1709












1710



      if (p->ts > pRecord->firstKey.key.ts && index > 0) {




×







1711



        index -= 1;




×







1712



      } else {  // find the first point that is smaller than the minKey.ts of dataBlock.










1713



        if (p->ts == pRecord->firstKey.key.ts && index > 0) {




×







1714



          index -= 1;




×







1715



        }










1716



        break;




×







1717



      }










1718



    }










1719












1720



    return doCheckDatablockOverlapWithoutVersion(pBlockScanInfo, pRecord, index);




×







1721



  }










1722



}
























    

  • 

    •