如何在MassTransit(RabbitMQ 7.3.1)中将EntityNameAttribute替换为自定义NameAttribute

我在应用里用不同接口处理消息，为了避免重复实现、减少DTO维护工作，想用自定义属性处理MassTransit的Exchange名称，同时降低公共模块对MassTransit的依赖。现在的问题是：在MassTransit（RabbitMQ 7.3.1版本）中，怎么把EntityNameAttribute替换成自定义的NameAttribute？

我试过重写MessageTopology的EntityNameFormatter，代码如下：

cfg.MessageTopology.SetEntityNameFormatter(new CustomExchangeNameFormatter(cfg.MessageTopology.EntityNameFormatter));  

发布消息的代码：

_asyncApi.PublishMessage<CustomMessage>(new CustomMessage(){Content = "TestMessage"});

报错信息：

RabbitMQ.Client.Exceptions.OperationInterruptedException: The AMQP operation was interrupted: AMQP close-reason, initiated by Peer, code=403, text='ACCESS_REFUSED - operation not permitted on the default exchange', classId=40, methodId=10 ...

自定义Formatter代码：

public class CustomExchangeNameFormatter: IEntityNameFormatter
{
    private IEntityNameFormatter _original;

    public CustomExchangeNameFormatter(IEntityNameFormatter original)
    {
        _original = original;
    }

    //Used to rename the exchanges
    public string FormatEntityName<T>()
    {
        string entityName;

        var exchangeNameAttribute = typeof(T).GetCustomAttribute<ExchangeNameAttribute>();
        if (exchangeNameAttribute != null)
        {
            entityName = exchangeNameAttribute.ExchangName;
        }
        else
        {
            entityName = _original.FormatEntityName<T>();
        }

        return entityName;
    }
}

问题分析与解决 #

1. 报错根源
   出现403权限错误是因为自定义Formatter返回了无效的Exchange名称（空字符串），导致MassTransit尝试使用RabbitMQ的默认交换机（""），而默认交换机不允许直接发布操作。检查代码发现exchangeNameAttribute.ExchangName存在拼写错误（应为ExchangeName），这会导致获取到的属性值为空，最终触发默认交换机的权限问题。

2. 正确替换自定义属性的步骤

   • 定义独立的自定义属性：在公共模块中创建自己的属性，无需依赖MassTransit：

     [AttributeUsage(AttributeTargets.Class, Inherited = false)]
     public class CustomExchangeNameAttribute : Attribute
     {
         public string Name { get; }
     
         public CustomExchangeNameAttribute(string name)
         {
             Name = name;
         }
     }
     

   • 修正自定义Formatter：改为读取自定义属性，同时增加空值校验逻辑：

     public class CustomExchangeNameFormatter : IEntityNameFormatter
     {
         private readonly IEntityNameFormatter _original;
     
         public CustomExchangeNameFormatter(IEntityNameFormatter original)
         {
             _original = original;
         }
     
         public string FormatEntityName<T>()
         {
             var customAttr = typeof(T).GetCustomAttribute<CustomExchangeNameAttribute>();
             if (customAttr != null && !string.IsNullOrWhiteSpace(customAttr.Name))
             {
                 return customAttr.Name;
             }
     
             // 回退到原始逻辑，确保不会返回空值
             var originalName = _original.FormatEntityName<T>();
             return string.IsNullOrWhiteSpace(originalName) ? typeof(T).FullName : originalName;
         }
     }
     

   • 配置MassTransit并应用属性：

     // 配置时设置格式化器
     cfg.MessageTopology.SetEntityNameFormatter(new CustomExchangeNameFormatter(cfg.MessageTopology.EntityNameFormatter));
     
     // 在消息DTO上标记自定义属性
     [CustomExchangeName("custom-message-exchange")]
     public class CustomMessage
     {
         public string Content { get; set; }
     }
     

3. 额外注意事项

   • 确保RabbitMQ用户拥有目标Exchange的configure、write权限，可通过RabbitMQ管理界面或命令行配置。
   • 始终保留有效的回退逻辑，避免返回空字符串作为Exchange名称。

内容的提问来源于stack exchange，提问作者KCB