EntityReferenceCreatorImpl.java

package org.molgenis.data;

import org.molgenis.data.meta.model.EntityType;
import org.molgenis.data.support.LazyEntity;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import static java.util.Objects.requireNonNull;
import static java.util.stream.StreamSupport.stream;

@Component
public class EntityReferenceCreatorImpl implements EntityReferenceCreator
{
	private final DataService dataService;
	private final EntityFactoryRegistry entityFactoryRegistry;

	@Autowired
	public EntityReferenceCreatorImpl(DataService dataService, EntityFactoryRegistry entityFactoryRegistry)
	{

		this.dataService = requireNonNull(dataService);
		this.entityFactoryRegistry = requireNonNull(entityFactoryRegistry);
	}

	@Override
	public Entity getReference(EntityType entityType, Object id)
	{
		Entity lazyEntity = new LazyEntity(entityType, dataService, id);

		EntityFactory<? extends Entity, ?> entityFactory = entityFactoryRegistry.getEntityFactory(entityType);
		if (entityFactory != null)
		{
			// create static entity (e.g. Tag, Language, Package) that wraps the constructed dynamic or partial entity.
			lazyEntity = entityFactory.create(lazyEntity);
		}

		return lazyEntity;
	}

	@Override
	public Iterable<Entity> getReferences(EntityType entityType, Iterable<?> ids)
	{
		EntityFactory<? extends Entity, ?> entityFactory = entityFactoryRegistry.getEntityFactory(entityType);
		return () -> stream(ids.spliterator(), false).map(id ->
		{
			Entity lazyEntity = getReference(entityType, id);
			if (entityFactory != null)
			{
				// create static entity (e.g. Tag, Language, Package) that wraps the constructed dynamic or partial entity.
				lazyEntity = entityFactory.create(lazyEntity);
			}
			return lazyEntity;
		}).iterator();
	}
}