package com.thhy.general.config.mysql;

import com.thhy.general.common.enums.DataSources;
import com.thhy.general.config.mysql.tos.MoreReflection;
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.SqlCommandType;
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 org.springframework.transaction.annotation.Transactional;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Properties;

@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}
        )
})
public class SwtichInterceptor implements Interceptor {

    private Logger logger = LoggerFactory.getLogger(SwtichInterceptor.class);

    @Override
    public Object intercept(Invocation invocation) throws Throwable {

        //System.out.println(invocation.getMethod().getName());
        MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
        //Connection connection = ((SqlSessionFactory) SpringContextUtils.getBean(SqlSessionFactory.class)).openSession().getConnection();
        String idclass = mappedStatement.getId();
        String className = idclass.substring(0,idclass.lastIndexOf("."));

        String serviceClass = MoreReflection.getCallerPlace(Class.forName(className)).getClassName();
        String serviceMethod = MoreReflection.getCallerPlace(Class.forName(className)).getMethodName();
        //System.out.println("========"+serviceClass+","+serviceMethod);
        Class clazz = Class.forName(serviceClass);
        Method[] methods = clazz.getMethods();
        for(Method m : methods){
            if(m.getName().equals(serviceMethod)){
                Annotation[] annotations = m.getDeclaredAnnotations();
                for(Annotation annotation : annotations){
                    if(annotation.annotationType().equals(Transactional.class)){
                        //System.out.println(mappedStatement.getBoundSql());
                        return invocation.proceed();
                    }
                }
            }
        }
        Object parameter = null;
        if (invocation.getArgs().length > 1) {
            parameter = invocation.getArgs()[1];
        }
        if (!SqlCommandType.SELECT.equals(mappedStatement.getSqlCommandType())) {
            //
            try {
                DynamicDataSource.setDataSource(DataSources.MASTER.name());
                System.out.println(mappedStatement.getSqlSource().toString());
                return invocation.proceed();
            } finally {
                DynamicDataSource.clearDataSource();
            }
        }else{
            //查询
            try {
                DynamicDataSource.setDataSource(DataSources.SLAVE.name());

                //printSql(serviceClass,serviceMethod,idclass,mappedStatement,parameter);
                return invocation.proceed();
            } finally {
                DynamicDataSource.clearDataSource();
            }
        }

    }



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

    @Override
    public void setProperties(Properties properties) {
        Interceptor.super.setProperties(properties);
    }

    public void printSql(String service,String serviceMethod,String mapperMethod,MappedStatement mappedStatement,Object parameter){

        logger.info("调用的Service：["+service+"]中的：["+serviceMethod+"]方法");
        logger.info("调用Mapper：["+mapperMethod+"]");
        String sql = mappedStatement.getBoundSql(parameter).getSql();
        logger.info("调用的sql：\n["+sql+"]");
    }
}