org.springframework.integration.kafka.support.ProducerConfiguration.java Source code

Introduction

Here is the source code for org.springframework.integration.kafka.support.ProducerConfiguration.java

Source

/*
 * Copyright 2002-2013 the original author or authors. 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
 * http://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 org.springframework.integration.kafka.support;

import java.io.*;

import kafka.javaapi.producer.Producer;
import kafka.producer.KeyedMessage;
import kafka.serializer.DefaultEncoder;

import org.apache.commons.lang.builder.EqualsBuilder;
import org.apache.commons.lang.builder.HashCodeBuilder;
import org.springframework.messaging.Message;

/**
 * @author Soby Chacko
 * @author Rajasekar Elango
 * @since 0.5
 */
public class ProducerConfiguration<K, V> {
    private final Producer<K, V> producer;
    private final ProducerMetadata<K, V> producerMetadata;

    public ProducerConfiguration(final ProducerMetadata<K, V> producerMetadata, final Producer<K, V> producer) {
        this.producerMetadata = producerMetadata;
        this.producer = producer;
    }

    public ProducerMetadata<K, V> getProducerMetadata() {
        return producerMetadata;
    }

    public void send(final Message<?> message) throws Exception {
        final V v = getPayload(message);

        String topic = message.getHeaders().get("topic", String.class);
        if (message.getHeaders().containsKey("messageKey")) {
            producer.send(new KeyedMessage<K, V>(topic, getKey(message), v));
        } else {
            producer.send(new KeyedMessage<K, V>(topic, v));
        }
    }

    @SuppressWarnings("unchecked")
    private V getPayload(final Message<?> message) throws Exception {
        if (producerMetadata.getValueEncoder().getClass().isAssignableFrom(DefaultEncoder.class)) {
            return (V) getByteStream(message.getPayload());
        } else if (message.getPayload().getClass().isAssignableFrom(producerMetadata.getValueClassType())) {
            return producerMetadata.getValueClassType().cast(message.getPayload());
        }

        throw new Exception("Message payload type is not matching with what is configured");
    }

    @SuppressWarnings("unchecked")
    private K getKey(final Message<?> message) throws Exception {
        final Object key = message.getHeaders().get("messageKey");

        if (producerMetadata.getKeyEncoder().getClass().isAssignableFrom(DefaultEncoder.class)) {
            return (K) getByteStream(key);
        }

        return message.getHeaders().get("messageKey", producerMetadata.getKeyClassType());
    }

    private static boolean isRawByteArray(final Object obj) {
        return obj instanceof byte[];
    }

    private static byte[] getByteStream(final Object obj) throws IOException {
        if (isRawByteArray(obj)) {
            return (byte[]) obj;
        }

        final ByteArrayOutputStream out = new ByteArrayOutputStream();
        final ObjectOutputStream os = new ObjectOutputStream(out);
        os.writeObject(obj);

        return out.toByteArray();
    }

    @Override
    public boolean equals(final Object obj) {
        return EqualsBuilder.reflectionEquals(this, obj);
    }

    @Override
    public int hashCode() {
        return HashCodeBuilder.reflectionHashCode(this);
    }

    @Override
    public String toString() {
        StringBuilder builder = new StringBuilder();
        builder.append("ProducerConfiguration [producerMetadata=").append(producerMetadata).append("]");
        return builder.toString();
    }
}