class/System.ServiceModel.Routing/System.ServiceModel.Routing/RoutingExtension.cs

   1 using System;
   2 using System.Collections.Generic;
   3 using System.Collections.ObjectModel;
   4 using System.ServiceModel;
   5 using System.ServiceModel.Channels;
   6 using System.ServiceModel.Description;
   7 using System.ServiceModel.Dispatcher;
   8
   9 namespace System.ServiceModel.Routing
  10 {
  11         public sealed class RoutingExtension : IExtension<ServiceHostBase>
  12         {
  13                 class InstanceProvider : IInstanceProvider
  14                 {
  15                         public InstanceProvider (RoutingConfiguration config)
  16                         {
  17                                 this.config = config;
  18                         }
  19
  20                         RoutingConfiguration config;
  21
  22                         public object GetInstance (InstanceContext instanceContext)
  23                         {
  24                                 return new RoutingService () { Configuration = config };
  25                         }
  26
  27                         public object GetInstance (InstanceContext instanceContext, Message message)
  28                         {
  29                                 return GetInstance (instanceContext);
  30                         }
  31
  32                         public void ReleaseInstance (InstanceContext instanceContext, object instance)
  33                         {
  34                         }
  35                 }
  36
  37                 internal RoutingExtension ()
  38                 {
  39                 }
  40
  41                 ServiceHostBase host;
  42                 RoutingConfiguration configuration;
  43
  44                 public void ApplyConfiguration (RoutingConfiguration routingConfiguration)
  45                 {
  46                         if (routingConfiguration == null)
  47                                 throw new ArgumentNullException ("routingConfiguration");
  48                         configuration = routingConfiguration;
  49
  50                         if (host == null)
  51                                 return;
  52
  53                         host.Opened += delegate {
  54                                 foreach (ChannelDispatcher cd in host.ChannelDispatchers)
  55                                         foreach (var ed in cd.Endpoints)
  56                                                 if (ed.ContractNamespace == "http://schemas.microsoft.com/netfx/2009/05/routing")
  57                                                         ed.DispatchRuntime.InstanceProvider = new InstanceProvider (configuration);
  58                                 };
  59                 }
  60
  61                 void IExtension<ServiceHostBase>.Attach (ServiceHostBase owner)
  62                 {
  63                         host = owner;
  64                         if (configuration != null)
  65                                 ApplyConfiguration (configuration);
  66                 }
  67
  68                 void IExtension<ServiceHostBase>.Detach (ServiceHostBase owner)
  69                 {
  70                         host = null;
  71                 }
  72         }
  73 }