计算分页数据

yshop-app/src/main/java/co/yixiang/modules/activity/rest/StoreBargainController.java

@@ -54,6 +54,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 
@@ -97,7 +98,8 @@ public class StoreBargainController {
     @ApiOperation(value = "砍价产品列表",notes = "砍价产品列表")
     public ApiResult<Object> getYxStoreBargainPageList(@RequestParam(value = "page",defaultValue = "1") int page,
                                                        @RequestParam(value = "limit",defaultValue = "10") int limit){
-        return ApiResult.ok(storeBargainService.getList(page, limit));
+
+        return ApiResult.resultPage(Collections.singletonList(storeBargainService.getList(page, limit)),limit);
     }
 
     /**

yshop-common/src/main/java/co/yixiang/api/ApiResult.java

@@ -21,6 +21,7 @@ import org.apache.commons.lang3.StringUtils;
 import java.io.Serializable;
 import java.util.Date;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 
@@ -177,6 +178,52 @@ public class ApiResult<T> implements Serializable {
         return result(ApiCode.FAIL,map);
     }
 
+    public static <T> ApiResult<T> resultPage(List<T> list, int limit){
+        int count = list.size() / limit;
+        if (list.size() == 0) {
+            return (ApiResult<T>) ApiResult.builder()
+                    .total(0)
+                    .totalPage(0)
+                    .status(200)
+                    .msg(null)
+                    .data(list)
+                    .success(true)
+                    .time(new Date())
+                    .build();
+        }
+        if (list.size() <= limit) {
+            return (ApiResult<T>) ApiResult.builder()
+                    .total(list.size())
+                    .totalPage(1)
+                    .status(200)
+                    .msg(null)
+                    .data(list)
+                    .success(true)
+                    .time(new Date())
+                    .build();
+        } else if (count % limit == 0) {
+            return (ApiResult<T>) ApiResult.builder()
+                    .total(list.size())
+                    .totalPage(count)
+                    .status(200)
+                    .msg(null)
+                    .data(list)
+                    .success(true)
+                    .time(new Date())
+                    .build();
+        } else {
+             return (ApiResult<T>) ApiResult.builder()
+                    .total(list.size())
+                    .totalPage(count+1)
+                    .status(200)
+                    .msg(null)
+                    .data(list)
+                    .success(true)
+                    .time(new Date())
+                    .build();
+        }
+    }
+
     public static ApiResult<Boolean> fail() {
         return fail(ApiCode.FAIL);
     }