MyBatis拦截器:给参数对象属性赋值的实例

该拦截器的作用：在进行增加、修改等操作时，给数据模型的一些通用操作属性（如：创建人、创建时间、修改人、修改时间等）自动赋值。

该实现是在DAO层拦截，即存入DB前最后一层。后经分析，不是很合理，改为在service层拦截，用spring AOP来实现了，该代码遂弃用。不过已经测试可用，记录备忘。

package com.development;

import java.lang.reflect.InvocationTargetException;
import java.util.Date;
import java.util.Map;
import java.util.Properties;

import org.apache.commons.beanutils.BeanUtils;
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;

/**
 * 拦截器作用：给各实体对象在增加、修改时，自动添加操作属性信息。
 */
@Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class }) })
public class OpeInfoInterceptor implements Interceptor
{

  public Object intercept(Invocation invocation) throws Throwable
  {
    Object[] args = invocation.getArgs();

    System.out.println("-----------参数拦截---------------------------------------------------");
    System.out.println("02 当前线程ID:"+Thread.currentThread().getId());
    //遍历处理所有参数，update方法有两个参数，参见Executor类中的update()方法。
    for(int i=0;i<args.length;i++)
    {
      Object arg=args[i];
      String className=arg.getClass().getName();
      System.out.println(i + " 参数类型："+className);

      //第一个参数处理。根据它判断是否给“操作属性”赋值。
      if(arg instanceof MappedStatement)
      {//如果是第一个参数 MappedStatement
        MappedStatement ms = (MappedStatement)arg;
        SqlCommandType sqlCommandType = ms.getSqlCommandType();
        System.out.println("操作类型："+sqlCommandType);
        if(sqlCommandType == SqlCommandType.INSERT || sqlCommandType==SqlCommandType.UPDATE)
        {//如果是“增加”或“更新”操作，则继续进行默认操作信息赋值。否则，则退出
          continue;
        }
        else
        {
          break;
        }
      }

      //第二个参数处理。（只有第二个程序才能跑到这）
      if (arg instanceof Map)
      {//如果是map，有两种情况：（1）使用@Param多参数传入，由Mybatis包装成map。（2）原始传入Map
        System.out.println("这是一个包装过的类型!");
        Map map=(Map)arg;
        for (Object obj : map.values())
        {
          setProperty(obj);
        }
      }
      else
      {//原始参数传入
        setProperty(arg);
      }

    }

    return invocation.proceed();

  }

  /**
   * 为对象的操作属性赋值
   * @param obj
   */
  private void setProperty(Object obj)
  {
    try
    {
      //TODO: 根据需要，将相关属性赋上默认值
      BeanUtils.setProperty(obj, "createrUsername", "张三");
      BeanUtils.setProperty(obj, "createDT", new Date());
    }
    catch (IllegalAccessException e)
    {
      e.printStackTrace();
    }
    catch (InvocationTargetException e)
    {
      e.printStackTrace();
    }
  }

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

  public void setProperties(Properties properties)
  {

  }

}

以上这篇MyBatis拦截器:给参数对象属性赋值的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持我们。