reids 域名添加 添加时间映射
This commit is contained in:
19
app/utils.py
19
app/utils.py
@@ -265,7 +265,6 @@ def add_allowed_domain(domain):
|
||||
timestamp = datetime.now().isoformat()
|
||||
redis_client.rpush('allowed_domains_list', domain) # 添加到列表
|
||||
redis_client.hset(f'domain:{domain}', 'created_at', timestamp) # 记录创建时间
|
||||
redis_client.hset(f'domain_time:{domain}', 'timestamp', timestamp) # 记录时间
|
||||
logger.info(f'Added allowed domain: {domain} with timestamp: {timestamp}')
|
||||
except Exception as e:
|
||||
logger.error(f'Error adding allowed domain: {e}')
|
||||
@@ -275,33 +274,19 @@ def remove_allowed_domain(domain):
|
||||
"""删除允许的域名"""
|
||||
try:
|
||||
redis_client.srem('allowed_domains', domain)
|
||||
redis_client.delete(f'domain_time:{domain}') # 删除时间记录
|
||||
logger.info(f'Removed allowed domain: {domain}')
|
||||
except Exception as e:
|
||||
logger.error(f'Error removing allowed domain: {e}')
|
||||
|
||||
|
||||
def get_allowed_domains():
|
||||
"""获取当前允许的域名"""
|
||||
try:
|
||||
domains = redis_client.lrange('allowed_domains_list', 0, -1)
|
||||
domain_list = []
|
||||
for domain in domains:
|
||||
domain_list.append(domain.decode())
|
||||
return domain_list
|
||||
except Exception as e:
|
||||
logger.error(f'Error fetching allowed domains: {e}')
|
||||
return []
|
||||
|
||||
|
||||
def get_allowed_domains_for_time():
|
||||
"""获取当前允许的域名及其创建时间,按时间排序"""
|
||||
"""获取当前允许的域名及其创建时间"""
|
||||
try:
|
||||
domains = redis_client.lrange('allowed_domains_list', 0, -1)
|
||||
domain_info = {}
|
||||
for domain in domains:
|
||||
domain = domain.decode()
|
||||
created_at = redis_client.hget(f'domain_time:{domain}', 'timestamp')
|
||||
created_at = redis_client.hget(f'domain:{domain}', 'created_at')
|
||||
domain_info[domain] = created_at.decode() if created_at else None
|
||||
return domain_info
|
||||
except Exception as e:
|
||||
|
||||
Reference in New Issue
Block a user