Spring Boot从AWS参数存储读取参数

时间:2020-01-08 12:52:25

标签: java spring amazon-web-services spring-boot aws-parameter-store

我创建了spring boot(gradle)应用程序,并包含依赖项: org.springframework.cloud:spring-cloud-starter-aws-parameter-store-config

我想使用AWSSimpleSystemsManagement从AWS参数存储中读取配置,但是我被迫这样写(在AWS中):

 config/application_dev/server.port: 8080

有什么方法可以从Spring Boot中读取如下内容:dev.application.server.port:8080

目前我认为所有这些都是通过自动配置进行管理的,有没有办法覆盖它

3 个答案:

答案 0 :(得分:2)

application.properties 中,您可以定义属性server.port=8081

spring-cloud-starter-aws-parameter-store-config 支持的parameter formats是:

  • /config/application/server.port
  • /config/application_dev/server.port
  • /config/my-service/server.port
  • /config/my-service_dev/server.port

通过在 bootstrap.properties 中定义以下属性,您可以通过某种方式更改格式:

spring.application.name=my-service
aws.paramstore.prefix=/config
aws.paramstore.defaultContext=application
aws.paramstore.profileSeparator=_

但是仅支持简单的自定义设置,因为主要参数命名逻辑是AwsParamStorePropertySourceLocator中的硬编码。

要大大改变参数格式,您必须定义一个自定义PropertySourceLocator并将其注册为引导程序配置。

问题在于dev.application.server.port是无效的参数名称。

AWS Systems Manager参数存储使用/作为路径分隔符,而Spring使用操作get-parameters-by-path。 一种解决方法是使用名称dev.application/server.port

但是该名称也无效。 参数名称必须是标准名称,因此有效名称为 /dev.application/server.port

要支持这种参数格式,请定义一个自定义PropertySourceLocator

@Configuration
public class CustomAwsParamStorePropertySourceLocator implements PropertySourceLocator {

  private static final Logger LOGGER =
      LoggerFactory.getLogger(CustomAwsParamStorePropertySourceLocator.class);

  private AWSSimpleSystemsManagement ssmClient;

  private List<String> contexts = new ArrayList<>();

  public CustomAwsParamStorePropertySourceLocator(AWSSimpleSystemsManagement ssmClient) {
    this.ssmClient = ssmClient;
  }

  public List<String> getContexts() {
    return contexts;
  }

  @Override
  public PropertySource<?> locate(Environment environment) {
    if (!(environment instanceof ConfigurableEnvironment)) {
      return null;
    }

    ConfigurableEnvironment env = (ConfigurableEnvironment) environment;

    List<String> profiles = Arrays.asList(env.getActiveProfiles());

    String defaultAppName = "application";
    this.contexts.add("/" + defaultAppName + "/");
    addProfiles(this.contexts, defaultAppName, profiles);

    String appName = env.getProperty("spring.application.name");
    this.contexts.add("/" + appName + "/");
    addProfiles(this.contexts, appName, profiles);

    Collections.reverse(this.contexts);

    CompositePropertySource composite = new CompositePropertySource("custom-aws-param-store");

    for (String propertySourceContext : this.contexts) {
      try {
        composite.addPropertySource(create(propertySourceContext));
      } catch (Exception e) {
        LOGGER.warn("Unable to load AWS config from " + propertySourceContext, e);
      }
    }

    return composite;
  }

  private void addProfiles(List<String> contexts, String appName, List<String> profiles) {
    for (String profile : profiles) {
      contexts.add("/" + profile + "." + appName + "/");
    }
  }

  private AwsParamStorePropertySource create(String context) {
    AwsParamStorePropertySource propertySource =
        new AwsParamStorePropertySource(context, this.ssmClient);
    propertySource.init();
    return propertySource;
  }
}

并通过添加文件META-INF/spring.factories

在引导上下文中注册它
org.springframework.cloud.bootstrap.BootstrapConfiguration=\
com.example.CustomAwsParamStorePropertySourceLocator

答案 1 :(得分:0)

您需要创建bootstrap.propeties文件才能配置根路径 在文件中添加“ aws.paramstore.prefix = dev”以替换“ config”

请参考: https://cloud.spring.io/spring-cloud-static/spring-cloud-aws/2.0.0.RELEASE/multi/multi__cloud_environment.html 3.2节

答案 2 :(得分:0)