All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • de.adorsys.multibanking.conf.ContinueOnBatchErrorTemplate Maven / Gradle / Ivy

    package de.adorsys.multibanking.conf;

import com.mongodb.WriteConcern;
import com.mongodb.client.model.InsertManyOptions;
import org.bson.Document;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoAction;
import org.springframework.data.mongodb.core.MongoActionOperation;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.util.StreamUtils;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * Created by alexg on 01.12.17.
 */
public class ContinueOnBatchErrorTemplate extends MongoTemplate {

    private static List continueOnErrorCollections = Arrays.asList("bookingEntity", "anonymizedBookingEntity");
    private static final String ID_FIELD = "_id";

    public ContinueOnBatchErrorTemplate(MongoDbFactory mongoDbFactory) {
        super(mongoDbFactory);
    }

    protected List insertDocumentList(final String collectionName, final List documents) {

        if (documents.isEmpty()) {
            return Collections.emptyList();
        }

        execute(collectionName, collection -> {

            MongoAction mongoAction = new MongoAction(null, MongoActionOperation.INSERT_LIST, collectionName, null,
                    null, null);
            WriteConcern writeConcernToUse = prepareWriteConcern(mongoAction);
            //TODO remove this bullshit when springframework data supports InsertOptions
            InsertManyOptions insertOptions = (new InsertManyOptions()).ordered(!continueOnErrorCollections.contains(collectionName));

            if (writeConcernToUse == null) {
                collection.insertMany(documents, insertOptions);
            } else {
                collection.withWriteConcern(writeConcernToUse).insertMany(documents, insertOptions);
            }

            return null;
        });

        return documents.stream()//
                .map(it -> it.get(ID_FIELD))//
                .collect(StreamUtils.toUnmodifiableList());
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy