當我們用java,jdbc 或者 spring的jdbctemplate來操作數據時候,總有時候會碰到 存在就插入不存在就更新 這樣的需求,我們一般都是用下面的方法解決
public void save(){ JdbcTemplate jdbc = getJdbc(); String updateStatement=""; String insertStatement=""; int cout = jdbc.update(updateStatement); if(count<=0){ //不存在需要被更新的數據,那就插入 jdbc.update(insertStatement); } jdbc.close(); }
在這里僅僅只是一條數據,看着還行,但是如果是批量的數據呢?代碼就會變成下面這樣
public void save(List<Param> list){ JdbcTemplate jdbc = getJdbc(); String updateStatement=""; String insertStatement=""; for(Param p:list){ int cout = jdbc.update(updateStatement); if(count<=0){ //不存在需要被更新的數據,那就插入 jdbc.update(insertStatement); } } jdbc.close(); }
這樣感覺就不怎么好看了,而且效率也不怎么樣。
優化方法如下:
1. 可以在數據庫中建立存儲過程,邏輯為單條記錄的 存在就更新不存在就插入;
2. 在java中通過jdbc調用存儲過程,如果是批量數據的話,跟批量插入差不多。
實例代碼:
1. 存儲過程(postgres數據庫):
CREATE OR REPLACE FUNCTION insert_exists_device(p_deviceid integer, p_eventtype integer, p_state integer, p_eventvalue character varying) RETURNS integer AS $BODY$ BEGIN perform 1 FROM tablename WHERE deviceid=p_deviceid AND eventtype=p_eventtype; --判斷是否有該條記錄 IF NOT found THEN insertStatement; ELSE updateStatement; END IF; RETURN 1; END $BODY$ LANGUAGE plpgsql;
2. java中調用:
public void save(List<Param> list){ return jdbc.execute("{call insert_exists_device_monitor(?,?,?,?)}", new CallableStatementCallback<Boolean>() { @Override public Boolean doInCallableStatement(CallableStatement cs)throws SQLException, DataAccessException { for (Param p: list) { cs.setInt(1, deviceid); cs.setInt(2, p.getEventType()); cs.setInt(3, p.getState()); cs.setString(4, p.getEventValue()); cs.addBatch(); } s.executeBatch(); } }); }