github/ebean
4
0
Fork 0
mirror of https://github.com/ebean-orm/ebean.git synced 2024-04-21 10:51:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: For SpringJdbcTransactionManager support using CurrentTenantProvider (#3368)

Browse Source 
Note that this only works for Multi-Tenant PARTITION mode where the TenantId is a column on the tables (rather than schema or database multi-tenancy mode).
This commit is contained in:
Rob Bygrave
2024-03-26 22:45:39 +13:00
committed by GitHub
parent ab45ad34e0
commit 7adbdbf532
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ebean-spring-txn/src/main/java/io/ebean/spring/txn/SpringJdbcTransactionManager.java
+20 -1
View File
@@ -2,6 +2,7 @@ package io.ebean.spring.txn;
import io.avaje.applog.AppLog;
import io.ebean.TxScope;
import io.ebean.config.CurrentTenantProvider;
import io.ebean.config.ExternalTransactionManager;
import io.ebeaninternal.api.SpiTransaction;
import io.ebeaninternal.server.transaction.TransactionManager;
@@ -27,13 +28,25 @@ public final class SpringJdbcTransactionManager implements ExternalTransactionMa
private static final System.Logger log = AppLog.getLogger(SpringJdbcTransactionManager.class);
private final CurrentTenantProvider tenantProvider;
private DataSource dataSource;
private TransactionManager transactionManager;
/**
* Instantiate with a CurrentTenantProvider for Multi-Tenant support.
* <p>
* When a new transaction is detected, this will set the TenantId on that transaction.
*/
public SpringJdbcTransactionManager(CurrentTenantProvider tenantProvider) {
this.tenantProvider = tenantProvider;
}
/**
* Instantiates a new spring aware transaction scope manager.
*/
public SpringJdbcTransactionManager() {
this.tenantProvider = null;
}
/**
@@ -73,7 +86,13 @@ public final class SpringJdbcTransactionManager implements ExternalTransactionMa
} else {
// This is a new spring transaction that we have not seen before.
// "wrap" it in a SpringJdbcTransaction for use with Ebean
SpringJdbcTransaction newTrans = new SpringJdbcTransaction(holder, transactionManager);
final var newTrans = new SpringJdbcTransaction(holder, transactionManager);
if (tenantProvider != null) {
final var tenantId = tenantProvider.currentId();
if (tenantId != null) {
newTrans.setTenantId(tenantId);
}
}
// Create and register a Spring TransactionSynchronization for this transaction
springTxnLister = createListener(newTrans);