com.yomahub.liteflow.solon.integration.XPluginImpl Maven / Gradle / Ivy

Go to download
package com.yomahub.liteflow.solon.integration;

import com.yomahub.liteflow.annotation.LiteflowComponent;
import com.yomahub.liteflow.annotation.LiteflowMethod;
import com.yomahub.liteflow.core.NodeComponent;
import com.yomahub.liteflow.flow.FlowBus;
import com.yomahub.liteflow.solon.*;
import com.yomahub.liteflow.solon.config.LiteflowAutoConfiguration;
import com.yomahub.liteflow.solon.config.LiteflowMainAutoConfiguration;
import com.yomahub.liteflow.solon.config.LiteflowMonitorProperty;
import com.yomahub.liteflow.solon.config.LiteflowProperty;
import org.noear.solon.Utils;
import org.noear.solon.core.AppContext;
import org.noear.solon.core.Plugin;

import java.util.Properties;

/**
 * @author noear
 * @since 2.9
 */
public class XPluginImpl implements Plugin {

	@Override
	public void start(AppContext context) {
		// 加载默认配置
		Properties defProps = Utils.loadProperties("META-INF/liteflow-default.properties");
		if (defProps != null && defProps.size() > 0) {
			defProps.forEach((k, v) -> {
				context.cfg().putIfAbsent(k, v);
			});
		}

		// 是否启用
		boolean enable = context.cfg().getBool("liteflow.enable", false);

		if (!enable) {
			return;
		}

		// 放到前面
		context.beanMake(LiteflowProperty.class);
		context.beanMake(LiteflowMonitorProperty.class);
		context.beanMake(LiteflowAutoConfiguration.class);
		context.beanMake(LiteflowMainAutoConfiguration.class);

		// 订阅 NodeComponent 组件
		context.subWrapsOfType(NodeComponent.class, bw -> {
			NodeComponent node1 = bw.raw();
			node1.setNodeId(bw.name());

			FlowBus.addManagedNode(bw.name(), bw.raw());
		});

		context.beanExtractorAdd(LiteflowMethod.class, (bw, method, anno) -> {
			NodeComponent node1 = null;
			switch (anno.value()) {
				case PROCESS_SWITCH:
					node1 = new NodeSwitchComponentOfMethod(bw, method, anno.value());
					break;
				case PROCESS_BOOLEAN:
					node1 = new NodeBooleanComponentOfMethod(bw, method, anno.value());
					break;
				case PROCESS_FOR:
					node1 = new NodeForComponentOfMethod(bw, method, anno.value());
					break;
				default:
					node1 = new NodeComponentOfMethod(bw, method, anno.value());
			}

			String nodeId = Utils.annoAlias(anno.nodeId(), bw.name());
			node1.setNodeId(nodeId);
			node1.setType(anno.nodeType());

			FlowBus.addManagedNode(nodeId, node1);
		});

		context.beanBuilderAdd(LiteflowComponent.class, (clz, bw, anno) -> {
			if (NodeComponent.class.isAssignableFrom(clz)) {
				NodeComponent node1 = bw.raw();
				String nodeId = Utils.annoAlias(anno.id(), anno.value());

				node1.setNodeId(nodeId);
				node1.setName(anno.name());

				FlowBus.addManagedNode(nodeId, node1);
			}
			else {
				context.beanExtractOrProxy(bw); // 尝试提取 LiteflowMethod 函数，并支持自动代理
			}
		});
	}

}