co.paralleluniverse.fibers.kafka.FiberKafkaProducer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of comsat-kafka Show documentation
Fibers services for web
The newest version!
/*
 * COMSAT
 * Copyright (C) 2013-2016, Parallel Universe Software Co. All rights reserved.
 *
 * This program and the accompanying materials are dual-licensed under
 * either the terms of the Eclipse Public License v1.0 as published by
 * the Eclipse Foundation
 *
 *   or (per the licensee's choosing)
 *
 * under the terms of the GNU Lesser General Public License version 3.0
 * as published by the Free Software Foundation.
 */
package co.paralleluniverse.fibers.kafka;

import co.paralleluniverse.strands.SettableFuture;
import org.apache.kafka.clients.producer.*;
import org.apache.kafka.common.Metric;
import org.apache.kafka.common.MetricName;
import org.apache.kafka.common.PartitionInfo;

import java.util.List;
import java.util.Map;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;

public class FiberKafkaProducer implements Producer {

    private final Producer producer;

    public FiberKafkaProducer(Producer producer) {
        this.producer = producer;
    }

    @Override
    public Future send(ProducerRecord record) {
        return send(record, null);
    }

    @Override
    public Future send(ProducerRecord record, Callback callback) {
        final SettableFuture future = new SettableFuture<>();
        producer.send(record, new CallbackWrapper(future, callback));
        return future;
    }

    @Override
    public void flush() {
        producer.flush();
    }

    @Override
    public List partitionsFor(String topic) {
        return producer.partitionsFor(topic);
    }

    @Override
    public Map metrics() {
        return producer.metrics();
    }

    @Override
    public void close() {
        producer.close();
    }

    @Override
    public void close(long timeout, TimeUnit unit) {
        producer.close(timeout, unit);
    }

    private static class CallbackWrapper implements Callback {

        private final SettableFuture future;
        private final Callback callback;

        public CallbackWrapper(SettableFuture future, Callback callback) {
            this.future = future;
            this.callback = callback;
        }

        @Override
        public void onCompletion(RecordMetadata metadata, Exception exception) {
            if (metadata != null) {
                future.set(metadata);
            } else {
                future.setException(exception);
            }
            if (callback != null) {
                callback.onCompletion(metadata, exception);
            }
        }
    }
}