package com.thhy.general.utils;

import com.github.pagehelper.PageHelper;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

/**
 * @Author: zhang_xiao_bo
 * @Date: 2022/3/22 10:41
 * @description: 分页工具类
 */
public class PagingUtils {

    /**
     * 设置分页参数
     */
    public static void setPageParam(){
        String pageNum = "1";
        String pageSize = "10";
        pageNum =  pageNum == null?"1":pageNum;
        pageSize =  pageSize == null?"10":pageSize;
        PageHelper.startPage(Integer.parseInt(pageNum),Integer.parseInt(pageSize));
    }

    public static void setPageParamFromTread(){
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        String pageN = request.getHeader("pageNum");
        String pageS = request.getHeader("pageSize");
        Integer pageNum = StringUtils.isEmpty(pageN)?1:Integer.parseInt(pageN);
        pageNum =  pageNum <1?1:pageNum;
        Integer pageSize =  StringUtils.isEmpty(pageS)?10:Integer.parseInt(pageS);
        PageHelper.startPage(pageNum,pageSize);
    }

    /**
     * 设置分页参数
     */
    public static void setPageParam(HttpServletRequest request){
        String pageN = request.getHeader("pageNum");
        String pageS = request.getHeader("pageSize");
        Integer pageNum = StringUtils.isEmpty(pageN)?1:Integer.parseInt(pageN);
        pageNum =  pageNum <1?1:pageNum;
        Integer pageSize =  StringUtils.isEmpty(pageS)?10:Integer.parseInt(pageS);
        PageHelper.startPage(pageNum,pageSize);
    }

    public static void setPageParam(Integer pageNum,Integer pageSize){
        PageHelper.startPage(pageNum,pageSize);
    }
}