package com.xa.demo1.config;

import lombok.extern.slf4j.Slf4j;
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.mapping.ParameterMapping;
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.reflection.MetaObject;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.apache.ibatis.type.TypeHandlerRegistry;
import org.springframework.util.ObjectUtils;

import java.text.DateFormat;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Properties;
import java.util.regex.Matcher;

/**
 * @author AkazaAkari   實現(xiàn)一個優(yōu)雅打日志的功能
 * @Intercepts 注解：
 *
 * 這是 MyBatis 提供的一個注解，用于定義一個攔截器。一個攔截器可以攔截一個或多個方法。
 * @Signature 注解：
 *
 * 這是 @Intercepts 注解的子注解，用于指定要攔截的目標(biāo)方法。
 *
 * @Signature 注解的參數(shù)分別是：
 *
 *     type：指定要攔截的接口類型。
 *
 *     method：指定要攔截的方法名。
 *
 *     args：指定要攔截的方法的參數(shù)類型列表。
 *      todo： 這一章其他內(nèi)容之后補上（Intercepts注解、MyBatis插件開發(fā)過程中的各個對象
 */
@Intercepts({
        @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}),
        @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
@Slf4j
public class SqlInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        // 計算這一次SQL執(zhí)行錢后的時間，統(tǒng)計一下執(zhí)行耗時
        long startTime = System.currentTimeMillis();
        Object proceed = invocation.proceed();
        long endTime = System.currentTimeMillis();

        String printSql = null;
        try {
            // 通過generateSql方法拿到最終生成的SQL
            printSql = generateSql(invocation);
        }catch (Exception exception){
            log.error("獲取sql異常",exception);
        }finally {
            // 拼接日志打印過程
            long costTime = endTime - startTime;
            log.info("\n 執(zhí)行SQL耗時：{}ms \n 執(zhí)行SQL：{}",costTime,printSql);
        }
        return proceed;
    }

    private static String generateSql(Invocation invocation){
        // 獲取到BoundSql以及Configuration對象
        // BoundSql 對象存儲了一條具體的 SQL 語句及其相關(guān)參數(shù)信息。
        // Configuration 對象保存了 MyBatis 框架運行時所有的配置信息
        MappedStatement statement = (MappedStatement) invocation.getArgs()[0];
        Object parameter = null;
        if (invocation.getArgs().length>1){
            parameter = invocation.getArgs()[1];
        }
        Configuration configuration = statement.getConfiguration();
        BoundSql boundSql = statement.getBoundSql(parameter);

        // 獲取參數(shù)對象
        Object parameterObject = boundSql.getParameterObject();
        // 獲取參數(shù)映射
        List<ParameterMapping> params = boundSql.getParameterMappings();
        // 獲取到執(zhí)行的SQL
        String sql = boundSql.getSql();
        // SQL中多個空格使用一個空格代替
        sql = sql.replaceAll("[\\s]+", " ");
        if (!ObjectUtils.isEmpty(params) && !ObjectUtils.isEmpty(parameterObject)){
            // TypeHandlerRegistry 是 MyBatis 用來管理 TypeHandler 的注冊器。TypeHandler 用于在 Java 類型和 JDBC 類型之間進行轉(zhuǎn)換
            TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();
            // 如果參數(shù)對象的類型有對應(yīng)的 TypeHandler，則使用 TypeHandler 進行處理
            if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())){
                sql = sql.replaceFirst("\\?", Matcher.quoteReplacement(getParameterValue(parameterObject)));
            }else {
                // 否則，逐個處理參數(shù)映射
                for (ParameterMapping param : params) {
                    // 獲取參數(shù)的屬性名
                    String propertyName = param.getProperty();
                    MetaObject metaObject = configuration.newMetaObject(parameterObject);
                    // 檢查對象中是否存在該屬性的 getter 方法，如果存在就取出來進行替換
                    if (metaObject.hasGetter(propertyName)){
                        Object obj = metaObject.getValue(propertyName);
                        sql = sql.replaceFirst("\\?", Matcher.quoteReplacement(getParameterValue(obj)));
                        // 檢查 BoundSql 對象中是否存在附加參數(shù)。附加參數(shù)可能是在動態(tài) SQL 處理中生成的，有的話就進行替換
                    }else if (boundSql.hasAdditionalParameter(propertyName)){
                        Object obj = boundSql.getAdditionalParameter(propertyName);
                        sql = sql.replaceFirst("\\?", Matcher.quoteReplacement(getParameterValue(obj)));
                    }else {
                        // 如果都沒有，說明SQL匹配不上，帶上“缺失”方便找問題
                        sql = sql.replaceFirst("\\?", "缺失");
                    }
                }
            }
        }
        return sql;
    }

    private static String getParameterValue(Object object) {
        String value = "";
        if (object instanceof String){
            value = "'" + object.toString() + "'";
        }else if (object instanceof Date){
            DateFormat format = DateFormat.getDateTimeInstance(DateFormat.DEFAULT, DateFormat.DEFAULT, Locale.CHINA);
            value = "'" + format.format((Date) object) + "'";
        } else if (!ObjectUtils.isEmpty(object)) {
            value = object.toString();
        }
        return value;
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    @Override
    public void setProperties(Properties properties) {
        // 可以通過properties配置插件參數(shù)
    }
}

/**
     * @Title: mybatisConfigurationCustomizer
     * @Description: 編寫一個MyBatis的配置類，將這個插件注冊進去
     * @CreatedDate: 2024/06/16 00:06
     * @Param
     * @Return ConfigurationCustomizer
     **/
    @Bean
    public ConfigurationCustomizer mybatisConfigurationCustomizer() {
        return configuration -> {
            configuration.addInterceptor(new SqlInterceptor());
        };
    }