OrderSagaFinder.cs :  » Workflows » NServiceBus » OrderService » Persistence » C# / CSharp Open Source

using System;
using NServiceBus.Saga;
using OrderService.Messages;
using NHibernate;
using NHibernate.Criterion;

namespace OrderService.Persistence{
    public class OrderSagaFinder : 
        IFindSagas<OrderSagaData>.Using<OrderMessage>,
        IFindSagas<OrderSagaData>.Using<CancelOrderMessage>
    {
        public OrderSagaData FindBy(OrderMessage message)
        {
            return FindBy(message.PurchaseOrderNumber, message.PartnerId);
        }

        public OrderSagaData FindBy(CancelOrderMessage message)
        {
            return FindBy(message.PurchaseOrderNumber, message.PartnerId);
        }

        public OrderSagaData FindBy(string purchaseOrderNumber, Guid partnerId)
        {
            return sessionFactory.GetCurrentSession().CreateCriteria(typeof(OrderSagaData))
                .Add(Expression.Eq("PurchaseOrderNumber", purchaseOrderNumber))
                .Add(Expression.Eq("PartnerId", partnerId))
                .UniqueResult<OrderSagaData>();
        }

        private ISessionFactory sessionFactory;

        public virtual ISessionFactory SessionFactory
        {
            get { return sessionFactory; }
            set { sessionFactory = value; }
        }
    }
}