package com.aukey.example.converter;

import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.support.converter.MessageConversionException;
import org.springframework.amqp.support.converter.MessageConverter;

import java.io.UnsupportedEncodingException;

/**
 * @author: wgf
 * @create: 2020-06-10 22:39
 * @description: 消息类型转换器
 * 如果 spring boot 用的是 2.2.7.RELEASE以上的版本则不需要使用此转换器
 **/
public class CharArrayToStringConverter implements MessageConverter {

    private String defaultCharset = "utf-8";

    @Override
    public Message toMessage(Object object, MessageProperties messageProperties) throws MessageConversionException {
        // TODO 不发消息不实现转换
        return null;
    }

    @Override
    public Object fromMessage(Message message) throws MessageConversionException {
        Object content = null;
        MessageProperties properties = message.getMessageProperties();
        if (properties != null) {
            String contentType = properties.getContentType();
            if (StringUtils.isBlank(contentType) || contentType.startsWith("text")) {
                String encoding = properties.getContentEncoding();
                if (encoding == null) {
                    encoding = this.defaultCharset;
                }
                try {
                    content = new String(message.getBody(), encoding);
                } catch (UnsupportedEncodingException e) {
                    throw new MessageConversionException(
                            "failed to convert text-based Message content", e);
                }
            }
        } else {
            content = message.getBody();
        }
        return content;
    }
}