io.annot8.common.implementations.pipelines.management.SimplePipelineFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of annot8-common-implementations Show documentation
Common functionality used by Annot8 implementations
There is a newer version: 0.2.0
Show newest version
/* Annot8 (annot8.io) - Licensed under Apache-2.0. */
package io.annot8.common.implementations.pipelines.management;

import java.util.Collection;
import java.util.Optional;
import java.util.function.BiConsumer;
import java.util.function.Supplier;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.annot8.common.implementations.pipelines.Pipeline;
import io.annot8.common.implementations.pipelines.PipelineBuilder;
import io.annot8.common.implementations.pipelines.configuration.ComponentConfiguration;
import io.annot8.common.implementations.pipelines.configuration.PipelineConfiguration;
import io.annot8.common.implementations.pipelines.configuration.TypedComponentConfiguration;
import io.annot8.common.implementations.registries.Annot8ComponentRegistry;
import io.annot8.core.components.Annot8Component;
import io.annot8.core.components.Processor;
import io.annot8.core.components.Resource;
import io.annot8.core.components.Source;
import io.annot8.core.exceptions.Annot8Exception;
import io.annot8.core.exceptions.IncompleteException;
import io.annot8.core.settings.Settings;

public class SimplePipelineFactory implements PipelineFactory {

  private static final Logger LOGGER = LoggerFactory.getLogger(SimplePipelineFactory.class);

  private final Supplier builderFactory;
  private final Annot8ComponentRegistry componentRegistry;

  public SimplePipelineFactory(
      Supplier builderFactory, Annot8ComponentRegistry componentRegistry) {
    this.builderFactory = builderFactory;
    this.componentRegistry = componentRegistry;
  }

  @Override
  public Pipeline create(PipelineConfiguration pipelineConfiguration) throws IncompleteException {
    return createPipeline(pipelineConfiguration);
  }

  private Pipeline createPipeline(PipelineConfiguration configuration) throws IncompleteException {
    PipelineBuilder pipelineBuilder = builderFactory.get();

    configuration
        .getSources()
        .forEach(
            s -> addComponentToBuilder(Source.class, s, (i, c) -> pipelineBuilder.addSource(i, c)));

    configuration
        .getProcessors()
        .forEach(
            s ->
                addComponentToBuilder(
                    Processor.class, s, (i, c) -> pipelineBuilder.addProcessor(i, c)));

    configuration
        .getResources()
        .forEach(
            s ->
                addComponentToBuilder(
                    Resource.class, s, (i, c) -> pipelineBuilder.addResource(s.getName(), i, c)));

    return pipelineBuilder.build();
  }

  private  void addComponentToBuilder(
      Class clazz, ComponentConfiguration config, BiConsumer> consumer) {

    try {
      TypedComponentConfiguration tcc = validateComponent(config, clazz);

      T s = createInstance(tcc.getComponentClass());
      consumer.accept(s, tcc.getSettings());

    } catch (Annot8Exception e) {
      LOGGER.warn(e.getMessage());
      LOGGER.debug("Exception is:", e);
    }
  }

  private  TypedComponentConfiguration validateComponent(
      ComponentConfiguration config, Class componentType) throws Annot8Exception {
    try {
      Optional> optional =
          componentRegistry.getComponent(config.getComponent(), componentType);
      Class componentClass = optional.get();
      return new TypedComponentConfiguration<>(componentClass, config.getSettings());
    } catch (Exception e) {
      throw new Annot8Exception(
          "Could not find class implementation for component name " + config.getComponent(), e);
    }
  }

  private  T createInstance(Class clazz) throws Annot8Exception {
    try {
      return clazz.getConstructor().newInstance();
    } catch (Exception e) {
      throw new Annot8Exception("Could not create instance of component " + clazz.getName(), e);
    }
  }
}