Add null checks before attempting debug response writing

aws · Oct 20, 2024 · 02eb5b4 · 02eb5b4
1 parent d0a84f2
commit 02eb5b4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Libraries/src/Amazon.Lambda.Serialization.Json/JsonSerializer.cs b/Libraries/src/Amazon.Lambda.Serialization.Json/JsonSerializer.cs
@@ -94,7 +94,7 @@ public void Serialize<T>(T response, Stream responseStream)
         {
             try
             {
-                if (debug)
+                if (debug && response != null)
                 {
                     using (StringWriter debugWriter = new StringWriter())
                     {

diff --git a/Libraries/src/Amazon.Lambda.Serialization.SystemTextJson/AbstractLambdaJsonSerializer.cs b/Libraries/src/Amazon.Lambda.Serialization.SystemTextJson/AbstractLambdaJsonSerializer.cs
@@ -48,7 +48,7 @@ public void Serialize<T>(T response, Stream responseStream)
         {
             try
             {
-                if (_debug)
+                if (_debug && response != null)
                 {
                     using (var debugStream = new MemoryStream())
                     using (var utf8Writer = new Utf8JsonWriter(debugStream, WriterOptions))

diff --git a/Libraries/src/Amazon.Lambda.Serialization.SystemTextJson/LambdaJsonSerializer.cs b/Libraries/src/Amazon.Lambda.Serialization.SystemTextJson/LambdaJsonSerializer.cs
@@ -95,7 +95,7 @@ public void Serialize<T>(T response, Stream responseStream)
         {
             try
             {
-                if (_debug)
+                if (_debug && response != null)
                 {
                     using (var debugWriter = new StringWriter())
                     using (var utf8Writer = new Utf8JsonWriter(responseStream, WriterOptions))
-Original file line number
+Diff line change
@@ Expand Up / @@ -94,7 +94,7 @@ public void Serialize<T>(T response, Stream responseStream) @@
             {
                 try
                 {
-                    if (debug)
+                    if (debug && response != null)
                     {
                         using (StringWriter debugWriter = new StringWriter())
                         {
@@ Expand Down @@