代理模式 - java如何使用代理实现这个业务场景?

 诸葛烈火_220 发布于 2022-10-26 16:55

我想写一个数据库连接池来为其他数据库访问类提供Connection, 但是我这个数据库连接池的产生是在数据库访问类写完之后, 因此所有的访问在结尾都有Connection.close()的调用, 因此我就在想, 当调用connection.close()时, 将该connection加入数据库连接池中. 所以我就写了一个代理类, 但是想的很美, 写起来就不知道该怎么写了. 如何将这个proxy产生的代理对象加入连接池中呢?

我目前的实现如下:

class EmptyInvocationHandler implements InvocationHandler{
    private EmptyConnection emptyConnection ;
    private ConnectionPool connectionPool ;

    public EmptyInvocationHandler(EmptyConnection emptyConnection ,ConnectionPool connectionPool) {
        this.connectionPool = connectionPool ;
        this.emptyConnection = emptyConnection;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if(method.getName().equals("close")){
            close();
            return null ;
        }
        else
            return method.invoke(emptyConnection , args) ;
    }
    //如果这样实现的话, 就是把真实的对象加入了连接池中, 如何把这个代理对象加入池中呢?
    private void close(){
        connectionPool.recycleConnection(emptyConnection);
    }
}

//模拟Connecter接口
interface EmptyConnectionI{
    void doSomething() ;
    void close() ;
}

//connecter的默认实现
class EmptyConnection implements EmptyConnectionI{
    public void doSomething(){
        System.out.println("do something");
    }

    public void close(){
        System.out.println("real close");
    }
}

class ConnectionPool{
    private List list = new ArrayList<>() ;
    private boolean inited = false ;
    public EmptyConnectionI getConnection(){
        if(!inited)
            init() ;
        return list.remove(0) ;
    }

    private void init(){
        EmptyConnection emptyConnection = new EmptyConnection() ;
        try {
            Class proxyClass = Proxy.getProxyClass(EmptyConnection.class.getClassLoader()
                    , new Class[]{EmptyConnectionI.class});
            EmptyConnectionI emptyConnectionI = (EmptyConnectionI) proxyClass.getConstructor(new Class[]{InvocationHandler.class})
                    .newInstance(new EmptyInvocationHandler(emptyConnection, this));
            list.add(emptyConnectionI) ;
        }
        catch (Exception e){
            e.printStackTrace();
            System.exit(-1);
        }
    }

    public void recycleConnection(EmptyConnectionI connection){
        list.add(connection) ;
    }
}
2 个回答
  • 可以实现的,MyBatis的PooledConnection和PooledDataSource这两个类就做了这样的事情,先标个记,回头写篇这两个类的源码分析文章再贴回来。

    2022-10-27 01:03 回答
  • 你怎么知道什么时候是需要close连接?

    2022-10-27 01:03 回答
撰写答案
今天,你开发时遇到什么问题呢?
立即提问
热门标签
PHP1.CN | 中国最专业的PHP中文社区 | PNG素材下载 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有