spring注解驱动-@Profile

最新推荐文章于 2025-05-04 14:52:47 发布

原创最新推荐文章于 2025-05-04 14:52:47 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

spring 专栏收录该内容

10 篇文章

订阅专栏

本文介绍如何在Spring框架中使用@Profile注解来根据不同环境（如开发、测试、生产）配置不同的数据源，通过实战演示了如何在Java代码中实现数据源的灵活切换。

@Profile指定运行环境

package com.aaa.config;

import com.mchange.v2.c3p0.ComboPooledDataSource;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.EmbeddedValueResolverAware;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import org.springframework.context.annotation.PropertySource;
import org.springframework.util.StringValueResolver;

import javax.sql.DataSource;
import java.beans.PropertyVetoException;

@Configuration
@PropertySource(value = {"classpath:/jdbc.properties"})
@SuppressWarnings("all") //镇压所有的警告
public class MainDataSources implements EmbeddedValueResolverAware {

    @Value("${jdbc.driver}")
    private  String driver;
    @Value("${jdbc.username}")
    private String username;
    private  StringValueResolver valueResolver;
    private  String url;

    @Profile("test")
    @Bean("testDataSources")
    public DataSource dataSourceTest(@Value("${jdbc.password}") String password) throws Exception {
        ComboPooledDataSource com = new ComboPooledDataSource();
        com.setDriverClass(driver);
        com.setUser(username);
        com.setPassword(password);
        com.setJdbcUrl(url);
        return com;
    }
    @Profile("dev")
    @Bean("devDataSources")
    public DataSource dataSourcedev(@Value("${jdbc.password}") String password) throws Exception {
        ComboPooledDataSource com = new ComboPooledDataSource();
        com.setDriverClass(driver);
        com.setUser(username);
        com.setPassword(password);
        com.setJdbcUrl(url);
        return com;
    }
    @Profile("pro")
    @Bean("ProDataSources")
    public DataSource dataSourcePro(@Value("${jdbc.password}") String password) throws Exception {
        ComboPooledDataSource com = new ComboPooledDataSource();
        com.setDriverClass(driver);
        com.setUser(username);
        com.setPassword(password);
        com.setJdbcUrl(url);
        return com;
    }
    public void setEmbeddedValueResolver(StringValueResolver resolver) {
        this.valueResolver=resolver;
        url = resolver.resolveStringValue("${jdbc.url}");

    }
}

jdbc.driver=oracle.jdbc.driver.OracleDriver
jdbc.url=jdbc:oracle:thin:@localhost:1521:orcl
jdbc.username=scott
jdbc.password=tiger

import com.aaa.config.MainConfigOpenAutowried;
import com.aaa.config.MainDataSources;
import com.aaa.service.BookService;
import org.junit.Test;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;

import javax.sql.DataSource;


public class EnvTest {
   @Test
   public void Import(){
       AnnotationConfigApplicationContext apx = new AnnotationConfigApplicationContext();
       //创建一个applicationContext
       //设置激活环境 ,可以写多个
       apx.getEnvironment().setActiveProfiles("dev","test");
        //注册主配置类
       apx.register(MainDataSources.class);
       //启动刷新容器
       apx.refresh();
      //运行环境测试 
      //获得数据源的类型
       String[] beanNamesForType = apx.getBeanNamesForType(DataSource.class);
       for (String s : beanNamesForType) {
           System.out.println(s);
       }

   }





}