redis：每天重置计数器

Question

我希望每天使用 Redis 重置一个计数器。我是 Redis 的新手，所以我想确保我很好地理解事务和管道的工作原理。

以下代码是否确保我在多进程环境中工作时总是会得到一对唯一的（日期、数字），或者我是否需要使用 Redis 锁？

import datetime
import redis

r = redis.Redis(...)

def get_utc_date_now():
    return datetime.datetime.utcnow().date()

def get_daily_counter(r, dt_key='dt', counter_key='counter'):

    def incr_daily_number(pipe):
        dt_now = get_utc_date_now().isoformat()  # e.g.: "2014-10-18"
        dt = pipe.get(dt_key)    
        pipe.multi()    
        if dt != dt_now:
            pipe.set(dt_key, dt_now)
            pipe.set(counter_key, 0)
        pipe.get(dt_key)
        pipe.incr(counter_key)

    result = r.transaction(incr_daily_number, dt_key)
    return result[-2:]

# Get the (dt, number) couple
# 2014-10-18, 1
# 2014-10-18, 2
# etc.
dt, number = get_daily_counter(r)

更新

尝试使用 LUA 脚本：

r = redis.Redis(...)

incr_with_reset_on_change_lua_script = """

local dt = redis.call('GET', KEYS[2])
if dt ~= ARGV[2] then
  redis.call('MSET', KEYS[1], ARGV[1], KEYS[2], ARGV[2])
end
return redis.call('INCR', KEYS[1])

"""
# Incr KEYS1 but reset first if KEYS2 has changed.
incr_with_reset_on_change = r.register_script(incr_with_reset_on_change_lua_script)

counter_key = 'dcounterA'
watch_key = 'dcounterA_dt'
watch_value = get_utc_date_now().isoformat()

number = incr_with_reset_on_change(keys=[counter_key, watch_key], args=[reset_value, watch_value])

Answer 1

考虑发生在午夜的两个并发事务。两者都可以执行 get(dt_key)，但一个会先执行 MULTI/EXEC 块。它将重置计数器，设置新日期，增加计数器。第二个也会进入它的 MULTI/EXEC 块，但是因为 'dt' 的值发生了变化，执行将失败，并且将再次调用 incr_daily_number。这次 get(dt_key) 将返回新的日期，因此当 MULTI/EXEC 块将被执行时，计数器将在没有任何重置的情况下递增。这两个事务将返回具有不同计数器值的新日期。

所以，我相信这里没有种族条件，而且（日期，数量）夫妇将是独一无二的。

您也可以使用服务器端 Lua 脚本（其执行始终是原子的）来实现这一点。它通常更方便。

请注意，实际上，没有 Redis 锁这样的东西。API 中可用的锁定机制由 Python 客户端提供，而不是由 Redis 服务器提供。如果您查看它的实现，您会发现它也是基于 SETNX + WATCH/MULTI/EXEC 块或 Lua 脚本。