fix(stream): check return value.

This commit is contained in:
Haojun Liao 2024-07-27 01:17:50 +08:00
parent 3073d6d067
commit e3415199f5
1 changed files with 5 additions and 0 deletions

View File

@ -173,6 +173,11 @@ int32_t tsortGetSortedDataBlock(const SSortHandle* pSortHandle, SSDataBlock** pB
return TSDB_CODE_SUCCESS;
}
if (pSortHandle->pDataBlock == NULL) {
*pBlock = NULL;
return TSDB_CODE_SUCCESS;
}
*pBlock = createOneDataBlock(pSortHandle->pDataBlock, false);
if (*pBlock == NULL) {
return TSDB_CODE_OUT_OF_MEMORY;