cn.lzgabel.converter.processing.task.JobWorkerTaskProcessor Maven / Gradle / Ivy

Go to download
package cn.lzgabel.converter.processing.task;

import cn.lzgabel.converter.bean.BaseDefinition;
import cn.lzgabel.converter.bean.task.JobWorkerDefinition;
import cn.lzgabel.converter.processing.BpmnElementProcessor;
import com.google.common.collect.Maps;
import io.camunda.zeebe.model.bpmn.builder.AbstractFlowNodeBuilder;
import io.camunda.zeebe.model.bpmn.builder.AbstractJobWorkerTaskBuilder;
import io.camunda.zeebe.model.bpmn.instance.Task;
import java.lang.reflect.InvocationTargetException;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import org.apache.commons.lang3.StringUtils;

/**
 * 〈功能简述〉

 * 〈所有JobWorkerTask节点类型详情设置〉
 *
 * @author lizhi
 * @since 1.0.0
 */
public abstract class JobWorkerTaskProcessor<
        E extends JobWorkerDefinition, T extends AbstractFlowNodeBuilder>
    implements BpmnElementProcessor {

  @Override
  public String onComplete(AbstractFlowNodeBuilder flowNodeBuilder, JobWorkerDefinition flowNode)
      throws InvocationTargetException, IllegalAccessException {

    String nodeType = flowNode.getNodeType();
    String nodeName = flowNode.getNodeName();
    String jobType = flowNode.getJobType();
    String jobRetries = flowNode.getJobRetries();

    // 创建 Task
    AbstractJobWorkerTaskBuilder jobWorkerTaskBuilder =
        getJobWorkerTaskBuilder(createInstance(flowNodeBuilder, nodeType));
    String id = jobWorkerTaskBuilder.getElement().getId();

    // 补充 header
    handleJobWorkerTask(
        jobWorkerTaskBuilder, nodeName, jobType, jobRetries, flowNode.getTaskHeaders());

    // 如果当前任务还有后续任务，则遍历创建后续任务
    BaseDefinition nextNode = flowNode.getNextNode();
    if (Objects.nonNull(nextNode)) {
      return onCreate(moveToNode(flowNodeBuilder, id), nextNode);
    } else {
      return id;
    }
  }

  /**
   * 通过实例获取对应 jobWorkerTaskBuilder
   *
   * @param target 实例对象
   * @return jobWorkerTaskBuilder
   */
  @SuppressWarnings("unchecked")
  abstract AbstractJobWorkerTaskBuilder getJobWorkerTaskBuilder(Object target);

  private , E extends Task> void handleJobWorkerTask(
      AbstractJobWorkerTaskBuilder jobWorkerTaskBuilder,
      String nodeName,
      String jobType,
      String jobRetries,
      Map taskHeaders) {

    // set name
    jobWorkerTaskBuilder.name(nodeName);

    // set job type
    if (StringUtils.isNotBlank(jobType)) {
      jobWorkerTaskBuilder.zeebeJobType(jobType);
    }
    // set job retries
    if (StringUtils.isNotBlank(jobRetries)) {
      jobWorkerTaskBuilder.zeebeJobRetries(jobRetries);
    }

    // set task header
    Optional.ofNullable(taskHeaders).orElse(Maps.newHashMap()).entrySet().stream()
        .filter(entry -> StringUtils.isNotBlank(entry.getKey()))
        .forEach(entry -> jobWorkerTaskBuilder.zeebeTaskHeader(entry.getKey(), entry.getValue()));
  }
}