用API调用发送的参数配置bean

Question

我有一个配置类来配置sqlite数据库连接。在这里，我手动配置它。

@Configuration
@EnableJpaRepositories(basePackages = "fr.company.dashboard.io.repository")
public class DBConfiguration {
    private final ApplicationProperties applicationProperties;

    @Autowired
    public DBConfiguration(ApplicationProperties applicationProperties) {
        this.applicationProperties = applicationProperties;
    }

    @Bean
    public DataSource dataSource() {
        final DriverManagerDataSource dataSource = new DriverManagerDataSource();
        dataSource.setDriverClassName(applicationProperties.getProperty("hibernate.connection.driverClassName"));
        dataSource.setUrl(applicationProperties.getProperty("url"));
        dataSource.setUsername(applicationProperties.getProperty("user"));
        dataSource.setPassword(applicationProperties.getProperty("password"));
        return dataSource;
    }
}

我想通过使用API get方法来设置数据库，该方法包括URL中的数据库名称

Answer 1

首先，需要将bean的作用域更改为"request"，然后将您想要的参数传递到如下所示：

    @Scope("request")
    @bean
      public DataSource dataSource(String dbName){
           ......
           ......
        return dataSource;
      }

在你的控制器上，你应该这样做，

@Controller
public class YourController{

   @Autowired
   private BeanFactory beanFactory;

   @RequestMapping("/")
   public String exFunc(){
      String dbName = "....."
      DataSource dataSource = 
         beanFactory.getBean(DataSource.class, dbName);
   }
}

Answer 2

在我的例子中，我在同一个文件夹中有多个sqlite。为了从入口点检索数据库名称，我更改了控制器：

@RestController
@RequestMapping("/dbfiles")
public class AppController {
    private final AppService appService;
    private final DataSourceContextHolder dataSourceContextHolder;

    @Autowired
    public DatabaseListController(AppService AppService,
                                  DataSourceContextHolder dataSourceContextHolder) {
        this.appService= appService;
        this.dataSourceContextHolder = dataSourceContextHolder;
    }
   
    @GetMapping("/{dbName}")
    public List<AppResponse> getAll(@PathVariable("dbName") String dbName) throws IOException {
        Set<String> databases = Utils.getDatabaseList();
        if(databases.contains(dbName) || databases.contains(dbName + ".db")) {
            dataSourceContextHolder.setBranchContext(dbName);
        }

        List<AppResponse> returnValue = new ArrayList<>();
        ModelMapper modelMapper = new ModelMapper();
        modelMapper.getConfiguration().setMatchingStrategy(MatchingStrategies.STRICT);
        List<AppResponse> appResponses = appService.getAll();
        returnValue = modelMapper.map(
                appResponses ,
                new TypeToken<List<AppDto>>() {}.getType()
        );
        Gson gson = new GsonBuilder().setPrettyPrinting().create();
        String json = gson.toJson(returnValue);

        System.out.println(json);
        return returnValue;
    }
}

我使用了一个组件类来传递数据库名：

@Component
@Scope(value = ConfigurableBeanFactory.SCOPE_SINGLETON)
public class DataSourceContextHolder {
    private static ThreadLocal<String> threadLocal;

    public DataSourceContextHolder() {
        threadLocal = new ThreadLocal<>();
    }

    public void setBranchContext(String dbName) {
        threadLocal.set(dbName);
    }

    public String getBranchContext() {
        return threadLocal.get();
    }

    public static void clearBranchContext() {
        threadLocal.remove();
    }
}

我还创建了一个在启动数据库时处理的DataSourceRouting类：

@Component
public class DataSourceRouting extends AbstractRoutingDataSource {
    private DataSourceContextHolder dataSourceContextHolder;
    private DataSourceConfig dataSourceConfig;

    public DataSourceRouting(DataSourceContextHolder dataSourceContextHolder,
                             DataSourceConfig dataSourceConfig) throws IOException {
        this.dataSourceContextHolder = dataSourceContextHolder;
        this.dataSourceConfig = dataSourceConfig;

        Map<Object, Object> dataSourceMap = new HashMap<>();
        Set<String> databases = Utils.getDatabaseList();
        databases.forEach(database -> {
            DataSource dataSource = dataSourceBuilder(database);
            dataSourceMap.put(database, dataSource);
            this.setDefaultTargetDataSource(dataSource);

        });
        this.setTargetDataSources(dataSourceMap);

    }

    private DataSource dataSourceBuilder(String database) {
        DriverManagerDataSource dataSource = new DriverManagerDataSource();
        dataSource.setDriverClassName(dataSourceConfig.getDriverClassName());
        dataSource.setUrl(dataSourceConfig.getUrl() + database);
        dataSource.setUsername(dataSourceConfig.getUsername());
        dataSource.setPassword(dataSourceConfig.getPassword());
        return dataSource;
    }

    @Override
    protected Object determineCurrentLookupKey() {
        //System.out.println(dataSourceContextHolder.getBranchContext().toString());
        return dataSourceContextHolder.getBranchContext();
    }
}

我创建了一个在application.properties中读取的POJO：

@Component
@ConfigurationProperties(prefix="datasource")
@Getter @Setter
public class DataSourceConfig {
    private String url;
    private String password;
    private String username;
    private String driverClassName;
}

此代码允许传递数据库名称并对正确的名称进行查询。由于数据库具有相同的结构，所以我确实按照预期输出了正确的值。