在mybatis执行SQL语句之前进行拦击处理
转载自:http://blog.csdn.net/hfmbook/article/details/41985853
比较适用于在分页时候进行拦截。对分页的SQL语句通过封装处理,处理成不同的分页sql。
实用性比较强。
- import java.sql.Connection;
- import java.sql.PreparedStatement;
- import java.sql.ResultSet;
- import java.sql.SQLException;
- import java.util.List;
- import java.util.Properties;
- import org.apache.ibatis.executor.parameter.ParameterHandler;
- import org.apache.ibatis.executor.statement.RoutingStatementHandler;
- import org.apache.ibatis.executor.statement.StatementHandler;
- import org.apache.ibatis.mapping.BoundSql;
- import org.apache.ibatis.mapping.MappedStatement;
- import org.apache.ibatis.mapping.ParameterMapping;
- 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.apache.ibatis.scripting.defaults.DefaultParameterHandler;
- import com.yidao.utils.Page;
- import com.yidao.utils.ReflectHelper;
- /**
- *
- * 分页拦截器,用于拦截需要进行分页查询的操作,然后对其进行分页处理。
- * 利用拦截器实现Mybatis分页的原理:
- * 要利用JDBC对数据库进行操作就必须要有一个对应的Statement对象,Mybatis在执行Sql语句前就会产生一个包含Sql语句的Statement对象,而且对应的Sql语句
- * 是在Statement之前产生的,所以我们就可以在它生成Statement之前对用来生成Statement的Sql语句下手。在Mybatis中Statement语句是通过RoutingStatementHandler对象的
- * prepare方法生成的。所以利用拦截器实现Mybatis分页的一个思路就是拦截StatementHandler接口的prepare方法,然后在拦截器方法中把Sql语句改成对应的分页查询Sql语句,之后再调用
- * StatementHandler对象的prepare方法,即调用invocation.proceed()。
- * 对于分页而言,在拦截器里面我们还需要做的一个操作就是统计满足当前条件的记录一共有多少,这是通过获取到了原始的Sql语句后,把它改为对应的统计语句再利用Mybatis封装好的参数和设
- * 置参数的功能把Sql语句中的参数进行替换,之后再执行查询记录数的Sql语句进行总记录数的统计。
- *
- */
- @Intercepts({@Signature(type=StatementHandler.class,method="prepare",args={Connection.class})})
- public class PageInterceptor implements Interceptor {
- private String dialect = ""; //数据库方言
- private String pageSqlId = ""; //mapper.xml中需要拦截的ID(正则匹配)
- public Object intercept(Invocation invocation) throws Throwable {
- //对于StatementHandler其实只有两个实现类,一个是RoutingStatementHandler,另一个是抽象类BaseStatementHandler,
- //BaseStatementHandler有三个子类,分别是SimpleStatementHandler,PreparedStatementHandler和CallableStatementHandler,
- //SimpleStatementHandler是用于处理Statement的,PreparedStatementHandler是处理PreparedStatement的,而CallableStatementHandler是
- //处理CallableStatement的。Mybatis在进行Sql语句处理的时候都是建立的RoutingStatementHandler,而在RoutingStatementHandler里面拥有一个
- //StatementHandler类型的delegate属性,RoutingStatementHandler会依据Statement的不同建立对应的BaseStatementHandler,即SimpleStatementHandler、
- //PreparedStatementHandler或CallableStatementHandler,在RoutingStatementHandler里面所有StatementHandler接口方法的实现都是调用的delegate对应的方法。
- //我们在PageInterceptor类上已经用@Signature标记了该Interceptor只拦截StatementHandler接口的prepare方法,又因为Mybatis只有在建立RoutingStatementHandler的时候
- //是通过Interceptor的plugin方法进行包裹的,所以我们这里拦截到的目标对象肯定是RoutingStatementHandler对象。
- if(invocation.getTarget() instanceof RoutingStatementHandler){
- RoutingStatementHandler statementHandler = (RoutingStatementHandler)invocation.getTarget();
- StatementHandler delegate = (StatementHandler) ReflectHelper.getFieldValue(statementHandler, "delegate");
- BoundSql boundSql = delegate.getBoundSql();
- Object obj = boundSql.getParameterObject();
- if (obj instanceof Page<?>) {
- Page<?> page = (Page<?>) obj;
- //通过反射获取delegate父类BaseStatementHandler的mappedStatement属性
- MappedStatement mappedStatement = (MappedStatement)ReflectHelper.getFieldValue(delegate, "mappedStatement");
- //拦截到的prepare方法参数是一个Connection对象
- Connection connection = (Connection)invocation.getArgs()[0];
- //获取当前要执行的Sql语句,也就是我们直接在Mapper映射语句中写的Sql语句
- String sql = boundSql.getSql();
- //给当前的page参数对象设置总记录数
- this.setTotalRecord(page,
- mappedStatement, connection);
- //获取分页Sql语句
- String pageSql = this.getPageSql(page, sql);
- //利用反射设置当前BoundSql对应的sql属性为我们建立好的分页Sql语句
- ReflectHelper.setFieldValue(boundSql, "sql", pageSql);
- }
- }
- return invocation.proceed();
- }
- /**
- * 给当前的参数对象page设置总记录数
- *
- * @param page Mapper映射语句对应的参数对象
- * @param mappedStatement Mapper映射语句
- * @param connection 当前的数据库连接
- */
- private void setTotalRecord(Page<?> page,
- MappedStatement mappedStatement, Connection connection) {
- //获取对应的BoundSql,这个BoundSql其实跟我们利用StatementHandler获取到的BoundSql是同一个对象。
- //delegate里面的boundSql也是通过mappedStatement.getBoundSql(paramObj)方法获取到的。
- BoundSql boundSql = mappedStatement.getBoundSql(page);
- //获取到我们自己写在Mapper映射语句中对应的Sql语句
- String sql = boundSql.getSql();
- //通过查询Sql语句获取到对应的计算总记录数的sql语句
- String countSql = this.getCountSql(sql);
- //通过BoundSql获取对应的参数映射
- List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
- //利用Configuration、查询记录数的Sql语句countSql、参数映射关系parameterMappings和参数对象page建立查询记录数对应的BoundSql对象。
- BoundSql countBoundSql = new BoundSql(mappedStatement.getConfiguration(), countSql, parameterMappings, page);
- //通过mappedStatement、参数对象page和BoundSql对象countBoundSql建立一个用于设定参数的ParameterHandler对象
- ParameterHandler parameterHandler = new DefaultParameterHandler(mappedStatement, page, countBoundSql);
- //通过connection建立一个countSql对应的PreparedStatement对象。
- PreparedStatement pstmt = null;
- ResultSet rs = null;
- try {
- pstmt = connection.prepareStatement(countSql);
- //通过parameterHandler给PreparedStatement对象设置参数
- parameterHandler.setParameters(pstmt);
- //之后就是执行获取总记录数的Sql语句和获取结果了。
- rs = pstmt.executeQuery();
- if (rs.next()) {
- int totalRecord = rs.getInt(1);
- //给当前的参数page对象设置总记录数
- page.setTotalRecord(totalRecord);
- }
- } catch (SQLException e) {
- e.printStackTrace();
- } finally {
- try {
- if (rs != null)
- rs.close();
- if (pstmt != null)
- pstmt.close();
- } catch (SQLException e) {
- e.printStackTrace();
- }
- }
- }
- /**
- * 根据原Sql语句获取对应的查询总记录数的Sql语句
- * @param sql
- * @return
- */
- private String getCountSql(String sql) {
- int index = sql.indexOf("from");
- return "select count(*) " + sql.substring(index);
- }
- /**
- * 根据page对象获取对应的分页查询Sql语句,这里只做了两种数据库类型,Mysql和Oracle
- * 其它的数据库都 没有进行分页
- *
- * @param page 分页对象
- * @param sql 原sql语句
- * @return
- */
- private String getPageSql(Page<?> page, String sql) {
- StringBuffer sqlBuffer = new StringBuffer(sql);
- if ("mysql".equalsIgnoreCase(dialect)) {
- return getMysqlPageSql(page, sqlBuffer);
- } else if ("oracle".equalsIgnoreCase(dialect)) {
- return getOraclePageSql(page, sqlBuffer);
- }
- return sqlBuffer.toString();
- }
- /**
- * 获取Mysql数据库的分页查询语句
- * @param page 分页对象
- * @param sqlBuffer 包含原sql语句的StringBuffer对象
- * @return Mysql数据库分页语句
- */
- private String getMysqlPageSql(Page<?> page, StringBuffer sqlBuffer) {
- //计算第一条记录的位置,Mysql中记录的位置是从0开始的。
- // System.out.println("page:"+page.getPage()+"-------"+page.getRows());
- int offset = (page.getPage() - 1) * page.getRows();
- sqlBuffer.append(" limit ").append(offset).append(",").append(page.getRows());
- return sqlBuffer.toString();
- }
- /**
- * 获取Oracle数据库的分页查询语句
- * @param page 分页对象
- * @param sqlBuffer 包含原sql语句的StringBuffer对象
- * @return Oracle数据库的分页查询语句
- */
- private String getOraclePageSql(Page<?> page, StringBuffer sqlBuffer) {
- //计算第一条记录的位置,Oracle分页是通过rownum进行的,而rownum是从1开始的
- int offset = (page.getPage() - 1) * page.getRows() + 1;
- sqlBuffer.insert(0, "select u.*, rownum r from (").append(") u where rownum < ").append(offset + page.getRows());
- sqlBuffer.insert(0, "select * from (").append(") where r >= ").append(offset);
- //上面的Sql语句拼接之后大概是这个样子:
- //select * from (select u.*, rownum r from (select * from t_user) u where rownum < 31) where r >= 16
- return sqlBuffer.toString();
- }
- /**
- * 拦截器对应的封装原始对象的方法
- */
- public Object plugin(Object arg0) {
- // TODO Auto-generated method stub
- if (arg0 instanceof StatementHandler) {
- return Plugin.wrap(arg0, this);
- } else {
- return arg0;
- }
- }
- /**
- * 设置注册拦截器时设定的属性
- */
- public void setProperties(Properties p) {
- }
- public String getDialect() {
- return dialect;
- }
- public void setDialect(String dialect) {
- this.dialect = dialect;
- }
- public String getPageSqlId() {
- return pageSqlId;
- }
- public void setPageSqlId(String pageSqlId) {
- this.pageSqlId = pageSqlId;
- }
- }
xml配置:
- <!-- MyBatis 接口编程配置 -->
- <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
- <!-- basePackage指定要扫描的包,在此包之下的映射器都会被搜索到,可指定多个包,包与包之间用逗号或分号分隔-->
- <property name="basePackage" value="com.yidao.mybatis.dao" />
- <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
- </bean>
- <!-- MyBatis 分页拦截器-->
- <bean id="paginationInterceptor" class="com.mybatis.interceptor.PageInterceptor">
- <property name="dialect" value="mysql"/>
- <!-- 拦截Mapper.xml文件中,id包含query字符的语句 -->
- <property name="pageSqlId" value=".*query$"/>
- </bean>
在mybatis执行SQL语句之前进行拦击处理的更多相关文章
- springmvc 项目完整示例04 整合mybatis mybatis所需要的jar包 mybatis配置文件 sql语句 mybatis应用
百度百科: MyBatis 本是apache的一个开源项目iBatis, 2010年这个项目由apache software foundation 迁移到了google code,并且改名为MyBat ...
- 10.1(java学习笔记)JDBC基本操作(连接,执行SQL语句,获取结果集)
一.JDBC JDBC的全称是java database connection java数据库连接. 在java中需要对数据库进行一系列的操作,这时就需要使用JDBC. sun公司制定了关于数据库操作 ...
- Java-MyBatis:MyBatis 3 | SQL 语句构建器类
ylbtech-Java-MyBatis:MyBatis 3 | SQL 语句构建器类 1.返回顶部 1. SQL语句构建器类 问题 Java程序员面对的最痛苦的事情之一就是在Java代码中嵌入SQL ...
- 4.5 .net core下直接执行SQL语句并生成DataTable
.net core可以执行SQL语句,但是只能生成强类型的返回结果.例如var blogs = context.Blogs.FromSql("SELECT * FROM dbo.Blogs& ...
- 三种执行SQL语句的的JAVA代码
问题描述: 连接数据库,执行SQL语句是必不可少的,下面给出了三种执行不通SQL语句的方法. 1.简单的Statement执行SQL语句.有SQL注入,一般不使用. public static voi ...
- Entity Framework Code First执行SQL语句、视图及存储过程
1.Entity Framework Code First查询视图 Entity Framework Code First目前还没有特别针对View操作的方法,但对于可更新的视图,可以采用与Table ...
- EFCore执行Sql语句的方法:FromSql与ExecuteSqlCommand
前言 在EFCore中执行Sql语句的方法为:FromSql与ExecuteSqlCommand:在EF6中的为SqlQuery与ExecuteSqlCommand,而FromSql和SqlQuery ...
- sp_executesql得到执行sql语句的返回值
执行 sql语句,得到 变量的值 ' declare @Partition int; ); ); SET @SQLString = N'SELECT @RangeKeyOUT = $PARTITION ...
- EF中执行sql语句,以及事务
EF to sql string sql = "select T_Task.BSID,T_Task.CloseDate,T_Task.CompleteDate,T_Task.CloseUse ...
随机推荐
- 实现Server.UrlEncode和Server.UrlDecode功能的js代码
<script> var EncodeURI = function(unzipStr,isCusEncode){ if(isCusEncode){ var zipArr ...
- 使用Sass优雅并高效的实现CSS中的垂直水平居中(附带Flex布局,CSS3+SASS完美版)
实现css水平垂直居中的方法有很多,在这里我简单的说下四种比较常用的方法: 1.使用CSS3中的Flex布局 对于flex,我们要了解的是它是一个display的属性,而且必须要给他的父元素设置fle ...
- Oracle定义两个变量,并对两个变量的值的长度进行判断
这个例子其实很简单,但是往往简单的东西如果不用心就会漏洞百出,简单的一个逻辑判断,是为了给复杂逻辑判断做出铺垫 语法格式: if<condition_expression> then pl ...
- MYSQL创建多张表,相同表结构,不同表名
#!/bin/bashfor i in {0..63}domysql -u$1 -p$2 -h127.0.0.1 <<EOFuse yoon;create table ivc_pre_de ...
- ios/mac/COCOA系列 -- UIALertVIew 学习笔记
最近在学习ios开发,学习的书籍<ios7 Pragramming cookbook>,做笔记的目的以后方便查看.笔记形式是小例子,将书上的例子书写完整. UIAlertViewClass ...
- Struts之ForwardAction
在Struts中,通过action跳转jsp,但是有时希望仅仅只是跳转页面,而不需要action,这时可以用ForwardAction. 定义一个仅仅是跳转的ForwardAction如下: < ...
- margin-top相对谁的问题
根据规范,一个盒子如果没有上补白(padding-top)和上边框(border-top),那么这个盒子的上边距会和其内部文档流中的第一个子元素的上边距重叠.意思便是:如果你只想margin相对于父标 ...
- Team Homework #3: The feedback of predecessors
此次对学长的采访主要在QQ上进行,感谢陈宇宁学长的热情配合. 采访学长的问题及学长的答复如下: 1. 平均每周花在这门课上的时间 (包括上课/作业/上机) -大约15-20小时吧(学长个人花费时间) ...
- GitHub error “Failed to get HEAD”
cd /要提交的文件的文件夹下 比如要提交一个名为 demo的 程序, 那么先进入demo 的文件夹里面 然后 进行以下两步 git init (有时这个是不必要的,因为xcode 自 ...
- BOM-字节序标记
BOM——Byte Order Mark 字节序标记 首先是什么是字节序? 字节序:与二进制数据在机器存放位置相关的! 可分为两类: 1. 小端字节序: 低地址放低位数据. x86系列的计算机就使用这 ...