java添加数据后获取id

原创

mob64ca12e63b18 2023-10-09 13:50:19 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e63b18的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java添加数据后获取ID

在Java开发中，有时候我们需要在向数据库中添加数据后获取生成的ID。这在一些需要进行关联的数据表中尤为常见，比如添加一条新的用户记录后，需要获取该用户的ID以便在其他关联表中进行引用。本文将介绍在Java中如何实现添加数据后获取ID的方法，并给出相应的代码示例。

1. 使用自增主键

在数据库表中，我们通常会定义一个自增主键来标识每条记录的唯一性。当我们插入一条新的记录时，数据库会自动为该记录生成一个唯一的ID。在Java中，我们可以通过以下步骤来获取该ID：

执行插入操作，将数据添加到数据库中。
获取刚刚插入的记录的ID。

下面是一个使用自增主键的示例代码：

import java.sql.*;

public class Example {
    public static void main(String[] args) {
        String url = "jdbc:mysql://localhost:3306/test";
        String username = "root";
        String password = "123456";

        try (Connection conn = DriverManager.getConnection(url, username, password)) {
            String sql = "INSERT INTO users (name, age) VALUES (?, ?)";
            PreparedStatement stmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
            stmt.setString(1, "John");
            stmt.setInt(2, 30);
            stmt.executeUpdate();
            
            ResultSet rs = stmt.getGeneratedKeys();
            if (rs.next()) {
                int id = rs.getInt(1);
                System.out.println("Generated ID: " + id);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

在上述代码中，我们创建了一个PreparedStatement对象，并将Statement.RETURN_GENERATED_KEYS作为参数传递给它的构造方法。这样一来，当我们执行executeUpdate()方法后，就能够通过调用getGeneratedKeys()方法获取生成的ID。然后我们可以使用ResultSet对象来获取具体的ID值。

2. 使用UUID

除了使用自增主键外，我们还可以使用UUID（Universally Unique Identifier）作为记录的唯一标识。UUID是一种由网络中多台计算机生成的标识符，具有全球唯一性。在Java中，我们可以使用java.util.UUID类来生成UUID，并将其作为记录的ID。

下面是一个使用UUID的示例代码：

import java.sql.*;
import java.util.UUID;

public class Example {
    public static void main(String[] args) {
        String url = "jdbc:mysql://localhost:3306/test";
        String username = "root";
        String password = "123456";

        try (Connection conn = DriverManager.getConnection(url, username, password)) {
            String sql = "INSERT INTO users (id, name, age) VALUES (?, ?, ?)";
            PreparedStatement stmt = conn.prepareStatement(sql);
            stmt.setString(1, UUID.randomUUID().toString());
            stmt.setString(2, "John");
            stmt.setInt(3, 30);
            stmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

在上述代码中，我们在插入数据之前生成了一个UUID，并将其作为ID插入到数据库中。

类图

下面是一个简化的类图，展示了上述示例代码中的类及其关系：

classDiagram
    Example --> Connection
    Example --> DriverManager
    Example --> Statement
    Example --> PreparedStatement
    Example --> ResultSet
    Example --> SQLException

在类图中，Example类依赖于Connection、DriverManager、Statement、PreparedStatement、ResultSet和SQLException这些类，通过调用它们的方法来完成数据库操作。