使用mybatis Interceptor對UpdateTime自動處理

通常數據庫的表結構都會有update_time，修改時間，由於這個字段基本與業務沒有太大關聯，所以開發過程當中常常會忘記設置這兩個字段的值，本插件就是來解決這個問題。一樣的想生成id，create_time等操做都是能夠以一樣的方式解決。想折騰的同窗還能夠經過這中方式本身寫個分頁插件。閒話少說上代碼。

1. 先寫一個自定義註解標註是update_time

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 "";
}

複製代碼

2. 寫一個mybatis插件

使用@Intercepts標註這是個mybatis插件，@Signature標註要攔截的操做

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插件 用於執行Update時將當前時間加入 */
@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();
        // 獲取參數
        Object parameter = invocation.getArgs()[1];
        if (parameter != null) {
            // 獲取成員變量
            Field[] declaredFields = parameter.getClass().getDeclaredFields();
            for (Field field : declaredFields) {
                if (field.getAnnotation(UpdateTime.class) != null) { // update 語句插入 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) {
    }
}

複製代碼

最後在mybatis的配置文件中註冊插件，而後就大功告成

<?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>
    <!--插件註冊-->
    <plugins>
        <plugin interceptor="com.zb.iscrm.mybatisInterceptor.UpdateTimeInterceptor"/>
    </plugins>
</configuration>
複製代碼