wangli/msg-center-plat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

REQ-3284: 查询数量过大告警

Browse Source
This commit is contained in:
yanglin 2024-12-02 13:38:32 +08:00
parent d9b2b18913
commit f92d5e9c3c
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
inside-notices/src/main/java/cn/axzo/msg/center/message/service/todo/mybatis/rowcountwarn/ProxyStatement.java
View File

@ -57,7 +57,15 @@ class ProxyStatement extends StatementWrapper {
void rowAdvanced() {
rowCount++;
tryWarnFor("Threshold", true);
if (rowCountAboveThreshold() && canWarnPeriodically()) {
warnFor("Threshold");
periodWarnTimes++;
}
}
private boolean canWarnPeriodically() {
int maxWarnTimes = props.getRowCount().getPeriodMaxWarnTimes();
return rowCount % 2000 == 0 && periodWarnTimes <= maxWarnTimes;
}
@Override
@ -76,17 +84,13 @@ class ProxyStatement extends StatementWrapper {
@Override
public void close() throws SQLException {
super.close();
if (!closeWarned) {
tryWarnFor("StatementClose", false);
if (rowCountAboveThreshold() && !closeWarned) {
warnFor("StatementClose");
closeWarned = true;
}
}
private void tryWarnFor(String phase, boolean periodically) {
if (!rowCountAboveThreshold()) return;
if (periodically && !canWarnPeriodically()) return;
if (periodically)
periodWarnTimes++;
private void warnFor(String phase) {
try {
// make a copy
int rowCount = this.rowCount;
@ -109,11 +113,6 @@ class ProxyStatement extends StatementWrapper {
return rowCount > props.getRowCount().determineWarningThreshold();
}
private boolean canWarnPeriodically() {
int maxWarnTimes = props.getRowCount().getPeriodMaxWarnTimes();
return rowCount % 2000 == 0 && periodWarnTimes <= maxWarnTimes;
}
private void sendWarn(String phase, int currentCount, String traceId) {
DingTalkService dingTalkService = beanFactory
.getBeanProvider(DingTalkService.class).getIfAvailable();