DataRepository => JDBCRepository

/*

 * Copyright 2016-2018 shardingsphere.io.

 * <p>

 * 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.

 * </p>

 */

package io.shardingsphere.example.repository.jdbc;

import javax.sql.DataSource;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

public class JDBCRepository {

    private final DataSource dataSource;

    private final boolean isXA;

    public JDBCRepository(final DataSource dataSource) {

        this(dataSource, false);

    }

    public JDBCRepository(final DataSource dataSource, final boolean isXA) {

        this.dataSource = dataSource;

        this.isXA = isXA;

    }

    public void demo() throws SQLException {

        createTable();

        insertData();

        insertFailure();

        updateData();

        System.out.println("1.Query with EQUAL--------------");

        queryWithEqual();

        System.out.println("2.Query with IN--------------");

        queryWithIn();

        dropTable();

    }

    private void createTable() throws SQLException {

        execute("CREATE TABLE IF NOT EXISTS t_order (order_id BIGINT NOT NULL AUTO_INCREMENT, user_id INT NOT NULL, status VARCHAR(50), PRIMARY KEY (order_id))");

        execute("CREATE TABLE IF NOT EXISTS t_order_item (order_item_id BIGINT NOT NULL AUTO_INCREMENT, order_id BIGINT NOT NULL, user_id INT NOT NULL, PRIMARY KEY (order_item_id))");

    }

    private void insertData() throws SQLException {

        Connection connection = dataSource.getConnection();

        setAutoCommit(connection);

        Statement statement = connection.createStatement();

        try {

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

                long orderId = insertAndGetGeneratedKey(statement,"INSERT INTO t_order (user_id, status) VALUES (10, 'INIT')");

                statement.execute(String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 10)", orderId));

                orderId = insertAndGetGeneratedKey(statement,"INSERT INTO t_order (user_id, status) VALUES (11, 'INIT')");

                statement.execute(String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 11)", orderId));

            }

            commit(connection);

        } catch (SQLException ex) {

            rollback(connection);

        }

        finally {

            connection.close();

            statement.close();

        }

    }

    private void insertFailure() throws SQLException {

        Connection connection = dataSource.getConnection();

        setAutoCommit(connection);

        Statement statement = connection.createStatement();

        try {

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

                long orderId = insertAndGetGeneratedKey(statement,"INSERT INTO t_order (user_id, status) VALUES (10, 'INIT')");

                statement.execute(String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 10)", orderId));

                orderId = insertAndGetGeneratedKey(statement,"INSERT INTO t_order (user_id, status) VALUES (11, 'INIT')");

                statement.execute(String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 11)", orderId));

            }

            makeException();

            commit(connection);

        } catch (Exception ex) {

            rollback(connection);

        }

        finally {

            connection.close();

            statement.close();

        }

    }

    private void makeException() {

        System.out.println(10 / 0);

    }

    private long insertAndGetGeneratedKey(final Statement statement, final String sql) throws SQLException {

        long result = -1;

        statement.executeUpdate(sql, Statement.RETURN_GENERATED_KEYS);

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

            if (resultSet.next()) {

                result = resultSet.getLong(1);

            }

        }

        return result;

    }

    private void updateData() throws SQLException {

        Connection connection = dataSource.getConnection();

        setAutoCommit(connection);

        try {

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

                Long orderId = getRandomOrderId(connection, 10);

                String sql = "UPDATE t_order SET status='UPDATE_1' WHERE user_id=? and order_id=?";

                PreparedStatement preparedStatement = connection.prepareStatement(sql);

                preparedStatement.setObject(1, 10);

                preparedStatement.setObject(2, orderId);

                preparedStatement.executeUpdate();

            }

            commit(connection);

        } catch (SQLException ex) {

            rollback(connection);

        }

        finally {

            connection.close();

        }

    }

    private Long getRandomOrderId(Connection connection, int userId) throws SQLException {

        Statement statement = connection.createStatement();

        int index = (int) (Math.random() * 10);

        ResultSet resultSet = statement.executeQuery("SELECT order_id FROM t_order WHERE user_id=" + userId);

        long location = 0;

        while (resultSet.next()) {

            if (++location == index) {

                return resultSet.getLong(1);

            }

        }

        return 0L;

    }

    private void setAutoCommit(final Connection connection) throws SQLException {

        if (isXA) {

            connection.setAutoCommit(false);

        }

    }

    private void commit(final Connection connection) throws SQLException {

        if (isXA) {

            connection.commit();

        }

    }

    private void rollback(final Connection connection) throws SQLException {

        if (isXA) {

            connection.rollback();

        }

    }

    private void queryWithEqual() throws SQLException {

        String sql = "SELECT i.* FROM t_order o JOIN t_order_item i ON o.order_id=i.order_id WHERE o.user_id=?";

        try (

                Connection connection = dataSource.getConnection();

                PreparedStatement preparedStatement = connection.prepareStatement(sql)) {

            preparedStatement.setInt(1, 10);

            printQuery(preparedStatement);

        }

    }

    private void queryWithIn() throws SQLException {

        String sql = "SELECT i.* FROM t_order o JOIN t_order_item i ON o.order_id=i.order_id WHERE o.user_id IN (?, ?)";

        try (

                Connection connection = dataSource.getConnection();

                PreparedStatement preparedStatement = connection.prepareStatement(sql)) {

            preparedStatement.setInt(1, 10);

            preparedStatement.setInt(2, 11);

            printQuery(preparedStatement);

        }

    }

    private void printQuery(final PreparedStatement preparedStatement) throws SQLException {

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

            while (resultSet.next()) {

                System.out.print("order_item_id:" + resultSet.getLong(1) + ", ");

                System.out.print("order_id:" + resultSet.getLong(2) + ", ");

                System.out.print("user_id:" + resultSet.getInt(3));

                System.out.println();

            }

        }

    }

    private void dropTable() throws SQLException {

        execute("DROP TABLE t_order_item");

        execute("DROP TABLE t_order");

    }

    private void execute(final String sql) throws SQLException {

        try (

                Connection connection = dataSource.getConnection();

                Statement statement = connection.createStatement()) {

            statement.execute(sql);

        }

    }

}