redis中如何根据key的存在设置hash key过期时间

Question

我想设置一些hash key的过期时间，如果是第一次hset key，我希望设置一个过期时间，否则我宁愿保留第一次设置的过期时间.

由于有大量的哈希键，我更喜欢在管道中进行，但是下面的函数不能很好地工作。

pipe.exists(hkey)这行返回一个管道的obj，它总是True，所以if子句总是去一个部分，不管hash key是否存在。

我的问题是：有没有一种方法可以通过管道的哈希键的存在来设置哈希键的到期？

def test1(hkey, v):
    with r.pipeline() as pipe:
        # tmp = pipe.exists(hkey)
        # pipe.exists(hkey) is a pipe obj, which is always True, 
        # this line not work as expected and the two lines below it will never be excuted.
        if not pipe.exists(hkey):
            pipe.hset(hkey, v, v)
            pipe.expire(hkey, 3600)
        else:
            # no matter whether the hash key is exist or not, the if else statment always goes to this line.
            pipe.hset(hkey, v, v)
        pipe.execute()

Answer 1

您无法通过管道实现这一点，因为在整个管道执行之前您永远不知道密钥是否存在。相反，您可以使用Lua scripting 来完成这项工作：

local key = KEYS[1]
local field = ARGV[1]
local value = ARGV[2]
local ttl = ARGV[3]

local exist = redis.call('exists', key)

redis.call('hset', key, field, value)

if exist == 0 then
    redis.call('expire', key, ttl)
end

查看this 了解如何使用 redis-py 运行 Lua 脚本。然后使用管道运行脚本以减少RTT。

更新

如果你坚持使用WATCH来做这个工作，你可以试试下面的代码：

with r.pipeline() as pipe:
    while 1:
        try:
            pipe.watch(hkey)

            exist = pipe.exists(hkey)

            pipe.multi()

            if not exist:
                pipe.hset(hkey, v, v)
                pipe.expire(hkey, 3600)
            else:
                pipe.hset(hkey, v, v)

            pipe.execute()
            break;
        except WatchError:
            continue