SystemEntityTypeRegistrar.java

package org.molgenis.data.meta.system;

import org.molgenis.data.meta.SystemEntityType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Component;

import java.util.Map;

import static java.util.Objects.requireNonNull;

/**
 * Discovers and registers system entity meta data with the system entity meta data registry.
 *
 * @see SystemEntityType
 * @see SystemEntityTypeRegistry
 */
@Component
public class SystemEntityTypeRegistrar
{
	private final SystemEntityTypeRegistry systemEntityTypeRegistry;

	@Autowired
	public SystemEntityTypeRegistrar(SystemEntityTypeRegistry systemEntityTypeRegistry)
	{
		this.systemEntityTypeRegistry = requireNonNull(systemEntityTypeRegistry);
	}

	public void register(ContextRefreshedEvent event)
	{
		ApplicationContext ctx = event.getApplicationContext();
		Map<String, SystemEntityType> systemEntityTypeMap = ctx.getBeansOfType(SystemEntityType.class);
		systemEntityTypeMap.values().forEach(this::register);
	}

	private void register(SystemEntityType systemEntityType)
	{
		systemEntityTypeRegistry.addSystemEntityType(systemEntityType);
	}
}