SpringBoot中,如何把自定义的yml配置文件中的内容和对应类的bean对象取出
首先 yml文件中的自定义配置如下
login-type-config:
types:
k1: "yuheng0"
k2: "yuheng1"
我们有两个对应的类,yuheng0 和 yuheng1 ,他们都实现了say接口,并且重写了say方法。下面要通过请求k1 k2 获取yuheng0 和 yuheng1类的bean对象。
注意,要记得加上Component注解 交给IOC容器管理他们
这两个类的具体内容如下
package com.example.demo;
import org.springframework.stereotype.Component;
@Component
public class yuheng0 implements say{
public void say(){
System.out.println("yuheng0");
}
@Override
public String toString() {
return "k1 to string";
}
}
package com.example.demo;
import org.springframework.stereotype.Component;
@Component
public class yuheng1 implements say{
public void say(){
System.out.println("yuheng1");
}
@Override
public String toString() {
return "k2 to string";
}
}
package com.example.demo;
public interface say {
void say();
}
下面是配置类,可以这么写配置来读取自定义配置的信息。这边的map是string string类型的,也就是单纯的把配置类的信息以String的形式读到map里面
package com.example.demo;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.Map;
@Component
@ConfigurationProperties(prefix = "login-type-config") // 注意要对应
public class LoginTypeConfig {
private Map<String,String> types;
public Map<String, String> getTypes() {
return types;
}
public void setTypes(Map<String, String> types) {
this.types = types;
}
}
最后就是最关键的逻辑,实现ApplicationContextAware接口,他可以根据名称从ioc容器中取出对应的bean
这次我们自定义的map类型是string 和 say。根据名称拿到对应的bean对象,也就是yuheng0和yuheng1的对象。放到自定义的map里面。在请求test的时候,从map中拿出已经准备好的bean对象就可以了。
package com.example.demo;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.HashMap;
import java.util.Map;
@RestController
public class demo implements ApplicationContextAware {
@Autowired
private LoginTypeConfig loginTypeConfig;
private static Map<String,say> res = new HashMap<>();
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
loginTypeConfig.getTypes().forEach((k, v) -> {
res.put(k, (say)applicationContext.getBean(v));
});
}
@GetMapping("/test")
public say test(String type)
{
say s = res.get(type);
System.out.println(s.toString());
s.say();
return s;
}
}