From c9ebfbd68abd74157aaec690e4e21703ceeb9358 Mon Sep 17 00:00:00 2001
From: "udf_admin@163.com" <udf_admin@163.com>
Date: Wed, 31 Jan 2018 23:39:23 +0800
Subject: [PATCH] =?UTF-8?q?=E9=98=BF=E9=87=8C=E4=BA=91=E7=9F=AD=E4=BF=A1?=
 =?UTF-8?q?=E5=B9=B3=E5=8F=B0=E9=9B=86=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../udf/aliyun/dysms/api/DysmsWarpper.java    |  4 ++
 .../dysms/web/AliuyunDysmsController.java     | 49 ++++++++++++++-----
 2 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/udf-starter-aliyun-dysms-api/src/main/java/org/itkk/udf/aliyun/dysms/api/DysmsWarpper.java b/udf-starter-aliyun-dysms-api/src/main/java/org/itkk/udf/aliyun/dysms/api/DysmsWarpper.java
index 5f39c04..fa2fc4e 100644
--- a/udf-starter-aliyun-dysms-api/src/main/java/org/itkk/udf/aliyun/dysms/api/DysmsWarpper.java
+++ b/udf-starter-aliyun-dysms-api/src/main/java/org/itkk/udf/aliyun/dysms/api/DysmsWarpper.java
@@ -8,6 +8,7 @@ import com.aliyuncs.exceptions.ClientException;
 import com.aliyuncs.http.MethodType;
 import com.aliyuncs.profile.DefaultProfile;
 import com.aliyuncs.profile.IClientProfile;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
@@ -17,6 +18,7 @@ import java.util.UUID;
  * DysmsWarpper
  */
 @Component
+@Slf4j
 public class DysmsWarpper {
 
     /**
@@ -73,6 +75,8 @@ public class DysmsWarpper {
         if (sendSmsResponse.getCode() != null && sendSmsResponse.getCode().equals("OK")) {
             //请求成功
             result = true;
+        } else {
+            log.warn("send dysms failure ---> requestId : {} , bizId : {} , code : {} , message : {}", sendSmsResponse.getRequestId(), sendSmsResponse.getBizId(), sendSmsResponse.getCode(), sendSmsResponse.getMessage());
         }
         //返回
         return result;
diff --git a/udf-starter-aliyun-dysms/src/main/java/org/itkk/udf/aliyun/dysms/web/AliuyunDysmsController.java b/udf-starter-aliyun-dysms/src/main/java/org/itkk/udf/aliyun/dysms/web/AliuyunDysmsController.java
index bfa21f7..494e7dc 100644
--- a/udf-starter-aliyun-dysms/src/main/java/org/itkk/udf/aliyun/dysms/web/AliuyunDysmsController.java
+++ b/udf-starter-aliyun-dysms/src/main/java/org/itkk/udf/aliyun/dysms/web/AliuyunDysmsController.java
@@ -5,6 +5,7 @@ import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import org.apache.commons.lang3.StringUtils;
 import org.itkk.udf.aliyun.dysms.api.DysmsWarpper;
+import org.itkk.udf.cache.redis.lock.RedisBasedDistributedLock;
 import org.itkk.udf.core.RestResponse;
 import org.itkk.udf.core.exception.ParameterValidException;
 import org.itkk.udf.core.utils.IdUtil;
@@ -29,6 +30,12 @@ public class AliuyunDysmsController implements IAliuyunDysmsController {
      */
     private static final String VERIFICATION_CODE_CACHE_KEY_PREFIX = ":VERIFICATION_CODE:";
 
+    /**
+     * profiles
+     */
+    @Value("${spring.profiles.active}")
+    private String profiles;
+
     /**
      * redisTemplate
      */
@@ -36,10 +43,10 @@ public class AliuyunDysmsController implements IAliuyunDysmsController {
     private RedisTemplate<Object, Object> redisTemplate;
 
     /**
-     * profiles
+     * redisBasedDistributedLock
      */
-    @Value("${spring.profiles.active}")
-    private String profiles;
+    @Autowired
+    private RedisBasedDistributedLock redisBasedDistributedLock;
 
     /**
      * dysmsWarpper
@@ -66,22 +73,37 @@ public class AliuyunDysmsController implements IAliuyunDysmsController {
         final String templateCode = "SMS_123798312";
         final int num4 = 4;
         final int num5 = 5;
+        final int num50 = 50;
+        //定义返回值
+        boolean result = false;
         //获得手机号
         String phoneNumber = httpServletRequest.getHeader("phoneNumber");
         //判空
         if (StringUtils.isBlank(phoneNumber)) {
             throw new ParameterValidException("手机号不能为空", null);
         }
-        //获得验证码
-        String code = IdUtil.genRandomNumber(num4);
-        //放入缓存
-        String key = getKey(phoneNumber);
-        redisTemplate.opsForValue().set(key, code, num5, TimeUnit.MINUTES);
-        //构造参数
-        Map<String, String> param = new HashMap<>();
-        param.put("code", code);
-        //发送 & 返回
-        return new RestResponse<>(dysmsWarpper.send(phoneNumber, signName, templateCode, objectMapper.writeValueAsString(param)));
+        //锁定 (同一个手机号50秒只能发送成功一次验证码短消息)
+        String lockKey = "VERIFICATION_CODE:" + phoneNumber;
+        if (redisBasedDistributedLock.lock(lockKey, num50, TimeUnit.SECONDS, null)) {
+            //获得验证码
+            String code = IdUtil.genRandomNumber(num4);
+            //构造参数
+            Map<String, String> param = new HashMap<>();
+            param.put("code", code);
+            //发送
+            result = dysmsWarpper.send(phoneNumber, signName, templateCode, objectMapper.writeValueAsString(param));
+            //判断是否成功,成功则缓存code,失败则解锁
+            if (result) {
+                //放入缓存
+                String key = getKey(phoneNumber);
+                redisTemplate.opsForValue().set(key, code, num5, TimeUnit.MINUTES);
+            } else {
+                //解锁
+                redisBasedDistributedLock.unlock(lockKey);
+            }
+        }
+        //返回
+        return new RestResponse<>(result);
     }
 
     @Override
@@ -120,4 +142,5 @@ public class AliuyunDysmsController implements IAliuyunDysmsController {
     private String getKey(String phoneNumber) {
         return profiles + VERIFICATION_CODE_CACHE_KEY_PREFIX + phoneNumber;
     }
+
 }