我正在使用ConcurrentDictionary通过并行访问来缓存数据,有时新项目可以存储在db中,并且它们不会加载到缓存中.这就是我使用GetOrAdd的原因
public User GetUser(int userId)
{
return _user.GetOrAdd(userId,GetUserFromDb);
}
private User GetUserFromDb(int userId)
{
var user = _unitOfWork.UserRepository.GetById(userId);
// if user is null,it is stored to dictionary
return user;
}
但是,如果用户不为空,我如何检查用户是否从db获取并将用户存储到字典？
可能我可以在GetOrAdd之后立即从ConcurrentDictionary中删除null但它看起来不是线程安全的并且它不是非常优雅的解决方案.无用插入和从字典中删除.你知道怎么做吗？
解决方法
public User GetUser(int userId)
{
var user = _user.GetOrAdd(userId,GetUserFromDb);
if (user == null) _user.TryRemove(userId,out user);
}
您还可以将其包装到扩展方法中：
public static TValue GetOrAddIfNotNull<TKey,TValue>(
this ConcurrentDictionary<TKey,TValue> dictionary,TKey key,Func<TKey,TValue> valueFactory) where TValue : class
{
var value = dictionary.GetOrAdd(key,valueFactory);
if (value == null) dictionary.TryRemove(key,out value);
return value;
}
然后你的代码看起来像：
public User GetUser(int userId)
{
var user = _user.GetOrAddIfNotNull(userId,GetUserFromDb)
}
UPDATE
根据@usr评论,可能有以下情况：
>线程1执行GetOrAdd,将null添加到字典并暂停.
>用户已添加到数据库中.
>线程2执行GetOrAdd并从字典中检索null而不是命中数据库.
>线程1和线程2执行TryRemove并从字典中删除记录.
有了这个时间,线程2将变为null而不是命中数据库并获取用户记录.如果这个边缘情况对您很重要并且您仍然想使用ConcurrentDictionary,那么您可以在扩展方法中使用lock：
public static class ConcurrentDictionaryExtensions
{
private static readonly object myLock = new object();
public static TValue GetOrAddIfNotNull<TKey,TValue>(
this ConcurrentDictionary<TKey,TValue> valueFactory) where TValue : class
{
lock (myLock)
{
var value = dictionary.GetOrAdd(key,valueFactory);
if (value == null) dictionary.TryRemove(key,out value);
return value;
}
}
}
