动态代理模拟连接池

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;
}

}

package 动态代理实现连接池;


//异常类
public class HaveBeenClosed extends Exception{
private static final long serialVersionUID = 1L;
public HaveBeenClosed(String message){
super(message);
}
}