這篇文章將為大家詳細(xì)講解有關(guān)sharding-jdbc中ShardingTransactionManager有什么用,文章內(nèi)容質(zhì)量較高,因此小編分享給大家做個(gè)參考,希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。

十多年的子洲網(wǎng)站建設(shè)經(jīng)驗(yàn),針對(duì)設(shè)計(jì)、前端、開發(fā)、售后、文案、推廣等六對(duì)一服務(wù),響應(yīng)快,48小時(shí)及時(shí)工作處理。營(yíng)銷型網(wǎng)站建設(shè)的優(yōu)勢(shì)是能夠根據(jù)用戶設(shè)備顯示端的尺寸不同,自動(dòng)調(diào)整子洲建站的顯示方式,使網(wǎng)站能夠適用不同顯示終端,在瀏覽器中調(diào)整網(wǎng)站的寬度,無(wú)論在任何一種瀏覽器上瀏覽網(wǎng)站,都能展現(xiàn)優(yōu)雅布局與設(shè)計(jì),從而大程度地提升瀏覽體驗(yàn)。成都創(chuàng)新互聯(lián)從事“子洲網(wǎng)站設(shè)計(jì)”,“子洲網(wǎng)站推廣”以來(lái),每個(gè)客戶項(xiàng)目都認(rèn)真落實(shí)執(zhí)行。
incubator-shardingsphere-4.0.0-RC1/sharding-transaction/sharding-transaction-core/src/main/java/org/apache/shardingsphere/transaction/spi/ShardingTransactionManager.java
public interface ShardingTransactionManager extends AutoCloseable {
/**
* Initialize sharding transaction manager.
*
* @param databaseType database type
* @param resourceDataSources resource data sources
*/
void init(DatabaseType databaseType, Collection<ResourceDataSource> resourceDataSources);
/**
* Get transaction type.
*
* @return transaction type
*/
TransactionType getTransactionType();
/**
* Judge is in transaction or not.
*
* @return in transaction or not
*/
boolean isInTransaction();
/**
* Get transactional connection.
*
* @param dataSourceName data source name
* @return connection
* @throws SQLException SQL exception
*/
Connection getConnection(String dataSourceName) throws SQLException;
/**
* Begin transaction.
*/
void begin();
/**
* Commit transaction.
*/
void commit();
/**
* Rollback transaction.
*/
void rollback();
}ShardingTransactionManager繼承了AutoCloseable接口,它定義了init、getTransactionType、isInTransaction、getConnection、begin、commit、rollback方法
incubator-shardingsphere-4.0.0-RC1/sharding-transaction/sharding-transaction-2pc/sharding-transaction-xa/sharding-transaction-xa-core/src/main/java/org/apache/shardingsphere/transaction/xa/XAShardingTransactionManager.java
public final class XAShardingTransactionManager implements ShardingTransactionManager {
private final Map<String, SingleXADataSource> singleXADataSourceMap = new HashMap<>();
private final XATransactionManager xaTransactionManager = XATransactionManagerLoader.getInstance().getTransactionManager();
private ThreadLocal<List<String>> enlistedXAResource = new ThreadLocal<List<String>>() {
@Override
public List<String> initialValue() {
return new LinkedList<>();
}
};
@Override
public void init(final DatabaseType databaseType, final Collection<ResourceDataSource> resourceDataSources) {
for (ResourceDataSource each : resourceDataSources) {
DataSource dataSource = each.getDataSource();
if (dataSource instanceof AtomikosDataSourceBean) {
continue;
}
SingleXADataSource singleXADataSource = new SingleXADataSource(databaseType, each.getUniqueResourceName(), dataSource);
singleXADataSourceMap.put(each.getOriginalName(), singleXADataSource);
xaTransactionManager.registerRecoveryResource(each.getUniqueResourceName(), singleXADataSource.getXaDataSource());
}
xaTransactionManager.init();
}
@Override
public TransactionType getTransactionType() {
return TransactionType.XA;
}
@SneakyThrows
@Override
public boolean isInTransaction() {
return Status.STATUS_NO_TRANSACTION != xaTransactionManager.getTransactionManager().getStatus();
}
@SneakyThrows
@Override
public Connection getConnection(final String dataSourceName) {
SingleXAConnection singleXAConnection = singleXADataSourceMap.get(dataSourceName).getXAConnection();
if (!enlistedXAResource.get().contains(dataSourceName)) {
xaTransactionManager.enlistResource(singleXAConnection.getXAResource());
enlistedXAResource.get().add(dataSourceName);
}
return singleXAConnection.getConnection();
}
@SneakyThrows
@Override
public void begin() {
xaTransactionManager.getTransactionManager().begin();
}
@SneakyThrows
@Override
public void commit() {
try {
xaTransactionManager.getTransactionManager().commit();
} finally {
enlistedXAResource.remove();
}
}
@SneakyThrows
@Override
public void rollback() {
try {
xaTransactionManager.getTransactionManager().rollback();
} finally {
enlistedXAResource.remove();
}
}
@Override
public void close() throws Exception {
for (SingleXADataSource each : singleXADataSourceMap.values()) {
xaTransactionManager.removeRecoveryResource(each.getResourceName(), each.getXaDataSource());
}
singleXADataSourceMap.clear();
xaTransactionManager.close();
enlistedXAResource = null;
}
}XAShardingTransactionManager實(shí)現(xiàn)了ShardingTransactionManager接口,它擁有singleXADataSourceMap、xaTransactionManager、enlistedXAResource三個(gè)屬性,其init方法對(duì)這三個(gè)屬性進(jìn)行了初始化
getTransactionType方法返回的是TransactionType.XA;isInTransaction方法則依據(jù)xaTransactionManager.getTransactionManager().getStatus()來(lái)判斷;getConnection方法從singleXADataSourceMap取出singleXAConnection,然后執(zhí)行其getConnection方法
begin方法執(zhí)行的是xaTransactionManager.getTransactionManager().begin();commit方法則調(diào)用了xaTransactionManager.getTransactionManager().commit(),其會(huì)在finally中執(zhí)行enlistedXAResource.remove();rollback方法調(diào)用了xaTransactionManager.getTransactionManager().rollback(),其也在finally中執(zhí)行enlistedXAResource.remove();close方法則執(zhí)行xaTransactionManager.removeRecoveryResource、singleXADataSourceMap.clear()、xaTransactionManager.close()
incubator-shardingsphere-4.0.0-RC1/sharding-transaction/sharding-transaction-2pc/sharding-transaction-xa/sharding-transaction-xa-core/src/test/java/org/apache/shardingsphere/transaction/xa/XAShardingTransactionManagerTest.java
@RunWith(MockitoJUnitRunner.class)
public final class XAShardingTransactionManagerTest {
private XAShardingTransactionManager xaShardingTransactionManager = new XAShardingTransactionManager();
@Mock
private XATransactionManager xaTransactionManager;
@Mock
private TransactionManager transactionManager;
@Before
public void setUp() {
when(xaTransactionManager.getTransactionManager()).thenReturn(transactionManager);
ReflectiveUtil.setProperty(xaShardingTransactionManager, "xaTransactionManager", xaTransactionManager);
}
@Test
public void assertGetTransactionType() {
assertThat(xaShardingTransactionManager.getTransactionType(), is(TransactionType.XA));
}
@Test
public void assertRegisterXATransactionalDataSources() {
Collection<ResourceDataSource> resourceDataSources = createResourceDataSources(DruidXADataSource.class, DatabaseType.MySQL);
xaShardingTransactionManager.init(DatabaseType.MySQL, resourceDataSources);
for (ResourceDataSource each : resourceDataSources) {
verify(xaTransactionManager).registerRecoveryResource(each.getUniqueResourceName(), (XADataSource) each.getDataSource());
}
}
@Test
public void assertRegisterAtomikosDataSourceBeans() {
xaShardingTransactionManager.init(DatabaseType.MySQL, createAtomikosDataSourceBeanResource());
verify(xaTransactionManager, times(0)).registerRecoveryResource(anyString(), any(XADataSource.class));
}
@Test
public void assertRegisterNoneXATransactionalDAtaSources() {
Collection<ResourceDataSource> resourceDataSources = createResourceDataSources(HikariDataSource.class, DatabaseType.MySQL);
xaShardingTransactionManager.init(DatabaseType.MySQL, resourceDataSources);
Map<String, SingleXADataSource> cachedXADatasourceMap = getCachedSingleXADataSourceMap();
assertThat(cachedXADatasourceMap.size(), is(2));
}
@Test
public void assertIsInTransaction() throws SystemException {
when(transactionManager.getStatus()).thenReturn(Status.STATUS_ACTIVE);
assertTrue(xaShardingTransactionManager.isInTransaction());
}
@Test
public void assertIsNotInTransaction() throws SystemException {
when(transactionManager.getStatus()).thenReturn(Status.STATUS_NO_TRANSACTION);
assertFalse(xaShardingTransactionManager.isInTransaction());
}
@Test
public void assertGetConnection() {
setCachedSingleXADataSourceMap("ds1");
Connection actual = xaShardingTransactionManager.getConnection("ds1");
assertThat(actual, instanceOf(Connection.class));
verify(xaTransactionManager).enlistResource(any(SingleXAResource.class));
}
@Test
public void assertGetConnectionWithoutEnlist() {
setCachedSingleXADataSourceMap("ds1");
Connection actual = xaShardingTransactionManager.getConnection("ds1");
assertThat(actual, instanceOf(Connection.class));
xaShardingTransactionManager.getConnection("ds1");
assertThat(actual, instanceOf(Connection.class));
verify(xaTransactionManager).enlistResource(any(SingleXAResource.class));
}
@Test
public void assertClose() throws Exception {
setCachedSingleXADataSourceMap("ds1");
xaShardingTransactionManager.close();
Map<String, SingleXADataSource> cachedSingleXADataSourceMap = getCachedSingleXADataSourceMap();
verify(xaTransactionManager).removeRecoveryResource(anyString(), any(XADataSource.class));
assertThat(cachedSingleXADataSourceMap.size(), is(0));
}
@SneakyThrows
@SuppressWarnings("unchecked")
private Map<String, SingleXADataSource> getCachedSingleXADataSourceMap() {
Field field = xaShardingTransactionManager.getClass().getDeclaredField("singleXADataSourceMap");
field.setAccessible(true);
return (Map<String, SingleXADataSource>) field.get(xaShardingTransactionManager);
}
@SneakyThrows
private void setCachedSingleXADataSourceMap(final String datasourceName) {
Field field = xaShardingTransactionManager.getClass().getDeclaredField("singleXADataSourceMap");
field.setAccessible(true);
field.set(xaShardingTransactionManager, createMockSingleXADataSourceMap(datasourceName));
}
@SneakyThrows
private Map<String, SingleXADataSource> createMockSingleXADataSourceMap(final String datasourceName) {
SingleXADataSource singleXADataSource = mock(SingleXADataSource.class);
SingleXAConnection singleXAConnection = mock(SingleXAConnection.class);
XADataSource xaDataSource = mock(XADataSource.class);
SingleXAResource singleXAResource = mock(SingleXAResource.class);
Connection connection = mock(Connection.class);
when(singleXAConnection.getConnection()).thenReturn(connection);
when(singleXAConnection.getXAResource()).thenReturn(singleXAResource);
when(singleXADataSource.getXAConnection()).thenReturn(singleXAConnection);
when(singleXADataSource.getResourceName()).thenReturn(datasourceName);
when(singleXADataSource.getXaDataSource()).thenReturn(xaDataSource);
Map<String, SingleXADataSource> result = new HashMap<>();
result.put(datasourceName, singleXADataSource);
return result;
}
private Collection<ResourceDataSource> createResourceDataSources(final Class<? extends DataSource> dataSourceClass, final DatabaseType databaseType) {
List<ResourceDataSource> result = new LinkedList<>();
result.add(new ResourceDataSource("ds1", DataSourceUtils.build(dataSourceClass, databaseType, "demo_ds_1")));
result.add(new ResourceDataSource("ds2", DataSourceUtils.build(dataSourceClass, databaseType, "demo_ds_2")));
return result;
}
private Collection<ResourceDataSource> createAtomikosDataSourceBeanResource() {
List<ResourceDataSource> result = new LinkedList<>();
result.add(new ResourceDataSource("ds1", new AtomikosDataSourceBean()));
result.add(new ResourceDataSource("ds2", new AtomikosDataSourceBean()));
return result;
}
}這里mock了xaTransactionManager、transactionManager
XAShardingTransactionManager實(shí)現(xiàn)了ShardingTransactionManager接口,它擁有singleXADataSourceMap、xaTransactionManager、enlistedXAResource三個(gè)屬性,其init方法對(duì)這三個(gè)屬性進(jìn)行了初始化
getTransactionType方法返回的是TransactionType.XA;isInTransaction方法則依據(jù)xaTransactionManager.getTransactionManager().getStatus()來(lái)判斷;getConnection方法從singleXADataSourceMap取出singleXAConnection,然后執(zhí)行其getConnection方法
begin方法執(zhí)行的是xaTransactionManager.getTransactionManager().begin();commit方法則調(diào)用了xaTransactionManager.getTransactionManager().commit(),其會(huì)在finally中執(zhí)行enlistedXAResource.remove();rollback方法調(diào)用了xaTransactionManager.getTransactionManager().rollback(),其也在finally中執(zhí)行enlistedXAResource.remove();close方法則執(zhí)行xaTransactionManager.removeRecoveryResource、singleXADataSourceMap.clear()、xaTransactionManager.close()
關(guān)于sharding-jdbc中ShardingTransactionManager有什么用就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò),可以把它分享出去讓更多的人看到。
當(dāng)前題目:sharding-jdbc中ShardingTransactionManager有什么用
文章來(lái)源:http://www.chinadenli.net/article24/isphje.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供小程序開發(fā)、網(wǎng)站設(shè)計(jì)、動(dòng)態(tài)網(wǎng)站、響應(yīng)式網(wǎng)站、域名注冊(cè)、搜索引擎優(yōu)化
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)