python分散式鎖

在進行某些比較耗時的查詢時,為了避免進行重複計算,可以採用分佈式鎖定服務, 
在同一個時間只有一個操作在進行,同類的操作進行等待重試. 
下面的程式碼(fetch_with_dist_lock )定義了一個fetcher,一個updater. 
如果fetcher獲取不到數據,則使用updater進行更新.更新成功之後通過fetcher返回結果. 
也有一些情況,我們只想更新某個數據,更新者是多個,但是更新操作不是原子的.那麼 
我們會透過update_with_dist_lock來進行. 

def fetch_with_dist_lock(mc_store, mutex_key, fetcher, updater,
                        lock_time=3*60*1000, sleep_time=100, retry_times=3):
    i = 0
    while i < retry_times:
        i += 1
        need_update, results = fetcher()
        if need_update:
            if(mc_store.add(mutex_key, lock_time)):
                try:
                    updater()
                    continue
                finally:
                    #release the distribute mutex anyway
                    mc_store.delete(mutex_key)
            else:
                time.sleep((sleep_time*1.0)/1000)
                continue
        return results
    #too much tries, but failed still.
    return None
def f_wrapper(f, *args, **kwargs):
    def _():
        return f(*args, **kwargs)
    return _
def update_with_dist_lock(mc_store, mutex_key, updater, lock_time=60*1000, sleep_time=100, retry_times=5):
    i = 0
    while i < retry_times:
        i += 1
        if (mc_store.add(mutex_key, lock_time)):
            try:
                updater()
                return True
            finally:
                mc_store.delete(mutex_key)
        else:
            time.sleep((sleep_time*1.0)/1000)
            continue
    return False