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

2024-09-18 13:44:31 浏览数 (2)

开发中一般我们使用配置文件或系统属性(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程序中,你可以这样获取它们:

代码语言:javascript复制
// 获取系统属性
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程序中,系统属性我们可以这样设置:

代码语言:javascript复制
 System.setProperty("name","认知科技技术团队");
 System.out.println(System.getProperty("name"));

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

源码:

代码语言:javascript复制
   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);

    }

测试:

源码:

代码语言:javascript复制
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中添加命令行参数:

代码语言:javascript复制
--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来的比较方便

代码语言:javascript复制

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

0 人点赞