xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix(query): check for the handle value.

This commit is contained in:
Haojun Liao 2023-02-28 16:28:02 +08:00
parent 281c774f6e
commit 95c6629937
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
source/libs/executor/src/dataSinkMgt.c
View File

@ -37,10 +37,18 @@ int32_t dsCreateDataSinker(const SDataSinkNode* pDataSink, DataSinkHandle* pHand
switch ((int)nodeType(pDataSink)) {
case QUERY_NODE_PHYSICAL_PLAN_DISPATCH:
return createDataDispatcher(&gDataSinkManager, pDataSink, pHandle);
case QUERY_NODE_PHYSICAL_PLAN_DELETE:
case QUERY_NODE_PHYSICAL_PLAN_DELETE: {
if (pParam == NULL) {
return TSDB_CODE_QRY_INVALID_INPUT;
}
return createDataDeleter(&gDataSinkManager, pDataSink, pHandle, pParam);
case QUERY_NODE_PHYSICAL_PLAN_QUERY_INSERT:
}
case QUERY_NODE_PHYSICAL_PLAN_QUERY_INSERT: {
if (pParam == NULL) {
return TSDB_CODE_QRY_INVALID_INPUT;
}
return createDataInserter(&gDataSinkManager, pDataSink, pHandle, pParam);
}
}
qError("invalid input node type:%d, %s", nodeType(pDataSink), id);

2
source/libs/executor/src/executor.c
View File

@ -469,7 +469,7 @@ int32_t qCreateExecTask(SReadHandle* readHandle, int32_t vgId, uint64_t taskId,
if (handle) {
void* pSinkParam = NULL;
code = createDataSinkParam(pSubplan->pDataSink, &pSinkParam, pTaskInfo, readHandle);
if (code != TSDB_CODE_SUCCESS || pSinkParam == NULL) {
if (code != TSDB_CODE_SUCCESS) {
qError("failed to createDataSinkParam, vgId:%d, code:%s, %s", vgId, tstrerror(code), (*pTask)->id.str);
goto _error;
}