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

09-12

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

  该实现是在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

  {

  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)

  {

  }

  }