diff --git a/src/main/java/cn/axzo/pokonyan/dao/converter/PageConverter.java b/src/main/java/cn/axzo/pokonyan/dao/converter/PageConverter.java index 3770871..72dd9c4 100644 --- a/src/main/java/cn/axzo/pokonyan/dao/converter/PageConverter.java +++ b/src/main/java/cn/axzo/pokonyan/dao/converter/PageConverter.java @@ -17,6 +17,8 @@ import java.util.Optional; import java.util.function.Function; import java.util.stream.Collectors; +import static java.util.stream.Collectors.toList; + @UtilityClass public class PageConverter { @@ -88,6 +90,14 @@ public class PageConverter { return totalData; } + public static Page convert(Page page, Function mapper) { + List collect = page.getRecords().stream().map(mapper).collect(toList()); + + Page result = new Page<>(page.getCurrent(), page.getSize(), page.getTotal()); + result.setRecords(collect); + return result; + } + /** * 将MybatisPlus的IPage转换为spring的Page, 用于返回 *