WCF在多个客户端之间共享IClientMessageInspector实例

时间:2011-05-27 20:12:23

标签: wcf dependency-injection structuremap wcf-client endpointbehavior

我在通过位于此处的标题为“集中式Cookie管理”标题下列出的方法进行WCF服务调用时管理共享身份验证Cookie:http://megakemp.com/2009/02/06/managing-shared-cookies-in-wcf/

我已经设置了自定义IClientMessageInspectorIEndpointBehaviorBehaviorExtensionElement这些作品。我的端点行为添加了一个消息检查器,如下所示:

public class MyEndpointBehavior : IEndpointBehavior
{
    public void AddBindingParameters(ServiceEndpoint endpoint, System.ServiceModel.Channels.BindingParameterCollection bindingParameters)
    {
    }
    public void ApplyClientBehavior(ServiceEndpoint endpoint, System.ServiceModel.Dispatcher.ClientRuntime clientRuntime)
    {
        // yuck.. Wish I had an instance of MyClientMessageInspector
        // (which has the auth cookie already) so I could just inject that
        // instance here instead of creating a new instance
        clientRuntime.MessageInspectors.Add(new MyClientMessageInspector());
    }
    public void ApplyDispatchBehavior(ServiceEndpoint endpoint, System.ServiceModel.Dispatcher.EndpointDispatcher endpointDispatcher)
    {
    }
    public void Validate(ServiceEndpoint endpoint)
    {
    }
}

这一切都可以完美无缺,但是当您希望在多个客户端上共享Cookie 时,此解决方案会崩溃。由于ApplyDispatchBehavior()方法创建了一个新实例,因此任何其他客户端都不会获取该消息检查器实例,因此也不会获取该身份验证票证。

那么我想尝试创建一个自定义构造函数,我可以像这样注入实例:

MyEndpointBehavior(MyClientMessageInspector msgInspector) { ... }

但是,WCF需要无参数构造函数。通过互联网除草,WCF有钩子允许依赖注入,创建IInstanceProviderIServiceBehavior等。但我不认为这就是我在这里寻找的。

任何人都可以帮助指导我朝正确的方向发展吗?

3 个答案:

答案 0 :(得分:5)

您只需要扩展概念,以便将cookie存储在消息检查器本身之外,以便消息检查器的所有实例共享相同的存储。

穷人的方式就是开始使用静态字段而不是实例字段。显然,如果你有多个线程,你需要在更新字段时提供并发性。如果你将它推广到一个cookie容器概念,然后确保你与所有客户共享同一个容器,你可以从那里获得更好的体验。共享容器可以通过获取客户端通道的ChannelParameterCollection并向其添加属性来完成,然后您的行为在检查消息并从中拉出cookie时查找该属性。这看起来有点像这样:

应用逻辑

// Hold onto a static cookie container
public static CookieContainer MyCookieContainer;

// When instantiating the client add the cookie container to the channel parameters
MyClient client = new MyClient();
client.InnerChannel.GetProperty<ChannelParameterCollection>().Add(MyCookieContainer);

消息检查器逻辑

public void BeforeSendMessage(ref Message, IClientChannel clientChannel)
{
    // Find the cookie container for the current channel
    CookieContainer cookieContainer = clientChannel.GetProperty<ChannelParameterCollection>().Select(p => p as CookieContainer).Where(cc => cc != null).First();

    // ... use the cookie container to set header on outgoing context ...
}

答案 1 :(得分:2)

你是对的,IInstanceProvider在你的情况下无济于事 - 它仅用于提供服务实例。您的行为不需要无参数构造函数。你需要一个config元素的无参数构造函数,这个类可以使用一些依赖注入类(见下文)来创建行为所需的相应检查器类。

namespace ConsoleApplication4
{
    public class MyEndpointBehavior : IEndpointBehavior
    {
        IClientMessageInspector inspector;
        public MyEndpointBehavior(IClientMessageInspector inspector)
        {
            this.inspector = inspector;
        }

        public void AddBindingParameters(ServiceEndpoint endpoint, BindingParameterCollection bindingParameters)
        {
        }

        public void ApplyClientBehavior(ServiceEndpoint endpoint, ClientRuntime clientRuntime)
        {
            clientRuntime.MessageInspectors.Add(this.inspector);
        }

        public void ApplyDispatchBehavior(ServiceEndpoint endpoint, EndpointDispatcher endpointDispatcher)
        {
        }

        public void Validate(ServiceEndpoint endpoint)
        {
        }
    }
    public class MyEndpointBehaviorElement : BehaviorExtensionElement
    {
        public override Type BehaviorType
        {
            get { return typeof(MyEndpointBehavior); }
        }

        protected override object CreateBehavior()
        {
            return new MyEndpointBehavior(ClientInspectorFactory.GetClientInspector());
        }
    }
    public class MyClientInspector : IClientMessageInspector
    {
        public MyClientInspector()
        {
        }

        public void AfterReceiveReply(ref Message reply, object correlationState)
        {
            Console.WriteLine("AfterReceiveReply");
        }

        public object BeforeSendRequest(ref Message request, IClientChannel channel)
        {
            Console.WriteLine("BeforeSendRequest");
            return null;
        }
    }
    public static class ClientInspectorFactory
    {
        static IClientMessageInspector instance;
        public static IClientMessageInspector GetClientInspector()
        {
            if (instance == null)
            {
                instance = new MyClientInspector();
            }

            return instance;
        }
    }
    [ServiceContract]
    public interface ITest
    {
        [OperationContract]
        int Add(int x, int y);
    }
    public class Service : ITest
    {
        public int Add(int x, int y) { return x + y; }
    }
    class Program
    {
        static void Main(string[] args)
        {
            ServiceHost host = new ServiceHost(typeof(Service));
            host.Open();
            Console.WriteLine("Host opened");

            ChannelFactory<ITest> factory = new ChannelFactory<ITest>("client1");
            ITest proxy = factory.CreateChannel();
            Console.WriteLine(proxy.Add(3, 4));
            ((IClientChannel)proxy).Close();
            factory.Close();

            factory = new ChannelFactory<ITest>("client2");
            proxy = factory.CreateChannel();
            Console.WriteLine(proxy.Add(5, 8));
            ((IClientChannel)proxy).Close();
            factory.Close();

            host.Close();
        }
    }
}

答案 2 :(得分:0)

我喜欢@carlosfigueira和@drew提供的答案,但我最终想出了一个稍微不同的方法。我选择配置我的IEndpointBehavior PROGRAMMATICY,vs via config。使事情变得简单得多。我更改了我的端点行为以存储我的客户端消息检查器,如下所示:

public class MyEndpointBehavior : IEndpointBehavior
{
    private MyClientMessageInspector_myClientMessageInspector;

    public MyClientMessageInspector MyClientMessageInspector
    {
        get
        {
            if (_myClientMessageInspector == null)
            {
                _myClientMessageInspector = new MyClientMessageInspector();
            }
            return _myClientMessageInspector;
         }
    }
    public void AddBindingParameters(ServiceEndpoint endpoint, System.ServiceModel.Channels.BindingParameterCollection bindingParameters)
    {
    }
    public void ApplyClientBehavior(ServiceEndpoint endpoint, System.ServiceModel.Dispatcher.ClientRuntime clientRuntime)
    {
        clientRuntime.MessageInspectors.Add(MyClientMessageInspector);
    }
public void ApplyDispatchBehavior(ServiceEndpoint endpoint, System.ServiceModel.Dispatcher.EndpointDispatcher endpointDispatcher)
    {
    }
    public void Validate(ServiceEndpoint endpoint)
    {
    }
}

然后我简单地在客户端之间共享此行为,如下所示:

var behavior = new MyEndpointBehavior();
client1.Endpoint.Behaviors.Add(behavior);
client2.Endpoint.Behaviors.Add(behavior);

现在两个客户端将共享相同的身份验证cookie。