ServiceBusPlugin Tricks

sender-validation-plugin-1.cs

public class SenderValidationPlugin : ServiceBusPlugin
{
}

sender-validation-plugin-2.cs

public class SenderValidationPlugin : ServiceBusPlugin
{
    // Overrides abstract property
    public override string Name => this.GetType().FullName;
}

sender-validation-plugin-3.cs

public class SenderValidationPlugin : ServiceBusPlugin
{
    public override string Name => this.GetType().FullName;

    // Overrides virtual property
    public override bool ShouldContinueOnException { get; }

    // Overrides virtual methods
    public override async Task<Message> BeforeMessageSend(Message message)
    {
        throw new NotImplementedException();
    }

    public override async Task<Message> AfterMessageReceive(Message message)
    {
        throw new NotImplementedException();
    }
}

sender-validation-plugin-4.cs

public class SenderValidationPlugin : ServiceBusPlugin
{
    public override string Name => this.GetType().FullName;

    public override bool ShouldContinueOnException { get; }

    public override async Task<Message> BeforeMessageSend(Message message)
    {
        return await this.ValidateAsync(message)
                         .ConfigureAwait(false);
    }

    public override async Task<Message> AfterMessageReceive(Message message)
    {
        return await this.ValidateAsync(message)
                         .ConfigureAwait(false);
    }

    // Implements private method for both BeforeMessageSend and AfterMessageReceive.
    private async Task<Message> ValidateAsync(Message message)
    {
        throw new NotImplementedException();
    }
}

subscription-receive.cs

var plugin = new SenderValidationPlugin();
var subscription = new SubscriptionClient("/* CONNECTION STRING */", "my-topic", "my-subscription", ReceiveMode.PeekLock);
subscription.RegisterPlugin(plugin);

subscription.RegisterMessageHandler(async (message, token) =>
{
    await subscription.CompleteAsync(message.SystemProperties.LockToken)
                      .ConfigureAwait(false);
},
new MessageHandlerOptions(args => {
    Console.WriteLine(args.Exception.Message);
    Console.WriteLine(args.ExceptionReceivedContext.EntityPath);
}));

topic-send.cs

var plugin = new SenderValidationPlugin();
var topic = new TopicClient("/* CONNECTION STRING */", "my-topic");
topic.RegisterPlugin(plugin);

var payload = "{ \"hello\": \"world\" }";
var body = Encoding.UTF8.GetBytes(serialised);
var message = new Message(body);
message.UserProperties.Add("sender", "lorem");

await topic.SendAsync(message)
           .ConfigureAwait(false);

validate-async.cs

private async Task<Message> ValidateAsync(Message message)
{
    var cloned = message.Clone()

    var body = cloned.Body;
    if (!body.Any())
    {
        throw new InvalidOperationException("Message body not exists");
    }

    var payload = Encoding.UTF8.GetString(cloned.Body);
    var sender = cloned.UserProperties["sender"] as string;
    if (string.IsNullOrWhiteSpace(sender))
    {
        throw new InvalidOperationException("Sender not defined");
    }

    var senders = new List<string>() { "app1", "app2", "app3" };
    if (!senders.Contains(sender))
    {
        throw new InvalidOperationException("Invalid sender");
    }

    return message;
}