[Stuck] - Change schema after log-in

[Castiel Snow]

Problem

Stuck on implementing change of schema after user login, following a StackOverFlow.

Description


Im following the stackoverflow solution at: https://stackoverflow.com/questions/39357367/change-database-schema-during-runtime-based-on-logged-in-user

The spring version im using is: :

   : Spring Boot ::        (v2.3.3.RELEASE)

I've created the class below. But I have no idea on how to use it properly.

What im trying to achieve is:

Springboot initializes in a default login database URL.
After login, it should use the class below to define which schema it should use based on the UserDetails class.

Im stuck. Im reading every tutorial, book I can find but im stuck. Please helpp

       import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import java.sql.Connection;
import java.sql.ConnectionBuilder;
import java.sql.SQLException;
import java.util.concurrent.TimeUnit;
import javax.sql.DataSource;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.core.env.Environment;
import org.springframework.jdbc.datasource.AbstractDataSource;


public class UserSchemaAwareRoutingDataSource extends AbstractDataSource {

    @Autowired
    UsuarioProvider customUserDetails;

    @Autowired
    Environment env;

    private LoadingCache<String, DataSource> dataSources = createCache();

    public UserSchemaAwareRoutingDataSource() {
    }

    public UserSchemaAwareRoutingDataSource(UsuarioProvider customUserDetails, Environment env) {
        this.customUserDetails = customUserDetails;
        this.env = env;
    }

    private LoadingCache<String, DataSource> createCache() {
        return CacheBuilder.newBuilder()
                .maximumSize(100)
                .expireAfterWrite(10, TimeUnit.MINUTES)
                .build(
                        new CacheLoader<String, DataSource>() {
                    public DataSource load(String key) throws Exception {
                        return buildDataSourceForSchema(key);
                    }
                });
    }

    private DataSource buildDataSourceForSchema(String schema) {
        System.out.println("schema:" + schema);
        String url = "jdbc:mysql://REDACTED.com/" + schema;
        String username = env.getRequiredProperty("spring.datasource.username");
        String password = env.getRequiredProperty("spring.datasource.password");

        System.out.println("Flag A");

        DataSource build = (DataSource) DataSourceBuilder.create()
                .driverClassName(env.getRequiredProperty("spring.datasource.driverClassName"))
                .username(username)
                .password(password)
                .url(url)
                .build();

        System.out.println("Flag B");

        return build;
    }

    @Override
    public Connection getConnection() throws SQLException {
        return determineTargetDataSource().getConnection();
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return determineTargetDataSource().getConnection(username, password);
    }

    private DataSource determineTargetDataSource() {
        try {
            Usuario usuario = customUserDetails.customUserDetails();
            //
            String db_schema = usuario.getTunnel().getDb_schema();
            //

            String schema = db_schema;
            return dataSources.get(schema);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return null;
    }

    @Override
    public ConnectionBuilder createConnectionBuilder() throws SQLException {
        return super.createConnectionBuilder();
    }

}