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

fix mem leak

This commit is contained in:
yihaoDeng 2023-05-06 13:57:33 +00:00
parent 08c3a63341
commit 163a862f5f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
source/libs/stream/src/streamBackendRocksdb.c
View File

@ -1798,10 +1798,14 @@ int32_t streamStateSessionClear_rocksdb(SStreamState* pState) {
int32_t code = streamStateSessionGetKVByCur_rocksdb(pCur, &delKey, &buf, &size); int32_t code = streamStateSessionGetKVByCur_rocksdb(pCur, &delKey, &buf, &size);
if (code == 0 && size > 0) { if (code == 0 && size > 0) {
memset(buf, 0, size); memset(buf, 0, size);
// refactor later
streamStateSessionPut_rocksdb(pState, &delKey, buf, size); streamStateSessionPut_rocksdb(pState, &delKey, buf, size);
} else { } else {
taosMemoryFreeClear(buf);
break; break;
} }
taosMemoryFreeClear(buf);
streamStateCurNext_rocksdb(pState, pCur); streamStateCurNext_rocksdb(pState, pCur);
} }
streamStateFreeCur(pCur); streamStateFreeCur(pCur);
@ -1842,7 +1846,7 @@ int32_t streamStateStateAddIfNotExist_rocksdb(SStreamState* pState, SSessionKey*
streamStateFreeCur(pCur); streamStateFreeCur(pCur);
pCur = streamStateSessionSeekKeyNext_rocksdb(pState, key); pCur = streamStateSessionSeekKeyNext_rocksdb(pState, key);
} }
taosMemoryFreeClear(*pVal);
code = streamStateSessionGetKVByCur_rocksdb(pCur, key, pVal, pVLen); code = streamStateSessionGetKVByCur_rocksdb(pCur, key, pVal, pVLen);
if (code == 0) { if (code == 0) {
void* stateKey = (char*)(*pVal) + (valSize - keyDataLen); void* stateKey = (char*)(*pVal) + (valSize - keyDataLen);
@ -1852,13 +1856,14 @@ int32_t streamStateStateAddIfNotExist_rocksdb(SStreamState* pState, SSessionKey*
goto _end; goto _end;
} }
} }
taosMemoryFreeClear(*pVal);
*key = tmpKey; *key = tmpKey;
res = 1; res = 1;
memset(tmp, 0, valSize); memset(tmp, 0, valSize);
_end: _end:
taosMemoryFreeClear(*pVal);
*pVal = tmp; *pVal = tmp;
streamStateFreeCur(pCur); streamStateFreeCur(pCur);
return res; return res;