package 动态代理实现连接池;
/**
* 动态代理自定义连接池:
* 技术难题:1.增强close方法,使得close方法不在 关闭连接而是归还连接
* 解决办法:使用动态代理增强close方法
* 2.归还连接后,连接未真正关闭,那么原来指向该连接的变量con仍可以掉用该连接的方法
* 解决办法:在代理上添加标记,通过判断标记来决定原来指向该连接的con是否可以继续调用该连接的方法;
* 如果单纯标记true或false,那么当下一个线程获取该连接后原来指向该连接的con又可以调用该连接了,
* 所以要跟线程联系起来,当线程获取连接时,以当前线程作为键,代理类作为值放入map集合中,归还时
* 从集合中删除,然后在调用连接的方法前进行判断,如果可以从map中取到代理类就可以执行,取不到就
* 抛出异常。
* */
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.ResourceBundle;
public class 动态代理实现数据库连接池 {
public static final Map<Thread,Connection> map = new HashMap<>();
private final static String url;
private final static String user;
private final static String password;
private static List<Connection> list =new ArrayList<>();
static{
ResourceBundle rb = ResourceBundle.getBundle("jdbc");
url = rb.getString("url");
user = rb.getString("user");
password = rb.getString("password");
try {
Class.forName(rb.getString("driverClass"));
} catch (Exception e) {
e.printStackTrace();
}
try {
for(int i = 0;i<5;i++){
Connection con = DriverManager.getConnection(url, user, password);
Connection conn = getProxy(con);
list.add(conn);
}
} catch (SQLException e) {
System.out.println(e);
throw new RuntimeException("获取连接失败!!!");
}
}
//获取连接的方法
public static Connection getCon(){
Connection con = (Connection) list.remove(0);
map.put(Thread.currentThread(),con) ;
return con;
}
//关闭资源的方法
public static void colse(ResultSet rs,Statement stmt,Connection con){
if(rs!=null){
try {
rs.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if(stmt!=null){
try {
stmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if(con!=null){
try {
con.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
public static void colse(Statement stmt,Connection con){
colse(null,stmt,con);
}
//下面是使用动态代理增强close的代码
//还链接方法
private static void back(Connection proxy) {
list.add(proxy);
}
//获取代理类方法
public static Connection getProxy(final Connection con){
Object conn = Proxy.newProxyInstance(动态代理实现数据库连接池.class.getClassLoader(),new Class[]{Connection.class}, new InvocationHandler(){
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
if(method.getName().equals("close")){
back((Connection)map.remove(Thread.currentThread()));
}else{
if(map.get(Thread.currentThread())==proxy){
return method.invoke(con, args);
}else{
throw new HaveBeenClosed("Connection used after been closed!");
}
}
return null;
}
});
return (Connection)conn;
}
/**
* 动态代理自定义连接池:
* 技术难题:1.增强close方法,使得close方法不在 关闭连接而是归还连接
* 解决办法:使用动态代理增强close方法
* 2.归还连接后,连接未真正关闭,那么原来指向该连接的变量con仍可以掉用该连接的方法
* 解决办法:在代理上添加标记,通过判断标记来决定原来指向该连接的con是否可以继续调用该连接的方法;
* 如果单纯标记true或false,那么当下一个线程获取该连接后原来指向该连接的con又可以调用该连接了,
* 所以要跟线程联系起来,当线程获取连接时,以当前线程作为键,代理类作为值放入map集合中,归还时
* 从集合中删除,然后在调用连接的方法前进行判断,如果可以从map中取到代理类就可以执行,取不到就
* 抛出异常。
* */
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.ResourceBundle;
public class 动态代理实现数据库连接池 {
public static final Map<Thread,Connection> map = new HashMap<>();
private final static String url;
private final static String user;
private final static String password;
private static List<Connection> list =new ArrayList<>();
static{
ResourceBundle rb = ResourceBundle.getBundle("jdbc");
url = rb.getString("url");
user = rb.getString("user");
password = rb.getString("password");
try {
Class.forName(rb.getString("driverClass"));
} catch (Exception e) {
e.printStackTrace();
}
try {
for(int i = 0;i<5;i++){
Connection con = DriverManager.getConnection(url, user, password);
Connection conn = getProxy(con);
list.add(conn);
}
} catch (SQLException e) {
System.out.println(e);
throw new RuntimeException("获取连接失败!!!");
}
}
//获取连接的方法
public static Connection getCon(){
Connection con = (Connection) list.remove(0);
map.put(Thread.currentThread(),con) ;
return con;
}
//关闭资源的方法
public static void colse(ResultSet rs,Statement stmt,Connection con){
if(rs!=null){
try {
rs.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if(stmt!=null){
try {
stmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
if(con!=null){
try {
con.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
public static void colse(Statement stmt,Connection con){
colse(null,stmt,con);
}
//下面是使用动态代理增强close的代码
//还链接方法
private static void back(Connection proxy) {
list.add(proxy);
}
//获取代理类方法
public static Connection getProxy(final Connection con){
Object conn = Proxy.newProxyInstance(动态代理实现数据库连接池.class.getClassLoader(),new Class[]{Connection.class}, new InvocationHandler(){
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
if(method.getName().equals("close")){
back((Connection)map.remove(Thread.currentThread()));
}else{
if(map.get(Thread.currentThread())==proxy){
return method.invoke(con, args);
}else{
throw new HaveBeenClosed("Connection used after been closed!");
}
}
return null;
}
});
return (Connection)conn;
}
}
package 动态代理实现连接池;
//异常类
public class HaveBeenClosed extends Exception{
private static final long serialVersionUID = 1L;
public HaveBeenClosed(String message){
super(message);
}
}