package com.wgcloud.util;


import com.github.pagehelper.PageInfo;
import org.springframework.ui.Model;

import java.util.ArrayList;
import java.util.List;


/**
 * @version v3.3
 * @ClassName:FormatUtil.java
 * @author: http://www.wgstart.com
 * @date: 2021年1月16日
 * @Description: FormatUtil.java
 * @Copyright: 2019-2021 wgcloud. All rights reserved.
 */
public class PageUtil {

    public static void initPageNumber(PageInfo pageInfo, Model model) {
        List<String> pageNumbers = new ArrayList<String>();
        if (pageInfo.getPages() >= 10 && pageInfo.getPageNum() <= 5) {
            for (int i = 0; i < 10; i++) {
                pageNumbers.add((1 + i) + "");
            }
        }else {
            for (int i = 5; i > 0; i--) {
                if (pageInfo.getPageNum() - i > 0) {
                    pageNumbers.add((pageInfo.getPageNum() - i) + "");
                }
            }
            for (int i = 0; i < 5; i++) {
                if (pageInfo.getPageNum() + i <= pageInfo.getPages()) {
                    pageNumbers.add((pageInfo.getPageNum() + i) + "");
                }
            }
        }
        model.addAttribute("pageNumbers", pageNumbers);
    }

}