How to use Spring AbstractRoutingDataSource with dynamic datasources?

by

Nothing in AbstractRoutingDataSource forces you to use a static map of DataSourceS. It is up to you to contruct a bean implementing Map<Object, Object>, where key is what you use to select the DataSource, and value is a DataSource or (by default) a String referencing a JNDI defined data source. You can even modify it dynamically since, as the map is stored in memory, AbstractRoutingDataSource does no caching.

I have no full example code. But here is what I can imagine. In a web application, you have one database per client, all with same structure – ok, it would be a strange design, say it is just for the example. At login time, the application creates the datasource for the client and stores it in a map indexed by sessionId – The map is a bean in root context named dataSources

@Autowired
@Qualifier("dataSources");
Map<String, DataSource> sources;

// I assume url, user and password have been found from connected user
// I use DriverManagerDataSource for the example because it is simple to setup
DataSource dataSource = new DriverManagerDataSource(url, user, password);
sources.put(request.getSession.getId(), dataSource);

You also need a session listener to cleanup dataSources in its destroy method

@Autowired
@Qualifier("dataSources");
Map<String, DataSource> sources;

public void sessionDestroyed(HttpSessionEvent se)  {
    // eventually cleanup the DataSource if appropriate (nothing to do for DriverManagerDataSource ...)
    sources.remove(se.getSession.getId());
}

The routing datasource could be like :

public class SessionRoutingDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        HttpServletRequest request = ((ServletRequestAttributes)
                RequestContextHolder.getRequestAttributes()).getRequest();
        return request.getSession().getId();
    }

    @Autowired
    @Qualifier("dataSources")
    public void setDataSources(Map<String, DataSource> dataSources) {
        setTargetDataSources(dataSources);
}

I have not tested anything because it would be a lot of work to setting the different database, but I thing that it should be Ok. In real world there would not be a different data source per session but one per user with a count of session per user but as I said it is an over simplified example.

More Related Contents:

  1. Joining two table entities in Spring Data JPA
  2. Spring data JPA query with parameter properties
  3. Spring Data JPA – Multiple EnableJpaRepositories
  4. Set default page size for JPA Pageable Object
  5. How does spring.jpa.hibernate.ddl-auto property exactly work in Spring?
  6. setMaxResults for Spring-Data-JPA annotation?
  7. How does the FetchMode work in Spring Data JPA
  8. Handle spring security authentication exceptions with @ExceptionHandler
  9. Handling soft-deletes with Spring JPA
  10. How do I get the Session Object in Spring?
  11. Spring Boot CORS filter – CORS preflight channel did not succeed
  12. How can I have list of all users logged in (via spring security) my web application
  13. Spring Security exclude url patterns in security annotation configurartion
  14. Spring Security 3.2 CSRF support for multipart requests
  15. Spring CSRF token does not work, when the request to be sent is a multipart request
  16. Spring-Data FETCH JOIN with Paging is not working
  17. Why use returned instance after save() on Spring Data JPA Repository?
  18. Use abstract super class as parameter for Spring data repository
  19. How to use @Transactional with Spring Data?
  20. Spring Security and @Async (Authenticated Users mixed up)
  21. Spring-boot: required a bean named ‘entityManagerFactory’ that could not be found
  22. Spring @Transactional with a transaction across multiple data sources
  23. Spring Security: mapping OAuth2 claims with roles to secure Resource Server endpoints
  24. How to use multiple login pages one for admin and the other one for user
  25. @Autowired vs @PersistenceContext for EntityManager bean
  26. Spring Social Authentication Filter for Stateless REST Endpoints which use Facebook Token for authentication
  27. How to enable LockModeType.PESSIMISTIC_WRITE when looking up entities with Spring Data JPA?
  28. Spring Data + JPA with multiple datasources but only one set of Repositories
  29. Spring security 4 custom login j_spring_security_check return http 302
  30. LazyInitializationException in JPA and Hibernate

Leave a Comment