io.camunda.exporter.handlers.ProcessHandler 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.handlers;

import io.camunda.exporter.cache.ExporterEntityCache;
import io.camunda.exporter.cache.process.CachedProcessEntity;
import io.camunda.exporter.store.BatchRequest;
import io.camunda.exporter.utils.ExporterUtil;
import io.camunda.exporter.utils.ProcessModelReader;
import io.camunda.exporter.utils.XMLUtil;
import io.camunda.webapps.schema.entities.operate.ProcessEntity;
import io.camunda.zeebe.protocol.record.Record;
import io.camunda.zeebe.protocol.record.ValueType;
import io.camunda.zeebe.protocol.record.intent.ProcessIntent;
import io.camunda.zeebe.protocol.record.value.deployment.Process;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Optional;

public class ProcessHandler implements ExportHandler {

  private final String indexName;
  private final XMLUtil xmlUtil;
  private final ExporterEntityCache processCache;

  public ProcessHandler(
      final String indexName,
      final XMLUtil xmlUtil,
      final ExporterEntityCache processCache) {
    this.indexName = indexName;
    this.xmlUtil = xmlUtil;
    this.processCache = processCache;
  }

  @Override
  public ValueType getHandledValueType() {
    return ValueType.PROCESS;
  }

  @Override
  public Class getEntityType() {
    return ProcessEntity.class;
  }

  @Override
  public boolean handlesRecord(final Record record) {
    return record.getIntent().equals(ProcessIntent.CREATED);
  }

  @Override
  public List generateIds(final Record record) {
    return List.of(String.valueOf(record.getValue().getProcessDefinitionKey()));
  }

  @Override
  public ProcessEntity createNewEntity(final String id) {
    return new ProcessEntity().setId(id);
  }

  @Override
  public void updateEntity(final Record record, final ProcessEntity entity) {
    final Process process = record.getValue();
    entity
        .setId(String.valueOf(process.getProcessDefinitionKey()))
        .setKey(process.getProcessDefinitionKey())
        .setBpmnProcessId(process.getBpmnProcessId())
        .setVersion(process.getVersion())
        .setTenantId(ExporterUtil.tenantOrDefault(process.getTenantId()));
    final byte[] byteArray = process.getResource();

    final String bpmn = new String(byteArray, StandardCharsets.UTF_8);
    entity.setBpmnXml(bpmn);

    final String resourceName = process.getResourceName();
    entity.setResourceName(resourceName);

    final var versionTag = process.getVersionTag();
    if (!ExporterUtil.isEmpty(versionTag)) {
      entity.setVersionTag(versionTag);
    }

    linkStartFormIfPresent(record, entity);

    final Optional diagramData =
        xmlUtil.extractDiagramData(byteArray, process.getBpmnProcessId());
    diagramData.ifPresent(
        processEntity ->
            entity
                .setName(processEntity.getName())
                .setFlowNodes(processEntity.getFlowNodes())
                .setIsPublic(processEntity.getIsPublic())
                .setCallActivityIds(processEntity.getCallActivityIds()));

    // update local cache so that the process info is available immediately to the process instance
    // record handler
    final var cachedProcessEntity =
        new CachedProcessEntity(
            entity.getName(), entity.getVersionTag(), entity.getCallActivityIds());
    processCache.put(process.getProcessDefinitionKey(), cachedProcessEntity);
  }

  @Override
  public void flush(final ProcessEntity entity, final BatchRequest batchRequest) {
    batchRequest.add(indexName, entity);
  }

  @Override
  public String getIndexName() {
    return indexName;
  }

  private void linkStartFormIfPresent(final Record record, final ProcessEntity entity) {
    final var process = record.getValue();
    final var bpmnProcessId = process.getBpmnProcessId();
    final var byteArray = process.getResource();

    xmlUtil
        .createProcessModelReader(byteArray, bpmnProcessId)
        .flatMap(ProcessModelReader::extractStartFormLink)
        .ifPresent(
            startFormLink ->
                entity
                    .setFormId(startFormLink.formId())
                    .setFormKey(startFormLink.formKey())
                    .setIsFormEmbedded(!ExporterUtil.isEmpty(startFormLink.formKey())));
  }
}