10. Redis implements likes (Set) and obtains the total number of likes

实现点赞

No need to write the data access layer,直接在Servicelayer can be written.

1、创建RedisKeyUtil工具类生成key：

public class RedisKeyUtil {
    
    private static final String SPLIT = ":";
    private static final String PREFIX_ENTITY_LIKE = "like:entity";

    // 某个实体的赞
    // like:entity:entityType:entityId -> set(userId)
    public static String getEntityLikeKey(int entityType, int entityId) {
    
        return PREFIX_ENTITY_LIKE + SPLIT + entityType + SPLIT + entityId;
    }
}

2、LikeService

@Service
public class LikeService {
    

    @Autowired
    private RedisTemplate redisTemplate;

    // 点赞,查看该用户userId是否在集合里,Not in the set,加进去,like;在集合里,取消点赞,remove掉
    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);
        }
    }

    // 查询某实体点赞的数量,See how many are in the setuserId
    public long findEntityLikeCount(int entityType, int entityId) {
    
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().size(entityLikeKey);
    }

    // 查询某人对某实体的点赞状态,看某userId是否在集合里,返回1点赞了,返回0没点赞（Tap to go back-1）
    public int findEntityLikeStatus(int userId, int entityType, int entityId) {
    
        String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
        return redisTemplate.opsForSet().isMember(entityLikeKey, userId) ? 1 : 0;
    }
    
}

3、LikeController

@Controller
public class LikeController {
    

    @Autowired
    private LikeService likeService;

    @Autowired
    private HostHolder hostHolder;

    @RequestMapping(path = "/like", method = RequestMethod.POST)
    @ResponseBody
    public String like(int entityType, int entityId) {
    
        User user = hostHolder.getUser();

        // 点赞
        likeService.like(user.getId(), entityType, entityId);

        // 重新计算数量
        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);

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

}

4、异步请求,discuss.js

function like(btn, entityType, entityId) {
    
    $.post(
        CONTEXT_PATH + "/like",
        {
    "entityType":entityType,"entityId":entityId},
        function(data) {
    
            data = $.parseJSON(data);
            if(data.code == 0) {
    
                $(btn).children("i").text(data.likeCount);
                $(btn).children("b").text(data.likeStatus==1?'已赞':"赞");
            } else {
    
                alert(data.msg);
            }
        }
    );
}

Get total likes

1、改key拼接工具类RedisKeyUtil

	private static final String PREFIX_USER_LIKE = "like:user";
    // The total number of likes for a user, value是Object->强转成Integer->intValue()转成int型
    // like:user:userId -> int
    public static String getUserLikeKey(int userId){
    
        return PREFIX_USER_LIKE + SPLIT + userId;
    }

2、LikeService,编程式事务

保证事务性,Add two likes at once,编程式事务

public void like(int userId, int entityType, int entityId, int entityUserId) {
    

        // 保证事务性,Two likes at a time increase,编程式事务
        redisTemplate.execute(new SessionCallback() {
    
            @Override
            public Object execute(RedisOperations operations) throws DataAccessException {
    
                String entityLikeKey = RedisKeyUtil.getEntityLikeKey(entityType, entityId);
                String userLikeKey = RedisKeyUtil.getUserLikeKey(entityUserId);

                // The query should precede the transaction,It cannot be placed in a transaction,Otherwise, the transaction will be executed after committing.
                boolean isMember = operations.opsForSet().isMember(entityLikeKey, userId);

                operations.multi();

                if (isMember) {
    
                    operations.opsForSet().remove(entityLikeKey, userId);// 取消点赞
                    operations.opsForValue().decrement(userLikeKey);// Decrease the number of likes for an entity
                }else{
    
                    operations.opsForSet().add(entityLikeKey, userId);// 点赞
                    operations.opsForValue().increment(userLikeKey);// Increase the number of likes for an entity
                }

                return operations.exec();
            }
        });

    }

Query the total number of likes a user has received

// Query the total number of likes a user has received
public int findUserLikeCount(int userId) {
    
    String userLikeKey = RedisKeyUtil.getUserLikeKey(userId);
    Integer count = (Integer) redisTemplate.opsForValue().get(userLikeKey);
    return count == null ? 0 : count.intValue();
}