mysql - 如何找出哪个表使用特定的外键值？

Question

在我的 MariaDB 数据库中，我有三个表，Users、Workers 和 Admins，其中 Users表将 ID 列作为主键。 Workers 和 Admins 使用 Users.ID 作为外键(和主键)。用户可以是工作人员或管理员，但不能同时是两者。我如何使用 Users.ID 找出他们是工作人员还是管理员？

我试过这个功能:

CREATE FUNCTION workeroradmin(id INT)
    RETURNS VARCHAR(20)
BEGIN
    DECLARE type VARCHAR(20)
    IF (SELECT COUNT(*) FROM Workers WHERE ID = id) = 1 
    THEN
        SET type='Worker'
    ELSEIF (SELECT COUNT(*) FROM Admins WHERE ID = id) = 1
    THEN
        SET type='Admin'
    RETURN type;
END;

但这只会给我带来 SQL 语法错误。函数是到达这里的方式还是一个简单的 IF 子句就足够了？

Answer 1

您确定一个用户不能同时是 Worker 和 Admin 吗？

我会做这样的事情:

SELECT
      Users.Id
    , IF(Workers.Id IS NOT NULL, 1, 0) AS IsWorker
    , IF(Admins.Id IS NOT NULL, 1, 0) AS IsAdmin
FROM
    Users
    LEFT JOIN Workers ON (Workers.Id = Users.Id)
    LEFT JOIN Admins ON (Admins.Id = Users.Id)
WHERE
    ...
;

您可以使用子查询而不是 LEFT JOIN 来做同样的事情。

如果你真的想用你的方式，你可以这样做:

DELIMITER $$

CREATE FUNCTION `workeroradmin`(`_id` INT)
RETURNS varchar(20)
LANGUAGE SQL
BEGIN
    IF((SELECT `Id` FROM `Workers` WHERE `Id` = _id) IS NOT NULL) THEN
        SET @result = 'Worker';
    ELSEIF((SELECT `Id` FROM `Admins` WHERE `Id` = _id) IS NOT NULL) THEN
        SET @result = 'Admin';
    END IF;

    RETURN @result;
END$$

DELIMITER ;

我不会只使用 COUNT 来检查是否存在，因为这是次优的，我也会这样做。