admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

set "auditEnable" default value to false

Browse Source
This commit is contained in:
开源海哥 2023-03-28 16:31:56 +08:00
parent e33b9a93e1
commit 1b4b55a038
2 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/AuditManager.java
View File

@ -28,10 +28,18 @@ import java.util.Map;
*/ */
public class AuditManager { public class AuditManager {
private static boolean auditEnable = false;
private static Clock clock = System::currentTimeMillis; private static Clock clock = System::currentTimeMillis;
private static MessageCreator messageCreator = new DefaultMessageCreator(); private static MessageCreator messageCreator = new DefaultMessageCreator();
private static MessageCollector messageCollector = new ScheduledMessageCollector(); private static MessageCollector messageCollector = new ScheduledMessageCollector();
public static boolean isAuditEnable() {
return auditEnable;
}
public static void setAuditEnable(boolean auditEnable) {
AuditManager.auditEnable = auditEnable;
}
public static Clock getClock() { public static Clock getClock() {
return clock; return clock;

20
mybatis-flex-core/src/main/java/com/mybatisflex/core/mybatis/FlexStatementHandler.java
View File

@ -43,6 +43,7 @@ public class FlexStatementHandler implements StatementHandler {
private final StatementHandler delegate; private final StatementHandler delegate;
private final BoundSql boundSql; private final BoundSql boundSql;
private final boolean auditEnable = AuditManager.isAuditEnable();
public FlexStatementHandler(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) { public FlexStatementHandler(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
@ -77,25 +78,32 @@ public class FlexStatementHandler implements StatementHandler {
@Override @Override
public void batch(Statement statement) throws SQLException { public void batch(Statement statement) throws SQLException {
AuditManager.startAudit(() -> { if (auditEnable) {
AuditManager.startAudit(() -> {
delegate.batch(statement);
return null;
}, boundSql);
} else {
delegate.batch(statement); delegate.batch(statement);
return null; }
}, boundSql);
} }
@Override @Override
public int update(Statement statement) throws SQLException { public int update(Statement statement) throws SQLException {
return AuditManager.startAudit(() -> delegate.update(statement), boundSql); return auditEnable ? AuditManager.startAudit(() -> delegate.update(statement), boundSql)
: delegate.update(statement);
} }
@Override @Override
public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException { public <E> List<E> query(Statement statement, ResultHandler resultHandler) throws SQLException {
return AuditManager.startAudit(() -> delegate.query(statement, resultHandler), boundSql); return auditEnable ? AuditManager.startAudit(() -> delegate.query(statement, resultHandler), boundSql)
: delegate.query(statement, resultHandler);
} }
@Override @Override
public <E> Cursor<E> queryCursor(Statement statement) throws SQLException { public <E> Cursor<E> queryCursor(Statement statement) throws SQLException {
return AuditManager.startAudit(() -> delegate.queryCursor(statement), boundSql); return auditEnable ? AuditManager.startAudit(() -> delegate.queryCursor(statement), boundSql)
: delegate.queryCursor(statement);
} }
@Override @Override