eu.drus.jpa.unit.mongodb.operation.UpdateOperation Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jpa-unit-mongodb Show documentation

JUnit extension for simple testing of JPA entities and components

The newest version!

package eu.drus.jpa.unit.mongodb.operation;

import java.util.List;

import org.bson.Document;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;

public class UpdateOperation implements MongoDbOperation {

    @Override
    public void execute(final MongoDatabase connection, final Document data) {
        for (final String collectionName : data.keySet()) {
            @SuppressWarnings("unchecked")
            final List documents = data.get(collectionName, List.class);

            final MongoCollection collection = connection.getCollection(collectionName);
            documents.forEach(d -> collection.replaceOne(Filters.eq(d.get("_id")), d));
        }
    }

}