JDBC以及相关技术学习(十)----拦截Connection的close方法

上一节的例子中，可能存在用户调用Connection后，直接调用close 的方法进行关闭，我们重新实现一个Connection，然后重新写close方法，代码如下：

public class MyConnection implements Connection {

//写这个类的目的是为了拦截Close方法 //类的组合

private Connection realConnection;//注意，我们是拦截close方法，对于其他方法，我们还是要用真实的Connection处理

private MyDataSource2 dataSource;

private int currentUserCount = 0;

public MyConnection(Connection realConnection,MyDataSource2 dataSource) {

this.realConnection = realConnection;

this.dataSource = dataSource;

// TODO Auto-generated constructor stub

}

@Override

public void clearWarnings() throws SQLException {

// TODO Auto-generated method stub

realConnection.clearWarnings();

}

@Override

public void close() throws SQLException {

// TODO Auto-generated method stub

this.dataSource.connectionsPool.add(this);//注意这里不能放realConnection，否则下次又自动关闭了

}

@Override

public void commit() throws SQLException {

// TODO Auto-generated method stub

realConnection.commit();

}

@Override

public Array createArrayOf(String typeName, Object[] elements)

throws SQLException {

return realConnection.createArrayOf(typeName, elements);

// TODO Auto-generated method stub

}

}

当然，这样要实现大量的代码，包括Connection的其他方法我们都要处理，现在换一种更简洁的方法，就是使用动态代理。

public class MyConnectionHandler implements InvocationHandler {

private Connection realConnection;

private Connection warpedConnection;

private MyDataSource2 dataSource;

MyConnectionHandler(MyDataSource2 dataSource){

this.dataSource = dataSource;

}

Connection bind(Connection realConn){

this.realConnection = realConn;

//Proxy可以理解为一个程序员,也会写代码，不过要告诉他怎么去写

Connection warpedConnection
= (Connection)Proxy.newProxyInstance(

this.getClass().getClassLoader(),

new Class[]{Connection.class},//接口可以很多个

this//调用处理器

);

this.warpedConnection = warpedConnection;

return warpedConnection;

}

@Override

public Object invoke(Object proxy, Method method, Object[] args)

throws Throwable {

if("close".equals(method.getName())){

dataSource.free(warpedConnection);

}

return method.invoke(realConnection, args);

}

}

MyConnectionHandler 类相当于对Connection进行重新包装，把close()拦截了。