init

ruoyi-common/src/main/java/com/ruoyi/common/core/page/TableDataInfo.java

@@ -2,11 +2,13 @@ package com.ruoyi.common.core.page;
 
 import cn.hutool.http.HttpStatus;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.ruoyi.common.utils.BeanConvertUtil;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
 import java.io.Serializable;
 import java.util.List;
+import java.util.function.Supplier;
 
 /**
  * 表格分页数据对象
@@ -59,6 +61,15 @@ public class TableDataInfo<T> implements Serializable {
         return rspData;
     }
 
+    public static <T,S> TableDataInfo<S> build(IPage<T> page, Supplier<S> targetSupplier) {
+        TableDataInfo<S> rspData = new TableDataInfo<>();
+        rspData.setCode(HttpStatus.HTTP_OK);
+        rspData.setMsg("查询成功");
+        rspData.setRows(BeanConvertUtil.convertListTo(page.getRecords(),targetSupplier));
+        rspData.setTotal(page.getTotal());
+        return rspData;
+    }
+
     public static <T> TableDataInfo<T> build(List<T> list) {
         TableDataInfo<T> rspData = new TableDataInfo<>();
         rspData.setCode(HttpStatus.HTTP_OK);