json body

Author: Ihar Yakimush

Commmunity.AspNetCore.ExceptionHandling.Integration/Startup.cs

@@ -31,21 +31,20 @@ public void ConfigureServices(IServiceCollection services)
             services.AddExceptionHandlingPolicies(options =>
             {
                 options.For<DuplicateNameException>().Retry().NextChain();
+
                 options.For<DuplicateWaitObjectException>().Retry();
 
                 options.For<ArgumentOutOfRangeException>().Log().Rethrow();
 
                 options.For<InvalidCastException>()
                     .Response(e => 400)
-                    .WithHeaders((h, e) => h["X-qwe"] = e.Message)
-                    .WithBody((req,stream, exception) =>
-                    {
-                        using (StreamWriter sw = new StreamWriter(stream))
-                        {
-                            return sw.WriteAsync(exception.ToString());
-                        }                                                
+                    .Headers((h, e) => h["X-qwe"] = e.Message)
+                    .WithBody((req,sw, exception) =>
+                    {                       
+                        return sw.WriteAsync(exception.ToString());                                                                        
                     })
                     .NextChain();
+
                 options.For<Exception>()
                     .Log(lo => { lo.Formatter = (o, e) => "qwe"; })
                     .Response(e => 500, RequestStartedBehaviour.SkipHandler).ClearCacheHeaders().WithBodyJson((r, e) => new { msg = e.Message, path = r.Path })

Commmunity.AspNetCore.ExceptionHandling/ExceptionHandlingPolicyMiddleware.cs

@@ -172,6 +172,8 @@ private async Task InvokeWithRetryAsync(HttpContext context, RequestDelegate nex
                             iteration,
                             context.TraceIdentifier);
 
+                    context.Response.Headers.Clear();
+
                     await InvokeWithRetryAsync(context, next, logger, iteration + 1);
                 }
 

Commmunity.AspNetCore.ExceptionHandling/PolicyBuilderExtensions.cs

@@ -136,7 +136,7 @@ public static IResponseHandlers<TException> Response<TException>(
             return builder as IResponseHandlers<TException>;
         }
 
-        public static IResponseHandlers<TException> WithHeaders<TException>(
+        public static IResponseHandlers<TException> Headers<TException>(
             this IResponseHandlers<TException> builder, Action<IHeaderDictionary,TException> settings, int index = -1)
             where TException : Exception
         {
@@ -180,22 +180,19 @@ public static IResponseHandlers<TException> ClearCacheHeaders<TException>(
         }
 
         public static IResponseHandlers<TException> WithBody<TException>(
-            this IResponseHandlers<TException> builder, Func<HttpRequest, Stream, TException, Task> settings, int index = -1)
+            this IResponseHandlers<TException> builder, Func<HttpRequest, StreamWriter, TException, Task> settings, int index = -1)
             where TException : Exception
         {
             if (settings == null)
             {
                 throw new ArgumentNullException(nameof(settings));
-            }
+            }            
 
             builder.Services.Configure<RawResponseHandlerOptions<TException>>(responceOptions =>
             {
                 responceOptions.SetResponse.Add((context, exception) =>
                 {
-                    if (context.Response.Body.CanSeek)
-                        context.Response.Body.SetLength(0L);
-
-                    return settings(context.Request,context.Response.Body, exception);
+                    return RawResponseExceptionHandler<TException>.SetBody(context, exception, settings);
                 });
             });
 
@@ -206,7 +203,7 @@ public static IResponseHandlers<TException> WithBodyJson<TException, TObject>(
             this IResponseHandlers<TException> builder, Func<HttpRequest, TException, TObject> value, JsonSerializerSettings settings = null, int index = -1)
             where TException : Exception
         {
-            return builder.WithBody((request, stream, exception) =>
+            return builder.WithBody((request, streamWriter, exception) =>
             {
                 if (settings == null)
                 {                    
@@ -226,12 +223,23 @@ public static IResponseHandlers<TException> WithBodyJson<TException, TObject>(
                     headers[HeaderNames.ContentType] = "application/json";
                 }
 
-                using (TextWriter textWriter = new StreamWriter(stream))
+                TObject val = value(request, exception);
+
+                //return Task.CompletedTask;
+                using (MemoryStream ms = new MemoryStream())
                 {
-                    jsonSerializer.Serialize(textWriter, value(request, exception));
-                }
+                    using (StreamWriter sw = new StreamWriter(ms, streamWriter.Encoding, 1024, true))
+                    {
+                        jsonSerializer.Serialize(sw, val);
+                    }
 
-                return Task.CompletedTask;
+                    ms.Seek(0, SeekOrigin.Begin);
+                    byte[] array = ms.ToArray();
+                    BinaryWriter binaryWriter = new BinaryWriter(streamWriter.BaseStream, streamWriter.Encoding, true);
+                    binaryWriter.Write(array);
+
+                    return Task.CompletedTask;
+                }
             });
         }
     }

Commmunity.AspNetCore.ExceptionHandling/Response/RawResponseExceptionHandler.cs

@@ -1,7 +1,10 @@
 using System;
+using System.IO;
 using System.Net;
+using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using Commmunity.AspNetCore.ExceptionHandling.Builder;
 using Commmunity.AspNetCore.ExceptionHandling.Handlers;
 using Commmunity.AspNetCore.ExceptionHandling.Logs;
 using Microsoft.AspNetCore.Http;
@@ -15,6 +18,8 @@ public class RawResponseExceptionHandler<TException> : HandlerStrongType<TExcept
     {
         public const string StatusCodeSetKey = "5D1CFED34A39";
 
+        public const string BodySetKey = "6D1CFED34A39";
+
         private readonly RawResponseHandlerOptions<TException> _options;
 
         private static readonly EventId ResponseHasStarted = new EventId(127, "ResponseAlreadyStarted");
@@ -40,8 +45,7 @@ protected override async Task<HandlerResult> HandleStrongType(HttpContext httpCo
                 else
                 {
                     return HandlerResult.NextHandler;
-                }
-                
+                }                
             }
 
             await HandleResponseAsync(httpContext, exception);
@@ -78,5 +82,31 @@ public static Task SetStatusCode(HttpContext context, TException exception, Func
 
             return Task.CompletedTask;
         }
+
+        public static Task SetBody<TException>(HttpContext context, TException exception, Func<HttpRequest, StreamWriter, TException, Task> settings) 
+        where TException : Exception
+        {
+            if (!context.Items.ContainsKey(BodySetKey))
+            {
+                context.Items[BodySetKey] = true;
+
+                Stream stream = context.Response.Body;
+
+                if (stream.CanSeek)
+                {
+                    stream.Seek(0, SeekOrigin.Begin);
+                    //stream.SetLength(0);
+                }
+
+                if (stream.CanWrite)
+                {
+                    StreamWriter writer = new StreamWriter(stream,Encoding.UTF8,1024, true);
+
+                    return settings(context.Request, writer, exception);                    
+                }
+            }
+
+            return Task.CompletedTask;
+        }
     }
 }