shuhao 1 день назад
Родитель
Сommit
b415a5e632

+ 14 - 15
src/main/java/com/xiesx/fastboot/base/pagination/PaginationHelper.java

@@ -6,7 +6,6 @@ import java.util.Map;
 import org.springframework.data.domain.Page;
 
 import com.google.common.collect.Lists;
-
 import lombok.NonNull;
 
 /**
@@ -23,8 +22,8 @@ public class PaginationHelper {
      * @param page
      * @return
      */
-    public static PaginationResult create(Page<?> page) {
-        return create(page.toList(), (int) page.getTotalElements());
+    public static <T> PaginationResult<T> create(Page<T> page) {
+        return create(page.getContent(), (int) page.getTotalElements());
     }
 
     /**
@@ -33,8 +32,8 @@ public class PaginationHelper {
      * @param page
      * @return
      */
-    public static PaginationResult create(Page<?> page, Map<String, Object> extData) {
-        PaginationResult paginationResult = create(page.toList(), (int) page.getTotalElements());
+    public static <T> PaginationResult<T> create(Page<T> page, Map<String, Object> extData) {
+        PaginationResult<T> paginationResult = create(page.getContent(), (int) page.getTotalElements());
         paginationResult.extData = extData;
         return paginationResult;
     }
@@ -43,8 +42,8 @@ public class PaginationHelper {
      * 构造空的返回
      * @return
      */
-    public static PaginationResult createEmpty() {
-        return  PaginationResult.builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).build();
+    public static <T> PaginationResult<T> createEmpty() {
+        return PaginationResult.<T>builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).build();
     }
 
     /**
@@ -54,12 +53,12 @@ public class PaginationHelper {
      * @param total
      * @return
      */
-    public static PaginationResult create(@NonNull List<?> data, Integer total) {
-        List<?> list = Lists.newArrayList(data);
+    public static <T> PaginationResult<T> create(@NonNull List<T> data, Integer total) {
+        List<T> list = Lists.newArrayList(data);
         if (list.isEmpty()) {
-            return PaginationResult.builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).build();
+            return PaginationResult.<T>builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).build();
         } else {
-            return PaginationResult.builder().code(0).data(list).count(total).build();
+            return PaginationResult.<T>builder().code(0).data(list).count(total).build();
         }
     }
 
@@ -70,12 +69,12 @@ public class PaginationHelper {
      * @param total
      * @return
      */
-    public static PaginationResult create(@NonNull List<?> data, Integer total, Integer page) {
-        List<?> list = Lists.newArrayList(data);
+    public static <T> PaginationResult<T> create(@NonNull List<T> data, Integer total, Integer page) {
+        List<T> list = Lists.newArrayList(data);
         if (list.isEmpty()) {
-            return PaginationResult.builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).page(1).build();
+            return PaginationResult.<T>builder().code(1).msg("No data").data(Lists.newArrayList()).count(0).page(1).build();
         } else {
-            return PaginationResult.builder().code(0).data(list).count(total).page(page).build();
+            return PaginationResult.<T>builder().code(0).data(list).count(total).page(page).build();
         }
     }
 }

+ 2 - 2
src/main/java/com/xiesx/fastboot/base/pagination/PaginationResult.java

@@ -18,7 +18,7 @@ import lombok.experimental.Accessors;
 @Data
 @Builder
 @Accessors(fluent = true)
-public class PaginationResult {
+public class PaginationResult<T> {
 
     @JSONField(ordinal = 1)
     public Integer code;
@@ -27,7 +27,7 @@ public class PaginationResult {
     public String msg;
 
     @JSONField(ordinal = 3)
-    public List<?> data;
+    public List<T> data;
 
     @JSONField(ordinal = 4)
     public Integer count;

+ 4 - 3
src/main/java/com/xiesx/fastboot/base/result/BaseResult.java

@@ -13,7 +13,8 @@ import lombok.Data;
  */
 @Data
 @Builder
-public class BaseResult {
+public class BaseResult<T,E> {
+
 
     public static String OP_MSG_SUCC = "success";
 
@@ -40,11 +41,11 @@ public class BaseResult {
 
     // 数据
     @JSONField(ordinal = 3)
-    private Object data;
+    private T data;
 
     // 数据
     @JSONField(ordinal = 4)
-    private Object extData;
+    private E extData;
 
     /**
      * 判断是否成功并返回