batch_send_message.go

package alimns

import (
	"context"
	"encoding/xml"
	"errors"
	"fmt"
	"net/http"
	"time"

	"go.uber.org/zap"
)

// SendMessage 发送消息
type SendMessage struct {
	XMLName        xml.Name `xml:"Message"`
	XMLNs          string   `xml:"xmlns,attr"`
	ErrorCode      string   `xml:"ErrorCode"`
	ErrorMessage   string   `xml:"ErrorMessage"`
	MessageID      string   `xml:"MessageId"`
	MessageBodyMD5 string   `xml:"MessageBodyMD5"`
	ReceiptHandle  string   `xml:"ReceiptHandle"` // 发送延时消息才有返回
}

// BatchSendMessageResponse 批量发送消息回复
type BatchSendMessageResponse struct {
	XMLName      xml.Name       `xml:"Messages"`
	XMLNs        string         `xml:"xmlns,attr"`
	SendMessages []*SendMessage `xml:"Message"`
}

// BatchSendMessage 批量发送消息
func (c *Client) BatchSendMessage(name string, messageList ...*Message) (*BatchSendMessageResponse, error) {
	if len(messageList) > 16 {
		return nil, batchSendMessageNumLimitError
	}

	var (
		try            = 0
		sendedMessages = make([]*SendMessage, len(messageList))
		roundIndexList []int
		err            error
	)

	originalMessageList := messageList

	for idx := range originalMessageList {
		roundIndexList = append(roundIndexList, idx)
	}

start:
	requestLine := fmt.Sprintf(mnsBatchSendMessage, name)
	req := c.ca.NewRequest().Post().WithPath(requestLine).WithXMLBody(&messageList).WithTimeout(apiTimeout)

	body, err := req.ReqBody()
	if err != nil {
		return nil, err
	}

	resp, err := c.ca.Do(context.TODO(), req)
	if err != nil {
		return nil, err
	}
	try++

	switch resp.StatusCode() {
	case http.StatusCreated,
		http.StatusInternalServerError:
		var batchSendMessageResponse BatchSendMessageResponse
		err = resp.DecodeFromXML(&batchSendMessageResponse)
		if err != nil {
			return nil, err
		}
		var retryIdx []int
		for seq, sendMessage := range batchSendMessageResponse.SendMessages {
			idx := roundIndexList[seq]

			switch sendMessage.ErrorCode {
			case "":
				sendedMessages[idx] = sendMessage
			case internalError.Error():
				retryIdx = append(retryIdx, idx)
			default:
				c.logger.Warn(fmt.Sprintf("fail to call BatchSendMessage for %s", name), zap.Error(err), zap.String("body", string(body)))
			}
		}
		if len(retryIdx) == 0 {
			batchSendMessageResponse.SendMessages = sendedMessages
			return &batchSendMessageResponse, nil
		}

		roundIndexList = retryIdx

		var retryMessageList []*Message
		for _, idx := range retryIdx {
			retryMessageList = append(retryMessageList, originalMessageList[idx])
		}

		messageList = retryMessageList

		if try > 4 {
			return nil, batchSendMessageTryLimitError
		}

		time.Sleep(100 * time.Millisecond * time.Duration(try))
		goto start

	default:
		var respErr RespErr
		if err := resp.DecodeFromXML(&respErr); err != nil {
			return nil, err
		}
		return nil, errors.New(respErr.Message)
	}
}