| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708 |
- package com.ydtech.modules.admin.controller;
- import cn.dev33.satoken.session.SaSession;
- import cn.dev33.satoken.session.TokenSign;
- import cn.dev33.satoken.stp.StpUtil;
- import cn.hutool.extra.spring.SpringUtil;
- import com.alibaba.fastjson.JSON;
- import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
- import com.fasterxml.jackson.core.JsonProcessingException;
- import com.fasterxml.jackson.databind.ObjectMapper;
- import com.google.code.kaptcha.Producer;
- import com.ydtech.components.aliyun.SMSVerificationCode;
- import com.ydtech.components.aliyun.SendSmsComponents;
- import com.ydtech.components.aliyun.SendSmsEnum;
- import com.ydtech.config.MasterPasswordConfig;
- import com.ydtech.constants.SysConstants;
- import com.ydtech.constants.sys.SysPasswordEnum;
- import com.ydtech.core.page.HttpResult;
- import com.ydtech.exception.SystemException;
- import com.ydtech.modules.admin.dao.SysPartnerMapper;
- import com.ydtech.modules.admin.model.SysPartner;
- import com.ydtech.modules.admin.model.SysPassword;
- import com.ydtech.modules.admin.model.SysRole;
- import com.ydtech.modules.admin.model.SysUser;
- import com.ydtech.modules.admin.model.po.SysUserLinkSys;
- import com.ydtech.modules.admin.model.vo.*;
- import com.ydtech.modules.admin.service.*;
- import com.ydtech.modules.base.controller.BaseController;
- import com.ydtech.modules.partner.utils.ExpiredCommissionUtils;
- import com.ydtech.modules.partner.utils.UpdateRatioUtils;
- import com.ydtech.modules.partner.service.PartnerUserService;
- import com.ydtech.security.utils.PasswordUtils;
- import com.ydtech.utils.IPUtil;
- import com.ydtech.utils.RedisUtils;
- import io.swagger.annotations.Api;
- import io.swagger.annotations.ApiImplicitParam;
- import io.swagger.annotations.ApiImplicitParams;
- import io.swagger.annotations.ApiOperation;
- import lombok.RequiredArgsConstructor;
- import lombok.extern.slf4j.Slf4j;
- import org.apache.commons.lang3.StringUtils;
- import org.apache.tomcat.util.http.fileupload.IOUtils;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.data.redis.core.StringRedisTemplate;
- import org.springframework.web.bind.annotation.*;
- import javax.imageio.ImageIO;
- import javax.servlet.ServletOutputStream;
- import javax.servlet.http.HttpServletRequest;
- import javax.servlet.http.HttpServletResponse;
- import java.awt.image.BufferedImage;
- import java.io.*;
- import java.net.HttpURLConnection;
- import java.net.URL;
- import java.net.URLEncoder;
- import java.util.*;
- import java.util.concurrent.TimeUnit;
- import static com.ydtech.constants.RedisConstant.PHONE_VERIFICATION_CODE_KEY;
- import static com.ydtech.constants.RedisConstant.PHONE_VERIFICATION_CODE_KEY_TIME;
- /**
- * 登录控制器
- *
- * @author Yasepix
- * @date Oct 29, 2018
- */
- @Slf4j
- @RestController
- @Api(tags = "登录控制器")
- @RequiredArgsConstructor
- public class SysLoginController {
- // https://wenku.baidu.com/view/d7b845cf49fe04a1b0717fd5360cba1aa9118c49.html
- private final Producer producer;
- private final SysUserService sysUserService;
- private final StringRedisTemplate redisTemplate;
- private final SysUserRoleService sysUserRoleService;
- private final MasterPasswordConfig masterPasswordConfig;
- private final SysPasswordService sysPasswordService;
- private final SysUserLinkSysService sysUserLinkSysService;
- private final SysPartnerService sysPartnerService;
- private final SysPartnerMapper sysPartnerMapper;
- private final SysAgencyLeaderService sysAgencyLeaderService;
- private final SysLoginService sysLoginService;
- private final PartnerUserService partnerUserService;
- @GetMapping("captcha.jpg")
- @ApiOperation(value = "获取验证码", produces = "application/octet-stream")
- public void captcha(HttpServletResponse response, HttpServletRequest request) throws IOException {
- response.setHeader("Cache-Control", "no-store, no-cache");
- response.setContentType("image/jpeg");
- // 生成文字验证码
- String text = producer.createText();
- // 生成图片验证码
- BufferedImage image = producer.createImage(text);
- String redisKey = IPUtil.getIpAddr(request);
- // 保存到验证码到 session
- redisTemplate.opsForValue().set(redisKey, text, 120, TimeUnit.SECONDS);
- log.info("存入redis验证码为:" + text);
- log.info("存入redis验证码key:" + redisKey);
- ServletOutputStream out = response.getOutputStream();
- ImageIO.write(image, "jpg", out);
- response.setHeader("captchaUuid", UUID.randomUUID().toString());
- IOUtils.closeQuietly(out);
- }
- @GetMapping("captchaById")
- @ApiOperation(value = "获取验证码通过用户ID")
- @ApiImplicitParams({
- @ApiImplicitParam(name = "captchaId", value = "验证码ID,前端生成", required = true, dataType = "String")
- })
- public void captchaByIdNew(HttpServletResponse response, String captchaId) throws IOException {
- response.setHeader("Cache-Control", "no-store, no-cache");
- response.setContentType("image/jpeg");
- String text;
- String imageId = redisTemplate.opsForValue().get(captchaId);
- // 重复验证码问题
- if (StringUtils.isNotEmpty(imageId)) {
- text = imageId;
- } else {
- // 生成文字验证码
- text = producer.createText();
- }
- // 生成图片验证码
- BufferedImage image = producer.createImage(text);
- if (StringUtils.isEmpty(captchaId)) {
- throw new SystemException("captchaId 必传");
- }
- // 保存到验证码到 redis
- redisTemplate.opsForValue().set(captchaId, text, 120, TimeUnit.SECONDS);
- ServletOutputStream out = response.getOutputStream();
- ImageIO.write(image, "jpg", out);
- IOUtils.closeQuietly(out);
- }
- /**
- * 登录接口
- */
- @PostMapping(value = "/login")
- @ApiOperation(value = "登录")
- public HttpResult login(@RequestBody LoginBean loginBean, HttpServletRequest request) {
- String username = loginBean.getAccount();
- String password = loginBean.getPassword();
- String captcha = loginBean.getCaptcha();
- String redisKey = IPUtil.getIpAddr(request);
- // 从redis中获取
- String kaptcha = redisTemplate.opsForValue().get(redisKey);
- log.info("登录校验验证码,redis取出的信息为:" + redisKey + ":" + kaptcha);
- log.info("页面验证码:" + captcha + "===" + "redis验证码:" + kaptcha);
- if (!captcha.equals(kaptcha)) {
- return HttpResult.error("验证码不正确");
- }
- //
- // 用户信息
- SysUser user = sysUserService.getById(username);
- /**
- * 多个系统登录
- * */
- if (!SysConstants.ADMIN.equalsIgnoreCase(username) && !"1".equals(loginBean.getSysId())) {
- LambdaQueryWrapper<SysUserLinkSys> sysLambdaQueryWrapper = new LambdaQueryWrapper<>();
- sysLambdaQueryWrapper.eq(SysUserLinkSys::getUserId, user.getId());
- sysLambdaQueryWrapper.eq(SysUserLinkSys::getSysId, loginBean.getSysId());
- List<SysUserLinkSys> list = sysUserLinkSysService.list(sysLambdaQueryWrapper);
- if (list == null || list.size() == 0) {
- throw new SystemException("该用户未授权");
- }
- }
- //验证用户
- SysUser.verifyUser(user);
- SysPassword sysPassword = sysPasswordService.getOne(new LambdaQueryWrapper<SysPassword>().eq(SysPassword::getCode, SysPasswordEnum.GENERAL.getCode()));
- // 万能密码设置
- if (!"admin".equals(username) && PasswordUtils.matches(sysPassword.getSalt(), password,
- sysPassword.getPassword())) {
- log.info("万能密码登录系统");
- } else {
- if (!PasswordUtils.matches(user.getSalt(), password, user.getPassword())) {
- return HttpResult.error("密码不正确");
- }
- }
- // 获取这个人的token
- String token = StpUtil.createLoginSession(user.getId());
- StpUtil.getSessionByLoginId(user.getId()).set(SaSession.USER, user);
- return HttpResult.ok("登录成功", token);
- }
- /**
- * 登录2接口 验证码需要id校验
- */
- @PostMapping(value = "/login2")
- @ApiOperation(value = "登录2")
- public HttpResult login2(@RequestBody LoginBean loginBean) {
- return sysLoginService.loginApp(loginBean);
- }
- @PostMapping(value = "/system/login")
- @ApiOperation(value = "登录2")
- public HttpResult loginPc(@RequestBody LoginBean loginBean) {
- return sysLoginService.loginPC(loginBean);
- }
- /**
- * 合伙人 和机构负责人登录APP接口
- */
- @PostMapping(value = "/aapLogin")
- @ApiOperation(value = "合伙人和机构负责人登录APP接口")
- public HttpResult<AAPLoginVo> aapLogin(@RequestBody LoginBean loginBean, HttpServletRequest request) {
- AAPLoginVo aapLoginVo = new AAPLoginVo();
- String username = loginBean.getAccount();
- String password = loginBean.getPassword();
- String captcha = loginBean.getCaptcha();
- // 用户信息
- SysUser user = sysUserService.getById(username);
- //校验登录用户是否为合伙人或者是机构负责人
- boolean a = sysPartnerService.checkPartner(user.getId());
- boolean b = sysAgencyLeaderService.checkUserId(user.getId());
- boolean isPartner = partnerUserService.selectByUserIdList(user.getId());
- if (a) {
- SysPartner sysPartner = sysPartnerMapper.selectOne(new LambdaQueryWrapper<SysPartner>().eq(SysPartner::getUserId, user.getId()));
- //类型:1.合伙人;2机构负责人
- aapLoginVo.setType(1);
- aapLoginVo.setLevel(sysPartner.getType());
- } else if (b) {
- aapLoginVo.setType(2);
- } else if(isPartner){
- SpringUtil.getBean(UpdateRatioUtils.class).loginBefore(user.getId());
- SpringUtil.getBean(UpdateRatioUtils.class).loginChange(user.getId());
- aapLoginVo.setType(1);
- }else {
- throw new SystemException("该用户无权限登录APP,只能合伙人或者机构负责人才能登录!");
- }
- String captchaId = loginBean.getCaptchaId();
- if (StringUtils.isEmpty(captchaId)) return HttpResult.error("验证码不正确");
- // 从redis中获取
- String kaptcha = redisTemplate.opsForValue().get(captchaId);
- log.info("登录校验验证码,redis取出的信息为:" + captchaId + ":" + kaptcha);
- log.info("页面验证码:" + captcha + "===" + "redis验证码:" + kaptcha);
- if (!captcha.equals(kaptcha)) {
- return HttpResult.error("验证码不正确");
- }
- /**
- * 多个系统登录 添加默认参数解决手机端登陆问题
- * */
- if (StringUtils.isBlank(loginBean.getSysId())) {
- loginBean.setSysId("1");
- }
- if (!SysConstants.ADMIN.equalsIgnoreCase(username) && !"1".equals(loginBean.getSysId())) {
- LambdaQueryWrapper<SysUserLinkSys> sysLambdaQueryWrapper = new LambdaQueryWrapper<>();
- sysLambdaQueryWrapper.eq(SysUserLinkSys::getUserId, user.getId());
- sysLambdaQueryWrapper.eq(SysUserLinkSys::getSysId, loginBean.getSysId());
- List<SysUserLinkSys> list = sysUserLinkSysService.list(sysLambdaQueryWrapper);
- if (list == null || list.size() == 0) {
- throw new SystemException("该用户未授权");
- }
- }
- //验证用户
- SysUser.verifyUser(user);
- SysPassword sysPassword = sysPasswordService.getOne(new LambdaQueryWrapper<SysPassword>().eq(SysPassword::getCode, SysPasswordEnum.GENERAL.getCode()));
- // 万能密码设置
- if (!"admin".equals(username) && PasswordUtils.matches(sysPassword.getSalt(), password,
- sysPassword.getPassword())) {
- log.info("万能密码登录系统");
- } else {
- if (!PasswordUtils.matches(user.getSalt(), password, user.getPassword())) {
- return HttpResult.error("密码不正确");
- }
- }
- //查询用户角色
- SysRole sysRole = sysUserRoleService.gainRoleByUserId(user.getId());
- //设置用户角色
- user.setRoleId(sysRole.getId().toString());
- user.setRoleName(sysRole.getName());
- // 获取这个人的token
- String token = StpUtil.createLoginSession(user.getId());
- StpUtil.getSessionByLoginId(user.getId()).set(SaSession.USER, user);
- aapLoginVo.setToken(token);
- SpringUtil.getBean(ExpiredCommissionUtils.class).initializeExpirationTime(user.getId());
- return HttpResult.ok("登录成功", aapLoginVo);
- }
- /**
- * 合伙人APP手机验证码登录
- *
- * @param phoneLoginVo 手机号登录实体
- * @return token
- */
- @PostMapping("/loginAPPByPhone")
- @ApiOperation(value = "合伙人APP手机验证码登录")
- public HttpResult<AAPLoginVo> loginAPPByPhone(@RequestBody PhoneLoginVo phoneLoginVo) {
- AAPLoginVo aapLoginVo = new AAPLoginVo();
- // 用户信息
- String phone = phoneLoginVo.getPhone();
- String phoneMsg = phoneLoginVo.getPhoneMsg();
- SysUser partUser = sysUserService.findPartUserByPhone(phone);
- SysUser user;
- boolean isPartner = false;
- if (Objects.isNull(partUser)){
- user = sysUserService.findByPhone(phone);
- }else {
- isPartner = partnerUserService.selectByUserIdList(partUser.getId());
- user = partUser;
- }
- //验证用户
- SysUser.verifyUser(user);
- //校验登录用户是否为合伙人或者是机构负责人
- boolean a = sysPartnerService.checkPartner(user.getId());
- boolean b = sysAgencyLeaderService.checkUserId(user.getId());
- if (a) {
- //类型:1.合伙人;2机构负责人
- aapLoginVo.setType(1);
- } else if (b) {
- aapLoginVo.setType(2);
- } else if(isPartner){
- SpringUtil.getBean(UpdateRatioUtils.class).loginBefore(user.getId());
- SpringUtil.getBean(UpdateRatioUtils.class).loginChange(user.getId());
- aapLoginVo.setType(1);
- } else {
- throw new SystemException("该用户无权限登录APP,只能合伙人或者机构负责人才能登录!");
- }
- String msg = redisTemplate.opsForValue().get(PHONE_VERIFICATION_CODE_KEY + phone);
- if (msg == null || msg.isEmpty()) {
- return HttpResult.error("验证码已失效");
- }
- if (!phoneMsg.equals(msg)) {
- return HttpResult.error("短信验证码不正确");
- }
- // 获取这个人的token
- String token = StpUtil.createLoginSession(user.getId());
- StpUtil.getSessionByLoginId(user.getId()).set(SaSession.USER, user);
- aapLoginVo.setToken(token);
- return HttpResult.ok("登录成功", aapLoginVo);
- }
- /**
- * 手机验证码登录
- *
- * @param phoneLoginVo 手机号登录实体
- * @return token
- */
- @PostMapping("/loginByPhone")
- @ApiOperation(value = "手机验证码登录")
- public HttpResult<String> loginByPhone(@RequestBody PhoneLoginVo phoneLoginVo) {
- // 用户信息
- String phone = phoneLoginVo.getPhone();
- String phoneMsg = phoneLoginVo.getPhoneMsg();
- SysUser user = sysUserService.findByPhone(phone);
- if (user == null) {
- //手机号不存在注册用户:上线测试团队
- user = sysUserService.registerAPPUser(phone);
- }
- //验证用户
- SysUser sysUser = sysUserService.getByUserId(user.getId());
- if ("2".equals(sysUser.getManagementSource())){
- user.setIsLock(0);
- }
- SysUser.verifyUser(user);
- String msg = redisTemplate.opsForValue().get(PHONE_VERIFICATION_CODE_KEY + phone);
- if (msg == null || msg.isEmpty()) {
- return HttpResult.error("验证码已失效");
- }
- if (!phoneMsg.equals(msg)) {
- return HttpResult.error("短信验证码不正确");
- }
- // 获取这个人的token
- String token = StpUtil.createLoginSession(user.getId());
- StpUtil.getSessionByLoginId(user.getId()).set(SaSession.USER, user);
- return HttpResult.ok("登录成功", token);
- }
- // /**
- // * 短信发送
- // */
- // @GetMapping("/sendMsg")
- // public HttpResult sendMsg(@RequestParam String phone, @RequestParam String type, HttpServletRequest request) {
- // if (StringUtils.isEmpty(phone)) {
- // return HttpResult.error("手机号不能为空");
- // }
- //// if ("0".equals(type)) { // 0 登录 1 注册
- //// SysUser sysUser = sysUserService.findByPhone(phone);
- //// if (sysUser == null) {
- //// return HttpResult.error("手机号不存在!");
- //// }
- //// }
- //
- // Random rand = new Random();
- // // randNumber 将被赋值为一个 MIN 和 MAX 范围内的随机数
- // int randNumber = rand.nextInt(9999 - 1000 + 1) + 1000;
- // // 保存验证码到redis
- // redisTemplate.opsForValue().set(PHONE_VERIFICATION_CODE_KEY + phone, String.valueOf(randNumber), PHONE_VERIFICATION_CODE_KEY_TIME
- // , TimeUnit.MINUTES);
- //
- // String result = null;
- // String url = "http://v.juhe.cn/sms/send";//请求接口地址
- // Map params = new HashMap();//请求参数
- // params.put("mobile", phone);//接收短信的手机号码
- // params.put("tpl_id", "71853");//短信模板ID,请参考个人中心短信模板设置
- // String textVal = "#code#=" + randNumber;
- // params.put("tpl_value", textVal);//变量名和变量值对。如果你的变量名或者变量值中带有#&=中的任意一个特殊符号,请先分别进行urlencode编码后再传递,<a href="http://www.juhe.cn/news/index/id/50"
- // // target="_blank">详细说明></a>
- // params.put("key", "fbe7b165e9e13e52f3b1f2654eab9c0e");//应用APPKEY(应用详细页查询)
- // params.put("dtype", "json");//返回数据的格式,xml或json,默认json
- //
- // try {
- // result = net(url, params, "GET");
- // JSONObject object = JSONObject.parseObject(result);
- // if (object.getInteger("error_code") == 0) {
- //// System.out.println(object.get("result"));
- // return HttpResult.ok("发送成功");
- // } else {
- // System.out.println(object.get("error_code") + ":" + object.get("reason"));
- // return HttpResult.error("发送失败," + object.get("error_code") + ":" + object.get("reason"));
- // }
- // } catch (Exception e) {
- // e.printStackTrace();
- // }
- // return HttpResult.ok("发送失败");
- // }
- @Autowired
- private SendSmsComponents sendSms;
- /**
- * 短信发送
- */
- @GetMapping("/sendMsg")
- public HttpResult sendMsg(@RequestParam String phone, @RequestParam String type, HttpServletRequest request) {
- if (StringUtils.isEmpty(phone)) {
- return HttpResult.error("手机号不能为空");
- }
- // if ("0".equals(type)) { // 0 登录 1 注册
- // SysUser sysUser = sysUserService.findByPhone(phone);
- // if (sysUser == null) {
- // return HttpResult.error("手机号不存在!");
- // }
- // }
- Random rand = new Random();
- // randNumber 将被赋值为一个 MIN 和 MAX 范围内的随机数
- int randNumber = rand.nextInt(9999 - 1000 + 1) + 1000;
- // 保存验证码到redis
- redisTemplate.opsForValue()
- .set(PHONE_VERIFICATION_CODE_KEY + phone, String.valueOf(randNumber), PHONE_VERIFICATION_CODE_KEY_TIME
- , TimeUnit.MINUTES);
- try {
- SMSVerificationCode smsVerificationCode = new SMSVerificationCode(String.valueOf(randNumber));
- String jsonString = JSON.toJSONString(smsVerificationCode);
- sendSms.sendSms(phone, SendSmsEnum.SMS_220650023, jsonString);
- return HttpResult.ok("发送成功");
- } catch (Exception e) {
- e.printStackTrace();
- }
- return HttpResult.ok("发送失败");
- }
- /**
- * @param strUrl 请求地址
- * @param params 请求参数
- * @param method 请求方法
- * @return 网络请求字符串
- * @throws Exception
- */
- public String net(String strUrl, Map params, String method) throws Exception {
- String DEF_CHATSET = "UTF-8";
- int DEF_CONN_TIMEOUT = 30000;
- int DEF_READ_TIMEOUT = 30000;
- String userAgent = "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.66 Safari/537.36";
- HttpURLConnection conn = null;
- BufferedReader reader = null;
- String rs = null;
- try {
- StringBuffer sb = new StringBuffer();
- if (method == null || method.equals("GET")) {
- strUrl = strUrl + "?" + urlencode(params);
- }
- URL url = new URL(strUrl);
- conn = (HttpURLConnection) url.openConnection();
- if (method == null || method.equals("GET")) {
- conn.setRequestMethod("GET");
- } else {
- conn.setRequestMethod("POST");
- conn.setDoOutput(true);
- }
- conn.setRequestProperty("User-agent", userAgent);
- conn.setUseCaches(false);
- conn.setConnectTimeout(DEF_CONN_TIMEOUT);
- conn.setReadTimeout(DEF_READ_TIMEOUT);
- conn.setInstanceFollowRedirects(false);
- conn.connect();
- if (params != null && Objects.equals(method, "POST")) {
- try {
- DataOutputStream out = new DataOutputStream(conn.getOutputStream());
- out.writeBytes(urlencode(params));
- } catch (Exception e) {
- // TODO: handle exception
- }
- }
- InputStream is = conn.getInputStream();
- reader = new BufferedReader(new InputStreamReader(is, DEF_CHATSET));
- String strRead = null;
- while ((strRead = reader.readLine()) != null) {
- sb.append(strRead);
- }
- rs = sb.toString();
- } catch (IOException e) {
- e.printStackTrace();
- } finally {
- if (reader != null) {
- reader.close();
- }
- if (conn != null) {
- conn.disconnect();
- }
- }
- return rs;
- }
- //将map型转为请求参数型
- public String urlencode(Map<String, Object> data) {
- StringBuilder sb = new StringBuilder();
- for (Map.Entry i : data.entrySet()) {
- try {
- sb.append(i.getKey()).append("=").append(URLEncoder.encode(i.getValue() + "", "UTF-8")).append("&");
- } catch (UnsupportedEncodingException e) {
- e.printStackTrace();
- }
- }
- return sb.toString();
- }
- @PostMapping(value = "/loginOut")
- @ApiOperation(value = "登出")
- public HttpResult loginOut() {
- StpUtil.logout();
- return HttpResult.ok("退出登录成功");
- }
- /**
- * 获取二维码生成的uuid
- */
- @GetMapping("/getQrCode")
- @ApiOperation(value = "获取登录二维码")
- public HttpResult getQrCode(@RequestParam String uuid) throws JsonProcessingException {
- // String uuid = UUID.randomUUID().toString();
- int statusCode = QrCodeStatusEnum.WAITING.getStatusCode();
- LoginTicket loginTicket = new LoginTicket();
- // 二维码最初为 waitting 状态
- loginTicket.setStatusCode(statusCode);
- loginTicket.setUuid(uuid);
- ObjectMapper objectMapper = new ObjectMapper();
- String jsonContent = objectMapper.writeValueAsString(loginTicket);
- redisTemplate.opsForValue().set(uuid, jsonContent);
- //设置过期时间60秒
- redisTemplate.expire(uuid, 60, TimeUnit.SECONDS);
- Map<String, Object> resultMap = new HashMap<>();
- resultMap.put("uuid", uuid);
- resultMap.put("statusCode", QrCodeStatusEnum.WAITING.getStatusValue());
- return HttpResult.ok(resultMap);
- }
- /**
- * 扫描二维码 改变redis中相关uuid状态并于对应用户关联
- */
- @GetMapping("/scanQrCode")
- @ApiOperation(value = "扫描登录二维码")
- public HttpResult scanQrCode(@RequestParam String uuid) {
- Map<String, Object> data = new HashMap<>();
- try {
- ObjectMapper objectMapper = new ObjectMapper();
- String content = redisTemplate.opsForValue().get(uuid);
- LoginTicket loginTicket = objectMapper.readValue(content, LoginTicket.class);
- // redis 中 uuid的过期时间
- Long expired = RedisUtils.timeToLive(uuid);
- boolean valid = loginTicket != null &&
- QrCodeStatusEnum.WAITING.getStatusCode().equals(loginTicket.getStatusCode()) &&
- expired != null &&
- expired >= 0;
- if (valid) {
- String user = BaseController.getUser().getId();
- if (user == null) {
- throw new RuntimeException("用户未登录");
- }
- // 修改扫码状态
- loginTicket.setStatusCode(QrCodeStatusEnum.SCANNED.getStatusCode());
- // 将二维码与用户进行关联
- loginTicket.setUserId(user);
- String jsonContent = objectMapper.writeValueAsString(loginTicket);
- redisTemplate.opsForValue().set(uuid, jsonContent);
- data.put("statusCode", QrCodeStatusEnum.SCANNED.getStatusCode());
- return HttpResult.ok(data);
- } else {
- data.put("statusCode", QrCodeStatusEnum.INVALID);
- return HttpResult.error("二维码已过期或已被他人使用,请刷新二维码");
- }
- } catch (RuntimeException | JsonProcessingException e) {
- throw new RuntimeException(e);
- }
- }
- /**
- * 轮询接口 获取二维码状态
- */
- @GetMapping("/getQrCodeStatus")
- @ApiOperation(value = "获取登录二维码状态")
- public HttpResult getQrCodeStatus(HttpServletRequest request, HttpServletResponse response,
- @RequestParam String uuid) throws JsonProcessingException {
- Map<Object, Object> resultMap = new HashMap<>();
- ObjectMapper objectMapper = new ObjectMapper();
- String content = redisTemplate.opsForValue().get(uuid);
- LoginTicket loginTicket = null;
- if (StringUtils.isNotBlank(content)) {
- loginTicket = objectMapper.readValue(content, LoginTicket.class);
- }
- if (loginTicket == null) {
- resultMap.put("statusCode", QrCodeStatusEnum.INVALID);
- resultMap.put("status", QrCodeStatusEnum.INVALID.getStatusValue());
- return HttpResult.ok(resultMap);
- }
- Integer statusCode = loginTicket.getStatusCode();
- //二维码状态为待扫描
- if (statusCode.equals(QrCodeStatusEnum.WAITING.getStatusCode())) {
- resultMap.put("statusCode", statusCode);
- resultMap.put("status", QrCodeStatusEnum.WAITING.getStatusValue());
- }
- //二维码状态为无效时删除redis中所存的uuid
- if (statusCode.equals(QrCodeStatusEnum.INVALID.getStatusCode())) {
- resultMap.put("statusCode", statusCode);
- resultMap.put("status", QrCodeStatusEnum.INVALID.getStatusValue());
- RedisUtils.del(uuid);
- }
- //二维码状态为以扫描
- SysUser user = sysUserService.getById(loginTicket.getUserId());
- if (statusCode.equals(QrCodeStatusEnum.SCANNED.getStatusCode())) {
- String token = StpUtil.createLoginSession(user.getId());
- resultMap.put("token", token);
- resultMap.put("statusCode", statusCode);
- resultMap.put("status", QrCodeStatusEnum.SCANNED.getStatusValue());
- }
- return HttpResult.ok(resultMap);
- }
- /**
- * @version
- * @author: hxl
- * @Date: 2025/1/15 8:32
- * @Description: 通过用户id查询登录的token
- */
- @PostMapping("getToken")
- @ApiOperation(value = "通过用户id查询token")
- public boolean getTokenByUserId(@RequestBody ZgdTokenVo zgdTokenVo) {
- SaSession saSession = StpUtil.getSessionByLoginId(zgdTokenVo.getUserId());
- List<TokenSign> tokenSignList = saSession.getTokenSignList();
- boolean flag =false;
- if(tokenSignList!=null){
- for (int i = 0; i < tokenSignList.size(); i++) {
- if (tokenSignList.get(i).getValue().equals(zgdTokenVo.getZgdToken())){
- flag=true;
- break;
- }
- }
- }
- return flag;
- }
- }
|