Umbraco-CMS/src/Umbraco.PublishedCache.NuCache/PublishedSnapshot.cs

using System;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Cache;
using Umbraco.Cms.Core.PublishedCache;

namespace Umbraco.Cms.Infrastructure.PublishedCache
{
    // implements published snapshot
    public class PublishedSnapshot : IPublishedSnapshot, IDisposable
    {
        private readonly PublishedSnapshotService _service;
        private bool _defaultPreview;
        private PublishedSnapshotElements _elements;

        #region Constructors

        public PublishedSnapshot(IPublishedSnapshotService service, bool defaultPreview)
        {
            _service = service as PublishedSnapshotService;
            _defaultPreview = defaultPreview;
        }

        public class PublishedSnapshotElements : IDisposable
        {
#pragma warning disable IDE1006 // Naming Styles
            public ContentCache ContentCache;
            public MediaCache MediaCache;
            public MemberCache MemberCache;
            public DomainCache DomainCache;
            public IAppCache SnapshotCache;
            public IAppCache ElementsCache;
#pragma warning restore IDE1006 // Naming Styles

            public void Dispose()
            {
                ContentCache.Dispose();
                MediaCache.Dispose();
                MemberCache.Dispose();
            }
        }

        private PublishedSnapshotElements Elements
        {
            get
            {
                if (_service == null)
                {
                    throw new InvalidOperationException($"The {typeof(PublishedSnapshot)} cannot be used when the {typeof(IPublishedSnapshotService)} is not the default type {typeof(PublishedSnapshotService)}");
                }
                return _elements ??= _service.GetElements(_defaultPreview);
            }
        }

        public void Resync()
        {
            // no lock - published snapshots are single-thread
            _elements?.Dispose();
            _elements = null;
        }

        #endregion

        #region Caches

        public IAppCache SnapshotCache => Elements.SnapshotCache;

        public IAppCache ElementsCache => Elements.ElementsCache;

        #endregion

        #region IPublishedSnapshot

        public IPublishedContentCache Content => Elements.ContentCache;

        public IPublishedMediaCache Media => Elements.MediaCache;

        public IPublishedMemberCache Members => Elements.MemberCache;

        public IDomainCache Domains => Elements.DomainCache;

        public IDisposable ForcedPreview(bool preview, Action<bool> callback = null)
        {
            return new ForcedPreviewObject(this, preview, callback);
        }

        private class ForcedPreviewObject : DisposableObjectSlim
        {
            private readonly PublishedSnapshot _publishedSnapshot;
            private readonly bool _origPreview;
            private readonly Action<bool> _callback;

            public ForcedPreviewObject(PublishedSnapshot publishedShapshot, bool preview, Action<bool> callback)
            {
                _publishedSnapshot = publishedShapshot;
                _callback = callback;

                // save and force
                _origPreview = publishedShapshot._defaultPreview;
                publishedShapshot._defaultPreview = preview;
            }

            protected override void DisposeResources()
            {
                // restore
                _publishedSnapshot._defaultPreview = _origPreview;
                _callback?.Invoke(_origPreview);
            }
        }

        #endregion

        #region IDisposable

        private bool _disposed;

        public void Dispose()
        {
            if (_disposed) return;
            _disposed = true;
            _elements?.Dispose();
        }

        #endregion
    }
}