PostgreSql JDBC事务操作方法详解

时间:2022-11-19 Zeuss 人气:0

JDBC事务相关方法简介

本文将借助示例，简单讲解下JDBC操作Pg事务的流程。

首先来简单讲解下事务的定义：为了确保两个（多个）数据库操作都生效，或者两个操作都不发生，可以使用事务。根据定义，事务是作为单个单元执行的一组语句。换句话说，要么所有语句都成功执行，要么没有执行。

禁用自动提交模式

当建立与PostgreSQL数据库的连接时,它处于自动提交模式。这意味着每个SQL语句都被视为事务并自动提交。

如果要在事务中封装一个或多个语句，则必须禁用自动提交模式。为此，我们可以调用Connection.setAutoCommit()方法来修改SQL提交模式:

Connection.setAutoCommit(false);

最佳做法是仅对事务模式禁用自动提交模式。它避免为多个语句保留数据库锁。

提交事务

要提交事务，请调用Connection对象的commit方法，如下所示:

Connection.commit();

当调用commit()方法，所有前面的SQL语句作为一个单元一起提交。

回滚事务

既然使用了事务，那我们肯定会有回滚的时候，我们可以使用rollback()方法来中止当前事务并将值恢复为原始值。

Connection.rollback();

PostgreSQL JDBC 事务示例

让我们举一个使用JDBC API执行PostgreSQL事务的示例。

首先，创建一个表示ProRank的实体类，如下所示:

import lombok.Data;
@Data
public class ProRank {
    Integer id;
    String name;
    String team;
    String line;
    Integer rank;
}

然后，编写以下代码，供我们测试事务操作。

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import java.sql.*;
@SpringBootTest
class JdbcTrasationTests {
    private final String url = "jdbc:p6spy:postgresql://localhost:5432/postgres";
    private final String user = "postgres";
    private final String password = "112233";
    /**
     * 连接PostgreSql数据库
     *
     * @return Connection
     * @throws SQLException
     */
    public Connection connect() throws SQLException {
        return DriverManager.getConnection(url, user, password);
    }
    @Test
    void testTrasation() {
        ProRank proRank = new ProRank();
        proRank.setLine("Mid");
        proRank.setName("Faker");
        proRank.setTeam("T1");
        proRank.setRank(0);
        //调用
        addProAndUpdateRank(proRank,2222);
    }
    /**
     * 关闭一个AutoCloseable对象
     *
     * @param closeable
     */
    private void close(AutoCloseable closeable) {
        try {
            if (closeable != null) {
                closeable.close();
            }
        } catch (Exception e) {
            System.out.println(e.getMessage());
        }
    }
    /**
     * 插入一条选手记录，更新他的rank值
     *
     * @param proRank
     * @param rank
     */
    public void addProAndUpdateRank(ProRank proRank, Integer rank) {
        Connection conn = null;
        PreparedStatement pstmt = null;
        PreparedStatement pstmt2 = null;
        ResultSet rs = null;
        // 插入一条数据
        String SQL = "INSERT INTO pro_rank(name,team,line,rank) VALUES(?,?,?,?)";
        // 更新他的rank值
        String SQLUpdateRank = "UPDATE pro_rank SET rank = ? WHERE id = ?;";
        int id = 0;
        try {
            // 链接数据库
            conn = connect();
            conn.setAutoCommit(false);
            // 插入一条数据
            pstmt = conn.prepareStatement(SQL, Statement.RETURN_GENERATED_KEYS);
            pstmt.setString(1, proRank.getName());
            pstmt.setString(2, proRank.getTeam());
            pstmt.setString(3, proRank.getLine());
            pstmt.setInt(4, proRank.getRank());
            int affectedRows = pstmt.executeUpdate();
            // 判断是否生效
            if (affectedRows > 0) {
                // 获取返回的id
                rs = pstmt.getGeneratedKeys();
                if (rs.next()) {
                    id = rs.getInt(1);
                    if (id > 0) {
                        pstmt2 = conn.prepareStatement(SQLUpdateRank);
                        pstmt2.setInt(2, id);
                        pstmt2.setInt(1, rank);
                        pstmt2.executeUpdate();
                    }
                }
            } else {
                // 如果新增数据失败，回滚
                conn.rollback();
            }
            // 提交事务
            conn.commit();
            System.out.println("插入选手数据成功！更新选手rank成功，数据id：" + id);
        } catch (SQLException sqlException) {
            System.out.println(sqlException.getMessage());
            sqlException.printStackTrace();
            // 回滚事务
            System.out.println("回滚事务...");
            try {
                if (conn != null) {
                    conn.rollback();
                }
            } catch (SQLException e) {
                System.out.println(e.getMessage());
                e.printStackTrace();
            }
        } finally {
            close(rs);close(pstmt);close(pstmt2);close(conn);
        }
    }
}

让我们看一下上面的代码，他包含三个方法

connect() 方法建立与数据库连接，并返回连接对象。

close() 方法关闭数据库操作可关闭的对象，如Resultset、Statement和Connection。

addProAndUpdateRank()方法插入新的选手，并在事务中更新选手的rank字段。此方法包含逻辑如下：

首先，在pro_rank表中插入一条新的选手数据。
接下来，获取新插入的选手数据的id
然后，更新插入选手的rank值。
之后，如果步骤2和3均成功，则提交事务。否则，回滚事务
最后，关闭ResultSet、PreparedStatement和Connection对象。

如果我们在第一个场景中执行程序，我们会得到以下结果:

插入选手数据成功！更新选手rank成功，数据id：14

我们可以通过查询pro_rank表格，来查看上述代码执行结果:

SELECT * FROM "public"."pro_rank" LIMIT 1000 OFFSET 0;

现在，让我们测试一下事务回滚的情况，比如，我们可以在插入一条数据的时候，将name字段赋值为超出数据库长度的字串，运行程序结果如下:

ERROR: value too long for type character varying(7)

事务将回滚，并且没有任何内容插入pro_rank表

加载全部内容