本文目录
- 一、投票功能
- 投票流程
- 实现代码
- redis投票
一、投票功能
投票流程
首先我们要明确,就是 谁(哪个用户:userID)
给 哪个帖子(postID)
投了 什么票
(赞成票or反对票)。
赞成票越多,热度越高,就会越展示在前面。
在redis中可以用zset
存储帖子,那么有两种存储方式。
一是根据帖子的发布时间存储帖子(根据时间戳来,时间越新,时间戳越大,越在前面),或者是按照评分来存储帖子。
然后还可以设计一个zset,用来存储给某个帖子投票用户。哪个用户投了赞成票就记为+1,投了反对票就记为-1。
所以总的来设计了一个帖子算法,投一张赞成票就加对应的分数,比如400分。
时间戳+赞成票*400分=评分,评分越高越放前面。
实现代码
首先我们封装了投票数据的结构体。
然后是controller层。
然后就是具体投票的业务逻辑实现部分Logic.VoteForPost
。
strconv.Itoa
是一个函数,用于将整数转换为字符串。它要求输入必须是 int 类型,因此这里使用了 int(userId) 将 userId 转换为 int。将用户 ID 转换为字符串格式,因为 Redis 的键通常是以字符串形式存储的。
redis投票
来看看投票的情况:
v=1时,有两种情况1.之前没投过票,现在要投赞成票 --> 更新分数和投票记录 差值的绝对值:1 +4322.之前投过反对票,现在要改为赞成票 --> 更新分数和投票记录 差值的绝对值:2 +432*2v=0时,有两种情况1.之前投过反对票,现在要取消 --> 更新分数和投票记录 差值的绝对值:1 +4322.之前投过赞成票,现在要取消 --> 更新分数和投票记录 差值的绝对值:1 -432v=-1时,有两种情况1.之前没投过票,现在要投反对票 --> 更新分数和投票记录 差值的绝对值:1 -4322.之前投过赞成票,现在要改为反对票 --> 更新分数和投票记录 差值的绝对值:2 -432*2
除此之外,我们还有对投票的限制:
每个帖子自发起之日起,一个星期之内允许用户投票,超过一个星期就不允许投票了。同时到期之后将redis中保存的赞成票数及反对票数存储到mysql表中。到期之后删除 KeyPostVotedZSetPrefix
。
这里的 KeyPostVotedZSetPrefix
就是记录用户及投票类型。
我们来看看redis.go
的相关代码,其中client
就是redis
客户端。
在service
层中,设置了相关的逻辑代码,来看看处理流程。
首先我们需要去redis中获取帖子的发布时间,从client中拿即可。并检查当前时间与帖子发布时间的差值是否超过一周(OneWeekInSeconds
)。如果超过一周,返回错误 ErrorVoteTimeExpire
,表示投票已过期。
我们在redis
的目录路径下,封装了error错误
,声明了几个自定义错误变量。这些错误变量用于在 Redis 相关的操作中表示特定的错误情况。errors.New
是一个常用的error函数,用于创建一个新的错误对象。
通过定义全局的错误变量,为 Redis 相关的操作提供了一致的错误处理机制。使用 errors.New
创建的错误对象可以在整个包中复用,避免了重复创建相同的错误信息,提高了代码的可维护性和一致性。
比如当投票时间已经过期了,我们就需要返回投票过期的错误。
postTime := client.ZScore(KeyPostTimeZSet, postID).Val()
ZScore 方法查询帖子 ID 对应的分数(发布时间),并通过 Val() 方法获取该分数的实际值。返回的 postTime 是一个浮点数,表示帖子的发布时间(通常是 Unix 时间戳)。
ZScore 是 Redis 客户端提供的一个方法,用于从有序集合(ZSet)中获取某个成员的分数。它的签名通常是:
func (c *Client) ZScore(key string, member string) *FloatCmd
key:有序集合的键名(唯一标识:键名是 Redis 数据库中唯一标识有序集合的字符串。通过键名,你可以访问和操作特定的有序集合。)
member:有序集合中的成员(在这里是帖子的 ID)。
在刚刚我们有提到,const KeyPostTimeZSet = "bluebell:post:time"
,这是一个常量,定义了存储帖子发布时间的有序集合的键名。
然后就是更新帖子分数,注意更新帖子分数+记录用户为该帖子投票的数据 是要放在一个redis事务中完成的。
在 Redis 中,Pipeline
(管道) 是一种用于将多个命令发送到服务器的技术,而 事务(Transaction
) 是一种将多个命令打包并一次性执行的机制。在 Redis 的上下文中,Pipeline 和事务经常结合使用,以提高性能和确保操作的原子性
。
事务 是一种将多个命令打包,并一次性、顺序地执行的机制。Redis 的事务通过 MULTI、EXEC、DISCARD 和 WATCH 命令实现。事务的主要特点包括:事务中的所有命令要么全部执行,要么全部不执行。这确保了操作的原子性,避免了部分执行导致的数据不一致问题。事务中的命令会按照顺序执行,不会被其他客户端的命令打断。
KeyPostVotedZSetPrefix = "bluebell:post:voted:" // zset;记录用户及投票类型;参数是post_id
刚刚我们定义了redis的常量,所以我们需要进行下面的操作:
key := KeyPostVotedZSetPrefix + postIDov := client.ZScore(key, userID).Val()
也就是从 redis中获取 某个帖子 的 用户投票类型,根据用户ID来获取Val值。
也就是下面这个图所示:
ov := client.ZScore(key, userID).Val()// 更新:如果这一次投票的值和之前保存的值一致,就提示不允许重复投票if v == ov {return ErrVoteRepested}var op float64if v > ov {op = 1} else {op = -1}diffAbs := math.Abs(ov - v) // 计算两次投票的差值pipeline := client.TxPipeline() // 事务操作_, err = pipeline.ZIncrBy(KeyPostScoreZSet, VoteScore*diffAbs*op, postID).Result() // 更新分数
通过Redis事务来更新分数。
TxPipeline() 是 Redis 客户端提供的一个方法,用于创建一个事务性 Pipeline。这个 Pipeline 允许将多个命令打包在一起,并作为一个事务发送到 Redis 服务器。
ZIncrBy 是 Redis 的一个命令,用于在有序集合(ZSet)中增加某个成员的分数。
if v == 0 {_, err = client.ZRem(key, userID).Result()} else {pipeline.ZAdd(key, redis.Z{ // 记录已投票Score: v, // 赞成票还是反对票Member: userID,})}_, err = pipeline.Exec() //执行pipeline中的所有命令
如果v=0,那么从有序集合中移除指定的成员。
client.ZRem:Redis 客户端提供的方法,用于从有序集合中移除指定的成员。
pipeline.ZAdd:Redis 客户端提供的方法,用于将一个成员及其分数添加到有序集合中。这里使用了事务性 Pipeline,确保操作的原子性。
redis.Z:一个结构体,包含成员(Member)和分数(Score)。
Score:用户的投票值(v),表示赞成票(1)或反对票(-1)。
Member:用户的唯一标识符(userID)。
在 Redis 中,有序集合(ZSet)相关的命令都以 Z 开头,例如 ZADD、ZSCORE、ZINCRBY、ZREM 等。
Redis 的有序集合命令都以 Z 开头,例如:
ZADD:将一个或多个成员及其分数添加到有序集合中。
ZSCORE:获取有序集合中成员的分数。
ZINCRBY:增加有序集合中成员的分数。
ZREM:从有序集合中移除成员。
可以看到,再投出一票之后,在原先的redis基础上加了432分。