Java 小案例：动态加载数据库信息

---

theme: smartblue

参考

Java 加密 - 掘金 (juejin.cn)

Java 读取配置文件 - 掘金 (juejin.cn)

环境搭建

创建application.properties，内容为数据库信息，账号密码已加密(加密参考开头链接)

```

测试数据库信息

db1.dev.dbname = dev_db db1.dev.username = MTN5u+gCyJI4zhXkJreKpA== db1.dev.password = Rlsy5KIMbP73BLKKWEvkPw==

生产数据库信息

db1.pro.dbname = pro_db db1.pro.username = MTN5u+gCyJI4zhXkJreKpA== db1.pro.password = wiv2NB5Tr24P0qKQTXerYg==

`` 创建DBUtil`工具类，动态加载数据库信息

``` package demo1;

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

public class DBUtil {

private static boolean isProduction() { 
    // 读取机器环境
    String env = System.getenv("Env");
    if("dev".equals(env)) {
        // 测试环境
        return false;
    }
    // 生产环境
    return true;
}


private static String getInfo(String value) throws IOException {
    InputStream inputStream = Main.class.getResourceAsStream("/application.properties") ;
    Properties properties = new Properties();
    properties.load(inputStream);
    String property = properties.getProperty(value);
    return property;
}

public static HashMap<String, String> getDBInfo() throws Exception {

    // 判断当前环境
    String env =  isProduction()? "pro":"dev";
    String dbname = "db1." + env + ".dbname";
    String username = "db1." + env + ".username";
    String password = "db1." + env + ".password";

    HashMap<String, String> res = new HashMap<String, String>();
    res.put("dbname", getInfo(dbname));
    res.put("username",EncryptUtil.decrypt( getInfo(username)));
    res.put("password", EncryptUtil.decrypt(getInfo(password)));

    return res;
}

}

```

创建EncryptUtil加密工具类

``` package demo1;

import java.nio.charset.StandardCharsets; import java.util.Base64;

import javax.crypto.Cipher; import javax.crypto.spec.IvParameterSpec; import javax.crypto.spec.SecretKeySpec;

public class EncryptUtil {

private static final String AES_ALGORITHM = "AES";
    // AES加密模式为CBC，填充方式为PKCS5Padding
    private static final String AES_TRANSFORMATION = "AES/CBC/PKCS5Padding";
    // AES密钥为16位
    private static final String AES_KEY = System.getenv("AES_KEY");
    // AES初始化向量为16位
    private static final String AES_IV = System.getenv("AES_IV");

    /**
     * AES加密
     *
     * @param data 待加密的数据
     * @return 加密后的数据，使用Base64编码
     */
    public static String encrypt(String data) throws Exception {
        // 将AES密钥转换为SecretKeySpec对象
        SecretKeySpec secretKeySpec = new SecretKeySpec(AES_KEY.getBytes(), AES_ALGORITHM);
        // 将AES初始化向量转换为IvParameterSpec对象
        IvParameterSpec ivParameterSpec = new IvParameterSpec(AES_IV.getBytes());
        // 根据加密算法获取加密器
        Cipher cipher = Cipher.getInstance(AES_TRANSFORMATION);
        // 初始化加密器，设置加密模式、密钥和初始化向量
        cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, ivParameterSpec);
        // 加密数据
        byte[] encryptedData = cipher.doFinal(data.getBytes(StandardCharsets.UTF_8));
        // 对加密后的数据使用Base64编码
        return Base64.getEncoder().encodeToString(encryptedData);
    }

    /**
     * AES解密
     *
     * @param encryptedData 加密后的数据，使用Base64编码
     * @return 解密后的数据
     */
    public static String decrypt(String encryptedData) throws Exception {
        // 将AES密钥转换为SecretKeySpec对象
        SecretKeySpec secretKeySpec = new SecretKeySpec(AES_KEY.getBytes(), AES_ALGORITHM);
        // 将AES初始化向量转换为IvParameterSpec对象
        IvParameterSpec ivParameterSpec = new IvParameterSpec(AES_IV.getBytes());
        // 根据加密算法获取解密器
        Cipher cipher = Cipher.getInstance(AES_TRANSFORMATION);
        // 初始化解密器，设置解密模式、密钥和初始化向量
        cipher.init(Cipher.DECRYPT_MODE, secretKeySpec, ivParameterSpec);
        // 对加密后的数据使用Base64解码
        byte[] decodedData = Base64.getDecoder().decode(encryptedData);
        // 解密数据
        byte[] decryptedData = cipher.doFinal(decodedData);
        // 返回解密后的数据
        return new String(decryptedData, StandardCharsets.UTF_8);
    }

}

```

测试

设置当前环境为测试环境

设置当前环境为生产环境