企业云盘集群搭建

from flask import Flask, request
import redis
import time

app = Flask(__name__)
redis_client = redis.StrictRedis(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 redis_client.setnx(lock_name, identifier):
            return identifier
        time.sleep(0.001)
    return False

def release_lock(lock_name, identifier):
    with redis_client.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

@app.route('/lock', methods=['POST'])
def lock():
    lock_name = request.json.get('lock_name')
    identifier = acquire_lock(lock_name)
    if identifier:
        return {'status': 'acquired', 'identifier': identifier}
    else:
        return {'status': 'failed', 'message': 'Could not acquire lock'}, 503

@app.route('/unlock', methods=['POST'])
def unlock():
    lock_name = request.json.get('lock_name')
    identifier = request.json.get('identifier')
    if release_lock(lock_name, identifier):
        return {'status': 'released'}
    else:
        return {'status': 'failed', 'message': 'Could not release lock'}, 503

if __name__ == '__main__':
    app.run(debug=True)