com.yahoo.elide.spring.datastore.config.DataStoreBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elide-spring-boot-autoconfigure Show documentation

Elide Spring Boot Autoconfigure

The newest version!

/*
 * Copyright 2023, the original author or authors.
 * Licensed under the Apache License, Version 2.0
 * See LICENSE file in project root for terms.
 */
package com.yahoo.elide.spring.datastore.config;

import com.yahoo.elide.core.datastore.DataStore;
import com.yahoo.elide.datastores.multiplex.MultiplexManager;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.function.Function;

/**
 * Builder used a build the DataStore.
 * 
 * By default this will use the {@link MultiplexManager} when there are multiple
 * DataStores.
 */
public class DataStoreBuilder {
    private final List dataStores = new ArrayList<>();
    private Function multiplexer = null;

    public DataStoreBuilder dataStores(List dataStores) {
        this.dataStores.clear();
        this.dataStores.addAll(dataStores);
        return this;
    }

    public DataStoreBuilder dataStores(Consumer> customizer) {
        customizer.accept(this.dataStores);
        return this;
    }

    public DataStoreBuilder dataStore(DataStore dataStore) {
        this.dataStores.add(dataStore);
        return this;
    }

    public DataStoreBuilder multiplexer(Function multiplexer) {
        this.multiplexer = Objects.requireNonNull(multiplexer, "Multiplexer cannot be null");
        return this;
    }

    public DataStore build() {
        if (this.dataStores.isEmpty()) {
            return null;
        }
        if (this.dataStores.size() == 1) {
            return this.dataStores.get(0);
        }
        if (multiplexer != null) {
            return multiplexer.apply(this.dataStores.toArray(DataStore[]::new));
        } else {
            // Not set in ctor as multiplexer = MultiplexManager::new to make elide-datastore-multiplex optional
            return new MultiplexManager(this.dataStores.toArray(DataStore[]::new));
        }
    }
}