init

DouyinApi.Extensions/AOP/BlogCacheAOP.cs

@@ -0,0 +1,84 @@
+using System;
+using DouyinApi.Common;
+using Castle.DynamicProxy;
+using System.Linq;
+using System.Threading.Tasks;
+using DouyinApi.Common.Caches;
+using DouyinApi.Common.Caches.Interface;
+
+namespace DouyinApi.AOP
+{
+	/// <summary>
+	/// 面向切面的缓存使用
+	/// </summary>
+	public class BlogCacheAOP : CacheAOPbase
+	{
+		//通过注入的方式，把缓存操作接口通过构造函数注入
+		private readonly ICaching _cache;
+
+		public BlogCacheAOP(ICaching cache)
+		{
+			_cache = cache;
+		}
+
+		//Intercept方法是拦截的关键所在，也是IInterceptor接口中的唯一定义
+		public override void Intercept(IInvocation invocation)
+		{
+			var method = invocation.MethodInvocationTarget ?? invocation.Method;
+			//对当前方法的特性验证
+			//如果需要验证
+			var CachingAttribute = method.GetCustomAttributes(true).FirstOrDefault(x => x.GetType() == typeof(CachingAttribute));
+			if (CachingAttribute is CachingAttribute qCachingAttribute)
+			{
+				//获取自定义缓存键
+				var cacheKey = CustomCacheKey(invocation);
+				if (_cache.Exists(cacheKey))
+				{
+					//将当前获取到的缓存值，赋值给当前执行方法
+					Type returnType;
+					if (typeof(Task).IsAssignableFrom(method.ReturnType))
+					{
+						returnType = method.ReturnType.GenericTypeArguments.FirstOrDefault();
+					}
+					else
+					{
+						returnType = method.ReturnType;
+					}
+
+					//根据key获取相应的缓存值
+					dynamic cacheValue = _cache.Get(returnType, cacheKey);
+					invocation.ReturnValue = (typeof(Task).IsAssignableFrom(method.ReturnType)) ? Task.FromResult(cacheValue) : cacheValue;
+					return;
+				}
+
+				//去执行当前的方法
+				invocation.Proceed();
+				//存入缓存
+				if (!string.IsNullOrWhiteSpace(cacheKey))
+				{
+					object response;
+
+					//Type type = invocation.ReturnValue?.GetType();
+					var type = invocation.Method.ReturnType;
+					if (typeof(Task).IsAssignableFrom(type))
+					{
+						dynamic result = invocation.ReturnValue;
+						response = result.Result;
+					}
+					else
+					{
+						response = invocation.ReturnValue;
+					}
+
+					if (response == null) response = string.Empty;
+
+					_cache.Set(cacheKey, response, TimeSpan.FromMinutes(qCachingAttribute.AbsoluteExpiration));
+				}
+			}
+			else
+			{
+				invocation.Proceed(); //直接执行被拦截方法
+			}
+		}
+	}
+}