Skip to content

Commit dc822e9

Browse files
committed
refactor: 精简代码逻辑
1 parent 7f514a3 commit dc822e9

File tree

2 files changed

+10
-13
lines changed

2 files changed

+10
-13
lines changed

src/BootstrapBlazor/Services/TcpSocket/DefaultTcpSocketClient.cs

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ namespace BootstrapBlazor.Components;
1515
sealed class DefaultTcpSocketClient(IPEndPoint localEndPoint) : TcpSocketClientBase
1616
{
1717
private TcpClient? _client;
18-
private IDataPackageHandler? _dataPackageHandler;
1918
private CancellationTokenSource? _receiveCancellationTokenSource;
2019
private IPEndPoint? _remoteEndPoint;
2120

@@ -24,13 +23,6 @@ sealed class DefaultTcpSocketClient(IPEndPoint localEndPoint) : TcpSocketClientB
2423
[NotNull]
2524
public ILogger<DefaultTcpSocketClient>? Logger { get; set; }
2625

27-
public override void SetDataHandler(IDataPackageHandler handler)
28-
{
29-
base.SetDataHandler(handler);
30-
31-
_dataPackageHandler = handler;
32-
}
33-
3426
public override async ValueTask<bool> ConnectAsync(IPEndPoint endPoint, CancellationToken token = default)
3527
{
3628
var ret = false;
@@ -99,9 +91,9 @@ public override async ValueTask<bool> SendAsync(ReadOnlyMemory<byte> data, Cance
9991
sendToken = CancellationTokenSource.CreateLinkedTokenSource(token, sendTokenSource.Token).Token;
10092
}
10193

102-
if (_dataPackageHandler != null)
94+
if (DataPackageHandler != null)
10395
{
104-
data = await _dataPackageHandler.SendAsync(data, sendToken);
96+
data = await DataPackageHandler.SendAsync(data, sendToken);
10597
}
10698

10799
await stream.WriteAsync(data, sendToken);
@@ -192,9 +184,9 @@ private async ValueTask<int> ReceiveCoreAsync(TcpClient client, Memory<byte> buf
192184
await ReceivedCallBack(buffer);
193185
}
194186

195-
if (_dataPackageHandler != null)
187+
if (DataPackageHandler != null)
196188
{
197-
await _dataPackageHandler.ReceiveAsync(buffer, receiveToken);
189+
await DataPackageHandler.ReceiveAsync(buffer, receiveToken);
198190
}
199191
}
200192
}

src/BootstrapBlazor/Services/TcpSocket/TcpSocketClientBase.cs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,17 @@ public abstract class TcpSocketClientBase : ITcpSocketClient
5656
/// </summary>
5757
public int ReceiveTimeout { get; set; }
5858

59+
/// <summary>
60+
/// Gets or sets the handler responsible for processing data packages.
61+
/// </summary>
62+
public IDataPackageHandler? DataPackageHandler { get; protected set; }
63+
5964
/// <summary>
6065
/// <inheritdoc/>
6166
/// </summary>
6267
public virtual void SetDataHandler(IDataPackageHandler handler)
6368
{
64-
69+
DataPackageHandler = handler;
6570
}
6671

6772
/// <summary>

0 commit comments

Comments
 (0)