支持将文章加入到分组集合,也支持将文章从分组集合中删除。
// 设置分组 func AddToGroup(conn redis.Conn, groupId int, articleIds ...int) { groupKey := GROUP_PREFIX + RedisKey(strconv.Itoa(groupId)) args := make([]interface{}, 1 + len(articleIds)) args[0] = groupKey // []int 转换成 []interface{} for i, articleId := range articleIds { args[i + 1] = articleId } // 不支持 []int 直接转 []interface{} // 也不支持 groupKey, articleIds... 这样传参(这样匹配的参数是 interface{}, ...interface{}) _, _ = conn.Do("SADD", args...) } // 取消分组 func RemoveFromGroup(conn redis.Conn, groupId int, articleIds ...int) { groupKey := GROUP_PREFIX + RedisKey(strconv.Itoa(groupId)) args := make([]interface{}, 1 + len(articleIds)) args[0] = groupKey // []int 转换成 []interface{} for i, articleId := range articleIds { args[i + 1] = articleId } // 不支持 []int 直接转 []interface{} // 也不支持 groupKey, articleIds... 这样传参(这样匹配的参数是 interface{}, ...interface{}) _, _ = conn.Do("SREM", args...) } 分组中分页获取文章 P20分组信息和排序信息在不同的(有序)集合中,所以需要取两个(有序)集合的交集,再进行分页获取。
取交集比较耗时,所以缓存 60s,不实时生成。
// 缓存过期时间 60s const EXPIRE_SECONDS = 60 // 分页获取某分组下的文章(忽略简单的参数校验等逻辑;过期设置没有在事务里) func ListArticlesFromGroup(conn redis.Conn, groupId int, page int, pageSize int, articleOrder ArticleOrder) []map[string]string { // 获取 ArticleOrder 对应的 命令 和 RedisKey command, redisKey := getCommandAndRedisKey(articleOrder) // ArticleOrder 不对,返回空列表,防止多做取交集操作 if command == "" || redisKey == ""{ return nil } groupKey := GROUP_PREFIX + RedisKey(strconv.Itoa(groupId)) targetRedisKey := redisKey + RedisKey("-inter-") + groupKey exists, err := redis.Int(conn.Do("EXISTS", targetRedisKey)) // 交集不存在或已过期,则取交集 if err == nil || exists != 1 { _, err := conn.Do("ZINTERSTORE", targetRedisKey, 2, redisKey, groupKey) if err != nil { return nil } } // 设置过期时间(过期设置失败,不影响查询) _, _ = conn.Do("EXPIRE", targetRedisKey, EXPIRE_SECONDS) // 执行分页获取文章逻辑,并返回结果 return doListArticles(conn, page, pageSize, command, targetRedisKey) } 练习题:投反对票 P21增加投反对票功能,并支持支持票和反对票互转。
看到这个练习和相应的提示后,又联系平日里投票的场景,觉得题目中的方式并不合理。在投支持/反对票时处理相应的转换逻辑符合用户习惯,也能又较好的扩展性。
更改处
文章 HASH,增加一个 downvoteNum 字段,用于记录投反对票人数
文章投票用户集合 SET 改为 HASH,用于存储用户投票的类型
UpvoteArticle 函数换为 VoteArticle,同时增加一个类型为 VoteType 的入参。函数功能不仅支持投支持/反对票,还支持取消投票
// redis key type RedisKey string const ( // 发布时间 有序集合 POST_TIME RedisKey = "postTime" // 文章评分 有序集合 SCORE RedisKey = "score" // 文章投票用户集合前缀 VOTED_USER_PREFIX RedisKey = "votedUser:" // 发布文章数 字符串 ARTICLE_COUNT RedisKey = "articleCount" // 发布文章哈希表前缀 ARTICLE_PREFIX RedisKey = "article:" // 分组前缀 GROUP_PREFIX RedisKey = "group:" ) type VoteType string const ( // 未投票 NONVOTE VoteType = "" // 投支持票 UPVOTE VoteType = "1" // 投反对票 DOWNVOTE VoteType = "2" ) const ONE_WEEK_SECONDS = int64(7 * 24 * 60 * 60) const UPVOTE_SCORE = 432 // 根据 原有投票类型 和 新投票类型,获取 分数、支持票数、反对票数 的增量(暂未处理“枚举”不对的情况,直接全返回 0) func getDelta(oldVoteType VoteType, newVoteType VoteType) (scoreDelta, upvoteNumDelta, downvoteNumDelta int) { // 类型不变,相关数值不用改变 if oldVoteType == newVoteType { return 0, 0, 0 } switch oldVoteType { case NONVOTE: if newVoteType == UPVOTE { return UPVOTE_SCORE, 1, 0 } if newVoteType == DOWNVOTE { return -UPVOTE_SCORE, 0, 1 } case UPVOTE: if newVoteType == NONVOTE { return -UPVOTE_SCORE, -1, 0 } if newVoteType == DOWNVOTE { return -(UPVOTE_SCORE << 1), -1, 1 } case DOWNVOTE: if newVoteType == NONVOTE { return UPVOTE_SCORE, 0, -1 } if newVoteType == UPVOTE { return UPVOTE_SCORE << 1, 1, -1 } default: return 0, 0, 0 } return 0, 0, 0 } // 为 投票 更新数据(忽略部分参数校验;没有事务控制,第 4 章会介绍 Redis 事务) func doVoteArticle(conn redis.Conn, userId int, articleId int, oldVoteType VoteType, voteType VoteType) { // 获取 分数、支持票数、反对票数 增量 scoreDelta, upvoteNumDelta, downvoteNumDelta := getDelta(oldVoteType, voteType) // 更新当前用户投票类型 votedUserKey := VOTED_USER_PREFIX + RedisKey(strconv.Itoa(articleId)) _, err := conn.Do("HSET", votedUserKey, userId, voteType) // 设置错误,则返回 if err != nil { return } // 更新 当前文章 得分 _, err = conn.Do("ZINCRBY", SCORE, scoreDelta, articleId) // 自增错误,则返回 if err != nil { return } // 更新 当前文章 支持票数 articleKey := ARTICLE_PREFIX + RedisKey(strconv.Itoa(articleId)) _, err = conn.Do("HINCRBY", articleKey, "upvoteNum", upvoteNumDelta) // 自增错误,则返回 if err != nil { return } // 更新 当前文章 反对票数 _, err = conn.Do("HINCRBY", articleKey, "downvoteNum", downvoteNumDelta) // 自增错误,则返回 if err != nil { return } } // 执行投票逻辑(忽略部分参数校验;没有事务控制,第 4 章会介绍 Redis 事务) func VoteArticle(conn redis.Conn, userId int, articleId int, voteType VoteType) { // 计算当前时间能投票的文章的最早发布时间 earliestPostTime := time.Now().Unix() - ONE_WEEK_SECONDS // 获取 当前文章 的发布时间 postTime, err := redis.Int64(conn.Do("ZSCORE", POST_TIME, articleId)) // 获取错误 或 文章 articleId 的投票截止时间已过,则返回 if err != nil || postTime < earliestPostTime { return } // 获取集合中投票类型 votedUserKey := VOTED_USER_PREFIX + RedisKey(strconv.Itoa(articleId)) result, err := conn.Do("HGET", votedUserKey, userId) // 查询错误,则返回 if err != nil { return } // 转换后 oldVoteType 必为 "", "1", "2" 其中之一 oldVoteType, err := redis.String(result, err) // 如果投票类型不变,则不进行处理 if VoteType(oldVoteType) == voteType { return } // 执行投票修改数据逻辑 doVoteArticle(conn, userId, articleId, VoteType(oldVoteType), voteType) } 小结Redis 特性
内存存储:Redis 速度非常快
远程:Redis 可以与多个客户端和服务器进行连接
持久化:服务器重启之后仍然保持重启之前的数据
可扩展:主从复制和分片
所思所想代码不是一次成形的,会在写新功能的过程中不断完善以前的逻辑,并抽取公共方法以达到较高的可维护性和可扩展性。
感觉思路还是没有转过来(不知道还是这个 Redis 开源库的问题),一直运用 Java 的思想,很多地方写着不方便。
虽然自己写的一些私有的方法保证不会出现某些异常数据,但是还是有一些会进行相应的处理,以防以后没注意调用了出错。