以DB表示的状态机，执行状态转换

Question

我有一个有限状态机，它表示一个作业的各个阶段。我需要在Postgres数据库中表示状态。我希望通过禁止从一种状态更新到另一种状态来强制代码正确性，除非状态机允许这样做。

实现我的目标的一种简单的方法是获取表上的排他锁，在事务中检查当前状态和下一个状态，在无效更新的情况下错误地中止。

这显然是一个性能杀手，因为我要在每次状态转换时锁定Job表。

是否有一种方法可以通过约束来实现相同的目标?

Answer 1

触发是你问题的答案。

让我们考虑一个简单的表:

CREATE TABLE world (id serial PRIMARY KEY, state VARCHAR);
insert into world (state) values ('big bang');
insert into world (state) values ('stars formation');
insert into world (state) values ('human era');

将被触发器调用的函数。在这里定义状态机逻辑。RAISE EXCEPTION很有用，因为你可以在这里提供自定义消息。

CREATE FUNCTION check_world_change() RETURNS trigger as $check_world_change$
BEGIN
  IF OLD.state = 'big bang' AND NEW.state = 'human era' THEN
    RAISE EXCEPTION 'Dont skip stars';
  END IF;
  IF OLD.state = 'stars formation' AND NEW.state = 'big bang' THEN
    RAISE EXCEPTION 'Impossible to reverse order of things';
  END IF;
  RETURN NEW;
END;
$check_world_change$ LANGUAGE plpgsql;

为你的表定义触发器:

CREATE TRIGGER check_world_change BEFORE UPDATE ON world 
  FOR EACH ROW EXECUTE PROCEDURE check_world_change();

现在，当您尝试更新其中一行的状态时，您将得到错误:

world=# select * from world;
 id |      state
----+-----------------
  2 | stars formation
  1 | human era
  3 | big bang
(3 rows)
world=# update world set state='human era' where state='big bang';
ERROR:  Wrong transition
world=# select * from world;
 id |      state
----+-----------------
  2 | stars formation
  1 | human era
  3 | big bang
(3 rows)

引用:

https://www.postgresql.org/docs/9.5/static/plpgsql-trigger.htmlhttps://www.postgresql.org/docs/9.5/static/sql-createtrigger.html