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

news2024/11/28 6:11:23

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

Spring Boot 的 JdbcTemplate 实现 INSERT ... ON DUPLICATE KEY UPDATE

如果你使用的是 Spring Boot 的 JdbcTemplate，可以这样实现：

代码示例

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

注意事项

主键或唯一索引
- 确保 user_id 和 data_time 组成的联合主键或唯一索引已正确设置，否则 ON DUPLICATE KEY UPDATE 无法生效。
```
ALTER TABLE table1 ADD UNIQUE INDEX idx_unique_user_data (user_id, data_time);
```
字段更新逻辑
- 如果部分字段不需要更新，可以省略对应字段的 ON DUPLICATE KEY UPDATE 语句。
性能优化
- 对于大批量插入操作，可以考虑一次性提交多个记录，以减少网络和数据库连接开销。