edu.stanford.protege.webprotege.index.impl.EntitiesInProjectSignatureByIriIndexImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of webprotege-backend-service Show documentation

The newest version!

package edu.stanford.protege.webprotege.index.impl;

import edu.stanford.protege.webprotege.index.*;
import edu.stanford.protege.webprotege.inject.ProjectSingleton;
import org.semanticweb.owlapi.model.IRI;
import org.semanticweb.owlapi.model.OWLEntity;

import javax.annotation.Nonnull;
import javax.inject.Inject;
import java.util.Collection;
import java.util.List;
import java.util.stream.Stream;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * Matthew Horridge
 * Stanford Center for Biomedical Informatics Research
 * 2019-08-13
 */
@ProjectSingleton
public class EntitiesInProjectSignatureByIriIndexImpl implements EntitiesInProjectSignatureByIriIndex, DependentIndex {

    @Nonnull
    private final ProjectOntologiesIndex projectOntologiesIndex;

    @Nonnull
    private final EntitiesInOntologySignatureByIriIndex entitiesInOntologySignatureByIriIndex;

    @Inject
    public EntitiesInProjectSignatureByIriIndexImpl(@Nonnull ProjectOntologiesIndex projectOntologiesIndex,
                                                    @Nonnull EntitiesInOntologySignatureByIriIndex entitiesInOntologySignatureByIriIndex) {
        this.projectOntologiesIndex = checkNotNull(projectOntologiesIndex);
        this.entitiesInOntologySignatureByIriIndex = checkNotNull(entitiesInOntologySignatureByIriIndex);
    }

    @Nonnull
    @Override
    public Collection getDependencies() {
        return List.of(projectOntologiesIndex, entitiesInOntologySignatureByIriIndex);
    }

    @Nonnull
    @Override
    public Stream getEntitiesInSignature(@Nonnull IRI entityIri) {
        checkNotNull(entityIri);
        return projectOntologiesIndex.getOntologyIds()
                                     .flatMap(ontId -> entitiesInOntologySignatureByIriIndex.getEntitiesInSignature(entityIri, ontId))
                                     .distinct();
    }
}