io.camunda.operate.webapp.zeebe.operation.UpdateVariableHandler 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.zeebe.operation;

import static io.camunda.webapps.schema.entities.operation.OperationType.ADD_VARIABLE;
import static io.camunda.webapps.schema.entities.operation.OperationType.UPDATE_VARIABLE;

import io.camunda.webapps.schema.entities.operation.OperationEntity;
import io.camunda.webapps.schema.entities.operation.OperationType;
import io.camunda.zeebe.client.api.response.SetVariablesResponse;
import java.util.Set;
import org.springframework.stereotype.Component;

/** Update the variable. */
@Component
public class UpdateVariableHandler extends AbstractOperationHandler implements OperationHandler {

  @Override
  public void handleWithException(final OperationEntity operation) throws Exception {
    final String updateVariableJson =
        mergeVariableJson(operation.getVariableName(), operation.getVariableValue());
    final SetVariablesResponse response =
        zeebeClient
            .newSetVariablesCommand(operation.getScopeKey())
            .variables(updateVariableJson)
            .local(true)
            .send()
            .join();
    markAsSent(operation, response.getKey());
  }

  @Override
  public Set getTypes() {
    return Set.of(UPDATE_VARIABLE, ADD_VARIABLE);
  }

  private String mergeVariableJson(final String variableName, final String variableValue) {
    return String.format("{\"%s\":%s}", variableName, variableValue);
  }
}