PostgreSql JDBC事务操作方法详解

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);
}
}
}