mybatis Interceptor对UpdateTime自动处理的实现方法

时间:2022-09-01 13:57:15

前言

一般数据库的表结构都会有update_time,修改时间,因为这个字段基本与业务没有太大关联,因此开发过程中经常会忘记设置这两个字段的值,本插件就是来解决这个问题。同样的想生成id,create_time等操作都是可以以同样的方式解决。想折腾的同学还可以通过这中方式自己写个分页插件。

闲话少说上代码。

1. 先写一个自定义注解标注是update_time

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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标注要拦截的操作

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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的配置文件中注册插件,然后就大功告成

?
1
2
3
4
5
6
7
8
9
10
<?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>

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对服务器之家的支持。

原文链接:https://juejin.im/post/5c036bf4f265da61524d23a3