import java.sql.Statement; import java.util.Properties; import org.apache.ibatis.executor.statement.StatementHandler; import org.apache.ibatis.logging.commons.JakartaCommonsLoggingImpl; import org.apache.ibatis.mapping.BoundSql; import org.apache.ibatis.plugin.Interceptor; import org.apache.ibatis.plugin.Intercepts; import org.apache.ibatis.plugin.Invocation; import org.apache.ibatis.plugin.Plugin; import org.apache.ibatis.plugin.Signature; import org.apache.ibatis.session.ResultHandler; /** * Sql執行時間記錄攔截器 */ @Intercepts({ @Signature(type = StatementHandler.class, method = "query", args = { Statement.class, ResultHandler.class }), @Signature(type = StatementHandler.class, method = "update", args = { Statement.class }), @Signature(type = StatementHandler.class, method = "batch", args = { Statement.class }) }) public class SqlCostInterceptor implements Interceptor { private static final JakartaCommonsLoggingImpl logger = new JakartaCommonsLoggingImpl("SqlCostInterceptor"); @Override public Object intercept(Invocation invocation) throws Throwable { long startTime = System.currentTimeMillis(); StatementHandler statementHandler = (StatementHandler) invocation.getTarget(); final BoundSql boundSql = statementHandler.getBoundSql(); try { return invocation.proceed(); } finally { final StringBuilder build = new StringBuilder(7); build.append("<== Preparing: "); build.append("\n"); build.append(boundSql.getSql()); build.append("\n"); build.append("MYBATIS-SQL執行耗時["); build.append((System.currentTimeMillis() - startTime)); build.append("ms]"); logger.debug(build.toString()); } } @Override public Object plugin(Object target) { return Plugin.wrap(target, this); } @Override public void setProperties(Properties properties) { } }