Fixes a NullReferenceExceptio error.

NerosoftDev · Dec 12, 2023 · ead9d75 · ead9d75
1 parent c2e0b46
commit ead9d75
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/Source/Euonia.Bus.InMemory/InMemoryDispatcher.cs b/Source/Euonia.Bus.InMemory/InMemoryDispatcher.cs
@@ -25,7 +25,7 @@ public InMemoryDispatcher(IServiceProvider provider)
 	public async Task PublishAsync<TMessage>(RoutedMessage<TMessage> message, CancellationToken cancellationToken = default)
 		where TMessage : class
 	{
-		var context = new MessageContext(message, _identity.GetIdentity);
+		var context = new MessageContext(message, authorization => _identity?.GetIdentity(authorization));
 		var pack = new MessagePack(message, context)
 		{
 			Aborted = cancellationToken
@@ -39,7 +39,7 @@ public async Task PublishAsync<TMessage>(RoutedMessage<TMessage> message, Cancel
 	public async Task SendAsync<TMessage>(RoutedMessage<TMessage> message, CancellationToken cancellationToken = default)
 		where TMessage : class
 	{
-		var context = new MessageContext(message, _identity.GetIdentity);
+		var context = new MessageContext(message, authorization => _identity?.GetIdentity(authorization));
 		var pack = new MessagePack(message, context)
 		{
 			Aborted = cancellationToken
@@ -73,7 +73,7 @@ public async Task SendAsync<TMessage>(RoutedMessage<TMessage> message, Cancellat
 	public async Task<TResponse> SendAsync<TMessage, TResponse>(RoutedMessage<TMessage, TResponse> message, CancellationToken cancellationToken = default)
 		where TMessage : class
 	{
-		var context = new MessageContext(message, _identity.GetIdentity);
+		var context = new MessageContext(message, authorization => _identity?.GetIdentity(authorization));
 		var pack = new MessagePack(message, context)
 		{
 			Aborted = cancellationToken

diff --git a/Source/Euonia.Bus.RabbitMq/RabbitMqQueueConsumer.cs b/Source/Euonia.Bus.RabbitMq/RabbitMqQueueConsumer.cs
@@ -76,7 +76,7 @@ protected override async void HandleMessageReceived(object sender, BasicDeliverE
 
 		var props = args.BasicProperties;
 
-		var context = new MessageContext(message, _identity.GetIdentity);
+		var context = new MessageContext(message, authorization => _identity?.GetIdentity(authorization));
 
 		OnMessageReceived(new MessageReceivedEventArgs(message.Data, context));
 

diff --git a/Source/Euonia.Bus.RabbitMq/RabbitMqTopicSubscriber.cs b/Source/Euonia.Bus.RabbitMq/RabbitMqTopicSubscriber.cs
@@ -83,7 +83,7 @@ protected override async void HandleMessageReceived(object sender, BasicDeliverE
 
 		var message = DeserializeMessage(args.Body.ToArray(), type);
 
-		var context = new MessageContext(message, _identity.GetIdentity);
+		var context = new MessageContext(message, authorization => _identity?.GetIdentity(authorization));
 
 		OnMessageReceived(new MessageReceivedEventArgs(message.Data, context));