com.taotao.boot.sms.upyun.UpyunSendHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of taotao-boot-starter-sms-upyun Show documentation
taotao-boot-starter-sms-upyun
The newest version!
/*
 * 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.boot.sms.upyun;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.taotao.boot.common.utils.log.LogUtils;
import com.taotao.boot.sms.common.exception.SendFailedException;
import com.taotao.boot.sms.common.handler.AbstractSendHandler;
import com.taotao.boot.sms.common.model.NoticeData;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * 又拍云发送处理
 *
 * @author shuigedeng
 * @version 2022.04
 * @since 2022-04-27 17:52:21
 */
public class UpyunSendHandler extends AbstractSendHandler {

    private static final String API_URL = "https://sms-api.upyun.com/api/messages";

    private final ObjectMapper objectMapper;

    private final RestTemplate restTemplate;

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

    @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;
        }

        ArrayList params = buildParams(noticeData);

        UpyunSendRequest request = new UpyunSendRequest();
        request.setMobile(StringUtils.join(phones, ","));
        request.setTemplateId(templateId);
        request.setVars(StringUtils.join(params, "|"));

        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.add(HttpHeaders.AUTHORIZATION, properties.getToken());

        ResponseEntity httpResponse = null;
        try {
            HttpEntity httpEntity = new HttpEntity<>(objectMapper.writeValueAsString(request), headers);

            httpResponse = restTemplate.exchange(API_URL, 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();

            boolean isJson = responseContent.startsWith("{") && responseContent.endsWith("}");
            boolean sendFail = !responseContent.contains("message_ids");
            if (!isJson || sendFail) {
                LogUtils.debug("send fail: {}", responseContent);
                publishSendFailEvent(noticeData, phones, new SendFailedException(responseContent), httpResponse);
                return false;
            }

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

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

            Collection messageIds = result.getMessageIds();

            if (messageIds == null || messageIds.isEmpty()) {
                publishSendFailEvent(
                        noticeData, phones, new SendFailedException("empty messageIds list"), httpResponse);
                return false;
            }

            boolean succeed = messageIds.stream().filter(Objects::nonNull).anyMatch(MessageId::succeed);

            if (succeed) {
                publishSendSuccessEvent(noticeData, phones, httpResponse);
            } else {
                publishSendFailEvent(noticeData, phones, new SendFailedException("templateId invalid"), httpResponse);
            }

            return succeed;
        } catch (Exception e) {
            LogUtils.error(e.getLocalizedMessage(), e);
            publishSendFailEvent(noticeData, phones, e, httpResponse);
            return false;
        }
    }

    private ArrayList buildParams(NoticeData noticeData) {
        List paramsOrder = properties.getParamsOrder(noticeData.getType());

        ArrayList params = new ArrayList<>();

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

                params.add(paramValue);
            }
        }

        return params;
    }

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