xtdb.kafka.connect.XtdbSinkTask Maven / Gradle / Ivy

Go to download
package xtdb.kafka.connect;

import org.apache.kafka.clients.consumer.OffsetAndMetadata;
import org.apache.kafka.common.TopicPartition;
import org.apache.kafka.connect.sink.SinkRecord;
import org.apache.kafka.connect.sink.SinkTask;
import clojure.java.api.Clojure;
import clojure.lang.IFn;

import java.io.Closeable;
import java.io.IOException;
import java.util.Collection;
import java.util.Map;


public class XtdbSinkTask extends SinkTask {
    private Map props;
    private Closeable api;
    private static IFn submitSinkRecords;

    static {
        Clojure.var("clojure.core/require").invoke(Clojure.read("xtdb.kafka.connect"));
        submitSinkRecords = Clojure.var("xtdb.kafka.connect/submit-sink-records");
    }

    @Override
    public String version() {
        return new XtdbSinkConnector().version();
    }

    @Override
    public void start(Map props) {
        this.props = props;
        this.api = (Closeable) Clojure.var("xtdb.api/new-api-client").invoke(props.get(XtdbSinkConnector.URL_CONFIG));
    }

    @Override
    public void put(Collection sinkRecords) {
        submitSinkRecords.invoke(api, props, sinkRecords);
    }

    @Override
    public void flush(Map offsets) {
    }

    @Override
    public void stop() {
        if (api != null)
            try {
                api.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
    }
}