在Spring-Boot中如何使用@Value注解注入集合類

更新時間：2021年08月02日 11:56:13 作者：m0_38063172

這篇文章主要介紹了在Spring-Boot中如何使用@Value注解注入集合類的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

我們在使用spring框架進(jìn)行開發(fā)時，有時候需要在properties文件中配置集合內(nèi)容并注入到代碼中使用。本篇文章的目的就是給出一種可行的方式。

1.注入

通常來說，我們都使用@Value注解來注入properties文件中的內(nèi)容，注入集合類時，我們也使用@Value來注入。

properties文件中的內(nèi)容如下：

my.set=foo,bar
my.list=foo,bar
my.map={"foo": "bar"}

分別是我們要注入的Set，List，Map中的內(nèi)容。

注入方式如下：

@Value("#{${my.map}}")
private Map<String, String> map;
@Value("#{'${my.set}'}")
private Set<String> set;
@Value("#{'${my.list}'}")
private List<String> list;

2.驗證

我們寫一個單測類來驗證上面的注入是否可行。

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE,
        classes = PropertiesTest.ClassUsingProperties.class)
@TestPropertySource(locations = "classpath:test.properties")
public class PropertiesTest {

    @Autowired
    private ClassUsingProperties classUsingProperties;

    @Test
    public void testInjectCollectionFieldsUsingPropertiesFile() {
        Map<String, String> map = classUsingProperties.getMap();
        Set<String> set = classUsingProperties.getSet();
        List<String> list = classUsingProperties.getList();
        asserts(map, set, list);
    }

    private void asserts(Map<String, String> map, Set<String> set, List<String> list) {
        Assert.assertEquals(map.get("foo"), "bar");
        Assert.assertTrue(set.contains("foo"));
        Assert.assertTrue(set.contains("bar"));
        Assert.assertTrue(list.contains("foo"));
        Assert.assertTrue(list.contains("bar"));
    }

    @Data
    @Component
    public static class ClassUsingProperties {
        @Value("#{${my.map}}")
        private Map<String, String> map;
        @Value("#{'${my.set}'}")
        private Set<String> set;
        @Value("#{'${my.list}'}")
        private List<String> list;
    }
}

test.properties中的內(nèi)容已經(jīng)在上面給出，位置在test文件夾下的resources文件夾下面（maven項目的文件夾結(jié)構(gòu)）。

3.原理

在我們使用的@Value注解中，每一個開頭都有個#，這其實就是說明我們使用了SpEL，如果直接使用SpEL，

就是下面的代碼：

ExpressionParser parser = new SpelExpressionParser();
Map<String, String> map =
                (Map<String, String>) parser
                      .parseExpression({'foo':'bar'}")
                      .getValue(Map.class);
Set<String> set =
                (Set<String>) parser
                        .parseExpression("'foo,bar'")
                        .getValue(Set.class);
List<String> list =
                (List<String>) parser
                        .parseExpression("'foo,bar'")
                        .getValue(List.class);

我們也使用單元測試來驗證：

@Test
@SuppressWarnings("unchecked")
public void testInitCollectionUsingSpEL() {
    ExpressionParser parser = new SpelExpressionParser();
    Map<String, String> map =
            (Map<String, String>) parser
                    .parseExpression("{'foo':'bar'}")
                    .getValue(Map.class);
    Set<String> set =
            (Set<String>) parser
                    .parseExpression("'foo,bar'")
                    .getValue(Set.class);
    List<String> list =
            (List<String>) parser
                    .parseExpression("'foo,bar'")
                    .getValue(List.class);
    asserts(map, set, list);
}

asserts方法的代碼已經(jīng)在驗證使用@Value注解方式的單元測試中給出。