配置 JPA 让 PostgreSQL 生成主键值

Question

所以我们的项目使用 PostgreSQL 数据库，我们使用 JPA 来操作数据库。 我们已经在 Netbeans 7.1.2 中使用自动创建者从数据库中创建了实体。

经过小的改动，我们的主键值被描述为：

@Id
@GeneratedValue(strategy=GenerationType.IDENTITY)
@Basic(optional = false)
@NotNull
@Column(name = "idwebuser", nullable = false)
private Integer idwebuser;

问题是现在应用程序不灵活，因为当我们直接修改数据库（使用 SQL 或其他工具）而不是通过 Java 应用程序时 - 生成的值低于实际的数据库 ID 值 - 所以我们在创建新实体时出错。

JPA 是否有可能让数据库自动生成 ID，然后在创建过程后获取它？ 或者有什么更好的解决方案？ 谢谢。

编辑 进一步来说： 我们有一个用户表，我的问题是使用任何类型的策略生成类型，JPA 正在插入一个由其生成器 ID 指定的新实体。这对我来说是错误的，因为如果我自己更改表，通过添加新条目，应用程序的 GeneratedValue 低于当前 ID - 这导致我们出现重复 ID 的异常。 我们可以修复它;)？

关于答案的简短说明 我有一个小谎言，因为我们使用了 PG Admin -> 查看前 100 行并从那里编辑行，而不是使用 select。无论如何，事实证明这个编辑器以某种方式跳过了更新 ID 的过程，因此即使在 DB 中，当我们编写正确的 INSERT 时，它也会以不正确的 ID 执行！所以基本上是我们使用的编辑器的问题，而不是数据库和应用的问题……

现在它甚至可以使用 @GeneratedValue(strategy=GenerationType.IDENTITY)

Answer 1

给定表定义：

CREATE TABLE webuser(
    idwebuser SERIAL PRIMARY KEY,
    ...
)

使用映射：

@Entity
@Table(name="webuser")
class Webuser {

    @Id
    @SequenceGenerator(name="webuser_idwebuser_seq",
                       sequenceName="webuser_idwebuser_seq",
                       allocationSize=1)
    @GeneratedValue(strategy = GenerationType.SEQUENCE,
                    generator="webuser_idwebuser_seq")
    @Column(name = "idwebuser", updatable=false)
    private Integer id;

    // ....

}

tablename_columname_seq的命名是SERIAL的PostgreSQL默认序列命名，我建议你坚持下去。

如果您需要 Hibernate 与数据库的其他客户端合作，allocationSize=1 很重要。

请注意，如果事务回滚，此序列中会有“间隙”。交易可以出于各种原因回滚。您的应用程序应设计为应对此问题。

• 永远不要假设任何 id n 都有一个 id n-1 或 n+1
• 永远不要假设在小于n 的id 之前或大于n 的id 之后添加或提交了id n。如果你真的很小心如何使用序列，你可以这样做，但你永远不应该尝试；而是在您的表中记录一个时间戳。
• 切勿添加或减去 ID。比较它们是否相等。

请参阅PostgreSQL documentation for sequences 和the serial data types。

他们解释说，上面的表定义基本上是一个快捷方式：

CREATE SEQUENCE idwebuser_id_seq;
CREATE TABLE webuser(
    idwebuser integer primary key default nextval('idwebuser_id_seq'),
    ...
)
ALTER SEQUENCE idwebuser_id_seq OWNED BY webuser.idwebuser;

...这应该有助于解释为什么我们添加了@SequenceGenerator 注释来描述序列。

---

如果您确实必须有一个无间隙序列（例如，支票或发票编号），请参阅gapless sequences，但请认真避免这种设计，并且永远不要将其用作主键。

---

注意：如果您的表定义如下所示：

CREATE TABLE webuser(
    idwebuser integer primary key,
    ...
)

并且您正在使用（不安全，请勿使用）插入其中：

INSERT INTO webuser(idwebuser, ...) VALUES ( 
    (SELECT max(idwebuser) FROM webuser)+1, ...
);

或（不安全，永远不要这样做）：

INSERT INTO webuser(idwebuser, ...) VALUES ( 
    (SELECT count(idwebuser) FROM webuser), ...
);

然后你做错了，应该切换到一个序列（如上所示）或一个正确使用锁定计数器表的无间隙序列实现（再次，见上文并在 Google 中查看“gapless sequence postgresql”）。如果有多个连接在数据库上工作，则上述两种方法都会出错。

Answer 2

看来你必须像这样使用序列生成器：

@GeneratedValue(generator="YOUR_SEQ",strategy=GenerationType.SEQUENCE)

Answer 3

请尝试使用GenerationType.TABLE 而不是GenerationType.IDENTITY。数据库将创建单独的表，用于生成唯一的主键，它还将存储上次使用的 id 号。

Answer 4

您还可以通过编写脚本将通用GenerationType.IDENTITY 批量转换为所选答案提出的解决方案，从而节省一些精力。下面的脚本对 Java 源文件的格式有一些轻微的依赖性，并且会在没有备份的情况下进行修改。警告购买者！

运行脚本后：

1. 搜索并将import javax.persistence.Table;替换为 import javax.persistence.Table; import javax.persistence.SequenceGenerator;。
2. 在 NetBeans 中重新格式化源代码，如下所示：
   1. 选择所有要格式化的源文件。
   2. 按 Alt+Shift+F
   3. 确认重新格式化。

将以下脚本另存为update-sequences.sh 或类似名称：

#!/bin/bash

# Change this to the directory name (package name) where the entities reside.
PACKAGE=com/domain/project/entities

# Change this to the path where the Java source files are located.
cd src/main/java

for i in $(find $PACKAGE/*.java -type f); do
  # Only process classes that have an IDENTITY sequence.
  if grep "GenerationType.IDENTITY" $i > /dev/null; then
    # Extract the table name line.
    LINE_TABLE_NAME=$(grep -m 1 @Table $i | awk '{print $4;}')
    # Trim the quotes (if present).
    TABLE_NAME=${LINE_TABLE_NAME//\"}
    # Trim the comma (if present).
    TABLE_NAME=${TABLE_NAME//,}

    # Extract the column name line.
    LINE_COLUMN_NAME=$(grep -m 1 -C1 -A3 @Id $i | tail -1)
    COLUMN_NAME=$(echo $LINE_COLUMN_NAME | awk '{print $4;}')
    COLUMN_NAME=${COLUMN_NAME//\"}
    COLUMN_NAME=${COLUMN_NAME//,}

    # PostgreSQL sequence name.
    SEQUENCE_NAME="${TABLE_NAME}_${COLUMN_NAME}_seq"

    LINE_SEQ_GENERATOR="@SequenceGenerator( name = \"$SEQUENCE_NAME\", sequenceName = \"$SEQUENCE_NAME\", allocationSize = 1 )"
    LINE_GENERATED_VAL="@GeneratedValue( strategy = GenerationType.SEQUENCE, generator = \"$SEQUENCE_NAME\" )"
    LINE_COLUMN="@Column( name = \"$COLUMN_NAME\", updatable = false )\n"

    # These will depend on source code formatting.
    DELIM_BEGIN="@GeneratedValue( strategy = GenerationType.IDENTITY )"
    # @Basic( optional = false ) is also replaced.
    DELIM_ENDED="@Column( name = \"$COLUMN_NAME\" )"

    # Replace these lines...
    #
    # $DELIM_BEGIN
    # $DELIM_ENDED
    #
    # With these lines...
    #
    # $LINE_SEQ_GENERATOR
    # $LINE_GENERATED_VAL
    # $LINE_COLUMN

    sed -i -n "/$DELIM_BEGIN/{:a;N;/$DELIM_ENDED/!ba;N;s/.*\n/$LINE_SEQ_GENERATOR\n$LINE_GENERATED_VAL\n$LINE_COLUMN/};p" $i
  else
    echo "Skipping $i ..."
  fi
done

使用 NetBeans 生成 CRUD 应用程序时，ID 属性将不包括可编辑的输入字段。

Answer 5

对我有用

1. 像这样创建表，使用 SERIAL。

CREATE TABLE webuser(
    idwebuser SERIAL PRIMARY KEY,
    ...
)

1. 在 id 字段添加@GeneratedValue(strategy = GenerationType.IDENTITY)。

@Entity
@Table(name="webuser")
class Webuser {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    // ....

}