squidex/src/Squidex.Infrastructure/Caching/InvalidatingMemoryCache.cs

// ==========================================================================
//  InvalidatingCache.cs
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex Group
//  All rights reserved.
// ==========================================================================

using Microsoft.Extensions.Caching.Memory;

namespace Squidex.Infrastructure.Caching
{
    public class InvalidatingMemoryCache : IMemoryCache
    {
        private const string ChannelName = "CacheInvalidations";
        private readonly IMemoryCache inner;
        private readonly IPubSub invalidator;

        public InvalidatingMemoryCache(IMemoryCache inner, IPubSub invalidator)
        {
            Guard.NotNull(inner, nameof(inner));
            Guard.NotNull(invalidator, nameof(invalidator));

            this.inner = inner;
            this.invalidator = invalidator;

            invalidator.Subscribe(ChannelName, inner.Remove);
        }

        public void Dispose()
        {
            inner.Dispose();
        }

        public ICacheEntry CreateEntry(object key)
        {
            return new WrapperCacheEntry(inner.CreateEntry(key), Invalidate);
        }

        public bool TryGetValue(object key, out object value)
        {
            return inner.TryGetValue(key, out value);
        }

        public void Remove(object key)
        {
            inner.Remove(key);

            Invalidate(key);
        }

        private void Invalidate(object key)
        {
            if (key is string)
            {
                invalidator.Publish(ChannelName, key.ToString(), false);
            }
        }
    }
}