大数据处理平台限时秒杀

import redis
import time

r = redis.Redis(host='localhost', port=6379, db=0)

def acquire_lock(lock_name, acquire_timeout=10):
    identifier = str(uuid.uuid4())
    end = time.time() + acquire_timeout
    while time.time() < end:
        if r.setnx(lock_name, identifier):
            return identifier
        time.sleep(0.001)
    return False

def release_lock(lock_name, identifier):
    with r.pipeline() as pipe:
        while True:
            try:
                pipe.watch(lock_name)
                if pipe.get(lock_name) == identifier:
                    pipe.multi()
                    pipe.delete(lock_name)
                    pipe.execute()
                    return True
                pipe.unwatch()
                break
            except redis.WatchError:
                pass
    return False

# 使用示例
lock_identifier = acquire_lock('stock_lock')
if lock_identifier:
    try:
        # 执行库存扣减操作
        stock = int(r.get('stock'))
        if stock > 0:
            r.decr('stock')
            print("库存扣减成功")
        else:
            print("库存不足")
    finally:
        release_lock('stock_lock', lock_identifier)