fix: stabilize idempotent decorator cache key
This commit is contained in:
parent
fbe12249f8
commit
71e91efe29
|
|
@ -2,6 +2,7 @@ import logging
|
|||
from functools import wraps
|
||||
from apps.utils.tasks import send_mail_task
|
||||
import traceback
|
||||
import hashlib
|
||||
import json
|
||||
from django.core.cache import cache
|
||||
from rest_framework.exceptions import ParseError
|
||||
|
|
@ -30,18 +31,23 @@ def idempotent(seconds=4):
|
|||
def decorate(func):
|
||||
@wraps(func)
|
||||
def wrapper(*args, **kwargs):
|
||||
rdata = args[1].data
|
||||
rdata['request_userid'] = getattr(args[1], 'user').id
|
||||
rdata['request_path'] = getattr(args[1], 'path')
|
||||
hash_k = hash(json.dumps(rdata))
|
||||
request = args[1]
|
||||
rdata = dict(request.data)
|
||||
rdata['request_userid'] = getattr(request, 'user').id
|
||||
rdata['request_path'] = getattr(request, 'path')
|
||||
payload = json.dumps(rdata, sort_keys=True, ensure_ascii=False, default=str)
|
||||
hash_k = hashlib.sha256(payload.encode('utf-8')).hexdigest()
|
||||
hash_v_e = cache.get(hash_k, None)
|
||||
if hash_v_e is None:
|
||||
cache.set(hash_k, 'o', seconds)
|
||||
real_func = func(*args, **kwargs)
|
||||
# real_func.render()
|
||||
# cache.set(hash_k, real_func, seconds)
|
||||
if hasattr(real_func, 'data'):
|
||||
cache.set(hash_k, real_func.data, seconds)
|
||||
else:
|
||||
cache.delete(hash_k)
|
||||
return real_func
|
||||
elif hash_v_e == 'o': # 说明请求正在处理
|
||||
raise ParseError(f'请求忽略,请{seconds}秒后重试')
|
||||
return hash_v_e
|
||||
return wrapper
|
||||
return decorate
|
||||
return decorate
|
||||
|
|
|
|||
Loading…
Reference in New Issue