spring boot判断是否线上环境

版权声明：本文为博主原创文章，欢迎注明出处转载 https://blog.csdn.net/shepherd_dirk/article/details/85340996

package com.guang.easygo.util;

import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;

/**
 * @Author: Dirk
 * @Description:
 * @Date: Created in 13:42 2018/4/14
 */

public class PropertyUtil {

    private static Properties loadPropertiesByFileName(String fileName) {
        Properties prop = new Properties();
        InputStream inputStream = PropertyUtil.class.getClassLoader().getResourceAsStream(fileName);

        try {
            prop.load(inputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }

        return prop;

    }

    /**
     * 是否是线上环境
     */
    public static boolean isProduction() {
        Properties properties = loadPropertiesByFileName("application.properties");
        return properties.get("spring.profiles.active").toString().equals("production");
    }

    public static boolean isDev() {
        return !isProduction();
    }

    public static String get(String propertyName) {
        Properties properties = loadPropertiesByFileName("application.properties");

        String config = properties.getProperty("spring.profiles.active");

        String fileName = "application-" + config + ".properties";

        return loadPropertiesByFileName(fileName).get(propertyName).toString();
    }
}