artoria.message.rocketmq.RocketMqMessage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artoria-extend Show documentation
Artoria is a java technology framework based on the facade pattern.
The newest version!
package artoria.message.rocketmq;

import artoria.data.json.JsonUtils;
import artoria.exception.ExceptionUtils;
import artoria.io.util.IOUtils;
import artoria.message.mq.MqMessage;
import artoria.util.Assert;
import artoria.util.ClassUtils;
import artoria.util.CloseUtils;

import java.io.InputStream;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.Collection;

import static artoria.common.constant.Charsets.UTF_8;

public class RocketMqMessage implements MqMessage {

    public static RocketMqMessage of(String topic, Object body) {

        return new RocketMqMessage().setTopic(topic).setBody(body);
    }

    public static RocketMqMessage of(String topic) {

        return new RocketMqMessage().setTopic(topic);
    }

    public static RocketMqMessage of() {

        return new RocketMqMessage();
    }

    private Collection keys;
    private Integer flag;
    private String topic;
    private String tags;
    private byte[] body;

    public Collection getKeys() {

        return keys;
    }

    public RocketMqMessage setKeys(Collection keys) {
        this.keys = keys;
        return this;
    }

    public Integer getFlag() {

        return flag;
    }

    public RocketMqMessage setFlag(Integer flag) {
        this.flag = flag;
        return this;
    }

    public String getTopic() {

        return topic;
    }

    public RocketMqMessage setTopic(String topic) {
        this.topic = topic;
        return this;
    }

    public String getTags() {

        return tags;
    }

    public RocketMqMessage setTags(String tags) {
        this.tags = tags;
        return this;
    }

    public byte[] getBody() {

        return body;
    }

    public RocketMqMessage setBody(byte[] body) {
        this.body = body;
        return this;
    }

    public RocketMqMessage setBody(Object body) {

        return setBody(body, UTF_8);
    }

    public RocketMqMessage setBody(Object body, Charset charset) {
        Assert.notNull(charset, "Parameter \"charset\" must not null. ");
        Assert.notNull(body, "Parameter \"body\" must not null. ");
        if (body instanceof byte[]) { return setBody((byte[]) body); }
        if (ClassUtils.isSimpleValueType(body.getClass())) {
            String valueOf = String.valueOf(body);
            return setBody(valueOf.getBytes(charset));
        }
        if (body instanceof InputStream) {
            try {
                return setBody(IOUtils.toByteArray((InputStream) body));
            }
            catch (Exception e) { throw ExceptionUtils.wrap(e); }
            finally { CloseUtils.closeQuietly((InputStream) body); }
        }
        if (body instanceof Reader) {
            try {
                return setBody(IOUtils.toByteArray((Reader) body, charset.name()));
            }
            catch (Exception e) { throw ExceptionUtils.wrap(e); }
            finally { CloseUtils.closeQuietly((Reader) body); }
        }
        //
        String jsonString = JsonUtils.toJsonString(body);
        return setBody(jsonString.getBytes(charset));
    }

    @Override
    public String getDestination() {

        return topic;
    }

    @Override
    public byte[] getPayload() {

        return body;
    }

}