com.jd.laf.web.vertx.spring.SpringEnvironment Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of laf-vertx-web-springboot-starter Show documentation

The newest version!

package com.jd.laf.web.vertx.spring;

import com.jd.laf.web.vertx.Environment;
import org.springframework.context.ApplicationContext;

import java.util.concurrent.ConcurrentHashMap;

/**
 * Created by yangyang36 on 2018/10/10.
 */
public class SpringEnvironment implements Environment {

    protected ConcurrentHashMap params = new ConcurrentHashMap<>();

    protected org.springframework.core.env.Environment environment;
    protected ApplicationContext applicationContext;

    public SpringEnvironment(org.springframework.core.env.Environment environment, ApplicationContext applicationContext) {
        this.environment = environment;
        this.applicationContext = applicationContext;
    }

    @Override
    public  T getObject(final String name) {
        Object value = params.get(name);
        if (value == null) {
            value = environment.getProperty(name);
            if (value == null) {
                value = applicationContext.containsBean(name) ? applicationContext.getBean(name) : null;
            }
            if (value != null) {
                Object exists = params.put(name, value);
                if (exists != null) {
                    value = exists;
                }
            }
        }
        return (T) value;
    }

    @Override
    public void put(String name, Object obj) {
        params.put(name, obj);
    }
}