基于redis实现【最热搜索】和【最近搜索】功能

import redis
 
# 连接到Redis
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
 
def record_search(user_id, search_term, timestamp, max_searches=10):
    """记录用户搜索并在Redis中更新相关列表"""
    # 使用用户ID和搜索词作为key来记录搜索
    user_search_key = f"user_search:{user_id}:{search_term}"
    # 更新用户搜索计数
    redis_client.zincrby(f"searches", 1, user_search_key)
    # 限制列表长度，只保留最新的max_searches次搜索
    redis_client.zremrangebyrank(f"searches", 0, -max_searches - 1)
 
    # 存储最新的搜索以便于检索
    redis_client.lpush(f"recent_searches:{user_id}", search_term)
    redis_client.ltrim(f"recent_searches:{user_id}", 0, max_searches - 1)
 
    # 存储所有用户的最新搜索以便于检索
    redis_client.lpush("recent_searches", search_term)
    redis_client.ltrim("recent_searches", 0, max_searches - 1)
 
def get_top_searches(max_results=10):
    """获取最热搜索列表"""
    return redis_client.zrevrange("searches", 0, max_results - 1)
 
def get_user_recent_searches(user_id, max_results=10):
    """获取用户最近搜索列表"""
    return redis_client.lrange(f"recent_searches:{user_id}", 0, max_results - 1)
 
def get_global_recent_searches(max_results=10):
    """获取全局最近搜索列表"""
    return redis_client.lrange("recent_searches", 0, max_results - 1)
 
# 示例使用
record_search("user123", "python", 1610000000)
record_search("user456", "java", 1610000000)
record_search("user123", "python", 1610003000)
 
top_searches = get_top_searches()
print("Top Searches:", top_searches)
 
user_recent_searches = get_user_recent_searches("user123")
print("User Recent Searches:", user_recent_searches)
 
global_recent_searches = get_global_recent_searches()
print("Global Recent Searches:", global_recent_searches)

这段代码使用了Redis的有序集合（sorted set）和列表（list）来分别存储最热搜索和最新搜索。record_search函数用于记录用户的搜索并更新相关的有序集合和列表。get_top_searches和get_user_recent_searches函数用于获取最热搜索和用户最新搜索的列表。