Spring Boot的JdbcTemplate实现“不存在即插入,存在即更新”

发布于:2024-11-28 ⋅ 阅读:(32) ⋅ 点赞:(0)

在 Java 中实现【不存在即插入,存在即更新】 INSERT ... ON DUPLICATE KEY UPDATE

Spring BootJdbcTemplate 实现 INSERT ... ON DUPLICATE KEY UPDATE

如果你使用的是 Spring BootJdbcTemplate,可以这样实现:

代码示例
@Autowired
private JdbcTemplate jdbcTemplate;

public void upsertUserData(String userId, LocalDateTime dataTime, int field1, int field2, int field3) {
    String sql = "INSERT INTO table1 (user_id, data_time, field1, field2, field3) " +
                 "VALUES (?, ?, ?, ?, ?) " +
                 "ON DUPLICATE KEY UPDATE " +
                 "field1 = VALUES(field1), " +
                 "field2 = VALUES(field2), " +
                 "field3 = VALUES(field3)";
                 
    jdbcTemplate.update(sql, devId, dataTime, field1, field2, field3);
}
使用示例
upsertUserData("user_1", LocalDateTime.of(2024, 11, 18, 15, 0), 10, 20, 30);

在批量操作中实现

如果有多条数据需要同时插入或更新,可以使用 batchUpdate

@Autowired
private JdbcTemplate jdbcTemplate;

public void batchUpsert(List<Data> dataList) {
    String sql = "INSERT INTO table1 (dev_id, data_time, field1, field2, field3) " +
                 "VALUES (?, ?, ?, ?, ?) " +
                 "ON DUPLICATE KEY UPDATE " +
                 "field1 = VALUES(field1), " +
                 "field2 = VALUES(field2), " +
                 "field3 = VALUES(field3)";
    
    jdbcTemplate.batchUpdate(sql, dataList, dataList.size(), (ps, data) -> {
        ps.setString(1, data.getUserId());
        ps.setTimestamp(2, Timestamp.valueOf(data.getDataTime()));
        ps.setInt(3, data.getField1());
        ps.setInt(4, data.getField2());
        ps.setInt(5, data.getField3());
    });
}
数据模型
@Data
@AllArgsConstructor
public class Data {
    private String userId;
    private LocalDateTime dataTime;
    private int field1;
    private int field2;
    private int field3;
}
调用示例
List<Data> dataList = Arrays.asList(
    new Data("user_1", LocalDateTime.of(2024, 11, 18, 15, 0), 10, 20, 30),
    new Data("user_2", LocalDateTime.of(2024, 11, 18, 15, 15), 15, 25, 35)
);
batchUpsert(dataList);

注意事项

  1. 主键或唯一索引

    • 确保 user_iddata_time 组成的联合主键或唯一索引已正确设置,否则 ON DUPLICATE KEY UPDATE 无法生效。
    ALTER TABLE table1 ADD UNIQUE INDEX idx_unique_user_data (user_id, data_time);
    
  2. 字段更新逻辑

    • 如果部分字段不需要更新,可以省略对应字段的 ON DUPLICATE KEY UPDATE 语句。
  3. 性能优化

    • 对于大批量插入操作,可以考虑一次性提交多个记录,以减少网络和数据库连接开销。

这样,你就可以灵活地实现对 table1 表的插入和更新逻辑了!