178.java

package com.github.mjdbc;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

/**
 * Lightweight java.sql.Connection wrapper.
 * Used in pair with DbPreparedStatement and DbOpX classes for automatic statements cleanup when connection is closed.
 */
public class DbConnection {

    @NotNull
    protected final DbImpl db;

    /**
     * Native SQL connection.
     */
    @Nullable
    private Connection sqlConnection;

    /**
     * List of statements to close when connection is closed.
     */
    @NotNull
    protected final List<DbPreparedStatement> statementsToClose = new ArrayList<>();

    protected DbConnection(@NotNull DbImpl db) {
        this.db = db;
    }

    @NotNull
    public Connection getConnection() {
        return ensureConnection();
    }

    @NotNull
    private Connection ensureConnection() {
        if (sqlConnection == null) {
            try {
                sqlConnection = db.dataSource.getConnection();
            } catch (SQLException e) {
                throw new RuntimeException("Failed to get connection from DataSource!", e);
            }
        }
        return sqlConnection;
    }

    public void commit() throws SQLException {
        // commit only if connection was opened.
        if (sqlConnection != null) {
            sqlConnection.commit();
        }
    }

    public void rollback() throws SQLException {
        // rollback only if connection was opened.
        if (sqlConnection != null) {
            sqlConnection.rollback();
        }
    }

    public void close() throws SQLException {
        for (DbPreparedStatement s : statementsToClose) {
            try {
                s.close();
            } catch (SQLException ignored) {
            }
        }
        if (sqlConnection != null) {
            sqlConnection.close();
        }
    }
}