+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta; + +import lombok.Getter; +import lombok.RequiredArgsConstructor; + +import javax.transaction.xa.XAException; +import javax.transaction.xa.XAResource; +import javax.transaction.xa.Xid; + +/** + * Sharding XA resource. + * + * @author zhaojun + */ +@RequiredArgsConstructor +@Getter +public final class ShardingXAResource implements XAResource { + + private final String resourceName; + + private final XAResource delegate; + + @Override + public void commit(final Xid xid, final boolean b) throws XAException { + delegate.commit(xid, b); + } + + @Override + public void end(final Xid xid, final int i) throws XAException { + delegate.end(xid, i); + } + + @Override + public void forget(final Xid xid) throws XAException { + delegate.forget(xid); + } + + @Override + public int getTransactionTimeout() throws XAException { + return delegate.getTransactionTimeout(); + } + + @Override + public boolean isSameRM(final XAResource xaResource) { + ShardingXAResource shardingXAResource = (ShardingXAResource) xaResource; + return resourceName.equals(shardingXAResource.getResourceName()); + } + + @Override + public int prepare(final Xid xid) throws XAException { + return delegate.prepare(xid); + } + + @Override + public Xid[] recover(final int i) throws XAException { + return delegate.recover(i); + } + + @Override + public void rollback(final Xid xid) throws XAException { + delegate.rollback(xid); + } + + @Override + public boolean setTransactionTimeout(final int i) throws XAException { + return delegate.setTransactionTimeout(i); + } + + @Override + public void start(final Xid xid, final int i) throws XAException { + delegate.start(xid, i); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnection.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnection.java new file mode 100644 index 0000000000000000000000000000000000000000..fd12d965fe9f26496ac65b073222d97969123a28 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnection.java @@ -0,0 +1,78 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection; + +import io.shardingsphere.transaction.xa.jta.ShardingXAResource; +import lombok.Getter; +import lombok.RequiredArgsConstructor; + +import javax.sql.ConnectionEventListener; +import javax.sql.StatementEventListener; +import javax.sql.XAConnection; +import javax.transaction.xa.XAResource; +import java.sql.Connection; +import java.sql.SQLException; + +/** + * Sharding XA Connection. + * + * @author zhaojun + */ +@RequiredArgsConstructor +@Getter +public final class ShardingXAConnection implements XAConnection { + + private final String resourceName; + + private final XAConnection delegate; + + @Override + public XAResource getXAResource() throws SQLException { + return new ShardingXAResource(resourceName, delegate.getXAResource()); + } + + @Override + public Connection getConnection() throws SQLException { + return delegate.getConnection(); + } + + @Override + public void close() throws SQLException { + delegate.close(); + } + + @Override + public void addConnectionEventListener(final ConnectionEventListener listener) { + delegate.addConnectionEventListener(listener); + } + + @Override + public void removeConnectionEventListener(final ConnectionEventListener listener) { + delegate.removeConnectionEventListener(listener); + } + + @Override + public void addStatementEventListener(final StatementEventListener listener) { + delegate.addStatementEventListener(listener); + } + + @Override + public void removeStatementEventListener(final StatementEventListener listener) { + delegate.removeStatementEventListener(listener); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactory.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactory.java new file mode 100644 index 0000000000000000000000000000000000000000..5ff80580e0d31fdf5c6fe724d02e765a6b341230 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactory.java @@ -0,0 +1,56 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection; + +import io.shardingsphere.core.constant.DatabaseType; +import io.shardingsphere.transaction.xa.jta.connection.dialect.H2ShardingXAConnectionWrapper; +import io.shardingsphere.transaction.xa.jta.connection.dialect.MySQLShardingXAConnectionWrapper; +import lombok.AccessLevel; +import lombok.NoArgsConstructor; + +import javax.sql.XADataSource; +import java.sql.Connection; + +/** + * Sharding XA connection wrapper. + * + * @author zhaojun + */ +@NoArgsConstructor(access = AccessLevel.PRIVATE) +public class ShardingXAConnectionFactory { + + /** + * Create a sharding XA connection from normal connection. + * + * @param databaseType database type + * @param connection normal connection + * @param resourceName resource name + * @param xaDataSource XA data source + * @return sharding XA connection + */ + public static ShardingXAConnection createShardingXAConnection(final DatabaseType databaseType, final String resourceName, final XADataSource xaDataSource, final Connection connection) { + switch (databaseType) { + case MySQL: + return new MySQLShardingXAConnectionWrapper().wrap(resourceName, xaDataSource, connection); + case H2: + return new H2ShardingXAConnectionWrapper().wrap(resourceName, xaDataSource, connection); + default: + throw new UnsupportedOperationException(String.format("Cannot support database type: `%s`", databaseType)); + } + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionWrapper.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionWrapper.java new file mode 100644 index 0000000000000000000000000000000000000000..be85f99f3f09316ce3817e6ab1f1b2c15ff7cfce --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionWrapper.java @@ -0,0 +1,39 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection; + +import javax.sql.XADataSource; +import java.sql.Connection; + +/** + * Sharding XA connection wrapper. + * + * @author zhaojun + */ +public interface ShardingXAConnectionWrapper { + + /** + * Wrap a normal connection to sharding XA connection. + * + * @param resourceName resource name + * @param xaDataSource XA data source + * @param connection connection + * @return sharding XA connection + */ + ShardingXAConnection wrap(String resourceName, XADataSource xaDataSource, Connection connection); +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/dialect/H2ShardingXAConnectionWrapper.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/dialect/H2ShardingXAConnectionWrapper.java new file mode 100644 index 0000000000000000000000000000000000000000..6b166e0a0014686b6f0d76dac51e5bc511b23398 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/connection/dialect/H2ShardingXAConnectionWrapper.java @@ -0,0 +1,84 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection.dialect; + +import io.shardingsphere.core.exception.ShardingException; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnection; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnectionWrapper; +import lombok.extern.slf4j.Slf4j; +import org.h2.jdbc.JdbcConnection; +import org.h2.jdbcx.JdbcDataSourceFactory; +import org.h2.jdbcx.JdbcXAConnection; +import org.h2.message.TraceObject; + +import javax.sql.XADataSource; +import java.lang.reflect.Constructor; +import java.lang.reflect.InvocationTargetException; +import java.lang.reflect.Method; +import java.sql.Connection; +import java.sql.SQLException; + + +/** + * H2 sharding XA connection wrapper. + * + * @author zhaojun + */ +@Slf4j +public final class H2ShardingXAConnectionWrapper implements ShardingXAConnectionWrapper { + + private static final int XA_DATA_SOURCE = 13; + + private static final Constructor+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection.dialect; + +import io.shardingsphere.core.exception.ShardingException; +import io.shardingsphere.core.util.ReflectiveUtil; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnection; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnectionWrapper; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; + +import javax.sql.XAConnection; +import javax.sql.XADataSource; +import java.lang.reflect.Method; +import java.sql.Connection; + +/** + * MySQL sharding XA connection wrapper. + * + * @author zhaojun + */ +@RequiredArgsConstructor +@Slf4j +public final class MySQLShardingXAConnectionWrapper implements ShardingXAConnectionWrapper { + + @Override + public ShardingXAConnection wrap(final String resourceName, final XADataSource xaDataSource, final Connection connection) { + try { + Connection mysqlPhysicalConnection = (Connection) connection.unwrap(Class.forName("com.mysql.jdbc.Connection")); + Method wrapConnectionMethod = ReflectiveUtil.findMethod(xaDataSource, "wrapConnection", Connection.class); + XAConnection xaConnection = (XAConnection) wrapConnectionMethod.invoke(xaDataSource, mysqlPhysicalConnection); + return new ShardingXAConnection(resourceName, xaConnection); + } catch (final Exception ex) { + log.error("Failed to wrap a connection to ShardingXAConnection"); + throw new ShardingException(ex); + } + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/AbstractUnsupportedShardingXADataSource.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/AbstractUnsupportedShardingXADataSource.java new file mode 100644 index 0000000000000000000000000000000000000000..db4debe5d6bcbdd6667f1933d19ae35629a07893 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/AbstractUnsupportedShardingXADataSource.java @@ -0,0 +1,63 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.datasource; + +import javax.sql.XAConnection; +import javax.sql.XADataSource; +import java.io.PrintWriter; +import java.sql.SQLException; +import java.sql.SQLFeatureNotSupportedException; +import java.util.logging.Logger; + +/** + * Abstract unsupported sharding XA data source. + * + * @author zhaojun + */ +public abstract class AbstractUnsupportedShardingXADataSource implements XADataSource { + + @Override + public final XAConnection getXAConnection(final String user, final String password) throws SQLException { + throw new SQLFeatureNotSupportedException("getXAConnection by user and password"); + } + + @Override + public final PrintWriter getLogWriter() throws SQLException { + throw new SQLFeatureNotSupportedException("getLogWriter"); + } + + @Override + public final void setLogWriter(final PrintWriter out) throws SQLException { + throw new SQLFeatureNotSupportedException("setLogWriter"); + } + + @Override + public final void setLoginTimeout(final int seconds) throws SQLException { + throw new SQLFeatureNotSupportedException("setLoginTimeout"); + } + + @Override + public final int getLoginTimeout() throws SQLException { + throw new SQLFeatureNotSupportedException("getLoginTimeout"); + } + + @Override + public final Logger getParentLogger() throws SQLFeatureNotSupportedException { + throw new SQLFeatureNotSupportedException("getParentLogger"); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/ShardingXADataSource.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/ShardingXADataSource.java new file mode 100644 index 0000000000000000000000000000000000000000..cc31a9fc4dba67dc58243e94cc729887065c3ecf --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/ShardingXADataSource.java @@ -0,0 +1,77 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.datasource; + +import io.shardingsphere.core.constant.DatabaseType; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnection; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnectionFactory; +import lombok.Getter; +import lombok.extern.slf4j.Slf4j; + +import javax.sql.DataSource; +import javax.sql.XADataSource; +import java.sql.Connection; +import java.sql.SQLException; + +/** + * Sharding XA data source. + * + * @author zhaojun + */ +@Getter +@Slf4j +public final class ShardingXADataSource extends AbstractUnsupportedShardingXADataSource { + + private final DatabaseType databaseType; + + private final String resourceName; + + private final DataSource originalDataSource; + + private final XADataSource xaDataSource; + + private boolean isOriginalXADataSource; + + public ShardingXADataSource(final DatabaseType databaseType, final String resourceName, final DataSource dataSource) { + this.databaseType = databaseType; + this.resourceName = resourceName; + this.originalDataSource = dataSource; + if (dataSource instanceof XADataSource) { + this.xaDataSource = (XADataSource) dataSource; + this.isOriginalXADataSource = true; + } else { + this.xaDataSource = XADataSourceFactory.build(databaseType, dataSource); + } + } + + @Override + public ShardingXAConnection getXAConnection() throws SQLException { + return isOriginalXADataSource ? new ShardingXAConnection(resourceName, xaDataSource.getXAConnection()) + : ShardingXAConnectionFactory.createShardingXAConnection(databaseType, resourceName, xaDataSource, originalDataSource.getConnection()); + } + + /** + * Get connection from original data source. + * + * @return connection + * @throws SQLException SQL exception + */ + public Connection getConnectionFromOriginalDataSource() throws SQLException { + return originalDataSource.getConnection(); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/convert/datasource/XADataSourceFactory.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/XADataSourceFactory.java similarity index 65% rename from sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/convert/datasource/XADataSourceFactory.java rename to sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/XADataSourceFactory.java index 163aa56b63717d4f1152627c950b0359e4395538..227b96d8a7c5c55596d33be01ac0f85775834b53 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/convert/datasource/XADataSourceFactory.java +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/java/io/shardingsphere/transaction/xa/jta/datasource/XADataSourceFactory.java @@ -15,16 +15,23 @@ * */ -package io.shardingsphere.transaction.xa.convert.datasource; +package io.shardingsphere.transaction.xa.jta.datasource; +import com.atomikos.beans.PropertyException; +import com.atomikos.beans.PropertyUtils; import io.shardingsphere.core.constant.DatabaseType; import io.shardingsphere.core.exception.ShardingException; +import io.shardingsphere.core.rule.DataSourceParameter; +import io.shardingsphere.transaction.xa.convert.swap.DataSourceSwapperRegistry; import lombok.AccessLevel; import lombok.NoArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import javax.sql.DataSource; import javax.sql.XADataSource; import java.util.HashMap; import java.util.Map; +import java.util.Properties; /** * XA data source factory. @@ -32,6 +39,7 @@ import java.util.Map; * @author zhaojun */ @NoArgsConstructor(access = AccessLevel.PRIVATE) +@Slf4j public final class XADataSourceFactory { private static final Map+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.manager; + +import com.atomikos.datasource.xa.jdbc.JdbcTransactionalResource; +import io.shardingsphere.transaction.xa.jta.ShardingXAResource; + +import javax.sql.XADataSource; +import javax.transaction.xa.XAResource; + +/** + * Sharding recovery resource. + * + * @author zhaojun + */ +public final class AtomikosXARecoverableResource extends JdbcTransactionalResource { + + private final String resourceName; + + public AtomikosXARecoverableResource(final String serverName, final XADataSource xaDataSource) { + super(serverName, xaDataSource); + resourceName = serverName; + } + + @Override + public boolean usesXAResource(final XAResource xaResource) { + return resourceName.equals(((ShardingXAResource) xaResource).getResourceName()); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/resources/jta.properties b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/resources/jta.properties index a1ac075fa680f518e80b5ebf055e61239bf82460..ab3d9e0432b5a819de271e85cb9b7c38f86569e4 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/resources/jta.properties +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/main/resources/jta.properties @@ -1,4 +1,5 @@ com.atomikos.icatch.serial_jta_transactions = false +com.atomikos.icatch.automatic_resource_registration = false com.atomikos.icatch.default_jta_timeout = 300000 com.atomikos.icatch.max_actives = 10000 com.atomikos.icatch.checkpoint_interval = 50000 diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/convert/AllConvertTests.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/convert/AllConvertTests.java index 326d96e67c7c722bfea3492c3a691bf0423200e7..4ed10f8e786b44a79ee0ba5ae1c428a5194af2be 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/convert/AllConvertTests.java +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/convert/AllConvertTests.java @@ -17,14 +17,9 @@ package io.shardingsphere.transaction.xa.convert; -import io.shardingsphere.transaction.xa.convert.datasource.XADataSourceFactoryTest; -import io.shardingsphere.transaction.xa.convert.datasource.XAPropertiesFactoryTest; -import io.shardingsphere.transaction.xa.convert.datasource.dialect.H2XAPropertiesTest; -import io.shardingsphere.transaction.xa.convert.datasource.dialect.MySQLXAPropertiesTest; -import io.shardingsphere.transaction.xa.convert.datasource.dialect.OracleXAPropertiesTest; -import io.shardingsphere.transaction.xa.convert.datasource.dialect.PostgreSQLXAPropertiesTest; -import io.shardingsphere.transaction.xa.convert.datasource.dialect.SQLServerXAPropertiesTest; import io.shardingsphere.transaction.xa.convert.swap.DataSourceSwapperRegistryTest; +import io.shardingsphere.transaction.xa.jta.datasource.XADataSourceFactoryTest; +import io.shardingsphere.transaction.xa.jta.datasource.XAPropertiesFactoryTest; import org.junit.runner.RunWith; import org.junit.runners.Suite; @@ -33,11 +28,6 @@ import org.junit.runners.Suite; XADataSourceMapConverterTest.class, XADataSourceFactoryTest.class, XAPropertiesFactoryTest.class, - H2XAPropertiesTest.class, - MySQLXAPropertiesTest.class, - PostgreSQLXAPropertiesTest.class, - OracleXAPropertiesTest.class, - SQLServerXAPropertiesTest.class, DataSourceSwapperRegistryTest.class }) public final class AllConvertTests { diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/DataSourceUtils.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/DataSourceUtils.java index 266478c5b8706a91a5eb75136ef8acb7f047e53d..2dedc1f566af6f2558f94fa33b58d37f52d055ba 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/DataSourceUtils.java +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/DataSourceUtils.java @@ -18,6 +18,7 @@ package io.shardingsphere.transaction.xa.fixture; import com.alibaba.druid.pool.DruidDataSource; +import com.alibaba.druid.pool.xa.DruidXADataSource; import com.zaxxer.hikari.HikariDataSource; import io.shardingsphere.core.constant.DatabaseType; import io.shardingsphere.core.constant.PoolType; @@ -50,6 +51,8 @@ public final class DataSourceUtils { return newHikariDataSource(databaseType, databaseName); case DRUID: return newDruidDataSource(databaseType, databaseName); + case DRUID_XA: + return newDruidXADataSource(databaseType, databaseName); default: return Mockito.mock(DataSource.class); } @@ -71,17 +74,27 @@ public final class DataSourceUtils { private static DruidDataSource newDruidDataSource(final DatabaseType databaseType, final String databaseName) { DruidDataSource result = new DruidDataSource(); - result.setUrl(getUrl(databaseType, databaseName)); - result.setUsername("root"); - result.setPassword("root"); - result.setMaxActive(10); - result.setMinIdle(2); - result.setMaxWait(15 * 1000); - result.setMinEvictableIdleTimeMillis(40 * 1000); - result.setTimeBetweenEvictionRunsMillis(20 * 1000); + configDruidDataSource(result, databaseType, databaseName); return result; } + private static DruidXADataSource newDruidXADataSource(final DatabaseType databaseType, final String databaseName) { + DruidXADataSource result = new DruidXADataSource(); + configDruidDataSource(result, databaseType, databaseName); + return result; + } + + private static void configDruidDataSource(final DruidDataSource druidDataSource, final DatabaseType databaseType, final String databaseName) { + druidDataSource.setUrl(getUrl(databaseType, databaseName)); + druidDataSource.setUsername("root"); + druidDataSource.setPassword("root"); + druidDataSource.setMaxActive(10); + druidDataSource.setMinIdle(2); + druidDataSource.setMaxWait(15 * 1000); + druidDataSource.setMinEvictableIdleTimeMillis(40 * 1000); + druidDataSource.setTimeBetweenEvictionRunsMillis(20 * 1000); + } + private static HikariDataSource newHikariDataSource(final DatabaseType databaseType, final String databaseName) { HikariDataSource result = new HikariDataSource(); result.setJdbcUrl(getUrl(databaseType, databaseName)); diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/FixtureXATransactionManager.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/FixtureXATransactionManager.java index b1d8e4a57943634afb7922943b1b627104072659..f272b79d5306e543f672631b133f8ef3488c6d25 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/FixtureXATransactionManager.java +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/fixture/FixtureXATransactionManager.java @@ -19,7 +19,6 @@ package io.shardingsphere.transaction.xa.fixture; import io.shardingsphere.core.constant.DatabaseType; import io.shardingsphere.core.rule.DataSourceParameter; -import io.shardingsphere.transaction.core.context.XATransactionContext; import io.shardingsphere.transaction.spi.xa.XATransactionManager; import javax.sql.DataSource; @@ -28,20 +27,24 @@ import javax.transaction.Status; import javax.transaction.TransactionManager; public final class FixtureXATransactionManager implements XATransactionManager { + @Override + public void startup() { + } + @Override public void destroy() { } @Override - public void begin(final XATransactionContext transactionContext) { + public void begin() { } @Override - public void commit(final XATransactionContext transactionContext) { + public void commit() { } @Override - public void rollback(final XATransactionContext transactionContext) { + public void rollback() { } @Override @@ -58,4 +61,14 @@ public final class FixtureXATransactionManager implements XATransactionManager { public TransactionManager getUnderlyingTransactionManager() { return null; } + + @Override + public void registerRecoveryResource(final String dataSourceName, final XADataSource xaDataSource) { + + } + + @Override + public void removeRecoveryResource(final String dataSourceName, final XADataSource xaDataSource) { + + } } diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/handler/XAShardingTransactionHandlerTest.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/handler/XAShardingTransactionHandlerTest.java index e71029ab3dde9e7db98523e44dbd7f6270e8f686..80bce2a396f7fc3047b0685e9491e6b6635f4df1 100644 --- a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/handler/XAShardingTransactionHandlerTest.java +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/handler/XAShardingTransactionHandlerTest.java @@ -17,33 +17,76 @@ package io.shardingsphere.transaction.xa.handler; +import com.atomikos.jdbc.AtomikosDataSourceBean; +import io.shardingsphere.core.constant.DatabaseType; +import io.shardingsphere.core.constant.PoolType; import io.shardingsphere.transaction.api.TransactionType; -import io.shardingsphere.transaction.core.TransactionOperationType; -import io.shardingsphere.transaction.core.context.XATransactionContext; import io.shardingsphere.transaction.core.manager.ShardingTransactionManager; -import io.shardingsphere.transaction.xa.manager.AtomikosTransactionManager; +import io.shardingsphere.transaction.spi.xa.XATransactionManager; +import io.shardingsphere.transaction.xa.fixture.DataSourceUtils; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnection; +import io.shardingsphere.transaction.xa.jta.datasource.ShardingXADataSource; +import lombok.SneakyThrows; +import org.junit.Before; import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; +import javax.sql.DataSource; +import javax.sql.XADataSource; import javax.transaction.Status; +import javax.transaction.Transaction; +import javax.transaction.TransactionManager; +import javax.transaction.xa.XAResource; +import java.lang.reflect.Field; +import java.sql.Connection; +import java.util.HashMap; +import java.util.Map; import static org.hamcrest.CoreMatchers.instanceOf; import static org.hamcrest.CoreMatchers.is; import static org.junit.Assert.assertThat; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.ArgumentMatchers.anyString; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.times; +import static org.mockito.Mockito.verify; +import static org.mockito.Mockito.when; +@RunWith(MockitoJUnitRunner.class) public class XAShardingTransactionHandlerTest { private XAShardingTransactionHandler xaShardingTransactionHandler = new XAShardingTransactionHandler(); - private XATransactionContext beginContext = new XATransactionContext(TransactionOperationType.BEGIN); + @Mock + private XATransactionManager xaTransactionManager; - private XATransactionContext commitContext = new XATransactionContext(TransactionOperationType.COMMIT); + @Mock + private TransactionManager transactionManager; - private XATransactionContext rollbackContext = new XATransactionContext(TransactionOperationType.ROLLBACK); + @Mock + private Transaction transaction; + + @Before + @SneakyThrows + public void setUp() { + setMockXATransactionManager(xaShardingTransactionHandler, xaTransactionManager); + when(xaTransactionManager.getUnderlyingTransactionManager()).thenReturn(transactionManager); + when(transactionManager.getTransaction()).thenReturn(transaction); + } + + @SneakyThrows + private void setMockXATransactionManager(final XAShardingTransactionHandler xaShardingTransactionHandler, final XATransactionManager xaTransactionManager) { + Field field = xaShardingTransactionHandler.getClass().getDeclaredField("xaTransactionManager"); + field.setAccessible(true); + field.set(xaShardingTransactionHandler, xaTransactionManager); + } @Test public void assertGetTransactionManager() { ShardingTransactionManager shardingTransactionManager = xaShardingTransactionHandler.getShardingTransactionManager(); - assertThat(shardingTransactionManager, instanceOf(AtomikosTransactionManager.class)); + assertThat(shardingTransactionManager, instanceOf(XATransactionManager.class)); } @Test @@ -52,62 +95,103 @@ public class XAShardingTransactionHandlerTest { } @Test - public void assertDoXATransactionBegin() throws InterruptedException { - Thread thread = new Thread(new Runnable() { - @Override - public void run() { - xaShardingTransactionHandler.doInTransaction(beginContext); - int actualStatus = xaShardingTransactionHandler.getShardingTransactionManager().getStatus(); - assertThat(actualStatus, is(Status.STATUS_ACTIVE)); - } - }); - thread.start(); - thread.join(); + public void assertRegisterXATransactionalDataSource() { + Map+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta; + +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import javax.transaction.xa.XAException; +import javax.transaction.xa.XAResource; +import javax.transaction.xa.Xid; + +import static org.junit.Assert.assertTrue; +import static org.mockito.Mockito.verify; + +@RunWith(MockitoJUnitRunner.class) +public class ShardingXAResourceTest { + + @Mock + private XAResource xaResource; + + @Mock + private Xid xid; + + private ShardingXAResource shardingXAResource; + + @Before + public void setUp() { + shardingXAResource = new ShardingXAResource("ds1", xaResource); + } + + @Test + public void assertCommit() throws XAException { + shardingXAResource.commit(xid, true); + verify(xaResource).commit(xid, true); + } + + @Test + public void assertEnd() throws XAException { + shardingXAResource.end(xid, 1); + verify(xaResource).end(xid, 1); + } + + @Test + public void assertForget() throws XAException { + shardingXAResource.forget(xid); + verify(xaResource).forget(xid); + } + + @Test + public void assertGetTransactionTimeout() throws XAException { + shardingXAResource.getTransactionTimeout(); + verify(xaResource).getTransactionTimeout(); + } + + @Test + public void assertIsSameRM() { + assertTrue(shardingXAResource.isSameRM(new ShardingXAResource("ds1", xaResource))); + } + + @Test + public void assertPrepare() throws XAException { + shardingXAResource.prepare(xid); + verify(xaResource).prepare(xid); + } + + @Test + public void assertRecover() throws XAException { + shardingXAResource.recover(1); + verify(xaResource).recover(1); + } + + @Test + public void assertRollback() throws XAException { + shardingXAResource.rollback(xid); + verify(xaResource).rollback(xid); + } + + @Test + public void assertSetTransactionTimeout() throws XAException { + shardingXAResource.setTransactionTimeout(1); + verify(xaResource).setTransactionTimeout(1); + } + + @Test + public void assertStart() throws XAException { + shardingXAResource.start(xid, 1); + verify(xaResource).start(xid, 1); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactoryTest.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactoryTest.java new file mode 100644 index 0000000000000000000000000000000000000000..ebc05940d89ae84f4d4cd0c4714dd0d982861402 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionFactoryTest.java @@ -0,0 +1,54 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection; + +import io.shardingsphere.core.constant.DatabaseType; +import io.shardingsphere.core.exception.ShardingException; +import lombok.SneakyThrows; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import javax.sql.XADataSource; +import java.sql.Connection; + +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertThat; + +@RunWith(MockitoJUnitRunner.class) +public class ShardingXAConnectionFactoryTest { + + @Mock + private XADataSource xaDataSource; + + @Mock + private Connection connection; + + @Test(expected = ShardingException.class) + public void assertCreateMysqlMySQLShardingXAConnection() { + ShardingXAConnectionFactory.createShardingXAConnection(DatabaseType.MySQL, "ds1", xaDataSource, connection); + } + + @Test + @SneakyThrows + public void assertCreateH2ShardingXAConnection() { + ShardingXAConnection actual = ShardingXAConnectionFactory.createShardingXAConnection(DatabaseType.H2, "ds1", xaDataSource, connection); + assertThat(actual.getResourceName(), is("ds1")); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionTest.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionTest.java new file mode 100644 index 0000000000000000000000000000000000000000..e14b7b877f03b7dcf060ff31570ac67799a7eb57 --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/ShardingXAConnectionTest.java @@ -0,0 +1,96 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection; + +import io.shardingsphere.transaction.xa.jta.ShardingXAResource; +import lombok.SneakyThrows; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import javax.sql.ConnectionEventListener; +import javax.sql.StatementEventListener; +import javax.sql.XAConnection; +import javax.transaction.xa.XAResource; + +import static org.hamcrest.CoreMatchers.instanceOf; +import static org.junit.Assert.assertThat; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.verify; + +@RunWith(MockitoJUnitRunner.class) +public class ShardingXAConnectionTest { + + @Mock + private XAConnection xaConnection; + + private ShardingXAConnection shardingXAConnection; + + @Before + public void setUp() { + shardingXAConnection = new ShardingXAConnection("ds1", xaConnection); + } + + @Test + @SneakyThrows + public void assertGetConnection() { + shardingXAConnection.getConnection(); + verify(xaConnection).getConnection(); + } + + @Test + @SneakyThrows + public void assertGetXAResource() { + XAResource actual = shardingXAConnection.getXAResource(); + assertThat(actual, instanceOf(ShardingXAResource.class)); + } + + @Test + @SneakyThrows + public void close() { + shardingXAConnection.close(); + verify(xaConnection).close(); + } + + @Test + public void assertAddConnectionEventListener() { + shardingXAConnection.addConnectionEventListener(mock(ConnectionEventListener.class)); + verify(xaConnection).addConnectionEventListener(any(ConnectionEventListener.class)); + } + + @Test + public void assertRemoveConnectionEventListener() { + shardingXAConnection.removeConnectionEventListener(mock(ConnectionEventListener.class)); + verify(xaConnection).removeConnectionEventListener(any(ConnectionEventListener.class)); + } + + @Test + public void assertAddStatementEventListener() { + shardingXAConnection.addStatementEventListener(mock(StatementEventListener.class)); + verify(xaConnection).addStatementEventListener(any(StatementEventListener.class)); + } + + @Test + public void removeStatementEventListener() { + shardingXAConnection.removeStatementEventListener(mock(StatementEventListener.class)); + verify(xaConnection).removeStatementEventListener(any(StatementEventListener.class)); + } +} diff --git a/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/dialect/MySQLShardingXAConnectionWrapperTest.java b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/dialect/MySQLShardingXAConnectionWrapperTest.java new file mode 100644 index 0000000000000000000000000000000000000000..4ef109de06ea2b54393f806fe67c4ef1f0af630f --- /dev/null +++ b/sharding-transaction/sharding-transaction-2pc/sharding-transaction-2pc-xa/src/test/java/io/shardingsphere/transaction/xa/jta/connection/dialect/MySQLShardingXAConnectionWrapperTest.java @@ -0,0 +1,70 @@ +/* + * Copyright 2016-2018 shardingsphere.io. + *+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + *
+ */ + +package io.shardingsphere.transaction.xa.jta.connection.dialect; + +import io.shardingsphere.core.constant.DatabaseType; +import io.shardingsphere.core.constant.PoolType; +import io.shardingsphere.transaction.xa.fixture.DataSourceUtils; +import io.shardingsphere.transaction.xa.jta.connection.ShardingXAConnection; +import io.shardingsphere.transaction.xa.jta.datasource.XADataSourceFactory; +import lombok.SneakyThrows; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import javax.sql.DataSource; +import javax.sql.XADataSource; +import javax.transaction.xa.XAResource; +import java.sql.Connection; + +import static org.hamcrest.CoreMatchers.instanceOf; +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertThat; +import static org.mockito.ArgumentMatchers.anyObject; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + +@RunWith(MockitoJUnitRunner.class) +public class MySQLShardingXAConnectionWrapperTest { + + private XADataSource xaDataSource; + + @Mock + private Connection connection; + + @Before + @SneakyThrows + @SuppressWarnings("unchecked") + public void setUp() { + Connection mysqlConnection = (Connection) mock(Class.forName("com.mysql.jdbc.Connection")); + DataSource dataSource = DataSourceUtils.build(PoolType.HIKARI, DatabaseType.MySQL, "ds1"); + xaDataSource = XADataSourceFactory.build(DatabaseType.MySQL, dataSource); + when(connection.unwrap((Class