Facilities/Wcf/Castle.Facilities.WcfIntegration/Behaviors/WcfExplicitBehavior.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
  16 {
  17         using System;
  18         using System.ServiceModel.Description;
  19         using Castle.MicroKernel;
  20
  21         internal abstract class WcfExplcitBehavior : IWcfServiceBehavior, IWcfEndpointBehavior
  22         {
  23
  24                 #region IWcfServiceBehavior
  25
  26                 public void Install(ServiceDescription description, IKernel kernel)
  27                 {
  28                         object behavior = GetBehaviorInstance(kernel);
  29
  30                         if (behavior is IServiceBehavior)
  31                         {
  32                                 description.Behaviors.Add((IServiceBehavior)behavior);
  33                         }
  34                 }
  35
  36                 #endregion
  37
  38                 #region IWcfEndpointBehavior
  39
  40                 public void Install(ServiceEndpoint endpoint, IKernel kernel)
  41                 {
  42                         object behavior = GetBehaviorInstance(kernel);
  43
  44                         if (behavior is IEndpointBehavior)
  45                         {
  46                                 endpoint.Behaviors.Add((IEndpointBehavior)behavior);
  47                         }
  48                         else if (behavior is IOperationBehavior)
  49                         {
  50                                 foreach (OperationDescription operation in endpoint.Contract.Operations)
  51                                 {
  52                                         operation.Behaviors.Add((IOperationBehavior)behavior);
  53                                 }
  54                         }
  55                 }
  56
  57                 #endregion
  58
  59                 protected abstract object GetBehaviorInstance(IKernel kernel);
  60
  61                 internal static WcfExplcitBehavior CreateFrom(object behavior)
  62                 {
  63                         if (behavior is Type)
  64                         {
  65                                 return new WcfServiceTypeBehavior((Type)behavior);
  66                         }
  67                         else if (behavior is string)
  68                         {
  69                                 return new WcfServiceKeyBehavior((string)behavior);
  70                         }
  71                         else
  72                         {
  73                                 return new WcfInstanceBehavior(behavior);
  74                         }
  75                 }
  76
  77                 public void Accept(IWcfBehaviorVisitor visitor)
  78                 {
  79                         visitor.VisitServiceBehavior(this);
  80                         visitor.VisitEndpointBehavior(this);
  81                 }
  82         }
  83
  84         #region Class: WcfServiceKeyBehavior
  85
  86         internal class WcfServiceKeyBehavior : WcfExplcitBehavior
  87         {
  88                 private readonly string key;
  89
  90                 internal WcfServiceKeyBehavior(string key)
  91                 {
  92                         this.key = key;
  93                 }
  94
  95                 protected override object GetBehaviorInstance(IKernel kernel)
  96                 {
  97                         return kernel[key];
  98                 }
  99         }
 100
 101         #endregion
 102
 103         #region Class: WcfServiceTypeBehavior
 104
 105         internal class WcfServiceTypeBehavior : WcfExplcitBehavior
 106         {
 107                 private readonly Type service;
 108
 109                 internal WcfServiceTypeBehavior(Type service)
 110                 {
 111                         this.service = service;
 112                 }
 113
 114                 protected override object GetBehaviorInstance(IKernel kernel)
 115                 {
 116                         return kernel.Resolve(service);
 117                 }
 118         }
 119
 120         #endregion
 121
 122         #region Class: WcfInstanceBehavior
 123
 124         internal class WcfInstanceBehavior : WcfExplcitBehavior
 125         {
 126                 private readonly object instance;
 127
 128                 internal WcfInstanceBehavior(object instance)
 129                 {
 130                         this.instance = instance;
 131                 }
 132
 133                 protected override object GetBehaviorInstance(IKernel kernel)
 134                 {
 135                         return instance;
 136                 }
 137         }
 138
 139         #endregion
 140 }