hcq/openGauss-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix the codegen memory leak

Browse Source
This commit is contained in:
gentle_hu
2022-11-10 10:05:22 +08:00
parent 93ad4ac372
commit 8878fec290
3 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/gausskernel/runtime/codegen/gscodegen.cpp
View File

@ -1036,7 +1036,8 @@ void CodeGenThreadInitialize()
bool CodeGenThreadObjectReady()
{
return t_thrd.codegen_cxt.thr_codegen_obj != NULL && !t_thrd.codegen_cxt.g_runningInFmgr;
return t_thrd.codegen_cxt.thr_codegen_obj != NULL && !t_thrd.codegen_cxt.g_runningInFmgr
&& !((dorado::GsCodeGen*)t_thrd.codegen_cxt.thr_codegen_obj)->IsCompiled();
}
/**

5
src/gausskernel/runtime/executor/execMain.cpp
View File

@ -787,7 +787,8 @@ void standard_ExecutorEnd(QueryDesc *queryDesc)
UnregisterSnapshot(estate->es_crosscheck_snapshot);
#ifdef ENABLE_LLVM_COMPILE
if (!t_thrd.codegen_cxt.g_runningInFmgr) {
/* Do not release codegen in Fmgr and Procedure */
if (!t_thrd.codegen_cxt.g_runningInFmgr && u_sess->SPI_cxt._connected == -1) {
CodeGenThreadTearDown();
}
#endif
@ -3826,4 +3827,4 @@ TupleTableSlot* FetchPlanSlot(PlanState* subPlanState, ProjectionInfo** projInfo
} else {
return ExecProcNode(subPlanState);
}
}
}

8
src/include/codegen/gscodegen.h
View File

@ -539,6 +539,14 @@ public:
m_currentModule = NULL;
}
/*
* @Description : Get m_moduleCompiled.
*/
inline bool IsCompiled()
{
return m_moduleCompiled;
}
/*
* @Description : IR compile to machine code.
*/