attempt with multiple grpc clients

This commit is contained in:
Stanley Dimant
2022-10-09 15:58:36 +02:00
parent 031d21e087
commit 1c8f16fac0

View File

@@ -17,6 +17,8 @@ public class GrpcClientIdentificationService : IHostedService
private readonly string _shardName; private readonly string _shardName;
private readonly ILogger<GrpcClientIdentificationService> _logger; private readonly ILogger<GrpcClientIdentificationService> _logger;
private readonly IdentificationService.IdentificationServiceClient _grpcIdentClient; private readonly IdentificationService.IdentificationServiceClient _grpcIdentClient;
private readonly IdentificationService.IdentificationServiceClient grpcIdentClientStreamOut;
private readonly IdentificationService.IdentificationServiceClient grpcIdentClientStreamIn;
private readonly MareMetrics _metrics; private readonly MareMetrics _metrics;
protected readonly ConcurrentDictionary<string, UidWithIdent> OnlineClients = new(StringComparer.Ordinal); protected readonly ConcurrentDictionary<string, UidWithIdent> OnlineClients = new(StringComparer.Ordinal);
private readonly ConcurrentDictionary<string, UidWithIdent> RemoteCachedIdents = new(StringComparer.Ordinal); private readonly ConcurrentDictionary<string, UidWithIdent> RemoteCachedIdents = new(StringComparer.Ordinal);
@@ -25,12 +27,16 @@ public class GrpcClientIdentificationService : IHostedService
private CancellationTokenSource _streamCts = new(); private CancellationTokenSource _streamCts = new();
private CancellationTokenSource _faultCheckCts = new(); private CancellationTokenSource _faultCheckCts = new();
public GrpcClientIdentificationService(ILogger<GrpcClientIdentificationService> logger, IdentificationService.IdentificationServiceClient gprcIdentClient, MareMetrics metrics, IConfiguration configuration) public GrpcClientIdentificationService(ILogger<GrpcClientIdentificationService> logger, IdentificationService.IdentificationServiceClient gprcIdentClient,
IdentificationService.IdentificationServiceClient gprcIdentClientStreamOut,
IdentificationService.IdentificationServiceClient gprcIdentClientStreamIn, MareMetrics metrics, IConfiguration configuration)
{ {
var config = configuration.GetSection("MareSynchronos"); var config = configuration.GetSection("MareSynchronos");
_shardName = config.GetValue("ShardName", "Main"); _shardName = config.GetValue("ShardName", "Main");
_logger = logger; _logger = logger;
_grpcIdentClient = gprcIdentClient; _grpcIdentClient = gprcIdentClient;
this.grpcIdentClientStreamOut = gprcIdentClientStreamOut;
this.grpcIdentClientStreamIn = gprcIdentClientStreamIn;
_metrics = metrics; _metrics = metrics;
} }
@@ -184,7 +190,7 @@ public class GrpcClientIdentificationService : IHostedService
{ {
try try
{ {
using var stream = _grpcIdentClient.SendStreamIdentStatusChange(cancellationToken: cts); using var stream = grpcIdentClientStreamOut.SendStreamIdentStatusChange(cancellationToken: cts);
await stream.RequestStream.WriteAsync(new IdentChangeMessage() await stream.RequestStream.WriteAsync(new IdentChangeMessage()
{ {
Server = new ServerMessage() Server = new ServerMessage()
@@ -216,7 +222,7 @@ public class GrpcClientIdentificationService : IHostedService
{ {
try try
{ {
using var stream = _grpcIdentClient.ReceiveStreamIdentStatusChange(new ServerMessage() using var stream = grpcIdentClientStreamIn.ReceiveStreamIdentStatusChange(new ServerMessage()
{ {
ServerId = _shardName, ServerId = _shardName,
}); });
@@ -233,6 +239,7 @@ public class GrpcClientIdentificationService : IHostedService
RemoteCachedIdents.TryRemove(cur.UidWithIdent.Uid.Uid, out _); RemoteCachedIdents.TryRemove(cur.UidWithIdent.Uid.Uid, out _);
} }
} }
_logger.LogCritical("Receive Online Client Data Stream ended");
} }
catch (OperationCanceledException) { return; } catch (OperationCanceledException) { return; }
catch catch