move JDBCBackendHandler.getConnection => ConnectionManager

 * </p>

 */

package io.shardingsphere.proxy.backend.common;

package io.shardingsphere.proxy.backend.common.jdbc;

import io.shardingsphere.proxy.backend.common.ProxyMode;

import io.shardingsphere.proxy.config.RuleRegistry;

import lombok.AccessLevel;

import lombok.NoArgsConstructor;

import javax.sql.DataSource;

import java.sql.Connection;

import java.sql.SQLException;

import java.util.HashMap;

import java.util.Map;

/**

 * Hold the connection when proxy mode is CONNECTION_STRICTLY.

 * Connection manager.

 *

 * @author zhaojun

 */

@NoArgsConstructor(access = AccessLevel.PRIVATE)

public final class ProxyConnectionHolder {

public final class ConnectionManager {

    private static final ThreadLocal<Map<DataSource, Connection>> RESOURCE = new ThreadLocal<Map<DataSource, Connection>>() {

        }

    };

    /**

     * Set connection associate with datasource into Thread.

     *

     * @param dataSource DataSource

     * @param connection Connection

     */

    public static void setConnection(final DataSource dataSource, final Connection connection) {

        RESOURCE.get().put(dataSource, connection);

    }

    /**

     * Get connection of current thread datasource.

     *

     * @param dataSource Datasource

     * @return Connection

     * @throws SQLException SQL exception

     */

    public static Connection getConnection(final DataSource dataSource) {

        return RESOURCE.get().get(dataSource);

    public static Connection getConnection(final DataSource dataSource) throws SQLException {

        if (ProxyMode.MEMORY_STRICTLY == RuleRegistry.getInstance().getProxyMode()) {

            return dataSource.getConnection();

        }

        return RESOURCE.get().containsKey(dataSource) ? RESOURCE.get().get(dataSource) : RESOURCE.get().put(dataSource, dataSource.getConnection());

    }

    /**

import io.shardingsphere.core.routing.SQLUnit;

import io.shardingsphere.core.routing.router.masterslave.MasterSlaveRouter;

import io.shardingsphere.proxy.backend.common.BackendHandler;

import io.shardingsphere.proxy.backend.common.ProxyConnectionHolder;

import io.shardingsphere.proxy.backend.common.ProxyMode;

import io.shardingsphere.proxy.backend.common.ResultList;

import io.shardingsphere.proxy.backend.resource.BaseJDBCResource;

import io.shardingsphere.proxy.config.RuleRegistry;

import lombok.Getter;

import lombok.Setter;

import javax.sql.DataSource;

import javax.transaction.Status;

import javax.transaction.SystemException;

import java.sql.Connection;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

    }

    protected abstract DatabaseProtocolPacket newDatabaseProtocolPacket(int sequenceId, List<Object> data);

    protected final Connection getConnection(final DataSource dataSource) throws SQLException {

        if (ProxyMode.MEMORY_STRICTLY == ruleRegistry.getProxyMode()) {

            return dataSource.getConnection();

        }

        Connection result = ProxyConnectionHolder.getConnection(dataSource);

        if (null == result) {

            result = dataSource.getConnection();

            ProxyConnectionHolder.setConnection(dataSource, result);

        }

        return result;

    }

}

import io.shardingsphere.core.routing.PreparedStatementRoutingEngine;

import io.shardingsphere.core.routing.SQLExecutionUnit;

import io.shardingsphere.core.routing.SQLRouteResult;

import io.shardingsphere.proxy.backend.common.jdbc.ConnectionManager;

import io.shardingsphere.proxy.backend.common.ProxyMode;

import io.shardingsphere.proxy.backend.common.jdbc.JDBCBackendHandler;

import io.shardingsphere.proxy.backend.mysql.MySQLPacketStatementExecuteQueryResult;

import io.shardingsphere.proxy.transport.mysql.packet.command.statement.execute.PreparedStatementParameter;

import lombok.Getter;

import javax.sql.DataSource;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.SQLException;

    @Override

    protected PreparedStatement prepareResource(final SQLExecutionUnit sqlExecutionUnit, final SQLStatement sqlStatement) throws SQLException {

        DataSource dataSource = ruleRegistry.getDataSourceMap().get(sqlExecutionUnit.getDataSource());

        Connection connection = getConnection(dataSource);

        Connection connection = ConnectionManager.getConnection(ruleRegistry.getDataSourceMap().get(sqlExecutionUnit.getDataSource()));

        PreparedStatement result = sqlStatement instanceof InsertStatement

                ? connection.prepareStatement(sqlExecutionUnit.getSqlUnit().getSql(), Statement.RETURN_GENERATED_KEYS) : connection.prepareStatement(sqlExecutionUnit.getSqlUnit().getSql());

        for (int i = 0; i < preparedStatementParameters.size(); i++) {

import io.shardingsphere.core.routing.SQLExecutionUnit;

import io.shardingsphere.core.routing.SQLRouteResult;

import io.shardingsphere.core.routing.StatementRoutingEngine;

import io.shardingsphere.proxy.backend.common.jdbc.ConnectionManager;

import io.shardingsphere.proxy.backend.common.ProxyMode;

import io.shardingsphere.proxy.backend.common.jdbc.JDBCBackendHandler;

import io.shardingsphere.proxy.backend.mysql.MySQLPacketQueryResult;

    @Override

    protected Statement prepareResource(final SQLExecutionUnit sqlExecutionUnit, final SQLStatement sqlStatement) throws SQLException {

        Connection connection = getConnection(ruleRegistry.getDataSourceMap().get(sqlExecutionUnit.getDataSource()));

        Connection connection = ConnectionManager.getConnection(ruleRegistry.getDataSourceMap().get(sqlExecutionUnit.getDataSource()));

        Statement result = connection.createStatement();

        ProxyJDBCResource proxyJDBCResource = (ProxyJDBCResource) getJdbcResource();

        proxyJDBCResource.addConnection(connection);

import io.netty.buffer.ByteBuf;

import io.netty.channel.ChannelHandlerContext;

import io.shardingsphere.core.routing.router.masterslave.MasterVisitedManager;

import io.shardingsphere.proxy.backend.common.ProxyConnectionHolder;

import io.shardingsphere.proxy.backend.common.jdbc.ConnectionManager;

import io.shardingsphere.proxy.frontend.common.FrontendHandler;

import io.shardingsphere.proxy.frontend.common.executor.ExecutorGroup;

import io.shardingsphere.proxy.transport.common.packet.DatabaseProtocolPacket;

                    }

                } finally {

                    MasterVisitedManager.clear();

                    ProxyConnectionHolder.clear();

                    ConnectionManager.clear();

                }

            }

        });