From 4d9e3c5a13fdabf4191b813fa46291335b1d10f7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ladislav=20He=C5=99man?= <ladislav.herman@data3s.com>
Date: Tue, 9 Jun 2020 14:22:48 +0200
Subject: [PATCH] AsyncLocal storage

---
 build-support/tools/nant/bin/NAnt.exe.config  |  8 +--
 .../Threading/AsyncLocalStorage.cs            | 57 +++++++++++++++++++
 .../Threading/LogicalThreadContext.cs         |  3 +-
 3 files changed, 63 insertions(+), 5 deletions(-)
 create mode 100644 src/Spring/Spring.Core/Threading/AsyncLocalStorage.cs

diff --git a/build-support/tools/nant/bin/NAnt.exe.config b/build-support/tools/nant/bin/NAnt.exe.config
index 04f38c2d3..1054409b0 100644
--- a/build-support/tools/nant/bin/NAnt.exe.config
+++ b/build-support/tools/nant/bin/NAnt.exe.config
@@ -577,7 +577,7 @@
                             </strict>
                         </modes>
                     </runtime>
-                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFilesx86')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
+                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFiles')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
                         <include name="Accessibility.dll" />
                         <include name="Microsoft.Build.Conversion.v4.0.dll" />
                         <include name="Microsoft.Build.dll" />
@@ -683,7 +683,7 @@
                         <include name="System.Xml.Linq.dll" />
                     </reference-assemblies>
                     <!-- WPF Assemblies -->
-                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFilesx86')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
+                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFiles')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
                         <include name="NaturalLanguage6.dll" />
                         <include name="NlsData0009.dll" />
                         <include name="NlsLexicons0009.dll" />
@@ -711,7 +711,7 @@
                         <include name="wpfgfx_v0400.dll" />
                         <include name="wpftxt_v0400.dll" />
                     </reference-assemblies>
-                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFilesx86')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
+                    <reference-assemblies basedir="${environment::get-folder-path('ProgramFiles')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.0">
                         <include name="Microsoft.Build.Conversion.v4.0.dll" />
                         <include name="Microsoft.Build.dll" />
                         <include name="Microsoft.Build.Engine.dll" />
@@ -1045,7 +1045,7 @@
     <include name="wpfgfx_v0400.dll" />
     <include name="wpftxt_v0400.dll" />
   </reference-assemblies>
-  <reference-assemblies basedir="${environment::get-folder-path('ProgramFilesx86')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.5">
+  <reference-assemblies basedir="${environment::get-folder-path('ProgramFiles')}/Reference Assemblies/Microsoft/Framework/.NETFramework/v4.5">
     <include name="Microsoft.Build.Conversion.v4.0.dll" />
     <include name="Microsoft.Build.dll" />
     <include name="Microsoft.Build.Engine.dll" />
diff --git a/src/Spring/Spring.Core/Threading/AsyncLocalStorage.cs b/src/Spring/Spring.Core/Threading/AsyncLocalStorage.cs
new file mode 100644
index 000000000..b27b1932d
--- /dev/null
+++ b/src/Spring/Spring.Core/Threading/AsyncLocalStorage.cs
@@ -0,0 +1,57 @@
+using System;
+using System.Collections;
+using System.Threading;
+
+namespace Spring.Threading
+{
+    /// <summary>
+    /// Implements <see cref="IThreadStorage"/> by using a <see cref="ThreadStaticAttribute"/> hashtable.
+    /// </summary>
+    /// <author>Erich Eichinger</author>
+
+#if NETSTANDARD
+    public class AsyncLocalStorage : IThreadStorage
+    {
+
+        static AsyncLocal<Hashtable> data;
+
+        private static Hashtable Data
+        {
+            get
+            {
+                if (data == null) data = new AsyncLocal<Hashtable>();
+                return data.Value;
+            }
+        }
+
+        /// <summary>
+        /// Retrieves an object with the specified name.
+        /// </summary>
+        /// <param name="name">The name of the item.</param>
+        /// <returns>The object in the call context associated with the specified name or null if no object has been stored previously</returns>
+        public object GetData(string name)
+        {
+            return Data[name];
+        }
+
+        /// <summary>
+        /// Stores a given object and associates it with the specified name.
+        /// </summary>
+        /// <param name="name">The name with which to associate the new item.</param>
+        /// <param name="value">The object to store in the call context.</param>
+        public void SetData(string name, object value)
+        {
+            Data[name] = value;
+        }
+
+        /// <summary>
+        /// Empties a data slot with the specified name.
+        /// </summary>
+        /// <param name="name">The name of the data slot to empty.</param>
+        public void FreeNamedDataSlot(string name)
+        {
+            Data.Remove(name);
+        }
+    }
+#endif
+}
\ No newline at end of file
diff --git a/src/Spring/Spring.Core/Threading/LogicalThreadContext.cs b/src/Spring/Spring.Core/Threading/LogicalThreadContext.cs
index 0f80c4d7f..ba807c340 100644
--- a/src/Spring/Spring.Core/Threading/LogicalThreadContext.cs
+++ b/src/Spring/Spring.Core/Threading/LogicalThreadContext.cs
@@ -48,7 +48,8 @@ public sealed class LogicalThreadContext
         /// </remarks>
         private static IThreadStorage threadStorage =
 #if NETSTANDARD
-            new ThreadStaticStorage();
+            //new ThreadStaticStorage();
+            new AsyncLocalStorage();
 #else
             new CallContextStorage();
 #endif