huyao
2022-09-29 50e0d4741b19a5071741b200ff76a4d69504fce3
cloud-server-upms/src/main/java/com/dsh/upms/controller/LoginController.java
@@ -4,7 +4,6 @@
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;
@@ -30,10 +29,6 @@
    @Autowired
    private SysUserService sysUserService;
    @Autowired
    private IRegionMemberService regionMemberService;
    @Autowired
    private IStoresService storesService;
    /**
     * 登录
@@ -58,53 +53,6 @@
        }
    }
    /**
     * 获取当前用户信息
     *
     * @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());
        }
    }
    /**
     * 退出登录