| | |
| | | import cn.mb.cloud.common.core.util.ResponseData; |
| | | import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
| | | import com.dsh.upms.model.vo.LoginUserVo; |
| | | import com.dsh.upms.model.vo.region.RegionVo; |
| | | import com.dsh.upms.service.SysUserService; |
| | | import com.dsh.utils.login.LoginHelper; |
| | | import io.swagger.annotations.Api; |
| | |
| | | |
| | | @Autowired |
| | | private SysUserService sysUserService; |
| | | @Autowired |
| | | private IRegionMemberService regionMemberService; |
| | | @Autowired |
| | | private IStoresService storesService; |
| | | |
| | | /** |
| | | * 登录 |
| | |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 获取当前用户信息 |
| | | * |
| | | * @return |
| | | */ |
| | | //@SysLog("获取当前用户信息") |
| | | @GetMapping("/getUser") |
| | | @ApiOperation(value = "获取当前用户信息", tags = {"后台登录控制器"}, notes = "", response = LoginUserVo.class) |
| | | @ApiImplicitParams({ |
| | | @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....") |
| | | }) |
| | | public ResponseData getUser() { |
| | | try { |
| | | LoginUserVo user = LoginHelper.getUser(); |
| | | List<RegionMember> regionMemberList = regionMemberService |
| | | .list(Wrappers.<RegionMember>query().lambda() |
| | | .eq(RegionMember::getMemberId, user.getId()) |
| | | .eq(RegionMember::getRegionType, 2) |
| | | .eq(RegionMember::getState, 0) |
| | | .eq(RegionMember::getDelFlag, 0)); |
| | | List<Long> regionIdList = new ArrayList<>(); |
| | | for (RegionMember regionMember : regionMemberList) { |
| | | if (!regionIdList.contains(regionMember.getRegionId())) { |
| | | regionIdList.add(regionMember.getRegionId()); |
| | | } |
| | | } |
| | | List<RegionVo> regionVoList = new ArrayList<>(); |
| | | if (!regionIdList.isEmpty()) { |
| | | List<Stores> storesList = storesService |
| | | .list(Wrappers.<Stores>query().lambda() |
| | | .in(Stores::getId, regionIdList) |
| | | .eq(Stores::getDelFlag, 0) |
| | | .orderByAsc(Stores::getCreateDate)); |
| | | for (Stores stores : storesList) { |
| | | RegionVo regionVo = new RegionVo(); |
| | | regionVo.setRegionId(stores.getId()); |
| | | regionVo.setRegionName(stores.getStoresName()); |
| | | regionVoList.add(regionVo); |
| | | } |
| | | } |
| | | user.setStoresId(regionVoList); |
| | | return new ResponseData<>(user); |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | return ResponseData.fail(e.getMessage()); |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 退出登录 |