Merge pull request #841 from bollhals/patch-1

getting rid of a Func allocation

Author: lukebakken

projects/RabbitMQ.Client/client/impl/AsyncConsumerWorkService.cs

@@ -9,13 +9,20 @@ namespace RabbitMQ.Client.Impl
     internal sealed class AsyncConsumerWorkService : ConsumerWorkService
     {
         private readonly ConcurrentDictionary<IModel, WorkPool> _workPools = new ConcurrentDictionary<IModel, WorkPool>();
+        private readonly Func<IModel, WorkPool> _startNewWorkPoolFunc = model => StartNewWorkPool(model);
 
         public void Schedule<TWork>(ModelBase model, TWork work) where TWork : Work
         {
-            _workPools.GetOrAdd(model, StartNewWorkPool).Enqueue(work);
+            /*
+             * rabbitmq/rabbitmq-dotnet-client#841
+             * https://docs.microsoft.com/en-us/dotnet/api/system.collections.concurrent.concurrentdictionary-2.getoradd
+             * Note that the value delegate is not atomic but the Schedule method will not be called concurrently.
+             */
+            WorkPool workPool = _workPools.GetOrAdd(model, _startNewWorkPoolFunc);
+            workPool.Enqueue(work);
         }
 
-        private WorkPool StartNewWorkPool(IModel model)
+        private static WorkPool StartNewWorkPool(IModel model)
         {
             var newWorkPool = new WorkPool(model as ModelBase);
             newWorkPool.Start();

projects/RabbitMQ.Client/client/impl/ConsumerWorkService.cs

@@ -5,16 +5,23 @@
 
 namespace RabbitMQ.Client.Impl
 {
-    class ConsumerWorkService
+    internal class ConsumerWorkService
     {
         private readonly ConcurrentDictionary<IModel, WorkPool> _workPools = new ConcurrentDictionary<IModel, WorkPool>();
+        private readonly Func<IModel, WorkPool> _startNewWorkPoolFunc = model => StartNewWorkPool(model);
 
         public void AddWork(IModel model, Action fn)
         {
-            _workPools.GetOrAdd(model, StartNewWorkPool).Enqueue(fn);
+            /*
+             * rabbitmq/rabbitmq-dotnet-client#841
+             * https://docs.microsoft.com/en-us/dotnet/api/system.collections.concurrent.concurrentdictionary-2.getoradd
+             * Note that the value delegate is not atomic but the AddWork method will not be called concurrently.
+             */
+            WorkPool workPool = _workPools.GetOrAdd(model, _startNewWorkPoolFunc);
+            workPool.Enqueue(fn);
         }
 
-        private WorkPool StartNewWorkPool(IModel model)
+        private static WorkPool StartNewWorkPool(IModel model)
         {
             var newWorkPool = new WorkPool();
             newWorkPool.Start();