OntologyRepository.java

package org.molgenis.ontology.core.repository;

import org.molgenis.data.DataService;
import org.molgenis.data.Entity;
import org.molgenis.data.support.QueryImpl;
import org.molgenis.ontology.core.model.Ontology;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.stream.Stream;

import static org.molgenis.ontology.core.meta.OntologyMetaData.*;

/**
 * Maps OntologyMetaData {@link Entity} <-> {@link Ontology}
 */
public class OntologyRepository
{
	@Autowired
	private DataService dataService;

	/**
	 * Retrieves all {@link Ontology}s.
	 */
	public Stream<Ontology> getOntologies()
	{
		return dataService.findAll(ONTOLOGY).map(OntologyRepository::toOntology);
	}

	/**
	 * Retrieves an ontology with a specific IRI.
	 *
	 * @param IRI the IRI of the ontology
	 * @return
	 */
	public Ontology getOntology(String IRI)
	{
		return toOntology(dataService.findOne(ONTOLOGY, QueryImpl.EQ(ONTOLOGY_IRI, IRI)));
	}

	private static Ontology toOntology(Entity entity)
	{
		if (entity == null)
		{
			return null;
		}
		return Ontology.create(entity.getString(ID), entity.getString(ONTOLOGY_IRI), entity.getString(ONTOLOGY_NAME));
	}

}