分页工具 显示当前页，前几页，后几页

有次面试，说让写一段代码，输入当前页码，显示当前页码前几页，后几页。当时时间紧张，逻辑是写出来了，可是边界有问题，现在闲来无事，就写了一个。下面的代码显示的是当前页码的前两页，当前页，后两页。

package test;

import java.util.Scanner;

public class TestPageSize {

	/**
	 * 根据当前页码，显示页码工具栏，显示当前页码的前两页、当前页和后两页
	 * @param curPage   当前页码
	 * @param totalPage  总条数
	 * @return
	 */
	public String getPageSize(int curPage,int totalPage){
		
		String pages = "";
		if(totalPage>5){//总条数大于5
			pages = "1 2 3 4 5";
			if(1==curPage||2==curPage||3==curPage){//当前页码1，2，3都直接显示1-5
				return pages;
			}else{//当前页码>=4
				if(curPage>(totalPage-2)){//当 当前页码是最后两页，不再计算，就显示 总条数-4 - 总条数
					pages = (totalPage-4)+" "+(totalPage-3)+" "+(totalPage-2)+" "+(totalPage-1)+" "+totalPage;
				}else{//当  4=<当前页码<总条数-2  计算显示 当前页码的前两页、当前页和后两页
					pages = (curPage-2)+" "+(curPage-1)+" "+curPage+" "+(curPage+1)+" "+(curPage+2);
				}
				return pages;
			}		
		}else {//总条数小于5，直接显示1-总条数
			for(int i=1;i<totalPage+1;i++){
				pages += i+" ";
			}
			return pages;
		}
	}
	
	
	public static void main(String[] args) {
		System.out.println("请输入总页码：");
		Scanner in = new Scanner(System.in);
		String total = in.nextLine();
		while(true){
			System.out.println("请输入当前页码：");
			String curPage = in.nextLine();
			TestPageSize testO = new TestPageSize();
			String pages = testO.getPageSize(Integer.parseInt(curPage),Integer.parseInt(total));
			System.out.println(pages);
		}
	}
}