[Java] Mybatis Interceptor 拦截器的实现

@intercepts({
  @signature(type = executor.class, method = "update", args = {mappedstatement.class, object.class})
})

object parameter = invocation.getargs()[1];
if (parameter instanceof baseentity) {
  baseentity entity = (baseentity) parameter;
}

object parameter = invocation.getargs()[1];
if (parameter instanceof map) {
  map map = (map) parameter;
}

public enum sqlcommandtype {
 unknown, insert, update, delete, select, flush;
}

mappedstatement ms = (mappedstatement) invocation.getargs()[0];
sqlcommandtype commandtype = ms.getsqlcommandtype();

public class baseentity {
  private int id;
  private int creator;
  private int updater;
  private long createtime;
  private long updatetime;
}

int add(bookentity entity);

int update(@param("id") int id, @param("url") string url, @param("description") string description, @param("playcount") int playcount, @param("creator") int creator, @param("updatetime") long updatetime);

package com.github.zhgxun.talk.common.plugin;
 
import com.github.zhgxun.talk.common.util.dateutil;
import com.github.zhgxun.talk.common.util.userutil;
import com.github.zhgxun.talk.entity.baseentity;
import com.github.zhgxun.talk.entity.userentity;
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.interceptor;
import org.apache.ibatis.plugin.intercepts;
import org.apache.ibatis.plugin.invocation;
import org.apache.ibatis.plugin.plugin;
import org.apache.ibatis.plugin.signature;
import org.springframework.stereotype.component;
 
import java.util.map;
import java.util.properties;
 
/**
 * 全局拦截数据库创建和更新
 * <p>
 * signature 对应 invocation 构造器, type 为 invocation.object, method 为 invocation.method, args 为 invocation.object[]
 * method 对应的 update 包括了最常用的 insert/update/delete 三种操作, 因此 update 本身无法直接判断sql为何种执行过程
 * args 包含了其余多有的操作信息, 按数组进行存储, 不同的拦截方式有不同的参数顺序, 具体看type接口的方法签名, 然后根据签名解析, 参见官网
 *
 * @link http://www.mybatis.org/mybatis-3/zh/configuration.html#plugins 插件
 * <p>
 * mappedstatement 包括了sql具体操作类型, 需要通过该类型判断当前sql执行过程
 */
@component
@intercepts({
  @signature(type = executor.class, method = "update", args = {mappedstatement.class, object.class})
})
@slf4j
public class normalplugin implements interceptor {
 
  @override
  @suppresswarnings("unchecked")
  public object intercept(invocation invocation) throws throwable {
  // 根据签名指定的args顺序获取具体的实现类
  // 1. 获取mappedstatement实例, 并获取当前sql命令类型
  mappedstatement ms = (mappedstatement) invocation.getargs()[0];
  sqlcommandtype commandtype = ms.getsqlcommandtype();
 
  // 2. 获取当前正在被操作的类, 有可能是java bean, 也可能是普通的操作对象, 比如普通的参数传递
  // 普通参数, 即是 @param 包装或者原始 map 对象, 普通参数会被 mybatis 包装成 map 对象
  // 即是 org.apache.ibatis.binding.mappermethod$parammap
  object parameter = invocation.getargs()[1];
  // 获取拦截器指定的方法类型, 通常需要拦截 update
  string methodname = invocation.getmethod().getname();
  log.info("normalplugin, methodname; {}, commandtype: {}", methodname, commandtype);
 
  // 3. 获取当前用户信息
  userentity userentity = userutil.getcurrentuser();
  // 默认测试参数值
  int creator = 2, updater = 3;
 
  if (parameter instanceof baseentity) {
    // 4. 实体类
    baseentity entity = (baseentity) parameter;
    if (userentity != null) {
    creator = entity.getcreator();
    updater = entity.getupdater();
    }
    if (methodname.equals("update")) {
    if (commandtype.equals(sqlcommandtype.insert)) {
      entity.setcreator(creator);
      entity.setupdater(updater);
      entity.setcreatetime(dateutil.gettimestamp());
      entity.setupdatetime(dateutil.gettimestamp());
    } else if (commandtype.equals(sqlcommandtype.update)) {
      entity.setupdater(updater);
      entity.setupdatetime(dateutil.gettimestamp());
    }
    }
  } else if (parameter instanceof map) {
    // 5. @param 等包装类
    // 更新时指定某些字段的最新数据值
    if (commandtype.equals(sqlcommandtype.update)) {
    // 遍历参数类型, 检查目标参数值是否存在对象中, 该方式需要应用编写有一些统一的规范
    // 否则均统一为实体对象, 就免去该重复操作
    map map = (map) parameter;
    if (map.containskey("creator")) {
      map.put("creator", creator);
    }
    if (map.containskey("updatetime")) {
      map.put("updatetime", dateutil.gettimestamp());
    }
    }
  }
  // 6. 均不是需要被拦截的类型, 不做操作
  return invocation.proceed();
  }
 
  @override
  public object plugin(object target) {
  return plugin.wrap(target, this);
  }
 
  @override
  public void setproperties(properties properties) {
  }
}

for (field field:parameter.getclass().getdeclaredfields()) {
 
}