每组值的自定义序列/自动增量

Question

我正在尝试制作一个类似的博客系统，但遇到了一个小问题。

简单地说，我的article 表中有 3 列：

id SERIAL,
category VARCHAR FK,
category_id INT

id 列显然是 PK，它被用作所有文章的全局标识符。

category 栏目很好.. 分类。

category_id 用作类别中的UNIQUE ID，因此目前存在UNIQUE(category, category_id) 约束。

不过，我也想让category_id自动递增。

我想要这样每次我执行类似的查询

INSERT INTO article(category) VALUES ('stackoverflow');

我希望category_id 列根据'stackoverflow' 类别的最新category_id 自动填充。

在我的逻辑代码中实现这一点非常容易。我只是选择最新的 num 并插入其中的 +1，但这涉及两个单独的查询。 我正在寻找一种可以在一个查询中完成所有这些操作的 SQL 解决方案。

Answer 1

这个问题已经被问过很多次了，一般的想法是在多用户环境中肯定会失败 - 博客系统听起来就是这样的情况。

所以最好的答案是：不要。考虑不同的方法。

从您的表中完全删除列 category_id - 它不存储其他两列 (id, category) 不会存储的任何信息。

您的 id 是一个 serial 列，并且已经以可靠的方式自动递增。

• Auto increment SQL function

如果您需要某种category_id，每个category 没有间隙，请使用row_number() 即时生成它：

• Serial numbers per group of rows for compound key

Answer 2

Postgresql 使用序列来实现这一点；这与您在 MySQL 中使用的方法不同。请查看http://www.postgresql.org/docs/current/static/sql-createsequence.html 以获取完整参考。

基本上，您通过以下方式创建序列（数据库对象）：

CREATE SEQUENCE serials;

然后当你想添加到你的表中时，你将拥有：

INSERT INTO mytable (name, id) VALUES ('The Name', NEXTVAL('serials')

Answer 3

概念

至少有几种方法可以解决这个问题。我想到的第一个：

通过覆盖INSERT 语句中的输入值，为每行执行的触发器内的category_id 列分配一个值。

动作

这里是 SQL Fiddle 查看实际代码

---

对于一个简单的测试，我正在创建article 表保存类别及其id，每个category 应该是唯一的。我省略了约束创建 - 这与提出要点无关。

create table article ( id serial, category varchar, category_id int )

使用generate_series() 函数为两个不同的类别插入一些值，以实现自动增量。

insert into article(category, category_id)
  select 'stackoverflow', i from generate_series(1,1) i
  union all
  select 'stackexchange', i from generate_series(1,3) i

创建一个触发函数，它将选择MAX(category_id) 并将其值增加1 以获得category，我们插入一行，然后覆盖该值，然后继续使用实际的INSERT 到表（BEFORE INSERT 触发器负责处理）。

CREATE OR REPLACE FUNCTION category_increment()
RETURNS trigger
LANGUAGE plpgsql
AS
$$
DECLARE
  v_category_inc int := 0;
BEGIN
  SELECT MAX(category_id) + 1 INTO v_category_inc FROM article WHERE category = NEW.category;
  IF v_category_inc is null THEN
    NEW.category_id := 1;
  ELSE
    NEW.category_id := v_category_inc;
  END IF;
RETURN NEW;
END;
$$ 

使用函数作为触发器。

CREATE TRIGGER trg_category_increment 
  BEFORE INSERT ON article 
  FOR EACH ROW EXECUTE PROCEDURE category_increment()

为已存在的类别和不存在的类别插入更多值（触发后设备）。

INSERT INTO article(category) VALUES 
  ('stackoverflow'),
  ('stackexchange'),
  ('nonexisting');

查询用于选择数据：

select category, category_id From article order by 1,2

初始插入结果：

category    category_id
stackexchange   1
stackexchange   2
stackexchange   3
stackoverflow   1

最终插入后的结果：

category    category_id
nonexisting     1
stackexchange   1
stackexchange   2
stackexchange   3
stackexchange   4
stackoverflow   1
stackoverflow   2