StackExchange.Redis - LockTake / LockRelease用法

时间:2014-08-04 20:41:37

标签: redis stackexchange stackexchange.redis

我正在使用Redis和StackExchange.Redis。我有多个线程,在某些时候会访问和编辑相同键的值,所以我需要同步数据的操作。

查看可用的函数,我看到有两个函数,TakeLock和ReleaseLock。但是,这些函数同时使用键和值参数,而不是锁定预期的单个键。关于GitHub的intellisene文档和源代码没有解释如何使用LockTake和LockRelease函数或传入的键和值参数。

问:StackExchange.Redis中LockTake和LockRelease的正确用法是什么?

我打算做的伪代码示例:

//Add Items Before Parallel Execution
redis.StringSet("myJSONKey", myJSON);

//Parallel Execution
Parallel.For(0, 100, i =>
    {
        //Some work here
        //....

        //Lock
        redis.LockTake("myJSONKey");

        //Manipulate
        var myJSONObject = redis.StringGet("myJSONKey");
        myJSONObject.Total++;
        Console.WriteLine(myJSONObject.Total);
        redis.StringSet("myJSONKey", myNewJSON);

        //Unlock
        redis.LockRelease("myJSONKey");

        //More work here
        //...
    });

2 个答案:

答案 0 :(得分:40)

锁有三个部分:

  • 密钥(数据库中锁的唯一名称)
  • 该值(一个调用者定义的标记,可用于指示谁拥有"锁定,并检查释放和扩展锁定是否正确完成)
  • 持续时间(锁意图是有限持续时间)

如果没有其他任何价值,guid可能会产生一个合适的价值"。我们倾向于使用机器名称(如果多个进程可以在同一台机器上竞争,则使用机器名称的munged版本。)

另外,请注意,锁定是推测,而不是阻止。完全有可能失败获取锁定,因此您可能需要对此进行测试,并可能添加一些重试逻辑。

一个典型的例子可能是:

RedisValue token = Environment.MachineName;
if(db.LockTake(key, token, duration)) {
    try {
        // you have the lock do work
    } finally {
        db.LockRelease(key, token);
    }
}

请注意,如果工作很长(尤其是循环),您可能希望在中间添加一些偶尔的LockExtend次呼叫 - 再次记住检查是否成功(如果它超时)。< / p>

另请注意,所有个别redis命令都是原子的,因此您不必担心两个谨慎的操作竞争。对于更复杂的多操作单元,事务脚本是选项。

答案 1 :(得分:2)

我的部分代码是lock-&gt; get-&gt; modify(如果需要) - &gt;解锁带注释的操作。

    public static T GetCachedAndModifyWithLock<T>(string key, Func<T> retrieveDataFunc, TimeSpan timeExpiration, Func<T, bool> modifyEntityFunc,
       TimeSpan? lockTimeout = null, bool isSlidingExpiration=false) where T : class
    {

        int lockCounter = 0;//for logging in case when too many locks per key
        Exception logException = null;

        var cache = Connection.GetDatabase();
        var lockToken = Guid.NewGuid().ToString(); //unique token for current part of code
        var lockName = key + "_lock"; //unique lock name. key-relative.
        T tResult = null;

        while ( lockCounter < 20)
        {
            //check for access to cache object, trying to lock it
            if (!cache.LockTake(lockName, lockToken, lockTimeout ?? TimeSpan.FromSeconds(10)))
            {
                lockCounter++;
                Thread.Sleep(100); //sleep for 100 milliseconds for next lock try. you can play with that
                continue;
            }

            try
            {
                RedisValue result = RedisValue.Null;

                if (isSlidingExpiration)
                {
                    //in case of sliding expiration - get object with expiry time
                    var exp = cache.StringGetWithExpiry(key);

                    //check ttl.
                    if (exp.Expiry.HasValue && exp.Expiry.Value.TotalSeconds >= 0)
                    {
                        //get only if not expired
                        result = exp.Value;
                    }
                }
                else //in absolute expiration case simply get
                {
                    result = cache.StringGet(key);
                }

                //"REDIS_NULL" is for cases when our retrieveDataFunc function returning null (we cannot store null in redis, but can store pre-defined string :) )
                if (result.HasValue && result == "REDIS_NULL") return null;
                //in case when cache is epmty
                if (!result.HasValue)
                {
                    //retrieving data from caller function (from db from example)
                    tResult = retrieveDataFunc();

                    if (tResult != null)
                    {
                        //trying to modify that entity. if caller modifyEntityFunc returns true, it means that caller wants to resave modified entity.
                        if (modifyEntityFunc(tResult))
                        {
                            //json serialization
                            var json = JsonConvert.SerializeObject(tResult);
                            cache.StringSet(key, json, timeExpiration);
                        }
                    }
                    else
                    {
                        //save pre-defined string in case if source-value is null.
                        cache.StringSet(key, "REDIS_NULL", timeExpiration);
                    }
                }
                else
                {
                    //retrieve from cache and serialize to required object
                    tResult = JsonConvert.DeserializeObject<T>(result);
                    //trying to modify
                    if (modifyEntityFunc(tResult))
                    {
                        //and save if required
                        var json = JsonConvert.SerializeObject(tResult);
                        cache.StringSet(key, json,  timeExpiration);
                    }
                }

                //refresh exiration in case of sliding expiration flag
                if(isSlidingExpiration)
                    cache.KeyExpire(key, timeExpiration);
            }
            catch (Exception ex)
            {
                logException = ex;
            }
            finally
            {                    
                cache.LockRelease(lockName, lockToken);
            }
            break;
        }

        if (lockCounter >= 20 || logException!=null)
        {
            //log it
        }

        return tResult;
    }

和用法:

public class User
{
    public int ViewCount { get; set; }
}

var cachedAndModifiedItem = GetCachedAndModifyWithLock<User>( "MyAwesomeKey", () =>
        {
            //return from db or kind of that
            return new User() { ViewCount = 0 };
        }, TimeSpan.FromMinutes(10), user=>
        {
            if (user.ViewCount< 3)
            {
                user.ViewCount++;
                return true; //save it to cache
            }
            return false; //do not update it in cache
        }, TimeSpan.FromSeconds(10),true);

该代码可以改进(例如,您可以为缓存等减少计数调用添加事务),但我很高兴它对您有所帮助。