//------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//------------------------------------------------------------
namespace System.ServiceModel.Dispatcher
{
    using System.ServiceModel.Channels;
 
    class DurableInstanceProvider : IInstanceProvider
    {
        DurableInstanceContextProvider durableInstanceContextProvider;
 
        public DurableInstanceProvider(DurableInstanceContextProvider instanceContextProvider)
        {
            this.durableInstanceContextProvider = instanceContextProvider;
        }
 
        public object GetInstance(InstanceContext instanceContext)
        {
            return ((IInstanceProvider) this).GetInstance(instanceContext, null);
        }
 
        public object GetInstance(InstanceContext instanceContext, Message message)
        {
            if (instanceContext == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("instanceContext");
            }
 
            return instanceContext.Extensions.Find<DurableInstance>();
        }
 
        public virtual void ReleaseInstance(InstanceContext instanceContext, object instance)
        {
            if (instanceContext == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("instanceContext");
            }
 
            if (instance == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("instance");
            }
 
            DurableInstance durableInstance = (DurableInstance) instance;
 
            if (instanceContext.State == CommunicationState.Faulted || instanceContext.Aborted)
            {
                durableInstance.Abort();
                this.durableInstanceContextProvider.UnbindAbortedInstance(instanceContext, durableInstance.InstanceId);
            }
            else if (instanceContext.State == CommunicationState.Closed)
            {
                durableInstance.Close();
            }
        }
    }
}