Umbraco-CMS/src/Umbraco.Web/PublishedCache/NuCache/PublishedShapshot.cs

using System;
using Umbraco.Core;
using Umbraco.Core.Cache;

namespace Umbraco.Web.PublishedCache.NuCache
{
    // implements published snapshot
    internal class PublishedShapshot : IPublishedShapshot, IDisposable
    {
        private readonly PublishedSnapshotService _service;
        private bool _defaultPreview;
        private PublishedSnapshotElements _elements;

        #region Constructors

        public PublishedShapshot(PublishedSnapshotService service, bool defaultPreview)
        {
            _service = service;
            _defaultPreview = defaultPreview;
            SnapshotCache = new ObjectCacheRuntimeCacheProvider();
        }

        public class PublishedSnapshotElements : IDisposable
        {
            public ContentCache ContentCache;
            public MediaCache MediaCache;
            public MemberCache MemberCache;
            public DomainCache DomainCache;
            public ICacheProvider SnapshotCache;
            public ICacheProvider ElementsCache;

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

        private PublishedSnapshotElements Elements => _elements ?? (_elements = _service.GetElements(_defaultPreview));

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

        #endregion

        #region Caches

        public ICacheProvider SnapshotCache { get; }

        public ICacheProvider ElementsCache => Elements.ElementsCache;

        #endregion

        #region IPublishedSnapshot

        public IPublishedContentCache ContentCache => Elements.ContentCache;

        public IPublishedMediaCache MediaCache => Elements.MediaCache;

        public IPublishedMemberCache MemberCache => Elements.MemberCache;

        public IDomainCache DomainCache => Elements.DomainCache;

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

        private class ForcedPreviewObject : DisposableObject
        {
            private readonly PublishedShapshot _publishedShapshot;
            private readonly bool _origPreview;
            private readonly Action<bool> _callback;

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

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

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

        #endregion

        #region IDisposable

        private bool _disposed;

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

        #endregion
    }
}