perf: Parsing Tokens (#3964)

v3.2
shaohuzhang1 2025-08-29 11:54:16 +08:00 committed by GitHub
parent 75c461f385
commit f6aaac4e00
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 31 additions and 6 deletions

View File

@ -6,27 +6,52 @@
@date2025/6/6 19:55 @date2025/6/6 19:55
@desc: @desc:
""" """
import hashlib
import json import json
import threading
from django.core import signing from django.core import signing, cache
from common.constants.cache_version import Cache_Version
from common.utils.rsa_util import encrypt, decrypt from common.utils.rsa_util import encrypt, decrypt
authentication_cache = cache.cache
lock = threading.Lock()
def _decrypt(authentication: str):
cache_key = hashlib.sha256(authentication.encode()).hexdigest()
result = authentication_cache.get(key=cache_key, version=Cache_Version.CHAT.value)
if result is None:
with lock:
result = authentication_cache.get(cache_key, version=Cache_Version.CHAT.value)
if result is None:
result = decrypt(authentication)
authentication_cache.set(cache_key, result, version=Cache_Version.CHAT.value, timeout=60 * 60 * 2)
return result
class ChatAuthentication: class ChatAuthentication:
def __init__(self, auth_type: str | None): def __init__(self, auth_type: str | None, **kwargs):
self.auth_type = auth_type self.auth_type = auth_type
for k, v in kwargs.items():
self.__setattr__(k, v)
def to_dict(self): def to_dict(self):
return {'auth_type': self.auth_type} return self.__dict__
def to_string(self): def to_string(self):
return encrypt(json.dumps(self.to_dict())) value = json.dumps(self.to_dict())
authentication = encrypt(value)
cache_key = hashlib.sha256(authentication.encode()).hexdigest()
authentication_cache.set(cache_key, value, version=Cache_Version.CHAT.value, timeout=60 * 60 * 2)
return authentication
@staticmethod @staticmethod
def new_instance(authentication: str): def new_instance(authentication: str):
auth = json.loads(decrypt(authentication)) auth = json.loads(_decrypt(authentication))
return ChatAuthentication(auth.get('auth_type')) return ChatAuthentication(**auth)
class ChatUserToken: class ChatUserToken: