eu.drus.jpa.unit.mongodb.operation.IndexOptionsUtils 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.concurrent.TimeUnit;

import org.bson.Document;
import org.bson.conversions.Bson;

import com.mongodb.client.model.Collation;
import com.mongodb.client.model.CollationAlternate;
import com.mongodb.client.model.CollationCaseFirst;
import com.mongodb.client.model.CollationMaxVariable;
import com.mongodb.client.model.CollationStrength;
import com.mongodb.client.model.IndexOptions;

public class IndexOptionsUtils {

    private IndexOptionsUtils() {}

    public static IndexOptions toIndexOptions(final Document options) {
        final IndexOptions indexOptions = new IndexOptions();

        applyIfTrue(options.containsKey("background"), () -> indexOptions.background(options.getBoolean("background")));
        applyIfTrue(options.containsKey("bits"), () -> indexOptions.bits(options.getInteger("bits")));
        applyIfTrue(options.containsKey("bucketSize"), () -> indexOptions.bucketSize(options.getDouble("bucketSize")));
        applyIfTrue(options.containsKey("collation"), () -> {
            final Document collationData = options.get("collation", Document.class);
            final Collation.Builder builder = Collation.builder();
            applyIfTrue(collationData.containsKey("backwards"), () -> builder.backwards(collationData.getBoolean("backwards")));
            applyIfTrue(collationData.containsKey("caseLevel"), () -> builder.caseLevel(collationData.getBoolean("caseLevel")));
            applyIfTrue(collationData.containsKey("alternate"),
                    () -> builder.collationAlternate(CollationAlternate.fromString(collationData.getString("alternate"))));
            applyIfTrue(collationData.containsKey("caseFirst"),
                    () -> builder.collationCaseFirst(CollationCaseFirst.fromString(collationData.getString("caseFirst"))));
            applyIfTrue(collationData.containsKey("maxVariable"),
                    () -> builder.collationMaxVariable(CollationMaxVariable.fromString(collationData.getString("maxVariable"))));
            applyIfTrue(collationData.containsKey("strength"),
                    () -> builder.collationStrength(CollationStrength.fromInt(collationData.getInteger("strength"))));
            applyIfTrue(collationData.containsKey("locale"), () -> builder.locale(collationData.getString("locale")));
            applyIfTrue(collationData.containsKey("normalization"), () -> builder.normalization(collationData.getBoolean("normalization")));
            applyIfTrue(collationData.containsKey("numericOrdering"),
                    () -> builder.numericOrdering(collationData.getBoolean("numericOrdering")));
            indexOptions.collation(builder.build());
        });
        applyIfTrue(options.containsKey("default_language"), () -> indexOptions.defaultLanguage(options.getString("default_language")));
        applyIfTrue(options.containsKey("expireAfterSeconds"),
                () -> indexOptions.expireAfter(options.getLong("expireAfterSeconds"), TimeUnit.SECONDS));
        applyIfTrue(options.containsKey("language_override"), () -> indexOptions.languageOverride(options.getString("language_override")));
        applyIfTrue(options.containsKey("max"), () -> indexOptions.max(options.getDouble("max")));
        applyIfTrue(options.containsKey("min"), () -> indexOptions.min(options.getDouble("min")));
        applyIfTrue(options.containsKey("name"), () -> indexOptions.name(options.getString("name")));
        applyIfTrue(options.containsKey("partialFilterExpression"),
                () -> indexOptions.partialFilterExpression(options.get("partialFilterExpression", Bson.class)));
        applyIfTrue(options.containsKey("sparse"), () -> indexOptions.sparse(options.getBoolean("sparse")));
        applyIfTrue(options.containsKey("sphereVersion"), () -> indexOptions.sphereVersion(options.getInteger("sphereVersion")));
        applyIfTrue(options.containsKey("storageEngine"), () -> indexOptions.storageEngine(options.get("storageEngine", Bson.class)));
        applyIfTrue(options.containsKey("textVersion"), () -> indexOptions.textVersion(options.getInteger("textVersion")));
        applyIfTrue(options.containsKey("unique"), () -> indexOptions.unique(options.getBoolean("unique")));
        applyIfTrue(options.containsKey("version"), () -> indexOptions.version(options.getInteger("version")));
        applyIfTrue(options.containsKey("weights"), () -> indexOptions.weights(options.get("weights", Bson.class)));

        return indexOptions;
    }

    private static void applyIfTrue(final boolean shouldApply, final Runnable run) {
        if (shouldApply) {
            run.run();
        }
    }
}