mybatis 自定义 拦截器

时间:2021-06-01 10:21:29

转载的。。。。。http://zhenghuazhi.iteye.com/blog/1468992

页面输入:男,数据库保存male,女,数据库保存为female。

使用interceptor,typeHandler

Java代码  mybatis 自定义 拦截器
  1. package cn.dcr.mybatis.util;  
  2.   
  3. import java.util.Properties;  
  4.   
  5. import org.apache.ibatis.executor.Executor;  
  6. import org.apache.ibatis.mapping.MappedStatement;  
  7. import org.apache.ibatis.plugin.Interceptor;  
  8. import org.apache.ibatis.plugin.Intercepts;  
  9. import org.apache.ibatis.plugin.Invocation;  
  10. import org.apache.ibatis.plugin.Plugin;  
  11. import org.apache.ibatis.plugin.Signature;  
  12. import org.apache.ibatis.session.ResultHandler;  
  13. import org.apache.ibatis.session.RowBounds;  
  14.   
  15. import com.test.pojos.User;  
  16.   
  17. @Intercepts({  
  18.         @Signature(type = Executor.class, method = "update", args = {  
  19.                 MappedStatement.class, Object.class }),  
  20.         @Signature(type = Executor.class, method = "query", args = {  
  21.                 MappedStatement.class, Object.class, RowBounds.class,  
  22.                 ResultHandler.class }) })  
  23. public class MyInterceptor implements Interceptor {  
  24.   
  25.     private static final String R_FEMALE = "女";  
  26.   
  27.     private static final String R_MALE = "男";  
  28.   
  29.     private static final String FEMALE = "female";  
  30.   
  31.     private static final String MALE = "male";  
  32.     private Properties properties;  
  33.   
  34.     /* 
  35.      * (non-Javadoc) 
  36.      *  
  37.      * @see 
  38.      * org.apache.ibatis.plugin.Interceptor#intercept(org.apache.ibatis.plugin 
  39.      * .Invocation) 
  40.      */  
  41.     public Object intercept(Invocation invocation) throws Throwable {  
  42.   
  43.         MappedStatement mappedStatement = (MappedStatement) invocation  
  44.                 .getArgs()[0];  
  45.         String sqlId = mappedStatement.getId();  
  46.         String namespace = sqlId.substring(0, sqlId.indexOf('.'));  
  47.         Executor exe = (Executor) invocation.getTarget();  
  48.         String methodName = invocation.getMethod().getName();  
  49.   
  50.         if (methodName.equals("query")) {  
  51.                 Object parameter = invocation.getArgs()[1];  
  52.                 RowBounds rowBounds = (RowBounds) invocation.getArgs()[2];  
  53.                    
  54.         }  
  55.         else if(methodName.equals("update")){  
  56.             Object parameter = invocation.getArgs()[1];  
  57.             if(parameter instanceof User)  
  58.                 ((User)parameter).setGender(saveValueToDb(((User)parameter).getGender()));  
  59.               
  60.         }  
  61.         return invocation.proceed();  
  62.   
  63.     }  
  64.   
  65.     /* 
  66.      * (non-Javadoc) 
  67.      *  
  68.      * @see org.apache.ibatis.plugin.Interceptor#plugin(java.lang.Object) 
  69.      */  
  70.     public Object plugin(Object target) {  
  71.         return Plugin.wrap(target, this);  
  72.     }  
  73.   
  74.     /* 
  75.      * (non-Javadoc) 
  76.      *  
  77.      * @see 
  78.      * org.apache.ibatis.plugin.Interceptor#setProperties(java.util.Properties) 
  79.      */  
  80.     public void setProperties(Properties properties) {  
  81.         this.properties = properties;  
  82.   
  83.     }  
  84.     /**插入数据库 
  85.      * @param value 
  86.      * @return 
  87.      */  
  88.     private String saveValueToDb(String value) {  
  89.         if (value.equals(R_FEMALE)) {  
  90.             return FEMALE;  
  91.         } else if (value.equals(R_MALE)) {  
  92.             return MALE;  
  93.         } else {  
  94.             throw new IllegalArgumentException("数据库异常!" + value);  
  95.         }  
  96.     }  
  97.   
  98. }  

 

Java代码  mybatis 自定义 拦截器
  1.    

 

Java代码  mybatis 自定义 拦截器
  1. @Intercepts({  
  2.         @Signature(type = Executor.class, method = "update", args = {  
  3.                 MappedStatement.class, Object.class }),  
  4.         @Signature(type = Executor.class, method = "query", args = {  
  5.                 MappedStatement.class, Object.class, RowBounds.class,  
  6.                 ResultHandler.class }) })  

 mybatis 拦截器好像只能使用注解,而且对于接口Executor,method只定义了update,query,flushStatements,commit,rollback,createCacheKey,isCached,clearLocalCache,deferLoad,getTransaction,close,isClosed这几个方法,没有delete和insert方法。

具体详见:

Java代码  mybatis 自定义 拦截器
  1. package org.apache.ibatis.executor;  
  2.   
  3. import org.apache.ibatis.cache.CacheKey;  
  4. import org.apache.ibatis.mapping.MappedStatement;  
  5. import org.apache.ibatis.reflection.MetaObject;  
  6. import org.apache.ibatis.session.ResultHandler;  
  7. import org.apache.ibatis.session.RowBounds;  
  8. import org.apache.ibatis.transaction.Transaction;  
  9.   
  10. import java.sql.SQLException;  
  11. import java.util.List;  
  12.   
  13. public interface Executor {  
  14.   
  15.   ResultHandler NO_RESULT_HANDLER = null;  
  16.   
  17.   int update(MappedStatement ms, Object parameter) throws SQLException;  
  18.   
  19.   List query(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler) throws SQLException;  
  20.   
  21.   List<BatchResult> flushStatements() throws SQLException;  
  22.   
  23.   void commit(boolean required) throws SQLException;  
  24.   
  25.   void rollback(boolean required) throws SQLException;  
  26.   
  27.   CacheKey createCacheKey(MappedStatement ms, Object parameterObject, RowBounds rowBounds);  
  28.   
  29.   boolean isCached(MappedStatement ms, CacheKey key);  
  30.   
  31.   void clearLocalCache();  
  32.   
  33.   void deferLoad(MappedStatement ms, MetaObject resultObject, String property, CacheKey key);  
  34.   
  35.   Transaction getTransaction();  
  36.   
  37.   void close(boolean forceRollback);  
  38.     
  39.   boolean isClosed();  
  40.   
  41. }  

    当我疑惑为什么mybatis没有在insert和delete的时候拦截呢?

看到了mybatis googleCode上的issue16。

Issue 16: Update interceptor (plugin) fires for inserts
Html代码  mybatis 自定义 拦截器
  1. Reporter:    Eli Kizhnerman  
  2.   
  3. The iBatis plugin does not seem to be resolving the target methods to  
  4. intercept properly. An interceptor that is defined to intercept only  
  5. updates is also executed on inserts.  
  6.   
  7. I have the following definition:  
  8. @Intercepts({@Signature(  
  9.         type=Executor.class,  
  10.         method="update",  
  11.         args={MappedStatement.class,Object.class})})  
  12. public class UpdateInterceptor implements Interceptor {  
  13. ...  
  14. }  
  15.   
  16. In Plugin.invoke you have the following code:  
  17.   
  18.       Set<Methodmethods = signatureMap.get(method.getDeclaringClass());  
  19.       if (methods != null && methods.contains(method)) {  
  20.         return interceptor.intercept(new Invocation(target, method, args));  
  21.       }  
  22.   
  23. DefaultSqlSession.insert internally calls an update method:  
  24.   
  25.   public int insert(String statement, Object parameter) {  
  26.     return update(statement, parameter);  
  27.   }  
  28.   
  29. I suspect that the Method that is being passed to plugin.invoke is actually  
  30. the update method and therefore the interceptor is executed.   

  原来在DefaultSqlSession的insert,delete方法也是调用了update方法。

Java代码  mybatis 自定义 拦截器
  1. public int insert(String statement, Object parameter) {  
  2.     return update(statement, parameter);  
  3.   }  

  

Java代码  mybatis 自定义 拦截器
  1. public int delete(String statement, Object parameter) {  
  2.     return update(statement, wrapCollection(parameter));  
  3.   }  

   mybatis太不厚道了,它的文档也没有具体的说明。

    有人提出这是个bug,不过mybatis没有承认。

Java代码  mybatis 自定义 拦截器
  1. Clinton Begin added a comment - 10/Dec/09 03:58 PM  
  2. Executor only has update and query methods. Thus inserts and deletes are also  
  3. considered updates. It sounds like to do what you want, we need another interception  
  4. point, at the session level. You can do so right now with a proxy class between the  
  5. SqlSession interface and the default implementation.  
  6. [ Show » ]  
  7. Clinton Begin added a comment - 10/Dec/09 03:58 PM Executor only has update and query  
  8. methods. Thus inserts and deletes are also considered updates. It sounds like to do  
  9. what you want, we need another interception point, at the session level. You can do  
  10. so right now with a proxy class between the SqlSession interface and the default  
  11. implementation.  
  12.   
  13. [ Permalink | Delete | « Hide ]  
  14. Eli Kizhnerman added a comment - 11/Dec/09 12:28 PM  
  15. This is not working the way I expected it but that it is not a bug. I can get what I  
  16. need from the MappedStatement SqlCommandType.  

 

Java代码  mybatis 自定义 拦截器
  1. package cn.dcr.mybatis.util;  
  2.   
  3. import java.sql.CallableStatement;  
  4. import java.sql.PreparedStatement;  
  5. import java.sql.ResultSet;  
  6. import java.sql.SQLException;  
  7.   
  8. import org.apache.ibatis.type.BaseTypeHandler;  
  9. import org.apache.ibatis.type.JdbcType;  
  10.   
  11. /**自定义typeHandler<br/> 
  12.  * 1 插入数据库,男转为male 
  13.  * 2 查询,maile转为男 
  14.  * @author Administrator 
  15.  * 
  16.  */  
  17. public class GenderTypeHandlerCallback extends BaseTypeHandler<String> {  
  18.   
  19.     private static final String R_FEMALE = "女";  
  20.   
  21.     private static final String R_MALE = "男";  
  22.   
  23.     private static final String FEMALE = "female";  
  24.   
  25.     private static final String MALE = "male";  
  26.   
  27.     @Override  
  28.     public String getNullableResult(CallableStatement cs, int columnIndex)  
  29.             throws SQLException {  
  30.   
  31.         return cs.getString(columnIndex);  
  32.     }  
  33.   
  34.     @Override  
  35.     public String getNullableResult(ResultSet rs, String columnName)  
  36.             throws SQLException {  
  37.         String t = rs.getString(columnName);  
  38.         return converSex(t);  
  39.     }  
  40.   
  41.     @Override  
  42.     public void setNonNullParameter(PreparedStatement preparedStatement, int i,  
  43.             String str, JdbcType jdbcType) throws SQLException {  
  44.   
  45.         preparedStatement.setString(i, saveValueToDb(str));  
  46.   
  47.     }  
  48.   
  49.     /**插入数据库 
  50.      * @param value 
  51.      * @return 
  52.      */  
  53.     private String saveValueToDb(String value) {  
  54.         if (value.equals(R_FEMALE)) {  
  55.             return FEMALE;  
  56.         } else if (value.equals(R_MALE)) {  
  57.             return MALE;  
  58.         } else {  
  59.             throw new IllegalArgumentException("数据库异常!" + value);  
  60.         }  
  61.     }  
  62.       
  63.     /** 从数据库读出 
  64.      * @param value 
  65.      * @return 
  66.      */  
  67.     private String converSex(String value){  
  68.         if (value.equals(FEMALE)) {  
  69.             return R_FEMALE;  
  70.         } else if (value.equals(MALE)) {  
  71.             return R_MALE;  
  72.         } else {  
  73.             throw new IllegalArgumentException("数据库异常!" + value);  
  74.         }  
  75.     }  
  76.   
  77. }  

  在自定义typeHandler的getNullableResult方法中调整自己的结果集,可以说是在sql执行后,IOC的理论叫后置通知,interceptor应该是前置通知。

Java代码  mybatis 自定义 拦截器
  1. @Override  
  2.     public String getNullableResult(ResultSet rs, String columnName)  
  3.             throws SQLException {  
  4.         String t = rs.getString(columnName);  
  5.         return converSex(t);  
  6.     }  

 在typeHandler的setNonNullParameter,应该是参数传入前转换参数的功能,但是实际操作中没有被调用,不知道为什么。

Java代码  mybatis 自定义 拦截器
  1. @Override  
  2.     public void setNonNullParameter(PreparedStatement preparedStatement, int i,  
  3.             String str, JdbcType jdbcType) throws SQLException {  
  4.   
  5.         preparedStatement.setString(i, saveValueToDb(str));  
  6.   
  7.     }  

 以上是对mybatis 的切入点的实践。总体感觉和struts2的拦截器比,还有差距,至少我理解struts2的拦截器比理解它的要快。