Spring:如何将KeyHolder与PostgreSQL一起使用

时间:2013-07-21 10:22:05

标签: java spring postgresql spring-jdbc

最近迁移到POSTGRESQL,我试图在创建db表的新条目时获取唯一生成的密钥。表screenstable看起来像这样:

CREATE TABLE screenstable
(
  id serial NOT NULL,
  screenshot bytea,
  CONSTRAINT screen_id PRIMARY KEY (id )
)

将数据插入screenstable的方法如下:

@Autowired NamedParameterJDBCTemplate template;
public int insertImage(ImageBean imageBean){
        String query = "insert into screenstable (screenshot) values (:image)";
        SqlParameterSource data = new BeanPropertySqlParameterSource(imageBean);
        KeyHolder keyHolder = new GeneratedKeyHolder();
        template.update(query, data, keyHolder);
        return keyHolder.getKey().intValue();
    }

ImageBean

import java.util.Arrays;

public class ImageBean {
    private int id;
    private byte[] image;
    @Override
    public String toString() {
        return "ImageBean [id=" + id + ", image=" + Arrays.toString(image)
                + "]";
    }
    public int getId() {
        return id;
    }
    public void setId(int id) {
        this.id = id;
    }
    public byte[] getImage() {
        return image;
    }
    public void setImage(byte[] image) {
        this.image = image;
    }
}

但是运行代码会产生以下异常

15:33:20,953 ERROR JsonParseExceptionMapper:15 - org.springframework.dao.InvalidDataAccessApiUsageException: The getKey method should only be used when a single key is returned.  The current key entry contains multiple keys: [{id=3, screenshot=[B@db59df}]
org.springframework.dao.InvalidDataAccessApiUsageException: The getKey method should only be used when a single key is returned.  The current key entry contains multiple keys: [{id=3, screenshot=[B@db59df}]
        at org.springframework.jdbc.support.GeneratedKeyHolder.getKey(GeneratedKeyHolder.java:65)
        at some.project.model.FeedbackDao.insertImage(FeedbackDao.java:20)
        at some.project.rest.FeedsRest.pluginCheck(FeedsRest.java:62)
        at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
        at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:39)
        at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:25)
        at java.lang.reflect.Method.invoke(Method.java:597).....

MySQL的情况下,用于运行正常的相同代码,但在与POSTGRES一起使用时,键失败。数据类型serial是否以某种方式导致代码失败,或者我是否正确使用主键功能?

请建议。

3 个答案:

答案 0 :(得分:13)

如果没有通知框架哪个列是键,它将返回表的所有列作为键。

您可以通过将新参数传递给update方法来通知它,如下所示:

template.update(query, data, keyHolder, new String[] { "id" });

请参阅NamedParameterJdbcTemplate.update(sql, paramSource, generatedKeyHolder, keyColumnNames)

答案 1 :(得分:1)

您也可以继续使用JdbcTemplate,但在这种情况下您必须检查它:

    jdbcTemplate.update(this, holder);
    Long newId;
    if (holder.getKeys().size() > 1) {
        newId = (Long)holder.getKeys().get("your_id_column");
    } else {
        newId= holder.getKey().longValue();
    }

GeneratedKeyHolder 类抛出异常,因为类只知道当有一个键时应该返回什么(并且该键需要是Number实例,因为getKey方法返回Number对象)。

在jdbc驱动程序返回多个键时,您必须确定生成的键列(在更新之前或之后)。请查看GeneratedKeyHolder#getKey的源代码 - 分析非常简单。检查密钥列表的大小和密钥以上的密钥Spring不知道应该返回哪个密钥以及返回异常的原因。

注意:请记住,这种方法不能与Oracle合作,因为Oracle会返回类似ROWID的内容。使用Oracle,您必须使用NamedParameterJdbcTemplate。

答案 2 :(得分:0)

快速查看一下...... keyHolder.getKey()。intValue()应该是keyHolder.getKey()[“id”]。intValue()(或类似的东西)? getKey()函数返回一个具有多个键(id和image)的对象,你试图用intValue()函数将对象转换为int,我猜这个函数不起作用,除非你指向返回的对象中的“id”键“信息getKey()” 干杯, 弗朗西斯