src/RoslynAnalyzers/Utilities/Compiler/PooledObjects/PooledDictionary.cs - main - githubmirror/roslyn

src/RoslynAnalyzers/Utilities/Compiler/PooledObjects/PooledDictionary.cs
87 строк3 KB
Jan Jones
Pool validation: add more tracking (#83799)
24 июн 2026, 12:36
24 июн 2026, 12:3628c44d1
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Diagnostics;

#if DEBUG
using System.Runtime.CompilerServices;
#endif

#pragma warning disable CA1000 // Do not declare static members on generic types

namespace Microsoft.CodeAnalysis.PooledObjects
{
    // Dictionary that can be recycled via an object pool
    // NOTE: these dictionaries always have the default comparer.
    internal sealed partial class PooledDictionary<K, V>
    {
        public ImmutableDictionary<TKey, TValue> ToImmutableDictionaryAndFree<TKey, TValue>(
           Func<KeyValuePair<K, V>, TKey> keySelector, Func<KeyValuePair<K, V>, TValue> elementSelector, IEqualityComparer<TKey> comparer)
            where TKey : notnull
        {
            ImmutableDictionary<TKey, TValue> result;
            if (Count == 0)
            {
                result = ImmutableDictionary<TKey, TValue>.Empty;
            }
            else
            {
                result = this.ToImmutableDictionary(keySelector, elementSelector, comparer);
                this.Clear();
            }

            _pool?.Free(this);
            return result;
        }

        private static readonly ConcurrentDictionary<IEqualityComparer<K>, ObjectPool<PooledDictionary<K, V>>> s_poolInstancesByComparer
            = new();

        public static PooledDictionary<K, V> GetInstance(
            IEqualityComparer<K>? keyComparer
#if DEBUG
            , [CallerFilePath] string filePath = ""
            , [CallerLineNumber] int lineNumber = 0
#endif
            )
        {
            var pool = keyComparer == null ?
                s_poolInstance :
                s_poolInstancesByComparer.GetOrAdd(keyComparer, CreatePool);
            var instance = pool.Allocate(
#if DEBUG
                filePath, lineNumber
#endif
                );
            Debug.Assert(instance.Count == 0);
            return instance;
        }

        public static PooledDictionary<K, V> GetInstance(
            IEnumerable<KeyValuePair<K, V>> initializer,
            IEqualityComparer<K>? keyComparer = null
#if DEBUG
            , [CallerFilePath] string filePath = ""
            , [CallerLineNumber] int lineNumber = 0
#endif
            )
        {
            var instance = GetInstance(keyComparer
#if DEBUG
                , filePath, lineNumber
#endif
                );
            foreach (var kvp in initializer)
            {
                instance.Add(kvp.Key, kvp.Value);
            }

            return instance;
        }
    }
}