如何检查值是否存在于其中一列中以及 MySQL 中的哪一列中？

Question

我正在尝试在 MySQL 中为我有 4 列的“友谊”表创建一个函数：

friendship_id、status、user1_id 和 user2_id（最后两个也是指向另一个表 'users' 的外键）。

函数应该接收一个参数，即用户ID，并返回该用户有多少朋友。

我创建了一个这样的函数：

CREATE FUNCTION num_of_friends(id int) 
RETURNS int(11)
BEGIN

SET @friends_count = (SELECT COUNT(user2_id) FROM friendships WHERE user1_id = id);
RETURN @friends_count;
END

因此，如果该特定用户的 id 在 user1_id 列中，则此函数会返回该特定用户的朋友数。

但是有些用户的 id 只在 user2_id 列中。

如何检查我们作为参数传递的用户 id 是否存在于任一列中并据此计算友谊？

Answer 1

您可以将 id 值与OR 的两列进行比较

CREATE FUNCTION num_of_friends(id int) 
RETURNS int(11)
BEGIN

SET @friends_count = (SELECT COUNT(*) from (select * from friendship  where user1_id=id
                      union
                      select user2_id,user1_id from friendship where user2_id=id)data);
RETURN @friends_count;
END

检查这个 SQL 小提琴：http://sqlfiddle.com/#!9/c5928ca/8