我正在使用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;
- }
- }
- }