postgresql – 在数据库中强制执行“至少一个”或“恰好一个”的约束

前端之家收集整理的这篇文章主要介绍了postgresql – 在数据库中强制执行“至少一个”或“恰好一个”的约束前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
假设我们有用户,每个用户可以有多个电子邮件地址
  1. CREATE TABLE emails (
  2. user_id integer,email_address text,is_active boolean
  3. )

一些示例行

  1. user_id | email_address | is_active
  2. 1 | foo@bar.com | t
  3. 1 | baz@bar.com | f
  4. 1 | bar@foo.com | f
  5. 2 | ccc@ddd.com | t

我想强制执行一个约束,每个用户只有一个活动地址.我怎样才能在Postgres中做到这一点?我能做到这一点:

  1. CREATE UNIQUE INDEX "user_email" ON emails(user_id) WHERE is_active=true;

这可以防止具有多个活动地址的用户,但我相信,它不会防止所有地址设置为false.

如果可能的话,我宁愿避免触发器或pl / pgsql脚本,因为我们目前没有任何这样的&设置起来会很困难.但我很高兴知道“只有这样才能使用触发器或pl / pgsql”,如果是这样的话.

你根本不需要触发器或PL / pgsql.
您甚至不需要DEFERRABLE约束.
而且您不需要冗余地存储任何信息.

在users表中包含活动电子邮件的ID,从而产生相互引用.有人可能会认为我们需要一个DEFERRABLE约束来解决插入用户及其活动电子邮件的鸡与蛋问题,但使用数据修改CTE我们甚至不需要它.

这会始终为每个用户强制执行一封有效电子邮件

  1. CREATE TABLE users (
  2. user_id serial PRIMARY KEY,username text NOT NULL,email_id int NOT NULL -- FK to active email,constraint added below
  3. );
  4.  
  5. CREATE TABLE email (
  6. email_id serial PRIMARY KEY,user_id int NOT NULL REFERENCES users ON DELETE CASCADE ON UPDATE CASCADE,email text NOT NULL,CONSTRAINT email_fk_uni UNIQUE(user_id,email_id) -- for FK constraint below
  7. );
  8.  
  9. ALTER TABLE users ADD CONSTRAINT active_email_fkey
  10. FOREIGN KEY (user_id,email_id) REFERENCES email(user_id,email_id);

从users.email_id中删除NOT NULL约束,使其“最多只能有一个活动的电子邮件”. (您仍然可以为每个用户存储多封电子邮件,但这些电子邮件都不是“活动”.)

您可以使active_email_fkey DEFERRABLE允许更多余地(在同一事务的单独命令中插入用户和电子邮件),但这不是必需的.

我将user_id放在UNIQUE约束email_fk_uni中以优化索引覆盖率.细节:

> Is a composite index also good for queries on the first field?

可选视图:

  1. CREATE VIEW user_with_active_email AS
  2. SELECT * FROM users JOIN email USING (user_id,email_id);

以下是使用活动电子邮件插入新用户的方式(根据需要):

  1. WITH new_data(username,email) AS (
  2. VALUES
  3. ('usr1','abc@d.com') -- new users with *1* active email,('usr2','def3@d.com'),('usr3','ghi1@d.com')
  4. ),u AS (
  5. INSERT INTO users(username,email_id)
  6. SELECT n.username,nextval('email_email_id_seq'::regclass)
  7. FROM new_data n
  8. RETURNING *
  9. )
  10. INSERT INTO email(email_id,user_id,email)
  11. SELECT u.email_id,u.user_id,n.email
  12. FROM u
  13. JOIN new_data n USING (username);

具体的困难是我们既没有user_id也没有email_id.两者都是从各自的SEQUENCE提供的序列号.单个RETURNING子句(另一个鸡和蛋问题)无法解决它.解决方案是nextval(),如explained in detail in the linked answer below.

如果您不知道串行列email.email_id的附加序列的名称,则可以替换:

  1. nextval('email_email_id_seq'::regclass)

  1. nextval(pg_get_serial_sequence('email','email_id'))

以下是添加新“有效”电子邮件的方式:

  1. WITH e AS (
  2. INSERT INTO email (user_id,email)
  3. VALUES (3,'new_active@d.com')
  4. RETURNING *
  5. )
  6. UPDATE users u
  7. SET email_id = e.email_id
  8. FROM e
  9. WHERE u.user_id = e.user_id;

SQL Fiddle.

如果一些头脑简单的ORM不够聪明,无法应对,那么您可以将sql命令封装在服务器端函数中.

密切相关,有充分的解释:

> How to deal with mutually recursive inserts
> Complex foreign key constraint in SQLAlchemy

还有关系:

> Enforcing constraints “two tables away”

关于DEFERRABLE约束:

> Constraint defined DEFERRABLE INITIALLY IMMEDIATE is still DEFERRED?

关于nextval()和pg_get_serial_sequence():

> PostgreSQL SELECT primary key as “serial” or “bigserial”
> Duplicate row with Primary Key in PostgreSQL

猜你在找的Postgre SQL相关文章