postgresql - 基于外键在 PostgreSQL 中增加一个序列

Question

我想使用序列为我的数据库中的某些对象创建友好的 ID。我的问题是我不希望序列对表来说是全局的，而是我想根据外键增加值。

例如，我的表定义为：

CREATE TABLE foo (id numeric PRIMARY KEY, friendly_id SERIAL, bar_id numeric NOT NULL)

我想friendly_id分别增加每个bar_id这样的语句：

INSERT INTO foo (123, DEFAULT, 345)
INSERT INTO foo (124, DEFAULT, 345)
INSERT INTO foo (125, DEFAULT, 346)
INSERT INTO foo (126, DEFAULT, 345)

将导致（期望的行为）：

id         | friendly_id      | bar_id
-----------+------------------+-----------------
123        | 1                | 345
124        | 2                | 345
125        | 1                | 346
126        | 3                | 345

而不是（当前行为）：

id         | friendly_id      | bar_id
-----------+------------------+-----------------
123        | 1                | 345
124        | 2                | 345
125        | 3                | 346
126        | 4                | 345

这可能使用序列还是有更好的方法来实现这一点？

Answer 1

create table foo (
    id serial primary key,
    friendly_id integer not null,
    bar_id integer not null,
    unique(friendly_id, bar_id)
);

在应用程序中，如果引发重复键异常，则将插入包装在异常捕获循环中以重试

insert into foo (friendly_id, bar_id)
select
    coalesce(max(friendly_id), 0) + 1,
    346
from foo
where bar_id = 346