在 Java 中实现【不存在即插入,存在即更新】 INSERT ... ON DUPLICATE KEY UPDATE
Spring Boot 的 JdbcTemplate
实现 INSERT ... ON DUPLICATE KEY UPDATE
如果你使用的是 Spring Boot 的 JdbcTemplate
,可以这样实现:
代码示例
java
@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);
}
使用示例
java
upsertUserData("user_1", LocalDateTime.of(2024, 11, 18, 15, 0), 10, 20, 30);
在批量操作中实现
如果有多条数据需要同时插入或更新,可以使用 batchUpdate
:
java
@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());
});
}
数据模型
java
@Data
@AllArgsConstructor
public class Data {
private String userId;
private LocalDateTime dataTime;
private int field1;
private int field2;
private int field3;
}
调用示例
java
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
无法生效。
sqlALTER TABLE table1 ADD UNIQUE INDEX idx_unique_user_data (user_id, data_time);
- 确保
-
字段更新逻辑
- 如果部分字段不需要更新,可以省略对应字段的
ON DUPLICATE KEY UPDATE
语句。
- 如果部分字段不需要更新,可以省略对应字段的
-
性能优化
- 对于大批量插入操作,可以考虑一次性提交多个记录,以减少网络和数据库连接开销。
这样,你就可以灵活地实现对 table1
表的插入和更新逻辑了!