Quarkus Multitenancy not referencing Connection Provider (even with explicit hibernate property setting)

[harikrishnajiju]

I am trying to integrate multitenancy to my Quarkus application. I have been facing some difficulties.

PFA my classes and logs for a minimalistic multitenancy project I setup:

Hibernate Config:

`package com.sc.faas.multitenant;

import io.quarkus.hibernate.orm.PersistenceUnitExtension;
import io.quarkus.runtime.Startup;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
import org.hibernate.cfg.Environment;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

@ApplicationScoped
@startup
public class HibernateConfig {

@Inject
DataSource dataSource;

@PersistenceUnitExtension
public Map<String, Object> hibernateProperties() {
    Map<String, Object> properties = new HashMap<>();
    properties.put("hibernate.multiTenancy", "SCHEMA");
    properties.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, new CustomMultiTenantConnectionProvider(dataSource));
    properties.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, new SettlementTenantResolver());
    return properties;
}

}`

Tenant Context

`package com.sc.faas.multitenant;

import jakarta.enterprise.context.ApplicationScoped;
import lombok.extern.slf4j.Slf4j;

@ApplicationScoped
@slf4j
public class TenantContext {
private static final InheritableThreadLocal CURRENT_TENANT = new InheritableThreadLocal<>();

public static String getCurrentTenantId() {
    log.info("Getting Current tenant: {}", CURRENT_TENANT.get());
    return CURRENT_TENANT.get();
}

public static void setCurrentTenantId(String tenantId) {

    log.info("Setting Current tenant: {}", tenantId);
    CURRENT_TENANT.set(tenantId);
}

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

}`

TenantResolver

`package com.sc.faas.multitenant;

import io.quarkus.hibernate.orm.runtime.tenant.TenantResolver;
import jakarta.enterprise.context.ApplicationScoped;
import lombok.extern.slf4j.Slf4j;

@ApplicationScoped
@slf4j
public class SettlementTenantResolver implements TenantResolver {

@Override
public String getDefaultTenantId() {
    log.info("Get defualt tenant Id method invoced, returning: public");
    return "public";
}

@Override
public String resolveTenantId() {
    log.info("Resolving Tenant Id: {}", TenantContext.getCurrentTenantId());
    String tenantId = TenantContext.getCurrentTenantId();
    tenantId = tenantId != null ? tenantId : getDefaultTenantId();
    log.info("Tenant Id resolved: {}", tenantId);
    return tenantId;

}

}`

Connection Provider

`package com.sc.faas.multitenant;

import lombok.extern.slf4j.Slf4j;
import org.hibernate.engine.jdbc.connections.spi.MultiTenantConnectionProvider;
import org.hibernate.service.spi.Wrapped;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

@slf4j
public class CustomMultiTenantConnectionProvider implements MultiTenantConnectionProvider {

private final DataSource defaultDataSource;

public CustomMultiTenantConnectionProvider(DataSource defaultDataSource) {
    this.defaultDataSource = defaultDataSource;
}


@Override
public Connection getConnection(Object o) throws SQLException {

    String tenantId = (String) o;
    log.info("Get Connection Method invoked, tenant ID = {}", tenantId);
    Connection connection = defaultDataSource.getConnection();
    connection.setSchema(tenantId);
    return connection;
}

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

@Override
public void releaseAnyConnection(Connection connection) throws SQLException {
    log.info("ReleaseAnyConnection method invoked");
    log.info("Current Connection schema: {}", connection.getSchema());
    connection.close();
}

@Override
public void releaseConnection(Object o, Connection connection) throws SQLException {
    log.info("ReleaseAnyConnection method invoked");
    log.info("Current Connection schema: {}", connection.getSchema());
    connection.close();
}

@Override
public boolean supportsAggressiveRelease() {
    return false;
}


@Override
public boolean isUnwrappableAs(Class unwrapType) {
    return MultiTenantConnectionProvider.class.isAssignableFrom(unwrapType) || Wrapped.class.isAssignableFrom(unwrapType);
}

@Override
public <T> T unwrap(Class<T> unwrapType) {
    if (isUnwrappableAs(unwrapType)) {
        return (T) this;
    } else {
        throw new IllegalArgumentException("Unknown unwrap type: " + unwrapType);
    }
}

}`

RequestResolver

`package com.sc.faas.multitenant;

import jakarta.ws.rs.container.ContainerRequestContext;
import jakarta.ws.rs.container.ContainerRequestFilter;
import jakarta.ws.rs.core.PathSegment;
import jakarta.ws.rs.core.UriInfo;
import jakarta.ws.rs.ext.Provider;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.util.List;

@Provider
@slf4j
public class TenantRequestFilter implements ContainerRequestFilter {

private static final String TENANT_ID_HEADER = "X-Tenant-ID";

@Override
public void filter(ContainerRequestContext requestContext) throws IOException {
    String tenantId = null;

    // Try to get tenant ID from header first
    tenantId = requestContext.getHeaderString(TENANT_ID_HEADER);
    log.info("Tenant ID from header: {}", tenantId);

    // If header not present, try to extract from URL Path
    if (tenantId == null || tenantId.isEmpty()) {
        UriInfo uriInfo = requestContext.getUriInfo();
        List<PathSegment> pathSegments = uriInfo.getPathSegments();

        // Assuming URL format: https://domain.com/{tenantId}/rest...
        if (!pathSegments.isEmpty()) {
            tenantId = pathSegments.get(0).getPath();
            log.info("Tenant ID from URL: {}", tenantId);
        }
    }

    if (tenantId != null && !tenantId.isEmpty()) {
        log.info("Setting tenant ID: {}", tenantId);
        TenantContext.setCurrentTenantId(tenantId);
    } else {
        log.warn("No tenant ID found in request");
    }
}

}`

Response Filter

`package com.sc.faas.multitenant;

import jakarta.ws.rs.container.ContainerRequestContext;
import jakarta.ws.rs.container.ContainerResponseContext;
import jakarta.ws.rs.container.ContainerResponseFilter;
import jakarta.ws.rs.ext.Provider;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;

@Provider
@slf4j
public class TenantResponseFilter implements ContainerResponseFilter {

@Override
public void filter(ContainerRequestContext requestContext,
                   ContainerResponseContext responseContext) throws IOException {
    log.info("Cleaning tenant context");
    TenantContext.clear();
}

}`

Product Controller:

`@Path("/products")
@produces(MediaType.APPLICATION_JSON)
@consumes(MediaType.APPLICATION_JSON)
public class ProductController {
private static final Logger log = LoggerFactory.getLogger(ProductController.class);

@Inject
ProductRepository productRepository;

@GET
@Path("/tenant-info")
public Response getTenantInfo() {
    String currentTenant = TenantContext.getCurrentTenantId();
    return Response.ok("Current tenant: " + (currentTenant != null ? currentTenant : "default")).build();
}`

PFA the logs below hitting the tenant-info endpoint:
2025-05-17 22:58:37,447 INFO [com.sc.faa.mul.TenantRequestFilter] (executor-thread-1) Tenant ID from header: hkd 2025-05-17 22:58:37,448 INFO [com.sc.faa.mul.TenantRequestFilter] (executor-thread-1) Setting tenant ID: hkd 2025-05-17 22:58:37,448 INFO [com.sc.faa.mul.TenantContext] (executor-thread-1) Setting Current tenant: hkd 2025-05-17 22:58:37,448 INFO [com.sc.faa.mul.TenantContext] (executor-thread-1) Getting Current tenant: hkd 2025-05-17 22:58:37,448 INFO [com.sc.faa.mul.TenantResponseFilter] (executor-thread-1) Cleaning tenant context

So the issue is that, the connection provider is not called when the API endpoint is hit. This is creating issues on the actual application I am workingon, please help.

[quarkus-bot[bot]]

/cc @gsmet (hibernate-orm)