com.yomahub.liteflow.spi.solon.SolonContextAware Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of liteflow-solon-plugin Show documentation
liteflow solon plugin
There is a newer version: 2.12.4.2
Show newest version
package com.yomahub.liteflow.spi.solon;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.yomahub.liteflow.core.NodeComponent;
import com.yomahub.liteflow.core.proxy.DeclWarpBean;
import com.yomahub.liteflow.core.proxy.LiteFlowProxyUtil;
import com.yomahub.liteflow.spi.ContextAware;
import org.noear.solon.Solon;
import org.noear.solon.core.BeanWrap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 基于代码形式的 Solon 上下文工具类
 *
 * @author Bryan.Zhang
 */
public class SolonContextAware implements ContextAware {
    private static final Logger LOG = LoggerFactory.getLogger(SolonContextAware.class);

    @Override
    public  T getBean(String name) {
        try {
            return Solon.context().getBean(name);
        } catch (Exception e) {
            return null;
        }
    }

    @Override
    public  T getBean(Class clazz) {
        try {
            return Solon.context().getBean(clazz);
        } catch (Exception e) {
            return null;
        }
    }

    private  T getBean(String beanName, Class clazz) {
        try {
            return Solon.context().getBean(beanName);
        } catch (Exception e) {
            return null;
        }
    }

    @Override
    public  T registerBean(String beanName, Class c) {
        BeanWrap beanWrap = Solon.context().getWrap(beanName);
        if (beanWrap == null) {
            beanWrap = new BeanWrap(Solon.context(), c, null, beanName);
            Solon.context().putWrap(beanName, beanWrap);
            Solon.context().putWrap(c, beanWrap);
        }

        return beanWrap.get();
    }

    @Override
    public  T registerBean(Class c) {
        return registerBean(c.getName(), c);
    }

    @Override
    public  T registerBean(String beanName, Object bean) {
        BeanWrap beanWrap = Solon.context().getWrap(beanName);
        if (beanWrap == null) {
            beanWrap = Solon.context().wrap(beanName, bean);
            Solon.context().putWrap(beanName, beanWrap);
        }

        return beanWrap.get();
    }

    @Override
    public Object registerDeclWrapBean(String beanName, DeclWarpBean declWarpBean) {
        BeanWrap beanWrap = Solon.context().getWrap(beanName);
        if (beanWrap == null) {
            NodeComponent nodeComponent = LiteFlowProxyUtil.proxy2NodeComponent(declWarpBean);
            beanWrap = Solon.context().wrap(beanName, nodeComponent);
            Solon.context().putWrap(beanName, beanWrap);
        }

        LOG.info("proxy component[{}] has been found", beanName);
        return beanWrap.get();
    }

    @Override
    public  T registerOrGet(String beanName, Class clazz) {
        T t = getBean(beanName, clazz);
        if (ObjectUtil.isNull(t)) {
            t = registerBean(beanName, clazz);
        }
        return t;
    }

    @Override
    public  Map getBeansOfType(Class type) {
        List wrapsOfType = Solon.context().getWrapsOfType(type);
        return CollUtil.toMap(wrapsOfType, new HashMap(), BeanWrap::name, BeanWrap::get);
    }

    @Override
    public boolean hasBean(String beanName) {
        return Solon.context().hasWrap(beanName);
    }

    @Override
    public int priority() {
        return 1;
    }
}