package com.zb.iscrm.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @Auther: 楊紅星
 * @Date: 2018/11/28 09:38
 * @Description:
 */
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD})
public @interface UpdateTime {
 String value() default "";
}

package com.zb.iscrm.mybatisInterceptor;

import com.zb.iscrm.annotation.UpdateTime;
import com.zb.iscrm.utils.DateUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.*;

import java.lang.reflect.Field;
import java.util.Properties;

/**
 * @Auther: 楊紅星
 * @Date: 2018/11/28 09:41
 * @Description: mybatis插件 用于執(zhí)行Update時(shí)將當(dāng)前時(shí)間加入
 */
@Slf4j
@Intercepts({ @Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }) })
public class UpdateTimeInterceptor implements Interceptor {

 @Override
 public Object intercept(Invocation invocation) throws Throwable {
  MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
  // 獲取 SQL 命令
  SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
  // 獲取參數(shù)
  Object parameter = invocation.getArgs()[1];
  if (parameter != null) {
   // 獲取成員變量
   Field[] declaredFields = parameter.getClass().getDeclaredFields();
   for (Field field : declaredFields) {
    if (field.getAnnotation(UpdateTime.class) != null) { // update 語(yǔ)句插入 updateTime
     if (SqlCommandType.INSERT.equals(sqlCommandType) || SqlCommandType.UPDATE.equals(sqlCommandType)) {
      field.setAccessible(true);
      if (field.get(parameter) == null) {
       field.set(parameter, DateUtils.dateTimeNow(DateUtils.YYYY_MM_DD_HH_MM_SS));
      }
     }
    }
   }
  }
  //同樣的方式也可以在這里添加create_time或者是id的生成等處理
  
  return invocation.proceed();
 }

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

 @Override
 public void setProperties(Properties properties) {
 }
}

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration
  PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
  "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
 <!--插件注冊(cè)-->
 <plugins>
  <plugin interceptor="com.zb.iscrm.mybatisInterceptor.UpdateTimeInterceptor"/>
 </plugins>
</configuration>