add

Author: Y-Sindo

src/Microsoft.Azure.SignalR.Common/Utilities/AckHandler.cs

@@ -17,7 +17,7 @@ namespace Microsoft.Azure.SignalR
     internal sealed class AckHandler : IDisposable
     {
         public static readonly AckHandler Singleton = new();
-        public static readonly ServiceModelProtocol _serviceModelProtocol = new();
+        public static readonly ServiceProtocol _serviceProtocol = new();
         private readonly ConcurrentDictionary<int, IAckInfo> _acks = new();
         private readonly Timer _timer;
         private readonly TimeSpan _defaultAckTimeout;
@@ -226,7 +226,7 @@ public override bool Ack(AckStatus status, ReadOnlySequence<byte>? payload = nul
 
                 try
                 {
-                    var result = _serviceModelProtocol.ParseModel<T>(payload.Value);
+                    var result = _serviceProtocol.ParseMessagePayload<T>(payload.Value);
                     return _tcs.TrySetResult(result);
                 }
                 catch (Exception e)

src/Microsoft.Azure.SignalR.Protocols/Models/ServiceModelProtocol.cs

@@ -21,6 +21,35 @@ public class ServiceProtocol : IServiceProtocol
     /// <inheritdoc />
     public int Version => ProtocolVersion;
 
+    public T ParseMessagePayload<T>(ReadOnlySequence<byte> input) where T : notnull, new()
+    {
+        if (typeof(IMessagePackSerializable).IsAssignableFrom(typeof(T)))
+        {
+            var model = new T();
+            var reader = new MessagePackReader(input);
+            ((IMessagePackSerializable)model).Load(ref reader, typeof(T).Name);
+            return model;
+        }
+        else
+        {
+            throw new NotSupportedException($"Type {typeof(T).Name} is not supported.");
+        }
+    }
+
+    public void WriteMessagePayload<T>(T model, IBufferWriter<byte> output) where T : notnull, new()
+    {
+        if (typeof(IMessagePackSerializable).IsAssignableFrom(typeof(T)))
+        {
+            var writer = new MessagePackWriter(output);
+            ((IMessagePackSerializable)model).Serialize(ref writer);
+            writer.Flush();
+        }
+        else
+        {
+            throw new NotSupportedException($"Type {typeof(T).Name} is not supported.");
+        }
+    }
+
     /// <inheritdoc />
     public bool TryParseMessage(ref ReadOnlySequence<byte> input, out ServiceMessage? message)
     {