对资源加读写锁的容器
之前写了一篇《对不能用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;