com.github.wz2cool.dynamic.mybatis.mapper.DynamicQueryMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-dynamic-query Show documentation
There is a newer version: 3.2.32
Show newest version
package com.github.wz2cool.dynamic.mybatis.mapper;

import com.github.wz2cool.dynamic.NormPagingQuery;
import com.github.wz2cool.dynamic.model.NormPagingResult;
import org.apache.ibatis.session.RowBounds;
import tk.mybatis.mapper.annotation.RegisterMapper;
import tk.mybatis.mapper.common.BaseMapper;
import tk.mybatis.mapper.common.special.InsertListMapper;

import java.util.List;


/**
 * @author Frank
 */
@RegisterMapper
public interface DynamicQueryMapper extends
        BaseMapper,
        InsertListMapper,
        SelectCountByDynamicQueryMapper,
        DeleteByDynamicQueryMapper,
        SelectByDynamicQueryMapper,
        SelectRowBoundsByDynamicQueryMapper,
        UpdateSelectiveByDynamicQueryMapper,
        UpdateByDynamicQueryMapper,
        SelectMaxByDynamicQueryMapper,
        SelectMinByDynamicQueryMapper,
        SelectSumByDynamicQueryMapper,
        SelectAvgByDynamicQueryMapper,
        UpdateByUpdateQueryMapper {


    /**
     * select by norm paging
     *
     * @param normPagingQuery norm paging query
     * @return norm paging result
     */
    default NormPagingResult selectByNormalPaging(NormPagingQuery normPagingQuery) {
        NormPagingResult result = new NormPagingResult<>();
        int pageNum = normPagingQuery.getPageNum() < 1 ? 1 : normPagingQuery.getPageNum();
        int pageSize = normPagingQuery.getPageSize();
        int queryPageSize = pageSize + 1;
        int offset = (pageNum - 1) * pageSize;
        List dataList = selectRowBoundsByDynamicQuery(normPagingQuery.getDynamicQuery(), new RowBounds(offset, queryPageSize));
        // 补偿当前页没有需要到上一页
        if (normPagingQuery.isAutoBackIfEmpty() && dataList.isEmpty() && pageNum > 1) {
            int newPageNum = pageNum - 1;
            NormPagingQuery newNormalPagingQuery = NormPagingQuery.createQuery(normPagingQuery.getEntityClass(),
                    newPageNum, normPagingQuery.getPageSize(),
                    normPagingQuery.isAutoBackIfEmpty(), normPagingQuery.isCalcTotal());
            newNormalPagingQuery.setDistinct(normPagingQuery.isDistinct());
            newNormalPagingQuery.setFilters(normPagingQuery.getFilters());
            newNormalPagingQuery.setSorts(normPagingQuery.getSorts());
            newNormalPagingQuery.setSelectedProperties(normPagingQuery.getSelectedProperties());
            newNormalPagingQuery.setIgnoredProperties(normPagingQuery.getIgnoredProperties());
            return selectByNormalPaging(newNormalPagingQuery);
        }
        if (normPagingQuery.isCalcTotal()) {
            int totalCount = selectCountByDynamicQuery(normPagingQuery.getDynamicQuery());
            int pages = (int) Math.ceil((double) totalCount / pageSize);
            result.setTotal(totalCount);
            result.setPages(pages);
        }
        boolean hasNext = dataList.size() > pageSize;
        boolean hasPre = pageNum > 1;
        result.setHasNextPage(hasNext);
        result.setHasPreviousPage(hasPre);
        if (dataList.size() > pageSize) {
            result.setList(dataList.subList(0, pageSize));
        } else {
            result.setList(dataList);
        }
        result.setPageNum(pageNum);
        result.setPageSize(normPagingQuery.getPageSize());
        return result;
    }
}