java：自定义变量加载到系统变量后替换shell模版并执行shell

清疚 2023-09-24 20:37 38阅读 0赞

> 这里的需求前提是，在项目中进行某些操作前，需要在命令后对shell配置文件的进行修改（如ip、port），这个对于用户是不友好的，需要改为用户页面输入ip、port，后台自动去操作修改配置；那么这篇博客的由来就有了。

![在这里插入图片描述][898bbeab426342998927978d7a7b4481.jpeg_pic_center]  
上面图片是AI创作，未经允许，不可商用哦！

#### 目录 ####

*  实现功能描述
 *  功能实现拆解
 *   *  项目加载自定义变量
     *   *  项目加载自定义变量文件到系统变量中
         *   *  自定义变量文件如下
             *  java代码如下
             *  实操验证如下
         *  shell通过source .env环境变量文件方式加载
         *   *  test.env环境变量文件如下
             *  shell模版文件如下
             *  实操验证如下
     *  替换shell模版并运行
     *   *  项目加载自定义变量文件到系统变量中
         *  shell通过source .env环境变量文件方式加载

以下为实际工作简化后描述

## 实现功能描述 ##

1.  项目加载自定义变量
2.  自定义变量替换shell模版内容
3.  项目运行shell

## 功能实现拆解 ##

针对上述功能描述，其实可以拆解为三步

1.  实现项目加载自定义变量
2.  实现替换shell模版内容
3.  运行shell

### 项目加载自定义变量 ###

这里有两种实现，  
一种是项目加载自定义变量文件到系统变量中；  
另一种是shell通过source .env环境变量方式加载。

#### 项目加载自定义变量文件到系统变量中 ####

> 这种方法的好处是项目中可以获取到自定义变量，如果项目代码需要获取自定义变量进行操作，可以考虑使用此方式

新增一个自定义变量文件（当然也可以在代码中固定写入变量直接加载到系统中，这样的缺点就是不灵活，如果有变动就需要改代码）

##### 自定义变量文件如下 #####

BASE_ROOT_DIR=/home
    LICENSE_PATH=/home/LICENSE
    PACKAGE_PATH=/home/test.tar.gz

##### java代码如下 #####

/**
         * 加载自定义变量文件到系统变量中
         * @param configFile
         */
        public static void loadProperties(String configFile){
        
            try {
        
                FileInputStream propFile = new FileInputStream(configFile);
                Properties p = new Properties(System.getProperties());
                p.load(propFile);
                System.setProperties(p);
            } catch (Exception e) {
        
                e.printStackTrace();
            }
        }

调用loadProperties方法后，就可以通过`System.getProperty("key")`获取自定义的变量了  
**如获取BASE\_ROOT\_DIR**：`System.getProperty("BASE_ROOT_DIR")`

##### 实操验证如下 #####

**自定义变量文件**  
![在这里插入图片描述][83108383db5f4886afd10e4286e070ec.png]

**测试代码**

import java.io.FileInputStream;
    import java.util.Properties;
    
    public class LoadProperties {
        
    
        public static void main(String[] args) {
        
    
            String configFile = "/Users/kunzhi/Downloads/learn/linuxTest/config.sh";
            loadProperties(configFile);
            System.out.println("BASE_ROOT_DIR: " + System.getProperty("BASE_ROOT_DIR"));
            System.out.println("LICENSE_PATH: " + System.getProperty("LICENSE_PATH"));
            System.out.println("PACKAGE_PATH: " + System.getProperty("PACKAGE_PATH"));
        }
    
        public static void loadProperties(String configFile){
        
            try {
        
                FileInputStream propFile = new FileInputStream(configFile);
                Properties p = new Properties(System.getProperties());
                p.load(propFile);
                System.setProperties(p);
            } catch (Exception e) {
        
                e.printStackTrace();
            }
        }
    }

**运行输出结果**  
![在这里插入图片描述][1fb11a602bad4fe5a69724ed65d7aac7.png]

#### shell通过source .env环境变量文件方式加载 ####

> 这种方法的好处是在shell执行时直接获取环境变量，不需要项目代码做多余的操作

1.  新增一个.env环境变量文件（如test.env）
2.  当在执行的shell文件中首行加入source命令（`source test.env`）或者先在命令行执行命令`source test.env`后，执行shell文件

##### test.env环境变量文件如下 #####

BASE_ROOT_DIR=/home
    LICENSE_PATH=/home/LICENSE
    PACKAGE_PATH=/home/test.tar.gz

##### shell模版文件如下 #####

cd ${DATASYS_BASE_ROOT_DIR}
    pwd
    
    echo "DATASYS_PACKAGE_PATH: ${DATASYS_PACKAGE_PATH}"
    echo "PACKAGE_PATH: ${PACKAGE_PATH}"
    echo "LICENSE_PATH: ${LICENSE_PATH}"

##### 实操验证如下 #####

创建shell模版文件 test.sh，并执行  
![在这里插入图片描述][d0dd28d747254fdda8cfb9cbb945c264.png]  
创建环境变量文件 test.env，并在test.sh文件首行加入命令：`source test.env`  
![在这里插入图片描述][1de12858373742159e82c9e8a4c9d601.png]

### 替换shell模版并运行 ###

> 因为替换是否成功，只有通过运行结果才能有力证明，所以我们放在一起看

由于替换shell模版内容的具体操作受自定义变量加载方式影响，所以我们还是按照两种加载方式来看

#### 项目加载自定义变量文件到系统变量中 ####

如果是项目加载自定义变量文件到系统变量中，还需要项目获取变量并传入shell替换；

代码如下

import java.io.FileInputStream;
    import java.io.InputStreamReader;
    import java.io.LineNumberReader;
    import java.util.Properties;
    
    public class LoadProperties {
        
    
        public static void main(String[] args) {
        
    
            String configFile = "/Users/kunzhi/Downloads/learn/linuxTest/config.sh";
            //加载自定义环境变量
            loadProperties(configFile);
    
            String BASE_ROOT_DIR = System.getProperty("BASE_ROOT_DIR");
            String LICENSE_PATH = System.getProperty("LICENSE_PATH");
            String PACKAGE_PATH = System.getProperty("PACKAGE_PATH");
            System.out.println(BASE_ROOT_DIR + ": " + System.getProperty("BASE_ROOT_DIR"));
            System.out.println(LICENSE_PATH + ": " + System.getProperty("LICENSE_PATH"));
            System.out.println(PACKAGE_PATH + ": " + System.getProperty("PACKAGE_PATH"));
            
            //需要传入shell模版替换的内容
            String[] evnp = {
        "BASE_ROOT_DIR=" + BASE_ROOT_DIR, "LICENSE_PATH=" + LICENSE_PATH, "PACKAGE_PATH=" + PACKAGE_PATH};
            String shellFile = "/Users/kunzhi/Downloads/learn/linuxTest/test1.sh";
            //替换并运行shell
            replaceAndRun(shellFile, evnp);
        }
    
        //加载自定义环境变量
        public static void loadProperties(String configFile){
        
            try {
        
                FileInputStream propFile = new FileInputStream(configFile);
                Properties p = new Properties(System.getProperties());
                p.load(propFile);
                System.setProperties(p);
            } catch (Exception e) {
        
                e.printStackTrace();
            }
        }
    
        //替换并运行shell
        private static void replaceAndRun(String path, String[] evnp) {
        
            try {
        
                Process process = Runtime.getRuntime().exec(new String[]{
        "/bin/sh", "-c", path}, evnp,null);
                //只有process.waitFor() 结果为0，才说明执行成功了
                if (process.waitFor() != 0) {
        
                    throw new Exception(String.valueOf(process.waitFor()));
                }
    
                InputStreamReader ir = new InputStreamReader(process.getInputStream());
                LineNumberReader input = new LineNumberReader(ir);
                String line;
                //输入shell中执行的每一行
                while ((line = input.readLine()) != null){
        
                    if (line.contains("ERROR")) {
        
                        throw new Exception("ERROR");
                    }
                    System.out.println(line);
                }
            } catch (Exception e) {
        
                System.out.println(e);
            }
        }
    }

运行结果  
![在这里插入图片描述][2b53075b26a04c8d980f070915702b9b.png]

#### shell通过source .env环境变量文件方式加载 ####

如果是shell通过source .env环境变量方式加载，在上一步实操演示可以看出，其实已经达到替换效果，直接执行shell文件即可了

代码如下

package test;
    
    import java.io.FileInputStream;
    import java.io.InputStreamReader;
    import java.io.LineNumberReader;
    import java.util.Properties;
    
    /**
     * @author yanZhiHang
     * @date 2023/2/26 17:06
     */
    public class LoadProperties {
        
    
        public static void main(String[] args) {
        
            runShell(shellFile1);
        }
    
        private static void runShell(String path) {
        
            try {
        
                Process process = Runtime.getRuntime().exec(new String[]{
        "/bin/sh", "-c", path}, null,null);
    
                if (process.waitFor() != 0) {
        
                    throw new Exception(String.valueOf(process.waitFor()));
                }
    
                InputStreamReader ir = new InputStreamReader(process.getInputStream());
                LineNumberReader input = new LineNumberReader(ir);
                String line;
                while ((line = input.readLine()) != null){
        
                    if (line.contains("ERROR")) {
        
                        throw new Exception("ERROR");
                    }
                    System.out.println(line);
                }
            } catch (Exception e) {
        
                System.out.println(e);
            }
        }
    }

[898bbeab426342998927978d7a7b4481.jpeg_pic_center]: https://img-blog.csdnimg.cn/898bbeab426342998927978d7a7b4481.jpeg#pic_center
[83108383db5f4886afd10e4286e070ec.png]: https://img-blog.csdnimg.cn/83108383db5f4886afd10e4286e070ec.png
[1fb11a602bad4fe5a69724ed65d7aac7.png]: https://img-blog.csdnimg.cn/1fb11a602bad4fe5a69724ed65d7aac7.png
[d0dd28d747254fdda8cfb9cbb945c264.png]: https://img-blog.csdnimg.cn/d0dd28d747254fdda8cfb9cbb945c264.png
[1de12858373742159e82c9e8a4c9d601.png]: https://img-blog.csdnimg.cn/1de12858373742159e82c9e8a4c9d601.png
[2b53075b26a04c8d980f070915702b9b.png]: https://img-blog.csdnimg.cn/2b53075b26a04c8d980f070915702b9b.png