From 654dd616eb6d808b1484d56c207f5610ab335a8e Mon Sep 17 00:00:00 2001 From: puzhibing <393733352@qq.com> Date: 星期二, 01 八月 2023 10:29:19 +0800 Subject: [PATCH] 修改bug --- ManagementFBTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TLineController.java | 30 ++++++++++++++++++++++++++++-- 1 files changed, 28 insertions(+), 2 deletions(-) diff --git a/ManagementFBTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TLineController.java b/ManagementFBTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TLineController.java index 9f060a4..e8ca76b 100644 --- a/ManagementFBTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TLineController.java +++ b/ManagementFBTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/specialTrain/TLineController.java @@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.mapper.EntityWrapper; +import com.baomidou.mybatisplus.mapper.Wrapper; import com.baomidou.mybatisplus.plugins.Page; import com.stylefeng.guns.core.base.controller.BaseController; import com.stylefeng.guns.core.common.constant.factory.PageFactory; @@ -26,6 +27,7 @@ import java.util.Date; import java.util.List; import java.util.Map; +import java.util.stream.Collectors; /** * 跨城线路管理控制器 @@ -63,6 +65,14 @@ @Autowired private ITLineCompanyService itLineCompanyService; + @Autowired + private IUserService userService; + + + + + + /** * 跳转到跨城站点管理首页 */ @@ -77,7 +87,15 @@ @RequestMapping("/tLine_add") public String tLineAdd(Model model) { //站点 - List<TSite> siteList = itSiteService.selectList(new EntityWrapper<TSite>().eq("state", 1)); + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + Wrapper<TSite> wrapper = new EntityWrapper<TSite>().eq("state", 1); + if(roleType != 1){ + List<User> users = userService.selectList(new EntityWrapper<User>().ne("roleType", 1).eq("objectId", objectId)); + List<Integer> collect = users.stream().map(User::getId).collect(Collectors.toList()); + wrapper.in("insertUserId", collect); + } + List<TSite> siteList = itSiteService.selectList(wrapper); model.addAttribute("siteList",siteList); //跨城车型 List<TServerCarmodel> carmodelList = itServerCarmodelService.selectList(new EntityWrapper<TServerCarmodel>().eq("type", 2).eq("state", 1)); @@ -116,7 +134,15 @@ TLineSite end = itLineSiteService.selectOne(new EntityWrapper<TLineSite>().eq("lineId", tLineId).eq("type", 2)); model.addAttribute("end",end); //站点 - List<TSite> siteList = itSiteService.selectList(new EntityWrapper<TSite>().eq("state", 1)); + Integer objectId = ShiroKit.getUser().getObjectId(); + Integer roleType = ShiroKit.getUser().getRoleType(); + Wrapper<TSite> wrapper = new EntityWrapper<TSite>().eq("state", 1); + if(roleType != 1){ + List<User> users = userService.selectList(new EntityWrapper<User>().ne("roleType", 1).eq("objectId", objectId)); + List<Integer> collect = users.stream().map(User::getId).collect(Collectors.toList()); + wrapper.in("insertUserId", collect); + } + List<TSite> siteList = itSiteService.selectList(wrapper); model.addAttribute("siteList",siteList); //跨城车型 List<TServerCarmodel> carmodelList = itServerCarmodelService.selectList(new EntityWrapper<TServerCarmodel>().eq("type", 2).eq("state", 1)); -- Gitblit v1.7.1