io.camunda.exporter.cache.process.ElasticSearchProcessCacheLoader 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.cache.process;

import co.elastic.clients.elasticsearch.ElasticsearchClient;
import com.github.benmanes.caffeine.cache.CacheLoader;
import io.camunda.exporter.utils.XMLUtil;
import io.camunda.webapps.schema.entities.operate.ProcessEntity;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ElasticSearchProcessCacheLoader implements CacheLoader {

  private static final Logger LOG = LoggerFactory.getLogger(ElasticSearchProcessCacheLoader.class);

  private final ElasticsearchClient client;
  private final String processIndexName;

  private final XMLUtil xmlUtil;

  public ElasticSearchProcessCacheLoader(
      final ElasticsearchClient client, final String processIndexName, final XMLUtil xmlUtil) {
    this.client = client;
    this.processIndexName = processIndexName;
    this.xmlUtil = xmlUtil;
  }

  @Override
  public CachedProcessEntity load(final Long processDefinitionKey) throws IOException {
    final var response =
        client.get(
            request -> request.index(processIndexName).id(String.valueOf(processDefinitionKey)),
            ProcessEntity.class);
    if (response.found()) {
      final var processEntity = response.source();
      return new CachedProcessEntity(
          processEntity.getName(),
          processEntity.getVersionTag(),
          extractCallActivityIdsFromDiagram(processEntity));
    } else {
      // This should only happen if the process was deleted from ElasticSearch which should never
      // happen. Normally, the process is exported before the process instance is exporter. So the
      // process should be found in ElasticSearch index.
      LOG.debug("Process '{}' not found in Elasticsearch", processDefinitionKey);
      return null;
    }
  }

  private List extractCallActivityIdsFromDiagram(final ProcessEntity processEntity) {
    final String bpmnXml = processEntity.getBpmnXml();
    final Optional diagramData =
        xmlUtil.extractDiagramData(bpmnXml.getBytes(), processEntity.getBpmnProcessId());
    return diagramData.isPresent() ? diagramData.get().getCallActivityIds() : new ArrayList<>();
  }
}