Profile 注入不同環(huán)境下的 Bean

原理

在注入 Bean 的時(shí)候通過(guò) Profile 為 Bean 設(shè)定環(huán)境

實(shí)現(xiàn)

創(chuàng)建一個(gè)普通類(lèi)

DataSource.java

public class DataSource {
    private String url;
    private String username;
    private String password;
    private String driveClassName;

    @Override
    public String toString() {
        return "DataSource{" +
                "url='" + url + '\'' +
                ", username='" + username + '\'' +
                ", password='" + password + '\'' +
                ", driveClassName='" + driveClassName + '\'' +
                '}';
    }

    public DataSource(String url, String username, String password, String driveClassName) {
        this.url = url;
        this.username = username;
        this.password = password;
        this.driveClassName = driveClassName;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }

    public String getDriveClassName() {
        return driveClassName;
    }

    public void setDriveClassName(String driveClassName) {
        this.driveClassName = driveClassName;
    }
}

注入兩個(gè)相同類(lèi)型和名稱(chēng)的Bean，用Profile設(shè)定環(huán)境

Java配置

JavaConfig.java

@Configuration
public class JavaConfig {

    @Bean("ds")
    @Profile("dev")
    DataSource devDs(){
        return new DataSource("jdbc:mysql://localhost/dev", "dev", "dev", "com.mysql.jdbc.Driver");
    }

    @Bean("ds")
    @Profile("prod")
    DataSource prodDs(){
        return new DataSource("jdbc:mysql://localhost/prod", "prod", "prod", "com.mysql.jdbc.Driver");
    }
}

應(yīng)用

AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext();
context.getEnvironment().setActiveProfiles("prod"); // 先設(shè)定環(huán)境
context.register(JavaConfig.class); // 再加載Java配置類(lèi)
context.refresh(); // 最后刷新
DataSource ds = (DataSource) context.getBean("ds");
System.out.println(ds);

XML配置

applicationContext.xml

<beans profile="dev">
    <bean class="org.daistudy.bean.DataSource" id="ds">
        <constructor-arg name="url" value="jdbc:mysql://localhost/dev"/>
        <constructor-arg name="username" value="dev"/>
        <constructor-arg name="password" value="dev"/>
        <constructor-arg name="driveClassName" value="com.mysql.jdbc.Driver"/>
    </bean>
</beans>

<beans profile="prod">
    <bean class="org.daistudy.bean.DataSource" id="ds">
        <constructor-arg name="url" value="jdbc:mysql://localhost/prod"/>
        <constructor-arg name="username" value="prod"/>
        <constructor-arg name="password" value="prod"/>
        <constructor-arg name="driveClassName" value="com.mysql.jdbc.Driver"/>
    </bean>
</beans>

應(yīng)用

ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext();
context.getEnvironment().setActiveProfiles("dev");
context.setConfigLocation("applicationContext.xml");
context.refresh();
DataSource ds = (DataSource) context.getBean("ds");
System.out.println(ds);