sdk/samples/WCFSamples/TechnologySamples/Extensibility/Instancing/Durable/CS/extensions/DurableInstanceContextAttribute.cs

   1 #region using
   2
   3 using System;
   4 using System.Collections.Generic;
   5 using System.Text;
   6 using System.ServiceModel;
   7 using System.Collections.ObjectModel;
   8 using System.ServiceModel.Description;
   9 using System.ServiceModel.Channels;
  10 using System.ServiceModel.Dispatcher;
  11
  12 #endregion
  13
  14 namespace Microsoft.ServiceModel.Samples
  15 {
  16      //This class contains the implementation of the attribute that
  17      //can be used with the service class to enable
  18      //durable instancing.
  19     [AttributeUsage(AttributeTargets.Class)]
  20     public sealed class DurableInstanceContextAttribute : Attribute, IServiceBehavior
  21     {
  22         #region Private members
  23
  24         private Type storageManagerType;
  25
  26         #endregion
  27
  28         #region Constructors
  29
  30         public DurableInstanceContextAttribute()
  31         {
  32         }
  33
  34         #endregion
  35
  36         #region Properties
  37         public Type StorageManagerType
  38         {
  39             get { return this.storageManagerType; }
  40             set { this.storageManagerType = value; }
  41         }
  42
  43         #endregion
  44
  45         #region IServiceBehavior Members
  46
  47         public void AddBindingParameters(ServiceDescription serviceDescription, ServiceHostBase serviceHostBase,
  48             Collection<ServiceEndpoint> endpoints, BindingParameterCollection bindingParameters)
  49         {
  50         }
  51
  52         public void ApplyDispatchBehavior(ServiceDescription serviceDescription, ServiceHostBase serviceHostBase)
  53         {
  54             // Durable instancing could not be used with
  55             // singleton instancing.
  56             ServiceBehaviorAttribute serviceBehavior =
  57                 serviceDescription.Behaviors.Find<ServiceBehaviorAttribute>();
  58
  59             if (serviceBehavior != null &&
  60                 serviceBehavior.InstanceContextMode == InstanceContextMode.Single)
  61             {
  62                 throw new InvalidOperationException(
  63                     ResourceHelper.GetString("ExSingeltonInstancingNotSupported"));
  64             }
  65
  66             // Use the StorageManagerFactory to create an instance of a
  67             // storage manager.
  68             IStorageManager storageManager =
  69                 StorageManagerFactory.GetStorageManager(storageManagerType);
  70
  71             InstanceContextInitializer contextInitializer =
  72                 new InstanceContextInitializer(storageManager);
  73
  74             InstanceProvider instanceProvider =
  75                 new InstanceProvider(serviceDescription.ServiceType);
  76
  77             foreach (ChannelDispatcherBase cdb in serviceHostBase.ChannelDispatchers)
  78             {
  79                 ChannelDispatcher cd = cdb as ChannelDispatcher;
  80
  81                 if (cd != null)
  82                 {
  83                     foreach (EndpointDispatcher ed in cd.Endpoints)
  84                     {
  85                         ed.DispatchRuntime.InstanceContextInitializers.Add(contextInitializer);
  86                         ed.DispatchRuntime.InstanceProvider = instanceProvider;
  87                     }
  88                 }
  89             }
  90         }
  91
  92         public void Validate(ServiceDescription serviceDescription, ServiceHostBase serviceHostBase)
  93         {
  94         }
  95
  96         #endregion
  97     }
  98 }