java - 将行添加到具有唯一列 : Get existing ID values plus newly-created ones 的表

Question

表结构如下

  1. mail_contacts(id,mail_id);//stores mail id, id is primary key
                                  mail_id is unique key
  2. inv_table(mid,aid,add_date);//mapped mail id with user id. mid is 
                                   mail_contacts id and aid is userid
                                   (mid,aid) is primary key  

插入后，我将多个邮件 ID 存储在 mail_contacts 中，我正在获取其插入的 ID 并将其存储在 inv_table 的帮助下。 如果任何 mail_id 没有存储在 mail_contacts 中，那么它工作正常。 但如果 mail_id 存储在 mail_contacts 中，则插入将终止。

我想要什么 如果 mail_id 存储在 mail_contacts 中，那么它应该获取它的 id 以存储在 inv_table.

我在努力

    PreparedStatement ps = null;
    PreparedStatement ps1 = null;
    ResultSet rs = null, res = null;
    Connection con = null;
    String status = "success";
    ArrayList ar = null;
    Invitation invi = null;
    int i = 0;

    public String insert(List<MailidInvitation> invitationList, Long aid) {
        con = ConnectionFactory.getConnection();
        try {
            ps = con.prepareStatement("insert into mail_contacts(mail_id)"
                    + " values(?)", Statement.RETURN_GENERATED_KEYS);

            for (MailidInvitation a : invitationList) {
                ps.setString(1, a.getMailId());
                ps.addBatch();
            }
            ps.executeBatch();
            ResultSet keys = ps.getGeneratedKeys();
            while (keys.next()) {
                long id = keys.getLong(1);
                invitationList.get(i).setId(id);
                System.out.println("generated id is " + id);
                i++;
            }

            ps1 = con.prepareStatement("insert into inv_table(mid,aid)"
                    + " values(?,?)");
            for (MailidInvitation a : invitationList) {
                ps1.setLong(1, a.getId());
                ps1.setLong(2, aid);
                ps1.addBatch();
            }
            ps1.executeBatch();
        } catch (SQLException e) {
            status = "failure";
            System.out.println("SQLException1 " + e);
        } finally {
            try {
                con.close();
            } catch (SQLException e) {
                System.out.println("SQLException2 " + e);
            }
        }
        System.out.println("Status is " + status);

        return status;
    }

如何做到这一点？

解决这类问题最好的方法是什么

我使用的是mysql数据库

Answer 1

第一种方法在处理批量 INSERT 时不对 JDBC 驱动程序的行为做出任何假设。它通过

避免了潜在的 INSERT 错误

• 在表中查询任何现有的 mail_id我们当前数据集中的值，
• 记下相应的id那些的值(value)mail_id确实存在的值(value)观，
• 插入 mail_id不存在的值，并检索它们的(新)id值(value)观，然后
• 在另一个表 ( inv_table ) 中插入行。

try (Connection dbConn = DriverManager.getConnection(myConnectionString, "root", "usbw")) {
    dbConn.setAutoCommit(false);

    // test data and setup
    Long aid = 123L;
    List<MailidInvitation> invitationList = new ArrayList<MailidInvitation>();
    invitationList.add(new MailidInvitation(13L));
    invitationList.add(new MailidInvitation(11L));
    invitationList.add(new MailidInvitation(12L));
    // remove stuff from previous test run
    try (Statement s = dbConn.createStatement()) {
        s.executeUpdate("DELETE FROM mail_contacts WHERE mail_id IN (11,13)");
    }
    try (PreparedStatement ps = dbConn.prepareStatement(
            "DELETE FROM inv_table WHERE aid=?")) {
        ps.setLong(1, aid);
        ps.executeUpdate();
    }

    // real code starts here
    //
    // create a Map to hold `mail_id` and their corresponding `id` values 
    Map<Long, Long> mailIdMap = new TreeMap<Long, Long>();
    for (MailidInvitation a : invitationList) {
        // mail_id, id (id is null for now)
        mailIdMap.put(a.getId(), null);
    }

    // build an SQL statement to retrieve any existing values
    StringBuilder sb = new StringBuilder(
            "SELECT id, mail_id " +
            "FROM mail_contacts " +
            "WHERE mail_id IN (");
    int n = 0;
    for (Map.Entry<Long, Long> entry : mailIdMap.entrySet()) {
        if (n++ > 0) sb.append(',');
        sb.append(entry.getKey());
    }
    sb.append(')');
    String sql = sb.toString();

    // run the query and save the results (if any) to the Map
    try (Statement s = dbConn.createStatement()) {
        // <demo>
        System.out.println(sql);
        // </demo>
        try (ResultSet rs = s.executeQuery(sql)) {
            while (rs.next()) {
                mailIdMap.put(rs.getLong("mail_id"), rs.getLong("id"));
            }
        }
    }

    // <demo>
    System.out.println();
    System.out.println("mailIdMap now contains:");
    // </demo>

    // build a list of the `mail_id` values to INSERT (where id == null)
    //     ... and print the existing mailIdMap values for demo purposes
    List<Long> mailIdsToInsert = new ArrayList<Long>();
    for (Map.Entry<Long, Long> entry : mailIdMap.entrySet()) {
        String idValue = "";  // <demo />
        if (entry.getValue() == null) {
            mailIdsToInsert.add(entry.getKey());
            // <demo>
            idValue = "null";  
        } else {
            idValue = entry.getValue().toString();
            // </demo>
        }
        // <demo>
        System.out.println(String.format(
                "    %d - %s", 
                entry.getKey(),
                idValue));
        // </demo>
    }

    // batch insert `mail_id` values that don't already exist
    try (PreparedStatement ps = dbConn.prepareStatement(
            "INSERT INTO mail_contacts (mail_id) VALUES (?)", 
            PreparedStatement.RETURN_GENERATED_KEYS)) {
        for (Long mid : mailIdsToInsert) {
            ps.setLong(1, mid);
            ps.addBatch();
        }
        ps.executeBatch();
        // get generated keys and insert them into the Map
        try (ResultSet rs = ps.getGeneratedKeys()) {
            n = 0;
            while (rs.next()) {
                mailIdMap.put(mailIdsToInsert.get(n++), rs.getLong(1));
            }
        }
    }

    // <demo>
    System.out.println();
    System.out.println("After INSERT INTO mail_contacts, mailIdMap now contains:");
    for (Map.Entry<Long, Long> entry : mailIdMap.entrySet()) {
        System.out.println(String.format(
                "    %d - %s", 
                entry.getKey(),
                entry.getValue()));
    }
    // </demo>

    // now insert the `inv_table` rows
    try (PreparedStatement ps = dbConn.prepareStatement(
            "INSERT INTO inv_table (mid, aid) VALUES (?,?)")) {
        ps.setLong(2, aid);
        for (MailidInvitation a : invitationList) {
            ps.setLong(1, mailIdMap.get(a.getId()));
            ps.addBatch();
        }
        ps.executeBatch();
    }
    dbConn.commit();
}

最终的控制台输出如下所示:

SELECT id, mail_id FROM mail_contacts WHERE mail_id IN (11,12,13)

mailIdMap now contains:
    11 - null
    12 - 1
    13 - null

After INSERT INTO mail_contacts, mailIdMap now contains:
    11 - 15
    12 - 1
    13 - 16

如果批处理中的一个或多个语句失败，一些 JDBC 驱动程序允许批处理继续执行。例如，在 MySQL Connector/J 中，选项是 continueBatchOnError这是true默认情况下。在这些情况下，另一种方法是尝试插入所有 mail_id值并检查批处理返回的更新计数。成功的 INSERT 将返回 UpdateCount 1，而失败的 INSERT 将返回现有的 mail_id。会返回 EXECUTE_FAILED (-3)。然后我们可以检索(新)id通过 .getGeneratedKeys() 成功插入的值，然后继续构建一个 SELECT 语句以返回并检索 id mail_id 的值已经存在的条目。

所以代码是这样的

// create a Map to hold `mail_id` and their corresponding `id` values 
Map<Long, Long> mailIdMap = new TreeMap<Long, Long>();
for (MailidInvitation a : invitationList) {
    // mail_id, id (id is null for now)
    mailIdMap.put(a.getId(), null);
}

// try INSERTing all `mail_id` values
try (PreparedStatement ps = dbConn.prepareStatement(
        "INSERT INTO mail_contacts (mail_id) VALUES (?)", 
        PreparedStatement.RETURN_GENERATED_KEYS)) {
    for (Long mid : mailIdMap.keySet()) {
        ps.setLong(1, mid);
        ps.addBatch();
    }
    int[] updateCounts = null;
    try {
        updateCounts = ps.executeBatch();
    } catch (BatchUpdateException bue) {
        updateCounts = bue.getUpdateCounts();
    }
    // get generated keys and insert them into the Map
    try (ResultSet rs = ps.getGeneratedKeys()) {
        int i = 0;
        for (Long mid : mailIdMap.keySet()) {
            if (updateCounts[i++] == 1) {
                rs.next();
                mailIdMap.put(mid, rs.getLong(1));
            }
        }
    }
}

// <demo>
System.out.println("mailIdMap now contains:");
// </demo>

// build a SELECT statement to get the `id` values for `mail_id`s that already existed
//     ... and print the existing mailIdMap values for demo purposes
StringBuilder sb = new StringBuilder(
        "SELECT id, mail_id " +
        "FROM mail_contacts " +
        "WHERE mail_id IN (");
int n = 0;
for (Map.Entry<Long, Long> entry : mailIdMap.entrySet()) {
    String idValue = "";  // <demo />
    if (entry.getValue() == null) {
        if (n++ > 0) sb.append(',');
        sb.append(entry.getKey());
        // <demo>
        idValue = "null";  
    } else {
        idValue = entry.getValue().toString();
        // </demo>
    }
    // <demo>
    System.out.println(String.format(
            "    %d - %s", 
            entry.getKey(),
            idValue));
    // </demo>
}
sb.append(')');
String sql = sb.toString();

// <demo>
System.out.println();
System.out.println(sql);
// </demo>

会产生这样的控制台输出:

mailIdMap now contains:
    11 - 17
    12 - null
    13 - 19

SELECT id, mail_id FROM mail_contacts WHERE mail_id IN (12)

其余过程与之前相同:

• 填写剩余mailIdMap条目，和
• 使用 id 在另一个表上处理 INSERT mailIdMap 中的值.