com.jn.sqlhelper.mango.MangoCommonPagingInterceptor Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sqlhelper-mango Show documentation
Supports mango (an ORM framework)
There is a newer version: 3.6.16
package com.jn.sqlhelper.mango;

import com.jn.sqlhelper.dialect.pagination.RowSelection;
import org.jfaster.mango.binding.BoundSql;
import org.jfaster.mango.interceptor.Parameter;
import org.jfaster.mango.interceptor.QueryInterceptor;
import org.jfaster.mango.mapper.SingleColumnRowMapper;
import org.jfaster.mango.plugin.page.Page;
import org.jfaster.mango.plugin.page.PageException;
import org.jfaster.mango.transaction.DataSourceUtils;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.util.List;

public class MangoCommonPagingInterceptor extends QueryInterceptor {
    @Override
    public void interceptQuery(BoundSql boundSql, List parameters, DataSource dataSource) {
        Connection conn = DataSourceUtils.getConnection(dataSource);
        DatabaseMetaData databaseMetaData = null;
        try {
            databaseMetaData = conn.getMetaData();
        } catch (Throwable ex) {
            return;
        }
        if (!MangoPagingContext.instrumentor.beginIfSupportsLimit(databaseMetaData)) {
            return;
        }
        Page page = findPageRequest(parameters);
        if (page == null) {
            return;
        }
        // 参数检测
        int pageNum = page.getPageNum();
        int pageSize = page.getPageSize();
        if (pageNum < 0) {
            throw new PageException("pageNum need >= 0, but pageNum is " + pageNum);
        }
        if (pageSize < 0) {
            throw new PageException("pageSize need >= 0, but pageSize is " + pageSize);
        }

        if (page.isFetchTotal()) {
            String countSql = MangoPagingContext.instrumentor.countSql(boundSql.getSql());
            BoundSql countBoundSql = boundSql.copy();
            countBoundSql.setSql(countSql);
            SingleColumnRowMapper mapper = new SingleColumnRowMapper(int.class);
            int total = getJdbcOperations().queryForObject(dataSource, countBoundSql, mapper);
            page.setTotal(total);
        }

        long offset = pageNum <= 0 ? 0L : (pageNum - 1) * pageSize;
        RowSelection rowSelection = new RowSelection();
        rowSelection.setLimit(pageSize);
        rowSelection.setOffset(offset);
        boundSql.setSql(MangoPagingContext.instrumentor.instrumentLimitSql(boundSql.getSql(), rowSelection));
        MangoPagingContext.pagingRequest.set(rowSelection);
    }

    private Page findPageRequest(List parameters) {
        for (Parameter parameter : parameters) {
            if (Page.class.equals(parameter.getRawType())) {
                Object val = parameter.getValue();
                if (val == null) {
                    throw new PageException("Parameter page is null");
                }
                return (Page) val;
            }
        }
        return null;
    }
}