对资源加读写锁的容器

之前写了一篇《对不能用using的成对操作,快速扩展IDisposable的方法》提到了如何快速的把销毁操作用闭包的形式封装为IDisposable,并且实现了一个ReaderWriteerLockSlimHelper。

 

对于没有using的RWLock似乎已经很好用了, 但是我仍嫌弃其不够简单。

 

资源读写锁的应用,绝大多数是针对某批特定的资源。如果为这样的资源做一个封装,资源的引用需要被隐藏在封装内,封装外不通过读锁不可以访问资源。显然ReaderWriteerLockSlimHelper却无法做到这一点。

 

我们在上文中已经建立了一个 Disposable对象,取得该对象意味着我们可以加锁,将其销毁我们就可以解锁,其生存期与整个加锁周期吻合。对于读写锁来说具有这种行为的对象是一种作为读写钥匙/令牌的存在。解锁后的资源引用放在这个令牌上作为属性暴露出来再合适不过了。

以此为考量,我们先设计一个实现IDisposable,读写锁的令牌

/// <summary>
/// 读写资源的令牌
/// </summary>
public abstract class LockToken<T> : IDisposable
{
protected LockableObjectContainer<T> _container;

/// <summary>
/// 被锁住的资源引用
/// </summary>
public T Value { get; set; }
/// <summary>
/// 读写锁的原钥匙,对其销毁即为解锁
/// </summary>
protected IDisposable _innerTicket;


/// <summary>
/// 创建读写资源的令牌
/// </summary>
/// <param name="value">解锁的资源引用</param>
/// <param name="innerTicket">锁的真实引用</param>
/// <param name="beforeDispose">在解锁前需要做的操作 如把资源的新值覆盖回锁住的资源</param>
/// <param name="afterDispose">在解锁后需要做的操作 如写日志</param>
internal LockToken(LockableObjectContainer<T> container, IDisposable innerTicket, Action<LockToken<T>> beforeDispose, Action<LockToken<T>> afterDispose)
{
_container=container;
_innerTicket
= innerTicket;
Value
= container._value;
this._disposeAction =
()
=>
{
try
{

if (beforeDispose != null) beforeDispose(this);
this.Value = default(T);
_innerTicket.Dispose();
if (afterDispose != null) afterDispose(this);
}
catch
{


}
};
}

#region IDisposable Members

/// <summary>
/// 令牌销毁时要做的操作
/// </summary>
Action _disposeAction;

/// <summary>
/// 销毁
/// </summary>
public void Dispose()
{
_disposeAction();
}

#endregion
}

读写时,所做的操作略有不同。写令牌在销毁时要把令牌上的新引用/值 覆盖到容器内

依此我们可以做两个不同的令牌子类

 

class ReadLockToken<T> : LockToken<T>
{
public ReadLockToken(LockableObjectContainer<T> container, ReaderWriterLockSlim _lock) :
base(container, _lock.CreateLockScope(LockType.Read), null, null)
{
}

}



class WriteLockToken<T> : LockToken<T>
{
public WriteLockToken(LockableObjectContainer<T> container, ReaderWriterLockSlim _lock)
:
base
(
container,
_lock.CreateLockScope(LockType.Write),
lt
=>
{
//在解锁前 把新值保存给原对象
lt._container._value =lt.Value;
           },
          null
)
{
}

}

这些做好后,编写容器就很容易了

 

 

 

public class LockableObjectContainer<T>
{
internal protected T _value;


System.Threading.ReaderWriterLockSlim _lock
= new
System.Threading.ReaderWriterLockSlim();
public
LockableObjectContainer(T value)
{
_value
=
value;

}

public LockToken<T>
GetReadToken()
{
return new ReadLockToken<T>(this
, _lock);
}
public LockToken<T>
GetWriteToken()
{
return new WriteLockToken<T>(this
, _lock);

}

}

 

 

上面的内容很枯燥

写读写锁的时候很方便

 

static LockableObjectContainer <Dictionary <Type ,IFactoryContainer > > planConnectionGetters
= new LockableObjectContainer<Dictionary<Type, IFactoryContainer>>
(
new Dictionary<Type, IFactoryContainer>());


。。。。。。。。。。。。。。。。。。。。。。。。。



using (var token =planConnectionGetters.GetReadToken())
{
var dic
= token.Value;

if (dic.TryGetValue(contractType , out channelFactory ))
{
return channelFactory.GetChannel (uri);

}

}


using (var token = planConnectionGetters.GetWriteToken ())
{
var dic
= token.Value;

if (!dic.TryGetValue(contractType, out channelFactory))
{
Type t
= typeof(FactoryContainer<>);
channelFactory
= Activator.CreateInstance(t.MakeGenericType(contractType)) as IFactoryContainer;
dic.Add(contractType, channelFactory);

}



}
 如果容器经常保存字典等常用集合对象 我们也可以这样做一些扩展方法

public static TValue GetOrCreateValue<TKey, TValue>
(
this LockableObjectContainer<IDictionary<TKey, TValue>> container,
TKey key,
Predicate
<TValue > valueChecker,
Func
<TValue> valueFactory
)
{
TValue val
=default (TValue);
using (var token = container.GetReadToken())
{

var dic
= token.Value;
if (dic.TryGetValue(key, out val))
{
if (valueChecker (val))
return val;

}
}

using (var token = container.GetWriteToken())
{
var dic
= token.Value;
if (dic.TryGetValue(key, out val))
{
if (valueChecker(val))
return val;

}
val
=valueFactory();
dic.Add(key,val) ;

}
return val;

}




------------------------------------------------------------

TQueue q;
q
= instanceQueues.GetOrCreateValue(instance, _ => true, () => new TQueue());
return q;

posted @ 2011-03-24 14:57  MSFT:waywa 韦恩卑鄙  阅读(2229)  评论(3编辑  收藏  举报