AssemblyLoadContextManager.cs

Go to the documentation of this file.

#if NET5_0_OR_GREATER
//-----------------------------------------------------------------------
// <copyright file="AssemblyLoadContextManager.cs" company="Marimer LLC">
//     Copyright (c) Marimer LLC. All rights reserved.
//     Website: https://cslanet.com
// </copyright>
// <summary>Provides common access to "AssemblyLoadContext" logic</summary>
//-----------------------------------------------------------------------
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Runtime.Loader;
 
namespace Csla.Runtime
{
    public static class AssemblyLoadContextManager
  {
        public static Tuple<string, TValue> CreateCacheInstance<TValue>(
            Type objectType,
            TValue cachingItem,
            Action<AssemblyLoadContext> unloadAction,
            bool excludeNonCollectible = false)
        {
            if (objectType == null)
            {
                throw new ArgumentNullException(nameof(objectType));
            }
 
            if (cachingItem == null)
            {
              throw new ArgumentNullException(nameof(cachingItem));
            }
 
            if (unloadAction == null)
            {
              throw new ArgumentNullException(nameof(Action<AssemblyLoadContext>));
            }
 
            string assemblyLoadContextName = null;
 
            if ((!excludeNonCollectible || objectType.Assembly.IsCollectible)
                && AssemblyLoadContext.CurrentContextualReflectionContext != null
                && AssemblyLoadContext.CurrentContextualReflectionContext.Name != AssemblyLoadContext.Default.Name)
            {
                assemblyLoadContextName = AssemblyLoadContext.CurrentContextualReflectionContext.Name;
 
                AssemblyLoadContext.CurrentContextualReflectionContext.Unloading += unloadAction;
            }
 
            return new Tuple<string, TValue>(assemblyLoadContextName, cachingItem);
        }
 
        public static void RemoveFromCache<TKey, TValue>(
          IDictionary<TKey, Tuple<string, TValue>> dictionary,
          AssemblyLoadContext context,
          bool usingConcurrentDictionary = false)
        {
            if (dictionary == null)
            {
                throw new ArgumentNullException("IDictionary<T, Tuple<string, TC>>");
            }
 
            if (context == null)
            {
                return;
            }
 
            var obsoleteCacheKeys = new List<TKey>();
 
            foreach (var (cacheKey, cacheInstance) in dictionary)
            {
                if (cacheInstance.Item1 == context.Name)
                {
                    obsoleteCacheKeys.Add(cacheKey);
                }
            }
 
            foreach (var cacheKey in obsoleteCacheKeys)
            {
                _ = usingConcurrentDictionary
                  ? ((ConcurrentDictionary<TKey, Tuple<string, TValue>>)dictionary).TryRemove(cacheKey, out _)
                  : dictionary.Remove(cacheKey);
            }
        }
    }
}
#endif