Facilities/Wcf/Castle.Facilities.WcfIntegration.Tests/Behaviors/CallCountServiceBehavior.cs

   1 // Copyright 2004-2008 Castle Project - http://www.castleproject.org/
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 namespace Castle.Facilities.WcfIntegration.Tests.Behaviors
  16 {
  17         using System.Collections.ObjectModel;
  18         using System.ServiceModel;
  19         using System.ServiceModel.Channels;
  20         using System.ServiceModel.Description;
  21         using System.ServiceModel.Dispatcher;
  22
  23         public class CallCountServiceBehavior : IServiceBehavior
  24         {
  25                 public static int CallCount = 0;
  26
  27                 #region IServiceBehavior Members
  28
  29                 public void Validate(ServiceDescription serviceDescription, ServiceHostBase serviceHostBase)
  30                 {
  31                 }
  32
  33                 public void AddBindingParameters(
  34                         ServiceDescription serviceDescription, ServiceHostBase serviceHostBase, Collection<ServiceEndpoint> endpoints, BindingParameterCollection bindingParameters)
  35                 {
  36                 }
  37
  38                 public void ApplyDispatchBehavior(ServiceDescription serviceDescription, ServiceHostBase serviceHostBase)
  39                 {
  40                         foreach (ChannelDispatcher channelDispatcher in serviceHostBase.ChannelDispatchers)
  41                         {
  42                                 foreach (EndpointDispatcher endPointDispatcher in channelDispatcher.Endpoints)
  43                                 {
  44                                         endPointDispatcher.DispatchRuntime.MessageInspectors.Add(new CallCountMessageInspector());
  45                                 }
  46                         }
  47                 }
  48
  49                 #endregion
  50
  51                 #region Nested type: CallCountMessageInspector
  52
  53                 public class CallCountMessageInspector : IDispatchMessageInspector
  54                 {
  55                         #region IDispatchMessageInspector Members
  56
  57                         public object AfterReceiveRequest(ref Message request, IClientChannel channel, InstanceContext instanceContext)
  58                         {
  59                                 CallCount += 1;
  60                                 return null;
  61                         }
  62
  63                         public void BeforeSendReply(ref Message reply, object correlationState)
  64                         {
  65                         }
  66
  67                         #endregion
  68                 }
  69
  70                 #endregion
  71         }
  72 }