import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Properties;

@Intercepts({
    @Signature(type = Executor.class, method = "update", 
               args = {MappedStatement.class, Object.class}),
    @Signature(type = Executor.class, method = "query",
               args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
    @Signature(type = Executor.class, method = "query",
               args = {MappedStatement.class, Object.class, RowBounds.class, 
                       ResultHandler.class, CacheKey.class, BoundSql.class})
})
public class SqlCostInterceptor implements Interceptor {
    
    private static final Logger logger = LoggerFactory.getLogger("SQL_PERF");
    
    // 慢查詢(xún)閾值（毫秒）
    private long slowQueryThreshold = 1000;
    
    // 是否打印參數(shù)
    private boolean showParameters = true;
    
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        MappedStatement mappedStatement = (MappedStatement) args[0];
        Object parameterObject = args[1];
        
        // 獲取SQL信息
        BoundSql boundSql = mappedStatement.getBoundSql(parameterObject);
        String sqlId = mappedStatement.getId();
        String sql = boundSql.getSql().replaceAll("\\s+", " ");
        
        long start = System.currentTimeMillis();
        try {
            return invocation.proceed(); // 執(zhí)行目標(biāo)方法
        } finally {
            long cost = System.currentTimeMillis() - start;
            logSqlPerformance(sqlId, sql, parameterObject, cost);
        }
    }
    
    private void logSqlPerformance(String sqlId, String sql, 
                                   Object params, long cost) {
        // 基礎(chǔ)日志
        String logMsg = String.format("SQL執(zhí)行耗時(shí): %dms | ID: %s | SQL: %s",
                                     cost, sqlId, sql);
        
        // 參數(shù)日志（敏感數(shù)據(jù)需脫敏）
        if (showParameters && params != null) {
            String paramStr = params.toString();
            // 簡(jiǎn)單脫敏處理（實(shí)際項(xiàng)目應(yīng)使用專(zhuān)業(yè)脫敏工具）
            paramStr = paramStr.replaceAll("(\"password\":\")([^\"]+)(\")", 
                                          "$1****$3");
            logMsg += " | 參數(shù): " + paramStr;
        }
        
        // 分級(jí)日志輸出
        if (cost > slowQueryThreshold) {
            logger.warn("[慢查詢(xún)告警] {}", logMsg);
        } else if (cost > 500) {
            logger.info("{}", logMsg);
        } else {
            logger.debug("{}", logMsg);
        }
    }
    
    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }
    
    @Override
    public void setProperties(Properties properties) {
        // 從配置加載參數(shù)
        String threshold = properties.getProperty("slowQueryThreshold");
        if (threshold != null) {
            this.slowQueryThreshold = Long.parseLong(threshold);
        }
        
        String showParams = properties.getProperty("showParameters");
        if (showParams != null) {
            this.showParameters = Boolean.parseBoolean(showParams);
        }
    }
}

@Configuration
public class MyBatisConfig {
    
    @Bean
    public SqlCostInterceptor sqlCostInterceptor() {
        SqlCostInterceptor interceptor = new SqlCostInterceptor();
        Properties properties = new Properties();
        properties.setProperty("slowQueryThreshold", "500"); // 500ms以上視為慢查詢(xún)
        properties.setProperty("showParameters", "true");
        interceptor.setProperties(properties);
        return interceptor;
    }
}

private final ExecutorService logExecutor = 
    Executors.newSingleThreadExecutor();

private void logSqlPerformance(...) {
    logExecutor.submit(() -> {
        // 日志記錄邏輯
    });
}

// 在intercept方法中加入
if (ThreadLocalRandom.current().nextDouble() < 0.2) {
    // 只記錄20%的請(qǐng)求
}

private void recordMetrics(String sqlId, long cost) {
    // Prometheus指標(biāo)記錄
    Summary.builder("sql_execution_time")
           .tag("sql_id", sqlId)
           .register()
           .observe(cost);
    
    // 慢查詢(xún)計(jì)數(shù)器
    if (cost > slowQueryThreshold) {
        Counter.builder("slow_sql_count")
               .tag("sql_id", sqlId)
               .register()
               .inc();
    }
}

if (paramStr.length() > 1000) {
    paramStr = paramStr.substring(0, 1000) + "...[truncated]";
}