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