详解redis是如何实现队列消息的ack_Redis

详解redis是如何实现队列消息的ack

2019-11-05 12:14Daniel Cai Redis

这篇文章主要介绍了关于redis是如何实现队列消息的ack的相关资料，文中介绍的非常详细，需要的朋友可以参考借鉴，下面来一起看看吧。

前言

由于公司提供的队列实在太过于蛋疼而且还限制不能使用其他队列，但为了保证数据安全性需要一个可以有ack功能的队列。

原生的redis中通过L/R PUSH/POP方式来实现队列的功能，这个当然是没办法满足需求的（没有ack功能），所以需要自己对redis的list（队列）做个小小的调整。

大体思路为在POP时将pop出的数据放到备份的地方，当有ACK请求（确认消息被消耗）后将备份的信息删除掉；每次在pop前需要检查备份队列中有没有过期的数据没有ack的，如果有则PUSH到list中后再从list中POP出来。

以下脚本使用lua实现，只需要在执行前加载到redis中即可。

消息本身需要包含id属性

push没什么问题，原生即可（此处以LPUSH为例）

pop时脚本

				?

									local not_empty = function(x)

									 return (type(x) == "table") and (not x.err) and (#x ~= 0)

									end

									local qName = ARGV[1] --队列名称

									local currentTime = ARGV[2] --当前时间，这个需要从外部传入，不能使用redis自身时间，如果使用自身时间可能导致redis本身的backup在重放请求时出现不一致性

									local considerAsFailMaxTimeSpan = ARGV[3] --超时时间设定，当消息超过一定时间还没有ack则认为此消息需要再次入队

									local zsetName= qName ..'BACKUP'

									local hashName= qName ..'CONTEXT'

									local tmp = redis.call('ZRANGEBYSCORE',zsetName , '-INF', tonumber(currentTime) - tonumber(considerAsFailMaxTimeSpan), 'LIMIT', 0, 1)

									if (not_empty(tmp)) then

									 redis.call('ZREM', zsetName, tmp[1]) --此处拿出的为消息的唯一id

									 redis.call('LPUSH', qName, redis.call('HGET', hashName, tmp[1]))

									end

									tmp = redis.call('RPOP', qName)

									if (tmp) then

									 local msg = cjson.decode(tmp)

									 local id = msg['id']

									 redis.call('ZADD', zsetName, tonumber(currentTime), id)

									 redis.call('HSET',hashName , id, tmp)

									end

									return tmp

ack时候比较简单，只需要将指定id从set和hash中删除即可

				?

									local key = ARGV[1]

									local qName=ARGV[2]

									redis.call('ZREM', qName..'BACKUP', key)

									redis.call('HDEL', qName..'CONTEXT', key)