package com.dsh.upms.controller;
|
|
import cn.mb.cloud.common.core.exception.BusinessException;
|
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;
|
import io.swagger.annotations.ApiImplicitParam;
|
import io.swagger.annotations.ApiImplicitParams;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 后台登录控制器
|
*/
|
@RestController
|
@RequestMapping("/login")
|
@Api("后台登录控制器")
|
public class LoginController {
|
|
@Autowired
|
private SysUserService sysUserService;
|
@Autowired
|
private IRegionMemberService regionMemberService;
|
@Autowired
|
private IStoresService storesService;
|
|
/**
|
* 登录
|
*
|
* @param username
|
* @param password
|
* @return
|
*/
|
// @SysLog("管理员登录")
|
@PostMapping("/submit")
|
@ApiOperation(value = "登录", tags = {"后台登录控制器"}, notes = "")
|
@ApiImplicitParams({
|
@ApiImplicitParam(value = "账号", name = "username", required = true, dataType = "String"),
|
@ApiImplicitParam(value = "密码", name = "password", required = true, dataType = "String")
|
})
|
public ResponseData submit(String username, String password) {
|
try {
|
return new ResponseData<>(sysUserService.login(username, password).getToken());
|
} catch (BusinessException e) {
|
e.printStackTrace();
|
return ResponseData.fail(e.getMessage());
|
}
|
}
|
|
/**
|
* 获取当前用户信息
|
*
|
* @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());
|
}
|
}
|
|
/**
|
* 退出登录
|
*
|
* @return
|
*/
|
@GetMapping("/logout")
|
@ApiOperation(value = "退出登录", tags = {"后台登录控制器"}, notes = "")
|
@ApiImplicitParams({
|
@ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
|
})
|
public ResponseData Logout() {
|
try {
|
return new ResponseData<>(LoginHelper.Logout());
|
} catch (Exception e) {
|
e.printStackTrace();
|
return ResponseData.fail(e.getMessage());
|
}
|
}
|
}
|
