diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/LawsNewsFeed.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/LawsNewsFeed.java index 653af6b2..afb4208d 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/LawsNewsFeed.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/LawsNewsFeed.java @@ -106,4 +106,8 @@ public class LawsNewsFeed implements Serializable { @ApiModelProperty(value = "推送范围") @TableField(exist = false) private String pushRange; + + @ApiModelProperty(value = "是否可以编辑和删除") + @TableField(exist = false) + private Boolean isEditAndRemove; } diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/LawsNewsFeedServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/LawsNewsFeedServiceImpl.java index 9efb93cf..ac8dde88 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/LawsNewsFeedServiceImpl.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/LawsNewsFeedServiceImpl.java @@ -53,20 +53,14 @@ public class LawsNewsFeedServiceImpl extends ServiceImpl queryPage(LawsNewsFeed lawsNewsFeed, Integer pageNo, Integer pageSize) { Page page = new Page<>(pageNo, pageSize); - LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); - lawsNewsFeed.setWriter(loginUser.getId()); - if (loginUser.getRoleIds().contains(FieldCommon.ROLE_ADMIN)) { - lawsNewsFeed.setRoleCode(FieldCommon.ROLE_ADMIN); - } else { - lawsNewsFeed.setRoleCode("other"); - } + setQueryParam(lawsNewsFeed); IPage pageList = lawsNewsFeedMapper.getPageList(page, lawsNewsFeed); + setResultData(pageList.getRecords()); return pageList; } @@ -74,11 +68,22 @@ public class LawsNewsFeedServiceImpl extends ServiceImpl queryList(LawsNewsFeed lawsNewsFeed) { + setQueryParam(lawsNewsFeed); + List pageList = lawsNewsFeedMapper.getPageList(lawsNewsFeed); + setResultData(pageList); + return pageList; + } + + /** + * @Author: liao + * @Date: 2023/9/20 17:41 + * @Description: 设置查询条件 + **/ + private void setQueryParam(LawsNewsFeed lawsNewsFeed) { LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); lawsNewsFeed.setWriter(loginUser.getId()); if (loginUser.getRoleIds().contains(FieldCommon.ROLE_ADMIN)) { @@ -86,9 +91,26 @@ public class LawsNewsFeedServiceImpl extends ServiceImpl pageList = lawsNewsFeedMapper.getPageList(lawsNewsFeed); - return pageList; - } + } + + /** + * @Author: liao + * @Date: 2023/9/20 17:42 + * @Description: 设置返回结果 + **/ + private void setResultData(List pageList) { + LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal(); + pageList.forEach(lawsNewsFeed -> { + if (loginUser.getRoleIds().contains(FieldCommon.ROLE_ADMIN) + || loginUser.getId().equals(lawsNewsFeed.getWriter())) { + // 可以编辑和删除 + lawsNewsFeed.setIsEditAndRemove(true); + } else { + // 不能编辑和删除 + lawsNewsFeed.setIsEditAndRemove(false); + } + }); + } /** * 保存