DataRepository => JDBCRepository (2a27b137) · Commits · SUMMER2020 / students / proj-2004142

common-repository/jdbc-repository/src/main/java/io/shardingsphere/example/repository/jdbc/DataRepository.java

deleted100644 → 0

+0 −213

Original line number	Diff line number	Diff line
		/*
		* 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 DataRepository {

		private final DataSource dataSource;

		private final boolean isXA;

		public DataRepository(final DataSource dataSource) {
		this(dataSource, false);
		}

		public DataRepository(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);
		}
		}
		}

sharding-jdbc-example/raw-jdbc-example/src/main/java/io/shardingsphere/example/jdbc/main/nodep/java/MasterSlaveOnly.java

+2 −2

Original line number	Diff line number	Diff line
		@@ -19,7 +19,7 @@ package io.shardingsphere.example.jdbc.main.nodep.java;

		import io.shardingsphere.api.config.MasterSlaveRuleConfiguration;
		import io.shardingsphere.shardingjdbc.api.MasterSlaveDataSourceFactory;
		import io.shardingsphere.example.repository.jdbc.DataRepository;
		import io.shardingsphere.example.repository.jdbc.JDBCRepository;
		import io.shardingsphere.example.repository.jdbc.DataSourceUtil;

		import javax.sql.DataSource;
		@@ -36,7 +36,7 @@ import java.util.concurrent.ConcurrentHashMap;
		public class MasterSlaveOnly {

		public static void main(final String[] args) throws SQLException {
		new DataRepository(getDataSource()).demo();
		new JDBCRepository(getDataSource()).demo();
		}

		private static DataSource getDataSource() throws SQLException {

sharding-jdbc-example/raw-jdbc-example/src/main/java/io/shardingsphere/example/jdbc/main/nodep/java/ShardingAndMasterSlaveTogether.java

+2 −2

Original line number	Diff line number	Diff line
		@@ -25,7 +25,7 @@ import io.shardingsphere.api.config.strategy.StandardShardingStrategyConfigurati
		import io.shardingsphere.example.jdbc.fixture.algorithm.ModuloShardingDatabaseAlgorithm;
		import io.shardingsphere.example.jdbc.fixture.algorithm.ModuloShardingTableAlgorithm;
		import io.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;
		import io.shardingsphere.example.repository.jdbc.DataRepository;
		import io.shardingsphere.example.repository.jdbc.JDBCRepository;
		import io.shardingsphere.example.repository.jdbc.DataSourceUtil;

		import javax.sql.DataSource;
		@@ -42,7 +42,7 @@ import java.util.Properties;
		public class ShardingAndMasterSlaveTogether {

		public static void main(final String[] args) throws SQLException {
		new DataRepository(getDataSource()).demo();
		new JDBCRepository(getDataSource()).demo();
		}

		private static DataSource getDataSource() throws SQLException {

sharding-jdbc-example/raw-jdbc-example/src/main/java/io/shardingsphere/example/jdbc/main/nodep/java/ShardingOnlyWithDatabases.java

+2 −2

Original line number	Diff line number	Diff line
		@@ -21,7 +21,7 @@ import io.shardingsphere.api.config.ShardingRuleConfiguration;
		import io.shardingsphere.api.config.TableRuleConfiguration;
		import io.shardingsphere.api.config.strategy.InlineShardingStrategyConfiguration;
		import io.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;
		import io.shardingsphere.example.repository.jdbc.DataRepository;
		import io.shardingsphere.example.repository.jdbc.JDBCRepository;
		import io.shardingsphere.example.repository.jdbc.DataSourceUtil;

		import javax.sql.DataSource;
		@@ -33,7 +33,7 @@ import java.util.Properties;
		public class ShardingOnlyWithDatabases {

		public static void main(final String[] args) throws SQLException {
		new DataRepository(getDataSource()).demo();
		new JDBCRepository(getDataSource()).demo();
		}

		private static DataSource getDataSource() throws SQLException {

sharding-jdbc-example/raw-jdbc-example/src/main/java/io/shardingsphere/example/jdbc/main/nodep/java/ShardingOnlyWithDatabasesAndTables.java

+2 −2

Original line number	Diff line number	Diff line
		@@ -23,7 +23,7 @@ import io.shardingsphere.api.config.strategy.InlineShardingStrategyConfiguration
		import io.shardingsphere.api.config.strategy.StandardShardingStrategyConfiguration;
		import io.shardingsphere.example.jdbc.fixture.algorithm.ModuloShardingTableAlgorithm;
		import io.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;
		import io.shardingsphere.example.repository.jdbc.DataRepository;
		import io.shardingsphere.example.repository.jdbc.JDBCRepository;
		import io.shardingsphere.example.repository.jdbc.DataSourceUtil;

		import javax.sql.DataSource;
		@@ -35,7 +35,7 @@ import java.util.Properties;
		public class ShardingOnlyWithDatabasesAndTables {

		public static void main(final String[] args) throws SQLException {
		new DataRepository(getDataSource()).demo();
		new JDBCRepository(getDataSource()).demo();
		}

		private static DataSource getDataSource() throws SQLException {