org.rx.spring.SpringContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rxlib Show documentation

A set of utilities for Java

There is a newer version: 3.0.0

package org.rx.spring;

import lombok.Setter;
import org.rx.core.Linq;
import org.rx.core.Strings;
import org.rx.exception.InvalidException;
import org.rx.util.function.TripleFunc;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.util.Map;

import static org.rx.core.Extends.require;

/**
 * BeanPostProcessor 注册bean时变更
 * AopUtils
 */
@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class SpringContext implements ApplicationContextAware {
    private static ApplicationContext applicationContext;

    public static boolean isInitiated() {
        return applicationContext != null;
    }

    public static ApplicationContext getApplicationContext() {
        require(applicationContext);
        return applicationContext;
    }

    public static  T getBean(String name) {
        return (T) getApplicationContext().getBean(name);
    }

    public static  T getBean(Class clazz) {
        return getBean(clazz, true);
    }

    public static  T getBean(Class clazz, boolean throwOnEmpty) {
        Map beanMaps = getApplicationContext().getBeansOfType(clazz);
        if (throwOnEmpty && beanMaps.isEmpty()) {
            throw new InvalidException("Bean {} not registered", clazz);
        }
        return beanMaps.values().iterator().next();
    }

    public static String[] fromYamlArray(String yamlArray) {
        if (Strings.isEmpty(yamlArray)) {
            return new String[0];
        }
        return Linq.from(Strings.split(yamlArray, "\n")).select(p -> {
            if (p.startsWith("- ")) {
                return p.substring(2);
            }
            return p;
        }).toArray();
    }

    public void setApplicationContext(ApplicationContext applicationContext) {
        SpringContext.applicationContext = applicationContext;
    }

    @Setter
    static TripleFunc controllerExceptionHandler;
}