Как получить значение из файла yaml на основе метода param весной?

Question

Я работаю с yaml и весенней загрузкой, и я сделаю метод, который будет проверять мое поле.

1. Если значение поля не равно NULL, я должен вернуть это значение.
2. Если значение поля равно нулю, я должен загрузить значение по умолчанию из файла yaml на основе fieldName.

Пример моего метода:

@PropertySource("classpath:defaultValue.yml")
public final class ValidateAttributeValue {


    public static String validateAttributeValue(String attributeName, String attributeValue){
        if (nonNull(attributeValue)){
            return attributeValue;
        }
        //here I have to return default value from file based on attributeName
    }

файл yaml:

values:
  field: defaultValue
  field1: defaultValue1
  field2: defaultValue2
  field3: defaultValue3

Как это может быть реализовано в Spring boot + yaml?

Answer 1

Вы можете использовать YamlPropertiesFactoryBean для преобразования YAML в PropertySource.

public class YamlPropertySourceFactory implements PropertySourceFactory {

    @Override
    public PropertySource<?> createPropertySource(@Nullable String name, EncodedResource resource) throws IOException {
        Properties propertiesFromYaml = loadYamlIntoProperties(resource);
        String sourceName = name != null ? name : resource.getResource().getFilename();
        return new PropertiesPropertySource(sourceName, propertiesFromYaml);
    }

    private Properties loadYamlIntoProperties(EncodedResource resource) throws FileNotFoundException {
        try {
            YamlPropertiesFactoryBean factory = new YamlPropertiesFactoryBean();
            factory.setResources(resource.getResource());
            factory.afterPropertiesSet();
            return factory.getObject();
        } catch (IllegalStateException e) {
            // for ignoreResourceNotFound
            Throwable cause = e.getCause();
            if (cause instanceof FileNotFoundException)
                throw (FileNotFoundException) e.getCause();
            throw e;
        }
    }
}

И затем использовать его следующим образом:

@PropertySource(factory = YamlPropertySourceFactory.class, value = "classpath:blog.yaml")
public class YamlPropertysourceApplication {

Здесь вы найдете полное описание:

https://mdeinum.github.io/2018-07-04-PropertySource-with-yaml-files/