smart-flow-master.zip

package org.smartboot.flow.spring.extension; import org.smartboot.flow.core.attribute.AttributeHolder; import org.smartboot.flow.core.attribute.Attributes; import org.smartboot.flow.core.parser.DefinitionVisitor; import org.smartboot.flow.core.parser.ParserContext; import org.smartboot.flow.core.parser.definition.AdapterDefinition; import org.smartboot.flow.core.parser.definition.ChooseDefinition; import org.smartboot.flow.core.parser.definition.ElementDefinition; import org.smartboot.flow.core.parser.definition.EngineDefinition; import org.smartboot.flow.core.parser.definition.IfElementDefinition; import org.smartboot.flow.core.parser.definition.PipelineComponentDefinition; import org.smartboot.flow.core.parser.definition.PipelineDefinition; import org.smartboot.flow.core.parser.definition.ScriptDefinition; import org.smartboot.flow.core.script.ScriptCondition; import org.smartboot.flow.core.util.AssertUtil; import org.smartboot.flow.core.util.AuxiliaryUtils; import org.springframework.beans.PropertyValue; import org.springframework.beans.factory.config.BeanDefinition; import org.springframework.beans.factory.config.RuntimeBeanReference; import org.springframework.beans.factory.support.BeanDefinitionRegistry; import org.springframework.beans.factory.support.ManagedList; import org.springframework.beans.factory.support.ManagedMap; import org.springframework.beans.factory.support.RootBeanDefinition; import java.util.List; /** * @author qinluo * @date 2022/11/17 0:01 * @since 1.0.0 */ public class BeanDefinitionVisitor implements DefinitionVisitor { private final ParserContext context; /** * Temporary def register. */ private final BeanDefinitionRegister register; public BeanDefinitionVisitor(BeanDefinitionRegistry registry, ParserContext context) { this.context = context; this.register = new BeanDefinitionRegister(registry); } @Override public void visit(ElementDefinition ed) { if (ed instanceof ScriptDefinition) { ScriptDefinition sed = (ScriptDefinition)ed; Class<?> javaType = sed.getJavaType(); // Maybe a ref. if (javaType == null) { BeanDefinition beanDefinition = register.getBeanDefinition(sed.getName()); String beanClassName = beanDefinition.getBeanClassName(); Class<?> defJavaType = AuxiliaryUtils.asClass(beanClassName); AssertUtil.notNull(defJavaType, "bean " + sed.getName() + " javaType must be subclass of ScriptCondition"); AssertUtil.isTrue(ScriptCondition.class.isAssignableFrom(defJavaType), "bean " + sed.getName() + " javaType must be subclass of ScriptCondition"); // Register script content. beanDefinition.getPropertyValues().add("script", sed.getScript()); } else { RootBeanDefinition definition = new RootBeanDefinition(); definition.setBeanClass(javaType); PropertyValue script = new PropertyValue("script", sed.getScript()); PropertyValue name = new PropertyValue("name", sed.getName()); definition.getPropertyValues().addPropertyValue(name); definition.getPropertyValues().addPropertyValue(script); register.registerBeanDefinition(ed.getIdentifier(), definition, true); } } else { ed.visit(this); } } @Override public void visit(EngineDefinition ed) { RootBeanDefinition definition = new RootBeanDefinition(); PropertyValue name = new PropertyValue("name", ed.getName()); PropertyValue pipeline = new PropertyValue("pipeline", new RuntimeBeanReference(ed.getPipeline())); definition.setBeanClass(ed.resolveType()); definition.getPropertyValues().addPropertyValue(name); definition.getPropertyValues().addPropertyValue(pipeline); register.registerBeanDefinition(ed.getIdentifier(), definition, true); } @Override public void visit(PipelineDefinition ed) { RootBeanDefinition definition = new RootBeanDefinition(); PropertyValue name = new PropertyValue("name", ed.getName()); ManagedList<Object> components = new ManagedList<>(); definition.setBeanClass(ed.resolveType()); definition.getPropertyValues().addPropertyValue(name); definition.getPropertyValues().add("components", components); ed.getChildren().forEach(p -> p.visit(this)); for (ElementDefinition elementDef : ed.getChildren()) { // Fire sub visit. elementDef.visit(this); BeanDefinition beanDefinition = register.getBeanDefinition(elementDef.getIdentifier()); AssertUtil.notNull(beanDefinition, elementDef.getIdentifier() + " not exist"); components.add(beanDefinition); } register.registerBeanDefinition(ed.getIdentifier(), definition, !isAnonymous(ed.getIdentifier())); } @Override public void visit(PipelineComponentDefinition ed) { RootBeanDefinition definition = new RootBeanDefinition(); PropertyValue name = new PropertyValue("name", ed.getName()); PropertyValue pipeline = new PropertyValue("pipeline", new RuntimeBeanReference(ed.getPipeline())); definition.setBeanClass(ed.resolveType()); definition.getPropertyValues().addPropertyValue(name); if (isAnonymous(ed.getPipeline())) { context.getRegistered(ed.getPipeline()).visit(this); definition.getPropertyValues().add("pipeline", register.getBeanDefinition(ed.getPipeline())); } else { definition.getPropertyValues().addPropertyValue(pipeline); } appendAttributes(ed, definition); register.registerBeanDefinition(ed.getIdentifier(), definition); } private boolean isAnonymous(String name) { return name.contains("anonymous-pipeline"); } private void appendAttributes(ElementDefinition ed, RootBeanDefinition definition) { List<AttributeHolder> attributes = ed.getAttributes(); for (AttributeHolder holder : attributes) { Attributes attribute = holder.getAttribute(); if (attribute == Attributes.NAME || attribute == Attributes.DEGRADE_CALLBACK) { continue; } PropertyValue value = new PropertyValue(attribute.getName(), holder.getValue()); definition.getPropertyValues().addPropertyValue(value); } definition.getPropertyValues().add("attributes", ed.getAttributes()); } @Override public void visitBasic(ElementDefinition ed) { RootBeanDefinition definition = new RootBeanDefinition(); PropertyValue name = new PropertyValue("name", ed.getName()); definition.setBeanClass(ed.resolveType()); definition.getPropertyValues().addPropertyValue(name); if (isType(ed.getExecute())) { RootBeanDefinition conditionDef = new RootBeanDefinition(); conditionDef.setBeanClassName(ed.getExecute()); definition.getPropertyValues().add("executable", conditionDef); } else { AssertUtil.notBlank(ed.getExecute(), "component element ref must not be null!"); definition.getPropertyValues().add("executable", new RuntimeBeanReference(ed.getExecute())); } AttributeHolder degradeCallback = ed.getAttributes().stream() .filter(p -> p.getAttribute() == Attributes.DEGRADE_CALLBACK).findFirst().orElse(null); if (degradeCallback != null) { String callback = String.valueOf(degradeCallback.getValue()); if (isType(callback)) { RootBeanDefinition conditionDef = new RootBeanDefinition(); conditionDef.setBeanClassName(callback); definition.getPropertyValues().add("degradeCallback", conditionDef);