SpringBoot PostgreSQL upsert

新丨客2023-07-21 8:22

PostgreSQL 的 upsert 功能：当记录不存在时，执行插入；否则，进行更新。但是版本需要在9.5以上

upsert SQL示例

sql 复制代码

INSERT INTO table_name(column_list) 
VALUES(value_list)
ON CONFLICT target action;

SQL字符串定义

java 复制代码

    @Autowired
    private EntityManager entityManager;
	
	//具体业务SQL
    private static final String INSERT_QUERY_BASE_OFFLINE_AUTH = 
    "insert INTO etk_lg_life (fbh, offline_auth_id, offline_auth_time) VALUES  ";
    private static final String INSERT_PARAM_FOOTER_OFFLINE_AUTH = 
    "  on conflict (fbh)  do update set offline_auth_id = EXCLUDED.offline_auth_id , offline_auth_time=EXCLUDED.offline_auth_time ";
	
    private static final int INSERT_BATCH_SIZE = 1000;
    private static final String DELIMITER = ",";

函数定义&使用

java 复制代码

 private void insertAll1(List<String> lgs, Date dataDate, int linkId) {
 		//日期转时间字符串的函数而已
        String dataDateStr = DateUtil.format4Sql(dataDate);

        String values = lgs.stream().map(lg -> this.renderSqlForLgLife(lg, dataDateStr, linkId))
                .collect(Collectors.joining(DELIMITER));

        String insertSQL = INSERT_QUERY_BASE_OFFLINE_AUTH + values +INSERT_PARAM_FOOTER_OFFLINE_AUTH ;

        entityManager.createNativeQuery(insertSQL).executeUpdate();
        entityManager.flush();
        entityManager.clear();

    }

java 复制代码

   下方函数为拼接values-SQL///
    private String renderSqlForLgLife(String lg, String dataDateStr, int linkId) {
        StringBuilder sBuilder = new StringBuilder();

        sBuilder.append("('").append(lg).append("',").append(linkId).append(",'").append(dataDateStr).append("')");

        return sBuilder.toString();
    }

java 复制代码

ListUtils.partition(lgLifeModel.getLgs(), INSERT_BATCH_SIZE).forEach(lgs -> {
                insertAll1(lgs, lgLifeModel.getDataDates()[0], lgLifeModel.getLinkId());

                log.info("插入{}life数据成功", lgs.size());
            });