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.
There is a newer version: 8.4.5
package org.molgenis.data.index;

import org.molgenis.data.DataService;
import org.molgenis.data.index.job.*;
import org.molgenis.data.index.meta.*;
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;

import javax.annotation.PostConstruct;

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

//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, IndexJobExecutionMeta.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());
			}
		};
	}
}