好久不见,马甲哥封闭居家半个月,记录之前遇到的一件小事。
ConcurrentDictionary<TKey,TValue>绝大部分api都是线程安全且原子性的,
唯二的例外是接收工厂委托的api:AddOrUpdate
、GetOrAdd
,这两个api不是原子性的,需要引起重视。
之前有个同事就因为这个case背了一个P。
AddOrUpdate(TKey, TValue, Func<TKey,TValue,TValue> valueFactory);
GetOrAdd(TKey key, Func<TKey, TValue> valueFactory);
(注意,包括其他接收工厂委托的重载函数)
Q1: valueFactory工厂函数不在锁定范围,为什么不在锁范围?
A: 还不是因为微软不相信你能写出健壮的业务代码,未知的业务代码可能造成死锁。
Q2:带来的效果?
- valueFactory工厂函数可能会多次执行
- 虽然会多次执行, 但插入的值永远是一个,插入的值取决于哪个线程率先插入字典。
示例代码:
using System.Collections.Concurrent;
public class Program
{
private static int _runCount = 0;
private static readonly ConcurrentDictionary<string, string> _dictionary
= new ConcurrentDictionary<string, string>();
public static void Main(string[] args)
{
var task1 = Task.Run(() => PrintValue("The first value"));
var task2 = Task.Run(() => PrintValue("The second value"));
var task3 = Task.Run(() => PrintValue("The three value"));
var task4 = Task.Run(() => PrintValue("The four value"));
Task.WaitAll(task1, task2, task4,task4);
PrintValue("The five value");
Console.WriteLine($"Run count: {_runCount}");
}
public static void PrintValue(string valueToPrint)
{
var valueFound = _dictionary.GetOrAdd("key",
x =>
{
Interlocked.Increment(ref _runCount);
Thread.Sleep(100);
return valueToPrint;
});
Console.WriteLine(valueFound);
}
}
上面4个线程并发插入字典,每次随机输出,_runCount=4
显示工厂类执行4次。
Q4:如果工厂产值的代价很大,不允许多次创建,如何实现?
A: 有一个trick能解决这个问题: valueFactory工厂函数返回Lazy.
using System.Collections.Concurrent;
public class Program
{
private static int _runCount2 = 0;
private static readonly ConcurrentDictionary<string, Lazy<string>> _lazyDictionary
= new ConcurrentDictionary<string, Lazy<string>>();
public static void Main(string[] args)
{
task1 = Task.Run(() => PrintValueLazy("The first value"));
task2 = Task.Run(() => PrintValueLazy("The second value"));
task3 = Task.Run(() => PrintValueLazy("The three value"));
task4 = Task.Run(() => PrintValueLazy("The four value"));
Task.WaitAll(task1, task2, task4, task4);
PrintValue("The five value");
Console.WriteLine($"Run count: {_runCount2}");
}
public static void PrintValueLazy(string valueToPrint)
{
var valueFound = _lazyDictionary.GetOrAdd("key",
x => new Lazy<string>(
() =>
{
Interlocked.Increment(ref _runCount2);
Thread.Sleep(100);
return valueToPrint;
}));
Console.WriteLine(valueFound.Value);
}
}
上面示例,依旧会稳定随机输出,但是_runOut=1
表明产值动作只执行了一次、
valueFactory工厂函数返回Lazy
① 工厂函数依旧没进入锁定过程,会多次执行;
② 与最上面的例子类似,只会插入一个Lazy容器(后续线程依旧做double check发现字典key已经有Lazy容器了,会放弃插入);
③ 线程执行Lazy
④ 多个线程尝试执行Lazy
public Lazy(Func<T> valueFactory)
:this(valueFactory, LazyThreadSafetyMode.ExecutionAndPublication, useDefaultConstructor: false)
{
}
控制构造函数执行的枚举值描述ExecutionAndPublication能确保只有一个线程能够以线程安全方式执行构造函数None线程不安全Publication并发线程都会执行初始化函数,以先完成初始化的值为准总结
为解决ConcurrentDictionary GetOrAdd(key, valueFactory) 工厂函数在并发场景下被多次执行的问题。
① valueFactory工厂函数产生Lazy容器
② 将Lazy容器的值初始化姿势设定为ExecutionAndPublication
(线程安全且执行一次)。
两姿势缺一不可。