From 5b0374ca427f97288831cdd24424a06b2aaa7e86 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=92=B1=E6=B6=9B?= <15850646081@163.com>
Date: Thu, 17 Mar 2022 15:47:14 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=86=E9=A1=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/com/engine/salary/util/page/Column.java   | 16 +++++++
 .../engine/salary/util/page/DataSource.java   |  9 ++++
 src/com/engine/salary/util/page/PageInfo.java | 43 ++++++++++++++++++
 src/com/engine/salary/util/page/PageUtil.java | 45 +++++++++++++++++++
 4 files changed, 113 insertions(+)
 create mode 100644 src/com/engine/salary/util/page/Column.java
 create mode 100644 src/com/engine/salary/util/page/DataSource.java
 create mode 100644 src/com/engine/salary/util/page/PageInfo.java
 create mode 100644 src/com/engine/salary/util/page/PageUtil.java

diff --git a/src/com/engine/salary/util/page/Column.java b/src/com/engine/salary/util/page/Column.java
new file mode 100644
index 000000000..71dad2257
--- /dev/null
+++ b/src/com/engine/salary/util/page/Column.java
@@ -0,0 +1,16 @@
+package com.engine.salary.util.page;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class Column {
+    String title;
+    String dataIndex;
+    String key;
+}
diff --git a/src/com/engine/salary/util/page/DataSource.java b/src/com/engine/salary/util/page/DataSource.java
new file mode 100644
index 000000000..379b99023
--- /dev/null
+++ b/src/com/engine/salary/util/page/DataSource.java
@@ -0,0 +1,9 @@
+package com.engine.salary.util.page;
+
+import lombok.Data;
+
+@Data
+public class DataSource {
+    String key;
+    String title;
+}
diff --git a/src/com/engine/salary/util/page/PageInfo.java b/src/com/engine/salary/util/page/PageInfo.java
new file mode 100644
index 000000000..af8d06447
--- /dev/null
+++ b/src/com/engine/salary/util/page/PageInfo.java
@@ -0,0 +1,43 @@
+package com.engine.salary.util.page;
+
+import com.engine.salary.annotation.TableTitle;
+import lombok.Data;
+import lombok.ToString;
+
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.List;
+
+@Data
+@ToString
+public class PageInfo<T> extends com.github.pagehelper.PageInfo<T> {
+    Class<T> clazz;
+    List<Column> columns = new ArrayList<>();
+    List<DataSource> dataSource = new ArrayList<>();
+
+    public PageInfo(List<T> list) {
+        super(list);
+    }
+
+    public PageInfo(List<T> list, Class<T> clazz) {
+        super(list);
+        this.clazz = clazz;
+    }
+
+
+    public List<Column> getColumns() {
+        Field[] fields = clazz.getDeclaredFields();
+        for (Field f : fields) {
+            boolean isanno = f.isAnnotationPresent(TableTitle.class);
+            if (isanno) {
+                TableTitle annotation = f.getAnnotation(TableTitle.class);
+                String title = annotation.title();
+                String dataIndex = annotation.dataIndex();
+                String key = annotation.key();
+                Column column = Column.builder().title(title).dataIndex(dataIndex).key(key).build();
+                columns.add(column);
+            }
+        }
+        return columns;
+    }
+}
diff --git a/src/com/engine/salary/util/page/PageUtil.java b/src/com/engine/salary/util/page/PageUtil.java
new file mode 100644
index 000000000..ab4aa5314
--- /dev/null
+++ b/src/com/engine/salary/util/page/PageUtil.java
@@ -0,0 +1,45 @@
+package com.engine.salary.util.page;
+
+import com.github.pagehelper.Page;
+import com.github.pagehelper.PageHelper;
+import org.apache.commons.collections4.CollectionUtils;
+
+import java.util.Collections;
+import java.util.List;
+
+
+public class PageUtil {
+
+    public static void start(Integer pageNum, Integer pageSize) {
+        pageNum = pageNum == null || pageNum <= 0 ? 1 : pageNum;
+        pageSize = pageSize == null || pageSize <= 0 ? 10 : pageSize;
+        PageHelper.startPage(pageNum, pageSize);
+    }
+
+
+    public static <T> Page<T> buildPage(Integer pageNo, Integer pageSize) {
+        pageNo = pageNo == null || pageNo <= 0 ? 1 : pageNo;
+        pageSize = pageSize == null || pageSize <= 0 ? 10 : pageSize;
+        return new Page(pageNo, pageSize, true);
+    }
+
+    /**
+     * 分页
+     *
+     * @param pageNo   页码（从1开始）
+     * @param pageSize 每页条数
+     * @param source   待分页的数据
+     * @param <T>      范型制定类
+     * @return
+     */
+    public static <T> List<T> subList(int pageNo, int pageSize, List<T> source) {
+        if (CollectionUtils.isEmpty(source)) {
+            return Collections.emptyList();
+        }
+        int endIndex = pageNo * pageSize;
+        int startIndex = (pageNo - 1) * pageSize;
+        startIndex = startIndex < 0 ? 0 : startIndex;
+        return source.subList(startIndex > source.size() ? source.size() : startIndex,
+                endIndex > source.size() ? source.size() : endIndex);
+    }
+}