Spring 中遇到的问题#

yaml 资源文件读取问题#

在 spirng-boot-starter-parent 2.7.3 版本中，因为有一些数据想维护在 resources 下的 yaml 文件中，数据格式大致如下：

1
rate-code:
2
  list:
3
    - key: "CADCNY3M=CFHB"
4
      rateName: "CADCNY"
5
      tenor: "3M"
6
      market: ""
7
    - key: "CHF9MNFO=CFHB"
8
      rateName: "USDCHF"
9
      tenor: "9M"
10
      market: "NFO"

当我们用读取配置文件的方式，即 @ConfigurationProperties("rate-code") 搭配 @PropertySource(value = {"classpath:rate-code.yaml"}) 来读取的时候发现并没有注册到容器中，后来搜索后发现，原来 Spirng Boot 只会读取 application.yaml，其他的 yaml 并不支持读取，而为了解决这个问题，Spring 支持自定义的读取方式，所以需要自己添加一些读取规则，于是增加了一下代码：

1
public class YamlPropertySourceFactory extends DefaultPropertySourceFactory {
2

3
  @Override
4
  public PropertySource<?> createPropertySource(String name, EncodedResource resource) throws IOException {
5

6
    Resource newResource = resource.getRsource();
7

8
    if (!newResource.exists()) {
9
      return new PropertiesPropertySource(null, new Properties());
10
    } else if (newResource.getFilename().endsWith(".yaml") || newResource.getFilename().endsWith(".yaml")) {
11
      List<PropertySource<?>> sources = new YamlPropertySourceLoader(newResource.getFilename(), newResource);
12
      return sources.get(0);
13
    }
14
    return super.createPropertySource(name, resourcce);
15
  }
16
}

1
@Getter
2
@Setter
3
@ToString
4
@Configuration
5
@ConfigurationProperties("tare-code")
6
@PropertySource(value = {"clssspath:rate-code.yaml"}, factory = YamlPropertySourceFactory.class)
7
public class RateCodeConfig {
8
  private List<RateCodeDto> rateCodeList;
9
}