furkandeveloper/EasyCache

using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Filters;
using System;
using System.Collections.Generic;
using System.Text;
using Microsoft.Extensions.DependencyInjection;
using EasyCache.Core.Abstractions;

namespace EasyCache.AspNetCore.Attributes
{
    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = false)]
    public class AutoCacheAttribute : Attribute, IActionFilter
    {
        public AutoCacheAttribute(Type type, string key)
        {
            this.Type = type;
            this.Key = key;
        }

        public AutoCacheAttribute(Type type, string key, Type actionResult) : this(type, key)
        {
            this.ActionResult = actionResult;
        }

        public Type Type { get; set; }
        public string Key { get; set; }

        public Type ActionResult { get; set; } = typeof(OkObjectResult);

        public void OnActionExecuted(ActionExecutedContext context)
        {
            var cacheService = context.HttpContext.RequestServices.GetService<IEasyCacheService>();
            if (context.Result is ObjectResult objectResult && cacheService.Get<object>(Key) == null)
            {
                var data = objectResult.Value;

                cacheService.Set<object>(Key, data, TimeSpan.FromSeconds(30));
            }
        }

        public void OnActionExecuting(ActionExecutingContext context)
        {
            var cacheService = context.HttpContext.RequestServices.GetService<IEasyCacheService>();
            var data = cacheService.Get<object>(Key);
            if (data != null)
            {
                context.Result = (IActionResult)Activator.CreateInstance(ActionResult, args: new object[] { data });
                return;
            }
        }
    }
}