com.yomahub.liteflow.builder.LiteFlowNodeBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of liteflow-core Show documentation
liteflow core
The newest version!
package com.yomahub.liteflow.builder;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.util.StrUtil;
import com.yomahub.liteflow.enums.NodeTypeEnum;
import com.yomahub.liteflow.exception.NodeBuildException;
import com.yomahub.liteflow.flow.FlowBus;
import com.yomahub.liteflow.flow.element.Node;
import com.yomahub.liteflow.log.LFLog;
import com.yomahub.liteflow.log.LFLoggerManager;
import com.yomahub.liteflow.monitor.MonitorFile;
import com.yomahub.liteflow.spi.holder.PathContentParserHolder;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

public class LiteFlowNodeBuilder {

	private final LFLog LOG = LFLoggerManager.getLogger(this.getClass());

	private final Node node;

	public static LiteFlowNodeBuilder createNode() {
		return new LiteFlowNodeBuilder();
	}

	public static LiteFlowNodeBuilder createCommonNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.COMMON);
	}

	public static LiteFlowNodeBuilder createSwitchNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.SWITCH);
	}

	public static LiteFlowNodeBuilder createBooleanNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.BOOLEAN);
	}

	public static LiteFlowNodeBuilder createForNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.FOR);
	}

	public static LiteFlowNodeBuilder createIteratorNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.ITERATOR);
	}

	public static LiteFlowNodeBuilder createScriptNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.SCRIPT);
	}

	public static LiteFlowNodeBuilder createScriptSwitchNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.SWITCH_SCRIPT);
	}

	public static LiteFlowNodeBuilder createScriptBooleanNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.BOOLEAN_SCRIPT);
	}

	public static LiteFlowNodeBuilder createScriptForNode() {
		return new LiteFlowNodeBuilder(NodeTypeEnum.FOR_SCRIPT);
	}

	public LiteFlowNodeBuilder() {
		this.node = new Node();
	}

	public LiteFlowNodeBuilder(NodeTypeEnum type) {
		this.node = new Node();
		this.node.setType(type);
	}

	public LiteFlowNodeBuilder setId(String nodeId) {
		if (StrUtil.isBlank(nodeId)) {
			return this;
		}
		this.node.setId(nodeId.trim());
		return this;
	}

	public LiteFlowNodeBuilder setName(String name) {
		if (StrUtil.isBlank(name)) {
			return this;
		}
		this.node.setName(name.trim());
		return this;
	}

	public LiteFlowNodeBuilder setClazz(String clazz) {
		if (StrUtil.isBlank(clazz)) {
			return this;
		}
		this.node.setClazz(clazz.trim());
		return this;
	}

	public LiteFlowNodeBuilder setClazz(Class clazz) {
		assert clazz != null;
		setClazz(clazz.getName());
		return this;
	}

	public LiteFlowNodeBuilder setType(NodeTypeEnum type) {
		this.node.setType(type);
		return this;
	}

	public LiteFlowNodeBuilder setScript(String script) {
		this.node.setScript(script);
		return this;
	}

	public LiteFlowNodeBuilder setFile(String filePath) {
		if (StrUtil.isBlank(filePath)) {
			return this;
		}
		try {
			List scriptList = PathContentParserHolder.loadContextAware()
				.parseContent(ListUtil.toList(filePath));
			String script = CollUtil.getFirst(scriptList);
			setScript(script);

			// 添加脚本文件监听
			List fileAbsolutePath = PathContentParserHolder.loadContextAware()
				.getFileAbsolutePath(ListUtil.toList(filePath));
			MonitorFile.getInstance().addMonitorFilePaths(fileAbsolutePath);
		}
		catch (Exception e) {
			String errMsg = StrUtil.format("An exception occurred while building the node[{}],{}", this.node.getId(),
					e.getMessage());
			throw new NodeBuildException(errMsg);
		}
		return this;
	}

	public LiteFlowNodeBuilder setLanguage(String language) {
		if (StrUtil.isNotBlank(language)){
			this.node.setLanguage(language);
		}
		return this;
	}

	public void build() {
		checkBuild();
		try {
			// 用于处理脚本 node
			if (this.node.getType().isScript()) {
				FlowBus.addScriptNode(this.node.getId(), this.node.getName(), this.node.getType(),
						this.node.getScript(), this.node.getLanguage());
			}
			// 用于处理普通 node
			else {
				FlowBus.addNode(this.node.getId(), this.node.getName(), this.node.getType(), this.node.getClazz());
			}
		}
		catch (Exception e) {
			String errMsg = StrUtil.format("An exception occurred while building the node[{}],{}", this.node.getId(),
					e.getMessage());
			LOG.error(errMsg, e);
			throw new NodeBuildException(errMsg);
		}
	}

	/**
	 * build 前简单校验
	 */
	private void checkBuild() {
		List errorList = new ArrayList<>();
		if (StrUtil.isBlank(this.node.getId())) {
			errorList.add("id is blank");
		}
		if (Objects.isNull(this.node.getType())) {
			errorList.add("type is null");
		}
		if (CollUtil.isNotEmpty(errorList)) {
			throw new NodeBuildException(CollUtil.join(errorList, ",", "[", "]"));
		}
	}
}