org.reactivecommons.async.kafka.KafkaMessage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of async-kafka Show documentation

Abstract your broker with semantic async messages

There is a newer version: 5.2.2

package org.reactivecommons.async.kafka;

import lombok.Data;
import org.apache.kafka.common.header.Headers;
import org.reactivecommons.async.commons.communications.Message;
import reactor.kafka.receiver.ReceiverRecord;

import java.util.HashMap;
import java.util.Map;

import static org.reactivecommons.async.kafka.converters.json.KafkaJacksonMessageConverter.CONTENT_TYPE;


@Data
public class KafkaMessage implements Message {
    private final byte[] body;
    private final Properties properties;

    @Data
    public static class KafkaMessageProperties implements Properties {
        private long contentLength;
        private String key;
        private String topic;
        private Map headers = new HashMap<>();

        @Override
        public String getContentType() {
            return (String) headers.get(CONTENT_TYPE);
        }
    }

    public static KafkaMessage fromDelivery(ReceiverRecord record) {
        return new KafkaMessage(record.value(), createMessageProps(record));
    }

    private static Properties createMessageProps(ReceiverRecord record) {
        Map headers = parseHeaders(record.headers());

        final KafkaMessageProperties properties = new KafkaMessageProperties();
        properties.setHeaders(headers);
        properties.setKey(record.key());
        properties.setTopic(record.topic());
        properties.setContentLength(record.value().length);
        return properties;
    }

    private static Map parseHeaders(Headers headers) {
        Map parsedHeaders = new HashMap<>();
        headers.forEach(header -> parsedHeaders.put(header.key(), new String(header.value())));
        return parsedHeaders;
    }
}