com.taotao.cloud.sms.chinamobile.ChinaMobileSendHandler Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2020-2030, Shuigedeng ([email protected] & https://blog.taotaocloud.top/).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.taotao.cloud.sms.chinamobile;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.taotao.cloud.common.utils.lang.StringUtils;
import com.taotao.cloud.common.utils.log.LogUtils;
import com.taotao.cloud.sms.common.exception.SendFailedException;
import com.taotao.cloud.sms.common.handler.AbstractSendHandler;
import com.taotao.cloud.sms.common.model.NoticeData;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Base64;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.util.DigestUtils;
import org.springframework.web.client.RestTemplate;

/**
 * 移动云发送处理
 *
 * @author shuigedeng
 * @version 2022.04
 * @since 2022-04-27 17:50:42
 */
public class ChinaMobileSendHandler extends AbstractSendHandler {

    private static final String BODY_TEMPLATE =
            "{\"ecName\":\"%s\",\"apId\":\"%s\",\"templateId\":\"%s\",\"mobiles\":\"%s\",\"params\":\"%s\",\"sign\":\"%s\",\"addSerial\":\"\",\"mac\":\"%s\"}";

    private final ObjectMapper objectMapper;

    private final RestTemplate restTemplate;

    public ChinaMobileSendHandler(
            ChinaMobileProperties properties,
            ApplicationEventPublisher eventPublisher,
            ObjectMapper objectMapper,
            RestTemplate restTemplate) {
        super(properties, eventPublisher);
        this.objectMapper = objectMapper;
        this.restTemplate = restTemplate;
    }

    /**
     * 构造模板参数
     *
     * @param params 参数列表
     * @return 模板参数
     */
    private static String buildTemplateParas(Collection params) {
        if (params == null || params.isEmpty()) {
            return "[\"\"]";
        }

        boolean firstParam = true;
        StringBuilder builder = new StringBuilder();
        builder.append("[");
        for (String param : params) {
            if (!firstParam) {
                builder.append(",");
            }
            builder.append("\"");
            builder.append(param);
            builder.append("\"");
            firstParam = false;
        }
        builder.append("]");

        return builder.toString();
    }

    private static String buildMac(
            String ecName,
            String apId,
            String secretKey,
            String templateId,
            String mobiles,
            String params,
            String sign) {
        String origin = ecName + apId + secretKey + templateId + mobiles + params + sign;
        return DigestUtils.md5DigestAsHex(origin.getBytes(StandardCharsets.UTF_8));
    }

    @Override
    public boolean send(NoticeData noticeData, Collection phones) {
        String type = noticeData.getType();

        String templateId = properties.getTemplates(type);

        if (templateId == null) {
            LogUtils.debug("templateId invalid");
            publishSendFailEvent(noticeData, phones, new SendFailedException("templateId invalid"), null);
            return false;
        }

        List paramsOrder = properties.getParamsOrder(type);

        ArrayList params = new ArrayList<>();

        if (!paramsOrder.isEmpty()) {
            Map paramMap = noticeData.getParams();
            for (String paramName : paramsOrder) {
                String paramValue = paramMap.get(paramName);

                params.add(paramValue);
            }
        }

        // StringBuilder receiverBuilder = new StringBuilder();
        // for (String phone : phones) {
        //	if (StringUtils.isBlank(phone)) {
        //		continue;
        //	}
        //	receiverBuilder.append(phone.trim());
        //	receiverBuilder.append(",");
        // }
        // String mobiles = receiverBuilder.substring(0, receiverBuilder.length() - 1);

        String mobiles = StringUtils.join(phones, ",");
        String paramsString = buildTemplateParas(params);
        String body = buildRequestBody(mobiles, templateId, paramsString);

        ResponseEntity httpResponse = null;
        try {
            HttpEntity httpEntity = new HttpEntity<>(body, new HttpHeaders());

            httpResponse = restTemplate.exchange(properties.getUri(), HttpMethod.POST, httpEntity, String.class);

            if (httpResponse.getBody() == null) {
                LogUtils.debug("response body ie null");
                publishSendFailEvent(noticeData, phones, new SendFailedException("response body ie null"), null);
                return false;
            }

            String responseContent = httpResponse.getBody();

            LogUtils.debug("responseContent: {}", responseContent);

            ChinaMobileResult result = objectMapper.readValue(responseContent, ChinaMobileResult.class);

            boolean succeed = ChinaMobileResult.SUCCESS_RSPCOD.equals(result.getRspcod());
            if (succeed) {
                publishSendSuccessEvent(noticeData, phones, httpResponse);
            } else {
                publishSendFailEvent(noticeData, phones, new SendFailedException(result.getRspcod()), httpResponse);
            }
            return succeed;
        } catch (Exception e) {
            LogUtils.debug(e.getLocalizedMessage(), e);
            publishSendFailEvent(noticeData, phones, e, httpResponse);
            return false;
        }
    }

    private String buildRequestBody(String mobiles, String templateId, String paramsString) {
        if (com.taotao.cloud.sms.common.utils.StringUtils.isAnyBlank(mobiles, templateId)) {
            throw new SendFailedException("buildRequestBody(): mobiles or templateId is null.");
        }

        String ecName = StringUtils.trimToNull(properties.getEcName());
        String apId = StringUtils.trimToNull(properties.getApId());
        String secretKey = StringUtils.trimToNull(properties.getSecretKey());
        String sign = StringUtils.trimToNull(properties.getSign());
        String mac = buildMac(ecName, apId, secretKey, templateId, mobiles, paramsString, sign);

        String body = String.format(
                BODY_TEMPLATE, ecName, apId, templateId, mobiles, paramsString.replace("\"", "\\\""), sign, mac);

        return new String(Base64.getEncoder().encode(body.getBytes(StandardCharsets.UTF_8)));
    }

    @Override
    public String getChannelName() {
        return "chinaMobile";
    }
}