database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](profile) fix query instance profile may be lost. (#12418)

Browse Source
This commit is contained in:
luozenglin
2022-09-09 22:58:04 +08:00
committed by GitHub
parent a6a378c9ca
commit 0c260152b7
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
View File

@ -245,8 +245,12 @@ public class StmtExecutor implements ProfileWriter {
plannerProfile.initRuntimeProfile(plannerRuntimeProfile);
queryProfile.getCounterTotalTime().setValue(TimeUtils.getEstimatedTime(plannerProfile.getQueryBeginTime()));
if (coord != null) {
coord.endProfile(waiteBeReport);
endProfile(waiteBeReport);
}
private void endProfile(boolean waitProfileDone) {
if (context != null && context.getSessionVariable().enableProfile() && coord != null) {
coord.endProfile(waitProfileDone);
}
}
@ -466,6 +470,10 @@ public class StmtExecutor implements ProfileWriter {
throw e;
}
} finally {
// The final profile report occurs after be returns the query data, and the profile cannot be
// received after unregisterQuery(), causing the instance profile to be lost, so we should wait
// for the profile before unregisterQuery().
endProfile(true);
QeProcessorImpl.INSTANCE.unregisterQuery(context.queryId());
}
}
@ -1472,6 +1480,7 @@ public class StmtExecutor implements ProfileWriter {
*/
throwable = t;
} finally {
endProfile(true);
QeProcessorImpl.INSTANCE.unregisterQuery(context.queryId());
}