update example for proxy prepare statement (f7fbdd19) · Commits · SUMMER2020 / students / proj-2004142

sharding-jdbc-server-example/src/main/java/io/shardingjdbc/example/server/ServerMain.java

deleted100644 → 0

+0 −42

Original line number	Diff line number	Diff line
		package io.shardingjdbc.example.server;

		import io.shardingjdbc.example.server.repository.RawJdbcRepository;
		import io.shardingjdbc.server.ShardingJDBCServer;
		import org.apache.commons.dbcp.BasicDataSource;

		import javax.sql.DataSource;
		import java.sql.SQLException;

		public final class ServerMain {

		public static void main(String[] args) throws InterruptedException, SQLException {
		// startServer();
		RawJdbcRepository rawJdbcRepository = new RawJdbcRepository(createDataSource());
		rawJdbcRepository.demo();
		}

		private static void startServer() throws InterruptedException {
		new Thread() {

		@Override
		public void run() {
		ShardingJDBCServer server = new ShardingJDBCServer();
		try {
		server.start(3307);
		} catch (final InterruptedException ex) {
		ex.printStackTrace();
		}
		}
		}.start();
		Thread.sleep(1000L);
		}

		public static DataSource createDataSource() {
		BasicDataSource result = new BasicDataSource();
		result.setDriverClassName(com.mysql.jdbc.Driver.class.getName());
		result.setUrl("jdbc:mysql://localhost:3307/demo_ds");
		result.setUsername("root");
		result.setPassword("");
		return result;
		}
		}

sharding-jdbc-server-example/src/main/java/io/shardingjdbc/example/server/repository/RawJdbcRepository.java

deleted100644 → 0

+0 −132

Original line number	Diff line number	Diff line
		/*
		* Copyright 1999-2015 dangdang.com.
		* <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.shardingjdbc.example.server.repository;

		import io.shardingjdbc.core.api.HintManager;

		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 RawJdbcRepository {

		private final DataSource dataSource;

		public RawJdbcRepository(final DataSource dataSource) {
		this.dataSource = dataSource;
		}

		public void demo() throws SQLException {
		createTable();
		insertData();
		System.out.println("1.Equals Select--------------");
		printEqualsSelect();
		System.out.println("2.In Select--------------");
		printInSelect();
		System.out.println("3.Hint Select--------------");
		printHintSimpleSelect();
		dropTable();
		}

		public void createTable() throws SQLException {
		execute(dataSource, "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(dataSource, "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))");
		}

		public void dropTable() throws SQLException {
		execute(dataSource, "DROP TABLE t_order_item");
		execute(dataSource, "DROP TABLE t_order");
		}

		public void insertData() throws SQLException {
		for (int i = 1; i < 10; i++) {
		long orderId = executeAndGetGeneratedKey(dataSource, "INSERT INTO t_order (user_id, status) VALUES (10, 'INIT')");
		execute(dataSource, String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 10)", orderId));
		orderId = executeAndGetGeneratedKey(dataSource, "INSERT INTO t_order (user_id, status) VALUES (11, 'INIT')");
		execute(dataSource, String.format("INSERT INTO t_order_item (order_id, user_id) VALUES (%d, 11)", orderId));
		}
		}

		public void printEqualsSelect() 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 conn = dataSource.getConnection();
		PreparedStatement preparedStatement = conn.prepareStatement(sql)) {
		preparedStatement.setInt(1, 10);
		printSimpleSelect(preparedStatement);
		}
		}

		public void printInSelect() 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 conn = dataSource.getConnection();
		PreparedStatement preparedStatement = conn.prepareStatement(sql)) {
		preparedStatement.setInt(1, 10);
		preparedStatement.setInt(2, 11);
		printSimpleSelect(preparedStatement);
		}
		}

		public void printHintSimpleSelect() throws SQLException {
		String sql = "SELECT i.* FROM t_order o JOIN t_order_item i ON o.order_id=i.order_id";
		try (
		HintManager hintManager = HintManager.getInstance();
		Connection conn = dataSource.getConnection();
		PreparedStatement preparedStatement = conn.prepareStatement(sql)) {
		hintManager.addDatabaseShardingValue("t_order", "user_id", 11);
		printSimpleSelect(preparedStatement);
		}
		}

		private void printSimpleSelect(final PreparedStatement preparedStatement) throws SQLException {
		try (ResultSet rs = preparedStatement.executeQuery()) {
		while (rs.next()) {
		System.out.print("order_item_id:" + rs.getLong(1) + ", ");
		System.out.print("order_id:" + rs.getLong(2) + ", ");
		System.out.print("user_id:" + rs.getInt(3));
		System.out.println();
		}
		}
		}

		private void execute(final DataSource dataSource, final String sql) throws SQLException {
		try (
		Connection conn = dataSource.getConnection();
		Statement statement = conn.createStatement()) {
		statement.execute(sql);
		}
		}

		private long executeAndGetGeneratedKey(final DataSource dataSource, final String sql) throws SQLException {
		long result = -1;
		try (
		Connection conn = dataSource.getConnection();
		Statement statement = conn.createStatement()) {
		statement.executeUpdate(sql, Statement.RETURN_GENERATED_KEYS);
		ResultSet resultSet = statement.getGeneratedKeys();
		if (resultSet.next()) {
		result = resultSet.getLong(1);
		}
		}
		return result;
		}
		}

sharding-proxy-example/src/main/java/io/shardingjdbc/example/proxy/ServerMain.java

+3 −2

Original line number	Diff line number	Diff line
		package io.shardingjdbc.example.proxy;

		import io.shardingjdbc.example.proxy.repository.RawJdbcRepository;
		import io.shardingjdbc.proxy.transport.ShardingProxy;
		import io.shardingjdbc.proxy.frontend.ShardingProxy;
		import org.apache.commons.dbcp.BasicDataSource;

		import javax.sql.DataSource;
		@@ -10,7 +10,7 @@ import java.sql.SQLException;
		public final class ServerMain {

		public static void main(String[] args) throws InterruptedException, SQLException {
		startServer();
		// startServer();
		RawJdbcRepository rawJdbcRepository = new RawJdbcRepository(createDataSource());
		rawJdbcRepository.demo();
		}
		@@ -35,6 +35,7 @@ public final class ServerMain {
		BasicDataSource result = new BasicDataSource();
		result.setDriverClassName(com.mysql.jdbc.Driver.class.getName());
		result.setUrl("jdbc:mysql://localhost:3307/demo_ds");
		// result.setUrl("jdbc:mysql://localhost:3307/demo_ds?useServerPrepStmts=true");
		result.setUsername("root");
		result.setPassword("");
		return result;