Общая информация между MessageInspector и веб-сервисом

Question

У меня вопрос по поводу обмена информацией между MessageInspector и веб-сервисом.

У меня есть идентификатор (Guid), который я использую для "связывания" AfterReceiveRequest и BeforeSendReply. Работает нормально. Но я бы хотел, чтобы этот идентификатор был доступен и в методах, используемых в веб-сервисе. Это очень полезно, например, для отслеживания проблем в журнале.

Ниже приведен небольшой демонстрационный пример, иллюстрирующий эту идею. Моя проблема в том, чтобы найти решение, где можно получить доступ к возвращаемому объекту из метода AfterReceiveRequest. В соответствии с "MagicStuff".

namespace Demo.MessageInspector
{
    public class DemoMessageInspector : IDispatchMessageInspector
    {
        public object AfterReceiveRequest(ref Message request, IClientChannel channel, InstanceContext instanceContext)
        {
            Guid activityId = Guid.NewGuid();
            MyLog.Message("AfterReceiveRequest", activityId);
            return activityId;
        }

        public void BeforeSendReply(ref Message reply, object correlationState)
        {
            Guid activityId = (Guid)correlationState;
            MyLog.Message("BeforeSendReply", activityId);
        }
    }
}

namespace Demo.WebServices
{
    [ServiceBehavior]
    [MessageInterceptionServiceBehaviour]
    public class MyWebService : IMyWebService
    {
        public void MyWebServiceMethod()
        {
            Guid activityId = (Guid)MagicStuff; // <-- correlationState from AfterReceiveRequest

            bool success = DoSomthing();

            if (!success)
                MyLog.Message("Error happened in MyWebServiceMethod", activityId);
        }
    }
}

Буду признателен за любую помощь или подсказки.

Answer 1

Вы можете использовать свойство OperationContext.IncomingMessageProperties. Вот использование в вашем сценарии:

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.ServiceModel;
using System.ServiceModel.Channels;
using System.ServiceModel.Description;
using System.ServiceModel.Dispatcher;
using Demo.MessageInspector;
using Demo.Utilities;

namespace Demo.WebServices
{
    public class MyWebService : IMyWebService, IServiceBehavior
    {
        public void MyWebServiceMethod()
        {
            // get the activityId from the incoming message properties
            var activityIdProperty = OperationContext.Current.IncomingMessageProperties
                .FirstOrDefault(property => property.Key == Properties.ActivityId.ToString());

            // create an empty Guid
            Guid activityId = new Guid();
            if (activityIdProperty.Value != null)
            {
                // replace the empty Guid with the activityId
                activityId = (Guid)activityIdProperty.Value;
            }
            bool success = DoSomething();
            if (!success)
                MyLog.Message("Error happened in MyWebServiceMethod", activityId);
        }

        private bool DoSomething()
        {
            // TODO: implement
            return false;
        }

        public void AddBindingParameters(
          ServiceDescription serviceDescription,
          ServiceHostBase serviceHostBase,
          System.Collections.ObjectModel.Collection<ServiceEndpoint> endpoints,
          BindingParameterCollection bindingParameters
        )
        {
            return;
        }

        public void ApplyDispatchBehavior(ServiceDescription serviceDescription,
            ServiceHostBase serviceHostBase)
        {
            foreach (ChannelDispatcher chDisp in serviceHostBase.ChannelDispatchers)
            {
                foreach (EndpointDispatcher epDisp in chDisp.Endpoints)
                {
                    var messageInspector = new DemoMessageInspector();
                    epDisp.DispatchRuntime.MessageInspectors.Add(messageInspector);
                }
            }
        }

        public void Validate(ServiceDescription serviceDescription, 
            ServiceHostBase serviceHostBase)
        {
            // TODO: implement validation
            //throw new NotImplementedException();
        }
    }
}

namespace Demo.MessageInspector
{
    public class DemoMessageInspector : IDispatchMessageInspector
    {
        public object AfterReceiveRequest(ref Message request, 
            IClientChannel channel, InstanceContext instanceContext)
        {
            Guid activityId = Guid.NewGuid();

            // add the activityId to the incoming message properties
            OperationContext.Current.IncomingMessageProperties
                .Add(Properties.ActivityId.ToString(), activityId);

            MyLog.Message("AfterReceiveRequest", activityId);
            return activityId;
        }

        public void BeforeSendReply(ref Message reply, object correlationState)
        {
            Guid activityId = (Guid)correlationState;
            MyLog.Message("BeforeSendReply", activityId);
        }
    }
}

namespace Demo.Utilities
{
    public enum Properties
    {
        ActivityId
    }

    public class MyLog
    {
        internal static void Message(string p, Guid guid)
        {
            File.AppendAllText(@"C:\Temp\log.txt", 
                String.Format("{0} {1} {2}\r\n", DateTime.Now, p, guid));
        }
    }
}

[ОБНОВИТЬ]

В качестве альтернативы вы можете использовать свойство CorrelationManager.ActivityId. Для этого сначала добавьте следующее в ваши файлы конфигурации (как клиентские, так и серверные):

<system.diagnostics>
  <sources>
    <source name="System.ServiceModel" propagateActivity="true">
      <listeners>
        <add name="ignored" type="System.Diagnostics.DefaultTraceListener" />
      </listeners>
    </source>
  </sources>
</system.diagnostics>

Затем добавьте следующий метод в Demo.Utilities.Helper учебный класс:

namespace Demo.Utilities
{
    internal class Helper
    {
        internal static Guid GetCorrelationId()
        {
            var headerPosition = OperationContext.Current.IncomingMessageHeaders.FindHeader("ActivityId",
                "http://schemas.microsoft.com/2004/09/ServiceModel/Diagnostics");
            if (headerPosition > -1)
            {
                var activityIdHeader = OperationContext.Current.IncomingMessageHeaders
                    .GetReaderAtHeader(headerPosition);
                var activityIdAttribute = activityIdHeader.GetAttribute("CorrelationId");
                return Guid.Parse(activityIdAttribute);
            }
            else
            {
                return Guid.Empty;
            }
        }
    }
}

Теперь вы можете использовать этот метод в DemoMessageInspector:

public object AfterReceiveRequest(ref Message request,
    IClientChannel channel, InstanceContext instanceContext)
{
    var correlationId = Helper.GetCorrelationId();
    MyLog.Message("AfterReceiveRequest\tCorrelationId", correlationId);
    // ...
}

и вот так в методе вашего сервиса:

public void MyWebServiceMethod()
{
    var correlationId = Helper.GetCorrelationId();
    MyLog.Message("MyWebServiceMethod\tCorrelationId", correlationId);

    // ...
}