postgresql - 约束，基于与另一个表的连接

Question

我有表 tariffs，有两列:(tariff_id, reception)

我有表 users，有两列:(user_id, reception)

我的表 users_tariffs 有两列:(user_id, tariff_id)。

我想防止将一个接收的资费分配给另一个接收的用户的情况。我该怎么做？

E.G

用户:

user_id | reception
Putin   | Russia
Trump   | USA

关税:

tariff_id | reception
cheap     | USA
expensive | Russia

users_tariffs 的错误情况，因为 Cheap tariff 仅适用于美国:

user_id | tariff_id
Putin   | Cheap

Answer 1

解决方案 1:外键约束

我假设下表定义。特别是，user_tariffs 中的组合键使 users 和 tariffs 之间成为多对多关系。

CREATE TABLE tariffs (tariff_id int NOT NULL PRIMARY KEY,  
                      reception text NOT NULL);  
CREATE TABLE users (user_id int NOT NULL PRIMARY KEY,  
                    reception text NOT NULL);  
CREATE TABLE user_tariffs (tariff_id int NOT NULL REFERENCES tariffs (tariff_id),  
                           user_id int NOT NULL REFERENCES users (user_id),  
                           PRIMARY KEY (tariff_id, user_id));  

您可能在某处需要所有三列的组合，所以让我们创建它:

ALTER TABLE user_tariffs ADD COLUMN reception text;  
UPDATE user_tariffs a  
SET reception = b.reception  
FROM (SELECT * FROM tariffs) b  
WHERE a.tariff_id = b.tariff_id;  
ALTER TABLE user_tariffs ALTER COLUMN reception SET NOT NULL;  

现在我们可以使用 FOREIGN KEY REFERENCES (user_id, reception) 进入 users。

CREATE UNIQUE INDEX ON tariffs (tariff_id, reception);  
ALTER TABLE user_tariffs ADD FOREIGN KEY (tariff_id, reception)  
                      REFERENCES tariffs (tariff_id, reception);  

此外，我们可以使用 FK REFs (tariff_id, reception) 到 tariffs 中。

CREATE UNIQUE INDEX ON users (user_id, reception);  
ALTER TABLE user_tariffs ADD FOREIGN KEY (user_id, reception) 
                        REFERENCES users (user_id, reception);  

填充数据:

INSERT INTO users VALUES (1, 'cheap'), (2, 'expensive');
INSERT INTO tariffs VALUES (1, 'cheap'), (2, 'expensive');

现在假设我们有以下数据 (user_id, tariff_id) 要插入:

WITH data (user_id, tariff_id) 
       AS (VALUES (1, 2), (2, 1)),   -- here is your application data
     datas (user_id, tariff_id, reception) 
       AS (SELECT user_id, 
                  tariff_id, 
                  (SELECT u.reception  -- reception calculated by user
                   FROM users u 
                   WHERE u.user_id = d.user_id)  
           FROM data d) 
INSERT INTO user_tariffs SELECT * FROM datas ;

那你就不能插入数据了，因为你只能添加(1, 1)或者(2, 2)同一个reception，但不是 (1, 2) 或 (2, 1) 具有不同的 reception。错误信息是:

ERROR:  insert or update on table "user_tariffs" violates foreign key constraint "user_tariffs_user_id_fkey1"
DETAIL:  Key (user_id, reception)=(2, cheap) is not present in table "users".

但是您可以插入 data AS VALUES (1, 1), (2, 2)。我认为 FOREIGN KEY CONSTRAINT 解决方案是首选。

如果您想要更好的表格设计，请描述您的功能依赖项。

解决方案 2:触发器

-- DROP TABLE user_tariffs CASCADE;  
-- DROP TABLE users CASCADE;  
-- DROP TABLE tariffs CASCADE;  
CREATE TABLE tariffs (tariff_id int NOT NULL PRIMARY KEY,  
                      reception text NOT NULL);  
CREATE TABLE users (user_id int NOT NULL PRIMARY KEY,  
                    reception text NOT NULL);  
CREATE TABLE user_tariffs (tariff_id int NOT NULL REFERENCES tariffs (tariff_id),  
                           user_id int NOT NULL REFERENCES users (user_id),  
                           PRIMARY KEY (tariff_id, user_id));  
INSERT INTO users VALUES (1, 'cheap'), (2, 'expensive');  
INSERT INTO tariffs VALUES (1, 'cheap'), (2, 'expensive');  
-- table user_tariffs (user_id, tariff_id) only, without reception column.

创建一个返回类型触发器的函数:

CREATE OR REPLACE FUNCTION check_reception()  
RETURNS trigger AS $$
DECLARE valid boolean := false;
BEGIN 
SELECT (SELECT u.reception FROM users u WHERE u.user_id = NEW.user_id) 
     = (SELECT t.reception FROM tariffs t WHERE t.tariff_id = NEW.tariff_id) 
INTO valid FROM user_tariffs ;
IF valid = false  
THEN RAISE EXCEPTION '(user, tariff, reception) invalid.';  
END IF;  
RETURN NEW;
END; $$ LANGUAGE plpgsql ;

并注册:

CREATE TRIGGER reception_trigger  
AFTER INSERT OR UPDATE ON user_tariffs  
FOR EACH ROW EXECUTE PROCEDURE check_reception();

现在尝试插入 (1, 2)，这是 (cheap, expensive) 并且是不允许的:

INSERT INTO user_tariffs VALUES (1, 2);
ERROR:  (user, tariff, reception) invalid.
KONTEXT:  PL/pgSQL function check_reception() line 7 at RAISE

但是我们可以插入 (1, 1)，这是 (cheap, cheap) 没有问题的:

INSERT INTO user_tariffs VALUES (1, 1);
SELECT * FROM user_tariffs;  

备注

在我看来，触发器并不是这里的最佳解决方案。如果可能，尽量避免触发。它们可能有副作用(交易等)。检查 StackOverflow 以获取更多详细信息:)