io.camunda.zeebe.gateway.query.impl.QueryApiImpl Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of zeebe-gateway-grpc Show documentation

Zeebe Gateway gRPC

There is a newer version: 8.6.0-alpha5

/*
 * 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.zeebe.gateway.query.impl;

import io.camunda.zeebe.broker.client.api.BrokerClient;
import io.camunda.zeebe.gateway.query.QueryApi;
import io.camunda.zeebe.protocol.Protocol;
import io.camunda.zeebe.protocol.record.ValueType;
import java.time.Duration;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;

public final class QueryApiImpl implements QueryApi {
  private final BrokerClient client;

  public QueryApiImpl(final BrokerClient client) {
    this.client = client;
  }

  @Override
  public CompletionStage getBpmnProcessIdFromProcess(
      final long key, final Duration timeout) {
    return queryPartition(key, ValueType.PROCESS, timeout);
  }

  @Override
  public CompletionStage getBpmnProcessIdFromProcessInstance(
      final long key, final Duration timeout) {
    return queryPartition(key, ValueType.PROCESS_INSTANCE, timeout);
  }

  @Override
  public CompletionStage getBpmnProcessIdFromJob(final long key, final Duration timeout) {
    return queryPartition(key, ValueType.JOB, timeout);
  }

  private CompletionStage queryPartition(
      final long key, final ValueType valueType, final Duration timeout) {
    final CompletableFuture result = new CompletableFuture<>();

    try {
      sendRequest(key, valueType, timeout, result);
    } catch (final Exception e) {
      result.completeExceptionally(e);
    }

    return result;
  }

  private void sendRequest(
      final long key,
      final ValueType valueType,
      final Duration timeout,
      final CompletableFuture result) {
    final var request = new BrokerExecuteQuery();
    final var partitionId = Protocol.decodePartitionId(key);

    request.setKey(key);
    request.setPartitionId(partitionId);
    request.setValueType(valueType);

    client
        .sendRequestWithRetry(request, timeout)
        .whenComplete(
            (response, error) -> {
              if (error != null) {
                result.completeExceptionally(error);
              } else {
                result.complete(response.getResponse());
              }
            });
  }
}