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

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

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

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

/**
 * 基于代码形式的 Solon 上下文工具类
 *
 * @author Bryan.Zhang
 */
public class SolonContextAware implements ContextAware {

    @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 = new BeanWrap(Solon.context(), c, null, beanName);
        Solon.context().putWrap(beanName, 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 = new BeanWrap(Solon.context(), bean.getClass(), bean, beanName);
        Solon.context().putWrap(beanName, beanWrap);

        return beanWrap.get();
    }

    @Override
    public Object registerDeclWrapBean(String beanName, DeclWarpBean declWarpBean) {
        BeanWrap beanWrap = new BeanWrap(Solon.context(), declWarpBean.getClass(), declWarpBean, beanName);
        Solon.context().putWrap(beanName, beanWrap);
        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;
    }

}