使用redis处理帖子点赞

M

public class RedisKeyUtil {

    private static final String SPLIT = ":";
    private static final String PREFIX_ENTITY_LIKE = "like:entity";

    // a entity like
    // like:entity:entityType:entityId -> set(userId)
    public static String getEntityLikeKey(int entityType, int entityId){
        return PREFIX_ENTITY_LIKE + SPLIT + entityType + SPLIT + entityId;
    }
}

C

查看帖子是否已点赞，查询有多少个赞等

@Service
public class LikeService {

    @Autowired
    private RedisTemplate redisTemplate;

    // like
    // entityType：帖子 = 1，评论 = 2
    // entityId：该帖子或评论的ID
    public void like(int userId, int entityType, int entityId){
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        boolean isMember = redisTemplate.opsForSet().isMember(entityLikeKey, userId);
        if (isMember) {
            redisTemplate.opsForSet().remove(entityLikeKey, userId);
        }else{
            redisTemplate.opsForSet().add(entityLikeKey, userId);
        }
    }

    // query the number of like
    public long findEntityLikeCount(int entityType, int entityId){
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().size(entityLikeKey);
    }

    // Check whether the user has liked this comment
    // boolean only have two status, we can use int to show three status : like, dislike, none
    public int findEntityLikeStatus(int userId, int entityType, int entityId){
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().isMember(entityLikeKey, userId) ? 1 : 0;
    }
}

V

@Controller
public class LikeController {

    @Autowired
    private LikeService likeService;

    @Autowired
    private HostHolder hostHolder;

    @PostMapping(path = "/like")
    @ResponseBody
    @LoginRequired			// 需要登录才能点赞
    public String like(int entityType, int entityId){
        User user = hostHolder.getUser();

        // like
        likeService.like(user.getId(), entityType, entityId);
        // number
        long likeCount = likeService.findEntityLikeCount(entityType, entityId);
        // status
        int likeStatus = likeService.findEntityLikeStatus(user.getId(), entityType, entityId);

        // return the result
        Map<String, Object> map = new HashMap<>();
        map.put("likeCount", likeCount);
        map.put("likeStatus", likeStatus);

        return CommunityUtil.getJSONString(0, null, map);
    }
}