refactor: 精简代码

Author: ArgoZhang

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -308,22 +308,16 @@ public async Task FixLengthDataPackageHandler_Ok()
         // 设置数据适配器
         var adapter = new DataPackageAdapter
         {
-            DataPackageHandler = new FixLengthDataPackageHandler(7),
-            ReceivedCallBack = buffer =>
-            {
-                // buffer 即是接收到的数据
-                buffer.CopyTo(receivedBuffer);
-                receivedBuffer = receivedBuffer[..buffer.Length];
-                tcs.SetResult();
-                return ValueTask.CompletedTask;
-            }
+            DataPackageHandler = new FixLengthDataPackageHandler(7)
         };
-
-        client.ReceivedCallBack = async buffer =>
+        client.SetDataPackageAdapter(adapter, buffer =>
         {
-            // 将接收到的数据传递给 DataPackageAdapter
-            await adapter.HandlerAsync(buffer);
-        };
+            // buffer 即是接收到的数据
+            buffer.CopyTo(receivedBuffer);
+            receivedBuffer = receivedBuffer[..buffer.Length];
+            tcs.SetResult();
+            return ValueTask.CompletedTask;
+        });
 
         // 测试 ConnectAsync 方法
         var connect = await client.ConnectAsync("localhost", port);
@@ -350,30 +344,25 @@ public async Task FixLengthDataPackageHandler_Sticky()
         var server = StartTcpServer(port, MockStickyPackageAsync);
         var client = CreateClient();
         var tcs = new TaskCompletionSource();
-        var receivedBuffer = new byte[1024];
+        var receivedBuffer = new byte[128];
 
         // 连接 TCP Server
         var connect = await client.ConnectAsync("localhost", port);
 
         // 设置数据适配器
         var adapter = new DataPackageAdapter
         {
-            DataPackageHandler = new FixLengthDataPackageHandler(7),
-            ReceivedCallBack = buffer =>
-            {
-                // buffer 即是接收到的数据
-                buffer.CopyTo(receivedBuffer);
-                receivedBuffer = receivedBuffer[..buffer.Length];
-                tcs.SetResult();
-                return ValueTask.CompletedTask;
-            }
+            DataPackageHandler = new FixLengthDataPackageHandler(7)
         };
 
-        client.ReceivedCallBack = async buffer =>
+        client.SetDataPackageAdapter(adapter, buffer =>
         {
-            // 将接收到的数据传递给 DataPackageAdapter
-            await adapter.HandlerAsync(buffer);
-        };
+            // buffer 即是接收到的数据
+            buffer.CopyTo(receivedBuffer);
+            receivedBuffer = receivedBuffer[..buffer.Length];
+            tcs.SetResult();
+            return ValueTask.CompletedTask;
+        });
 
         // 发送数据
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
@@ -408,31 +397,25 @@ public async Task FixLengthDataPackageHandler_Sticky()
     [Fact]
     public async Task DelimiterDataPackageHandler_Ok()
     {
-        var port = 8886;
+        var port = 8883;
         var server = StartTcpServer(port, MockDelimiterPackageAsync);
         var client = CreateClient();
         var tcs = new TaskCompletionSource();
-        var receivedBuffer = new byte[1024];
+        var receivedBuffer = new byte[128];
 
         // 设置数据适配器
         var adapter = new DataPackageAdapter
         {
             DataPackageHandler = new DelimiterDataPackageHandler([13, 10]),
-            ReceivedCallBack = buffer =>
-            {
-                // buffer 即是接收到的数据
-                buffer.CopyTo(receivedBuffer);
-                receivedBuffer = receivedBuffer[..buffer.Length];
-                tcs.SetResult();
-                return ValueTask.CompletedTask;
-            }
         };
-
-        client.ReceivedCallBack = async buffer =>
+        client.SetDataPackageAdapter(adapter, buffer =>
         {
-            // 将接收到的数据传递给 DataPackageAdapter
-            await adapter.HandlerAsync(buffer);
-        };
+            // buffer 即是接收到的数据
+            buffer.CopyTo(receivedBuffer);
+            receivedBuffer = receivedBuffer[..buffer.Length];
+            tcs.SetResult();
+            return ValueTask.CompletedTask;
+        });
 
         // 连接 TCP Server
         var connect = await client.ConnectAsync("localhost", port);