为JdbcTemplate扩展获得插入数据的主键功能[转]

最新推荐文章于 2025-04-24 22:29:00 发布

iteye_20054

最新推荐文章于 2025-04-24 22:29:00 发布

阅读量109

点赞数

CC 4.0 BY-SA版权

文章标签： java

本文链接：https://blog.youkuaiyun.com/iteye_20054/article/details/82131298

本文介绍了一种基于Spring JdbcTemplate的数据访问适配器实现方式，该适配器提供了插入记录并返回主键的功能。通过扩展JdbcTemplate并重写部分方法，增强了其灵活性和实用性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >



package com.lianai88.common.access;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import javax.sql.DataSource;

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.ParameterDisposer;
import org.springframework.jdbc.core.PreparedStatementCreator;
import org.springframework.jdbc.core.PreparedStatementSetter;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;

/**
 * 数据存取适配器
 * @author David Day
 */
public class JdbcTemplateAdapter extends JdbcTemplate {

	public JdbcTemplateAdapter() {
		super();
	}

	public JdbcTemplateAdapter(DataSource ds) {
		super(ds);
	}

	/**
	 * 增加并且获取主键
	 * @param sql sql语句
	 * @param params 参数列表
	 * @return 主键
	 */
	public Object insertAndGetKey(final String sql, final Object... params) {
		logger.debug("Executing SQL update and returning generated keys");

		final KeyHolder key = new GeneratedKeyHolder();

		update(new PreparedStatementCreator() {

			public PreparedStatement createPreparedStatement(Connection con) throws SQLException {
				PreparedStatement ps = con.prepareStatement(sql, 
						PreparedStatement.RETURN_GENERATED_KEYS);
				PreparedStatementSetter pss = new ArgPreparedStatementSetter(params);
				try {
					if (pss != null) {
						pss.setValues(ps);
					}
				} finally {
					if (pss instanceof ParameterDisposer) {
						((ParameterDisposer) pss).cleanupParameters();
					}
				}
				return ps;
			}

		}, key);

		return key.getKey();
	}

}