java - 向界面添加信息

Question

连接.java

public interface Connect {
void commit();
void close();
void setTransactional();
void executeCommand();
}

MainApp.java

public class MainApp {
    public static void main(String[] args) {
        Connect conn=MainApp.getTransactionalConnection();
        conn.executeCommand();
        conn.executeCommand();
        conn.executeCommand();
        conn.executeCommand();
        closeConnection(conn);      
    }
    public static Connect getNormalConnection(){
        return DriverManager.getConnection(); //returns class i have no access to , dont even know its name
    }
    public static Connect getTransactionalConnection(){
        Connect conn =  getNormalConnection();
        conn.setTransactional();
        return conn;
    }
    private static void closeConnection(Connect conn) {
        //something like if connection is transactional then conn.commit();

        conn.close();
    }

}

我有大约 100 个看起来像 main() 的函数。问题是我不想更改这 100 个函数的代码，只想将 1 行 getNormalConnection 更改为 getTransactionalConnection 。我无权访问也不知道实现 Connect 接口(interface)的类的名称。我只想添加对 closeConnection() 的提交，无论是否创建了事务连接。

Answer 1

一个简单的答案是拥有自己的 Connect 实现，它要么公开 isTransactional 属性，要么在事务性时自动提交，将其他所有内容委托(delegate)给底层连接:

public class AutoCommitConnect implements Connect {
    private final Connect connect;
    private boolean transactional = false;
    private boolean committed = false;

    public AutoCommitConnect(Connect delegate) {
        this.delegate = delegate;
    }

    public void commit() {
        delegate.commit();
        committed = true;
    }

    public void setTransactional() {
        transactional = true;
        delegate.setTransactional();
    }

    public void close() {
        if (transactional && !committed) {
            commit();
        }
        delegate.close();
    }

    public void executeCommand() {
        delegate.executeCommand();
    }
}

然后更改连接返回方法来使用它:

public static Connect getNormalConnection() {
    // getTransactionalConnection will call setTransactional on the wrapper,
    // which will remember that it needs to commit
    return new AutoCommitConnect(DriverManager.getConnection());
}