覆盖默认数据源getConnection（）

Question

我试图将Spring应用程序（大部分）转换为Spring Boot应用程序。在应用程序中，我有一个HTTP基本过滤器，收集用户名和密码，然后在DataSource实现中作为变量传递。

在这个数据源中，getConnection（）方法是这样的：

@Override\n public Connection getConnection() throws SQLException {
Statement  stmt = null;

try {
    ConnectionWrapper connection = this.authenticatedConnection.get();
    if (connection == null) {
        connection = new ConnectionWrapper(this.dataSource.getConnection());

        StringBuilder command;

        // The CONNECT command allows indicating a user name, a password
        // and a database to initiate a
        // new session in the server with a new profile.
        command = new StringBuilder("CONNECT USER ").append(this.parameters.get().get(USER_NAME)).append(" PASSWORD ")
                .append("'").append(this.parameters.get().get(PASSWORD_NAME)).append("'").append(" DATABASE ")
                .append(this.parameters.get().get(DATA_BASE_NAME));

        this.authenticatedConnection.set(connection);

        stmt = connection.createStatement();
        stmt.execute(command.toString());
    }

    return connection;
} catch (final SQLException e) {...`

（由于StackOverflow格式问题，\n作为新行）

在Spring中，我可以毫无问题地实现@autowiledPrivate DataSource dataSource。据我所知，在Spring Boot中，对象需要是一个Bean才能使用@autowmed，但是当我在这个实现的DataSource之前添加@Bean时，我得到了“注释@Bean不允许用于这个位置”

我怎样才能得到它，这样我就可以做一个数据源了。getConnection（）；从主数据源获取连接，或者能够覆盖主数据源的方法？

在我看来，这里按优先顺序列出了4种可能的解决方案：

1. 创建一个实际上正在覆盖spring的数据源。数据源的方法
2. 让这个实现“Beanified”，这样我就可以再次@Autowired数据源了
3. 我想我可以跳过@Autowired并简单地设置它。dataSource=[在application.properties中定义的spring.dataSource的未知引用]
4. 创建另一个用spring配置的数据源类ProgrammedDataSource。数据源属性，然后将其设置为。dataSource=新编程的dataSource（）

但是我试图实施这些解决方案中的任何一个都产生了这个问题。

Answer 1

我想出来了。我不需要在那里生成Bean，尽管我仍然不确定为什么我不能在数据源之前调用@Bean，但无论如何。

在我的申请中：

public class ServiceApplication {

@Bean
@Primary
@ConfigurationProperties("spring.datasource")
 public DataSource dataSource(){
  return DataSourceBuilder.create().build();
 }

@Bean(name="AuthDataSource")
public DataSource authDataSource() {
        return new AuthDataSource();
}

public static void main(String[] args) {
    SpringApplication.run(ServiceApplication.class, args);
}

}

在控制器里我有：

@Controller
@RequestMapping("/service")
public class ServiceController {

      @Autowired
      public void MyBean(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = new JdbcTemplate(new AuthDataSource());
      } ...

但是，由于我在JdbcTemboard中调用了新的AuthDataSource（），它没有执行自动连接。现在控制器看起来像这样，它的工作原理是：

@Controller
@RequestMapping("/service")
public class ServiceController {

      @Autowired
      @Qualifier("AuthDataSource")
      private DataSource datasource;
      private JdbcTemplate jdbcTemplate;

      @Autowired
      public void MyBean(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = new JdbcTemplate(this.dataSource);
      } ...