fix(stream):check null pointer

This commit is contained in:
54liuyao 2024-09-18 16:22:14 +08:00
parent 6a1cde28fb
commit bb39402174
2 changed files with 12 additions and 0 deletions

View File

@ -81,6 +81,9 @@ bool inSessionWindow(SSessionKey* pKey, TSKEY ts, int64_t gap) {
SStreamStateCur* createSessionStateCursor(SStreamFileState* pFileState) { SStreamStateCur* createSessionStateCursor(SStreamFileState* pFileState) {
SStreamStateCur* pCur = createStreamStateCursor(); SStreamStateCur* pCur = createStreamStateCursor();
if (pCur == NULL) {
return NULL;
}
pCur->pStreamFileState = pFileState; pCur->pStreamFileState = pFileState;
return pCur; return pCur;
} }
@ -533,6 +536,9 @@ static SStreamStateCur* seekKeyCurrentPrev_buff(SStreamFileState* pFileState, co
if (index >= 0) { if (index >= 0) {
pCur = createSessionStateCursor(pFileState); pCur = createSessionStateCursor(pFileState);
if (pCur == NULL) {
return NULL;
}
pCur->buffIndex = index; pCur->buffIndex = index;
if (pIndex) { if (pIndex) {
*pIndex = index; *pIndex = index;
@ -634,6 +640,9 @@ SStreamStateCur* countWinStateSeekKeyPrev(SStreamFileState* pFileState, const SS
pBuffCur->buffIndex = 0; pBuffCur->buffIndex = 0;
} else if (taosArrayGetSize(pWinStates) > 0) { } else if (taosArrayGetSize(pWinStates) > 0) {
pBuffCur = createSessionStateCursor(pFileState); pBuffCur = createSessionStateCursor(pFileState);
if (pBuffCur == NULL) {
return NULL;
}
pBuffCur->buffIndex = 0; pBuffCur->buffIndex = 0;
} }

View File

@ -530,6 +530,9 @@ void streamStateCopyBackend(SStreamState* src, SStreamState* dst) {
} }
SStreamStateCur* createStreamStateCursor() { SStreamStateCur* createStreamStateCursor() {
SStreamStateCur* pCur = taosMemoryCalloc(1, sizeof(SStreamStateCur)); SStreamStateCur* pCur = taosMemoryCalloc(1, sizeof(SStreamStateCur));
if (pCur == NULL) {
return NULL;
}
pCur->buffIndex = -1; pCur->buffIndex = -1;
return pCur; return pCur;
} }