单元测试中，如何为java设置系统变量（System Variables）【建议使用配置文件或系统属性】

崔认知

发布于 2024-09-18 13:44:31

1690

发布于 2024-09-18 13:44:31

文章被收录于专栏：nobody

开发中一般我们使用配置文件或系统属性（System Properties）配置参数，而不用系统变量（System Variables），因为配置文件或系统属性（System Properties）比较灵活，系统变量（System Variables）是操作系统级别的环境变量，它不是Java特有的，而是操作系统的一部分。

系统属性是Java虚拟机（JVM）的配置参数，可以通过System.getProperty()方法来获取。它在JVM启动时设置，可以是JVM的内部属性，也可以是用户在启动JVM时通过命令行参数设置的属性。可以通过-Dproperty=value的形式在命令行中设置系统属性。

系统变量是操作系统级别的环境变量，可以在操作系统的任何地方使用，包括命令行、脚本和应用程序。在Java中，可以通过System.getenv()方法来访问这些环境变量。例如，PATH、HOME、USERPROFILE等都是常见的系统变量。

在Java程序中，你可以这样获取它们：

// 获取系统属性
String javaVersion = System.getProperty("java.version");
System.out.println("Java Version: " + javaVersion);

// 获取系统变量
String userHome = System.getenv("USERPROFILE"); // 在Windows上
// 或者
String userHome = System.getenv("HOME"); // 在Unix/Linux上
System.out.println("User Home Directory: " + userHome);

在Java程序中，系统属性我们可以这样设置：

 System.setProperty("name","认知科技技术团队");
 System.out.println(System.getProperty("name"));

但是系统变量没有办法直接去修改，只能通过反射的方式修改，下面代码通过两种不同的方式获取系统变量底层的数据结构map去修改（只针对当前进程生效）：

源码：

   public static void setEnv(String name, String value) throws Exception {
        getTheUnmodifiableEnvironment().put(name, value);
    }
    public static void setEnvV2(String name, String value) throws Exception {
        getTheUnmodifiableEnvironmentV2().put(name, value);
    }


    private static Map<String, String> getTheUnmodifiableEnvironment() throws Exception {
        Class<?> pe = Class.forName("java.lang.ProcessEnvironment");
        Method getenv = pe.getDeclaredMethod("getenv");
        getenv.setAccessible(true);
        Object unmodifiableEnvironment = getenv.invoke(null);
        Class<?> map = Class.forName("java.util.Collections$UnmodifiableMap");
        Field m = map.getDeclaredField("m");
        m.setAccessible(true);
        return (Map<String, String>) m.get(unmodifiableEnvironment);
    }

    private static Map<String, String> getTheUnmodifiableEnvironmentV2() throws Exception {
        Class<?> pe = Class.forName("java.lang.ProcessEnvironment");
        Field field = pe.getDeclaredField("theUnmodifiableEnvironment");
        field.setAccessible(true);
        Object unmodifiableEnvironment = field.get(null);
        Class<?> map = Class.forName("java.util.Collections$UnmodifiableMap");
        Field m = map.getDeclaredField("m");
        m.setAccessible(true);
        return (Map<String, String>) m.get(unmodifiableEnvironment);

    }

测试：

源码：

public static void main(String[] args) throws Exception {
        EnvUtils.setEnv("author","认知科技技术团队");

        System.out.println(System.getenv("author"));
        System.out.println(System.getenv().get("author"));

        EnvUtils.setEnvV2("微信公众号","认知科技技术团队");

        System.out.println(System.getenv("微信公众号"));
        System.out.println(System.getenv().get("微信公众号"));

        Thread thread = new Thread(new Runnable() {
            @Override
            public void run() {
                System.out.println(Thread.currentThread().getName()+System.getenv("微信公众号"));
                System.out.println(Thread.currentThread().getName()+System.getenv().get("微信公众号"));
            }
        });

        thread.start();
        thread.join();


    }

结果：

注：新版本java新增了模块化访问机制，本测试用的java21，所以需要IntelliJ IDEA中添加命令行参数：

--add-opens java.base/java.lang=ALL-UNNAMED --add-opens java.base/java.io=ALL-UNNAMED --add-opens java.base/java.math=ALL-UNNAMED --add-opens java.base/java.net=ALL-UNNAMED --add-opens java.base/java.nio=ALL-UNNAMED --add-opens java.base/java.security=ALL-UNNAMED --add-opens java.base/java.text=ALL-UNNAMED --add-opens java.base/java.time=ALL-UNNAMED --add-opens java.base/java.util=ALL-UNNAMED --add-opens java.base/jdk.internal.access=ALL-UNNAMED --add-opens java.base/jdk.internal.misc=ALL-UNNAMED

想想还是python来的比较方便

：


if __name__ == '__main__':
    os.environ['author'] = '认知科技技术团队'
    print(os.getenv('author'))
    print(os.environ.get('author'))
    print(os.environ['author'])