仿牛客网项目---点赞模块的实现

本篇文章介绍一下项目中的点赞模块。

点赞模块是一个通过使用Redis实现的功能模块，它提供了点赞操作的处理逻辑和数据存取功能。通过服务类和控制器类的配合，点赞模块实现了用户对实体的点赞、点赞数量的查询、点赞状态的查询等功能。该模块使用了Redis作为数据存储，通过Redis的事务支持和数据结构操作，实现了点赞操作的原子性和高效性。同时，通过依赖注入和控制反转的思想，实现了代码的解耦和灵活性，使得点赞模块具有良好的可维护性和可扩展性。

首先我们写一个RedisKeyUtil的工具类，这个工具类提供了一些用于在Redis中生成键的方法。

public class RedisKeyUtil {// 某个实体的赞// like:entity:entityType:entityId -> set(userId)public static String getEntityLikeKey(int entityType, int entityId) {return PREFIX_ENTITY_LIKE + SPLIT + entityType + SPLIT + entityId;}// 某个用户的赞// like:user:userId -> intpublic static String getUserLikeKey(int userId) {return PREFIX_USER_LIKE + SPLIT + userId;}// 某个用户关注的实体// followee:userId:entityType -> zset(entityId,now)public static String getFolloweeKey(int userId, int entityType) {return PREFIX_FOLLOWEE + SPLIT + userId + SPLIT + entityType;}// 某个实体拥有的粉丝// follower:entityType:entityId -> zset(userId,now)public static String getFollowerKey(int entityType, int entityId) {return PREFIX_FOLLOWER + SPLIT + entityType + SPLIT + entityId;}// 登录验证码public static String getKaptchaKey(String owner) {return PREFIX_KAPTCHA + SPLIT + owner;}// 登录的凭证public static String getTicketKey(String ticket) {return PREFIX_TICKET + SPLIT + ticket;}// 用户public static String getUserKey(int userId) {return PREFIX_USER + SPLIT + userId;}// 单日UVpublic static String getUVKey(String date) {return PREFIX_UV + SPLIT + date;}// 区间UVpublic static String getUVKey(String startDate, String endDate) {return PREFIX_UV + SPLIT + startDate + SPLIT + endDate;}// 单日活跃用户public static String getDAUKey(String date) {return PREFIX_DAU + SPLIT + date;}// 区间活跃用户public static String getDAUKey(String startDate, String endDate) {return PREFIX_DAU + SPLIT + startDate + SPLIT + endDate;}// 帖子分数public static String getPostScoreKey() {return PREFIX_POST + SPLIT + "score";}}

举例来说，通过使用getEntityLikeKey方法，可以生成一个用于存储特定实体点赞记录的键，通过使用getUserLikeKey方法，可以生成一个用于存储特定用户点赞记录的键。其他方法则可以生成用于存储用户数据、关注关系、验证码、登录凭证等的键。

看到这里，你应该会觉得很奇怪，为什么没有DAO层呢？

因为点赞数据的存储和查询需求可以完全由Redis满足，并且Redis作为一个高性能的内存数据库能够提供快速的读写操作，简化了数据访问的复杂性和开销，同时满足了性能要求。此外，点赞数据可能具有相对简单的数据结构，可以直接使用Redis的数据结构进行存储和查询，而不需要传统的数据库访问层。

所以说DAO层在这个项目就没有用到了，我这个项目的点赞模块用的是Redis，关于一些点赞的信息我是用redis来存的。

然后再来写LikeService文件，即service层。

@Service
public class LikeService {@Autowiredprivate RedisTemplate redisTemplate;// 点赞public void like(int userId, int entityType, int entityId, int entityUserId) {redisTemplate.execute(new SessionCallback() {@Overridepublic Object execute(RedisOperations operations) throws DataAccessException {String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);String userLikeKey = RedisKeyUtil.getUserLikeKey(entityUserId);boolean isMember = operations.opsForSet().isMember(entityLikeKey, userId);operations.multi();if (isMember) {operations.opsForSet().remove(entityLikeKey, userId);operations.opsForValue().decrement(userLikeKey);} else {operations.opsForSet().add(entityLikeKey, userId);operations.opsForValue().increment(userLikeKey);}return operations.exec();}});}// 查询某实体点赞的数量public long findEntityLikeCount(int entityType, int entityId) {String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);return redisTemplate.opsForSet().size(entityLikeKey);}// 查询某人对某实体的点赞状态public int findEntityLikeStatus(int userId, int entityType, int entityId) {String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);return redisTemplate.opsForSet().isMember(entityLikeKey, userId) ? 1 : 0;}// 查询某个用户获得的赞public int findUserLikeCount(int userId) {String userLikeKey = RedisKeyUtil.getUserLikeKey(userId);Integer count = (Integer) redisTemplate.opsForValue().get(userLikeKey);return count == null ? 0 : count.intValue();}}

1. like(int userId, int entityType, int entityId, int entityUserId)：点赞方法，用于给特定实体点赞。它首先生成实体点赞的键和用户点赞的键，然后通过Redis的事务支持（multi()和exec()方法）来保证点赞操作的原子性。根据用户是否已经点赞，执行相应的操作（添加或移除点赞），并更新用户点赞数。

2. findEntityLikeCount(int entityType, int entityId)：查询某实体的点赞数量。它使用RedisKeyUtil生成实体点赞的键，并通过Redis的opsForSet().size()方法获取集合的大小，即点赞数量。

3. findEntityLikeStatus(int userId, int entityType, int entityId)：查询某人对某实体的点赞状态。它使用RedisKeyUtil生成实体点赞的键，并通过Redis的opsForSet().isMember()方法判断用户是否已经点赞。

4. findUserLikeCount(int userId)：查询某个用户获得的赞数量。它使用RedisKeyUtil生成用户点赞的键，并通过Redis的opsForValue().get()方法获取用户点赞数。

最后写controller层，即LikeController。

@Controller
public class LikeController implements CommunityConstant {//省略了注入LikeService、HostHolder、EventProducer和RedisTemplate来处理点赞逻辑的代码@RequestMapping(path = "/like", method = RequestMethod.POST)@ResponseBodypublic String like(int entityType, int entityId, int entityUserId, int postId) {User user = hostHolder.getUser();// 点赞likeService.like(user.getId(), entityType, entityId, entityUserId);// 数量long likeCount = likeService.findEntityLikeCount(entityType, entityId);// 状态int likeStatus = likeService.findEntityLikeStatus(user.getId(), entityType, entityId);// 返回的结果Map<String, Object> map = new HashMap<>();map.put("likeCount", likeCount);map.put("likeStatus", likeStatus);// 触发点赞事件if (likeStatus == 1) {Event event = new Event().setTopic(TOPIC_LIKE).setUserId(hostHolder.getUser().getId()).setEntityType(entityType).setEntityId(entityId).setEntityUserId(entityUserId).setData("postId", postId);eventProducer.fireEvent(event);}if(entityType == ENTITY_TYPE_POST) {// 计算帖子分数String redisKey = RedisKeyUtil.getPostScoreKey();redisTemplate.opsForSet().add(redisKey, postId);}return CommunityUtil.getJSONString(0, null, map);}}

上面这段代码是一个名为LikeController的控制器类，用于处理点赞相关的请求。它通过注入LikeService、HostHolder、EventProducer和RedisTemplate来处理点赞逻辑。

控制器中的主要方法是like()方法，用于处理点赞请求。该方法接受实体类型、实体ID、实体创建者ID和帖子ID作为参数。在方法中，首先获取当前用户信息，然后调用likeService的点赞方法，对实体进行点赞操作。

接下来，通过调用likeService的方法获取点赞数量和点赞状态，并将结果存储在一个Map对象中。然后，根据点赞状态触发相应的点赞事件，创建一个Event对象，并通过eventProducer发送事件。

如果实体类型是帖子（ENTITY_TYPE_POST），则计算帖子的分数并存储到Redis中。

最后，通过CommunityUtil.getJSONString()方法将结果封装成JSON字符串并返回给前端。

控制器类通过调用likeService处理点赞逻辑，而不是使用DAO层进行数据访问。这是因为点赞操作的数据存储和查询可以完全由Redis满足，并且通过调用likeService提供的方法，可以直接在控制器中处理点赞逻辑，避免了引入额外的DAO层。