From cca012619ab220f1b79d3d52fbb5bb42f47e44a8 Mon Sep 17 00:00:00 2001
From: seepine <seepine@outlook.com>
Date: Wed, 23 Aug 2023 21:53:05 +0800
Subject: [PATCH] Support for multi tenancy root mode #35424

---
 .../HibernateCurrentTenantIdentifierResolver.java | 13 +++++++++++++
 .../orm/runtime/tenant/TenantResolver.java        | 11 +++++++++++
 .../multitenancy/fruit/CustomTenantResolver.java  |  7 +++++++
 .../fruit/HibernateTenancyFunctionalityTest.java  | 15 +++++++++++++++
 4 files changed, 46 insertions(+)

diff --git a/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/HibernateCurrentTenantIdentifierResolver.java b/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/HibernateCurrentTenantIdentifierResolver.java
index c7b2fdf6389db..1989224a06406 100644
--- a/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/HibernateCurrentTenantIdentifierResolver.java
+++ b/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/HibernateCurrentTenantIdentifierResolver.java
@@ -50,6 +50,19 @@ public boolean validateExistingCurrentSessions() {
         return false;
     }
 
+    @Override
+    public boolean isRoot(String tenantId) {
+        // Make sure that we're in a request
+        if (!Arc.container().requestContext().isActive()) {
+            return false;
+        }
+        TenantResolver resolver = tenantResolver(persistenceUnitName);
+        if (resolver == null) {
+            return false;
+        }
+        return resolver.isRoot(tenantId);
+    }
+
     private static TenantResolver tenantResolver(String persistenceUnitName) {
         InjectableInstance<TenantResolver> instance = PersistenceUnitUtil.legacySingleExtensionInstanceForPersistenceUnit(
                 TenantResolver.class, persistenceUnitName);
diff --git a/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/TenantResolver.java b/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/TenantResolver.java
index 3e628625c5c00..39d581b4f5264 100644
--- a/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/TenantResolver.java
+++ b/extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/tenant/TenantResolver.java
@@ -23,4 +23,15 @@ public interface TenantResolver {
      */
     String resolveTenantId();
 
+    /**
+     * Does the given tenant id represent a "root" tenant with access to all partitions?
+     *
+     * @param tenantId a tenant id produced by {@link #resolveTenantId()}
+     *
+     * @return true is this is root tenant
+     */
+    default boolean isRoot(String tenantId) {
+        return false;
+    }
+
 }
diff --git a/integration-tests/hibernate-orm-tenancy/discriminator/src/main/java/io/quarkus/it/hibernate/multitenancy/fruit/CustomTenantResolver.java b/integration-tests/hibernate-orm-tenancy/discriminator/src/main/java/io/quarkus/it/hibernate/multitenancy/fruit/CustomTenantResolver.java
index 424bb8087c0c3..c4c910189850a 100644
--- a/integration-tests/hibernate-orm-tenancy/discriminator/src/main/java/io/quarkus/it/hibernate/multitenancy/fruit/CustomTenantResolver.java
+++ b/integration-tests/hibernate-orm-tenancy/discriminator/src/main/java/io/quarkus/it/hibernate/multitenancy/fruit/CustomTenantResolver.java
@@ -29,6 +29,8 @@ public String resolveTenantId() {
         final String tenantId;
         if (path.startsWith("/mycompany")) {
             tenantId = "mycompany";
+        } else if (path.startsWith("/global")) {
+            tenantId = "global";
         } else {
             tenantId = getDefaultTenantId();
         }
@@ -36,4 +38,9 @@ public String resolveTenantId() {
         return tenantId;
     }
 
+    @Override
+    public boolean isRoot(String tenantId) {
+        return "global".equals(tenantId);
+    }
+
 }
diff --git a/integration-tests/hibernate-orm-tenancy/discriminator/src/test/java/io/quarkus/it/hibernate/multitenancy/fruit/HibernateTenancyFunctionalityTest.java b/integration-tests/hibernate-orm-tenancy/discriminator/src/test/java/io/quarkus/it/hibernate/multitenancy/fruit/HibernateTenancyFunctionalityTest.java
index 6d3090f0d0208..ba4556fc144a0 100644
--- a/integration-tests/hibernate-orm-tenancy/discriminator/src/test/java/io/quarkus/it/hibernate/multitenancy/fruit/HibernateTenancyFunctionalityTest.java
+++ b/integration-tests/hibernate-orm-tenancy/discriminator/src/test/java/io/quarkus/it/hibernate/multitenancy/fruit/HibernateTenancyFunctionalityTest.java
@@ -4,10 +4,13 @@
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.arrayContaining;
 import static org.hamcrest.Matchers.arrayWithSize;
+import static org.hamcrest.Matchers.hasItems;
 import static org.hamcrest.Matchers.is;
 import static org.hamcrest.Matchers.not;
 import static org.hamcrest.Matchers.nullValue;
 
+import java.util.Arrays;
+
 import jakarta.ws.rs.core.Response.Status;
 
 import org.junit.jupiter.api.BeforeEach;
@@ -147,6 +150,18 @@ public void testUpdateFruitDefaultTenant() throws Exception {
 
     }
 
+    @Test
+    public void testGetFruitsTenantRoot() throws Exception {
+
+        // Get all tenant fruits
+        Fruit[] fruits = given().when().get("/global/fruits").then().assertThat()
+                .statusCode(is(Status.OK.getStatusCode())).extract()
+                .as(Fruit[].class);
+        assertThat(fruits, arrayWithSize(6));
+        assertThat(Arrays.asList(fruits), hasItems(new Fruit(4, "Avocado"), new Fruit(1, "Cherry")));
+
+    }
+
     private Fruit findByName(String tenantPath, String name) {
         Response response = given().when().get(tenantPath + "/fruitsFindBy?type=name&value={name}", name);
         if (response.getStatusCode() == Status.OK.getStatusCode()) {