Spring启动自定义环境变量

￡神魔★判官ぃ 2024-04-01 13:23 93阅读 0赞

## Spring启动自定义环境变量 ##

### 1 自定义值实体 ###

import java.net.InetAddress;
    import java.net.NetworkInterface;
    import java.util.Enumeration;
    
    public class CustomValueDto {
        
        /**
         * 获取本机Ip.
         */
        public String getIp() {
        
            return IpUtil.getLocalIP();
        }
    }
    
    class IpUtil {
        
    
        /**
         * 获取本机IP，只返回一个.
         */
        static String getLocalIP() {
        
            String sIP = "";
            InetAddress ip = null;
            try {
        
                // 如果是Windows操作系统
                if (isWindowsOS()) {
        
                    ip = InetAddress.getLocalHost();
                    // 如果是Linux操作系统
                } else {
        
                    boolean bFindIP = false;
                    Enumeration<NetworkInterface> netInterfaces = NetworkInterface.getNetworkInterfaces();
                    while (netInterfaces.hasMoreElements()) {
        
                        if (bFindIP) {
        
                            break;
                        }
                        NetworkInterface ni = netInterfaces.nextElement();
                        if (ni.isLoopback() || ni.isVirtual() || !ni.isUp()) {
        
                            continue;
                        }
                        // ----------特定情况，可以考虑用ni.getName判断
                        // 遍历所有ip
                        Enumeration<InetAddress> ips = ni.getInetAddresses();
                        while (ips.hasMoreElements()) {
        
                            ip = ips.nextElement();
                            // 127.开头的都是lookback地址
                            if (ip.isSiteLocalAddress() && !ip.isLoopbackAddress()
                                    && !ip.getHostAddress().contains(":")) {
        
                                bFindIP = true;
                                break;
                            }
                        }
    
                    }
                }
            } catch (Exception e) {
        
                e.printStackTrace();
            }
    
            if (null != ip) {
        
                sIP = ip.getHostAddress();
            }
    
            return sIP;
        }
    
        /**
         * 判断是否为Windows系统.
         */
        private static boolean isWindowsOS() {
        
            boolean isWindowsOS = false;
            String osName = System.getProperty("os.name");
            if (osName.toLowerCase().contains("windows")) {
        
                isWindowsOS = true;
            }
            return isWindowsOS;
        }
    }

### 2 自定义属性资源类 ###

import org.apache.commons.logging.Log;
    import org.apache.commons.logging.LogFactory;
    import org.springframework.core.env.ConfigurableEnvironment;
    import org.springframework.core.env.MutablePropertySources;
    import org.springframework.core.env.PropertySource;
    import org.springframework.core.env.StandardEnvironment;
    import org.springframework.core.log.LogMessage;
    
    /**
     * 自定义属性资源类
     */
    public class CustomPropertySource extends PropertySource<CustomValueDto> {
        
        /**
         * Name of the random {@link PropertySource}.
         */
        public static final String MY_PROPERTY_SOURCE_NAME = "myVar";
    
        private static final String PREFIX = "myVar.";
    
        private static final Log logger = LogFactory.getLog(CustomPropertySource.class);
    
        /**
         * MyValuePropertySource.
         */
        public CustomPropertySource() {
        
            this(MY_PROPERTY_SOURCE_NAME);
        }
    
        /**
         * MyValuePropertySource.
         */
        public CustomPropertySource(String name) {
        
            super(name, new CustomValueDto());
        }
    
        /**
         * 增加自定义表达式到环境上下文.
         */
        public static void addToEnvironment(ConfigurableEnvironment environment, Log logger) {
        
    
            MutablePropertySources sources = environment.getPropertySources();
            PropertySource<?> existing = sources.get(MY_PROPERTY_SOURCE_NAME);
            if (existing != null) {
        
                logger.trace("RandomValuePropertySource already present");
                return;
            }
            CustomPropertySource randomSource = new CustomPropertySource(MY_PROPERTY_SOURCE_NAME);
            if (sources.get(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME) != null) {
        
                sources.addAfter(StandardEnvironment.SYSTEM_ENVIRONMENT_PROPERTY_SOURCE_NAME, randomSource);
            } else {
        
                sources.addLast(randomSource);
            }
            logger.trace("MyValuePropertySource add to Environment");
        }
    
        @Override
        public Object getProperty(String name) {
        
            if (!name.startsWith(PREFIX)) {
        
                return null;
            }
            logger.trace(LogMessage.format("Generating property for '%s'", name));
            return getValue(name.substring(PREFIX.length()));
        }
    
        /**
         * 目前仅支持ip.
         */
        private Object getValue(String type) {
        
            if (type.equalsIgnoreCase("ip")) {
        
                return getSource().getIp();
            }
            if (type.equalsIgnoreCase("name")) {
        
                return "zzh";
            }
            //todo 自行设置属性
            return null;
        }
    }

### 3 自定义环境后置处理器EnvironmentPostProcessor ###

import org.apache.commons.logging.Log;
    import org.apache.commons.logging.LogFactory;
    import org.springframework.boot.SpringApplication;
    import org.springframework.boot.env.EnvironmentPostProcessor;
    import org.springframework.core.Ordered;
    import org.springframework.core.env.ConfigurableEnvironment;
    
    public class CustomPropertySourceEnvironmentPostProcessor implements EnvironmentPostProcessor, Ordered {
        
    
    
        private Log logger;
    
        /**
         * empty.
         */
        public CustomPropertySourceEnvironmentPostProcessor() {
        
            logger = LogFactory.getLog(CustomPropertySourceEnvironmentPostProcessor.class);
        }
    
        /**
         * init。
         */
        public CustomPropertySourceEnvironmentPostProcessor(Log logger) {
        
            this.logger = logger;
        }
    
        @Override
        public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
        
            CustomPropertySource.addToEnvironment(environment, this.logger);
        }
    
        @Override
        public int getOrder() {
        
            return Ordered.HIGHEST_PRECEDENCE + 2;
        }
    }