io.camunda.operate.webapp.api.v1.dao.elasticsearch.ElasticsearchVariableDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of operate-webapp Show documentation
Operate Webapp
There is a newer version: 8.7.0-alpha2-rc1
/*
 * 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.webapp.api.v1.dao.elasticsearch;

import static io.camunda.operate.util.ElasticsearchUtil.joinWithAnd;
import static org.elasticsearch.index.query.QueryBuilders.termQuery;

import io.camunda.operate.conditions.ElasticsearchCondition;
import io.camunda.operate.util.ElasticsearchUtil;
import io.camunda.operate.webapp.api.v1.dao.VariableDao;
import io.camunda.operate.webapp.api.v1.entities.Query;
import io.camunda.operate.webapp.api.v1.entities.Results;
import io.camunda.operate.webapp.api.v1.entities.Variable;
import io.camunda.operate.webapp.api.v1.exceptions.APIException;
import io.camunda.operate.webapp.api.v1.exceptions.ResourceNotFoundException;
import io.camunda.operate.webapp.api.v1.exceptions.ServerException;
import io.camunda.webapps.schema.descriptors.operate.template.VariableTemplate;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Conditional;
import org.springframework.stereotype.Component;

@Conditional(ElasticsearchCondition.class)
@Component("ElasticsearchVariableDaoV1")
public class ElasticsearchVariableDao extends ElasticsearchDao implements VariableDao {

  @Autowired private VariableTemplate variableIndex;

  @Override
  protected void buildFiltering(
      final Query query, final SearchSourceBuilder searchSourceBuilder) {
    final Variable filter = query.getFilter();
    final List queryBuilders = new ArrayList<>();
    if (filter != null) {
      queryBuilders.add(buildTermQuery(Variable.KEY, filter.getKey()));
      queryBuilders.add(buildTermQuery(Variable.TENANT_ID, filter.getTenantId()));
      queryBuilders.add(
          buildTermQuery(Variable.PROCESS_INSTANCE_KEY, filter.getProcessInstanceKey()));
      queryBuilders.add(buildTermQuery(Variable.SCOPE_KEY, filter.getScopeKey()));
      queryBuilders.add(buildTermQuery(Variable.NAME, filter.getName()));
      queryBuilders.add(buildTermQuery(Variable.VALUE, filter.getValue()));
      queryBuilders.add(buildTermQuery(Variable.TRUNCATED, filter.getTruncated()));
    }
    searchSourceBuilder.query(joinWithAnd(queryBuilders.toArray(new QueryBuilder[] {})));
  }

  @Override
  public Variable byKey(final Long key) throws APIException {
    logger.debug("byKey {}", key);
    final List variables;
    try {
      variables = searchFor(new SearchSourceBuilder().query(termQuery(Variable.KEY, key)));
    } catch (final Exception e) {
      throw new ServerException(String.format("Error in reading variable for key %s", key), e);
    }
    if (variables.isEmpty()) {
      throw new ResourceNotFoundException(String.format("No variable found for key %s ", key));
    }
    if (variables.size() > 1) {
      throw new ServerException(String.format("Found more than one variables for key %s", key));
    }
    return variables.get(0);
  }

  @Override
  public Results search(final Query query) throws APIException {
    logger.debug("search {}", query);
    final SearchSourceBuilder searchSourceBuilder =
        buildQueryOn(query, Variable.KEY, new SearchSourceBuilder());
    try {
      final SearchRequest searchRequest =
          new SearchRequest().indices(variableIndex.getAlias()).source(searchSourceBuilder);
      final SearchResponse searchResponse = tenantAwareClient.search(searchRequest);
      final SearchHits searchHits = searchResponse.getHits();
      final SearchHit[] searchHitArray = searchHits.getHits();
      if (searchHitArray != null && searchHitArray.length > 0) {
        final Object[] sortValues = searchHitArray[searchHitArray.length - 1].getSortValues();
        final List variables =
            ElasticsearchUtil.mapSearchHits(
                searchHitArray, this::searchHitToVariableWithoutFullValue);
        return new Results()
            .setTotal(searchHits.getTotalHits().value)
            .setItems(variables)
            .setSortValues(sortValues);
      } else {
        return new Results().setTotal(searchHits.getTotalHits().value);
      }
    } catch (final Exception e) {
      throw new ServerException("Error in reading incidents", e);
    }
  }

  protected Variable searchHitToVariableWithoutFullValue(final SearchHit searchHit) {
    return searchHitToVariable(searchHit, false);
  }

  protected Variable searchHitToVariableWithFullValue(final SearchHit searchHit) {
    return searchHitToVariable(searchHit, true);
  }

  protected Variable searchHitToVariable(final SearchHit searchHit, final boolean isFullValue) {
    final Map searchHitAsMap = searchHit.getSourceAsMap();
    final Variable variable =
        new Variable()
            .setKey((Long) searchHitAsMap.get(Variable.KEY))
            .setProcessInstanceKey((Long) searchHitAsMap.get(Variable.PROCESS_INSTANCE_KEY))
            .setScopeKey((Long) searchHitAsMap.get(Variable.SCOPE_KEY))
            .setTenantId((String) searchHitAsMap.get(Variable.TENANT_ID))
            .setName((String) searchHitAsMap.get(Variable.NAME))
            .setValue((String) searchHitAsMap.get(Variable.VALUE))
            .setTruncated((Boolean) searchHitAsMap.get(Variable.TRUNCATED));
    if (isFullValue) {
      final String fullValue = (String) searchHitAsMap.get(Variable.FULL_VALUE);
      if (fullValue != null) {
        variable.setValue(fullValue);
      }
      variable.setTruncated(false);
    }
    return variable;
  }

  protected List searchFor(final SearchSourceBuilder searchSourceBuilder) {
    try {
      final SearchRequest searchRequest =
          new SearchRequest(variableIndex.getAlias()).source(searchSourceBuilder);
      final SearchResponse searchResponse = tenantAwareClient.search(searchRequest);
      final SearchHits searchHits = searchResponse.getHits();
      final SearchHit[] searchHitArray = searchHits.getHits();
      if (searchHitArray != null && searchHitArray.length > 0) {
        return ElasticsearchUtil.mapSearchHits(
            searchHitArray, this::searchHitToVariableWithFullValue);
      } else {
        return List.of();
      }
    } catch (final Exception e) {
      throw new ServerException("Error in reading variables", e);
    }
  }
}