io.camunda.operate.schema.migration.elasticsearch.ElasticsearchMigrationPlanFactory Maven / Gradle / Ivy

Go to download
/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
 * one or more contributor license agreements. See the NOTICE file distributed
 * with this work for additional information regarding copyright ownership.
 * Licensed under the Camunda License 1.0. You may not use this file
 * except in compliance with the Camunda License 1.0.
 */
package io.camunda.operate.schema.migration.elasticsearch;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.camunda.operate.conditions.ElasticsearchCondition;
import io.camunda.operate.property.MigrationProperties;
import io.camunda.operate.property.OperateProperties;
import io.camunda.operate.schema.migration.FillPostImporterQueuePlan;
import io.camunda.operate.schema.migration.MigrationPlanFactory;
import io.camunda.operate.schema.migration.ReindexPlan;
import io.camunda.operate.schema.migration.ReindexWithQueryAndScriptPlan;
import io.camunda.operate.store.elasticsearch.RetryElasticsearchClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Conditional;
import org.springframework.stereotype.Component;

@Component
@Conditional(ElasticsearchCondition.class)
public class ElasticsearchMigrationPlanFactory implements MigrationPlanFactory {

  private final OperateProperties operateProperties;
  private final MigrationProperties migrationProperties;
  private final ObjectMapper objectMapper;
  private final RetryElasticsearchClient retryElasticsearchClient;
  private final RestHighLevelClient esClient;

  @Autowired
  public ElasticsearchMigrationPlanFactory(
      final OperateProperties operateProperties,
      final MigrationProperties migrationProperties,
      @Qualifier("operateObjectMapper") final ObjectMapper objectMapper,
      final RetryElasticsearchClient retryElasticsearchClient,
      final RestHighLevelClient esClient) {
    this.operateProperties = operateProperties;
    this.migrationProperties = migrationProperties;
    this.objectMapper = objectMapper;
    this.retryElasticsearchClient = retryElasticsearchClient;
    this.esClient = esClient;
  }

  @Override
  public FillPostImporterQueuePlan createFillPostImporterQueuePlan() {
    return new ElasticsearchFillPostImporterQueuePlan(
        operateProperties, migrationProperties, objectMapper, esClient);
  }

  @Override
  public ReindexPlan createReindexPlan() {
    return new ElasticsearchPipelineReindexPlan(retryElasticsearchClient, migrationProperties);
  }

  @Override
  public ReindexWithQueryAndScriptPlan createReindexWithQueryAndScriptPlan() {
    return new ElasticsearchReindexWithQueryAndScriptPlan(
        migrationProperties, objectMapper, esClient, retryElasticsearchClient);
  }
}