【Redis】点赞功能

“千山万水总是情，点个关注行不行！”、“各位老铁点点小红心”、“欢迎来到xx直播间，主播带你装逼带你飞，喜欢主播的点亮噢。”各位程序猿们看到这样的话术是不是很熟悉。那么今天，咋们就透过本质看一下点赞功能到底是如何实现的。

在日常开发中，一般点赞功能的实现有两种方式。

第一种方式就是在数据库中建一张点赞表，用户点赞或者取消时，直接落库处理。不过，点赞业务的重要性并不是那么大，所以丢失几条数据也没什么影响，所以采用直接落库可能对服务的性能有一定的影响。

第二种方式就是使用Redis来进行存储，用户点赞或取消时，在Redis中进行处理。如果认为这个业务比较重要的话，那么可以写定时任务进行落库处理；如果并没有那么重要的话，也可以直接存储在Redis中不做其他处理，毕竟Redis也是有持久化的。

两者相比之下，肯定是Redis实现点赞功能的性能要比数据库实现点赞功能的性能高。根据标题也可以看出，今天主要是用Redis来实现点赞功能。

点赞/取消

在实现点赞功能时，选取的是Redis中集合这一数据结构。利用博客id作为key，集合中的内容就是点赞的用户id。

    /**
     * 点赞功能
     * @param id 博客id
     */
    public Boolean likeBlog(Long id) {
        String key = Constants.BLOG_LIKE_KEY + id;
        // TODO 从ThreadLocal中获取用户id
        Long userId = UserHolder.getId();
        if(userId == null) {
            return false;
        }
        // TODO 从Redis中判断该用户对该博客是否进行点赞
        Boolean isMember = this.stringRedisTemplate.opsForSet()
                .isMember(key, userId.toString());
        // TODO 判断是否点赞
        if(isMember == null || !isMember) {
            // 未点赞
            // TODO 博客总点赞数加一
            LambdaUpdateWrapper<Blog> blogLambdaUpdateWrapper = new LambdaUpdateWrapper<>();
            blogLambdaUpdateWrapper.setSql("liked = liked + 1")
                    .eq(Blog::getId, id);
            boolean add = this.update(blogLambdaUpdateWrapper);
            // TODO Redis缓存中添加该点赞
            if(add) {
                this.stringRedisTemplate.opsForSet()
                        .add(key, userId.toString());
            }
        } else {
            // 已经点赞
            // TODO 博客总点赞数减一
            LambdaUpdateWrapper<Blog> blogLambdaUpdateWrapper = new LambdaUpdateWrapper<>();
            blogLambdaUpdateWrapper.setSql("liked = liked - 1")
                    .eq(Blog::getId, id);
            boolean subtract = this.update(blogLambdaUpdateWrapper);
            // TODO Redis缓存中删除该点赞
            if(subtract) {
                this.stringRedisTemplate.opsForSet()
                        .remove(key, userId.toString());
            }
        }
        return true;
    }

查看是否点赞

    /**
     * 查询是否点赞
     * @param id 博客id
     */
    public Boolean queryLikeBlog(Long id) {
        String key = Constants.BLOG_LIKE_KEY + id;
        // TODO 从ThreadLocal中获取用户id
        Long userId = UserHolder.getId();
        if(userId == null) {
            return false;
        }
        // TODO 从Redis中判断是否点赞
        Boolean isMember = this.stringRedisTemplate.opsForSet()
                .isMember(key + id, userId.toString());
        // TODO 返回结果
        if(isMember == null) {
            isMember = false;
        }
        return isMember;
    }

点赞排行榜

对于博客来说，那些比较水的博客肯定是点赞数比较少，但是一些大V的博客点赞数可能是以万为单位来计算的，自然就不可能将所有的点赞用户都表示出来。所以，我借用了微信的朋友圈点赞，将最开始点赞的N个人进行展示。

由于这个功能要找到最开始点赞的N个人，所以就对点赞时间有了一个顺序之分，这时就不能使用集合来实现这个业务，转而换成有序集合来进行实现。

1. 将上述两个功能的代码进行修改

    /**
     * 点赞功能
     * @param id 博客id
     */
    public Boolean likeBlog(Long id) {
        String key = Constants.BLOG_LIKE_KEY + id;
        // TODO 从ThreadLocal中获取用户id
        Long userId = UserHolder.getId();
        if(userId == null) {
            return false;
        }
        // TODO 从Redis中判断该用户对该博客是否进行点赞
        Double score = this.stringRedisTemplate.opsForZSet()
                .score(key, userId.toString());
        // TODO 判断是否点赞
        if(score == null) {
            // 未点赞
            // TODO 博客总点赞数加一
            LambdaUpdateWrapper<Blog> blogLambdaUpdateWrapper = new LambdaUpdateWrapper<>();
            blogLambdaUpdateWrapper.setSql("liked = liked + 1")
                    .eq(Blog::getId, id);
            boolean add = this.update(blogLambdaUpdateWrapper);
            // TODO Redis缓存中添加该点赞
            if(add) {
                this.stringRedisTemplate.opsForZSet()
                        .add(key, userId.toString(), System.currentTimeMillis());
            }
        } else {
            // 已经点赞
            // TODO 博客总点赞数减一
            LambdaUpdateWrapper<Blog> blogLambdaUpdateWrapper = new LambdaUpdateWrapper<>();
            blogLambdaUpdateWrapper.setSql("liked = liked - 1")
                    .eq(Blog::getId, id);
            boolean subtract = this.update(blogLambdaUpdateWrapper);
            // TODO Redis缓存中删除该点赞
            if(subtract) {
                this.stringRedisTemplate.opsForZSet()
                        .remove(key, userId.toString());
            }
        }
        return true;
    }

    /**
     * 查询是否点赞
     * @param id 博客id
     */
    public Boolean queryLikeBlog(Long id) {
        String key = Constants.BLOG_LIKE_KEY + id;
        // TODO 从ThreadLocal中获取用户id
        Long userId = UserHolder.getId();
        if(userId == null) {
            return false;
        }
        // TODO 从Redis中判断是否点赞
        Double score = this.stringRedisTemplate.opsForZSet()
                .score(key, userId.toString());
        // TODO 返回结果
        if(score == null) {
            return false;
        }
        return true;
    }

2. 实现点赞排行榜功能

    /**
     * 点赞排行榜
     * 找出点赞的前5个人
     * @param id 博客id
     */
    public List<User> queryLikeBlogOfUsers(Long id) {
        String key = Constants.BLOG_LIKE_KEY + id;
        // TODO 按照业务要求找出前N个人，此处是5个人
        Set<String> top5 = this.stringRedisTemplate.opsForZSet()
                .range(key, 0, 4);
        // TODO 判断数据是否为空
        if(top5 == null || top5.isEmpty()) {
            return Collections.emptyList();
        }
        // TODO 将字符串类型转换为Long类型进行查询用户的详细信息
        List<Long> userIds = new ArrayList<>();
        for(String userId : top5) {
            userIds.add(Long.parseLong(userId));
        }
        // TODO 返回结果
        return this.userService.listByIds(userIds);
    }

对于点赞业务的讲解就这么多了，其实这里只是简单的实现了一个点赞业务而已。不同的业务自然就会出现不同的代码，即使实现的都是一个功能。所以，对代码的熟悉固然重要，但是对业务的熟悉也是非常重要的。