Adding cache to MediatR

CachePipelineBehavior.cs

public class CachePipelineBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
    {
        private readonly IDistributedCache cache;
        private readonly ILogger<SGUKAspNetCore> logger;

        public CachePipelineBehavior(
            IDistributedCache cache,
            ILogger<SGUKAspNetCore> logger)
        {
            Ensure.Argument.NotNull(cache, nameof(cache));
            Ensure.Argument.NotNull(logger, nameof(logger));
            this.cache = cache;
            this.logger = logger;
        }

        public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
        {
            if (request is ICacheable cacheableRequest)
            {
                var key = cacheableRequest.GetCacheKey();

                return await cache.GetOrSet(
                    key,
                    miss: () => { Log.CacheMiss(logger, key); return next(); },
                    hit: (data) => Log.CacheHit(logger, key),
                    cacheableRequest.GetExpirationTime(),
                    cancellationToken);
            }

            var response = await next();

            if (request is ICacheableInvalidation cacheInvalidationRequest)
            {
                foreach (var key in cacheInvalidationRequest.GetCacheKeys())
                {
                    await cache.Remove(key, cancellationToken);
                }
            }

            return response;
        }
    }

ICacheable.cs

public interface ICacheable
{
    public string GetCacheKey();
    public TimeSpan? GetExpirationTime() => null;
}

ICacheableInvalidation.cs

public interface ICacheableInvalidation
{
    public IEnumerable<string> GetCacheKeys();
}