启用缓存简略输出

Signed-off-by: Vicente.Yu <^@^>
vicenteyu · Dec 6, 2023 · 60f7298 · 60f7298
1 parent e2d002c
commit 60f7298
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/Attributes/EnableLoggingFilterAttribute.cs b/src/Attributes/EnableLoggingFilterAttribute.cs
@@ -13,6 +13,11 @@ public class EnableLoggingFilterAttribute : ApiFilterAttribute
     {
         private static readonly LoggingFilterAttribute _loggingFilter = new();
 
+        /// <summary>
+        /// 启用时，不再输出缓存内容。
+        /// </summary>
+        public bool EnableCacheLogging { get; set; } = true;
+
         /// <summary>
         /// 请求之前
         /// </summary>
@@ -28,6 +33,16 @@ public async override Task OnRequestAsync(ApiRequestContext context)
         public async override Task OnResponseAsync(ApiResponseContext context)
         {
             if (!IsEnable(context.HttpContext)) return;
+
+            if (EnableCacheLogging
+                && context.HttpContext.ResponseMessage?.Headers?
+                    .TryGetValues("Response-Cache-Provider", out var values) == true
+                && values?.Any() == true)
+            {
+                context.HttpContext.ResponseMessage ??= new HttpResponseMessage();
+                context.HttpContext.ResponseMessage.Content = new StringContent("...来自缓存...");
+            }
+
             await _loggingFilter.OnResponseAsync(context);
         }