io.camunda.exporter.adapters.ElasticsearchAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camunda-exporter Show documentation
Camunda Exporter
There is a newer version: 8.7.0-alpha2
/*
 * 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.exporter.adapters;

import co.elastic.clients.elasticsearch.ElasticsearchClient;
import co.elastic.clients.elasticsearch.core.BulkRequest;
import com.github.benmanes.caffeine.cache.CacheLoader;
import io.camunda.exporter.cache.ExporterEntityCacheProvider;
import io.camunda.exporter.cache.form.CachedFormEntity;
import io.camunda.exporter.cache.form.ElasticSearchFormCacheLoader;
import io.camunda.exporter.cache.process.CachedProcessEntity;
import io.camunda.exporter.cache.process.ElasticSearchProcessCacheLoader;
import io.camunda.exporter.config.ExporterConfiguration;
import io.camunda.exporter.schema.SearchEngineClient;
import io.camunda.exporter.schema.elasticsearch.ElasticsearchEngineClient;
import io.camunda.exporter.store.BatchRequest;
import io.camunda.exporter.store.ElasticsearchBatchRequest;
import io.camunda.exporter.utils.ElasticsearchScriptBuilder;
import io.camunda.exporter.utils.XMLUtil;
import io.camunda.search.connect.es.ElasticsearchConnector;
import java.io.IOException;

class ElasticsearchAdapter implements ClientAdapter {
  private final ElasticsearchClient client;
  private final ElasticsearchEngineClient searchEngineClient;
  private final ElasticsearchExporterEntityCacheProvider entityCacheLoader;

  ElasticsearchAdapter(final ExporterConfiguration configuration) {
    final var connector = new ElasticsearchConnector(configuration.getConnect());
    client = connector.createClient();
    searchEngineClient = new ElasticsearchEngineClient(client);
    entityCacheLoader = new ElasticsearchExporterEntityCacheProvider(client);
  }

  @Override
  public SearchEngineClient getSearchEngineClient() {
    return searchEngineClient;
  }

  @Override
  public BatchRequest createBatchRequest() {
    return new ElasticsearchBatchRequest(
        client, new BulkRequest.Builder(), new ElasticsearchScriptBuilder());
  }

  @Override
  public ExporterEntityCacheProvider getExporterEntityCacheProvider() {
    return entityCacheLoader;
  }

  @Override
  public void close() throws IOException {
    client._transport().close();
  }

  record ElasticsearchExporterEntityCacheProvider(ElasticsearchClient client)
      implements ExporterEntityCacheProvider {

    @Override
    public CacheLoader getProcessCacheLoader(
        final String processIndexName, final XMLUtil xmlUtil) {
      return new ElasticSearchProcessCacheLoader(client, processIndexName, xmlUtil);
    }

    @Override
    public CacheLoader getFormCacheLoader(final String formIndexName) {
      return new ElasticSearchFormCacheLoader(client, formIndexName);
    }
  }
}