@Component
public class TenantInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        String tenantId = request.getHeader("X-TenantID");
        if (tenantId != null) {
            TenantContext.setCurrentTenant(tenantId);
        }
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        TenantContext.clear();
    }
}

public class TenantContext {
    private static final ThreadLocal<String> currentTenant = new ThreadLocal<>();

    public static void setCurrentTenant(String tenantId) {
        currentTenant.set(tenantId);
    }

    public static String getCurrentTenant() {
        return currentTenant.get();
    }

    public static void clear() {
        currentTenant.remove();
    }
}

public class MultiTenantDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return TenantContext.getCurrentTenant();
    }
}

@Configuration
public class DataSourceConfig {

    @Bean
    public DataSource dataSource() {
        MultiTenantDataSource dataSource = new MultiTenantDataSource();

        Map<Object, Object> targetDataSources = new HashMap<>();
        // 假設(shè)已經(jīng)有一個(gè)方法來獲取所有租戶的數(shù)據(jù)源配置
        Map<String, DataSource> tenantDataSources = tenantDataSources();
        targetDataSources.putAll(tenantDataSources);

        dataSource.setTargetDataSources(targetDataSources);
        dataSource.setDefaultTargetDataSource(defaultDataSource()); // 默認(rèn)數(shù)據(jù)源
        dataSource.afterPropertiesSet();
        return dataSource;
    }

    private Map<String, DataSource> tenantDataSources() {
        // 這里應(yīng)該從配置文件或數(shù)據(jù)庫中加載所有租戶的數(shù)據(jù)源配置
        // 舉例，假設(shè)有兩個(gè)租戶tenantA和tenantB
        Map<String, DataSource> result = new HashMap<>();
        result.put("tenantA", createDataSource("jdbc:mysql://localhost:3306/tenantA"));
        result.put("tenantB", createDataSource("jdbc:mysql://localhost:3306/tenantB"));
        return result;
    }

    private DataSource createDataSource(String url) {
        DriverManagerDataSource dataSource = new DriverManagerDataSource();
        dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
        dataSource.setUrl(url);
        dataSource.setUsername("username");
        dataSource.setPassword("password");
        return dataSource;
    }

    private DataSource defaultDataSource() {
        // 定義默認(rèn)數(shù)據(jù)源
        return createDataSource("jdbc:mysql://localhost:3306/defaultDb");
    }
}

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {

    @Autowired
    private TenantInterceptor tenantInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(tenantInterceptor);
    }
}