org.molgenis.data.index.IndexConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of molgenis-data-index Show documentation
Data layer indexing framework.
The newest version!
package org.molgenis.data.index;

import static java.util.Objects.requireNonNull;
import static org.molgenis.data.index.job.IndexJobExecutionMetadata.INDEX_JOB_EXECUTION;

import javax.annotation.PostConstruct;
import org.molgenis.data.DataService;
import org.molgenis.data.index.job.IndexJobExecution;
import org.molgenis.data.index.job.IndexJobExecutionFactory;
import org.molgenis.data.index.job.IndexJobExecutionMetadata;
import org.molgenis.data.index.job.IndexJobScheduler;
import org.molgenis.data.index.job.IndexJobSchedulerImpl;
import org.molgenis.data.index.job.IndexJobService;
import org.molgenis.data.index.meta.IndexActionFactory;
import org.molgenis.data.index.meta.IndexActionGroupFactory;
import org.molgenis.data.index.meta.IndexActionGroupMetadata;
import org.molgenis.data.index.meta.IndexActionMetadata;
import org.molgenis.data.index.meta.IndexPackage;
import org.molgenis.data.index.transaction.IndexTransactionListener;
import org.molgenis.data.meta.model.EntityTypeFactory;
import org.molgenis.data.transaction.TransactionManager;
import org.molgenis.jobs.Job;
import org.molgenis.jobs.JobExecutor;
import org.molgenis.jobs.JobFactory;
import org.molgenis.jobs.model.JobExecutionMetaData;
import org.molgenis.jobs.model.JobPackage;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;

// TODO: These imported classes should be in separate config and this is the IndexJobConfig
@Import({
  IndexActionFactory.class,
  IndexActionGroupFactory.class,
  IndexActionMetadata.class,
  IndexActionGroupMetadata.class,
  IndexPackage.class,
  IndexJobExecutionFactory.class,
  IndexJobExecutionMetadata.class,
  JobPackage.class,
  JobExecutionMetaData.class,
  IndexActionRegisterServiceImpl.class,
  IndexingStrategy.class
})
@Configuration
public class IndexConfig {
  private final IndexActionRegisterService indexActionRegisterService;
  private final TransactionManager transactionManager;
  private final DataService dataService;
  private final IndexService indexService;
  private final IndexJobExecutionFactory indexJobExecutionFactory;
  private final EntityTypeFactory entityTypeFactory;
  private final JobExecutor jobExecutor;

  public IndexConfig(
      IndexActionRegisterService indexActionRegisterService,
      TransactionManager transactionManager,
      DataService dataService,
      IndexService indexService,
      IndexJobExecutionFactory indexJobExecutionFactory,
      EntityTypeFactory entityTypeFactory,
      JobExecutor jobExecutor) {
    this.indexActionRegisterService = requireNonNull(indexActionRegisterService);
    this.transactionManager = requireNonNull(transactionManager);
    this.dataService = requireNonNull(dataService);
    this.indexService = requireNonNull(indexService);
    this.indexJobExecutionFactory = requireNonNull(indexJobExecutionFactory);
    this.entityTypeFactory = requireNonNull(entityTypeFactory);
    this.jobExecutor = requireNonNull(jobExecutor);
  }

  @PostConstruct
  public void register() {
    indexActionRegisterService.addExcludedEntity(INDEX_JOB_EXECUTION);
  }

  @Bean
  public IndexTransactionListener indexTransactionListener() {
    final IndexTransactionListener indexTransactionListener =
        new IndexTransactionListener(indexJobScheduler(), indexActionRegisterService);
    transactionManager.addTransactionListener(indexTransactionListener);
    return indexTransactionListener;
  }

  @Bean
  public IndexJobScheduler indexJobScheduler() {
    return new IndexJobSchedulerImpl(dataService, indexJobExecutionFactory, jobExecutor);
  }

  @Bean
  public IndexJobService indexJobService() {
    return new IndexJobService(dataService, indexService, entityTypeFactory);
  }

  @Bean
  public JobFactory indexJobFactory() {
    return new JobFactory() {
      @Override
      public Job createJob(IndexJobExecution jobExecution) {
        return progress ->
            indexJobService().executeJob(progress, jobExecution.getIndexActionJobID());
      }
    };
  }
}