org.apache.ibatis.session.RowBounds#DEFAULT

Source File: PaginationInterceptor.java From joyqueue with Apache License 2.0

6 votes

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = statementHandler.getBoundSql();
    if (log.isDebugEnabled()) {
        log.debug("raw SQL : " + boundSql.getSql());
    }
    if (boundSql.getSql() == null || boundSql.getSql().isEmpty() || boundSql.getSql().contains(" limit ")) {
        return invocation.proceed();
    }
    MetaObject metaStatementHandler = null;
    RowBounds rowBounds = (RowBounds) metaStatementHandler.getValue("delegate.rowBounds");
    if (rowBounds == null || rowBounds == RowBounds.DEFAULT) {
        return invocation.proceed();
    }
    String originalSql = (String) metaStatementHandler.getValue("delegate.boundSql.sql");
    metaStatementHandler.setValue("delegate.boundSql.sql",
            getLimitString(originalSql, rowBounds.getOffset(), rowBounds.getLimit()));
    metaStatementHandler.setValue("delegate.rowBounds.offset", RowBounds.NO_ROW_OFFSET);
    metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT);
    if (log.isDebugEnabled()) {
        log.debug("pagination SQL : " + boundSql.getSql());
    }
    return invocation.proceed();
}

Source File: MySqlPagingPlugin.java From mybatis-test with Apache License 2.0

6 votes

@Override
public Object intercept(Invocation invocation) throws Throwable {
    Object[] args = invocation.getArgs();
    RowBounds rb = (RowBounds) args[ROW_BOUNDS_INDEX];
    if (rb == RowBounds.DEFAULT) {
        return invocation.proceed();
    }
    args[ROW_BOUNDS_INDEX] = RowBounds.DEFAULT;

    MappedStatement ms = (MappedStatement) args[MAPPED_STATEMENT_INDEX];
    BoundSql boundSql = ms.getBoundSql(args[PARAMETER_INDEX]);

    System.out.println();
    String sql = boundSql.getSql();
    String limit = String.format("LIMIT %d,%d", rb.getOffset(), rb.getLimit());
    sql = sql + " " + limit;

    SqlSource sqlSource = new StaticSqlSource(ms.getConfiguration(), sql, boundSql.getParameterMappings());

    Field field = MappedStatement.class.getDeclaredField("sqlSource");
    field.setAccessible(true);
    field.set(ms, sqlSource);

    return invocation.proceed();
}

Source File: PageInterceptor.java From QuickProject with Apache License 2.0

6 votes

@Override
public Object intercept(Invocation invocation) throws Throwable {
	StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
	BoundSql boundSql = statementHandler.getBoundSql();
	MetaObject metaStatementHandler = MetaObject.forObject(statementHandler, new DefaultObjectFactory(), new DefaultObjectWrapperFactory());
	RowBounds rowBounds = (RowBounds) metaStatementHandler.getValue("delegate.rowBounds");
	if ((rowBounds != null) && (rowBounds != RowBounds.DEFAULT)) {
           Configuration configuration = (Configuration) metaStatementHandler.getValue("delegate.configuration");
           Dialect dialect = DialectParser.parse(configuration);
           String sql = (String) metaStatementHandler.getValue("delegate.boundSql.sql");
           sql = dialect.addLimitString(sql, rowBounds.getOffset(), rowBounds.getLimit());

           metaStatementHandler.setValue("delegate.boundSql.sql", sql);
           metaStatementHandler.setValue("delegate.rowBounds.offset", RowBounds.NO_ROW_OFFSET);
           metaStatementHandler.setValue("delegate.rowBounds.limit", RowBounds.NO_ROW_LIMIT);
	}

	log.debug("SQL : " + boundSql.getSql());
	return invocation.proceed();
}

Source File: MapperInstrument.java From mybatis-boost with MIT License

5 votes

private static void addRowBoundsParameter(CtMethod ctMethod, RowBounds rowBounds) {
    if (rowBounds == RowBounds.DEFAULT) return;
    try {
        String newMethodName = ctMethod.getName() + "$" +
                UUID.randomUUID().toString().replace("-", "");
        CtMethod ctNewMethod = CtNewMethod.copy(ctMethod, newMethodName, ctMethod.getDeclaringClass(), null);
        ctNewMethod.addParameter(ClassPool.getDefault().get("org.apache.ibatis.session.RowBounds"));
        ctNewMethod.getMethodInfo().addAttribute
                (ctMethod.getMethodInfo().removeAttribute(AnnotationsAttribute.visibleTag));

        String body;
        if (rowBounds.getLimit() == 1) {
            MethodInfo methodInfo = ctNewMethod.getMethodInfo();
            String descriptor = methodInfo.getDescriptor();
            String returnType = descriptor.substring(descriptor.lastIndexOf(')') + 1);
            methodInfo.setDescriptor(descriptor =
                    descriptor.substring(0, descriptor.length() - returnType.length()) + "Ljava/util/List;");
            ctNewMethod.setGenericSignature
                    (descriptor.substring(0, descriptor.length() - 1) + "<" + returnType + ">;");

            body = "{ java.util.List list = %s($$, new org.apache.ibatis.session.RowBounds(%s, %s));" +
                    "return !list.isEmpty() ? (%s) list.get(0) : null; }";
            body = String.format(body, newMethodName, rowBounds.getOffset(), rowBounds.getLimit(),
                    returnType.substring(1, returnType.length() - 1).replace('/', '.'));
        } else {
            String signature = ctMethod.getGenericSignature();
            int index = signature.lastIndexOf(')');
            ctNewMethod.setGenericSignature(signature.substring(0, index) +
                    "Lorg/apache/ibatis/session/RowBounds;" + signature.substring(index));
            body = String.format("{ return %s($$, new org.apache.ibatis.session.RowBounds(%s, %s)); }",
                    newMethodName, rowBounds.getOffset(), rowBounds.getLimit());
        }
        ctMethod.getDeclaringClass().addMethod(ctNewMethod);
        ctMethod.setBody(body);
    } catch (CannotCompileException | NotFoundException e) {
        throw new RuntimeException(e);
    }
}

Source File: PaginationInterceptor.java From Mario with Apache License 2.0

5 votes

@Override
public Object intercept(Invocation invocation) throws Throwable {

    StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = statementHandler.getBoundSql();

    String sql = boundSql.getSql();
    if (StringUtils.isBlank(sql)) {
        return invocation.proceed();
    }

    //select sql do 
    if (sql.matches(SQL_SELECT_REGEX) && !Pattern.matches(SQL_COUNT_REGEX, sql)) {
        Object obj = FieldUtils.readField(statementHandler, "delegate", true);
        // 反射获取 RowBounds 对象。
        RowBounds rowBounds = (RowBounds) FieldUtils.readField(obj, "rowBounds", true);

        // 分页参数存在且不为默认值时进行分页SQL构造
        if (rowBounds != null && rowBounds != RowBounds.DEFAULT) {
            FieldUtils.writeField(boundSql, "sql", newSql(sql, rowBounds), true);

            // 一定要还原否则将无法得到下一组数据(第一次的数据被缓存了)
            FieldUtils.writeField(rowBounds, "offset", RowBounds.NO_ROW_OFFSET, true);
            FieldUtils.writeField(rowBounds, "limit", RowBounds.NO_ROW_LIMIT, true);
        }
    }

    return invocation.proceed();
}

Source File: MybatisUtils.java From sqlhelper with GNU Lesser General Public License v3.0

4 votes

public static boolean isPagingRowBounds(RowBounds rowBounds) {
    if (rowBounds == null || rowBounds == RowBounds.DEFAULT) {
        return false;
    }
    return rowBounds.getOffset() != RowBounds.NO_ROW_OFFSET || rowBounds.getLimit() != RowBounds.NO_ROW_LIMIT;
}

Source File: MethodNameParser.java From mybatis-boost with MIT License

4 votes

public RowBounds toRowBounds() {
    if (parsedSql == null) toSql();
    return offset > 0 || limit > 0 ? new RowBounds(offset, limit) : RowBounds.DEFAULT;
}

Source File: MybatisOrmImpl.java From tephra with MIT License

4 votes

private RowBounds getRowBounds(MybatisBuilder builder) {
    return builder.getRowBounds() == null ? RowBounds.DEFAULT : builder.getRowBounds();
}

Source File: AbstractRowBoundsDialect.java From Mybatis-PageHelper with MIT License

4 votes

@Override
public boolean skip(MappedStatement ms, Object parameterObject, RowBounds rowBounds) {
    return rowBounds == RowBounds.DEFAULT;
}

Source File: PaginationInterceptor.java From platform with Apache License 2.0

4 votes

@Override
@SuppressWarnings("unchecked")
public Object intercept(Invocation invocation) throws Throwable {
    Executor executor = MyBatisUtils.getRealTarget(invocation.getTarget());
    Object[] args = invocation.getArgs();
    MappedStatement mappedStatement = (MappedStatement) args[MAPPED_STATEMENT_INDEX];

    // 查询操作或者存储过程，无需分页
    if (SqlCommandType.SELECT != mappedStatement.getSqlCommandType()
            || StatementType.CALLABLE == mappedStatement.getStatementType()) {
        return invocation.proceed();
    }

    Object parameterObject = invocation.getArgs()[PARAMETER_INDEX];

    // 不包含PageableRequest或者Pageable参数时，无需分页
    PageableRequest<?> pageableRequest = MyBatisUtils.findPageableRequest(parameterObject).orElse(null);
    if (null == pageableRequest || pageableRequest.getPageable() == null ||
            pageableRequest.getPageable().getPageSize() < 0) {
        return invocation.proceed();
    }

    Connection connection = executor.getTransaction().getConnection();

    DbType dbType = this.dbType == null ? JdbcUtils.getDbType(connection) : this.dbType;
    DbDialect dialect = Optional.ofNullable(this.dbDialect).orElseGet(() -> JdbcUtils.getDialect(dbType));

    // 针对定义了rowBounds，做为mapper接口方法的参数
    BoundSql boundSql = mappedStatement.getBoundSql(parameterObject);
    String originalSql = boundSql.getSql();

    // 查询总记录数
    long total = 0;
    if (pageableRequest.isQueryTotalCount()) {
        String countSql = dialect.buildCountSql(boundSql.getSql());

        total = this.queryTotal(countSql, mappedStatement, boundSql, connection);

        if (total <= 0) {
            return null;
        }
    }

    Pageable pageable = pageableRequest.getPageable();
    String buildSql = concatOrderBy(originalSql, pageable);
    String paginationSql = dialect.buildPaginationSql(buildSql, pageable.getOffset(), pageable.getPageSize());

    BoundSql newBs = MyBatisUtils.copyFromBoundSql(mappedStatement, boundSql, paginationSql, parameterObject);
    MappedStatement newMs = MyBatisUtils.copyFromMappedStatement(mappedStatement, new BoundSqlSqlSource(newBs));
    args[MAPPED_STATEMENT_INDEX] = newMs;
    args[ROWBOUNDS_INDEX] = RowBounds.DEFAULT;

    Object result = invocation.proceed();
    pageableRequest.setTotal(total);
    pageableRequest.setRecords((List) result);
    return result;
}

Java Code Examples for org.apache.ibatis.session.RowBounds#DEFAULT