如何在 NServicebus 消息处理程序中注入多个存储库？

Question

我使用以下内容：

public interface IRepository<T>
{
   void Add(T entity);
}

public class Repository<T>
{
  private readonly ISession session;

  public Repository(ISession session)
  {
    this.session = session;
  }

  public void Add(T entity)
  {
     session.Save(entity);
  }
}

public class SomeHandler : IHandleMessages<SomeMessage>
{
  private readonly IRepository<EntityA> aRepository;
  private readonly IRepository<EntityB> bRepository;

  public SomeHandler(IRepository<EntityA> aRepository, IRepository<EntityB> bRepository)
  {
    this.aRepository = aRepository;
    this.bRepository = bRepository; 
  }

  public void Handle(SomeMessage message)
  {
   aRepository.Add(new A(message.Property);
   bRepository.Add(new B(message.Property);
  }
}

public class MessageEndPoint : IConfigureThisEndpoint, AsA_Server, IWantCustomInitialization
{
   public void Init()
   {
      ObjectFactory.Configure(config =>
        {
            config.For<ISession>()
                .CacheBy(InstanceScope.ThreadLocal)
                .TheDefault.Is.ConstructedBy(ctx => ctx.GetInstance<ISessionFactory>().OpenSession());
            config.ForRequestedType(typeof(IRepository<>))
                .TheDefaultIsConcreteType(typeof(Repository<>));
   }
}

我的线程本地存储问题是，在整个应用程序线程期间使用相同的会话。当我看到一级缓存没有被清除时我发现了这一点。我想要的是在每次调用 IHandleMessages<>.Handle 之前使用新的会话实例。 我怎样才能用结构图做到这一点？我必须创建消息模块吗？

Answer 1

您是对的，同一会话用于同一线程的所有请求。这是因为 NSB 不会为每个请求创建新线程。解决方法是添加自定义缓存模式并在消息处理完成时将其清除。

1.延长线程存储生命周期并将其连接到消息模块

public class NServiceBusThreadLocalStorageLifestyle : ThreadLocalStorageLifecycle, IMessageModule
{

    public void HandleBeginMessage(){}

    public void HandleEndMessage()
    {
        EjectAll();
    }

    public void HandleError(){}
}

2.按如下方式配置结构图：

For<<ISession>>
.LifecycleIs(new NServiceBusThreadLocalStorageLifestyle())
...

希望这会有所帮助！