需求 功能: P15
发布文章
获取文章
文章分组
投支持票
数值及限制条件 P15如果一篇文章获得了至少 200 张支持票,那么这篇文章就是一篇有趣的文章
如果这个网站每天有 50 篇有趣的文章,那么网站要把这 50 篇文章放到文章列表页前 100 位至少一天
支持文章评分(投支持票会加评分),且评分随时间递减
实现 投支持票 P15如果要实现评分实时随时间递减,且支持按评分排序,那么工作量很大而且不精确。可以想到只有时间戳会随时间实时变化,如果我们把发布文章的时间戳当作初始评分,那么后发布的文章初始评分一定会更高,从另一个层面上实现了评分随时间递减。按照每个有趣文章每天 200 张支持票计算,平均到一天(86400 秒)中,每张票可以将分提高 432 分。
为了按照评分和时间排序获取文章,需要文章 id 及相应信息存在两个有序集合中,分别为:postTime 和 score 。
为了防止统一用户对统一文章多次投票,需要记录每篇文章投票的用户id,存储在集合中,为:votedUser:{articleId} 。
同时规定一篇文章发布期满一周后不能再进行投票,评分将被固定下来,同时记录文章已经投票的用户名单集合也会被删除。
// 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:" ) const ONE_WEEK_SECONDS = int64(7 * 24 * 60 * 60) const UPVOTE_SCORE = 432 // 用户 userId 给文章 articleId 投赞成票(没有事务控制,第 4 章会介绍 Redis 事务) func UpvoteArticle(conn redis.Conn, userId int, articleId int) { // 计算当前时间能投票的文章的最早发布时间 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)) addedNum, err := redis.Int(conn.Do("SADD", votedUserKey, userId)) // 添加错误 或 当前已投过票,则返回 if err != nil || addedNum == 0 { return } // 用户已成功添加到当前文章的投票集合中,则增加 当前文章 得分 _, err = conn.Do("ZINCRBY", SCORE, UPVOTE_SCORE, articleId) // 自增错误,则返回 if err != nil { return } // 增加 当前文章 支持票数 articleKey := ARTICLE_PREFIX + RedisKey(strconv.Itoa(articleId)) _, err = conn.Do("HINCRBY", articleKey, 1) // 自增错误,则返回 if err != nil { return } } 发布文章 P17可以使用 INCR 命令为每个文章生成一个自增唯一 id 。
将发布者的 userId 记录到该文章的投票用户集合中(即发布者默认为自己投支持票),同时设置过期时间为一周。
存储文章相关信息,并将初始评分和发布时间记录下来。
// 发布文章(没有事务控制,第 4 章会介绍 Redis 事务) func PostArticle(conn redis.Conn, userId int, title string, link string) { // 获取当前文章自增 id articleId, err := redis.Int(conn.Do("INCR", ARTICLE_COUNT)) if err != nil { return } // 将作者加入到投票用户集合中 votedUserKey := VOTED_USER_PREFIX + RedisKey(strconv.Itoa(articleId)) _, err = conn.Do("SADD", votedUserKey, userId) if err != nil { return } // 设置 投票用户集合 过期时间为一周 _, err = conn.Do("EXPIRE", votedUserKey, ONE_WEEK_SECONDS) if err != nil { return } postTime := time.Now().Unix() articleKey := ARTICLE_PREFIX + RedisKey(strconv.Itoa(articleId)) // 设置文章相关信息 _, err = conn.Do("HMSET", articleKey, "title", title, "link", link, "userId", userId, "postTime", postTime, "upvoteNum", 1, ) if err != nil { return } // 设置 发布时间 _, err = conn.Do("ZADD", POST_TIME, postTime, articleId) if err != nil { return } // 设置 文章评分 score := postTime + UPVOTE_SCORE _, err = conn.Do("ZADD", SCORE, score, articleId) if err != nil { return } } 分页获取文章 P18分页获取支持四种排序,获取错误时返回空数组。
注意:ZRANGE 和 ZREVRANGE 的范围起止都是闭区间。
type ArticleOrder int const ( TIME_ASC ArticleOrder = iota TIME_DESC SCORE_ASC SCORE_DESC ) // 根据 ArticleOrder 获取相应的 命令 和 RedisKey func getCommandAndRedisKey(articleOrder ArticleOrder) (string, RedisKey) { switch articleOrder { case TIME_ASC: return "ZRANGE", POST_TIME case TIME_DESC: return "ZREVRANGE", POST_TIME case SCORE_ASC: return "ZRANGE", SCORE case SCORE_DESC: return "ZREVRANGE", SCORE default: return "", "" } } // 执行分页获取文章逻辑(忽略部分简单的参数校验等逻辑) func doListArticles(conn redis.Conn, page int, pageSize int, command string, redisKey RedisKey) []map[string]string { var articles []map[string]string // ArticleOrder 不对,返回空列表 if command == "" || redisKey == ""{ return nil } // 获取 起止下标(都是闭区间) start := (page - 1) * pageSize end := start + pageSize - 1 // 获取 文章id 列表 ids, err := redis.Ints(conn.Do(command, redisKey, start, end)) if err != nil { return articles } // 获取每篇文章信息 for _, id := range ids { articleKey := ARTICLE_PREFIX + RedisKey(strconv.Itoa(id)) article, err := redis.StringMap(conn.Do("HGETALL", articleKey)) if err == nil { articles = append(articles, article) } } return articles } // 分页获取文章 func ListArticles(conn redis.Conn, page int, pageSize int, articleOrder ArticleOrder) []map[string]string { // 获取 ArticleOrder 对应的 命令 和 RedisKey command, redisKey := getCommandAndRedisKey(articleOrder) // 执行分页获取文章逻辑,并返回结果 return doListArticles(conn, page, pageSize, command, redisKey) } 文章分组 P19