Unverified Commit e943f997 authored by zhaojun's avatar zhaojun Committed by GitHub
Browse files

Merge pull request #139 from cherrylzhao/dev 

add seataAT raw-jdbc example.
parents 9371693b 3ce311f5

pom.xml

+17 −0
Original line number Diff line number Diff line
@@ -36,6 +36,8 @@ 
        <jpa.version>1.0.0.Final</jpa.version>

        <hibernate.version>4.3.11.Final</hibernate.version>

    

        <seata.version>0.5.1</seata.version>



        <junit.version>4.12</junit.version>

        <hamcrest.version>1.3</hamcrest.version>

        <mockito.version>2.7.21</mockito.version>
@@ -80,6 +82,11 @@ 
                <artifactId>sharding-transaction-base-saga</artifactId>

                <version>${sharding-sphere.spi.impl.version}</version>

            </dependency>

            <dependency>

                <groupId>org.apache.shardingsphere</groupId>

                <artifactId>sharding-transaction-base-seata-at</artifactId>

                <version>${project.parent.version}</version>

            </dependency>

            <dependency>

                <groupId>org.apache.shardingsphere</groupId>

                <artifactId>sharding-jdbc-spring-boot-starter</artifactId>
@@ -196,6 +203,16 @@ 
                <artifactId>mybatis-spring-boot-starter</artifactId>

                <version>${mybatis-spring.version}</version>

            </dependency>

            <dependency>

                <groupId>io.seata</groupId>

                <artifactId>seata-rm-datasource</artifactId>

                <version>${seata.version}</version>

            </dependency>

            <dependency>

                <groupId>io.seata</groupId>

                <artifactId>seata-tm</artifactId>

                <version>${seata.version}</version>

            </dependency>

            <dependency>

                <groupId>org.springframework.boot</groupId>

                <artifactId>spring-boot-starter-test</artifactId>

sharding-jdbc-example/transaction-example/pom.xml

+1 −0
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ 
    <modules>

        <module>transaction-2pc-xa-example</module>

        <module>transaction-base-saga-example</module>

        <module>transaction-base-seata-example</module>

    </modules>



</project>

sharding-jdbc-example/transaction-example/transaction-base-seata-example/pom.xml

0 → 100644
+45 −0
Original line number Diff line number Diff line 
<?xml version="1.0" encoding="UTF-8"?>

<!--

  ~ Licensed to the Apache Software Foundation (ASF) under one or more

  ~ contributor license agreements.  See the NOTICE file distributed with

  ~ this work for additional information regarding copyright ownership.

  ~ The ASF licenses this file to You 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.

  -->



<project xmlns="http://maven.apache.org/POM/4.0.0"

         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

    <modelVersion>4.0.0</modelVersion>

    <parent>

        <groupId>org.apache.shardingsphere.example</groupId>

        <artifactId>transaction-example</artifactId>

        <version>4.0.0-RC2-SNAPSHOT</version>

    </parent>

    <artifactId>transaction-base-seata-example</artifactId>

    <packaging>pom</packaging>

    

    <modules>

        <module>transaction-base-seata-raw-jdbc-example</module>

    </modules>

    

    <dependencies>

        <dependency>

            <groupId>io.seata</groupId>

            <artifactId>seata-rm-datasource</artifactId>

        </dependency>

        <dependency>

            <groupId>io.seata</groupId>

            <artifactId>seata-tm</artifactId>

        </dependency>

    </dependencies>

</project>

sharding-jdbc-example/transaction-example/transaction-base-seata-example/transaction-base-seata-raw-jdbc-example/pom.xml

0 → 100644
+46 −0
Original line number Diff line number Diff line 
<?xml version="1.0" encoding="UTF-8"?>

<!--

  ~ Licensed to the Apache Software Foundation (ASF) under one or more

  ~ contributor license agreements.  See the NOTICE file distributed with

  ~ this work for additional information regarding copyright ownership.

  ~ The ASF licenses this file to You 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.

  -->



<project xmlns="http://maven.apache.org/POM/4.0.0"

         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

    <parent>

        <groupId>org.apache.shardingsphere.example</groupId>

        <artifactId>transaction-base-seata-example</artifactId>

        <version>4.0.0-RC2-SNAPSHOT</version>

    </parent>

    <modelVersion>4.0.0</modelVersion>

    

    <artifactId>transaction-base-seata-raw-jdbc-example</artifactId>

    

    <dependencies>

        <dependency>

            <groupId>org.apache.shardingsphere.example</groupId>

            <artifactId>repository-jdbc</artifactId>

            <version>${project.version}</version>

        </dependency>

        <dependency>

            <groupId>org.apache.shardingsphere</groupId>

            <artifactId>sharding-jdbc-core</artifactId>

        </dependency>

        <dependency>

            <groupId>org.apache.shardingsphere</groupId>

            <artifactId>sharding-transaction-base-seata-at</artifactId>

        </dependency>

    </dependencies>

</project>

sharding-jdbc-example/transaction-example/transaction-base-seata-example/transaction-base-seata-raw-jdbc-example/src/main/java/org/apache/shardingsphere/example/transaction/base/seata/raw/jdbc/YamlConfigurationTransactionExample.java

0 → 100755
+145 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You 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 org.apache.shardingsphere.example.transaction.base.seata.raw.jdbc;



import org.apache.shardingsphere.example.common.entity.Order;

import org.apache.shardingsphere.example.common.entity.OrderItem;

import org.apache.shardingsphere.example.common.jdbc.repository.OrderItemRepositoryImpl;

import org.apache.shardingsphere.example.common.jdbc.repository.OrderRepositoryImpl;

import org.apache.shardingsphere.example.common.jdbc.service.CommonServiceImpl;

import org.apache.shardingsphere.example.common.service.CommonService;

import org.apache.shardingsphere.shardingjdbc.api.yaml.YamlShardingDataSourceFactory;

import org.apache.shardingsphere.transaction.core.TransactionType;

import org.apache.shardingsphere.transaction.core.TransactionTypeHolder;



import javax.sql.DataSource;

import java.io.File;

import java.io.IOException;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.concurrent.locks.LockSupport;



/*

    Please startup seata-server( before running this example.

    Download seata-server from here https://github.com/seata/seata/releases

 */

public class YamlConfigurationTransactionExample {

    

    private static String configFile = "/META-INF/sharding-databases-tables.yaml";

//    private static String configFile = "/META-INF/master-slave.yaml";

    

    public static void main(final String[] args) throws SQLException, IOException {

        DataSource dataSource = YamlShardingDataSourceFactory.createDataSource(getFile(configFile));

        CommonService commonService = getCommonService(dataSource);

        commonService.initEnvironment();

        processSeataTransaction(dataSource, commonService);

        commonService.cleanEnvironment();

    }

    

    private static File getFile(final String fileName) {

        return new File(Thread.currentThread().getClass().getResource(fileName).getFile());

    }

    

    private static CommonService getCommonService(final DataSource dataSource) {

        return new CommonServiceImpl(new OrderRepositoryImpl(dataSource), new OrderItemRepositoryImpl(dataSource));

    }

    

    private static void processSeataTransaction(final DataSource dataSource, final CommonService commonService) throws SQLException {

        TransactionTypeHolder.set(TransactionType.BASE);

        System.out.println("------############## Start seata succeed transaction ##################------");

        try (Connection connection = dataSource.getConnection()) {

            connection.setAutoCommit(false);

            insertSuccess(connection, commonService);

            connection.commit();

        }

        LockSupport.parkUntil(System.currentTimeMillis() + 1000);

        truncateTable(dataSource);

        System.out.println("------############## End seata succeed transaction ######################------");

        System.out.println("------############## Start seata failure transaction ############------");

        TransactionTypeHolder.set(TransactionType.BASE);

        Connection connection = dataSource.getConnection();

        try {

            connection.setAutoCommit(false);

            insertSuccess(connection, commonService);

            throw new SQLException("exception occur!");

        } catch (final SQLException ex) {

            connection.rollback();

        }

        commonService.printData();

        System.out.println("------############# End seata failure transaction #############------");

        truncateTable(dataSource);

    }

    

    private static void insertSuccess(final Connection connection, final CommonService commonService) throws SQLException {

        for (int i = 0; i < 10; i++) {

            Order order = new Order();

            order.setUserId(i);

            order.setStatus("SEATA-INIT");

            insertOrder(connection, order);

            OrderItem item = new OrderItem();

            item.setUserId(i);

            item.setOrderId(order.getOrderId());

            item.setStatus("SEATA-INIT");

            insertOrderItem(connection, item);

        }

        commonService.printData();

    }

    

    private static Long insertOrder(final Connection connection, final Order order) {

        String sql = "INSERT INTO t_order (user_id, status) VALUES (?, ?)";

        try (PreparedStatement preparedStatement = connection.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {

            preparedStatement.setObject(1, order.getUserId());

            preparedStatement.setObject(2, order.getStatus());

            preparedStatement.executeUpdate();

            try (ResultSet resultSet = preparedStatement.getGeneratedKeys()) {

                if (resultSet.next()) {

                    order.setOrderId(resultSet.getLong(1));

                }

            }

        } catch (final SQLException ignored) {

        }

        return order.getOrderId();

    }

    

    private static Long insertOrderItem(final Connection connection, final OrderItem orderItem) {

        String sql = "INSERT INTO t_order_item (order_id, user_id, status) VALUES (?, ?, ?)";

        try (PreparedStatement preparedStatement = connection.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {

            preparedStatement.setObject(1, orderItem.getOrderId());

            preparedStatement.setObject(2, orderItem.getUserId());

            preparedStatement.setString(3, orderItem.getStatus());

            preparedStatement.executeUpdate();

            try (ResultSet resultSet = preparedStatement.getGeneratedKeys()) {

                if (resultSet.next()) {

                    orderItem.setOrderItemId(resultSet.getLong(1));

                }

            }

        } catch (final SQLException ignored) {

        }

        return orderItem.getOrderItemId();

    }

    

    private static void truncateTable(final DataSource dataSource) {

        OrderRepositoryImpl orderRepository = new OrderRepositoryImpl(dataSource);

        OrderItemRepositoryImpl orderItemRepository = new OrderItemRepositoryImpl(dataSource);

        orderRepository.truncateTable();

        orderItemRepository.truncateTable();

    }

}