org.molgenis.data.RepositoryCollectionRegistry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of molgenis-data Show documentation

There is a newer version: 8.4.5

package org.molgenis.data;

import static java.util.Objects.requireNonNull;

import java.util.HashMap;
import java.util.Map;
import java.util.stream.Stream;
import org.springframework.stereotype.Component;

/** Registry containing all {@link RepositoryCollection RepositoryCollections}. */
@Component
public class RepositoryCollectionRegistry {
  private final RepositoryCollectionDecoratorFactory repoCollectionDecoratorFactory;

  private Map repoCollectionMap;
  private RepositoryCollection defaultRepoCollection;

  public RepositoryCollectionRegistry(
      RepositoryCollectionDecoratorFactory repoCollectionDecoratorFactory) {
    this.repoCollectionDecoratorFactory = requireNonNull(repoCollectionDecoratorFactory);
    repoCollectionMap = new HashMap<>();
  }

  public Stream getRepositoryCollections() {
    return repoCollectionMap.values().stream()
        .map(repoCollectionDecoratorFactory::createDecoratedRepositoryCollection);
  }

  public void addRepositoryCollection(RepositoryCollection repoCollection) {
    repoCollectionMap.put(repoCollection.getName(), repoCollection);
  }

  public RepositoryCollection getRepositoryCollection(String name) {
    RepositoryCollection repoCollection = repoCollectionMap.get(name);
    return repoCollection != null
        ? repoCollectionDecoratorFactory.createDecoratedRepositoryCollection(repoCollection)
        : null;
  }

  public boolean hasRepositoryCollection(String name) {
    return repoCollectionMap.containsKey(name);
  }

  public RepositoryCollection getDefaultRepoCollection() {
    return repoCollectionDecoratorFactory.createDecoratedRepositoryCollection(
        defaultRepoCollection);
  }

  void setDefaultRepoCollection(RepositoryCollection defaultRepoCollection) {
    this.defaultRepoCollection = requireNonNull(defaultRepoCollection);
  }
}