Umbraco-CMS/src/Umbraco.Core/Cache/RepositoryCachePolicyBase.cs

using System;
using System.Collections.Generic;
using Umbraco.Core.Models.EntityBase;
using Umbraco.Core.Scoping;

namespace Umbraco.Core.Cache
{
    /// <summary>
    /// A base class for repository cache policies.
    /// </summary>
    /// <typeparam name="TEntity">The type of the entity.</typeparam>
    /// <typeparam name="TId">The type of the identifier.</typeparam>
    internal abstract class RepositoryCachePolicyBase<TEntity, TId> : IRepositoryCachePolicy<TEntity, TId>
        where TEntity : class, IAggregateRoot
    {
        private readonly IRuntimeCacheProvider _globalCache;
        private readonly IScopeAccessor _scopeAccessor;

        protected RepositoryCachePolicyBase(IRuntimeCacheProvider globalCache, IScopeAccessor scopeAccessor)
        {
            _globalCache = globalCache ?? throw new ArgumentNullException(nameof(globalCache));
            _scopeAccessor = scopeAccessor ?? throw new ArgumentNullException(nameof(scopeAccessor));
        }

        protected IRuntimeCacheProvider Cache
        {
            get
            {
                var ambientScope = _scopeAccessor.AmbientScope;
                switch (ambientScope.RepositoryCacheMode)
                {
                    case RepositoryCacheMode.Default:
                        return _globalCache;
                    case RepositoryCacheMode.Scoped:
                        return ambientScope.IsolatedRuntimeCache.GetOrCreateCache<TEntity>();
                    case RepositoryCacheMode.None:
                        return NullCacheProvider.Instance;
                    default:
                        throw new NotSupportedException($"Repository cache mode {ambientScope.RepositoryCacheMode} is not supported.");
                }
            }
        }

        /// <inheritdoc />
        public abstract TEntity Get(TId id, Func<TId, TEntity> performGet, Func<TId[], IEnumerable<TEntity>> performGetAll);

        /// <inheritdoc />
        public abstract TEntity GetCached(TId id);

        /// <inheritdoc />
        public abstract bool Exists(TId id, Func<TId, bool> performExists, Func<TId[], IEnumerable<TEntity>> performGetAll);

        /// <inheritdoc />
        public abstract void Create(TEntity entity, Action<TEntity> persistNew);

        /// <inheritdoc />
        public abstract void Update(TEntity entity, Action<TEntity> persistUpdated);

        /// <inheritdoc />
        public abstract void Delete(TEntity entity, Action<TEntity> persistDeleted);

        /// <inheritdoc />
        public abstract TEntity[] GetAll(TId[] ids, Func<TId[], IEnumerable<TEntity>> performGetAll);

        /// <inheritdoc />
        public abstract void ClearAll();
    }
}