mybatis实现SQL查询拦截修改详解
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了mybatis实现SQL查询拦截修改详解，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含4967字，纯文字阅读大概需要8分钟。
内容图文
前言
截器的一个作用就是我们可以拦截某些方法的调用，我们可以选择在这些被拦截的方法执行前后加上某些逻辑，也可以在执行这些被拦截的方法时执行自己的逻辑而不再执行被拦截的方法。
Mybatis拦截器设计的一个初衷就是为了供用户在某些时候可以实现自己的逻辑而不必去动Mybatis固有的逻辑。比如我想针对所有的SQL执行某个固定的操作，针对SQL查询执行安全检查，或者记录相关SQL查询日志等等。
Mybatis为我们提供了一个Interceptor接口，可以实现自定义的拦截器。
public interface Interceptor {
Object intercept(Invocation invocation) throws Throwable;
Object plugin(Object target);
void setProperties(Properties properties);
}
接口中包含了三个方法定义
intercept方法为具体的拦截对象的处理方法，传入的Invocation包含了拦截目标类的实力，拦截的方法和方法的入参数组。使用Invocation的procced执行原函数。
plugin 中执行判断是否要进行拦截进，如果不需要拦截，直接返回target，如果需要拦截则调用Plugin类中的wrap静态方法，如果当前拦截器实现了任意接口，则返回一个代理对象，否则直接返回（回忆代理模式的设计）。代理对象实际是一个Plugin类实例，它实现了InvocationHandler接口 ，InvocationHandler接口仅包含invoke方法用于回调方法。
当执行代理对象的接口方法时，会调用Plugin的invoke方法，它会把要执行的对象，方法和参数打包成Invocation对象传给拦截器的intercept方法。Invocation定义了一个procced方法，用于执行被拦截的原方法。
Plugin类定义
public class Plugin implements InvocationHandler {
private Object target;
private Interceptor interceptor;
private Map, Set> signatureMap;
private Plugin(Object target, Interceptor interceptor, Map, Set> signatureMap) {
this.target = target;
this.interceptor = interceptor;
this.signatureMap = signatureMap;
}
public static Object wrap(Object target, Interceptor interceptor) {
Map, Set> signatureMap = getSignatureMap(interceptor);
Class type = target.getClass();
Class[] interfaces = getAllInterfaces(type, signatureMap);
if (interfaces.length > 0) {
return Proxy.newProxyInstance(
type.getClassLoader(),
interfaces,
new Plugin(target, interceptor, signatureMap));
}
return target;
}
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
try {
Set methods = signatureMap.get(method.getDeclaringClass());
if (methods != null && methods.contains(method)) {
return interceptor.intercept(new Invocation(target, method, args));
}
return method.invoke(target, args);
} catch (Exception e) {
throw ExceptionUtil.unwrapThrowable(e);
}
}
private static Map, Set> getSignatureMap(Interceptor interceptor) {
Intercepts interceptsAnnotation = interceptor.getClass().getAnnotation(Intercepts.class);
if (interceptsAnnotation == null) { // issue #251
throw new PluginException("No @Intercepts annotation was found in interceptor " + interceptor.getClass().getName());
}
Signature[] sigs = interceptsAnnotation.value();
Map, Set> signatureMap = new HashMap, Set>();
for (Signature sig : sigs) {
Set methods = signatureMap.get(sig.type());
if (methods == null) {
methods = new HashSet();
signatureMap.put(sig.type(), methods);
}
try {
Method method = sig.type().getMethod(sig.method(), sig.args());
methods.add(method);
} catch (NoSuchMethodException e) {
throw new PluginException("Could not find method on " + sig.type() + " named " + sig.method() + ". Cause: " + e, e);
}
}
return signatureMap;
}
private static Class[] getAllInterfaces(Class type, Map, Set> signatureMap) {
Set> interfaces = new HashSet>();
while (type != null) {
for (Class c : type.getInterfaces()) {
if (signatureMap.containsKey(c)) {
interfaces.add(c);
}
}
type = type.getSuperclass();
}
return interfaces.toArray(new Class[interfaces.size()]);
}
}
setProperties 方法顾名思义，用于设置属性的。bean的属性初始化方法有很多，这是其中的一种。
mybatis提供了@Intercepts注解用于声明当前类是拦截器，其值为@Signature数组，表明要拦截的接口、方法以及对应的参数类型
@Intercepts({@Signature(method = "prepare", type = StatementHandler.class, args = {Connection.class}),
@Signature(method = "query", type = StatementHandler.class, args = {java.sql.Statement.class, ResultHandler.class})})
public class TenantInterceptor implements Interceptor {
.....
例如上面的类声明，第一个Signature标注拦截了StatementHandler类下的入参是一个Connection的名为prepare的方法。
第二个Signature标注拦截StatementHandler类中包含2个入参（分别为Statement和ResultHandler类型）的名为query的方法。
最后，声明的Interceptor需要注册到mybatis的plug中才能生效。
<!-- 配置mybatis -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource"/>
<property name="configLocation" value="classpath:mybatis/mybatis-config.xml"/>
<!-- mapper扫描 -->
<property name="mapperLocations" value="classpath:mybatis/*/*.xml"/>
<property name="plugins">
<array>
<!-- 注册自己的拦截器 -->
<bean id="paginationInterceptor" class="xxx.xxx.TenantInterceptor">
</bean>
</array>
</property>
</bean>
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
- MyBatis拦截器实现分页功能的实现方法
- MyBatis拦截器:给参数对象属性赋值的实例
- java利用mybatis拦截器统计sql执行时间示例
- MyBatis实现模糊查询的几种方式
- Mybatis批量修改的操作代码
- MyBatis批量添加、修改和删除
内容总结
以上是互联网集市为您收集整理的mybatis实现SQL查询拦截修改详解全部内容，希望文章能够帮你解决mybatis实现SQL查询拦截修改详解所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。