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