diff --git a/src/main/java/cn/celess/blog/controller/LinksController.java b/src/main/java/cn/celess/blog/controller/LinksController.java index f79c588..3c5db4b 100644 --- a/src/main/java/cn/celess/blog/controller/LinksController.java +++ b/src/main/java/cn/celess/blog/controller/LinksController.java @@ -11,8 +11,8 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletRequest; -import java.util.ArrayList; import java.util.List; +import java.util.stream.Collectors; /** * @author : xiaohai @@ -46,14 +46,7 @@ public class LinksController { @GetMapping("/links") public Response allForOpen() { - List sites = new ArrayList<>(); - for (PartnerSite p : partnerSiteService.findAll()) { - if (p.getOpen()) { - //隐藏open字段 - p.setOpen(null); - sites.add(p); - } - } + List sites = partnerSiteService.findAll().stream().peek(partnerSite -> partnerSite.setOpen(null)).collect(Collectors.toList()); return Response.success(sites); }