init

ruoyi-cai/src/main/java/com/ruoyi/cai/kit/AliSmsKit.java

@@ -0,0 +1,144 @@
+package com.ruoyi.cai.kit;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.aliyuncs.CommonRequest;
+import com.aliyuncs.CommonResponse;
+import com.aliyuncs.DefaultAcsClient;
+import com.aliyuncs.IAcsClient;
+import com.aliyuncs.exceptions.ClientException;
+import com.aliyuncs.http.MethodType;
+import com.aliyuncs.profile.DefaultProfile;
+import com.aliyuncs.profile.IClientProfile;
+import com.google.common.collect.Lists;
+import com.ruoyi.cai.config.AliSmsProperties;
+import com.ruoyi.cai.enums.CodeEnum;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.http.util.Asserts;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import org.springframework.util.CollectionUtils;
+
+import javax.annotation.PostConstruct;
+import java.util.Arrays;
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+@Slf4j
+public class AliSmsKit {
+	@Autowired
+	@Setter
+    private AliSmsProperties config;
+    private IClientProfile profile;
+
+    @PostConstruct
+    public void init() {
+		System.setProperty("sun.net.client.defaultConnectTimeout", "10000");
+		System.setProperty("sun.net.client.defaultReadTimeout", "10000");
+		profile = DefaultProfile.getProfile("cn-hangzhou", config.getAccessKeyId(), config.getAccessKeySecret());
+    }
+
+	private boolean batchSendMessage(List<String> phone,String messageTemplate){
+		Asserts.notBlank(messageTemplate, "短信模板不能为空");
+		IAcsClient client = new DefaultAcsClient(profile);
+		CommonRequest request = new CommonRequest();
+		request.setSysMethod(MethodType.POST);
+		request.setSysDomain("dysmsapi.aliyuncs.com");
+		request.setSysVersion("2017-05-25");
+		request.setSysAction("SendBatchSms");
+		List<String> signName = phone.stream().map(i -> config.getSignName()).collect(Collectors.toList());
+		request.putQueryParameter("TemplateCode", messageTemplate);
+		request.putQueryParameter("SignNameJson", JSON.toJSONString(signName));
+		request.putQueryParameter("PhoneNumberJson", JSON.toJSONString(phone));
+		try {
+			CommonResponse response = client.getCommonResponse(request);
+			String data = response.getData();
+			JSONObject jsonObject = JSONObject.parseObject(data);
+			if(jsonObject.get("Code") == null || !"OK".equals(jsonObject.get("Code"))){
+				log.error("短信发送失败 response={}", jsonObject.toJSONString());
+				return false;
+			}
+		} catch (ClientException e) {
+			log.error(e.getMessage(),e);
+			throw new RuntimeException("短信发送失败!");
+		}
+		return true;
+	}
+
+    private boolean sendMessage(String phone, String messageTemplate, String messageCode) {
+        Asserts.notBlank(messageTemplate, "短信模板不能为空");
+        IAcsClient client = new DefaultAcsClient(profile);
+        CommonRequest request = new CommonRequest();
+        request.setSysMethod(MethodType.POST);
+        request.setSysDomain("dysmsapi.aliyuncs.com");
+        request.setSysVersion("2017-05-25");
+        request.setSysAction("SendSms");
+        request.putQueryParameter("SignName", config.getSignName());
+        request.putQueryParameter("RegionId", "cn-hangzhou");
+        request.putQueryParameter("PhoneNumbers", phone);
+        request.putQueryParameter("TemplateCode", messageTemplate);
+        if (StringUtils.isNoneBlank(messageCode)) {
+            request.putQueryParameter("TemplateParam", "{\"code\":\"" + messageCode + "\"}");
+        }
+        CommonResponse response = null;
+        try {
+            response = client.getCommonResponse(request);
+            String data = response.getData();
+            JSONObject jsonObject = JSONObject.parseObject(data);
+            if(jsonObject.get("Code") == null || !"OK".equals(jsonObject.get("Code"))){
+            	log.error("短信发送失败 response={}", jsonObject.toJSONString());
+                return false;
+            }
+        } catch (ClientException e) {
+			log.error(e.getMessage(),e);
+            throw new RuntimeException("短信发送失败!");
+        }
+        return true;
+    }
+
+
+    public boolean sendMessage(String phone, String messageTemplate, String messageCode, boolean isAsy) {
+        if (isAsy) {
+            new Thread(() -> {
+                sendMessage(phone, messageTemplate, messageCode);
+            }).start();
+            return true;
+        } else {
+            return sendMessage(phone, messageTemplate, messageCode);
+        }
+    }
+
+    public boolean batchSendMessagePublic(List<String> phone,String messageTemplate){
+		if(CollectionUtils.isEmpty(phone)){
+			return false;
+		}
+		List<List<String>> list = Lists.partition(phone, 100);
+		for (List<String> phoneList : list) {
+			this.batchSendMessage(phoneList,messageTemplate);
+		}
+		return true;
+	}
+
+    /**
+     * 异步发送消息
+     * @param phone
+     * @param messageTemplate
+     * @return
+     */
+    public boolean sendMessageAsync(String phone, String messageTemplate) {
+        return sendMessage(phone,messageTemplate,null,true);
+    }
+
+	public static void main(String[] args) {
+		AliSmsKit messageSenderUtil = new AliSmsKit();
+		AliSmsProperties config = new AliSmsProperties();
+		messageSenderUtil.setConfig(config);
+		messageSenderUtil.init();
+		messageSenderUtil.batchSendMessagePublic(Arrays.asList("15302786929"), CodeEnum.REGISTER.getAliTemplate());
+
+	}
+
+}