at.grahsl.kafka.connect.mongodb.writemodel.strategy.DeleteOneDefaultStrategy Maven / Gradle / Ivy

Go to download

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

A Kafka connect MongoDB sink connector.

The newest version!

package at.grahsl.kafka.connect.mongodb.writemodel.strategy;

import at.grahsl.kafka.connect.mongodb.converter.SinkDocument;
import at.grahsl.kafka.connect.mongodb.processor.id.strategy.IdStrategy;
import com.mongodb.DBCollection;
import com.mongodb.client.model.DeleteOneModel;
import com.mongodb.client.model.WriteModel;
import org.apache.kafka.connect.errors.DataException;
import org.bson.BsonDocument;
import org.bson.BsonValue;

public class DeleteOneDefaultStrategy implements WriteModelStrategy {

    private IdStrategy idStrategy;

    @Deprecated
    public DeleteOneDefaultStrategy() {}

    public DeleteOneDefaultStrategy(IdStrategy idStrategy) {
        this.idStrategy = idStrategy;
    }

    @Override
    public WriteModel createWriteModel(SinkDocument document) {

        BsonDocument kd = document.getKeyDoc().orElseThrow(
                () -> new DataException("error: cannot build the WriteModel since"
                        + " the key document was missing unexpectedly")
        );

        //NOTE: fallback for backwards / deprecation compatibility
        if(idStrategy == null) {
            return kd.containsKey(DBCollection.ID_FIELD_NAME)
                    ? new DeleteOneModel<>(kd)
                    : new DeleteOneModel<>(new BsonDocument(DBCollection.ID_FIELD_NAME,kd));
        }

        //NOTE: current design doesn't allow to access original SinkRecord (= null)
        BsonValue _id = idStrategy.generateId(document,null);
        return new DeleteOneModel<>(
                new BsonDocument(DBCollection.ID_FIELD_NAME,_id)
        );

    }
}