io.camunda.service.ApiServices Maven / Gradle / Ivy

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

import static java.util.Optional.ofNullable;

import io.camunda.search.clients.CamundaSearchClient;
import io.camunda.service.security.auth.Authentication;
import io.camunda.service.transformers.ServiceTransformers;
import io.camunda.util.ObjectBuilder;
import io.camunda.zeebe.broker.client.api.BrokerClient;
import io.camunda.zeebe.broker.client.api.dto.BrokerRequest;
import io.camunda.zeebe.broker.client.api.dto.BrokerResponse;
import io.camunda.zeebe.msgpack.value.DocumentValue;
import io.camunda.zeebe.protocol.impl.encoding.MsgPackConverter;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.function.Function;
import org.agrona.DirectBuffer;
import org.agrona.concurrent.UnsafeBuffer;

public abstract class ApiServices> {

  protected final BrokerClient brokerClient;
  protected final CamundaSearchClient searchClient;
  protected final Authentication authentication;
  protected final ServiceTransformers transformers;

  protected ApiServices(
      final BrokerClient brokerClient,
      final CamundaSearchClient searchClient,
      final Authentication authentication) {
    this(brokerClient, searchClient, null, authentication);
  }

  protected ApiServices(
      final BrokerClient brokerClient,
      final CamundaSearchClient searchClient,
      final ServiceTransformers transformers,
      final Authentication authentication) {
    this.brokerClient = brokerClient;
    this.searchClient = searchClient;
    this.authentication = authentication;
    this.transformers = ofNullable(transformers).orElseGet(ServiceTransformers::newInstance);
  }

  public abstract T withAuthentication(final Authentication authentication);

  public T withAuthentication(
      final Function> fn) {
    return withAuthentication(fn.apply(new Authentication.Builder()).build());
  }

  protected  CompletableFuture sendBrokerRequest(final BrokerRequest brokerRequest) {
    return sendBrokerRequestWithFullResponse(brokerRequest).thenApply(BrokerResponse::getResponse);
  }

  protected  CompletableFuture> sendBrokerRequestWithFullResponse(
      final BrokerRequest brokerRequest) {
    brokerRequest.setAuthorization(authentication.token());
    return brokerClient
        .sendRequest(brokerRequest)
        .handleAsync(
            (response, error) -> {
              if (error != null) {
                throw new CamundaServiceException(error);
              }
              if (response.isError()) {
                throw new CamundaServiceException(response.getError());
              }
              if (response.isRejection()) {
                throw new CamundaServiceException(response.getRejection());
              }
              return response;
            });
  }

  protected DirectBuffer getDocumentOrEmpty(final Map value) {
    return value == null || value.isEmpty()
        ? DocumentValue.EMPTY_DOCUMENT
        : new UnsafeBuffer(MsgPackConverter.convertToMsgPack(value));
  }
}