0

我一直在使用 JPA 将实体插入数据库，但遇到了一个问题，我需要进行插入并获取最后插入的记录的主键。

使用 PostgreSQL，我将使用一个返回记录 ID 的 INSERT RETURNING 语句，但是通过实体管理器执行所有这些操作，我知道的唯一方法是使用 SELECT CURRVAL。

所以问题就变成了，我有几个数据源通过 OpenMQ 将数据发送到消息驱动的 bean（通常每个源一次发送 10-100 条消息），在这个 MDB 中，我通过实体管理器将其保存到 PostgreSQL。正是在这一点上，我认为会有这么多插入会产生“类似竞争条件”的效果，以至于我不一定会使用 SELECT CURRVAL 获得最后一条记录 id。

我的 MDB 通过如下所示的实体管理器保留 3 个实体 bean。

任何有关如何更好地做到这一点的帮助都非常感谢。

public void onMessage(Message msg) {
    Integer agPK = 0;
    Integer scanPK = 0;
    Integer lookPK = 0;

    Iterator iter = null;
    List<Ag> agKeys = null;
    List<Scan> scanKeys = null;

    try {
        iag = (IAgBean) (new InitialContext()).lookup(
                "java:comp/env/ejb/AgBean");

        TextMessage tmsg = (TextMessage) msg;


        // insert this into table only if doesn't exists
        Ag ag = new Ag(msg.getStringProperty("name"));

        agKeys = (List) (iag.getPKs(ag));

        iter = agKeys.iterator();

        if (iter.hasNext()) {
            agPK = ((Ag) iter.next()).getId();
        }
        else {
            // no PK found so not in dbase, insert new 
            iag.addAg(ag);
            agKeys = (List) (iag.getPKs(ag));
            iter = agKeys.iterator();

            if (iter.hasNext()) {
                agPK = ((Ag) iter.next()).getId();
            }
        }


        // insert this into table always
        iscan = (IScanBean) (new InitialContext()).lookup(
                "java:comp/env/ejb/ScanBean");

        Scan scan = new Scan();
        scan.setName(msg.getStringProperty("name"));
        scan.setCode(msg.getIntProperty("code"));

        iscan.addScan(scan);

        scanKeys = (List) iscan.getPKs(scan);

        iter = scanKeys.iterator();

        if (iter.hasNext()) {
            scanPK = ((Scan) iter.next()).getId();
        }


        // insert into this table the two primary keys above

        ilook = (ILookBean) (new InitialContext()).lookup(
                "java:comp/env/ejb/LookBean");

        Look look = new Look();

        if (agPK.intValue() != 0 && scanPK.intValue() != 0) {
            look.setAgId(agPK);
            look.setScanId(scanPK);

            ilook.addLook(look);
        }
// ...
4

1 回答 1

2

JPA 规范要求在持久化之后，如果正在使用 ID 生成策略，则使用有效 ID 填充实体。你不必做任何事情。

于 2012-10-04T03:05:31.957 回答