From 23c59ce9c773022e4354ddac13cd5c722f4efb0f Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Fri, 29 May 2020 15:31:01 +0800 Subject: [PATCH] modify --- src/main/java/com/xcong/excoin/common/system/controller/CommonController.java | 39 ++++++++++++++++++++++++++++++++------- 1 files changed, 32 insertions(+), 7 deletions(-) diff --git a/src/main/java/com/xcong/excoin/common/system/controller/CommonController.java b/src/main/java/com/xcong/excoin/common/system/controller/CommonController.java index db35bd2..8ba477e 100644 --- a/src/main/java/com/xcong/excoin/common/system/controller/CommonController.java +++ b/src/main/java/com/xcong/excoin/common/system/controller/CommonController.java @@ -1,14 +1,19 @@ package com.xcong.excoin.common.system.controller; +import cn.hutool.core.util.StrUtil; import com.xcong.excoin.common.contants.AppContants; import com.xcong.excoin.common.response.Result; +import com.xcong.excoin.common.system.service.CommonService; import com.xcong.excoin.utils.RedisUtils; import com.xcong.excoin.utils.SmsUtils; +import com.xcong.excoin.utils.mail.SubMailSend; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; +import io.swagger.annotations.ApiParam; import lombok.extern.slf4j.Slf4j; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; import javax.annotation.Resource; @@ -32,23 +37,38 @@ @Resource private RedisUtils redisUtils; + @Resource + private CommonService commonservice; + @ApiOperation(value = "获取验证码接口", notes = "获取验证码通用接口") @GetMapping(value = "/verifyCode") - public Result verifyCode(String account, String type) { + public Result verifyCode(@ApiParam(name = "account", value = "手机号或邮箱", required = true) @RequestParam(value = "account") String account, + @ApiParam(name = "type", value = "类型1-手机号2-邮箱", required = true) @RequestParam("type") String type) { log.info("#账号:{}, 类型:{}#", account, type); - int code = (int) ((Math.random() * 9 + 1) * 100000); + Integer code = (int) ((Math.random() * 9 + 1) * 100000); + if (StrUtil.isNotBlank(redisUtils.getString(AppContants.VERIFY_CODE_PREFIX + account))) { + return Result.fail("验证码已发送,请勿重复发"); + } + + // 发送手机验证码 if (AppContants.ACCOUNT_TYPE_MOBILE.equals(type)) { Map<String, Object> result = SmsUtils.sendVerifyCode(account, code); - if (SUCCESS.equals(result.get("resultstatus"))) { Map<String, Object> map = new HashMap<>(); - boolean flag = redisUtils.set("SMS_" + account, code, 120); + boolean flag = redisUtils.set(AppContants.VERIFY_CODE_PREFIX + account, code, 120); map.put("code", flag); return Result.ok("success", map); } + // 发送邮件验证码 } else if (AppContants.ACCOUNT_TYPE_EMAIL.equals(type)) { - return Result.ok("success"); + boolean flag = SubMailSend.sendMail(account, code.toString()); + if (flag) { + redisUtils.set(AppContants.VERIFY_CODE_PREFIX + account, code, 120); + return Result.ok("success"); + } else { + return Result.fail("发送失败"); + } } else { log.info("未定义账号类型"); return Result.fail("fail"); @@ -58,8 +78,13 @@ @ApiOperation(value = "验证验证码是否正确", notes = "验证验证码是否正确") @GetMapping(value = "/checkVerify") - public Result checkVerify() { - return null; + public Result checkVerify(@ApiParam(name = "account", value = "账号", required = true) @RequestParam("account") String account, + @ApiParam(name = "code", value = "验证码", required = true) @RequestParam("code") String code) { + boolean flag = commonservice.verifyCode(account, code); + if (flag) { + return Result.ok("success"); + } + return Result.fail("验证码错误"); } @ApiOperation(value = "文件上次接口", notes = "文件上传") -- Gitblit v1.9.1