TDengine/source/libs/executor/src/dynqueryctrloperator.c

3777 lines
137 KiB
C
Raw Normal View History

2023-09-01 05:24:47 +00:00
/*
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
*
* This program is free software: you can use, redistribute, and/or modify
* it under the terms of the GNU Affero General Public License, version 3
* or later ("AGPL"), as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "executorInt.h"
#include "filter.h"
2025-01-08 03:24:24 +00:00
#include "nodes.h"
2023-09-01 05:24:47 +00:00
#include "operator.h"
#include "os.h"
2025-01-08 03:24:24 +00:00
#include "plannodes.h"
#include "query.h"
2023-09-01 05:24:47 +00:00
#include "querynodes.h"
#include "querytask.h"
2025-01-08 03:24:24 +00:00
#include "tarray.h"
2023-09-01 05:24:47 +00:00
#include "tcompare.h"
#include "tdatablock.h"
#include "thash.h"
#include "tmsg.h"
#include "trpc.h"
2023-09-01 05:24:47 +00:00
#include "ttypes.h"
#include "tdataformat.h"
2023-09-01 05:24:47 +00:00
#include "dynqueryctrl.h"
int64_t gSessionId = 0;
void freeVgTableList(void* ptr) {
taosArrayDestroy(*(SArray**)ptr);
}
static void destroyStbJoinTableList(SStbJoinTableList* pListHead) {
SStbJoinTableList* pNext = NULL;
while (pListHead) {
taosMemoryFree(pListHead->pLeftVg);
taosMemoryFree(pListHead->pLeftUid);
taosMemoryFree(pListHead->pRightVg);
taosMemoryFree(pListHead->pRightUid);
pNext = pListHead->pNext;
taosMemoryFree(pListHead);
pListHead = pNext;
}
}
static void destroyStbJoinDynCtrlInfo(SStbJoinDynCtrlInfo* pStbJoin) {
2024-03-22 00:47:07 +00:00
qDebug("dynQueryCtrl exec info, prevBlk:%" PRId64 ", prevRows:%" PRId64 ", postBlk:%" PRId64 ", postRows:%" PRId64 ", leftCacheNum:%" PRId64 ", rightCacheNum:%" PRId64,
2023-09-01 05:24:47 +00:00
pStbJoin->execInfo.prevBlkNum, pStbJoin->execInfo.prevBlkRows, pStbJoin->execInfo.postBlkNum,
pStbJoin->execInfo.postBlkRows, pStbJoin->execInfo.leftCacheNum, pStbJoin->execInfo.rightCacheNum);
if (pStbJoin->basic.batchFetch) {
if (pStbJoin->ctx.prev.leftHash) {
tSimpleHashSetFreeFp(pStbJoin->ctx.prev.leftHash, freeVgTableList);
tSimpleHashCleanup(pStbJoin->ctx.prev.leftHash);
}
if (pStbJoin->ctx.prev.rightHash) {
tSimpleHashSetFreeFp(pStbJoin->ctx.prev.rightHash, freeVgTableList);
tSimpleHashCleanup(pStbJoin->ctx.prev.rightHash);
}
} else {
if (pStbJoin->ctx.prev.leftCache) {
tSimpleHashCleanup(pStbJoin->ctx.prev.leftCache);
}
if (pStbJoin->ctx.prev.rightCache) {
tSimpleHashCleanup(pStbJoin->ctx.prev.rightCache);
}
if (pStbJoin->ctx.prev.onceTable) {
tSimpleHashCleanup(pStbJoin->ctx.prev.onceTable);
}
}
destroyStbJoinTableList(pStbJoin->ctx.prev.pListHead);
}
void destroyColRefInfo(void *info) {
SColRefInfo *pColRefInfo = (SColRefInfo *)info;
if (pColRefInfo) {
taosMemoryFree(pColRefInfo->colName);
taosMemoryFree(pColRefInfo->colrefName);
2025-07-12 09:52:44 +00:00
}
}
void destroyColRefArray(void *info) {
SArray *pColRefArray = *(SArray **)info;
if (pColRefArray) {
taosArrayDestroyEx(pColRefArray, destroyColRefInfo);
2025-07-12 09:52:44 +00:00
}
}
void freeUseDbOutput(void* pOutput) {
SUseDbOutput *pOut = *(SUseDbOutput**)pOutput;
if (NULL == pOutput) {
return;
}
if (pOut->dbVgroup) {
freeVgInfo(pOut->dbVgroup);
}
taosMemFree(pOut);
}
void destroyOtbInfoArray(void *info) {
SArray *pOtbInfoArray = *(SArray **)info;
if (pOtbInfoArray) {
taosArrayDestroyEx(pOtbInfoArray, destroySOrgTbInfo);
}
}
void destroyOtbVgIdToOtbInfoArrayMap(void *info) {
SHashObj* pOtbVgIdToOtbInfoArrayMap = *(SHashObj **)info;
if (pOtbVgIdToOtbInfoArrayMap) {
taosHashSetFreeFp(pOtbVgIdToOtbInfoArrayMap, destroyOtbInfoArray);
taosHashCleanup(pOtbVgIdToOtbInfoArrayMap);
}
}
void destroyTagList(void *info) {
SArray *pTagList = *(SArray **)info;
if (pTagList) {
taosArrayDestroyEx(pTagList, destroyTagVal);
}
}
void destroyVtbUidTagListMap(void *info) {
SHashObj* pVtbUidTagListMap = *(SHashObj **)info;
if (pVtbUidTagListMap) {
taosHashSetFreeFp(pVtbUidTagListMap, destroyTagList);
taosHashCleanup(pVtbUidTagListMap);
}
}
static void destroyVtbScanDynCtrlInfo(SVtbScanDynCtrlInfo* pVtbScan) {
if (pVtbScan->dbName) {
taosMemoryFreeClear(pVtbScan->dbName);
}
if (pVtbScan->tbName) {
taosMemoryFreeClear(pVtbScan->tbName);
2025-07-12 09:52:44 +00:00
}
if (pVtbScan->childTableList) {
2025-07-12 09:52:44 +00:00
taosArrayDestroyEx(pVtbScan->childTableList, destroyColRefArray);
}
if (pVtbScan->colRefInfo) {
taosArrayDestroyEx(pVtbScan->colRefInfo, destroyColRefInfo);
pVtbScan->colRefInfo = NULL;
}
2025-07-12 09:52:44 +00:00
if (pVtbScan->childTableMap) {
taosHashCleanup(pVtbScan->childTableMap);
}
if (pVtbScan->readColList) {
taosArrayDestroy(pVtbScan->readColList);
}
if (pVtbScan->dbVgInfoMap) {
taosHashSetFreeFp(pVtbScan->dbVgInfoMap, freeUseDbOutput);
taosHashCleanup(pVtbScan->dbVgInfoMap);
}
if (pVtbScan->otbNameToOtbInfoMap) {
taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
}
if (pVtbScan->pRsp) {
tFreeSUsedbRsp(pVtbScan->pRsp);
taosMemoryFreeClear(pVtbScan->pRsp);
}
if (pVtbScan->existOrgTbVg) {
taosHashCleanup(pVtbScan->existOrgTbVg);
}
if (pVtbScan->curOrgTbVg) {
taosHashCleanup(pVtbScan->curOrgTbVg);
}
if (pVtbScan->newAddedVgInfo) {
taosHashCleanup(pVtbScan->newAddedVgInfo);
}
if (pVtbScan->otbVgIdToOtbInfoArrayMap) {
taosHashSetFreeFp(pVtbScan->otbVgIdToOtbInfoArrayMap, destroyOtbInfoArray);
taosHashCleanup(pVtbScan->otbVgIdToOtbInfoArrayMap);
}
if (pVtbScan->vtbUidToVgIdMapMap) {
taosHashSetFreeFp(pVtbScan->vtbUidToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
taosHashCleanup(pVtbScan->vtbUidToVgIdMapMap);
}
if (pVtbScan->vtbGroupIdToVgIdMapMap) {
taosHashSetFreeFp(pVtbScan->vtbGroupIdToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
taosHashCleanup(pVtbScan->vtbGroupIdToVgIdMapMap);
}
if (pVtbScan->vtbUidTagListMap) {
taosHashSetFreeFp(pVtbScan->vtbUidTagListMap, destroyTagList);
taosHashCleanup(pVtbScan->vtbUidTagListMap);
}
if (pVtbScan->vtbGroupIdTagListMap) {
taosHashCleanup(pVtbScan->vtbGroupIdTagListMap);
}
if (pVtbScan->vtbUidToGroupIdMap) {
taosHashCleanup(pVtbScan->vtbUidToGroupIdMap);
}
}
void destroyWinArray(void *info) {
SArray *pWinArray = *(SArray **)info;
if (pWinArray) {
taosArrayDestroy(pWinArray);
}
}
static void destroyVtbWindowDynCtrlInfo(SVtbWindowDynCtrlInfo* pVtbWindow) {
if (pVtbWindow->pRes) {
blockDataDestroy(pVtbWindow->pRes);
}
if (pVtbWindow->pWins) {
taosArrayDestroyEx(pVtbWindow->pWins, destroyWinArray);
}
}
2023-09-01 05:24:47 +00:00
static void destroyDynQueryCtrlOperator(void* param) {
SDynQueryCtrlOperatorInfo* pDyn = (SDynQueryCtrlOperatorInfo*)param;
switch (pDyn->qType) {
case DYN_QTYPE_STB_HASH:
destroyStbJoinDynCtrlInfo(&pDyn->stbJoin);
break;
case DYN_QTYPE_VTB_AGG:
case DYN_QTYPE_VTB_SCAN:
destroyVtbScanDynCtrlInfo(&pDyn->vtbScan);
break;
case DYN_QTYPE_VTB_WINDOW:
destroyVtbWindowDynCtrlInfo(&pDyn->vtbWindow);
break;
2023-09-01 05:24:47 +00:00
default:
qError("unsupported dynamic query ctrl type: %d", pDyn->qType);
break;
}
taosMemoryFreeClear(param);
}
static FORCE_INLINE bool tableNeedCache(int64_t uid, SStbJoinPrevJoinCtx* pPrev, SStbJoinPostJoinCtx* pPost, bool rightTable, bool batchFetch) {
if (batchFetch) {
return true;
}
if (rightTable) {
return pPost->rightCurrUid == pPost->rightNextUid;
}
uint32_t* num = tSimpleHashGet(pPrev->leftCache, &uid, sizeof(uid));
return (NULL == num) ? false : true;
}
2024-07-22 06:35:12 +00:00
static int32_t updatePostJoinCurrTableInfo(SStbJoinDynCtrlInfo* pStbJoin) {
2023-09-01 05:24:47 +00:00
SStbJoinPrevJoinCtx* pPrev = &pStbJoin->ctx.prev;
SStbJoinPostJoinCtx* pPost = &pStbJoin->ctx.post;
SStbJoinTableList* pNode = pPrev->pListHead;
int32_t* leftVgId = pNode->pLeftVg + pNode->readIdx;
int32_t* rightVgId = pNode->pRightVg + pNode->readIdx;
int64_t* leftUid = pNode->pLeftUid + pNode->readIdx;
int64_t* rightUid = pNode->pRightUid + pNode->readIdx;
int64_t readIdx = pNode->readIdx + 1;
int64_t rightPrevUid = pPost->rightCurrUid;
pPost->leftCurrUid = *leftUid;
pPost->rightCurrUid = *rightUid;
pPost->leftVgId = *leftVgId;
pPost->rightVgId = *rightVgId;
while (true) {
if (readIdx < pNode->uidNum) {
pPost->rightNextUid = *(pNode->pRightUid + readIdx);
break;
}
pNode = pNode->pNext;
if (NULL == pNode) {
pPost->rightNextUid = 0;
break;
}
rightUid = pNode->pRightUid;
readIdx = 0;
}
pPost->leftNeedCache = tableNeedCache(*leftUid, pPrev, pPost, false, pStbJoin->basic.batchFetch);
pPost->rightNeedCache = tableNeedCache(*rightUid, pPrev, pPost, true, pStbJoin->basic.batchFetch);
2024-07-22 06:35:12 +00:00
if (!pStbJoin->basic.batchFetch && pPost->rightNeedCache && rightPrevUid != pPost->rightCurrUid) {
QRY_ERR_RET(tSimpleHashPut(pPrev->rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid), NULL, 0));
2023-09-01 05:24:47 +00:00
pStbJoin->execInfo.rightCacheNum++;
}
2024-07-22 06:35:12 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
static int32_t copyOrgTbInfo(SOrgTbInfo* pSrc, SOrgTbInfo** ppDst) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SOrgTbInfo* pTbInfo = NULL;
qDebug("start to copy org table info, vgId:%d, tbName:%s", pSrc->vgId, pSrc->tbName);
pTbInfo = taosMemoryMalloc(sizeof(SOrgTbInfo));
QUERY_CHECK_NULL(pTbInfo, code, lino, _return, terrno)
pTbInfo->vgId = pSrc->vgId;
tstrncpy(pTbInfo->tbName, pSrc->tbName, TSDB_TABLE_FNAME_LEN);
pTbInfo->colMap = taosArrayDup(pSrc->colMap, NULL);
QUERY_CHECK_NULL(pTbInfo->colMap, code, lino, _return, terrno)
*ppDst = pTbInfo;
return code;
_return:
qError("failed to copy org table info, code:%d, line:%d", code, lino);
if (pTbInfo) {
if (pTbInfo->colMap) {
taosArrayDestroy(pTbInfo->colMap);
}
taosMemoryFreeClear(pTbInfo);
}
return code;
}
static int32_t buildTagListForExchangeBasicParam(SExchangeOperatorBasicParam* pBasic, SArray* pTagList) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
STagVal tmpTag;
pBasic->tagList = taosArrayInit(1, sizeof(STagVal));
QUERY_CHECK_NULL(pBasic->tagList, code, lino, _return, terrno)
for (int32_t i = 0; i < taosArrayGetSize(pTagList); ++i) {
STagVal* pSrcTag = (STagVal*)taosArrayGet(pTagList, i);
QUERY_CHECK_NULL(pSrcTag, code, lino, _return, terrno)
tmpTag.type = pSrcTag->type;
tmpTag.cid = pSrcTag->cid;
if (IS_VAR_DATA_TYPE(pSrcTag->type)) {
tmpTag.nData = pSrcTag->nData;
tmpTag.pData = taosMemoryMalloc(tmpTag.nData);
QUERY_CHECK_NULL(tmpTag.pData, code, lino, _return, terrno)
memcpy(tmpTag.pData, pSrcTag->pData, tmpTag.nData);
} else {
tmpTag.i64 = pSrcTag->i64;
}
QUERY_CHECK_NULL(taosArrayPush(pBasic->tagList, &tmpTag), code, lino, _return, terrno)
tmpTag = (STagVal){0};
}
return code;
_return:
if (pBasic->tagList) {
taosArrayDestroyEx(pBasic->tagList, destroyTagVal);
pBasic->tagList = NULL;
}
if (tmpTag.pData) {
taosMemoryFree(tmpTag.pData);
}
qError("%s failed at line: %d, code: %d", __func__, lino, code);
return code;
}
static int32_t buildBatchOrgTbInfoForExchangeBasicParam(SExchangeOperatorBasicParam* pBasic, SArray* pOrgTbInfoArray) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SOrgTbInfo batchInfo;
pBasic->batchOrgTbInfo = taosArrayInit(1, sizeof(SOrgTbInfo));
QUERY_CHECK_NULL(pBasic->batchOrgTbInfo, code, lino, _return, terrno)
for (int32_t i = 0; i < taosArrayGetSize(pOrgTbInfoArray); ++i) {
SOrgTbInfo* pSrc = (SOrgTbInfo*)taosArrayGet(pOrgTbInfoArray, i);
QUERY_CHECK_NULL(pSrc, code, lino, _return, terrno)
batchInfo.vgId = pSrc->vgId;
tstrncpy(batchInfo.tbName, pSrc->tbName, TSDB_TABLE_FNAME_LEN);
batchInfo.colMap = taosArrayDup(pSrc->colMap, NULL);
QUERY_CHECK_NULL(batchInfo.colMap, code, lino, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pBasic->batchOrgTbInfo, &batchInfo), code, lino, _return, terrno)
batchInfo = (SOrgTbInfo){0};
}
return code;
_return:
qError("%s failed at line: %d, code: %d", __func__, lino, code);
if (pBasic->batchOrgTbInfo) {
taosArrayDestroyEx(pBasic->batchOrgTbInfo, destroySOrgTbInfo);
pBasic->batchOrgTbInfo = NULL;
}
if (batchInfo.colMap) {
taosArrayDestroy(batchInfo.colMap);
batchInfo.colMap = NULL;
}
return code;
}
2023-09-01 05:24:47 +00:00
static int32_t buildGroupCacheOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, int64_t tbUid, bool needCache, SOperatorParam* pChild) {
2024-07-22 06:35:12 +00:00
int32_t code = TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
if (NULL == *ppRes) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(pChild, OP_GET_PARAM);
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
if (pChild) {
(*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
2023-09-06 02:57:49 +00:00
if (NULL == (*ppRes)->pChildren) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(pChild, OP_GET_PARAM);
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
if (NULL == taosArrayPush((*ppRes)->pChildren, &pChild)) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(pChild, OP_GET_PARAM);
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
} else {
(*ppRes)->pChildren = NULL;
}
SGcOperatorParam* pGc = taosMemoryMalloc(sizeof(SGcOperatorParam));
if (NULL == pGc) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
pGc->sessionId = atomic_add_fetch_64(&gSessionId, 1);
pGc->downstreamIdx = downstreamIdx;
pGc->vgId = vgId;
pGc->tbUid = tbUid;
pGc->needCache = needCache;
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE;
(*ppRes)->downstreamIdx = downstreamIdx;
(*ppRes)->value = pGc;
(*ppRes)->reUse = false;
2023-09-01 05:24:47 +00:00
return TSDB_CODE_SUCCESS;
}
static int32_t buildGroupCacheNotifyOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, int64_t tbUid) {
2024-07-22 06:35:12 +00:00
int32_t code = TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
if (NULL == *ppRes) {
2024-07-22 06:35:12 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
(*ppRes)->pChildren = NULL;
SGcNotifyOperatorParam* pGc = taosMemoryMalloc(sizeof(SGcNotifyOperatorParam));
if (NULL == pGc) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
pGc->downstreamIdx = downstreamIdx;
pGc->vgId = vgId;
pGc->tbUid = tbUid;
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE;
(*ppRes)->downstreamIdx = downstreamIdx;
(*ppRes)->value = pGc;
(*ppRes)->reUse = false;
2023-09-01 05:24:47 +00:00
return TSDB_CODE_SUCCESS;
}
static int32_t buildExchangeOperatorBasicParam(SExchangeOperatorBasicParam* pBasic, ENodeType srcOpType,
EExchangeSourceType exchangeType, int32_t vgId, uint64_t groupId,
SArray* pUidList, SOrgTbInfo* pOrgTbInfo, SArray* pTagList,
SArray* pOrgTbInfoArray, STimeWindow window,
SDownstreamSourceNode* pDownstreamSourceNode,
bool tableSeq, bool isNewParam, bool isNewDeployed) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
2023-09-01 05:24:47 +00:00
qDebug("buildExchangeOperatorBasicParam, srcOpType:%d, exchangeType:%d, vgId:%d, groupId:%" PRIu64 ", tableSeq:%d, "
"isNewParam:%d, isNewDeployed:%d", srcOpType, exchangeType, vgId, groupId, tableSeq, isNewParam, isNewDeployed);
pBasic->srcOpType = srcOpType;
pBasic->vgId = vgId;
pBasic->groupid = groupId;
pBasic->window = window;
pBasic->tableSeq = tableSeq;
pBasic->type = exchangeType;
pBasic->isNewParam = isNewParam;
if (pDownstreamSourceNode) {
pBasic->isNewDeployed = true;
pBasic->newDeployedSrc.type = QUERY_NODE_DOWNSTREAM_SOURCE;
pBasic->newDeployedSrc.clientId = pDownstreamSourceNode->clientId;// current task's taskid
pBasic->newDeployedSrc.taskId = pDownstreamSourceNode->taskId;
pBasic->newDeployedSrc.fetchMsgType = TDMT_STREAM_FETCH;
pBasic->newDeployedSrc.localExec = false;
pBasic->newDeployedSrc.addr.nodeId = pDownstreamSourceNode->addr.nodeId;
memcpy(&pBasic->newDeployedSrc.addr.epSet, &pDownstreamSourceNode->addr.epSet, sizeof(SEpSet));
} else {
pBasic->isNewDeployed = false;
pBasic->newDeployedSrc = (SDownstreamSourceNode){0};
2023-09-01 05:24:47 +00:00
}
if (pUidList) {
pBasic->uidList = taosArrayDup(pUidList, NULL);
QUERY_CHECK_NULL(pBasic->uidList, code, lino, _return, terrno)
} else {
pBasic->uidList = taosArrayInit(1, sizeof(int64_t));
QUERY_CHECK_NULL(pBasic->uidList, code, lino, _return, terrno)
2024-07-29 01:48:04 +00:00
}
2023-09-01 05:24:47 +00:00
if (pOrgTbInfo) {
code = copyOrgTbInfo(pOrgTbInfo, &pBasic->orgTbInfo);
QUERY_CHECK_CODE(code, lino, _return);
} else {
pBasic->orgTbInfo = NULL;
2023-09-01 05:24:47 +00:00
}
if (pTagList) {
code = buildTagListForExchangeBasicParam(pBasic, pTagList);
QUERY_CHECK_CODE(code, lino, _return);
} else {
pBasic->tagList = NULL;
2023-09-01 05:24:47 +00:00
}
if (pOrgTbInfoArray) {
code = buildBatchOrgTbInfoForExchangeBasicParam(pBasic, pOrgTbInfoArray);
QUERY_CHECK_CODE(code, lino, _return);
} else {
pBasic->batchOrgTbInfo = NULL;
2023-09-01 05:24:47 +00:00
}
return code;
2023-09-01 05:24:47 +00:00
_return:
qError("%s failed at line: %d, code: %d", __func__, lino, code);
freeExchangeGetBasicOperatorParam(pBasic);
return code;
2023-09-01 05:24:47 +00:00
}
static int32_t buildExchangeOperatorParamImpl(SOperatorParam** ppRes, int32_t downstreamIdx, ENodeType srcOpType,
EExchangeSourceType exchangeType, int32_t vgId, uint64_t groupId,
SArray* pUidList, SOrgTbInfo* pOrgTbInfo, SArray* pTagList,
SArray* pOrgTbInfoArray, STimeWindow window,
SDownstreamSourceNode* pDownstreamSourceNode,
bool tableSeq, bool isNewParam, bool reUse, bool isNewDeployed) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SOperatorParam* pParam = NULL;
SExchangeOperatorParam* pExc = NULL;
pParam = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
pParam->downstreamIdx = downstreamIdx;
pParam->reUse = reUse;
pParam->pChildren = NULL;
pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
pExc = (SExchangeOperatorParam*)pParam->value;
pExc->multiParams = false;
code = buildExchangeOperatorBasicParam(&pExc->basic, srcOpType, exchangeType, vgId, groupId,
pUidList, pOrgTbInfo, pTagList, pOrgTbInfoArray,
window, pDownstreamSourceNode, tableSeq, isNewParam, isNewDeployed);
*ppRes = pParam;
return code;
_return:
qError("%s failed at line: %d, code: %d", __func__, lino, code);
if (pParam) {
freeOperatorParam(pParam, OP_GET_PARAM);
}
return code;
}
static int32_t buildExchangeOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, const int32_t* pVgId, int64_t* pUid) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SArray* pUidList = taosArrayInit(1, sizeof(int64_t));
QUERY_CHECK_NULL(pUidList, code, lino, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pUidList, pUid), code, lino, _return, terrno);
code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_STB_JOIN_SCAN,
*pVgId, 0, pUidList, NULL, NULL, NULL, (STimeWindow){0}, NULL, true, false, false, false);
QUERY_CHECK_CODE(code, lino, _return);
_return:
if (code) {
qError("failed to build exchange operator param, code:%d", code);
}
taosArrayDestroy(pUidList);
return code;
}
static int32_t buildExchangeOperatorParamForExternalWindow(SOperatorParam** ppRes, int32_t downstreamIdx, STimeWindow win) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_VSTB_WIN_SCAN,
0, 0, NULL, NULL, NULL, NULL, win, NULL, true, true, true, false);
QUERY_CHECK_CODE(code, lino, _return);
return code;
_return:
qError("failed to build exchange operator param for external window, code:%d", code);
return code;
}
static int32_t buildExchangeOperatorParamForVTagScan(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, tb_uid_t uid) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SArray* pUidList = NULL;
pUidList = taosArrayInit(1, sizeof(int64_t));
QUERY_CHECK_NULL(pUidList, code, lino, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pUidList, &uid), code, lino, _return, terrno)
code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, EX_SRC_TYPE_VSTB_TAG_SCAN,
vgId, 0, pUidList, NULL, NULL, NULL, (STimeWindow){0}, NULL, false, false, true, false);
QUERY_CHECK_CODE(code, lino, _return);
_return:
if (code) {
qError("failed to build exchange operator param for tag scan, code:%d", code);
}
taosArrayDestroy(pUidList);
return code;
}
static int32_t buildExchangeOperatorParamForVScan(SOperatorParam** ppRes, int32_t downstreamIdx, SOrgTbInfo* pOrgTbInfo,
SDownstreamSourceNode* pNewSource) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_VSTB_SCAN,
pOrgTbInfo->vgId, 0, NULL, pOrgTbInfo, NULL, NULL, (STimeWindow){0}, pNewSource, false, true, true, true);
QUERY_CHECK_CODE(code, lino, _return);
return code;
_return:
qError("failed to build exchange operator param for vscan, code:%d", code);
return code;
}
2023-09-01 05:24:47 +00:00
static int32_t buildBatchExchangeOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, SSHashObj* pVg) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SOperatorParam* pParam = NULL;
SExchangeOperatorBatchParam* pExc = NULL;
SExchangeOperatorBasicParam basic = {0};
pParam = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(pParam, code, line, _return, terrno);
pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
pParam->downstreamIdx = downstreamIdx;
pParam->reUse = false;
pParam->pChildren = NULL;
pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
QUERY_CHECK_NULL(pParam->value, code, line, _return, terrno);
pExc = pParam->value;
pExc->multiParams = true;
pExc->pBatchs = tSimpleHashInit(tSimpleHashGetSize(pVg), taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
QUERY_CHECK_NULL(pExc->pBatchs, code, line, _return, terrno)
tSimpleHashSetFreeFp(pExc->pBatchs, freeExchangeGetBasicOperatorParam);
int32_t iter = 0;
void* p = NULL;
while (NULL != (p = tSimpleHashIterate(pVg, p, &iter))) {
int32_t* pVgId = tSimpleHashGetKey(p, NULL);
SArray* pUidList = *(SArray**)p;
code = buildExchangeOperatorBasicParam(&basic, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN,
EX_SRC_TYPE_STB_JOIN_SCAN, *pVgId, 0,
pUidList, NULL, NULL, NULL,
(STimeWindow){0}, NULL, false, false, false);
QUERY_CHECK_CODE(code, line, _return);
QRY_ERR_RET(tSimpleHashPut(pExc->pBatchs, pVgId, sizeof(*pVgId), &basic, sizeof(basic)));
basic = (SExchangeOperatorBasicParam){0};
qTrace("build downstreamIdx %d batch scan, vgId:%d, uidNum:%" PRId64, downstreamIdx, *pVgId, (int64_t)taosArrayGetSize(pUidList));
2026-01-28 06:36:54 +00:00
// already transferred to batch param, can free here
taosArrayDestroy(pUidList);
*(SArray**)p = NULL;
}
*ppRes = pParam;
return code;
_return:
qError("failed to build batch exchange operator param, code:%d", code);
freeOperatorParam(pParam, OP_GET_PARAM);
freeExchangeGetBasicOperatorParam(&basic);
return code;
}
static int32_t buildBatchExchangeOperatorParamForVSAgg(SOperatorParam** ppRes, int32_t downstreamIdx, SArray* pTagList, uint64_t groupid, SHashObj* pBatchMaps) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SOperatorParam* pParam = NULL;
SExchangeOperatorBatchParam* pExc = NULL;
SExchangeOperatorBasicParam basic = {0};
pParam = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
pExc = pParam->value;
pExc->multiParams = true;
pExc->pBatchs = tSimpleHashInit(taosHashGetSize(pBatchMaps), taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
QUERY_CHECK_NULL(pExc->pBatchs, code, lino, _return, terrno)
tSimpleHashSetFreeFp(pExc->pBatchs, freeExchangeGetBasicOperatorParam);
size_t keyLen = 0;
void* pIter = taosHashIterate(pBatchMaps, NULL);
while (pIter != NULL) {
SArray* pOrgTbInfoArray = *(SArray**)pIter;
int32_t* vgId = (int32_t*)taosHashGetKey(pIter, &keyLen);
STimeWindow win = {.skey = INT64_MAX, .ekey = INT64_MIN};
code = buildExchangeOperatorBasicParam(&basic, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN,
EX_SRC_TYPE_VSTB_AGG_SCAN, *vgId, groupid,
NULL, NULL, pTagList, pOrgTbInfoArray,
win, NULL, false, true, false);
QUERY_CHECK_CODE(code, lino, _return);
code = tSimpleHashPut(pExc->pBatchs, vgId, sizeof(*vgId), &basic, sizeof(basic));
QUERY_CHECK_CODE(code, lino, _return);
basic = (SExchangeOperatorBasicParam){0};
pIter = taosHashIterate(pBatchMaps, pIter);
}
pParam->pChildren = NULL;
pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
pParam->downstreamIdx = downstreamIdx;
pParam->reUse = false;
*ppRes = pParam;
return code;
_return:
qError("failed to build exchange operator param for vscan, code:%d", code);
freeOperatorParam(pParam, OP_GET_PARAM);
freeExchangeGetBasicOperatorParam(&basic);
return code;
}
2024-08-22 09:07:00 +00:00
static int32_t buildMergeJoinOperatorParam(SOperatorParam** ppRes, bool initParam, SOperatorParam** ppChild0, SOperatorParam** ppChild1) {
2024-07-22 06:35:12 +00:00
int32_t code = TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
if (NULL == *ppRes) {
2024-07-22 06:35:12 +00:00
code = terrno;
return code;
2023-09-01 05:24:47 +00:00
}
(*ppRes)->pChildren = taosArrayInit(2, POINTER_BYTES);
2024-08-20 06:13:06 +00:00
if (NULL == (*ppRes)->pChildren) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
2024-08-22 09:07:00 +00:00
if (NULL == taosArrayPush((*ppRes)->pChildren, ppChild0)) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
2024-08-22 09:07:00 +00:00
*ppChild0 = NULL;
if (NULL == taosArrayPush((*ppRes)->pChildren, ppChild1)) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
2024-08-22 09:07:00 +00:00
*ppChild1 = NULL;
2023-09-01 05:24:47 +00:00
SSortMergeJoinOperatorParam* pJoin = taosMemoryMalloc(sizeof(SSortMergeJoinOperatorParam));
if (NULL == pJoin) {
2024-07-22 06:35:12 +00:00
code = terrno;
2023-09-06 02:57:49 +00:00
freeOperatorParam(*ppRes, OP_GET_PARAM);
*ppRes = NULL;
2024-07-22 06:35:12 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
pJoin->initDownstream = initParam;
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN;
(*ppRes)->value = pJoin;
(*ppRes)->reUse = false;
return TSDB_CODE_SUCCESS;
}
static int32_t buildMergeJoinNotifyOperatorParam(SOperatorParam** ppRes, SOperatorParam* pChild0, SOperatorParam* pChild1) {
int32_t code = TSDB_CODE_SUCCESS;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
if (NULL == *ppRes) {
code = terrno;
freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
return code;
}
(*ppRes)->pChildren = taosArrayInit(2, POINTER_BYTES);
if (NULL == *ppRes) {
code = terrno;
taosMemoryFreeClear(*ppRes);
freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
return code;
}
if (pChild0 && NULL == taosArrayPush((*ppRes)->pChildren, &pChild0)) {
code = terrno;
freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
*ppRes = NULL;
return code;
}
if (pChild1 && NULL == taosArrayPush((*ppRes)->pChildren, &pChild1)) {
code = terrno;
freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
*ppRes = NULL;
return code;
}
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN;
(*ppRes)->value = NULL;
(*ppRes)->reUse = false;
return TSDB_CODE_SUCCESS;
}
static int32_t buildBatchTableScanOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, SSHashObj* pVg) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t vgNum = tSimpleHashGetSize(pVg);
if (vgNum <= 0 || vgNum > 1) {
qError("Invalid vgroup num %d to build table scan operator param", vgNum);
return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
}
int32_t iter = 0;
void* p = NULL;
while (NULL != (p = tSimpleHashIterate(pVg, p, &iter))) {
SArray* pUidList = *(SArray**)p;
code = buildTableScanOperatorParam(ppRes, pUidList, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false);
if (code) {
return code;
}
taosArrayDestroy(pUidList);
*(SArray**)p = NULL;
}
return TSDB_CODE_SUCCESS;
}
static int32_t buildSingleTableScanOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t* pVgId, int64_t* pUid) {
SArray* pUidList = taosArrayInit(1, sizeof(int64_t));
if (NULL == pUidList) {
return terrno;
}
if (NULL == taosArrayPush(pUidList, pUid)) {
return terrno;
}
int32_t code = buildTableScanOperatorParam(ppRes, pUidList, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, true);
taosArrayDestroy(pUidList);
if (code) {
return code;
}
return TSDB_CODE_SUCCESS;
}
static int32_t buildSeqStbJoinOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SStbJoinPrevJoinCtx* pPrev, SStbJoinPostJoinCtx* pPost, SOperatorParam** ppParam) {
int64_t rowIdx = pPrev->pListHead->readIdx;
SOperatorParam* pSrcParam0 = NULL;
SOperatorParam* pSrcParam1 = NULL;
SOperatorParam* pGcParam0 = NULL;
SOperatorParam* pGcParam1 = NULL;
int32_t* leftVg = pPrev->pListHead->pLeftVg + rowIdx;
int64_t* leftUid = pPrev->pListHead->pLeftUid + rowIdx;
int32_t* rightVg = pPrev->pListHead->pRightVg + rowIdx;
int64_t* rightUid = pPrev->pListHead->pRightUid + rowIdx;
int32_t code = TSDB_CODE_SUCCESS;
qDebug("start %" PRId64 ":%" PRId64 "th stbJoin, left:%d,%" PRIu64 " - right:%d,%" PRIu64,
rowIdx, pPrev->tableNum, *leftVg, *leftUid, *rightVg, *rightUid);
QRY_ERR_RET(updatePostJoinCurrTableInfo(&pInfo->stbJoin));
if (pInfo->stbJoin.basic.batchFetch) {
if (pPrev->leftHash) {
code = pInfo->stbJoin.basic.srcScan[0] ? buildBatchTableScanOperatorParam(&pSrcParam0, 0, pPrev->leftHash) : buildBatchExchangeOperatorParam(&pSrcParam0, 0, pPrev->leftHash);
if (TSDB_CODE_SUCCESS == code) {
code = pInfo->stbJoin.basic.srcScan[1] ? buildBatchTableScanOperatorParam(&pSrcParam1, 1, pPrev->rightHash) : buildBatchExchangeOperatorParam(&pSrcParam1, 1, pPrev->rightHash);
}
if (TSDB_CODE_SUCCESS == code) {
tSimpleHashCleanup(pPrev->leftHash);
tSimpleHashCleanup(pPrev->rightHash);
pPrev->leftHash = NULL;
pPrev->rightHash = NULL;
}
}
} else {
code = pInfo->stbJoin.basic.srcScan[0] ? buildSingleTableScanOperatorParam(&pSrcParam0, 0, leftVg, leftUid) : buildExchangeOperatorParam(&pSrcParam0, 0, leftVg, leftUid);
if (TSDB_CODE_SUCCESS == code) {
code = pInfo->stbJoin.basic.srcScan[1] ? buildSingleTableScanOperatorParam(&pSrcParam1, 1, rightVg, rightUid) : buildExchangeOperatorParam(&pSrcParam1, 1, rightVg, rightUid);
}
}
bool initParam = pSrcParam0 ? true : false;
if (TSDB_CODE_SUCCESS == code) {
code = buildGroupCacheOperatorParam(&pGcParam0, 0, *leftVg, *leftUid, pPost->leftNeedCache, pSrcParam0);
pSrcParam0 = NULL;
}
if (TSDB_CODE_SUCCESS == code) {
code = buildGroupCacheOperatorParam(&pGcParam1, 1, *rightVg, *rightUid, pPost->rightNeedCache, pSrcParam1);
pSrcParam1 = NULL;
}
if (TSDB_CODE_SUCCESS == code) {
code = buildMergeJoinOperatorParam(ppParam, initParam, &pGcParam0, &pGcParam1);
}
if (TSDB_CODE_SUCCESS != code) {
if (pSrcParam0) {
freeOperatorParam(pSrcParam0, OP_GET_PARAM);
}
if (pSrcParam1) {
freeOperatorParam(pSrcParam1, OP_GET_PARAM);
}
if (pGcParam0) {
freeOperatorParam(pGcParam0, OP_GET_PARAM);
}
if (pGcParam1) {
freeOperatorParam(pGcParam1, OP_GET_PARAM);
}
if (*ppParam) {
freeOperatorParam(*ppParam, OP_GET_PARAM);
*ppParam = NULL;
}
}
return code;
}
static int32_t buildVtbScanOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, uint64_t uid) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SVTableScanOperatorParam* pVScan = NULL;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
(*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
pVScan = taosMemoryMalloc(sizeof(SVTableScanOperatorParam));
QUERY_CHECK_NULL(pVScan, code, lino, _return, terrno)
pVScan->pOpParamArray = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL(pVScan->pOpParamArray, code, lino, _return, terrno)
pVScan->uid = uid;
pVScan->window = pInfo->vtbScan.window;
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN;
(*ppRes)->downstreamIdx = 0;
(*ppRes)->value = pVScan;
(*ppRes)->reUse = false;
return TSDB_CODE_SUCCESS;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (pVScan) {
taosArrayDestroy(pVScan->pOpParamArray);
taosMemoryFreeClear(pVScan);
}
if (*ppRes) {
taosArrayDestroy((*ppRes)->pChildren);
taosMemoryFreeClear(*ppRes);
}
return code;
}
static int32_t buildExternalWindowOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins, int32_t idx) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SExternalWindowOperatorParam* pExtWinOp = NULL;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
pExtWinOp = taosMemoryMalloc(sizeof(SExternalWindowOperatorParam));
QUERY_CHECK_NULL(pExtWinOp, code, lino, _return, terrno)
pExtWinOp->ExtWins = taosArrayDup(pWins, NULL);
QUERY_CHECK_NULL(pExtWinOp->ExtWins, code, lino, _return, terrno)
SExtWinTimeWindow *firstWin = (SExtWinTimeWindow *)taosArrayGet(pWins, 0);
SExtWinTimeWindow *lastWin = (SExtWinTimeWindow *)taosArrayGet(pWins, taosArrayGetSize(pWins) - 1);
(*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
SOperatorParam* pExchangeOperator = NULL;
STimeWindow twin = {.skey = firstWin->tw.skey, .ekey = lastWin->tw.ekey};
code = buildExchangeOperatorParamForExternalWindow(&pExchangeOperator, 0, twin);
QUERY_CHECK_CODE(code, lino, _return);
QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pExchangeOperator), code, lino, _return, terrno)
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW;
(*ppRes)->downstreamIdx = idx;
(*ppRes)->value = pExtWinOp;
(*ppRes)->reUse = false;
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (pExtWinOp) {
if (pExtWinOp->ExtWins) {
taosArrayDestroy(pExtWinOp->ExtWins);
}
taosMemoryFree(pExtWinOp);
}
if (*ppRes) {
if ((*ppRes)->pChildren) {
taosArrayDestroy((*ppRes)->pChildren);
}
taosMemoryFree(*ppRes);
*ppRes = NULL;
}
return code;
}
static int32_t buildMergeOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins,
int32_t numOfDownstream, int32_t numOfWins) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SMergeOperatorParam* pMergeOp = NULL;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
(*ppRes)->pChildren = taosArrayInit(numOfDownstream, POINTER_BYTES);
QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
pMergeOp = taosMemoryMalloc(sizeof(SMergeOperatorParam));
QUERY_CHECK_NULL(pMergeOp, code, lino, _return, terrno)
pMergeOp->winNum = numOfWins;
for (int32_t i = 0; i < numOfDownstream; i++) {
SOperatorParam* pExternalWinParam = NULL;
code = buildExternalWindowOperatorParam(pInfo, &pExternalWinParam, pWins, i);
QUERY_CHECK_CODE(code, lino, _return);
QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pExternalWinParam), code, lino, _return, terrno)
}
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE;
(*ppRes)->downstreamIdx = 0;
(*ppRes)->value = pMergeOp;
(*ppRes)->reUse = false;
return TSDB_CODE_SUCCESS;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (pMergeOp) {
taosMemoryFree(pMergeOp);
}
if (*ppRes) {
if ((*ppRes)->pChildren) {
for (int32_t i = 0; i < taosArrayGetSize((*ppRes)->pChildren); i++) {
SOperatorParam* pChildParam = (SOperatorParam*)taosArrayGet((*ppRes)->pChildren, i);
if (pChildParam) {
SExternalWindowOperatorParam* pExtWinOp = (SExternalWindowOperatorParam*)pChildParam->value;
if (pExtWinOp) {
if (pExtWinOp->ExtWins) {
taosArrayDestroy(pExtWinOp->ExtWins);
}
taosMemoryFree(pExtWinOp);
}
taosMemoryFree(pChildParam);
}
}
taosArrayDestroy((*ppRes)->pChildren);
}
taosMemoryFree(*ppRes);
*ppRes = NULL;
}
return code;
}
static int32_t buildAggOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SOperatorParam* pParam = NULL;
SOperatorParam* pExchangeParam = NULL;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
bool freeExchange = false;
pParam = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
pParam->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL(pParam->pChildren, code, lino, _return, terrno)
pParam->value = taosMemoryMalloc(sizeof(SAggOperatorParam));
QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
code = buildBatchExchangeOperatorParamForVSAgg(&pExchangeParam, 0, NULL, 0, pVtbScan->otbVgIdToOtbInfoArrayMap);
QUERY_CHECK_CODE(code, lino, _return);
2023-09-01 05:24:47 +00:00
freeExchange = true;
2023-09-01 05:24:47 +00:00
QUERY_CHECK_NULL(taosArrayPush(pParam->pChildren, &pExchangeParam), code, lino, _return, terrno)
freeExchange = false;
pParam->opType = QUERY_NODE_PHYSICAL_PLAN_HASH_AGG;
pParam->downstreamIdx = 0;
pParam->reUse = false;
*ppRes = pParam;
return code;
_return:
if (freeExchange) {
freeOperatorParam(pExchangeParam, OP_GET_PARAM);
2023-09-01 05:24:47 +00:00
}
if (pParam) {
freeOperatorParam(pParam, OP_GET_PARAM);
2023-09-01 05:24:47 +00:00
}
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
return code;
}
static int32_t buildAggOperatorParamWithGroupId(SDynQueryCtrlOperatorInfo* pInfo, uint64_t groupid, SOperatorParam** ppRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SOperatorParam* pParam = NULL;
SOperatorParam* pExchangeParam = NULL;
SHashObj* otbVgIdToOtbInfoArrayMap = NULL;
bool freeExchange = false;
void* pIter = taosHashGet(pVtbScan->vtbGroupIdToVgIdMapMap, &groupid, sizeof(groupid));
if (!pIter) {
2023-09-06 02:57:49 +00:00
*ppRes = NULL;
2024-07-22 07:07:01 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
otbVgIdToOtbInfoArrayMap = *(SHashObj**)pIter;
2023-09-01 05:24:47 +00:00
pParam = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
2023-09-01 05:24:47 +00:00
pParam->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL(pParam->pChildren, code, lino, _return, terrno)
2023-09-01 05:24:47 +00:00
code = buildBatchExchangeOperatorParamForVSAgg(&pExchangeParam, 0, NULL, groupid, otbVgIdToOtbInfoArrayMap);
QUERY_CHECK_CODE(code, lino, _return);
2023-09-01 05:24:47 +00:00
freeExchange = true;
2023-09-01 05:24:47 +00:00
QUERY_CHECK_NULL(taosArrayPush(pParam->pChildren, &pExchangeParam), code, lino, _return, terrno)
2023-09-01 05:24:47 +00:00
freeExchange = false;
pParam->opType = QUERY_NODE_PHYSICAL_PLAN_HASH_AGG;
pParam->downstreamIdx = 0;
pParam->value = NULL;
pParam->reUse = false;
*ppRes = pParam;
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (freeExchange) {
freeOperatorParam(pExchangeParam, OP_GET_PARAM);
2023-09-01 05:24:47 +00:00
}
if (pParam) {
freeOperatorParam(pParam, OP_GET_PARAM);
}
return code;
2023-09-01 05:24:47 +00:00
}
static int32_t buildAggOperatorParamForSingleChild(SDynQueryCtrlOperatorInfo* pInfo, tb_uid_t uid, uint64_t groupid, SArray* pTagList, SOperatorParam** ppRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SOperatorParam* pParam = NULL;
SHashObj* pOtbVgIdToOtbInfoArrayMap = NULL;
void* pIter = taosHashGet(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid));
2023-09-01 05:24:47 +00:00
if (pIter) {
pOtbVgIdToOtbInfoArrayMap = *(SHashObj**)taosHashGet(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid));
2023-09-01 05:24:47 +00:00
code = buildBatchExchangeOperatorParamForVSAgg(&pParam, 0, pTagList, groupid, pOtbVgIdToOtbInfoArrayMap);
QUERY_CHECK_CODE(code, lino, _return);
*ppRes = pParam;
2023-09-01 05:24:47 +00:00
} else {
*ppRes = NULL;
2023-09-01 05:24:47 +00:00
}
2024-03-06 03:18:57 +00:00
return code;
_return:
if (pParam) {
freeOperatorParam(pParam, OP_GET_PARAM);
2023-09-06 02:57:49 +00:00
}
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
2023-09-01 05:24:47 +00:00
return code;
}
static void seqJoinLaunchNewRetrieveImpl(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
SStbJoinPrevJoinCtx* pPrev = &pStbJoin->ctx.prev;
SStbJoinPostJoinCtx* pPost = &pStbJoin->ctx.post;
SOperatorParam* pParam = NULL;
int32_t code = buildSeqStbJoinOperatorParam(pInfo, pPrev, pPost, &pParam);
if (TSDB_CODE_SUCCESS != code) {
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
}
qDebug("%s dynamic post task begin", GET_TASKID(pOperator->pTaskInfo));
2024-07-24 09:08:08 +00:00
code = pOperator->pDownstream[1]->fpSet.getNextExtFn(pOperator->pDownstream[1], pParam, ppRes);
if (*ppRes && (code == 0)) {
2024-10-23 07:37:49 +00:00
code = blockDataCheck(*ppRes);
if (code) {
qError("Invalid block data, blockDataCheck failed, error:%s", tstrerror(code));
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
}
2023-09-01 05:24:47 +00:00
pPost->isStarted = true;
pStbJoin->execInfo.postBlkNum++;
pStbJoin->execInfo.postBlkRows += (*ppRes)->info.rows;
qDebug("%s join res block retrieved", GET_TASKID(pOperator->pTaskInfo));
2023-09-01 05:24:47 +00:00
} else {
qDebug("%s Empty join res block retrieved", GET_TASKID(pOperator->pTaskInfo));
2023-09-01 05:24:47 +00:00
}
}
static int32_t notifySeqJoinTableCacheEnd(SOperatorInfo* pOperator, SStbJoinPostJoinCtx* pPost, bool leftTable) {
SOperatorParam* pGcParam = NULL;
SOperatorParam* pMergeJoinParam = NULL;
int32_t downstreamId = leftTable ? 0 : 1;
int32_t vgId = leftTable ? pPost->leftVgId : pPost->rightVgId;
int64_t uid = leftTable ? pPost->leftCurrUid : pPost->rightCurrUid;
2024-03-22 00:47:07 +00:00
qDebug("notify table %" PRIu64 " in vgId %d downstreamId %d cache end", uid, vgId, downstreamId);
2023-09-01 05:24:47 +00:00
int32_t code = buildGroupCacheNotifyOperatorParam(&pGcParam, downstreamId, vgId, uid);
if (TSDB_CODE_SUCCESS != code) {
return code;
}
code = buildMergeJoinNotifyOperatorParam(&pMergeJoinParam, pGcParam, NULL);
if (TSDB_CODE_SUCCESS != code) {
return code;
}
return optrDefaultNotifyFn(pOperator->pDownstream[1], pMergeJoinParam);
}
2024-07-22 07:07:01 +00:00
static int32_t handleSeqJoinCurrRetrieveEnd(SOperatorInfo* pOperator, SStbJoinDynCtrlInfo* pStbJoin) {
2023-09-01 05:24:47 +00:00
SStbJoinPostJoinCtx* pPost = &pStbJoin->ctx.post;
2024-09-12 06:17:14 +00:00
int32_t code = 0;
2023-09-01 05:24:47 +00:00
pPost->isStarted = false;
if (pStbJoin->basic.batchFetch) {
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
if (pPost->leftNeedCache) {
uint32_t* num = tSimpleHashGet(pStbJoin->ctx.prev.leftCache, &pPost->leftCurrUid, sizeof(pPost->leftCurrUid));
2023-09-06 02:57:49 +00:00
if (num && --(*num) <= 0) {
2024-09-12 06:17:14 +00:00
code = tSimpleHashRemove(pStbJoin->ctx.prev.leftCache, &pPost->leftCurrUid, sizeof(pPost->leftCurrUid));
if (code) {
qError("tSimpleHashRemove leftCurrUid %" PRId64 " from leftCache failed, error:%s", pPost->leftCurrUid, tstrerror(code));
QRY_ERR_RET(code);
}
2024-07-22 07:07:01 +00:00
QRY_ERR_RET(notifySeqJoinTableCacheEnd(pOperator, pPost, true));
2023-09-01 05:24:47 +00:00
}
}
if (!pPost->rightNeedCache) {
void* v = tSimpleHashGet(pStbJoin->ctx.prev.rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid));
if (NULL != v) {
2024-09-12 06:17:14 +00:00
code = tSimpleHashRemove(pStbJoin->ctx.prev.rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid));
if (code) {
qError("tSimpleHashRemove rightCurrUid %" PRId64 " from rightCache failed, error:%s", pPost->rightCurrUid, tstrerror(code));
QRY_ERR_RET(code);
}
2024-07-22 07:07:01 +00:00
QRY_ERR_RET(notifySeqJoinTableCacheEnd(pOperator, pPost, false));
2023-09-01 05:24:47 +00:00
}
}
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
2024-07-22 07:07:01 +00:00
static FORCE_INLINE int32_t seqJoinContinueCurrRetrieve(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2023-09-01 05:24:47 +00:00
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinPostJoinCtx* pPost = &pInfo->stbJoin.ctx.post;
SStbJoinPrevJoinCtx* pPrev = &pInfo->stbJoin.ctx.prev;
if (!pPost->isStarted) {
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
qDebug("%s dynQueryCtrl continue to retrieve block from post op", GET_TASKID(pOperator->pTaskInfo));
*ppRes = getNextBlockFromDownstream(pOperator, 1);
if (NULL == *ppRes) {
2024-07-22 07:07:01 +00:00
QRY_ERR_RET(handleSeqJoinCurrRetrieveEnd(pOperator, &pInfo->stbJoin));
2023-09-01 05:24:47 +00:00
pPrev->pListHead->readIdx++;
} else {
pInfo->stbJoin.execInfo.postBlkNum++;
pInfo->stbJoin.execInfo.postBlkRows += (*ppRes)->info.rows;
}
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
static FORCE_INLINE int32_t addToJoinVgroupHash(SSHashObj* pHash, void* pKey, int32_t keySize, void* pVal, int32_t valSize) {
SArray** ppArray = tSimpleHashGet(pHash, pKey, keySize);
if (NULL == ppArray) {
SArray* pArray = taosArrayInit(10, valSize);
if (NULL == pArray) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
if (NULL == taosArrayPush(pArray, pVal)) {
taosArrayDestroy(pArray);
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
if (tSimpleHashPut(pHash, pKey, keySize, &pArray, POINTER_BYTES)) {
taosArrayDestroy(pArray);
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
return TSDB_CODE_SUCCESS;
}
if (NULL == taosArrayPush(*ppArray, pVal)) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
return TSDB_CODE_SUCCESS;
}
static FORCE_INLINE int32_t addToJoinTableHash(SSHashObj* pHash, SSHashObj* pOnceHash, void* pKey, int32_t keySize) {
2024-07-22 07:07:01 +00:00
int32_t code = TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
uint32_t* pNum = tSimpleHashGet(pHash, pKey, keySize);
if (NULL == pNum) {
uint32_t n = 1;
2024-07-22 07:07:01 +00:00
code = tSimpleHashPut(pHash, pKey, keySize, &n, sizeof(n));
if (code) {
return code;
2023-09-01 05:24:47 +00:00
}
2024-07-22 07:07:01 +00:00
code = tSimpleHashPut(pOnceHash, pKey, keySize, NULL, 0);
if (code) {
return code;
2023-09-01 05:24:47 +00:00
}
return TSDB_CODE_SUCCESS;
}
switch (*pNum) {
case 0:
break;
case UINT32_MAX:
*pNum = 0;
break;
default:
if (1 == (*pNum)) {
2024-09-12 06:17:14 +00:00
code = tSimpleHashRemove(pOnceHash, pKey, keySize);
if (code) {
qError("tSimpleHashRemove failed in addToJoinTableHash, error:%s", tstrerror(code));
QRY_ERR_RET(code);
}
2023-09-01 05:24:47 +00:00
}
(*pNum)++;
break;
}
return TSDB_CODE_SUCCESS;
}
static void freeStbJoinTableList(SStbJoinTableList* pList) {
if (NULL == pList) {
return;
}
taosMemoryFree(pList->pLeftVg);
taosMemoryFree(pList->pLeftUid);
taosMemoryFree(pList->pRightVg);
taosMemoryFree(pList->pRightUid);
taosMemoryFree(pList);
}
static int32_t appendStbJoinTableList(SStbJoinPrevJoinCtx* pCtx, int64_t rows, int32_t* pLeftVg, int64_t* pLeftUid, int32_t* pRightVg, int64_t* pRightUid) {
2024-07-22 07:07:01 +00:00
int32_t code = TSDB_CODE_SUCCESS;
2024-09-18 10:36:21 +00:00
SStbJoinTableList* pNew = taosMemoryCalloc(1, sizeof(SStbJoinTableList));
2023-09-01 05:24:47 +00:00
if (NULL == pNew) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
pNew->pLeftVg = taosMemoryMalloc(rows * sizeof(*pLeftVg));
2024-07-22 07:07:01 +00:00
if (NULL == pNew->pLeftVg) {
code = terrno;
freeStbJoinTableList(pNew);
return code;
}
2023-09-01 05:24:47 +00:00
pNew->pLeftUid = taosMemoryMalloc(rows * sizeof(*pLeftUid));
2024-07-22 07:07:01 +00:00
if (NULL == pNew->pLeftUid) {
code = terrno;
freeStbJoinTableList(pNew);
return code;
}
2023-09-01 05:24:47 +00:00
pNew->pRightVg = taosMemoryMalloc(rows * sizeof(*pRightVg));
2024-07-22 07:07:01 +00:00
if (NULL == pNew->pRightVg) {
code = terrno;
freeStbJoinTableList(pNew);
return code;
}
2023-09-01 05:24:47 +00:00
pNew->pRightUid = taosMemoryMalloc(rows * sizeof(*pRightUid));
2024-07-22 07:07:01 +00:00
if (NULL == pNew->pRightUid) {
code = terrno;
2023-09-01 05:24:47 +00:00
freeStbJoinTableList(pNew);
2024-07-22 07:07:01 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
2024-07-22 07:07:01 +00:00
TAOS_MEMCPY(pNew->pLeftVg, pLeftVg, rows * sizeof(*pLeftVg));
TAOS_MEMCPY(pNew->pLeftUid, pLeftUid, rows * sizeof(*pLeftUid));
TAOS_MEMCPY(pNew->pRightVg, pRightVg, rows * sizeof(*pRightVg));
TAOS_MEMCPY(pNew->pRightUid, pRightUid, rows * sizeof(*pRightUid));
2023-09-01 05:24:47 +00:00
pNew->readIdx = 0;
pNew->uidNum = rows;
pNew->pNext = NULL;
if (pCtx->pListTail) {
pCtx->pListTail->pNext = pNew;
pCtx->pListTail = pNew;
} else {
pCtx->pListHead = pNew;
pCtx->pListTail= pNew;
}
return TSDB_CODE_SUCCESS;
}
static void doBuildStbJoinTableHash(SOperatorInfo* pOperator, SSDataBlock* pBlock) {
int32_t code = TSDB_CODE_SUCCESS;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
SColumnInfoData* pVg0 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.vgSlot[0]);
2024-07-22 07:07:01 +00:00
if (NULL == pVg0) {
QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
}
2023-09-01 05:24:47 +00:00
SColumnInfoData* pVg1 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.vgSlot[1]);
2024-07-22 07:07:01 +00:00
if (NULL == pVg1) {
QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
}
2023-09-01 05:24:47 +00:00
SColumnInfoData* pUid0 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.uidSlot[0]);
2024-07-22 07:07:01 +00:00
if (NULL == pUid0) {
QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
}
2023-09-01 05:24:47 +00:00
SColumnInfoData* pUid1 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.uidSlot[1]);
2024-07-22 07:07:01 +00:00
if (NULL == pUid1) {
QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
}
2023-09-01 05:24:47 +00:00
if (pStbJoin->basic.batchFetch) {
for (int32_t i = 0; i < pBlock->info.rows; ++i) {
int32_t* leftVg = (int32_t*)(pVg0->pData + pVg0->info.bytes * i);
int64_t* leftUid = (int64_t*)(pUid0->pData + pUid0->info.bytes * i);
int32_t* rightVg = (int32_t*)(pVg1->pData + pVg1->info.bytes * i);
int64_t* rightUid = (int64_t*)(pUid1->pData + pUid1->info.bytes * i);
code = addToJoinVgroupHash(pStbJoin->ctx.prev.leftHash, leftVg, sizeof(*leftVg), leftUid, sizeof(*leftUid));
if (TSDB_CODE_SUCCESS != code) {
break;
}
code = addToJoinVgroupHash(pStbJoin->ctx.prev.rightHash, rightVg, sizeof(*rightVg), rightUid, sizeof(*rightUid));
if (TSDB_CODE_SUCCESS != code) {
break;
}
}
} else {
for (int32_t i = 0; i < pBlock->info.rows; ++i) {
int64_t* leftUid = (int64_t*)(pUid0->pData + pUid0->info.bytes * i);
code = addToJoinTableHash(pStbJoin->ctx.prev.leftCache, pStbJoin->ctx.prev.onceTable, leftUid, sizeof(*leftUid));
if (TSDB_CODE_SUCCESS != code) {
break;
}
}
}
if (TSDB_CODE_SUCCESS == code) {
code = appendStbJoinTableList(&pStbJoin->ctx.prev, pBlock->info.rows, (int32_t*)pVg0->pData, (int64_t*)pUid0->pData, (int32_t*)pVg1->pData, (int64_t*)pUid1->pData);
if (TSDB_CODE_SUCCESS == code) {
pStbJoin->ctx.prev.tableNum += pBlock->info.rows;
}
}
2024-07-22 07:07:01 +00:00
_return:
2023-09-01 05:24:47 +00:00
if (TSDB_CODE_SUCCESS != code) {
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
}
}
static void postProcessStbJoinTableHash(SOperatorInfo* pOperator) {
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
if (pStbJoin->basic.batchFetch) {
return;
}
if (tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache) == tSimpleHashGetSize(pStbJoin->ctx.prev.onceTable)) {
tSimpleHashClear(pStbJoin->ctx.prev.leftCache);
return;
}
uint64_t* pUid = NULL;
int32_t iter = 0;
2024-09-12 06:17:14 +00:00
int32_t code = 0;
2023-09-01 05:24:47 +00:00
while (NULL != (pUid = tSimpleHashIterate(pStbJoin->ctx.prev.onceTable, pUid, &iter))) {
2024-09-12 06:17:14 +00:00
code = tSimpleHashRemove(pStbJoin->ctx.prev.leftCache, pUid, sizeof(*pUid));
if (code) {
qError("tSimpleHashRemove failed in postProcessStbJoinTableHash, error:%s", tstrerror(code));
}
2023-09-01 05:24:47 +00:00
}
pStbJoin->execInfo.leftCacheNum = tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache);
2024-03-22 00:47:07 +00:00
qDebug("more than 1 ref build table num %" PRId64, (int64_t)tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache));
2023-09-01 05:24:47 +00:00
2024-08-21 06:41:14 +00:00
/*
2023-09-01 05:24:47 +00:00
// debug only
iter = 0;
uint32_t* num = NULL;
while (NULL != (num = tSimpleHashIterate(pStbJoin->ctx.prev.leftCache, num, &iter))) {
2024-08-21 06:41:14 +00:00
A S S E R T(*num > 1);
2023-09-01 05:24:47 +00:00
}
2024-08-21 06:41:14 +00:00
*/
2023-09-01 05:24:47 +00:00
}
static void buildStbJoinTableList(SOperatorInfo* pOperator) {
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
while (true) {
SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
if (NULL == pBlock) {
break;
}
pStbJoin->execInfo.prevBlkNum++;
pStbJoin->execInfo.prevBlkRows += pBlock->info.rows;
doBuildStbJoinTableHash(pOperator, pBlock);
}
postProcessStbJoinTableHash(pOperator);
pStbJoin->ctx.prev.joinBuild = true;
}
2024-07-22 07:07:01 +00:00
static int32_t seqJoinLaunchNewRetrieve(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2023-09-01 05:24:47 +00:00
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
SStbJoinPrevJoinCtx* pPrev = &pStbJoin->ctx.prev;
SStbJoinTableList* pNode = pPrev->pListHead;
while (pNode) {
if (pNode->readIdx >= pNode->uidNum) {
pPrev->pListHead = pNode->pNext;
freeStbJoinTableList(pNode);
pNode = pPrev->pListHead;
continue;
}
seqJoinLaunchNewRetrieveImpl(pOperator, ppRes);
if (*ppRes) {
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
2024-07-22 07:07:01 +00:00
QRY_ERR_RET(handleSeqJoinCurrRetrieveEnd(pOperator, pStbJoin));
2023-09-01 05:24:47 +00:00
pPrev->pListHead->readIdx++;
}
*ppRes = NULL;
setOperatorCompleted(pOperator);
2024-07-22 07:07:01 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
2025-01-08 03:24:24 +00:00
static int32_t seqStableJoinComposeRes(SStbJoinDynCtrlInfo* pStbJoin, SSDataBlock* pBlock) {
2025-01-08 07:22:19 +00:00
if (pBlock) {
if (pStbJoin && pStbJoin->pOutputDataBlockDesc) {
pBlock->info.id.blockId = pStbJoin->pOutputDataBlockDesc->dataBlockId;
if (!pBlock->pDataBlock) return TSDB_CODE_SUCCESS;
for (int i = (int)pBlock->pDataBlock->size; i < pStbJoin->pOutputDataBlockDesc->pSlots->length; i++) {
2025-01-08 07:22:19 +00:00
SSlotDescNode* pSlot = (SSlotDescNode*)nodesListGetNode(pStbJoin->pOutputDataBlockDesc->pSlots, i);
if (pSlot == NULL) {
qError("seqStableJoinComposeRes: pSlot is NULL, i:%d", i);
return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
}
SColumnInfoData colInfo = createColumnInfoData(pSlot->dataType.type, pSlot->dataType.bytes, pSlot->slotId);
int32_t code = colInfoDataEnsureCapacity(&colInfo, pBlock->info.rows, true);
if (code != TSDB_CODE_SUCCESS) {
return code;
}
code = blockDataAppendColInfo(pBlock, &colInfo);
2025-01-08 07:22:19 +00:00
if (code != TSDB_CODE_SUCCESS) {
2025-01-08 07:24:04 +00:00
return code;
2025-01-08 07:22:19 +00:00
}
2025-01-08 03:24:24 +00:00
}
2025-01-08 07:22:19 +00:00
} else {
qError("seqStableJoinComposeRes: pBlock or pStbJoin is NULL");
return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
2025-01-08 03:24:24 +00:00
}
}
2025-01-08 03:24:24 +00:00
return TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
}
2024-08-27 09:04:44 +00:00
int32_t seqStableJoin(SOperatorInfo* pOperator, SSDataBlock** pRes) {
int32_t code = TSDB_CODE_SUCCESS;
2023-09-01 05:24:47 +00:00
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SStbJoinDynCtrlInfo* pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
QRY_PARAM_CHECK(pRes);
2023-09-01 05:24:47 +00:00
if (pOperator->status == OP_EXEC_DONE) {
2024-08-27 09:04:44 +00:00
return code;
2023-09-01 05:24:47 +00:00
}
int64_t st = 0;
if (pOperator->cost.openCost == 0) {
st = taosGetTimestampUs();
}
if (!pStbJoin->ctx.prev.joinBuild) {
buildStbJoinTableList(pOperator);
if (pStbJoin->execInfo.prevBlkRows <= 0) {
setOperatorCompleted(pOperator);
goto _return;
}
}
QRY_ERR_JRET(seqJoinContinueCurrRetrieve(pOperator, pRes));
if (*pRes) {
goto _return;
}
QRY_ERR_JRET(seqJoinLaunchNewRetrieve(pOperator, pRes));
_return:
if (pOperator->cost.openCost == 0) {
pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
}
if (code) {
qError("%s failed since %s", __func__, tstrerror(code));
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, code);
} else {
code = seqStableJoinComposeRes(pStbJoin, *pRes);
}
return code;
}
int32_t dynProcessUseDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
int32_t lino = 0;
SOperatorInfo* operator=(SOperatorInfo*) param;
SDynQueryCtrlOperatorInfo* pScanResInfo = (SDynQueryCtrlOperatorInfo*)operator->info;
if (TSDB_CODE_SUCCESS != code) {
operator->pTaskInfo->code = rpcCvtErrCode(code);
if (operator->pTaskInfo->code != code) {
qError("load systable rsp received, error:%s, cvted error:%s", tstrerror(code),
tstrerror(operator->pTaskInfo->code));
} else {
qError("load systable rsp received, error:%s", tstrerror(code));
}
goto _return;
}
pScanResInfo->vtbScan.pRsp = taosMemoryMalloc(sizeof(SUseDbRsp));
QUERY_CHECK_NULL(pScanResInfo->vtbScan.pRsp, code, lino, _return, terrno)
code = tDeserializeSUseDbRsp(pMsg->pData, (int32_t)pMsg->len, pScanResInfo->vtbScan.pRsp);
QUERY_CHECK_CODE(code, lino, _return);
taosMemoryFreeClear(pMsg->pData);
code = tsem_post(&pScanResInfo->vtbScan.ready);
QUERY_CHECK_CODE(code, lino, _return);
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
return code;
}
static int32_t buildDbVgInfoMap(SOperatorInfo* pOperator, SMsgCb* pMsgCb, SName* name, SExecTaskInfo* pTaskInfo, SUseDbOutput* output) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
char* buf1 = NULL;
SUseDbReq* pReq = NULL;
SDynQueryCtrlOperatorInfo* pScanResInfo = (SDynQueryCtrlOperatorInfo*)pOperator->info;
pReq = taosMemoryMalloc(sizeof(SUseDbReq));
QUERY_CHECK_NULL(pReq, code, lino, _return, terrno)
code = tNameGetFullDbName(name, pReq->db);
QUERY_CHECK_CODE(code, lino, _return);
int32_t contLen = tSerializeSUseDbReq(NULL, 0, pReq);
buf1 = taosMemoryCalloc(1, contLen);
QUERY_CHECK_NULL(buf1, code, lino, _return, terrno)
int32_t tempRes = tSerializeSUseDbReq(buf1, contLen, pReq);
if (tempRes < 0) {
QUERY_CHECK_CODE(terrno, lino, _return);
}
// send the fetch remote task result request
SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
QUERY_CHECK_NULL(pMsgSendInfo, code, lino, _return, terrno)
pMsgSendInfo->param = pOperator;
pMsgSendInfo->msgInfo.pData = buf1;
pMsgSendInfo->msgInfo.len = contLen;
pMsgSendInfo->msgType = TDMT_MND_GET_DB_INFO;
pMsgSendInfo->fp = dynProcessUseDbRsp;
pMsgSendInfo->requestId = pTaskInfo->id.queryId;
code = asyncSendMsgToServer(pMsgCb->clientRpc, &pScanResInfo->vtbScan.epSet, NULL, pMsgSendInfo);
QUERY_CHECK_CODE(code, lino, _return);
code = tsem_wait(&pScanResInfo->vtbScan.ready);
QUERY_CHECK_CODE(code, lino, _return);
code = queryBuildUseDbOutput(output, pScanResInfo->vtbScan.pRsp);
QUERY_CHECK_CODE(code, lino, _return);
_return:
if (code) {
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
taosMemoryFree(buf1);
}
taosMemoryFree(pReq);
tFreeSUsedbRsp(pScanResInfo->vtbScan.pRsp);
taosMemoryFreeClear(pScanResInfo->vtbScan.pRsp);
return code;
}
int dynVgInfoComp(const void* lp, const void* rp) {
SVgroupInfo* pLeft = (SVgroupInfo*)lp;
SVgroupInfo* pRight = (SVgroupInfo*)rp;
if (pLeft->hashBegin < pRight->hashBegin) {
return -1;
} else if (pLeft->hashBegin > pRight->hashBegin) {
return 1;
}
return 0;
}
int32_t dynMakeVgArraySortBy(SDBVgInfo* dbInfo, __compar_fn_t sort_func) {
if (NULL == dbInfo) {
return TSDB_CODE_SUCCESS;
}
if (dbInfo->vgHash && NULL == dbInfo->vgArray) {
int32_t vgSize = taosHashGetSize(dbInfo->vgHash);
dbInfo->vgArray = taosArrayInit(vgSize, sizeof(SVgroupInfo));
if (NULL == dbInfo->vgArray) {
return terrno;
}
void* pIter = taosHashIterate(dbInfo->vgHash, NULL);
while (pIter) {
if (NULL == taosArrayPush(dbInfo->vgArray, pIter)) {
taosHashCancelIterate(dbInfo->vgHash, pIter);
return terrno;
}
pIter = taosHashIterate(dbInfo->vgHash, pIter);
}
taosArraySort(dbInfo->vgArray, sort_func);
}
return TSDB_CODE_SUCCESS;
}
int32_t dynHashValueComp(void const* lp, void const* rp) {
uint32_t* key = (uint32_t*)lp;
SVgroupInfo* pVg = (SVgroupInfo*)rp;
if (*key < pVg->hashBegin) {
return -1;
} else if (*key > pVg->hashEnd) {
return 1;
}
return 0;
}
int32_t getVgId(SDBVgInfo* dbInfo, char* dbFName, int32_t* vgId, char *tbName) {
int32_t code = 0;
int32_t lino = 0;
code = dynMakeVgArraySortBy(dbInfo, dynVgInfoComp);
QUERY_CHECK_CODE(code, lino, _return);
int32_t vgNum = (int32_t)taosArrayGetSize(dbInfo->vgArray);
if (vgNum <= 0) {
qError("db vgroup cache invalid, db:%s, vgroup number:%d", dbFName, vgNum);
QUERY_CHECK_CODE(code = TSDB_CODE_TSC_DB_NOT_SELECTED, lino, _return);
}
SVgroupInfo* vgInfo = NULL;
char tbFullName[TSDB_TABLE_FNAME_LEN];
(void)snprintf(tbFullName, sizeof(tbFullName), "%s.", dbFName);
int32_t offset = (int32_t)strlen(tbFullName);
(void)snprintf(tbFullName + offset, sizeof(tbFullName) - offset, "%s", tbName);
uint32_t hashValue = taosGetTbHashVal(tbFullName, (int32_t)strlen(tbFullName), dbInfo->hashMethod,
dbInfo->hashPrefix, dbInfo->hashSuffix);
vgInfo = taosArraySearch(dbInfo->vgArray, &hashValue, dynHashValueComp, TD_EQ);
if (NULL == vgInfo) {
qError("no hash range found for hash value [%u], db:%s, numOfVgId:%d", hashValue, dbFName,
(int32_t)taosArrayGetSize(dbInfo->vgArray));
return TSDB_CODE_CTG_INTERNAL_ERROR;
}
*vgId = vgInfo->vgId;
_return:
return code;
}
bool colNeedScan(SOperatorInfo* pOperator, col_id_t colId) {
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SArray * pColList = pVtbScan->readColList;
if (pVtbScan->scanAllCols) {
return true;
}
for (int32_t i = 0; i < taosArrayGetSize(pColList); i++) {
if (colId == *(col_id_t*)taosArrayGet(pColList, i)) {
return true;
}
}
return false;
}
int32_t getDbVgInfo(SOperatorInfo* pOperator, SName *name, SDBVgInfo **dbVgInfo) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SMsgCb* pMsgCb = pVtbScan->pMsgCb;
SUseDbOutput* output = NULL;
SUseDbOutput** find = (SUseDbOutput**)taosHashGet(pInfo->vtbScan.dbVgInfoMap, name->dbname, strlen(name->dbname));
QRY_PARAM_CHECK(dbVgInfo);
if (find == NULL) {
output = taosMemoryMalloc(sizeof(SUseDbOutput));
code = buildDbVgInfoMap(pOperator, pMsgCb, name, pTaskInfo, output);
QUERY_CHECK_CODE(code, line, _return);
code = taosHashPut(pInfo->vtbScan.dbVgInfoMap, name->dbname, strlen(name->dbname), &output, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
} else {
output = *find;
}
*dbVgInfo = output->dbVgroup;
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
freeUseDbOutput(output);
return code;
}
int32_t extractColRefName(const char *colref, char **refDb, char** refTb, char** refCol) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
const char *first_dot = strchr(colref, '.');
QUERY_CHECK_NULL(first_dot, code, line, _return, terrno)
const char *second_dot = strchr(first_dot + 1, '.');
QUERY_CHECK_NULL(second_dot, code, line, _return, terrno)
size_t db_len = first_dot - colref;
size_t table_len = second_dot - first_dot - 1;
size_t col_len = strlen(second_dot + 1);
*refDb = taosMemoryMalloc(db_len + 1);
*refTb = taosMemoryMalloc(table_len + 1);
*refCol = taosMemoryMalloc(col_len + 1);
QUERY_CHECK_NULL(*refDb, code, line, _return, terrno)
QUERY_CHECK_NULL(*refTb, code, line, _return, terrno)
QUERY_CHECK_NULL(*refCol, code, line, _return, terrno)
tstrncpy(*refDb, colref, db_len + 1);
tstrncpy(*refTb, first_dot + 1, table_len + 1);
tstrncpy(*refCol, second_dot + 1, col_len + 1);
return TSDB_CODE_SUCCESS;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
if (*refDb) {
taosMemoryFree(*refDb);
*refDb = NULL;
}
if (*refTb) {
taosMemoryFree(*refTb);
*refTb = NULL;
}
if (*refCol) {
taosMemoryFree(*refCol);
*refCol = NULL;
}
return code;
}
bool tableInfoNeedCollect(char *dbName, char *tbName, char *expectDbName, char *expectTbName) {
if (strncmp(varDataVal(tbName), expectTbName, varDataLen(tbName)) == 0 &&
strlen(expectTbName) == varDataLen(tbName) &&
strncmp(varDataVal(dbName), expectDbName, varDataLen(dbName)) == 0 &&
strlen(expectDbName) == varDataLen(dbName)) {
return true;
}
return false;
}
int32_t getColRefInfo(SColRefInfo *pInfo, SArray* pDataBlock, int32_t index) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SColumnInfoData *pColNameCol = taosArrayGet(pDataBlock, 3);
SColumnInfoData *pUidCol = taosArrayGet(pDataBlock, 4);
SColumnInfoData *pColIdCol = taosArrayGet(pDataBlock, 5);
SColumnInfoData *pRefCol = taosArrayGet(pDataBlock, 6);
SColumnInfoData *pVgIdCol = taosArrayGet(pDataBlock, 7);
SColumnInfoData *pRefVerCol = taosArrayGet(pDataBlock, 8);
QUERY_CHECK_NULL(pColNameCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pColIdCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pRefCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pVgIdCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pRefVerCol, code, line, _return, terrno)
if (colDataIsNull_s(pRefCol, index)) {
pInfo->colrefName = NULL;
} else {
pInfo->colrefName = taosMemoryCalloc(varDataTLen(colDataGetData(pRefCol, index)), 1);
QUERY_CHECK_NULL(pInfo->colrefName, code, line, _return, terrno)
memcpy(pInfo->colrefName, varDataVal(colDataGetData(pRefCol, index)), varDataLen(colDataGetData(pRefCol, index)));
pInfo->colrefName[varDataLen(colDataGetData(pRefCol, index))] = 0;
}
pInfo->colName = taosMemoryCalloc(varDataTLen(colDataGetData(pColNameCol, index)), 1);
QUERY_CHECK_NULL(pInfo->colName, code, line, _return, terrno)
memcpy(pInfo->colName, varDataVal(colDataGetData(pColNameCol, index)), varDataLen(colDataGetData(pColNameCol, index)));
pInfo->colName[varDataLen(colDataGetData(pColNameCol, index))] = 0;
if (!colDataIsNull_s(pUidCol, index)) {
GET_TYPED_DATA(pInfo->uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, index), 0);
}
if (!colDataIsNull_s(pColIdCol, index)) {
GET_TYPED_DATA(pInfo->colId, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pColIdCol, index), 0);
}
if (!colDataIsNull_s(pVgIdCol, index)) {
GET_TYPED_DATA(pInfo->vgId, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pVgIdCol, index), 0);
}
_return:
return code;
}
int32_t processOrgTbVg(SVtbScanDynCtrlInfo* pVtbScan, SExecTaskInfo* pTaskInfo, int32_t rversion) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
if (pTaskInfo->pStreamRuntimeInfo == NULL) {
return code;
}
if (pVtbScan->existOrgTbVg == NULL) {
pVtbScan->existOrgTbVg = pVtbScan->curOrgTbVg;
pVtbScan->curOrgTbVg = NULL;
}
if (pVtbScan->curOrgTbVg != NULL) {
// which means rversion has changed
void* pCurIter = NULL;
SArray* tmpArray = NULL;
while ((pCurIter = taosHashIterate(pVtbScan->curOrgTbVg, pCurIter))) {
int32_t* vgId = (int32_t*)taosHashGetKey(pCurIter, NULL);
if (taosHashGet(pVtbScan->existOrgTbVg, vgId, sizeof(int32_t)) == NULL) {
if (tmpArray == NULL) {
tmpArray = taosArrayInit(1, sizeof(int32_t));
QUERY_CHECK_NULL(tmpArray, code, line, _return, terrno)
}
QUERY_CHECK_NULL(taosArrayPush(tmpArray, vgId), code, line, _return, terrno)
}
}
if (tmpArray == NULL) {
return TSDB_CODE_SUCCESS;
}
if (tmpArray != NULL && pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addVgIds == NULL) {
SArray* expiredInfo = atomic_load_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo);
if (expiredInfo && expiredInfo == atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo, expiredInfo, NULL)) {
for (int32_t i = 0; i < taosArrayGetSize(expiredInfo); i++) {
SStreamTaskAddr* vgInfo = (SStreamTaskAddr*)taosArrayGet(expiredInfo, i);
QUERY_CHECK_NULL(taosArrayPush(tmpArray, &vgInfo->nodeId), code, line, _return, terrno)
}
taosArrayDestroy(expiredInfo);
}
if (atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addVgIds, NULL, tmpArray)) {
taosArrayDestroy(tmpArray);
}
}
atomic_store_64(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.uid, (int64_t)(pVtbScan->isSuperTable ? pVtbScan->suid : pVtbScan->uid));
(void)atomic_val_compare_exchange_8(pTaskInfo->pStreamRuntimeInfo->vtableDeployGot, 0, 1);
taosHashClear(pVtbScan->curOrgTbVg);
pVtbScan->needRedeploy = true;
pVtbScan->rversion = rversion;
return TSDB_CODE_STREAM_VTABLE_NEED_REDEPLOY;
}
return code;
_return:
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
return code;
}
int32_t getVgIdFromColref(SOperatorInfo* pOperator, const char* colRef, int32_t* vgId) {
int32_t code =TSDB_CODE_SUCCESS;
int32_t line = 0;
char* refDbName = NULL;
char* refTbName = NULL;
char* refColName = NULL;
SDBVgInfo* dbVgInfo = NULL;
SName name = {0};
char dbFname[TSDB_DB_FNAME_LEN] = {0};
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
code = extractColRefName(colRef, &refDbName, &refTbName, &refColName);
QUERY_CHECK_CODE(code, line, _return);
toName(pInfo->vtbScan.acctId, refDbName, refTbName, &name);
code = getDbVgInfo(pOperator, &name, &dbVgInfo);
QUERY_CHECK_CODE(code, line, _return);
code = tNameGetFullDbName(&name, dbFname);
QUERY_CHECK_CODE(code, line, _return);
code = getVgId(dbVgInfo, dbFname, vgId, name.tname);
QUERY_CHECK_CODE(code, line, _return);
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
taosMemoryFree(refDbName);
taosMemoryFree(refTbName);
taosMemoryFree(refColName);
return code;
}
static int32_t generateTagArrayByTagBlockAndSave(SHashObj* vtbUidTagListMap, tb_uid_t uid, SSDataBlock *pTagVal, int32_t rowIdx) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
STagVal tagVal = {0};
// last col is uid
SArray* pTagList = taosArrayInit(1, sizeof(STagVal));
QUERY_CHECK_NULL(pTagList, code, line, _return, terrno)
for (int32_t k = 0; k < taosArrayGetSize(pTagVal->pDataBlock) - 1; k++) {
SColumnInfoData *pTagCol = taosArrayGet(pTagVal->pDataBlock, k);
QUERY_CHECK_NULL(pTagCol, code, line, _return, terrno)
tagVal.type = pTagCol->info.type;
tagVal.cid = pTagCol->info.colId;
if (!colDataIsNull_s(pTagCol, rowIdx)) {
char* pData = colDataGetData(pTagCol, rowIdx);
if (IS_VAR_DATA_TYPE(pTagCol->info.type)) {
tagVal.nData = varDataLen(pData);
tagVal.pData = taosMemoryMalloc(tagVal.nData);
QUERY_CHECK_NULL(tagVal.pData, code, line, _return, terrno)
memcpy(tagVal.pData, varDataVal(pData), varDataLen(pData));
QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
} else {
memcpy(&tagVal.i64, pData, tDataTypes[pTagCol->info.type].bytes);
QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
}
} else {
tagVal.pData = NULL;
tagVal.nData = 0;
QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
}
tagVal = (STagVal){0};
}
code = taosHashPut(vtbUidTagListMap, &uid, sizeof(uid), &pTagList, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
return code;
_return:
if (tagVal.pData) {
taosMemoryFreeClear(tagVal.pData);
}
if (pTagList) {
taosArrayDestroyEx(pTagList, destroyTagVal);
}
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
return code;
}
int32_t virtualTableScanProcessColRefInfo(SOperatorInfo* pOperator, SArray* pColRefInfo, tb_uid_t* uid, int32_t* vgId) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SDBVgInfo* dbVgInfo = NULL;
for (int32_t j = 0; j < taosArrayGetSize(pColRefInfo); j++) {
SColRefInfo *pKV = (SColRefInfo*)taosArrayGet(pColRefInfo, j);
*uid = pKV->uid;
*vgId = pKV->vgId;
if (pKV->colrefName != NULL && colNeedScan(pOperator, pKV->colId)) {
char* refDbName = NULL;
char* refTbName = NULL;
char* refColName = NULL;
SName name = {0};
char dbFname[TSDB_DB_FNAME_LEN] = {0};
char orgTbFName[TSDB_TABLE_FNAME_LEN] = {0};
code = extractColRefName(pKV->colrefName, &refDbName, &refTbName, &refColName);
QUERY_CHECK_CODE(code, line, _return);
toName(pInfo->vtbScan.acctId, refDbName, refTbName, &name);
code = getDbVgInfo(pOperator, &name, &dbVgInfo);
QUERY_CHECK_CODE(code, line, _return);
code = tNameGetFullDbName(&name, dbFname);
QUERY_CHECK_CODE(code, line, _return);
code = tNameGetFullTableName(&name, orgTbFName);
QUERY_CHECK_CODE(code, line, _return);
void *pVal = taosHashGet(pVtbScan->otbNameToOtbInfoMap, orgTbFName, sizeof(orgTbFName));
if (!pVal) {
SOrgTbInfo orgTbInfo = {0};
code = getVgId(dbVgInfo, dbFname, &orgTbInfo.vgId, name.tname);
QUERY_CHECK_CODE(code, line, _return);
tstrncpy(orgTbInfo.tbName, orgTbFName, sizeof(orgTbInfo.tbName));
orgTbInfo.colMap = taosArrayInit(10, sizeof(SColIdNameKV));
QUERY_CHECK_NULL(orgTbInfo.colMap, code, line, _return, terrno)
SColIdNameKV colIdNameKV = {0};
colIdNameKV.colId = pKV->colId;
tstrncpy(colIdNameKV.colName, refColName, sizeof(colIdNameKV.colName));
QUERY_CHECK_NULL(taosArrayPush(orgTbInfo.colMap, &colIdNameKV), code, line, _return, terrno)
code = taosHashPut(pVtbScan->otbNameToOtbInfoMap, orgTbFName, sizeof(orgTbFName), &orgTbInfo, sizeof(orgTbInfo));
QUERY_CHECK_CODE(code, line, _return);
} else {
SOrgTbInfo *tbInfo = (SOrgTbInfo *)pVal;
SColIdNameKV colIdNameKV = {0};
colIdNameKV.colId = pKV->colId;
tstrncpy(colIdNameKV.colName, refColName, sizeof(colIdNameKV.colName));
QUERY_CHECK_NULL(taosArrayPush(tbInfo->colMap, &colIdNameKV), code, line, _return, terrno)
}
taosMemoryFree(refDbName);
taosMemoryFree(refTbName);
taosMemoryFree(refColName);
}
}
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
static int32_t getTagBlockAndProcess(SOperatorInfo* pOperator, bool hasPartition) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray* pColRefArray = NULL;
SOperatorInfo* pSystableScanOp = pOperator->pDownstream[0];
SOperatorInfo* pTagScanOp = pOperator->pDownstream[1];
pVtbScan->vtbUidTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->vtbUidTagListMap, code, line, _return, terrno)
taosHashSetFreeFp(pVtbScan->vtbUidTagListMap, destroyTagList);
if (hasPartition) {
pVtbScan->vtbUidToGroupIdMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
pVtbScan->vtbGroupIdTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->vtbUidToGroupIdMap, code, line, _return, terrno)
QUERY_CHECK_NULL(pVtbScan->vtbGroupIdTagListMap, code, line, _return, terrno)
taosHashSetFreeFp(pVtbScan->vtbGroupIdTagListMap, destroyVtbUidTagListMap);
}
while (true) {
SSDataBlock *pTagVal = NULL;
code = pTagScanOp->fpSet.getNextFn(pTagScanOp, &pTagVal);
QUERY_CHECK_CODE(code, line, _return);
if (pTagVal == NULL) {
break;
}
SHashObj *vtbUidTagListMap = NULL;
if (hasPartition) {
void* pIter = taosHashGet(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
if (pIter) {
vtbUidTagListMap = *(SHashObj**)pIter;
} else {
vtbUidTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(vtbUidTagListMap, code, line, _return, terrno)
taosHashSetFreeFp(vtbUidTagListMap, destroyTagList);
code = taosHashPut(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId), &vtbUidTagListMap, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
}
} else {
vtbUidTagListMap = pVtbScan->vtbUidTagListMap;
}
SColumnInfoData *pUidCol = taosArrayGetLast(pTagVal->pDataBlock);
QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
for (int32_t i = 0; i < pTagVal->info.rows; i++) {
tb_uid_t uid = 0;
if (!colDataIsNull_s(pUidCol, i)) {
GET_TYPED_DATA(uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, i), 0);
QUERY_CHECK_CODE(code, line, _return);
}
code = generateTagArrayByTagBlockAndSave(vtbUidTagListMap, uid, pTagVal, i);
QUERY_CHECK_CODE(code, line, _return);
if (hasPartition) {
code = taosHashPut(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid), &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
QUERY_CHECK_CODE(code, line, _return);
}
}
}
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
static int32_t processChildTableListAndGenerateOrgTbInfoMap(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray* pColRefArray = NULL;
SOperatorInfo* pSystableScanOp = pOperator->pDownstream[0];
SOperatorInfo* pTagScanOp = pOperator->pDownstream[1];
pVtbScan->vtbUidToVgIdMapMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->vtbUidToVgIdMapMap, code, line, _return, terrno)
for (int32_t i = 0; i < taosArrayGetSize(pVtbScan->childTableList); i++) {
SHashObj* otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
SArray* pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, i);
QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
tb_uid_t uid = 0;
int32_t vgId = 0;
code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
QUERY_CHECK_CODE(code, line, _return);
size_t len = 0;
void* pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
while (pOrgTbInfo != NULL) {
char* key = taosHashGetKey(pOrgTbInfo, &len);
SOrgTbInfo* orgTbInfo = (SOrgTbInfo*)pOrgTbInfo;
void* pIter = taosHashGet(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId));
if (!pIter) {
SArray* pOrgTbInfoArray = taosArrayInit(1, sizeof(SOrgTbInfo));
QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
code = taosHashPut(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId), &pOrgTbInfoArray, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
} else {
SArray* pOrgTbInfoArray = *(SArray**)pIter;
QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
}
pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pOrgTbInfo);
code = taosHashRemove(pVtbScan->otbNameToOtbInfoMap, key, len);
QUERY_CHECK_CODE(code, line, _return);
}
code = taosHashPut(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid), &otbVgIdToOtbInfoArrayMap, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
}
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
static int32_t buildOrgTbInfoSingle(SOperatorInfo* pOperator, bool hasPartition) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
code = processChildTableListAndGenerateOrgTbInfoMap(pOperator);
QUERY_CHECK_CODE(code, line, _return);
// process tag
code = getTagBlockAndProcess(pOperator, hasPartition);
QUERY_CHECK_CODE(code, line, _return);
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
static int32_t buildOrgTbInfoBatch(SOperatorInfo* pOperator, bool hasPartition) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray* pColRefArray = NULL;
SOperatorInfo* pSystableScanOp = pOperator->pDownstream[0];
SOperatorInfo* pTagScanOp = pOperator->pDownstream[1];
if (hasPartition) {
pVtbScan->vtbUidToGroupIdMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
pVtbScan->vtbGroupIdTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
pVtbScan->vtbGroupIdToVgIdMapMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->vtbGroupIdToVgIdMapMap, code, line, _return, terrno)
QUERY_CHECK_NULL(pVtbScan->vtbUidToGroupIdMap, code, line, _return, terrno)
QUERY_CHECK_NULL(pVtbScan->vtbGroupIdTagListMap, code, line, _return, terrno)
taosHashSetFreeFp(pVtbScan->vtbGroupIdToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
} else {
pVtbScan->otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
}
while (true && hasPartition) {
SSDataBlock* pTagVal = NULL;
code = pTagScanOp->fpSet.getNextFn(pTagScanOp, &pTagVal);
QUERY_CHECK_CODE(code, line, _return);
if (pTagVal == NULL) {
break;
}
SColumnInfoData *pUidCol = taosArrayGetLast(pTagVal->pDataBlock);
QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
for (int32_t i = 0; i < pTagVal->info.rows; i++) {
tb_uid_t uid = 0;
if (!colDataIsNull_s(pUidCol, i)) {
GET_TYPED_DATA(uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, i), 0);
QUERY_CHECK_CODE(code, line, _return);
}
code = taosHashPut(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid), &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
QUERY_CHECK_CODE(code, line, _return);
}
code = taosHashPut(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId), NULL, 0);
QUERY_CHECK_CODE(code, line, _return);
}
for (int32_t i = 0; i < taosArrayGetSize(pVtbScan->childTableList); i++) {
SArray* pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, i);
QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
tb_uid_t uid = 0;
int32_t vgId = 0;
code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
QUERY_CHECK_CODE(code, line, _return);
SHashObj* otbVgIdToOtbInfoArrayMap = NULL;
if (hasPartition) {
uint64_t* groupId = (uint64_t *)taosHashGet(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid));
QUERY_CHECK_NULL(groupId, code, line, _return, terrno)
void* pHashIter = taosHashGet(pVtbScan->vtbGroupIdToVgIdMapMap, groupId, sizeof(*groupId));
if (pHashIter) {
otbVgIdToOtbInfoArrayMap = *(SHashObj**)pHashIter;
} else {
otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
code = taosHashPut(pVtbScan->vtbGroupIdToVgIdMapMap, groupId, sizeof(*groupId), &otbVgIdToOtbInfoArrayMap, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
}
} else {
otbVgIdToOtbInfoArrayMap = pVtbScan->otbVgIdToOtbInfoArrayMap;
}
size_t len = 0;
void* pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
while (pOrgTbInfo != NULL) {
char* key = taosHashGetKey(pOrgTbInfo, &len);
SOrgTbInfo* orgTbInfo = (SOrgTbInfo*)pOrgTbInfo;
void* pIter = taosHashGet(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId));
if (!pIter) {
SArray* pOrgTbInfoArray = taosArrayInit(1, sizeof(SOrgTbInfo));
QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
code = taosHashPut(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId), &pOrgTbInfoArray, POINTER_BYTES);
QUERY_CHECK_CODE(code, line, _return);
} else {
SArray* pOrgTbInfoArray = *(SArray**)pIter;
QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
}
pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pOrgTbInfo);
code = taosHashRemove(pVtbScan->otbNameToOtbInfoMap, key, len);
QUERY_CHECK_CODE(code, line, _return);
}
}
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
int32_t buildVirtualSuperTableScanChildTableMap(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray* pColRefArray = NULL;
SOperatorInfo* pSystableScanOp = NULL;
pVtbScan->childTableMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->childTableMap, code, line, _return, terrno)
if (pInfo->qType == DYN_QTYPE_VTB_AGG) {
pVtbScan->otbNameToOtbInfoMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->otbNameToOtbInfoMap, code, line, _return, terrno)
pSystableScanOp = pOperator->pDownstream[0];
} else {
pSystableScanOp = pOperator->pDownstream[1];
}
while (true) {
SSDataBlock *pChildInfo = NULL;
code = pSystableScanOp->fpSet.getNextFn(pSystableScanOp, &pChildInfo);
QUERY_CHECK_CODE(code, line, _return);
if (pChildInfo == NULL) {
break;
}
SColumnInfoData *pTableNameCol = taosArrayGet(pChildInfo->pDataBlock, 0);
SColumnInfoData *pStbNameCol = taosArrayGet(pChildInfo->pDataBlock, 1);
SColumnInfoData *pDbNameCol = taosArrayGet(pChildInfo->pDataBlock, 2);
QUERY_CHECK_NULL(pTableNameCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pStbNameCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pDbNameCol, code, line, _return, terrno)
for (int32_t i = 0; i < pChildInfo->info.rows; i++) {
if (!colDataIsNull_s(pStbNameCol, i)) {
char* stbrawname = colDataGetData(pStbNameCol, i);
char* dbrawname = colDataGetData(pDbNameCol, i);
char *ctbName = colDataGetData(pTableNameCol, i);
if (tableInfoNeedCollect(dbrawname, stbrawname, pInfo->vtbScan.dbName, pInfo->vtbScan.tbName)) {
SColRefInfo info = {0};
code = getColRefInfo(&info, pChildInfo->pDataBlock, i);
QUERY_CHECK_CODE(code, line, _return);
if (pInfo->qType == DYN_QTYPE_VTB_SCAN) {
if (pInfo->vtbScan.dynTbUid != 0 && info.uid != pInfo->vtbScan.dynTbUid) {
qTrace("dynQueryCtrl tb uid filter, info uid:%" PRIu64 ", dyn tb uid:%" PRIu64, info.uid,
pInfo->vtbScan.dynTbUid);
destroyColRefInfo(&info);
continue;
}
if (pTaskInfo->pStreamRuntimeInfo) {
if (pVtbScan->curOrgTbVg == NULL) {
pVtbScan->curOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->curOrgTbVg, code, line, _return, terrno)
}
if (info.colrefName) {
int32_t vgId;
code = getVgIdFromColref(pOperator, info.colrefName, &vgId);
QUERY_CHECK_CODE(code, line, _return);
code = taosHashPut(pVtbScan->curOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
QUERY_CHECK_CODE(code, line, _return);
}
}
}
if (taosHashGet(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName)) == NULL) {
pColRefArray = taosArrayInit(1, sizeof(SColRefInfo));
QUERY_CHECK_NULL(pColRefArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pColRefArray, &info), code, line, _return, terrno)
int32_t tableIdx = (int32_t)taosArrayGetSize(pVtbScan->childTableList);
QUERY_CHECK_NULL(taosArrayPush(pVtbScan->childTableList, &pColRefArray), code, line, _return, terrno)
code = taosHashPut(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName), &tableIdx, sizeof(tableIdx));
QUERY_CHECK_CODE(code, line, _return);
} else {
int32_t *tableIdx = (int32_t*)taosHashGet(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName));
QUERY_CHECK_NULL(tableIdx, code, line, _return, terrno)
pColRefArray = (SArray *)taosArrayGetP(pVtbScan->childTableList, *tableIdx);
QUERY_CHECK_NULL(pColRefArray, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pColRefArray, &info), code, line, _return, terrno)
}
}
}
}
}
switch (pInfo->qType) {
case DYN_QTYPE_VTB_AGG: {
if (pVtbScan->batchProcessChild) {
code = buildOrgTbInfoBatch(pOperator, pVtbScan->hasPartition);
} else {
code = buildOrgTbInfoSingle(pOperator, pVtbScan->hasPartition);
}
break;
}
case DYN_QTYPE_VTB_SCAN: {
code = processOrgTbVg(pVtbScan, pTaskInfo, 1);
break;
}
default: {
code = TSDB_CODE_PLAN_INVALID_DYN_CTRL_TYPE;
break;
}
}
QUERY_CHECK_CODE(code, line, _return);
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t buildVirtualNormalChildTableScanChildTableMap(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray* pColRefInfo = pInfo->vtbScan.colRefInfo;
SOperatorInfo* pSystableScanOp = pOperator->pDownstream[1];
int32_t rversion = 0;
pInfo->vtbScan.colRefInfo = taosArrayInit(1, sizeof(SColRefInfo));
QUERY_CHECK_NULL(pInfo->vtbScan.colRefInfo, code, line, _return, terrno)
while (true) {
SSDataBlock *pTableInfo = NULL;
code = pSystableScanOp->fpSet.getNextFn(pSystableScanOp, &pTableInfo);
if (pTableInfo == NULL) {
break;
}
SColumnInfoData *pTableNameCol = taosArrayGet(pTableInfo->pDataBlock, 0);
SColumnInfoData *pDbNameCol = taosArrayGet(pTableInfo->pDataBlock, 2);
SColumnInfoData *pRefVerCol = taosArrayGet(pTableInfo->pDataBlock, 8);
QUERY_CHECK_NULL(pTableNameCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pDbNameCol, code, line, _return, terrno)
QUERY_CHECK_NULL(pRefVerCol, code, line, _return, terrno)
for (int32_t i = 0; i < pTableInfo->info.rows; i++) {
if (!colDataIsNull_s(pRefVerCol, i)) {
GET_TYPED_DATA(rversion, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pRefVerCol, i), 0);
}
if (!colDataIsNull_s(pTableNameCol, i)) {
char* tbrawname = colDataGetData(pTableNameCol, i);
char* dbrawname = colDataGetData(pDbNameCol, i);
QUERY_CHECK_NULL(tbrawname, code, line, _return, terrno)
QUERY_CHECK_NULL(dbrawname, code, line, _return, terrno)
if (tableInfoNeedCollect(dbrawname, tbrawname, pInfo->vtbScan.dbName, pInfo->vtbScan.tbName)) {
SColRefInfo info = {0};
code = getColRefInfo(&info, pTableInfo->pDataBlock, i);
QUERY_CHECK_CODE(code, line, _return);
if ((rversion != pVtbScan->rversion || pVtbScan->existOrgTbVg == NULL) && info.colrefName) {
if (pVtbScan->curOrgTbVg == NULL) {
pVtbScan->curOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->curOrgTbVg, code, line, _return, terrno)
}
int32_t vgId;
code = getVgIdFromColref(pOperator, info.colrefName, &vgId);
QUERY_CHECK_CODE(code, line, _return);
code = taosHashPut(pVtbScan->curOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
QUERY_CHECK_CODE(code, line, _return);
}
QUERY_CHECK_NULL(taosArrayPush(pInfo->vtbScan.colRefInfo, &info), code, line, _return, terrno)
}
}
}
}
code = processOrgTbVg(pVtbScan, pTaskInfo, rversion);
QUERY_CHECK_CODE(code, line, _return);
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t virtualTableScanCheckNeedRedeploy(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SArray *tmpArray = NULL;
tmpArray = atomic_load_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo);
if (tmpArray && tmpArray == atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo, tmpArray, NULL)) {
for (int32_t i = 0; i < taosArrayGetSize(tmpArray); i++) {
SStreamTaskAddr* pTaskAddr = (SStreamTaskAddr*)taosArrayGet(tmpArray, i);
code = taosHashPut(pVtbScan->existOrgTbVg, &pTaskAddr->nodeId, sizeof(pTaskAddr->nodeId), NULL, 0);
QUERY_CHECK_CODE(code, line, _return);
if (pVtbScan->newAddedVgInfo == NULL) {
pVtbScan->newAddedVgInfo = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->newAddedVgInfo, code, line, _return, terrno)
}
code = taosHashPut(pVtbScan->newAddedVgInfo, &pTaskAddr->nodeId, sizeof(pTaskAddr->nodeId), pTaskAddr, sizeof(SStreamTaskAddr));
QUERY_CHECK_CODE(code, line, _return);
}
pVtbScan->needRedeploy = false;
} else {
code = TSDB_CODE_STREAM_VTABLE_NEED_REDEPLOY;
QUERY_CHECK_CODE(code, line, _return);
}
_return:
taosArrayClear(tmpArray);
taosArrayDestroy(tmpArray);
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t virtualTableScanBuildDownStreamOpParam(SOperatorInfo* pOperator, tb_uid_t uid, int32_t vgId) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
pVtbScan->vtbScanParam = NULL;
code = buildVtbScanOperatorParam(pInfo, &pVtbScan->vtbScanParam, uid);
QUERY_CHECK_CODE(code, line, _return);
void* pIter = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
while (pIter != NULL) {
SOrgTbInfo* pMap = (SOrgTbInfo*)pIter;
SOperatorParam* pExchangeParam = NULL;
SStreamTaskAddr* addr = taosHashGet(pVtbScan->newAddedVgInfo, &pMap->vgId, sizeof(pMap->vgId));
if (addr != NULL) {
SDownstreamSourceNode newSource = {0};
newSource.type = QUERY_NODE_DOWNSTREAM_SOURCE;
newSource.clientId = pTaskInfo->id.taskId;// current task's taskid
newSource.taskId = addr->taskId;
newSource.fetchMsgType = TDMT_STREAM_FETCH;
newSource.localExec = false;
newSource.addr.nodeId = addr->nodeId;
memcpy(&newSource.addr.epSet, &addr->epset, sizeof(SEpSet));
code = buildExchangeOperatorParamForVScan(&pExchangeParam, 0, pMap, &newSource);
QUERY_CHECK_CODE(code, line, _return);
code = taosHashRemove(pVtbScan->newAddedVgInfo, &pMap->vgId, sizeof(pMap->vgId));
QUERY_CHECK_CODE(code, line, _return);
} else {
code = buildExchangeOperatorParamForVScan(&pExchangeParam, 0, pMap, NULL);
QUERY_CHECK_CODE(code, line, _return);
}
QUERY_CHECK_NULL(taosArrayPush(((SVTableScanOperatorParam*)pVtbScan->vtbScanParam->value)->pOpParamArray, &pExchangeParam), code, line, _return, terrno)
pIter = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pIter);
}
SOperatorParam* pExchangeParam = NULL;
code = buildExchangeOperatorParamForVTagScan(&pExchangeParam, 0, vgId, uid);
QUERY_CHECK_CODE(code, line, _return);
((SVTableScanOperatorParam*)pVtbScan->vtbScanParam->value)->pTagScanOp = pExchangeParam;
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t virtualTableScanGetNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SOperatorInfo* pVtbScanOp = pOperator->pDownstream[0];
pVtbScan->otbNameToOtbInfoMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pVtbScan->otbNameToOtbInfoMap, code, line, _return, terrno)
taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
while (true) {
if (pVtbScan->curTableIdx == pVtbScan->lastTableIdx) {
code = pVtbScanOp->fpSet.getNextFn(pVtbScanOp, pRes);
QUERY_CHECK_CODE(code, line, _return);
} else {
taosHashClear(pVtbScan->otbNameToOtbInfoMap);
SArray* pColRefInfo = NULL;
if (pVtbScan->isSuperTable) {
pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, pVtbScan->curTableIdx);
} else {
pColRefInfo = pInfo->vtbScan.colRefInfo;
}
QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
tb_uid_t uid = 0;
int32_t vgId = 0;
code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
QUERY_CHECK_CODE(code, line, _return);
code = virtualTableScanBuildDownStreamOpParam(pOperator, uid, vgId);
QUERY_CHECK_CODE(code, line, _return);
// reset downstream operator's status
pVtbScanOp->status = OP_NOT_OPENED;
code = pVtbScanOp->fpSet.getNextExtFn(pVtbScanOp, pVtbScan->vtbScanParam, pRes);
QUERY_CHECK_CODE(code, line, _return);
}
if (*pRes) {
// has result, still read data from this table.
pVtbScan->lastTableIdx = pVtbScan->curTableIdx;
break;
} else {
// no result, read next table.
pVtbScan->curTableIdx++;
if (pVtbScan->isSuperTable) {
if (pVtbScan->curTableIdx >= taosArrayGetSize(pVtbScan->childTableList)) {
setOperatorCompleted(pOperator);
break;
}
} else {
setOperatorCompleted(pOperator);
break;
}
}
}
_return:
taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
pVtbScan->otbNameToOtbInfoMap = NULL;
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t vtbScanOpen(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
int64_t st = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
if (OPTR_IS_OPENED(pOperator)) {
return code;
}
if (pOperator->cost.openCost == 0) {
st = taosGetTimestampUs();
}
if (pVtbScan->isSuperTable) {
code = buildVirtualSuperTableScanChildTableMap(pOperator);
QUERY_CHECK_CODE(code, line, _return);
} else {
code = buildVirtualNormalChildTableScanChildTableMap(pOperator);
QUERY_CHECK_CODE(code, line, _return);
}
OPTR_SET_OPENED(pOperator);
_return:
if (pOperator->cost.openCost == 0) {
pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
}
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, code);
}
return code;
}
int32_t vtbScanNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
QRY_PARAM_CHECK(pRes);
if (pOperator->status == OP_EXEC_DONE && !pOperator->pOperatorGetParam) {
return code;
}
if (pOperator->pOperatorGetParam) {
if (pOperator->status == OP_EXEC_DONE) {
pOperator->status = OP_OPENED;
}
pVtbScan->curTableIdx = 0;
pVtbScan->lastTableIdx = -1;
pVtbScan->window = ((SDynQueryCtrlOperatorParam *)(pOperator->pOperatorGetParam)->value)->window;
pOperator->pOperatorGetParam = NULL;
} else {
pVtbScan->window.skey = INT64_MAX;
pVtbScan->window.ekey = INT64_MIN;
}
if (pVtbScan->needRedeploy) {
code = virtualTableScanCheckNeedRedeploy(pOperator);
QUERY_CHECK_CODE(code, line, _return);
}
code = pOperator->fpSet._openFn(pOperator);
QUERY_CHECK_CODE(code, line, _return);
if (pVtbScan->isSuperTable && taosArrayGetSize(pVtbScan->childTableList) == 0) {
setOperatorCompleted(pOperator);
return code;
}
code = virtualTableScanGetNext(pOperator, pRes);
QUERY_CHECK_CODE(code, line, _return);
return code;
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, code);
}
return code;
}
2023-09-01 05:24:47 +00:00
int32_t initSeqStbJoinTableHash(SStbJoinPrevJoinCtx* pPrev, bool batchFetch) {
if (batchFetch) {
pPrev->leftHash = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
if (NULL == pPrev->leftHash) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
pPrev->rightHash = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
if (NULL == pPrev->rightHash) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
} else {
pPrev->leftCache = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
if (NULL == pPrev->leftCache) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
pPrev->rightCache = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
if (NULL == pPrev->rightCache) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
pPrev->onceTable = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
if (NULL == pPrev->onceTable) {
2024-07-22 07:07:01 +00:00
return terrno;
2023-09-01 05:24:47 +00:00
}
}
return TSDB_CODE_SUCCESS;
}
static void updateDynTbUidIfNeeded(SVtbScanDynCtrlInfo* pVtbScan, SStreamRuntimeInfo* pStreamRuntimeInfo) {
if (pStreamRuntimeInfo == NULL) {
return;
}
SArray* vals = pStreamRuntimeInfo->funcInfo.pStreamPartColVals;
for (int32_t i = 0; i < taosArrayGetSize(vals); ++i) {
SStreamGroupValue* pValue = taosArrayGet(vals, i);
if (pValue != NULL && pValue->isTbname && pValue->uid != pVtbScan->dynTbUid) {
qTrace("dynQueryCtrl dyn tb uid:%" PRIu64 " reset to:%" PRIu64, pVtbScan->dynTbUid, pValue->uid);
pVtbScan->dynTbUid = pValue->uid;
break;
}
}
}
static int32_t initVtbScanInfo(SDynQueryCtrlOperatorInfo* pInfo, SMsgCb* pMsgCb,
SDynQueryCtrlPhysiNode* pPhyciNode, SExecTaskInfo* pTaskInfo) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
code = tsem_init(&pInfo->vtbScan.ready, 0, 0);
QUERY_CHECK_CODE(code, line, _return);
pInfo->vtbScan.genNewParam = true;
pInfo->vtbScan.batchProcessChild = pPhyciNode->vtbScan.batchProcessChild;
pInfo->vtbScan.hasPartition = pPhyciNode->vtbScan.hasPartition;
pInfo->vtbScan.scanAllCols = pPhyciNode->vtbScan.scanAllCols;
pInfo->vtbScan.isSuperTable = pPhyciNode->vtbScan.isSuperTable;
pInfo->vtbScan.rversion = pPhyciNode->vtbScan.rversion;
pInfo->vtbScan.uid = pPhyciNode->vtbScan.uid;
pInfo->vtbScan.suid = pPhyciNode->vtbScan.suid;
pInfo->vtbScan.epSet = pPhyciNode->vtbScan.mgmtEpSet;
pInfo->vtbScan.acctId = pPhyciNode->vtbScan.accountId;
pInfo->vtbScan.needRedeploy = false;
pInfo->vtbScan.pMsgCb = pMsgCb;
pInfo->vtbScan.curTableIdx = 0;
pInfo->vtbScan.lastTableIdx = -1;
pInfo->vtbScan.dynTbUid = 0;
pInfo->vtbScan.dbName = taosStrdup(pPhyciNode->vtbScan.dbName);
pInfo->vtbScan.tbName = taosStrdup(pPhyciNode->vtbScan.tbName);
QUERY_CHECK_NULL(pInfo->vtbScan.dbName, code, line, _return, terrno)
QUERY_CHECK_NULL(pInfo->vtbScan.tbName, code, line, _return, terrno)
pInfo->vtbScan.existOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pInfo->vtbScan.existOrgTbVg, code, line, _return, terrno)
for (int32_t i = 0; i < LIST_LENGTH(pPhyciNode->vtbScan.pOrgVgIds); ++i) {
SValueNode* valueNode = (SValueNode*)nodesListGetNode(pPhyciNode->vtbScan.pOrgVgIds, i);
int32_t vgId = (int32_t)valueNode->datum.i;
code = taosHashPut(pInfo->vtbScan.existOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
QUERY_CHECK_CODE(code, line, _return);
}
if (pPhyciNode->dynTbname && pTaskInfo) {
updateDynTbUidIfNeeded(&pInfo->vtbScan, pTaskInfo->pStreamRuntimeInfo);
}
pInfo->vtbScan.readColList = taosArrayInit(LIST_LENGTH(pPhyciNode->vtbScan.pScanCols), sizeof(col_id_t));
QUERY_CHECK_NULL(pInfo->vtbScan.readColList, code, line, _return, terrno)
for (int32_t i = 0; i < LIST_LENGTH(pPhyciNode->vtbScan.pScanCols); ++i) {
SColumnNode* pNode = (SColumnNode*)nodesListGetNode(pPhyciNode->vtbScan.pScanCols, i);
QUERY_CHECK_NULL(pNode, code, line, _return, terrno)
QUERY_CHECK_NULL(taosArrayPush(pInfo->vtbScan.readColList, &pNode->colId), code, line, _return, terrno)
}
pInfo->vtbScan.childTableList = taosArrayInit(10, POINTER_BYTES);
QUERY_CHECK_NULL(pInfo->vtbScan.childTableList, code, line, _return, terrno)
pInfo->vtbScan.dbVgInfoMap = taosHashInit(taosArrayGetSize(pInfo->vtbScan.childTableList), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
QUERY_CHECK_NULL(pInfo->vtbScan.dbVgInfoMap, code, line, _return, terrno)
pInfo->vtbScan.otbNameToOtbInfoMap = NULL;
pInfo->vtbScan.otbVgIdToOtbInfoArrayMap = NULL;
pInfo->vtbScan.vtbUidToVgIdMapMap = NULL;
pInfo->vtbScan.vtbGroupIdToVgIdMapMap = NULL;
pInfo->vtbScan.vtbUidTagListMap = NULL;
pInfo->vtbScan.vtbGroupIdTagListMap = NULL;
pInfo->vtbScan.vtbUidToGroupIdMap = NULL;
return code;
_return:
// no need to destroy array and hashmap allocated in this function,
// since the operator's destroy function will take care of it
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
static int32_t initVtbWindowInfo(SDynQueryCtrlOperatorInfo* pInfo, SDynQueryCtrlPhysiNode* pPhyciNode,
SExecTaskInfo* pTaskInfo, SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDataBlockDescNode* pDescNode = pPhyciNode->node.pOutputDataBlockDesc;
pInfo->vtbWindow.wstartSlotId = pPhyciNode->vtbWindow.wstartSlotId;
pInfo->vtbWindow.wendSlotId = pPhyciNode->vtbWindow.wendSlotId;
pInfo->vtbWindow.wdurationSlotId = pPhyciNode->vtbWindow.wdurationSlotId;
pInfo->vtbWindow.pTargets = pPhyciNode->vtbWindow.pTargets;
pInfo->vtbWindow.isVstb = pPhyciNode->vtbWindow.isVstb;
pInfo->vtbWindow.singleWinMode = pPhyciNode->vtbWindow.singleWinMode;
pInfo->vtbWindow.extendOption = pPhyciNode->vtbWindow.extendOption;
pInfo->vtbWindow.pRes = createDataBlockFromDescNode(pDescNode);
QUERY_CHECK_NULL(pInfo->vtbWindow.pRes, code, line, _return, terrno)
pInfo->vtbWindow.pWins = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
QUERY_CHECK_NULL(pInfo->vtbWindow.pWins, code, line, _return, terrno)
pInfo->vtbWindow.outputWstartSlotId = -1;
pInfo->vtbWindow.outputWendSlotId = -1;
pInfo->vtbWindow.outputWdurationSlotId = -1;
pInfo->vtbWindow.curWinBatchIdx = 0;
initResultSizeInfo(&pOperator->resultInfo, 1);
code = blockDataEnsureCapacity(pInfo->vtbWindow.pRes, pOperator->resultInfo.capacity);
QUERY_CHECK_CODE(code, line, _return);
return code;
_return:
qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
return code;
}
static int32_t extractTsCol(SSDataBlock* pBlock, int32_t slotId, TSKEY** ppTsCols) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
if (pBlock->pDataBlock != NULL && pBlock->info.dataLoad) {
SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, slotId);
QUERY_CHECK_NULL(pColDataInfo, code, lino, _return, terrno)
*ppTsCols = (int64_t*)pColDataInfo->pData;
if ((*ppTsCols)[0] != 0 && (pBlock->info.window.skey == 0 && pBlock->info.window.ekey == 0)) {
code = blockDataUpdateTsWindow(pBlock, slotId);
QUERY_CHECK_CODE(code, lino, _return);
}
}
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
return code;
}
int32_t vtbWindowOpen(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SDynQueryCtrlOperatorInfo* pDynInfo = pOperator->info;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
SVtbWindowDynCtrlInfo* pInfo = &pDynInfo->vtbWindow;
int64_t st = 0;
if (OPTR_IS_OPENED(pOperator)) {
return code;
}
if (pOperator->cost.openCost == 0) {
st = taosGetTimestampUs();
}
while (1) {
SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
if (pBlock == NULL) {
break;
}
if (pInfo->outputWendSlotId == -1 && pInfo->outputWstartSlotId == -1 && pInfo->outputWdurationSlotId == -1) {
for (int32_t i = 0; i < LIST_LENGTH(pInfo->pTargets); ++i) {
STargetNode* pNode = (STargetNode*)nodesListGetNode(pInfo->pTargets, i);
if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pExpr)->dataBlockId == pBlock->info.id.blockId) {
if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wstartSlotId) {
pInfo->outputWstartSlotId = i;
} else if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wendSlotId) {
pInfo->outputWendSlotId = i;
} else if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wdurationSlotId) {
pInfo->outputWdurationSlotId = i;
}
}
}
}
TSKEY* wstartCol = NULL;
TSKEY* wendCol = NULL;
code = extractTsCol(pBlock, pDynInfo->vtbWindow.wstartSlotId, &wstartCol);
QUERY_CHECK_CODE(code, lino, _return);
code = extractTsCol(pBlock, pDynInfo->vtbWindow.wendSlotId, &wendCol);
QUERY_CHECK_CODE(code, lino, _return);
if (pDynInfo->vtbWindow.singleWinMode) {
for (int32_t i = 0; i < pBlock->info.rows; i++) {
SArray* pWin = taosArrayInit(pBlock->info.rows, sizeof(SExtWinTimeWindow));
QUERY_CHECK_NULL(pWin, code, lino, _return, terrno)
QUERY_CHECK_NULL(taosArrayReserve(pWin, 1), code, lino, _return, terrno);
SExtWinTimeWindow* pWindow = taosArrayGet(pWin, 0);
QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
pWindow->tw.skey = wstartCol[i];
pWindow->tw.ekey = wendCol[i] + 1;
pWindow->winOutIdx = -1;
QUERY_CHECK_NULL(taosArrayPush(pDynInfo->vtbWindow.pWins, &pWin), code, lino, _return, terrno);
}
} else {
SArray* pWin = taosArrayInit(pBlock->info.rows, sizeof(SExtWinTimeWindow));
QUERY_CHECK_NULL(pWin, code, lino, _return, terrno)
QUERY_CHECK_NULL(taosArrayReserve(pWin, pBlock->info.rows), code, lino, _return, terrno);
for (int32_t i = 0; i < pBlock->info.rows; i++) {
SExtWinTimeWindow* pWindow = taosArrayGet(pWin, i);
QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
pWindow->tw.skey = wstartCol[i];
pWindow->tw.ekey = wendCol[i] + 1;
pWindow->winOutIdx = -1;
}
QUERY_CHECK_NULL(taosArrayPush(pDynInfo->vtbWindow.pWins, &pWin), code, lino, _return, terrno);
}
}
// handle first window's start key and last window's end key
SArray* firstBatch = (SArray*)taosArrayGetP(pDynInfo->vtbWindow.pWins, 0);
SArray* lastBatch = (SArray*)taosArrayGetP(pDynInfo->vtbWindow.pWins, taosArrayGetSize(pDynInfo->vtbWindow.pWins) - 1);
QUERY_CHECK_NULL(firstBatch, code, lino, _return, terrno)
QUERY_CHECK_NULL(lastBatch, code, lino, _return, terrno)
SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(firstBatch, 0);
SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(lastBatch);
QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
if (pInfo->extendOption == STATE_WIN_EXTEND_OPTION_BACKWARD) {
lastWin->tw.ekey = INT64_MAX;
}
if (pInfo->extendOption == STATE_WIN_EXTEND_OPTION_FORWARD) {
firstWin->tw.skey = INT64_MIN;
}
OPTR_SET_OPENED(pOperator);
if (pOperator->cost.openCost == 0) {
pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
}
_return:
if (code != TSDB_CODE_SUCCESS) {
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
return code;
}
static int32_t buildDynQueryCtrlOperatorParamForExternalWindow(SOperatorParam** ppRes, int32_t downstreamIdx, int64_t skey, int64_t ekey) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SDynQueryCtrlOperatorParam* pDyn = NULL;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
(*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
pDyn = taosMemoryMalloc(sizeof(SDynQueryCtrlOperatorParam));
QUERY_CHECK_NULL(pDyn, code, lino, _return, terrno);
pDyn->window.skey = skey;
pDyn->window.ekey = ekey;
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL;
(*ppRes)->downstreamIdx = 0;
(*ppRes)->reUse = false;
(*ppRes)->value = pDyn;
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (pDyn) {
taosMemoryFree(pDyn);
}
if (*ppRes) {
if ((*ppRes)->pChildren) {
taosArrayDestroy((*ppRes)->pChildren);
}
taosMemoryFree(*ppRes);
*ppRes = NULL;
}
return code;
}
static int32_t buildExternalWindowOperatorParamEx(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins, int32_t idx) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SExternalWindowOperatorParam* pExtWinOp = NULL;
*ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
pExtWinOp = taosMemoryMalloc(sizeof(SExternalWindowOperatorParam));
QUERY_CHECK_NULL(pExtWinOp, code, lino, _return, terrno)
pExtWinOp->ExtWins = taosArrayDup(pWins, NULL);
QUERY_CHECK_NULL(pExtWinOp->ExtWins, code, lino, _return, terrno)
SExtWinTimeWindow *firstWin = (SExtWinTimeWindow *)taosArrayGet(pWins, 0);
SExtWinTimeWindow *lastWin = (SExtWinTimeWindow *)taosArrayGet(pWins, taosArrayGetSize(pWins) - 1);
(*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
SOperatorParam* pDynQueryCtrlParam = NULL;
code = buildDynQueryCtrlOperatorParamForExternalWindow(&pDynQueryCtrlParam, 0, firstWin->tw.skey, lastWin->tw.ekey);
QUERY_CHECK_CODE(code, lino, _return);
QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pDynQueryCtrlParam), code, lino, _return, terrno)
(*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW;
(*ppRes)->downstreamIdx = idx;
(*ppRes)->value = pExtWinOp;
(*ppRes)->reUse = false;
return code;
_return:
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
if (pExtWinOp) {
if (pExtWinOp->ExtWins) {
taosArrayDestroy(pExtWinOp->ExtWins);
}
taosMemoryFree(pExtWinOp);
}
if (*ppRes) {
if ((*ppRes)->pChildren) {
for (int32_t i = 0; i < taosArrayGetSize((*ppRes)->pChildren); i++) {
SOperatorParam* pChildParam = (SOperatorParam*)taosArrayGet((*ppRes)->pChildren, i);
if (pChildParam) {
SDynQueryCtrlOperatorParam* pDynParam = (SDynQueryCtrlOperatorParam*)pChildParam->value;
if (pDynParam) {
taosMemoryFree(pDynParam);
}
taosMemoryFree(pChildParam);
}
}
taosArrayDestroy((*ppRes)->pChildren);
}
taosMemoryFree(*ppRes);
*ppRes = NULL;
}
return code;
}
int32_t vtbWindowNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t lino = 0;
SDynQueryCtrlOperatorInfo* pDynInfo = pOperator->info;
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
int64_t st = taosGetTimestampUs();
int32_t numOfWins = 0;
SOperatorInfo* mergeOp = NULL;
SOperatorInfo* extWinOp = NULL;
SOperatorParam* pMergeParam = NULL;
SOperatorParam* pExtWinParam = NULL;
SVtbWindowDynCtrlInfo* pInfo = &pDynInfo->vtbWindow;
SSDataBlock* pRes = pInfo->pRes;
code = pOperator->fpSet._openFn(pOperator);
QUERY_CHECK_CODE(code, lino, _return);
if (pInfo->curWinBatchIdx >= taosArrayGetSize(pInfo->pWins)) {
*ppRes = NULL;
return code;
}
SArray* pWinArray = (SArray*)taosArrayGetP(pInfo->pWins, pInfo->curWinBatchIdx);
QUERY_CHECK_NULL(pWinArray, code, lino, _return, terrno)
numOfWins = (int32_t)taosArrayGetSize(pWinArray);
if (pInfo->isVstb) {
extWinOp = pOperator->pDownstream[1];
code = buildExternalWindowOperatorParamEx(pDynInfo, &pExtWinParam, pWinArray, extWinOp->numOfDownstream);
QUERY_CHECK_CODE(code, lino, _return);
SSDataBlock* pExtWinBlock = NULL;
code = extWinOp->fpSet.getNextExtFn(extWinOp, pExtWinParam, &pExtWinBlock);
QUERY_CHECK_CODE(code, lino, _return);
2026-01-12 09:32:39 +00:00
setOperatorCompleted(extWinOp);
blockDataCleanup(pRes);
code = blockDataEnsureCapacity(pRes, numOfWins);
QUERY_CHECK_CODE(code, lino, _return);
if (pExtWinBlock) {
code = copyColumnsValue(pInfo->pTargets, pExtWinBlock->info.id.blockId, pRes, pExtWinBlock, numOfWins);
QUERY_CHECK_CODE(code, lino, _return);
if (pInfo->curWinBatchIdx == 0) {
// first batch, get _wstart from pMergedBlock
SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(taosArrayGetP(pInfo->pWins, 0), 0);
QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
firstWin->tw.skey = pExtWinBlock->info.window.skey;
}
if (pInfo->curWinBatchIdx == taosArrayGetSize(pInfo->pWins) - 1) {
// last batch, get _wend from pMergedBlock
SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(taosArrayGetP(pInfo->pWins, taosArrayGetSize(pInfo->pWins) - 1));
QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
lastWin->tw.ekey = pExtWinBlock->info.window.ekey + 1;
}
}
} else {
mergeOp = pOperator->pDownstream[1];
code = buildMergeOperatorParam(pDynInfo, &pMergeParam, pWinArray, mergeOp->numOfDownstream, numOfWins);
QUERY_CHECK_CODE(code, lino, _return);
SSDataBlock* pMergedBlock = NULL;
code = mergeOp->fpSet.getNextExtFn(mergeOp, pMergeParam, &pMergedBlock);
QUERY_CHECK_CODE(code, lino, _return);
blockDataCleanup(pRes);
code = blockDataEnsureCapacity(pRes, numOfWins);
QUERY_CHECK_CODE(code, lino, _return);
if (pMergedBlock) {
code = copyColumnsValue(pInfo->pTargets, pMergedBlock->info.id.blockId, pRes, pMergedBlock, numOfWins);
QUERY_CHECK_CODE(code, lino, _return);
if (pInfo->curWinBatchIdx == 0) {
// first batch, get _wstart from pMergedBlock
SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(taosArrayGetP(pInfo->pWins, 0), 0);
QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
firstWin->tw.skey = pMergedBlock->info.window.skey;
}
if (pInfo->curWinBatchIdx == taosArrayGetSize(pInfo->pWins) - 1) {
// last batch, get _wend from pMergedBlock
SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(taosArrayGetP(pInfo->pWins, taosArrayGetSize(pInfo->pWins) - 1));
QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
lastWin->tw.ekey = pMergedBlock->info.window.ekey + 1;
}
}
}
if (pInfo->outputWstartSlotId != -1) {
SColumnInfoData* pWstartCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWstartSlotId);
QUERY_CHECK_NULL(pWstartCol, code, lino, _return, terrno)
for (int32_t i = 0; i < numOfWins; i++) {
SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
code = colDataSetVal(pWstartCol, i, (const char*)&pWindow->tw.skey, false);
QUERY_CHECK_CODE(code, lino, _return);
}
}
if (pInfo->outputWendSlotId != -1) {
SColumnInfoData* pWendCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWendSlotId);
QUERY_CHECK_NULL(pWendCol, code, lino, _return, terrno)
for (int32_t i = 0; i < numOfWins; i++) {
SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
TSKEY ekey = pWindow->tw.ekey - 1;
code = colDataSetVal(pWendCol, i, (const char*)&ekey, false);
QUERY_CHECK_CODE(code, lino, _return);
}
}
if (pInfo->outputWdurationSlotId != -1) {
SColumnInfoData* pWdurationCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWdurationSlotId);
QUERY_CHECK_NULL(pWdurationCol, code, lino, _return, terrno)
for (int32_t i = 0; i < numOfWins; i++) {
SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
int64_t duration = pWindow->tw.ekey - 1 - pWindow->tw.skey;
code = colDataSetVal(pWdurationCol, i, (const char*)&duration, false);
QUERY_CHECK_CODE(code, lino, _return);
}
}
pRes->info.rows = numOfWins;
*ppRes = pRes;
pInfo->curWinBatchIdx++;
return code;
_return:
if (code != TSDB_CODE_SUCCESS) {
qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
pTaskInfo->code = code;
T_LONG_JMP(pTaskInfo->env, code);
}
return code;
}
static int32_t resetDynQueryCtrlOperState(SOperatorInfo* pOper) {
SDynQueryCtrlOperatorInfo* pDyn = pOper->info;
SDynQueryCtrlPhysiNode const* pPhyciNode = pOper->pPhyNode;
SExecTaskInfo* pTaskInfo = pOper->pTaskInfo;
pOper->status = OP_NOT_OPENED;
switch (pDyn->qType) {
case DYN_QTYPE_STB_HASH:{
pDyn->stbJoin.execInfo = (SDynQueryCtrlExecInfo){0};
SStbJoinDynCtrlInfo* pStbJoin = &pDyn->stbJoin;
2025-07-29 09:04:03 +00:00
destroyStbJoinDynCtrlInfo(&pDyn->stbJoin);
2025-07-28 04:35:32 +00:00
int32_t code = initSeqStbJoinTableHash(&pDyn->stbJoin.ctx.prev, pDyn->stbJoin.basic.batchFetch);
if (TSDB_CODE_SUCCESS != code) {
qError("initSeqStbJoinTableHash failed since %s", tstrerror(code));
return code;
}
pStbJoin->ctx.prev.pListHead = NULL;
pStbJoin->ctx.prev.joinBuild = false;
pStbJoin->ctx.prev.pListTail = NULL;
pStbJoin->ctx.prev.tableNum = 0;
pStbJoin->ctx.post = (SStbJoinPostJoinCtx){0};
break;
}
case DYN_QTYPE_VTB_SCAN: {
SVtbScanDynCtrlInfo* pVtbScan = &pDyn->vtbScan;
if (pVtbScan->otbNameToOtbInfoMap) {
taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
pVtbScan->otbNameToOtbInfoMap = NULL;
}
if (pVtbScan->pRsp) {
tFreeSUsedbRsp(pVtbScan->pRsp);
taosMemoryFreeClear(pVtbScan->pRsp);
}
if (pVtbScan->colRefInfo) {
taosArrayDestroyEx(pVtbScan->colRefInfo, destroyColRefInfo);
pVtbScan->colRefInfo = NULL;
}
if (pVtbScan->childTableMap) {
taosHashCleanup(pVtbScan->childTableMap);
pVtbScan->childTableMap = NULL;
}
if (pVtbScan->childTableList) {
taosArrayClearEx(pVtbScan->childTableList, destroyColRefArray);
}
if (pPhyciNode->dynTbname && pTaskInfo) {
updateDynTbUidIfNeeded(pVtbScan, pTaskInfo->pStreamRuntimeInfo);
}
pVtbScan->curTableIdx = 0;
pVtbScan->lastTableIdx = -1;
break;
}
case DYN_QTYPE_VTB_WINDOW: {
SVtbWindowDynCtrlInfo* pVtbWindow = &pDyn->vtbWindow;
if (pVtbWindow->pRes) {
blockDataDestroy(pVtbWindow->pRes);
pVtbWindow->pRes = NULL;
}
if (pVtbWindow->pWins) {
taosArrayDestroyEx(pVtbWindow->pWins, destroyWinArray);
pVtbWindow->pWins = NULL;
}
pVtbWindow->outputWdurationSlotId = -1;
pVtbWindow->outputWendSlotId = -1;
pVtbWindow->outputWstartSlotId = -1;
pVtbWindow->curWinBatchIdx = 0;
break;
}
default:
qError("unsupported dynamic query ctrl type: %d", pDyn->qType);
break;
}
return 0;
2025-04-16 08:36:09 +00:00
}
int32_t vtbAggOpen(SOperatorInfo* pOperator) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
int64_t st = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
if (OPTR_IS_OPENED(pOperator)) {
return code;
}
if (pOperator->cost.openCost == 0) {
st = taosGetTimestampUs();
}
code = buildVirtualSuperTableScanChildTableMap(pOperator);
QUERY_CHECK_CODE(code, line, _return);
OPTR_SET_OPENED(pOperator);
_return:
if (pOperator->cost.openCost == 0) {
pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
}
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, code);
}
return code;
}
int32_t virtualTableAggGetNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
SOperatorInfo* pAggOp = pOperator->pDownstream[pOperator->numOfDownstream - 1];
SOperatorInfo* pTagScanOp = pOperator->pDownstream[1];
SOperatorParam* pAggParam = NULL;
if (pInfo->vtbScan.hasPartition) {
if (pInfo->vtbScan.batchProcessChild) {
void* pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, NULL);
while (pIter) {
size_t keyLen = 0;
uint64_t groupid = *(uint64_t*)taosHashGetKey(pIter, &keyLen);
code = buildAggOperatorParamWithGroupId(pInfo, groupid, &pAggParam);
QUERY_CHECK_CODE(code, line, _return);
if (pAggParam) {
code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
QUERY_CHECK_CODE(code, line, _return);
} else {
*pRes = NULL;
}
pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, pIter);
if (*pRes) {
(*pRes)->info.id.groupId = groupid;
code = taosHashRemove(pVtbScan->vtbGroupIdTagListMap, &groupid, keyLen);
QUERY_CHECK_CODE(code, line, _return);
break;
}
}
} else {
void *pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, NULL);
while (pIter) {
size_t keyLen = 0;
uint64_t* groupid = (uint64_t*)taosHashGetKey(pIter, &keyLen);
SHashObj* vtbUidTagListMap = *(SHashObj**)pIter;
void* pIter2 = taosHashIterate(vtbUidTagListMap, NULL);
while (pIter2) {
size_t keyLen2 = 0;
tb_uid_t uid = *(tb_uid_t*)taosHashGetKey(pIter2, &keyLen2);
SArray* pTagList = *(SArray**)pIter2;
if (pVtbScan->genNewParam) {
code = buildAggOperatorParamForSingleChild(pInfo, uid, *groupid, pTagList, &pAggParam);
QUERY_CHECK_CODE(code, line, _return);
if (pAggParam) {
code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
QUERY_CHECK_CODE(code, line, _return);
} else {
*pRes = NULL;
}
} else {
code = pAggOp->fpSet.getNextFn(pAggOp, pRes);
QUERY_CHECK_CODE(code, line, _return);
}
if (*pRes) {
pVtbScan->genNewParam = false;
(*pRes)->info.id.groupId = *groupid;
break;
}
pVtbScan->genNewParam = true;
pIter2 = taosHashIterate(vtbUidTagListMap, pIter2);
code = taosHashRemove(vtbUidTagListMap, &uid, keyLen);
QUERY_CHECK_CODE(code, line, _return);
}
if (*pRes) {
break;
}
pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, pIter);
code = taosHashRemove(pVtbScan->vtbGroupIdTagListMap, groupid, keyLen);
QUERY_CHECK_CODE(code, line, _return);
}
}
} else {
if (pInfo->vtbScan.batchProcessChild) {
code = buildAggOperatorParam(pInfo, &pAggParam);
QUERY_CHECK_CODE(code, line, _return);
code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
QUERY_CHECK_CODE(code, line, _return);
setOperatorCompleted(pOperator);
} else {
void* pIter = taosHashIterate(pVtbScan->vtbUidTagListMap, NULL);
while (pIter) {
size_t keyLen = 0;
tb_uid_t uid = *(tb_uid_t*)taosHashGetKey(pIter, &keyLen);
SArray* pTagList = *(SArray**)pIter;
if (pVtbScan->genNewParam) {
code = buildAggOperatorParamForSingleChild(pInfo, uid, 0, pTagList, &pAggParam);
QUERY_CHECK_CODE(code, line, _return);
if (pAggParam) {
code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
QUERY_CHECK_CODE(code, line, _return);
} else {
*pRes = NULL;
}
} else {
code = pAggOp->fpSet.getNextFn(pAggOp, pRes);
QUERY_CHECK_CODE(code, line, _return);
}
if (*pRes) {
pVtbScan->genNewParam = false;
break;
}
pVtbScan->genNewParam = true;
pIter = taosHashIterate(pVtbScan->vtbUidTagListMap, pIter);
code = taosHashRemove(pVtbScan->vtbUidTagListMap, &uid, keyLen);
QUERY_CHECK_CODE(code, line, _return);
}
}
}
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
}
return code;
}
int32_t vtbAggNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
SVtbScanDynCtrlInfo* pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
QRY_PARAM_CHECK(pRes);
if (pOperator->status == OP_EXEC_DONE) {
return code;
}
code = pOperator->fpSet._openFn(pOperator);
QUERY_CHECK_CODE(code, line, _return);
if (pVtbScan->isSuperTable && taosArrayGetSize(pVtbScan->childTableList) == 0) {
setOperatorCompleted(pOperator);
return code;
}
code = virtualTableAggGetNext(pOperator, pRes);
QUERY_CHECK_CODE(code, line, _return);
return code;
_return:
if (code) {
qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
pOperator->pTaskInfo->code = code;
T_LONG_JMP(pOperator->pTaskInfo->env, code);
}
return code;
}
2024-07-24 09:08:08 +00:00
int32_t createDynQueryCtrlOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream,
SDynQueryCtrlPhysiNode* pPhyciNode, SExecTaskInfo* pTaskInfo,
SMsgCb* pMsgCb, SOperatorInfo** pOptrInfo) {
QRY_PARAM_CHECK(pOptrInfo);
2023-09-01 05:24:47 +00:00
2024-07-24 09:08:08 +00:00
int32_t code = TSDB_CODE_SUCCESS;
int32_t line = 0;
2024-07-24 09:08:08 +00:00
__optr_fn_t nextFp = NULL;
__optr_open_fn_t openFp = NULL;
2024-08-19 01:15:36 +00:00
SOperatorInfo* pOperator = NULL;
2023-09-01 05:24:47 +00:00
SDynQueryCtrlOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SDynQueryCtrlOperatorInfo));
QUERY_CHECK_NULL(pInfo, code, line, _error, terrno)
2024-07-24 09:08:08 +00:00
2024-08-19 01:15:36 +00:00
pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
QUERY_CHECK_NULL(pOperator, code, line, _error, terrno)
2023-09-01 05:24:47 +00:00
pOperator->pPhyNode = pPhyciNode;
pTaskInfo->dynamicTask = (int8_t)pPhyciNode->node.dynamicOp;
2023-09-01 05:24:47 +00:00
code = appendDownstream(pOperator, pDownstream, numOfDownstream);
QUERY_CHECK_CODE(code, line, _error);
2023-09-01 05:24:47 +00:00
setOperatorInfo(pOperator, "DynQueryCtrlOperator", QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL, false, OP_NOT_OPENED,
pInfo, pTaskInfo);
2023-09-01 05:24:47 +00:00
pInfo->qType = pPhyciNode->qType;
switch (pInfo->qType) {
case DYN_QTYPE_STB_HASH:
2024-07-22 07:07:01 +00:00
TAOS_MEMCPY(&pInfo->stbJoin.basic, &pPhyciNode->stbJoin, sizeof(pPhyciNode->stbJoin));
2025-01-08 03:24:24 +00:00
pInfo->stbJoin.pOutputDataBlockDesc = pPhyciNode->node.pOutputDataBlockDesc;
2023-09-01 05:24:47 +00:00
code = initSeqStbJoinTableHash(&pInfo->stbJoin.ctx.prev, pInfo->stbJoin.basic.batchFetch);
QUERY_CHECK_CODE(code, line, _error);
2023-09-01 05:24:47 +00:00
nextFp = seqStableJoin;
openFp = optrDummyOpenFn;
2023-09-01 05:24:47 +00:00
break;
case DYN_QTYPE_VTB_SCAN:
code = initVtbScanInfo(pInfo, pMsgCb, pPhyciNode, pTaskInfo);
QUERY_CHECK_CODE(code, line, _error);
nextFp = vtbScanNext;
openFp = vtbScanOpen;
break;
case DYN_QTYPE_VTB_WINDOW:
code = initVtbWindowInfo(pInfo, pPhyciNode, pTaskInfo, pOperator);
QUERY_CHECK_CODE(code, line, _error);
nextFp = vtbWindowNext;
openFp = vtbWindowOpen;
break;
case DYN_QTYPE_VTB_AGG:
code = initVtbScanInfo(pInfo, pMsgCb, pPhyciNode, pTaskInfo);
QUERY_CHECK_CODE(code, line, _error);
nextFp = vtbAggNext;
openFp = vtbAggOpen;
break;
2023-09-01 05:24:47 +00:00
default:
qError("unsupported dynamic query ctrl type: %d", pInfo->qType);
code = TSDB_CODE_INVALID_PARA;
goto _error;
}
pOperator->fpSet = createOperatorFpSet(openFp, nextFp, NULL, destroyDynQueryCtrlOperator, optrDefaultBufFn,
2024-07-24 09:08:08 +00:00
NULL, optrDefaultGetNextExtFn, NULL);
2023-09-01 05:24:47 +00:00
setOperatorResetStateFn(pOperator, resetDynQueryCtrlOperState);
2024-07-24 09:08:08 +00:00
*pOptrInfo = pOperator;
2024-08-29 11:43:59 +00:00
return TSDB_CODE_SUCCESS;
2024-07-22 07:07:01 +00:00
2024-07-24 09:08:08 +00:00
_error:
2023-09-01 05:24:47 +00:00
if (pInfo != NULL) {
destroyDynQueryCtrlOperator(pInfo);
}
qError("failed to create dyn query ctrl operator, %s code:%s, line:%d", __func__, tstrerror(code), line);
2024-08-29 11:43:59 +00:00
destroyOperatorAndDownstreams(pOperator, pDownstream, numOfDownstream);
2023-09-01 05:24:47 +00:00
pTaskInfo->code = code;
2024-07-24 09:08:08 +00:00
return code;
2023-09-01 05:24:47 +00:00
}