diff --git a/src/activities/Elsa.Activities.Rebus/Bookmarks/MessageReceivedBookmark.cs b/src/activities/Elsa.Activities.Rebus/Bookmarks/MessageReceivedBookmark.cs index 6a652c7d6..89b83fc4e 100644 --- a/src/activities/Elsa.Activities.Rebus/Bookmarks/MessageReceivedBookmark.cs +++ b/src/activities/Elsa.Activities.Rebus/Bookmarks/MessageReceivedBookmark.cs @@ -1,4 +1,5 @@ using System.Collections.Generic; +using System.Linq; using System.Threading; using System.Threading.Tasks; using Elsa.Services; @@ -13,13 +14,20 @@ namespace Elsa.Activities.Rebus.Bookmarks public class MessageReceivedTriggerProvider : BookmarkProvider { - public override async ValueTask> GetBookmarksAsync(BookmarkProviderContext context, CancellationToken cancellationToken) => - new[] + public override async ValueTask> GetBookmarksAsync(BookmarkProviderContext context, CancellationToken cancellationToken) + { + var messageType = await context.ReadActivityPropertyAsync(x => x.MessageType, cancellationToken); + + if (messageType == null) + return Enumerable.Empty(); + + return new[] { Result(new MessageReceivedBookmark { - MessageType = (await context.ReadActivityPropertyAsync(x => x.MessageType, cancellationToken))!.Name + MessageType = messageType.Name }) }; + } } } \ No newline at end of file