使用Ecto，验证具有2个不同相关模型的变更集具有相同的父模型

Question

在我的应用中，我有一种创建新的response的方法。 response与belongs_to和player都具有match关系。

此外player和match都与belongs_to具有team关系。

看起来像这样：

[插入新的response时，我想验证变更集中具有player和match外键的player_id和match_id属于同一team。

目前，我正在实现以下目标。首先，定义一个定制验证来检查属于外键的记录：

def validate_match_player(changeset) do
  player_team =
    Player
    |> Repo.get(get_field(changeset, :player_id))
    |> Map.get(:team_id)

  match_team =
    Match
    |> Repo.get(get_field(changeset, :match_id))
    |> Map.get(:team_id)

  cond do
    match_team == player_team -> changeset
    true -> changeset |> add_error(:player, "does not belong to the same team as the match")
  end
end

并将验证用作变更集的一部分：

def changeset(model, params \\ %{}) do
  model
  |> cast(params, [:player_id, :match_id, :message])
  |> validate_required([:player_id, :match_id, :message])
  |> foreign_key_constraint(:match_id)
  |> foreign_key_constraint(:player_id)
  |> validate_match_player()
  |> unique_constraint(
    :player,
    name: :responses_player_id_match_id_unique,
    message: "already has an response for this match"
  )
end

这很好用，但是涉及几个额外的SQL查询来查找相关记录，以便获得它们的team_id外键进行比较。

是否有更好的方法来执行此操作，也许使用约束来避免额外的查询？

Answer 1

我有两个可能的改进：

• 应用程序级解决方案：您只需查询一次，而不是两个查询。
• 数据库级解决方案：您为数据库中的检查创建触发器。

应用程序级解决方案

现在，您有两个查询来检查球员和比赛是否属于同一支球队。这意味着需要两次往返数据库。如果您只使用一个查询，例如，可以将其减少一半。给出以下查询：

    SELECT COUNT(*)
      FROM players AS p
INNER JOIN matches AS m
        ON p.team_id = m.team_id
     WHERE p.id = NEW.player_id AND m.id = NEW.match_id

您将按以下方式更改功能：

def validate_match_player(changeset) do
  player_id = get_field(changeset, :player_id)
  match_id = get_field(changeset, :match_id)

  [result] =
    Player
    |> join(:inner, [p], m in Match, on: p.team_id == m.team_id)
    |> where([p, m], p.id == ^player_id and m.id == ^match_id)
    |> select([p, m], %{count: count(p.id)})
    |> Repo.all()

  case result do
    %{count: 0} ->
      add_error(changeset, :player, "does not belong to the same team as the match")
    _ ->
      changeset
  end
end

数据库级解决方案

我假设您使用的是PostgreSQL，所以我的答案将与您在PostgreSQL手册中可以找到的相对应。

没有（干净的）方式在执行此操作的表中定义约束。约束只能访问定义它们的表。某些约束只能通过定义的内容访问列，仅此而已（CHECK CONSTRAINT）。

最好的方法是编写一个用于验证两个字段的触发器，例如：

CREATE OR REPLACE FUNCTION trigger_validate_match_player()
  RETURNS TRIGGER AS $$
    IF (
         SELECT COUNT(*)
         FROM players AS p
         INNER JOIN matches AS m
         ON p.team_id = m.team_id
         WHERE p.id = NEW.player_id AND m.id = NEW.match_id
       ) = 0
    THEN
      RAISE 'does not belong to the same team as the match'
        USING ERRCODE 'invalid_match_player';
    END IF;

    RETURN NEW;
  $$ LANGUAGE plpgsql;

CREATE TRIGGER responses_validate_match_player
  BEFORE INSERT OR UPDATE ON responses
  FOR EACH ROW
  EXECUTE PROCEDURE trigger_validate_match_player();

上一个触发器失败时将引发异常。这也意味着Ecto将引发异常。您可以查看如何处理此异常here。

最后，维护触发器并不容易，除非您使用sqitch之类的东西进行数据库迁移。

PS：如果您很好奇，在CHECK约束中执行此操作的非常肮脏方法是通过定义一个基本上绕过该限制的PostgreSQL函数。我不推荐。

我希望这会有所帮助：）