diff --git a/src/main/java/com/example/demo/domain/vo/RoleVo.java b/src/main/java/com/example/demo/domain/vo/RoleVo.java index bbcaca4..fbb111f 100644 --- a/src/main/java/com/example/demo/domain/vo/RoleVo.java +++ b/src/main/java/com/example/demo/domain/vo/RoleVo.java @@ -31,6 +31,7 @@ public class RoleVo implements Serializable { private Integer priority;//优先级 private Integer fatherId;//上级角色id private String fatherName;//上级角色名称 + private String market;//所属市场 @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "Asia/Shanghai") private Date createTime; // 创建时间 @@ -40,4 +41,6 @@ public class RoleVo implements Serializable { List menuIds;//拥有的菜单id List tree;//菜单树 + + private List markets; // 地区列表 } diff --git a/src/main/java/com/example/demo/mapper/RoleMapper.java b/src/main/java/com/example/demo/mapper/RoleMapper.java index 51a5afa..51fb1e0 100644 --- a/src/main/java/com/example/demo/mapper/RoleMapper.java +++ b/src/main/java/com/example/demo/mapper/RoleMapper.java @@ -7,7 +7,6 @@ import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Param; import java.util.List; -import java.util.Random; /** * @program: gold-java @@ -27,8 +26,9 @@ public interface RoleMapper { //查找角色id Role selectByRoleId(Integer id); + //查找全部角色 - List selectAllRole(); + List selectAllRole(List markets); //新建角色 void addRole(RoleVo roleVo); diff --git a/src/main/java/com/example/demo/serviceImpl/RoleServiceImpl.java b/src/main/java/com/example/demo/serviceImpl/RoleServiceImpl.java index 3238626..386bde2 100644 --- a/src/main/java/com/example/demo/serviceImpl/RoleServiceImpl.java +++ b/src/main/java/com/example/demo/serviceImpl/RoleServiceImpl.java @@ -7,6 +7,8 @@ import com.example.demo.domain.vo.Result; import com.example.demo.domain.vo.RoleVo; import com.example.demo.mapper.RefundMapper; import com.example.demo.mapper.RoleMapper; +import com.example.demo.service.AdminService; +import com.example.demo.service.GeneralService; import com.example.demo.service.MenuService; import com.example.demo.service.RoleService; import com.github.pagehelper.PageHelper; @@ -33,6 +35,10 @@ public class RoleServiceImpl implements RoleService { private RoleMapper roleMapper; @Autowired private MenuService menuService; + @Autowired + private AdminService adminService; + @Autowired + private GeneralService generalService; //新增角色 @Override @@ -51,6 +57,9 @@ public class RoleServiceImpl implements RoleService { if (roleVo.getMenuIds() == null || roleVo.getMenuIds().isEmpty()) { return Result.error("权限为空"); } + if(roleVo.getMarket() == null|| roleVo.getMarket().isEmpty()){ + return Result.error("所属市场为空"); + } else { if(roleVo.getFatherId() != 2) { for(Integer menuId : roleVo.getMenuIds()){ @@ -106,7 +115,9 @@ public class RoleServiceImpl implements RoleService { //查找全部角色 @Override public List selectAllRole() { - return roleMapper.selectAllRole(); + String account = adminService.getCurrentAdminAccount(); + List markets = generalService.getRoleMarket(account); + return roleMapper.selectAllRole(markets); } @Override @@ -117,6 +128,9 @@ public class RoleServiceImpl implements RoleService { //分页查找角色 @Override public PageInfo selectBy(Integer pageNum, Integer pageSize, RoleVo roleVo) { + String account = adminService.getCurrentAdminAccount(); + List markets = generalService.getRoleMarket(account); + roleVo.setMarkets(markets); PageHelper.startPage(pageNum, pageSize); List roleVos = roleMapper.selectBy(roleVo); // 遍历 roleVos 列表 diff --git a/src/main/resources/mapper/RoleMapper.xml b/src/main/resources/mapper/RoleMapper.xml index 94f74e0..bf5a640 100644 --- a/src/main/resources/mapper/RoleMapper.xml +++ b/src/main/resources/mapper/RoleMapper.xml @@ -19,6 +19,15 @@