package com.hyx.study.commom.aop.interceptor;

    import com.hyx.study.commom.util.JacksonUtils;

    import com.xkzhangsan.time.calculator.DateTimeCalculatorUtil;

    import org.apache.ibatis.executor.Executor;

    import org.apache.ibatis.executor.statement.StatementHandler;

    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.slf4j.Logger;

    import org.slf4j.LoggerFactory;

    import org.apache.commons.beanutils.BeanUtils;

    import org.springframework.stereotype.Component;

    import java.lang.reflect.Field;

    import java.lang.reflect.InvocationTargetException;

    import java.sql.Connection;

    import java.sql.Statement;

    import java.util.*;

    /**

     * @author huangyaxiong

     * @version 1.0

     * @description: 利用mybatis拦截器，修改所有得Date类型得参数省略毫秒数

     * @date 2022-09-06 9:59

     */

    @Intercepts({

            @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})

    })

    @Component

    public class MybatisDateInterceptor implements Interceptor {

        private static Logger logger = LoggerFactory.getLogger(MybatisDateInterceptor.class);

        /**

         * 获取sql语句

         * @param invocation

         * @return

         */

        public static String getSqlByInvocation(Invocation invocation) {

            final Object[] args = invocation.getArgs();

            MappedStatement ms = (MappedStatement) args[0];

            Object parameterObject = args[1];

            BoundSql boundSql = ms.getBoundSql(parameterObject);

            return boundSql.getSql();

        }

        @Override

        public Object intercept(Invocation invocation) throws InvocationTargetException, IllegalAccessException {

            String processSql = getSqlByInvocation(invocation);

            logger.info("原始sql==>{}",processSql);

            Object[] args = invocation.getArgs();

            for(Object arg:args){

                if(arg instanceof MappedStatement){

                    MappedStatement mappedStatement = (MappedStatement) arg;

                    SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();

                    logger.info("操作类型==>{}",sqlCommandType);

                    if(sqlCommandType ==  SqlCommandType.INSERT || sqlCommandType == SqlCommandType.UPDATE){

                        continue;

                    }else{

                        break;

                    }

                }else if(arg instanceof Map){

                    //如果是map，有两种情况：（1）使用@Param多参数传入，由Mybatis包装成map。（2）原始传入Map

                    logger.info("这是一个包装过的类型!");

                    Map map=(Map)arg;

                    for (Object obj : map.values()){

                        tranDateProperty(obj);

                    }

                }else {

                    tranDateProperty(arg);

                }

            }

            return invocation.proceed ();

        }

        /**

         * @description: 修改对象中对的Date类型并且舍弃毫秒数

         * @param arg

         * @author 'huangyaxiong'

         * @date: 2022-09-06 11:16

         */

        private static void tranDateProperty(Object arg){

            try{

                Field[] declaredFields = arg.getClass().getDeclaredFields();

                for(Field field:declaredFields){

                    Class<?> type = field.getType();

                    if(Date.class == field.getType()){

                        field.setAccessible(true);

                        System.out.println(type);

                        String name = field.getName();

                        Object temp = field.get(arg);

                        if(temp != null){

                            Date value = (Date) field.get(arg);

                            System.out.println(value.getTime());

                            Date date = DateTimeCalculatorUtil.reduceAccuracyToSecond(value);

                            System.out.println(date.getTime());

                            field.set(arg,date);

    //                        BeanUtils.setProperty(arg,name,date);

                        }

                    }

                }

            }catch (Exception e){

                logger.error("日期数据转换出现了异常==>data:{}==>e:{}", JacksonUtils.object2Json(arg),e);

            }

        }

        @Override

        public Object plugin(Object target) {

            // 当目标类是StatementHandler类型时，才包装目标类，否者直接返回目标本身,减少目标被代理的次数

            return Plugin.wrap(target, this);

    //        return null;

        }

        @Override

        public void setProperties(Properties properties) {

        }

    }