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

fix: add 多数据源打印数据源名字 新增扩展 SqlDebugExtPrinter

Browse Source
This commit is contained in:
huangxy 2024-01-18 10:21:11 +08:00
parent a777a97c1b
commit f9502671cf
1 changed files with 27 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/ConsoleMessageCollector.java
View File

@ -18,12 +18,18 @@ package com.mybatisflex.core.audit;
public class ConsoleMessageCollector implements MessageCollector {
private SqlDebugPrinter printer = (sql, dsName, tookTimeMillis) -> {
private SqlDebugPrinter printer = (sql, tookTimeMillis) -> {
if (tookTimeMillis != null) {
System.out.println("Flex exec sql took " + tookTimeMillis + " ms >>> " + sql);
} else {
System.out.println("Flex exec sql >>> " + sql);
}
};
private SqlDebugExtPrinter extPrinter = (sql, dsName, tookTimeMillis) -> {
StringBuilder buffer = new StringBuilder();
buffer.append("Flex exec");
if (dsName != null) {
buffer.append("dsName >>> ").append(dsName);
}
buffer.append("dsName >>> ").append(dsName);
if (tookTimeMillis != null) {
buffer.append(" sql took ").append(tookTimeMillis).append(" ms >>> ").append(sql);
} else {
@ -39,13 +45,29 @@ public class ConsoleMessageCollector implements MessageCollector {
this.printer = printer;
}
public ConsoleMessageCollector(SqlDebugExtPrinter printer) {
this.extPrinter = printer;
}
@Override
public void collect(AuditMessage message) {
printer.print(message.getFullSql(), message.getDsName(), message.getElapsedTime());
if (message.getDsName() == null) {
printer.print(message.getFullSql(), message.getElapsedTime());
} else {
extPrinter.print(message.getFullSql(), message.getDsName(), message.getElapsedTime());
}
}
public interface SqlDebugPrinter {
void print(String sql, Long tookTimeMillis);
}
public interface SqlDebugExtPrinter {
void print(String sql, String dsName, Long tookTimeMillis);
}