前言
本文介绍 MySQL45 讲中提到的一个用户关注的案例,并记录下可行的处理方案。
业务背景
业务上有这样的需求,A、B两个用户,如果互相关注,则成为好友。存在两个表,
关系(relation)表:
CREATE TABLE `relation` (
`id` int NOT NULL AUTO_INCREMENT,
`user_id` int NOT NULL,
`liker_id` int NOT NULL,
PRIMARY KEY (`id`),
UNIQUE KEY `uk_user_id_liker_id` (`user_id`,`liker_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb3;
好友(friend)表:
CREATE TABLE `friend` (
`id` int NOT NULL AUTO_INCREMENT,
`friend1_id` int NOT NULL,
`friend2_id` int NOT NULL,
PRIMARY KEY (`id`),
KEY `uk_friend` (`friend1_id`,`friend2_id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb3;
以A关注B为例:
第一步,先查询对方有没有关注自己(B有没有关注A)
select * from relation where user_id = B and liker_id = A;
如果有,则成为好友
insert into friend;
没有,则只是单向关注关系
insert into like;
现在存在的问题是,如果 A 和 B 同时关注对方,在第一步判断对方没有关注自己,则只记录单向关注关系,然而实际上应该记录成好友的。
那通过对记录上锁可以解决么?不行的,因为记录不存在,行锁无法生效。
解决方案
将两条记录合并到一条,这样就可以充分利用唯一索引。此外,再新增一个字段来记录状态。
关系表新增一个 relation_ship 字段,tinyint 类型。当 A 关注 B 且 A < B 时,
INSERT INTO (user_id, liker_id, relation_ship) values(A, B, 1);
当 A 关注 B 且 B < A 时,
INSERT INTO (user_id, liker_id, relation_ship) values(B, A, 2);
当 relation_ship = 3,表示 A 和 B 互相关注。
执行逻辑如下
begin;
# 1. 插入记录或者更新 relation_ship
INSERT INTO relation(user_id, liker_id, relation_ship) values(A, B, 1) on duplicate key update relation_ship = (relation_ship | 1);
# 2. 查询 relation_ship
SELECT relation_ship FROM relation WHERE user_id = A AND liker_id = B;
# 3. 在业务逻辑中,如果 relation_ship = 3,则创建好友关系
INSERT INTO friend(friend1_id, friend2_id) values(A, B);
commit;
当两个用户同时关注对方时,对先执行到达的事务会成功插入关系记录,得到的 relation_ship 值为 1 或 2,不创建好友关系;而后到来的事务,因为重复不插入关系记录,而更新 relation_ship = 1 | 2 或者 2 | 1,总之结果都是 3,于是它会创建好友关系。
这样就解决了业务上存在的问题,主要借助了唯一索引来实现。
参考
- [1] 答疑文章(一):日志和索引相关问题