From 7e3924f4578c650f1c67e77c6bb185bd4cb08527 Mon Sep 17 00:00:00 2001
From: Andrei Grigorev <angrigor@microsoft.com>
Date: Fri, 23 Feb 2024 14:28:03 -0800
Subject: [PATCH] `GetRawBody` method

---
 .../ServiceBus/IReceivedBrokeredMessage.cs                  | 2 ++
 .../ServiceBusReceivedMessageWrapper.cs                     | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/src/NuGet.Services.Contracts/ServiceBus/IReceivedBrokeredMessage.cs b/src/NuGet.Services.Contracts/ServiceBus/IReceivedBrokeredMessage.cs
index adbdc7c6..0627a088 100644
--- a/src/NuGet.Services.Contracts/ServiceBus/IReceivedBrokeredMessage.cs
+++ b/src/NuGet.Services.Contracts/ServiceBus/IReceivedBrokeredMessage.cs
@@ -3,6 +3,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Threading.Tasks;
 
 namespace NuGet.Services.ServiceBus
@@ -20,5 +21,6 @@ public interface IReceivedBrokeredMessage
         Task AbandonAsync();
         string GetBody();
         TStream GetBody<TStream>();
+        Stream GetRawBody();
     }
 }
\ No newline at end of file
diff --git a/src/NuGet.Services.ServiceBus/ServiceBusReceivedMessageWrapper.cs b/src/NuGet.Services.ServiceBus/ServiceBusReceivedMessageWrapper.cs
index e19f4a75..506cbffd 100644
--- a/src/NuGet.Services.ServiceBus/ServiceBusReceivedMessageWrapper.cs
+++ b/src/NuGet.Services.ServiceBus/ServiceBusReceivedMessageWrapper.cs
@@ -5,6 +5,7 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
+using System.IO;
 using System.Linq;
 using System.Runtime.Serialization;
 using System.Threading.Tasks;
@@ -48,6 +49,11 @@ public TStream GetBody<TStream>()
             return ServiceBusClientHelper.DeserializeXmlDataContract<TStream>(ServiceBusReceivedMessage.Body);
         }
 
+        public Stream GetRawBody()
+        {
+            return ServiceBusReceivedMessage.Body.ToStream();
+        }
+
         public Task CompleteAsync()
         {
             return _args.CompleteMessageAsync(ServiceBusReceivedMessage);