org.molgenis.data.EntityFactoryRegistrar 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.Map;
import org.springframework.context.ApplicationContext;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Component;

/**
 * Discovers and registers entity factories with the entity factory registry.
 *
 * @see EntityFactory
 * @see EntityFactoryRegistry
 */
@Component
public class EntityFactoryRegistrar {
  private final EntityFactoryRegistry entityFactoryRegistry;

  public EntityFactoryRegistrar(EntityFactoryRegistry entityFactoryRegistry) {
    this.entityFactoryRegistry = requireNonNull(entityFactoryRegistry);
  }

  public void register(ContextRefreshedEvent event) {
    ApplicationContext ctx = event.getApplicationContext();
    Map entityFactoryMap = ctx.getBeansOfType(EntityFactory.class);
    entityFactoryMap.values().forEach(this::registerStaticEntityFactory);
  }

  private void registerStaticEntityFactory(EntityFactory untypedEntityFactory) {
    entityFactoryRegistry.registerStaticEntityFactory(
        (EntityFactory) untypedEntityFactory);
  }
}