Non-generic StringComparer conversion for dict and hashset

src/libraries/System.Private.CoreLib/src/System/Collections/Generic/Dictionary.cs

@@ -54,7 +54,7 @@ public Dictionary(int capacity, IEqualityComparer<TKey>? comparer)
                 Initialize(capacity);
             }
 
-            if (comparer != null && comparer != EqualityComparer<TKey>.Default) // first check for null to avoid forcing default comparer instantiation unnecessarily
+            if (comparer is not null && comparer != EqualityComparer<TKey>.Default) // first check for null to avoid forcing default comparer instantiation unnecessarily
             {
                 _comparer = comparer;
             }
@@ -65,18 +65,7 @@ public Dictionary(int capacity, IEqualityComparer<TKey>? comparer)
 
             if (typeof(TKey) == typeof(string))
             {
-                if (_comparer is null)
-                {
-                    _comparer = (IEqualityComparer<TKey>)NonRandomizedStringEqualityComparer.WrappedAroundDefaultComparer;
-                }
-                else if (ReferenceEquals(_comparer, StringComparer.Ordinal))
-                {
-                    _comparer = (IEqualityComparer<TKey>)NonRandomizedStringEqualityComparer.WrappedAroundStringComparerOrdinal;
-                }
-                else if (ReferenceEquals(_comparer, StringComparer.OrdinalIgnoreCase))
-                {
-                    _comparer = (IEqualityComparer<TKey>)NonRandomizedStringEqualityComparer.WrappedAroundStringComparerOrdinalIgnoreCase;
-                }
+                _comparer = (IEqualityComparer<TKey>?)NonRandomizedStringEqualityComparer.GetStringComparer((IEqualityComparer<string>?)_comparer);
             }
         }
 

src/libraries/System.Private.CoreLib/src/System/Collections/Generic/HashSet.cs

@@ -54,7 +54,7 @@ public HashSet() : this((IEqualityComparer<T>?)null) { }
 
         public HashSet(IEqualityComparer<T>? comparer)
         {
-            if (comparer != null && comparer != EqualityComparer<T>.Default) // first check for null to avoid forcing default comparer instantiation unnecessarily
+            if (comparer is not null && comparer != EqualityComparer<T>.Default) // first check for null to avoid forcing default comparer instantiation unnecessarily
             {
                 _comparer = comparer;
             }
@@ -65,18 +65,7 @@ public HashSet(IEqualityComparer<T>? comparer)
 
             if (typeof(T) == typeof(string))
             {
-                if (_comparer is null)
-                {
-                    _comparer = (IEqualityComparer<T>)NonRandomizedStringEqualityComparer.WrappedAroundDefaultComparer;
-                }
-                else if (ReferenceEquals(_comparer, StringComparer.Ordinal))
-                {
-                    _comparer = (IEqualityComparer<T>)NonRandomizedStringEqualityComparer.WrappedAroundStringComparerOrdinal;
-                }
-                else if (ReferenceEquals(_comparer, StringComparer.OrdinalIgnoreCase))
-                {
-                    _comparer = (IEqualityComparer<T>)NonRandomizedStringEqualityComparer.WrappedAroundStringComparerOrdinalIgnoreCase;
-                }
+                _comparer = (IEqualityComparer<T>?)NonRandomizedStringEqualityComparer.GetStringComparer((IEqualityComparer<string>?)_comparer);
             }
         }
 

src/libraries/System.Private.CoreLib/src/System/Collections/Generic/NonRandomizedStringEqualityComparer.cs

@@ -2,9 +2,7 @@
 // The .NET Foundation licenses this file to you under the MIT license.
 
 using System.Diagnostics;
-using System.Globalization;
 using System.Runtime.Serialization;
-using Internal.Runtime.CompilerServices;
 
 namespace System.Collections.Generic
 {
@@ -16,13 +14,7 @@ namespace System.Collections.Generic
     // Needs to be public to support binary serialization compatibility
     public class NonRandomizedStringEqualityComparer : IEqualityComparer<string?>, IInternalStringEqualityComparer, ISerializable
     {
-        // Dictionary<...>.Comparer and similar methods need to return the original IEqualityComparer
-        // that was passed in to the ctor. The caller chooses one of these singletons so that the
-        // GetUnderlyingEqualityComparer method can return the correct value.
-
-        internal static readonly NonRandomizedStringEqualityComparer WrappedAroundDefaultComparer = new OrdinalComparer(EqualityComparer<string?>.Default);
-        internal static readonly NonRandomizedStringEqualityComparer WrappedAroundStringComparerOrdinal = new OrdinalComparer(StringComparer.Ordinal);
-        internal static readonly NonRandomizedStringEqualityComparer WrappedAroundStringComparerOrdinalIgnoreCase = new OrdinalIgnoreCaseComparer(StringComparer.OrdinalIgnoreCase);
+        private static readonly StringComparers s_stringComparers = new StringComparers();
 
         private readonly IEqualityComparer<string?> _underlyingComparer;
 
@@ -109,5 +101,37 @@ internal override RandomizedStringEqualityComparer GetRandomizedEqualityComparer
                 return RandomizedStringEqualityComparer.Create(_underlyingComparer, ignoreCase: true);
             }
         }
+
+        public static IEqualityComparer<string> GetStringComparer(IEqualityComparer<string>? comparer)
+        {
+            // Special-case EqualityComparer<string>.Default, StringComparer.Ordinal, and StringComparer.OrdinalIgnoreCase.
+            // We use a non-randomized comparer for improved perf, falling back to a randomized comparer if the
+            // hash buckets become unbalanced.
+            StringComparers stringComparers = s_stringComparers;
+            if (comparer is null)
+            {
+                return stringComparers.WrappedAroundDefaultComparer;
+            }
+            else if (ReferenceEquals(comparer, StringComparer.Ordinal))
+            {
+                return stringComparers.WrappedAroundStringComparerOrdinal;
+            }
+            else if (ReferenceEquals(comparer, StringComparer.OrdinalIgnoreCase))
+            {
+                return stringComparers.WrappedAroundStringComparerOrdinalIgnoreCase;
+            }
+
+            return comparer;
+        }
+
+        private class StringComparers
+        {
+            // Dictionary<...>.Comparer and similar methods need to return the original IEqualityComparer
+            // that was passed in to the ctor. The caller chooses one of these singletons so that the
+            // GetUnderlyingEqualityComparer method can return the correct value.
+            public readonly NonRandomizedStringEqualityComparer WrappedAroundDefaultComparer = new OrdinalComparer(EqualityComparer<string?>.Default);
+            public readonly NonRandomizedStringEqualityComparer WrappedAroundStringComparerOrdinal = new OrdinalComparer(StringComparer.Ordinal);
+            public readonly NonRandomizedStringEqualityComparer WrappedAroundStringComparerOrdinalIgnoreCase = new OrdinalIgnoreCaseComparer(StringComparer.OrdinalIgnoreCase);
+        }
     }
 }