带有Spring Cloud配置的自定义属性加载器

Question

我在我的SpringBoot应用程序中使用SpringCloud配置，我需要编写一些自定义代码来处理属性，当属性被标记为这样时，这些属性将从我的公司密码库中读取。我知道SpringCloud支持Hashicorp Vault，但在这种情况下不是这样的。

我不想硬编码从不同的源检索特定的属性，例如，我会为应用程序app1创建一个属性文件，其中profiledev的值为：

spring.datasource.url=jdbc:mysql://localhost/test
spring.datasource.username=dbuser
spring.datasource.password=dbpass
spring.datasource.driver-class-name=com.mysql.jdbc.Driver

但对于其他一些配置文件，如prod，我会：

spring.datasource.url=jdbc:mysql://localhost/test
spring.datasource.username=prod-user
spring.datasource.password=[[vault]]
spring.datasource.driver-class-name=com.mysql.jdbc.Driver

因此，我需要自定义属性vault在发现返回值等于[[vault]]（或其他类型的标志）时拦截加载的属性，并从公司vault进行查询。在本例中，我的自定义属性加载器将查找属性spring的值。数据源。密码来自公司密码保险库。所有其他属性仍将按原样从标准SpringCloudConfigClient加载的值返回。

我希望只使用带注释的代码，而不使用XML配置。

Answer 1

在试图解决同样的问题时，我相信我已经找到了可以接受的解决办法。

下面是我的解决方案。

public class JBossVaultEnvironmentPostProcessor implements EnvironmentPostProcessor {
@Override
public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) {
    MutablePropertySources propertySources = environment.getPropertySources();
    Map<String, String> sensitiveProperties = propertySources.stream()
            .filter(propertySource -> propertySource instanceof EnumerablePropertySource)
            .map(propertySource -> (EnumerablePropertySource<?>) propertySource)
            .map(propertySource -> {
                Map<String, String> vaultProperties = new HashMap<>();
                String[] propertyNames = propertySource.getPropertyNames();
                for (String propertyName : propertyNames) {
                    String propertyValue = propertySource.getProperty(propertyName).toString();
                    if (propertyValue.startsWith("VAULT::")) {
                        vaultProperties.put(propertyName, propertyValue);
                    }
                }
                return vaultProperties;
            })
            .reduce(new HashMap<>(), (m1, m2) -> {
                m1.putAll(m2);
                return m1;
            });

    Map<String, Object> vaultProperties = new HashMap<>();
    sensitiveProperties.keySet().stream()
            .forEach(key -> {
                vaultProperties.put(key, VaultReader.readAttributeValue(sensitiveProperties.get(key)));
            });

    propertySources.addFirst(new MapPropertySource("vaultProperties", vaultProperties));
}

Answer 2

您可以实现自己的PropertySourceLocator并向spring添加条目。META-INF目录中的工厂。

#spring.factories

org.springframework.cloud.bootstrap.BootstrapConfiguration=/
foo.bar.MyPropertySourceLocator

然后你可以像Spring的普通属性一样引用公司密码库中的密钥。

spring.datasource.url=jdbc:mysql://localhost/test
spring.datasource.username=prod-user
spring.datasource.password=${lodaded.password.from.corporate.vault}
spring.datasource.driver-class-name=com.mysql.jdbc.Driver

HasiCorp的实现：Vault属性资源定位器支持