课程 8：PySpark 与 Redis 入门

import redis

# 连接到 Redis (确保 Redis 服务器正在运行)
try:
    r = redis.Redis(host='localhost', port=6379, db=0, decode_responses=True) # decode_responses=True 以获取字符串而非字节
    r.ping() # 检查连接
    print("成功连接到 Redis！") # "Successfully connected to Redis!"
except redis.exceptions.ConnectionError as e:
    print(f"无法连接到 Redis: {e}") # "Could not connect to Redis:"
    exit()

# 1. 字符串操作
print("\\n--- 字符串操作 ---")
r.set('user:1000:name', 'Alice Wonderland')
user_name = r.get('user:1000:name')
print(f"GET user:1000:name -> {user_name}")

r.set('counter', 10)
r.incr('counter') # 将值增加 1
r.incrby('counter', 5) # 将值增加 5
print(f"GET counter -> {r.get('counter')}")


# 2. 列表操作 (例如，用于简单队列)
print("\\n--- 列表操作 ---")
list_key = 'task_queue'
r.delete(list_key) # 清空列表以重新开始

r.lpush(list_key, 'task1', 'task2', 'task3') # 推入左侧 (头部)
print(f"LPUSH 到 {list_key}: task1, task2, task3")

r.rpush(list_key, 'task0') # 推入右侧 (尾部)
print(f"RPUSH 到 {list_key}: task0")

tasks = r.lrange(list_key, 0, -1) # 获取列表中的所有元素
print(f"LRANGE {list_key} (全部) -> {tasks}")

task = r.rpop(list_key) # 从右侧弹出 (如果是 LPUSH 推入则为 FIFO，如果是 RPUSH 推入则为 LIFO)
print(f"RPOP 从 {list_key} -> {task}")
print(f"LRANGE {list_key} (RPOP 之后) -> {r.lrange(list_key, 0, -1)}")


# 3. 哈希操作 (例如，用于存储对象属性)
print("\\n--- 哈希操作 ---")
hash_key = 'user:1001'
r.delete(hash_key)

r.hset(hash_key, mapping={ # 在新版 redis-py 中，hset 可以直接接受 mapping
    'name': 'Bob The Builder',
    'age': 40,
    'occupation': 'Builder'
})
print(f"HSET 操作 {hash_key}")

user_name_hash = r.hget(hash_key, 'name')
print(f"HGET {hash_key} 字段 'name' -> {user_name_hash}")

user_info = r.hgetall(hash_key) # 获取哈希中的所有字段和值
print(f"HGETALL {hash_key} -> {user_info}")


# 4. 集合操作 (例如，用于唯一项)
print("\\n--- 集合操作 ---")
set_key = "unique_visitors_today"
r.delete(set_key)
r.sadd(set_key, "user:A", "user:B", "user:C", "user:A") # "user:A" 只添加一次
print(f"SADD 到 {set_key}: user:A, user:B, user:C, user:A")
print(f"SMEMBERS {set_key} -> {r.smembers(set_key)}")
print(f"SISMEMBER {set_key} 'user:B' -> {r.sismember(set_key, 'user:B')}") # 检查成员是否存在
print(f"SCARD {set_key} (基数/数量) -> {r.scard(set_key)}")

# 清理一些键 (可选)
# r.delete('user:1000:name', 'counter', list_key, hash_key, set_key)