Don't call MemoryStream.ToArray, don't create MemoryStream, NetworkBinaryWriter

Author: yupliner

projects/client/RabbitMQ.Client/RabbitMQ.Client.csproj

@@ -40,6 +40,10 @@
     <DefineConstants>$(DefineConstants);CORECLR</DefineConstants>
   </PropertyGroup>
 
+  <PropertyGroup Condition=" '$(TargetFramework)' == 'netstandard1.5'">
+    <DefineConstants>$(DefineConstants);CORECLR15</DefineConstants>
+  </PropertyGroup>
+
   <PropertyGroup Condition=" '$(Configuration)' == 'SignedRelease' ">
     <DelaySign>true</DelaySign>
     <OutputType>Library</OutputType>

projects/client/RabbitMQ.Client/src/client/impl/Frame.cs

@@ -56,98 +56,115 @@ namespace RabbitMQ.Client.Impl
 {
     public class HeaderOutboundFrame : OutboundFrame
     {
+        private readonly ContentHeaderBase header;
+        private readonly int bodyLength;
+
         public HeaderOutboundFrame(int channel, ContentHeaderBase header, int bodyLength) : base(FrameType.FrameHeader, channel)
         {
-            NetworkBinaryWriter writer = base.GetWriter();
+            this.header = header;
+            this.bodyLength = bodyLength;
+        }
+
+        public override void WritePayload(NetworkBinaryWriter writer)
+        {
+            var ms = new MemoryStream();
+            var nw = new NetworkBinaryWriter(ms);
 
-            writer.Write(header.ProtocolClassId);
-            header.WriteTo(writer, (ulong)bodyLength);
+            nw.Write(header.ProtocolClassId);
+            header.WriteTo(nw, (ulong)bodyLength);
+
+#if CORECLR15
+            var payload = ms.ToArray();
+            writer.Write((uint)payload.Length);
+            writer.Write(payload);
+#else
+            var buffer = ms.GetBuffer();
+            writer.Write((uint)ms.Position);
+            writer.Write(buffer, 0, (int)ms.Position);
+#endif
         }
     }
 
     public class BodySegmentOutboundFrame : OutboundFrame
     {
+        private readonly byte[] body;
+        private readonly int offset;
+        private readonly int count;
+
         public BodySegmentOutboundFrame(int channel, byte[] body, int offset, int count) : base(FrameType.FrameBody, channel)
         {
-            NetworkBinaryWriter writer = base.GetWriter();
+            this.body = body;
+            this.offset = offset;
+            this.count = count;
+        }
 
+        public override void WritePayload(NetworkBinaryWriter writer)
+        {
+            writer.Write((uint)count);
             writer.Write(body, offset, count);
         }
     }
 
     public class MethodOutboundFrame : OutboundFrame
     {
+        private readonly MethodBase method;
+
         public MethodOutboundFrame(int channel, MethodBase method) : base(FrameType.FrameMethod, channel)
         {
-            NetworkBinaryWriter writer = base.GetWriter();
+            this.method = method;
+        }
 
-            writer.Write(method.ProtocolClassId);
-            writer.Write(method.ProtocolMethodId);
+        public override void WritePayload(NetworkBinaryWriter writer)
+        {
+            var ms = new MemoryStream();
+            var nw = new NetworkBinaryWriter(ms);
 
-            var argWriter = new MethodArgumentWriter(writer);
+            nw.Write(method.ProtocolClassId);
+            nw.Write(method.ProtocolMethodId);
 
+            var argWriter = new MethodArgumentWriter(nw);
             method.WriteArgumentsTo(argWriter);
-
             argWriter.Flush();
+
+#if CORECLR15
+            var payload = ms.ToArray();
+            writer.Write((uint)payload.Length);
+            writer.Write(payload);
+#else
+            var buffer = ms.GetBuffer();
+            writer.Write((uint)ms.Position);
+            writer.Write(buffer, 0, (int)ms.Position);
+#endif
         }
     }
 
     public class EmptyOutboundFrame : OutboundFrame
     {
-        private static readonly byte[] m_emptyByteArray = new byte[0];
-
         public EmptyOutboundFrame() : base(FrameType.FrameHeartbeat, 0)
         {
-            base.GetWriter().Write(m_emptyByteArray);
         }
 
-        public override string ToString()
+        public override void WritePayload(NetworkBinaryWriter writer)
         {
-            return base.ToString() + string.Format("(type={0}, channel={1}, {2} bytes of payload)",
-                Type,
-                Channel,
-                Payload == null
-                    ? "(null)"
-                    : Payload.Length.ToString());
+            writer.Write((uint)0);
         }
     }
 
-    public class OutboundFrame : Frame
+    public abstract class OutboundFrame : Frame
     {
-        private readonly MemoryStream m_accumulator;
-        private readonly NetworkBinaryWriter writer;
-
         public OutboundFrame(FrameType type, int channel) : base(type, channel)
         {
-            m_accumulator = new MemoryStream();
-            writer = new NetworkBinaryWriter(m_accumulator);
-        }
-
-        public NetworkBinaryWriter GetWriter()
-        {
-            return writer;
-        }
-
-        public override string ToString()
-        {
-            return base.ToString() + string.Format("(type={0}, channel={1}, {2} bytes of payload)",
-                Type,
-                Channel,
-                Payload == null
-                    ? "(null)"
-                    : Payload.Length.ToString());
         }
 
         public void WriteTo(NetworkBinaryWriter writer)
         {
-            var payload = m_accumulator.ToArray();
-
             writer.Write((byte)Type);
             writer.Write((ushort)Channel);
-            writer.Write((uint)payload.Length);
-            writer.Write(payload);
+            WritePayload(writer);
             writer.Write((byte)Constants.FrameEnd);
         }
+
+        public abstract void WritePayload(NetworkBinaryWriter writer);
     }
 
     public class InboundFrame : Frame
@@ -252,16 +269,6 @@ public NetworkBinaryReader GetReader()
         {
             return new NetworkBinaryReader(new MemoryStream(base.Payload));
         }
-
-        public override string ToString()
-        {
-            return base.ToString() + string.Format("(type={0}, channel={1}, {2} bytes of payload)",
-                base.Type,
-                base.Channel,
-                base.Payload == null
-                    ? "(null)"
-                    : base.Payload.Length.ToString());
-        }
     }
 
     public class Frame
@@ -288,7 +295,7 @@ public Frame(FrameType type, int channel, byte[] payload)
 
         public override string ToString()
         {
-            return base.ToString() + string.Format("(type={0}, channel={1}, {2} bytes of payload)",
+            return string.Format("(type={0}, channel={1}, {2} bytes of payload)",
                 Type,
                 Channel,
                 Payload == null