package org.sxkj.common.query;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.function.Consumer;
import java.util.function.Function;

/**
 * 分页处理工具类
 */
public class PaginationUtils {

	/**
	 * 通用分页处理方法
	 * @param mapper MyBatis-Plus Mapper接口
	 * @param pageSize 每页大小
	 * @param queryWrapperBuilder 查询条件构建函数
	 * @param pageDataProcessor 每页数据处理函数
	 * @param <T> 实体类型
	 * @param <M> Mapper类型
	 */
	public static <T, M extends BaseMapper<T>> void processInPages(
		M mapper,
		int pageSize,
		Function<Integer, Wrapper<T>> queryWrapperBuilder,
		Consumer<Page<T>> pageDataProcessor) {

		int currentPage = 1;
		boolean hasNext = true;

		while (hasNext) {
			// 创建分页对象
			Page<T> page = new Page<>(currentPage, pageSize);

			// 构建当前页的查询条件
			Wrapper<T> wrapper = queryWrapperBuilder.apply(currentPage);

			// 执行分页查询
			Page<T> resultPage = mapper.selectPage(page, wrapper);
			List<T> records = resultPage.getRecords();

			// 处理当前页数据
			if (!CollectionUtils.isEmpty(records)) {
				pageDataProcessor.accept(resultPage);
			}

			// 判断是否还有下一页
			hasNext = currentPage < resultPage.getPages() && !CollectionUtils.isEmpty(records);
			currentPage++;
		}
	}

	/**
	 * 简化版分页处理（不需要当前页码构建查询条件）
	 * @param mapper MyBatis-Plus Mapper接口
	 * @param pageSize 每页大小
	 * @param queryWrapper 固定查询条件
	 * @param pageDataProcessor 每页数据处理函数
	 * @param <T> 实体类型
	 * @param <M> Mapper类型
	 */
	public static <T, M extends BaseMapper<T>> void processInPages(
		M mapper,
		int pageSize,
		Wrapper<T> queryWrapper,
		Consumer<Page<T>> pageDataProcessor) {

		processInPages(mapper, pageSize, page -> queryWrapper, pageDataProcessor);
	}
}